summaryrefslogtreecommitdiff
path: root/sources/scalac/checkers/CheckOwners.java
blob: 8ebd0b9f8fd2179cf4ed8440cc2df3deaec733f3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
/*     ____ ____  ____ ____  ______                                     *\
**    / __// __ \/ __// __ \/ ____/    SOcos COmpiles Scala             **
**  __\_ \/ /_/ / /__/ /_/ /\_ \       (c) 2002, LAMP/EPFL              **
** /_____/\____/\___/\____/____/                                        **
\*                                                                      */

// $Id$

package scalac.checkers;

import scalac.util.*;
import scalac.ast.*;
import scalac.symtab.*;
import scalac.Global;
import scalac.util.Debug;
import Tree.*;

/**
 * Check that the owner of symbols is set correctly.
 *
 * @author Michel Schinz
 * @version 1.0
 */

public class CheckOwners extends Checker {
    protected Symbol currentOwner;

    public CheckOwners(Global global) {
        super(global);
	currentOwner = global.definitions.ROOT_CLASS;
    }

    protected void traverse(Tree tree, Symbol owner) {
        Symbol prevOwner = currentOwner;
        currentOwner = owner;
        traverse(tree);
        currentOwner = prevOwner;
    }

    protected void traverse(Tree[] array, Symbol owner) {
        Symbol prevOwner = currentOwner;
        currentOwner = owner;
        traverse(array);
        currentOwner = prevOwner;
    }

    protected void traverse(Tree[][] array, Symbol owner) {
        Symbol prevOwner = currentOwner;
        currentOwner = owner;
        traverse(array);
        currentOwner = prevOwner;
    }

    protected void traverse(Template templ, Symbol owner) {
	Symbol prevOwner = currentOwner;
	if (owner.kind == Kinds.CLASS)
	    currentOwner = owner.primaryConstructor();
	traverse(templ.parents);
	currentOwner = owner;

        Symbol templSymbol = templ.symbol();
        Tree[] body = templ.body;
        for (int i = 0; i < body.length; ++i) {
            switch (body[i]) {
            case PackageDef(_,_):
            case ClassDef(_,_,_,_,_,_):
            case ModuleDef(_,_,_,_):
            case DefDef(_,_,_,_,_,_):
            case ValDef(_,_,_,_):
            case TypeDef(_,_,_, _):
                traverse(body[i], owner);
                break;
            default:
                traverse(body[i], templSymbol);
            }
        }

	currentOwner = prevOwner;
    }

    protected void checkOwner(Tree tree, Symbol sym) {
        Symbol owner = sym.owner();
        verify(tree,
               owner == currentOwner,
               "owner",
               "incorrect owner for " + Debug.toString(sym) + ":\n"
               + "  found:    " + Debug.toString(owner) + "\n"
               + "  required: " + Debug.toString(currentOwner));
    }

    public void traverse(Tree tree) {
	switch(tree) {
	case PackageDef(Tree packaged, Template impl):
            check(tree);
            traverse(packaged);
            traverse(impl, packaged.symbol());
            break;

        case ClassDef(int mods,
                      Name name,
                      TypeDef[] tparams,
                      ValDef[][] vparams,
		      Tree tpe,
                      Template impl): {
            check(tree);
            traverse(tparams, tree.symbol().primaryConstructor());
            traverse(vparams, tree.symbol().primaryConstructor());
	    traverse(tpe);
            traverse(impl, tree.symbol());
        } break;

        case ModuleDef(int mods, Name name, Tree tpe, Template impl): {
            check(tree);
            traverse(tpe);
            traverse(impl, tree.symbol().moduleClass());
        } break;

        case DefDef(int mods,
                    Name name,
                    TypeDef[] tparams,
                    ValDef[][] vparams,
                    Tree tpe,
                    Tree rhs): {
            check(tree);
            traverse(tparams, tree.symbol());
            traverse(vparams, tree.symbol());
            traverse(tpe, tree.symbol());
            traverse(rhs, tree.symbol());
        } break;

        case ValDef(int mods, Name name, Tree tpe, Tree rhs): {
            check(tree);
            traverse(tpe);
            traverse(rhs, tree.symbol());
        } break;

        case TypeDef(int mods, Name name, Tree rhs, Tree lobound): {
            check(tree);
            traverse(rhs, tree.symbol());
	    // todo: we should do something about lobound here.
        } break;

	default:
	    super.traverse(tree);
        }
    }

    public void check(Tree tree) {
        if (tree.definesSymbol()) {
            Symbol sym = tree.symbol();
            if (sym != null && sym != Symbol.NONE) {
                checkOwner(tree, sym);
                if (sym.kind == Kinds.CLASS)
                    checkOwner(tree, sym.primaryConstructor());
            }
        }
    }
}