| // Copyright 2009 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| #include "go.h" |
| |
| static Type* sw1(Node*, Type*); |
| static Type* sw2(Node*, Type*); |
| static Type* sw3(Node*, Type*); |
| static Node* curfn; |
| |
| enum |
| { |
| Inone, |
| I2T, |
| I2T2, |
| I2I, |
| I2I2, |
| T2I, |
| }; |
| |
| // can this code branch reach the end |
| // without an undcontitional RETURN |
| // this is hard, so it is conservative |
| int |
| walkret(Node *n) |
| { |
| |
| loop: |
| if(n != N) |
| switch(n->op) { |
| case OLIST: |
| if(n->right == N) { |
| n = n->left; |
| goto loop; |
| } |
| n = n->right; |
| goto loop; |
| |
| // at this point, we have the last |
| // statement of the function |
| |
| case OGOTO: |
| case OPANIC: |
| case ORETURN: |
| return 0; |
| } |
| |
| // all other statements |
| // will flow to the end |
| return 1; |
| } |
| |
| void |
| walk(Node *fn) |
| { |
| char s[50]; |
| |
| curfn = fn; |
| if(debug['W']) { |
| snprint(s, sizeof(s), "\nbefore %S", curfn->nname->sym); |
| dump(s, curfn->nbody); |
| } |
| if(curfn->type->outtuple) |
| if(walkret(curfn->nbody)) |
| yyerror("function ends without a return statement"); |
| if(addtop != N) |
| fatal("addtop in walk"); |
| walkstate(curfn->nbody); |
| if(debug['W']) { |
| snprint(s, sizeof(s), "after %S", curfn->nname->sym); |
| dump(s, curfn->nbody); |
| } |
| } |
| |
| void |
| addtotop(Node *n) |
| { |
| Node *l; |
| |
| while(addtop != N) { |
| l = addtop; |
| addtop = N; |
| walktype(l, Etop); |
| n->ninit = list(n->ninit, l); |
| } |
| } |
| |
| void |
| gettype(Node *n, Node *a) |
| { |
| if(debug['W']) |
| dump("\nbefore gettype", n); |
| walktype(n, Erv); |
| if(a == N && addtop != N) |
| fatal("gettype: addtop"); |
| addtotop(a); |
| if(debug['W']) |
| dump("after gettype", n); |
| } |
| |
| void |
| walkstate(Node *n) |
| { |
| Node *more; |
| |
| loop: |
| if(n == N) |
| return; |
| |
| more = N; |
| switch(n->op) { |
| |
| case OLIST: |
| walkstate(n->left); |
| more = n->right; |
| break; |
| |
| default: |
| yyerror("walkstate: %O not a top level statement", n->op); |
| |
| case OASOP: |
| case OAS: |
| case OCALLMETH: |
| case OCALLINTER: |
| case OCALL: |
| case OSEND: |
| case ORECV: |
| case OPRINT: |
| case OPRINTN: |
| case OPANIC: |
| case OPANICN: |
| case OFOR: |
| case OIF: |
| case OSWITCH: |
| case OSELECT: |
| case OEMPTY: |
| case OBREAK: |
| case OCONTINUE: |
| case OGOTO: |
| case OLABEL: |
| case OFALL: |
| case OXCASE: |
| case OCASE: |
| case OXFALL: |
| case ORETURN: |
| case OPROC: |
| walktype(n, Etop); |
| break; |
| } |
| |
| addtotop(n); |
| |
| if(more != N) { |
| n = more; |
| goto loop; |
| } |
| } |
| |
| void |
| indir(Node *nl, Node *nr) |
| { |
| if(nr != N) |
| *nl = *nr; |
| } |
| |
| void |
| walktype(Node *n, int top) |
| { |
| Node *r, *l; |
| Type *t; |
| Sym *s; |
| int et, cl, cr; |
| int32 lno; |
| |
| if(n == N) |
| return; |
| lno = setlineno(n); |
| |
| /* |
| * walk the whole tree of the body of a function. |
| * the types expressions are calculated. |
| * compile-time constants are evaluated. |
| */ |
| |
| loop: |
| if(n == N) |
| goto ret; |
| |
| setlineno(n); |
| |
| if(debug['w'] > 1 && top == Etop && n->op != OLIST) |
| dump("walk-before", n); |
| |
| t = T; |
| et = Txxx; |
| |
| switch(n->op) { |
| default: |
| fatal("walktype: switch 1 unknown op %N", n); |
| goto ret; |
| |
| case OLIST: |
| case OKEY: |
| walktype(n->left, top); |
| n = n->right; |
| goto loop; |
| |
| case OPRINT: |
| if(top != Etop) |
| goto nottop; |
| walktype(n->left, Erv); |
| indir(n, prcompat(n->left, 0)); |
| goto ret; |
| |
| case OPRINTN: |
| if(top != Etop) |
| goto nottop; |
| walktype(n->left, Erv); |
| indir(n, prcompat(n->left, 1)); |
| goto ret; |
| |
| case OPANIC: |
| if(top != Etop) |
| goto nottop; |
| walktype(n->left, Erv); |
| indir(n, list(prcompat(n->left, 0), nodpanic(n->lineno))); |
| goto ret; |
| |
| case OPANICN: |
| if(top != Etop) |
| goto nottop; |
| walktype(n->left, Erv); |
| indir(n, list(prcompat(n->left, 1), nodpanic(n->lineno))); |
| goto ret; |
| |
| case OLITERAL: |
| if(top != Erv) |
| goto nottop; |
| n->addable = 1; |
| goto ret; |
| |
| case ONONAME: |
| s = n->sym; |
| if(s->undef == 0) { |
| s->undef = 1; |
| yyerror("%S: undefined", s); |
| goto ret; |
| } |
| if(top == Etop) |
| goto nottop; |
| goto ret; |
| |
| case ONAME: |
| if(top == Etop) |
| goto nottop; |
| n->addable = 1; |
| if(n->type == T) { |
| s = n->sym; |
| if(s->undef == 0) { |
| yyerror("walktype: %S undeclared", s); |
| s->undef = 1; |
| } |
| } |
| goto ret; |
| |
| case OFOR: |
| if(top != Etop) |
| goto nottop; |
| walkstate(n->ninit); |
| walkbool(n->ntest); |
| walkstate(n->nincr); |
| walkstate(n->nbody); |
| goto ret; |
| |
| case OSWITCH: |
| if(top != Etop) |
| goto nottop; |
| |
| if(!casebody(n->nbody)) |
| yyerror("switch statement must have case labels"); |
| |
| if(n->ntest == N) |
| n->ntest = booltrue; |
| walkstate(n->ninit); |
| walktype(n->ntest, Erv); |
| walkstate(n->nbody); |
| |
| // find common type |
| if(n->ntest->type == T) |
| n->ntest->type = walkswitch(n, sw1); |
| |
| // if that fails pick a type |
| if(n->ntest->type == T) |
| n->ntest->type = walkswitch(n, sw2); |
| |
| // set the type on all literals |
| if(n->ntest->type != T) |
| walkswitch(n, sw3); |
| walktype(n->ntest, Erv); // BOTCH is this right |
| walktype(n->nincr, Erv); |
| goto ret; |
| |
| case OSELECT: |
| if(top != Etop) |
| goto nottop; |
| |
| walkselect(n); |
| goto ret; |
| |
| case OIF: |
| if(top != Etop) |
| goto nottop; |
| walkstate(n->ninit); |
| walkbool(n->ntest); |
| walkstate(n->nbody); |
| walkstate(n->nelse); |
| goto ret; |
| |
| case OPROC: |
| if(top != Etop) |
| goto nottop; |
| walkstate(n->left); |
| goto ret; |
| |
| case OCALLMETH: |
| case OCALLINTER: |
| case OCALL: |
| if(top == Elv) |
| goto nottop; |
| |
| if(n->type != T) |
| goto ret; |
| |
| walktype(n->left, Erv); |
| if(n->left == N) |
| goto ret; |
| t = n->left->type; |
| if(t == T) |
| goto ret; |
| |
| if(n->left->op == ODOTMETH) |
| n->op = OCALLMETH; |
| if(n->left->op == ODOTINTER) |
| n->op = OCALLINTER; |
| |
| if(isptr[t->etype]) |
| t = t->type; |
| |
| if(t->etype != TFUNC) { |
| yyerror("call of a non-function %T", t); |
| goto ret; |
| } |
| |
| dowidth(t); |
| n->type = *getoutarg(t); |
| switch(t->outtuple) { |
| case 0: |
| if(top == Erv) { |
| yyerror("function requires a return type"); |
| n->type = types[TINT]; |
| } |
| break; |
| |
| case 1: |
| n->type = n->type->type->type; |
| break; |
| } |
| |
| walktype(n->right, Erv); |
| |
| switch(n->op) { |
| default: |
| fatal("walk: op: %O", n->op); |
| |
| case OCALLINTER: |
| l = ascompatte(n->op, getinarg(t), &n->right, 0); |
| n->right = reorder1(l); |
| break; |
| |
| case OCALL: |
| l = ascompatte(n->op, getinarg(t), &n->right, 0); |
| n->right = reorder1(l); |
| if(isselect(n)) { |
| // clear output bool - special prob with selectsend |
| r = ascompatte(n->op, getoutarg(t), &boolfalse, 0); |
| n->right = list(n->right, r); |
| } |
| break; |
| |
| case OCALLMETH: |
| l = ascompatte(n->op, getinarg(t), &n->right, 0); |
| r = ascompatte(n->op, getthis(t), &n->left->left, 0); |
| l = list(r, l); |
| n->left->left = N; |
| ullmancalc(n->left); |
| n->right = reorder1(l); |
| break; |
| } |
| goto ret; |
| |
| case OAS: |
| if(top != Etop) |
| goto nottop; |
| |
| addtop = list(addtop, n->ninit); |
| n->ninit = N; |
| |
| l = n->left; |
| r = n->right; |
| walktype(l, Elv); |
| if(l == N || r == N) |
| goto ret; |
| |
| cl = listcount(l); |
| cr = listcount(r); |
| |
| if(cl == cr) { |
| walktype(r, Erv); |
| l = ascompatee(n->op, &n->left, &n->right); |
| if(l != N) |
| indir(n, reorder3(l)); |
| goto ret; |
| } |
| |
| switch(r->op) { |
| |
| case OCALLMETH: |
| case OCALLINTER: |
| case OCALL: |
| if(cr == 1) { |
| // a,b,... = fn() |
| walktype(r, Erv); |
| l = ascompatet(n->op, &n->left, &r->type, 0); |
| if(l != N) |
| indir(n, list(r, reorder2(l))); |
| goto ret; |
| } |
| break; |
| |
| case OINDEX: |
| case OINDEXPTR: |
| if(cl == 2 && cr == 1) { |
| // a,b = map[] - mapaccess2 |
| walktype(r->left, Erv); |
| if(!isptrto(r->left->type, TMAP)) |
| break; |
| l = mapop(n, top); |
| if(l == N) |
| break; |
| indir(n, l); |
| goto ret; |
| } |
| break; |
| |
| case ORECV: |
| if(cl == 2 && cr == 1) { |
| // a,b = <chan - chanrecv2 |
| walktype(r->left, Erv); |
| if(!isptrto(r->left->type, TCHAN)) |
| break; |
| l = chanop(n, top); |
| if(l == N) |
| break; |
| indir(n, l); |
| goto ret; |
| } |
| break; |
| |
| case OCONV: |
| if(cl == 2 && cr == 1) { |
| // a,b = i.(T) |
| walktype(r->left, Erv); |
| if(r->left == N) |
| break; |
| et = isandss(r->type, r->left); |
| switch(et) { |
| case I2T: |
| et = I2T2; |
| break; |
| case I2I: |
| et = I2I2; |
| break; |
| default: |
| et = Inone; |
| break; |
| } |
| if(et == Inone) |
| break; |
| r = ifaceop(r->type, r->left, et); |
| l = ascompatet(n->op, &n->left, &r->type, 0); |
| if(l != N) |
| indir(n, list(r, reorder2(l))); |
| goto ret; |
| } |
| break; |
| } |
| |
| switch(l->op) { |
| case OINDEX: |
| case OINDEXPTR: |
| if(cl == 1 && cr == 2) { |
| // map[] = a,b - mapassign2 |
| if(!isptrto(l->left->type, TMAP)) |
| break; |
| l = mapop(n, top); |
| if(l == N) |
| break; |
| indir(n, l); |
| goto ret; |
| } |
| break; |
| } |
| |
| yyerror("bad shape across assignment - cr=%d cl=%d\n", cr, cl); |
| goto ret; |
| |
| case OBREAK: |
| case OCONTINUE: |
| case OGOTO: |
| case OLABEL: |
| if(top != Etop) |
| goto nottop; |
| goto ret; |
| |
| case OXCASE: |
| if(top != Etop) |
| goto nottop; |
| yyerror("case statement out of place"); |
| n->op = OCASE; |
| |
| case OCASE: |
| if(top != Etop) |
| goto nottop; |
| walktype(n->left, Erv); |
| walkstate(n->right); |
| goto ret; |
| |
| case OXFALL: |
| if(top != Etop) |
| goto nottop; |
| yyerror("fallthrough statement out of place"); |
| n->op = OFALL; |
| |
| case OFALL: |
| case OINDREG: |
| case OEMPTY: |
| goto ret; |
| |
| case OCONV: |
| if(top == Etop) |
| goto nottop; |
| |
| l = n->left; |
| if(l == N) |
| goto ret; |
| |
| walktype(l, Erv); |
| |
| t = n->type; |
| if(t == T) |
| goto ret; |
| |
| if(!iscomposite(t)) |
| convlit(l, t); |
| |
| // nil conversion |
| if(eqtype(t, l->type, 0)) { |
| if(l->op != ONAME) |
| indir(n, l); |
| goto ret; |
| } |
| |
| // simple fix-float |
| if(l->type != T) |
| if(isint[l->type->etype] || isfloat[l->type->etype]) |
| if(isint[t->etype] || isfloat[t->etype]) { |
| evconst(n); |
| goto ret; |
| } |
| |
| // to string |
| if(l->type != T) |
| if(isptrto(t, TSTRING)) { |
| if(isint[l->type->etype]) { |
| indir(n, stringop(n, top)); |
| goto ret; |
| } |
| if(bytearraysz(l->type) != -2) { |
| n->op = OARRAY; |
| indir(n, stringop(n, top)); |
| goto ret; |
| } |
| } |
| |
| // convert dynamic to static generated by ONEW |
| if(isptrarray(t) && isptrdarray(l->type)) |
| goto ret; |
| |
| // structure literal |
| if(t->etype == TSTRUCT) { |
| indir(n, structlit(n)); |
| goto ret; |
| } |
| |
| // array literal |
| if(t->etype == TARRAY) { |
| r = arraylit(n); |
| indir(n, r); |
| goto ret; |
| } |
| |
| // map literal |
| if(t->etype == TMAP) { |
| r = maplit(n); |
| indir(n, r); |
| goto ret; |
| } |
| |
| // interface and structure |
| et = isandss(n->type, l); |
| if(et != Inone) { |
| indir(n, ifaceop(n->type, l, et)); |
| goto ret; |
| } |
| |
| if(l->type != T) |
| yyerror("cannot convert %T to %T", l->type, t); |
| goto ret; |
| |
| case ORETURN: |
| if(top != Etop) |
| goto nottop; |
| walktype(n->left, Erv); |
| if(curfn->type->outnamed && n->left == N) { |
| // print("special return\n"); |
| goto ret; |
| } |
| l = ascompatte(n->op, getoutarg(curfn->type), &n->left, 1); |
| if(l != N) |
| n->left = reorder4(l); |
| goto ret; |
| |
| case ONOT: |
| if(top != Erv) |
| goto nottop; |
| walktype(n->left, Erv); |
| if(n->left == N || n->left->type == T) |
| goto ret; |
| et = n->left->type->etype; |
| break; |
| |
| case OASOP: |
| if(top != Etop) |
| goto nottop; |
| walktype(n->left, Elv); |
| l = n->left; |
| if(l->op != OINDEX) { |
| if(n->etype == OLSH || n->etype == ORSH) |
| goto shft; |
| goto com; |
| } |
| if(!isptrto(l->left->type, TMAP)) |
| goto com; |
| indir(n, mapop(n, top)); |
| goto ret; |
| |
| case OLSH: |
| case ORSH: |
| if(top != Erv) |
| goto nottop; |
| walktype(n->left, Erv); |
| |
| shft: |
| walktype(n->right, Erv); |
| if(n->left == N || n->right == N) |
| goto ret; |
| evconst(n); |
| if(n->op == OLITERAL) |
| goto ret; |
| if(n->left->type == T) |
| convlit(n->left, types[TINT]); |
| if(n->right->type == T) |
| convlit(n->right, types[TUINT]); |
| if(n->left->type == T || n->right->type == T) |
| goto ret; |
| if(issigned[n->right->type->etype]) |
| goto badt; |
| break; |
| |
| case OMOD: |
| case OAND: |
| case OOR: |
| case OXOR: |
| case OANDAND: |
| case OOROR: |
| case OEQ: |
| case ONE: |
| case OLT: |
| case OLE: |
| case OGE: |
| case OGT: |
| case OADD: |
| case OSUB: |
| case OMUL: |
| case ODIV: |
| if(top != Erv) |
| goto nottop; |
| walktype(n->left, Erv); |
| |
| com: |
| walktype(n->right, Erv); |
| if(n->left == N || n->right == N) |
| goto ret; |
| evconst(n); |
| if(n->op == OLITERAL) |
| goto ret; |
| convlit(n->left, n->right->type); |
| convlit(n->right, n->left->type); |
| if(n->left->type == T || n->right->type == T) |
| goto ret; |
| if(!eqtype(n->left->type, n->right->type, 0)) |
| goto badt; |
| |
| switch(n->op) { |
| case OEQ: |
| case ONE: |
| case OLT: |
| case OLE: |
| case OGE: |
| case OGT: |
| case OADD: |
| case OASOP: |
| if(isptrto(n->left->type, TSTRING)) { |
| indir(n, stringop(n, top)); |
| goto ret; |
| } |
| } |
| break; |
| |
| case OMINUS: |
| case OPLUS: |
| case OCOM: |
| if(top != Erv) |
| goto nottop; |
| walktype(n->left, Erv); |
| if(n->left == N) |
| goto ret; |
| evconst(n); |
| if(n->op == OLITERAL) |
| goto ret; |
| break; |
| |
| case OLEN: |
| if(top != Erv) |
| goto nottop; |
| walktype(n->left, Erv); |
| evconst(n); |
| t = n->left->type; |
| if(t != T && isptr[t->etype]) |
| t = t->type; |
| if(t == T) |
| goto ret; |
| switch(t->etype) { |
| default: |
| goto badt; |
| case TSTRING: |
| case TMAP: |
| break; |
| case TARRAY: |
| if(t->bound >= 0) |
| nodconst(n, types[TINT], t->bound); |
| break; |
| } |
| n->type = types[TINT]; |
| goto ret; |
| |
| case OCAP: |
| if(top != Erv) |
| goto nottop; |
| walktype(n->left, Erv); |
| evconst(n); |
| t = n->left->type; |
| if(t != T && isptr[t->etype]) |
| t = t->type; |
| if(t == T) |
| goto ret; |
| switch(t->etype) { |
| default: |
| goto badt; |
| case TARRAY: |
| if(t->bound >= 0) |
| nodconst(n, types[TINT], t->bound); |
| break; |
| } |
| n->type = types[TINT]; |
| goto ret; |
| |
| case OINDEX: |
| case OINDEXPTR: |
| if(top == Etop) |
| goto nottop; |
| |
| walktype(n->left, Erv); |
| walktype(n->right, Erv); |
| |
| if(n->left == N || n->right == N) |
| goto ret; |
| |
| defaultlit(n->left); |
| t = n->left->type; |
| if(t == T) |
| goto ret; |
| |
| // BOTCH - convert each index opcode |
| // to look like this and get rid of OINDEXPTR |
| if(isptr[t->etype]) |
| if(isptrto(t, TSTRING) || isptrto(t->type, TSTRING)) { |
| // right side must be an int |
| if(top != Erv) |
| goto nottop; |
| if(n->right->type == T) { |
| convlit(n->right, types[TINT]); |
| if(n->right->type == T) |
| goto ret; |
| } |
| if(!isint[n->right->type->etype]) |
| goto badt; |
| indir(n, stringop(n, top)); |
| goto ret; |
| } |
| |
| // left side is indirect |
| if(isptr[t->etype]) { |
| t = t->type; |
| n->op = OINDEXPTR; |
| } |
| |
| switch(t->etype) { |
| default: |
| goto badt; |
| |
| case TMAP: |
| // right side must be map type |
| if(n->right->type == T) { |
| convlit(n->right, t->down); |
| if(n->right->type == T) |
| break; |
| } |
| if(!eqtype(n->right->type, t->down, 0)) |
| goto badt; |
| if(n->op != OINDEXPTR) |
| goto badt; |
| n->op = OINDEX; |
| n->type = t->type; |
| if(top == Erv) |
| indir(n, mapop(n, top)); |
| break; |
| |
| case TARRAY: |
| // right side must be an int |
| if(n->right->type == T) { |
| convlit(n->right, types[TINT]); |
| if(n->right->type == T) |
| break; |
| } |
| if(!isint[n->right->type->etype]) |
| goto badt; |
| |
| n->type = t->type; |
| break; |
| } |
| goto ret; |
| |
| case OSEND: |
| if(top == Elv) |
| goto nottop; |
| walktype(n->left, Erv); // chan |
| walktype(n->right, Erv); // e |
| indir(n, chanop(n, top)); |
| goto ret; |
| |
| case ORECV: |
| if(top == Elv) |
| goto nottop; |
| if(n->right == N) { |
| walktype(n->left, Erv); // chan |
| indir(n, chanop(n, top)); // returns e blocking |
| goto ret; |
| } |
| walktype(n->left, Elv); // e |
| walktype(n->right, Erv); // chan |
| indir(n, chanop(n, top)); // returns bool non-blocking |
| goto ret; |
| |
| case OSLICE: |
| if(top == Etop) |
| goto nottop; |
| |
| walktype(n->left, top); |
| walktype(n->right, Erv); |
| if(n->left == N || n->right == N) |
| goto ret; |
| convlit(n->left, types[TSTRING]); |
| t = n->left->type; |
| if(t == T) |
| goto ret; |
| if(isptr[t->etype]) |
| t = t->type; |
| if(t->etype == TSTRING) { |
| indir(n, stringop(n, top)); |
| goto ret; |
| } |
| if(t->etype == TARRAY) { |
| indir(n, arrayop(n, top)); |
| goto ret; |
| } |
| badtype(OSLICE, n->left->type, T); |
| goto ret; |
| |
| case ODOT: |
| case ODOTPTR: |
| case ODOTMETH: |
| case ODOTINTER: |
| if(top == Etop) |
| goto nottop; |
| walkdot(n); |
| goto ret; |
| |
| case OADDR: |
| if(top != Erv) |
| goto nottop; |
| if(n->left->op == OCONV && iscomposite(n->left->type)) { |
| // turn &Point{1, 2} into allocation. |
| // initialize with |
| // nvar := new(Point); |
| // *nvar = Point{1, 2}; |
| // and replace expression with nvar |
| |
| // TODO(rsc): might do a better job (fewer copies) later |
| Node *nnew, *nvar, *nas; |
| |
| walktype(n->left, Elv); |
| if(n->left == N) |
| goto ret; |
| |
| nvar = nod(0, N, N); |
| tempname(nvar, ptrto(n->left->type)); |
| |
| nnew = nod(ONEW, N, N); |
| nnew->type = nvar->type; |
| nnew = newcompat(nnew); |
| |
| nas = nod(OAS, nvar, nnew); |
| addtop = list(addtop, nas); |
| |
| nas = nod(OAS, nod(OIND, nvar, N), n->left); |
| addtop = list(addtop, nas); |
| |
| indir(n, nvar); |
| goto ret; |
| } |
| walktype(n->left, Elv); |
| if(n->left == N) |
| goto ret; |
| t = n->left->type; |
| if(t == T) |
| goto ret; |
| n->type = ptrto(t); |
| goto ret; |
| |
| case OIND: |
| if(top == Etop) |
| goto nottop; |
| if(top == Elv) // even if n is lvalue, n->left is rvalue |
| top = Erv; |
| walktype(n->left, top); |
| if(n->left == N) |
| goto ret; |
| t = n->left->type; |
| if(t == T) |
| goto ret; |
| if(!isptr[t->etype]) |
| goto badt; |
| n->type = t->type; |
| goto ret; |
| |
| case ONEW: |
| if(top != Erv) |
| goto nottop; |
| indir(n, newcompat(n)); |
| goto ret; |
| } |
| |
| /* |
| * ======== second switch ======== |
| */ |
| |
| switch(n->op) { |
| default: |
| fatal("walktype: switch 2 unknown op %N", n); |
| goto ret; |
| |
| case OASOP: |
| break; |
| |
| case ONOT: |
| case OANDAND: |
| case OOROR: |
| if(n->left->type == T) |
| goto ret; |
| et = n->left->type->etype; |
| if(et != TBOOL) |
| goto badt; |
| t = types[TBOOL]; |
| break; |
| |
| case OEQ: |
| case ONE: |
| if(n->left->type == T) |
| goto ret; |
| et = n->left->type->etype; |
| if(!okforeq[et]) |
| goto badt; |
| if(isinter(n->left->type)) { |
| indir(n, ifaceop(T, n, n->op)); |
| goto ret; |
| } |
| t = types[TBOOL]; |
| break; |
| |
| case OLT: |
| case OLE: |
| case OGE: |
| case OGT: |
| if(n->left->type == T) |
| goto ret; |
| et = n->left->type->etype; |
| if(!okforadd[et]) |
| if(!isptrto(n->left->type, TSTRING)) |
| goto badt; |
| t = types[TBOOL]; |
| break; |
| |
| case OADD: |
| case OSUB: |
| case OMUL: |
| case ODIV: |
| case OPLUS: |
| if(n->left->type == T) |
| goto ret; |
| et = n->left->type->etype; |
| if(!okforadd[et]) |
| goto badt; |
| break; |
| |
| case OMINUS: |
| if(n->left->type == T) |
| goto ret; |
| et = n->left->type->etype; |
| if(!okforadd[et]) |
| goto badt; |
| if(!isfloat[et]) |
| break; |
| |
| l = nod(OLITERAL, N, N); |
| l->val.u.fval = mal(sizeof(*l->val.u.fval)); |
| l->val.ctype = CTFLT; |
| mpmovecflt(l->val.u.fval, 0.0); |
| |
| l = nod(OSUB, l, n->left); |
| indir(n, l); |
| walktype(n, Erv); |
| goto ret; |
| |
| case OLSH: |
| case ORSH: |
| case OAND: |
| case OOR: |
| case OXOR: |
| case OMOD: |
| case OCOM: |
| if(n->left->type == T) |
| goto ret; |
| et = n->left->type->etype; |
| if(!okforand[et]) |
| goto badt; |
| break; |
| } |
| |
| if(t == T) |
| t = n->left->type; |
| n->type = t; |
| goto ret; |
| |
| nottop: |
| dump("bad top", n); |
| fatal("walktype: top=%d %O", top, n->op); |
| goto ret; |
| |
| badt: |
| if(n->right == N) { |
| if(n->left == N) { |
| badtype(n->op, T, T); |
| goto ret; |
| } |
| badtype(n->op, n->left->type, T); |
| goto ret; |
| } |
| badtype(n->op, n->left->type, n->right->type); |
| goto ret; |
| |
| ret: |
| if(debug['w'] && top == Etop && n != N) |
| dump("walk", n); |
| |
| ullmancalc(n); |
| lineno = lno; |
| } |
| |
| void |
| walkbool(Node *n) |
| { |
| walktype(n, Erv); |
| addtotop(n); |
| if(n != N && n->type != T) |
| if(!eqtype(n->type, types[TBOOL], 0)) |
| yyerror("IF and FOR require a boolean type"); |
| } |
| |
| /* |
| * return the first type |
| */ |
| Type* |
| sw1(Node *c, Type *place) |
| { |
| if(place == T) |
| return c->type; |
| return place; |
| } |
| |
| /* |
| * return a suitable type |
| */ |
| Type* |
| sw2(Node *c, Type *place) |
| { |
| return types[TINT]; // botch |
| } |
| |
| /* |
| * check that switch type |
| * is compat with all the cases |
| */ |
| Type* |
| sw3(Node *c, Type *place) |
| { |
| if(place == T) |
| return c->type; |
| if(c->type == T) |
| c->type = place; |
| convlit(c, place); |
| if(!ascompat(place, c->type)) |
| badtype(OSWITCH, place, c->type); |
| return place; |
| } |
| |
| Type* |
| walkswitch(Node *sw, Type*(*call)(Node*, Type*)) |
| { |
| Node *n, *c; |
| Type *place; |
| place = call(sw->ntest, T); |
| |
| setlineno(sw); |
| |
| n = sw->nbody; |
| if(n->op == OLIST) |
| n = n->left; |
| if(n->op == OEMPTY) |
| return T; |
| |
| for(; n!=N; n=n->right) { |
| if(n->op != OCASE) |
| fatal("walkswitch: not case %O\n", n->op); |
| for(c=n->left; c!=N; c=c->right) { |
| if(c->op != OLIST) { |
| setlineno(c); |
| place = call(c, place); |
| break; |
| } |
| setlineno(c); |
| place = call(c->left, place); |
| } |
| } |
| return place; |
| } |
| |
| int |
| casebody(Node *n) |
| { |
| Node *oc, *ot, *t; |
| Iter save; |
| |
| /* |
| * look to see if statements at top level have |
| * case labels attached to them. convert the illegal |
| * ops XFALL and XCASE into legal ops FALL and CASE. |
| * all unconverted ops will thus be caught as illegal |
| */ |
| |
| oc = N; // last case statement |
| ot = N; // last statement (look for XFALL) |
| t = listfirst(&save, &n); |
| |
| loop: |
| if(t == N) { |
| /* empty switch */ |
| if(oc == N) |
| return 0; |
| return 1; |
| } |
| if(t->op == OXCASE) { |
| /* rewrite and link top level cases */ |
| t->op = OCASE; |
| if(oc != N) |
| oc->right = t; |
| oc = t; |
| |
| /* rewrite top fall that preceed case */ |
| if(ot != N && ot->op == OXFALL) |
| ot->op = OFALL; |
| } |
| |
| /* if first statement is not case */ |
| if(oc == N) |
| return 0; |
| |
| ot = t; |
| t = listnext(&save); |
| goto loop; |
| } |
| |
| Node* |
| selcase(Node *n, Node *var) |
| { |
| Node *a, *r, *on, *c; |
| Type *t; |
| |
| if(n->left == N) |
| goto dflt; |
| c = n->left; |
| if(c->op == ORECV) |
| goto recv; |
| |
| walktype(c->left, Erv); // chan |
| walktype(c->right, Erv); // elem |
| |
| t = fixchan(c->left->type); |
| if(t == T) |
| return N; |
| |
| convlit(c->right, t->type); |
| if(!ascompat(t->type, c->right->type)) { |
| badtype(c->op, t->type, c->right->type); |
| return N; |
| } |
| |
| // selectsend(sel *byte, hchan *chan any, elem any) (selected bool); |
| on = syslook("selectsend", 1); |
| argtype(on, t->type); |
| argtype(on, t->type); |
| |
| a = c->right; // elem |
| r = a; |
| a = c->left; // chan |
| r = list(a, r); |
| a = var; // sel-var |
| r = list(a, r); |
| |
| goto out; |
| |
| recv: |
| if(c->right != N) |
| goto recv2; |
| |
| walktype(c->left, Erv); // chan |
| |
| t = fixchan(c->left->type); |
| if(t == T) |
| return N; |
| |
| // selectrecv(sel *byte, hchan *chan any, elem *any) (selected bool); |
| on = syslook("selectrecv", 1); |
| argtype(on, t->type); |
| argtype(on, t->type); |
| |
| a = c->left; // nil elem |
| a = nod(OLITERAL, N, N); |
| a->val.ctype = CTNIL; |
| |
| r = a; |
| a = c->left; // chan |
| r = list(a, r); |
| a = var; // sel-var |
| r = list(a, r); |
| goto out; |
| |
| recv2: |
| walktype(c->right, Erv); // chan |
| |
| t = fixchan(c->right->type); |
| if(t == T) |
| return N; |
| |
| walktype(c->left, Elv); // elem |
| convlit(c->left, t->type); |
| if(!ascompat(t->type, c->left->type)) { |
| badtype(c->op, t->type, c->left->type); |
| return N; |
| } |
| |
| // selectrecv(sel *byte, hchan *chan any, elem *any) (selected bool); |
| on = syslook("selectrecv", 1); |
| argtype(on, t->type); |
| argtype(on, t->type); |
| |
| a = c->left; // elem |
| a = nod(OADDR, a, N); |
| r = a; |
| a = c->right; // chan |
| r = list(a, r); |
| a = var; // sel-var |
| r = list(a, r); |
| goto out; |
| |
| dflt: |
| // selectdefault(sel *byte); |
| on = syslook("selectdefault", 0); |
| a = var; |
| r = a; // sel-var |
| goto out; |
| |
| out: |
| a = nod(OCALL, on, r); |
| r = nod(OIF, N, N); |
| r->ntest = a; |
| |
| return r; |
| } |
| |
| Node* |
| selectas(Node *name, Node *expr) |
| { |
| Node *a; |
| Type *t; |
| |
| if(expr == N || expr->op != ORECV) |
| goto bad; |
| t = expr->left->type; |
| if(t == T) |
| goto bad; |
| if(isptr[t->etype]) |
| t = t->type; |
| if(t == T) |
| goto bad; |
| if(t->etype != TCHAN) |
| goto bad; |
| a = old2new(name, t->type); |
| return a; |
| |
| bad: |
| return name; |
| } |
| |
| void |
| walkselect(Node *sel) |
| { |
| Iter iter; |
| Node *n, *oc, *on, *r; |
| Node *var, *bod, *res, *def; |
| int count, op; |
| int32 lno; |
| |
| lno = setlineno(sel); |
| |
| // generate sel-struct |
| var = nod(OXXX, N, N); |
| tempname(var, ptrto(types[TUINT8])); |
| |
| n = listfirst(&iter, &sel->left); |
| if(n == N || n->op != OXCASE) |
| yyerror("first select statement must be a case"); |
| |
| count = 0; // number of cases |
| res = N; // entire select body |
| bod = N; // body of each case |
| oc = N; // last case |
| def = N; // default case |
| |
| for(count=0; n!=N; n=listnext(&iter)) { |
| setlineno(n); |
| |
| switch(n->op) { |
| default: |
| bod = list(bod, n); |
| break; |
| |
| case OXCASE: |
| if(n->left == N) { |
| op = ORECV; // actual value not used |
| if(def != N) |
| yyerror("only one default select allowed"); |
| def = n; |
| } else |
| op = n->left->op; |
| switch(op) { |
| default: |
| yyerror("select cases must be send, recv or default"); |
| break; |
| |
| case OAS: |
| // convert new syntax (a=recv(chan)) to (recv(a,chan)) |
| if(n->left->right == N || n->left->right->op != ORECV) { |
| yyerror("select cases must be send, recv or default"); |
| break; |
| } |
| n->left->right->right = n->left->right->left; |
| n->left->right->left = n->left->left; |
| n->left = n->left->right; |
| |
| case OSEND: |
| case ORECV: |
| if(oc != N) { |
| bod = list(bod, nod(OBREAK, N, N)); |
| oc->nbody = rev(bod); |
| } |
| oc = selcase(n, var); |
| res = list(res, oc); |
| break; |
| |
| |
| } |
| bod = N; |
| count++; |
| break; |
| } |
| } |
| if(oc != N) { |
| bod = list(bod, nod(OBREAK, N, N)); |
| oc->nbody = rev(bod); |
| } |
| setlineno(sel); |
| |
| // selectgo(sel *byte); |
| on = syslook("selectgo", 0); |
| r = nod(OCALL, on, var); // sel-var |
| res = list(res, r); |
| |
| // newselect(size uint32) (sel *byte); |
| on = syslook("newselect", 0); |
| |
| r = nod(OXXX, N, N); |
| nodconst(r, types[TINT], count); // count |
| r = nod(OCALL, on, r); |
| r = nod(OAS, var, r); |
| |
| sel->ninit = r; |
| sel->nbody = rev(res); |
| sel->left = N; |
| |
| walkstate(sel->ninit); |
| walkstate(sel->nbody); |
| |
| //dump("sel", sel); |
| |
| lineno = lno; |
| } |
| |
| Type* |
| lookdot1(Node *n, Type *f) |
| { |
| Type *r; |
| Sym *s; |
| |
| r = T; |
| s = n->sym; |
| |
| for(; f!=T; f=f->down) { |
| if(f->sym == S) |
| continue; |
| if(f->sym != s) |
| continue; |
| if(r != T) { |
| yyerror("ambiguous DOT reference %S", s); |
| break; |
| } |
| r = f; |
| } |
| return r; |
| } |
| |
| int |
| lookdot(Node *n, Type *t) |
| { |
| Type *f1, *f2; |
| |
| f1 = T; |
| if(t->etype == TSTRUCT || t->etype == TINTER) |
| f1 = lookdot1(n->right, t->type); |
| |
| f2 = methtype(n->left->type); |
| if(f2 != T) |
| f2 = lookdot1(n->right, f2->method); |
| |
| if(f1 != T) { |
| if(f2 != T) |
| yyerror("ambiguous DOT reference %S as both field and method", |
| n->right->sym); |
| n->right = f1->nname; // substitute real name |
| n->xoffset = f1->width; |
| n->type = f1->type; |
| if(t->etype == TINTER) |
| n->op = ODOTINTER; |
| return 1; |
| } |
| |
| if(f2 != T) { |
| if(needaddr(n->left->type)) { |
| n->left = nod(OADDR, n->left, N); |
| n->left->type = ptrto(n->left->left->type); |
| } |
| n->right = methodname(n->right, ismethod(n->left->type)); |
| n->xoffset = f2->width; |
| n->type = f2->type; |
| n->op = ODOTMETH; |
| return 1; |
| } |
| |
| return 0; |
| } |
| |
| void |
| walkdot(Node *n) |
| { |
| Type *t; |
| |
| addtop = list(addtop, n->ninit); |
| n->ninit = N; |
| |
| if(n->left == N || n->right == N) |
| return; |
| switch(n->op) { |
| case ODOTINTER: |
| case ODOTMETH: |
| return; // already done |
| } |
| |
| walktype(n->left, Erv); |
| if(n->right->op != ONAME) { |
| yyerror("rhs of . must be a name"); |
| return; |
| } |
| |
| t = n->left->type; |
| if(t == T) |
| return; |
| |
| // as a structure field or pointer to structure field |
| if(isptr[t->etype]) { |
| t = t->type; |
| if(t == T) |
| return; |
| n->op = ODOTPTR; |
| } |
| |
| if(!lookdot(n, t)) |
| yyerror("undefined DOT %S on %T", n->right->sym, n->left->type); |
| } |
| |
| Node* |
| ascompatee(int op, Node **nl, Node **nr) |
| { |
| Node *l, *r, *nn, *a; |
| Iter savel, saver; |
| |
| /* |
| * check assign expression list to |
| * a expression list. called in |
| * expr-list = expr-list |
| */ |
| l = listfirst(&savel, nl); |
| r = listfirst(&saver, nr); |
| nn = N; |
| |
| loop: |
| if(l == N || r == N) { |
| if(l != r) |
| yyerror("error in shape across %O", op); |
| return rev(nn); |
| } |
| |
| convlit(r, l->type); |
| if(!ascompat(l->type, r->type)) { |
| badtype(op, l->type, r->type); |
| return N; |
| } |
| |
| a = nod(OAS, l, r); |
| a = convas(a); |
| nn = list(a, nn); |
| |
| l = listnext(&savel); |
| r = listnext(&saver); |
| goto loop; |
| } |
| |
| Node* |
| ascompatet(int op, Node **nl, Type **nr, int fp) |
| { |
| Node *l, *nn, *a; |
| Type *r; |
| Iter savel, saver; |
| |
| /* |
| * check assign type list to |
| * a expression list. called in |
| * expr-list = func() |
| */ |
| l = listfirst(&savel, nl); |
| r = structfirst(&saver, nr); |
| nn = N; |
| |
| loop: |
| if(l == N || r == T) { |
| if(l != N || r != T) |
| yyerror("error in shape across %O", op); |
| return rev(nn); |
| } |
| |
| if(!ascompat(l->type, r->type)) { |
| badtype(op, l->type, r->type); |
| return N; |
| } |
| |
| a = nod(OAS, l, nodarg(r, fp)); |
| a = convas(a); |
| nn = list(a, nn); |
| |
| l = listnext(&savel); |
| r = structnext(&saver); |
| |
| goto loop; |
| } |
| |
| /* |
| * make a tsig for the structure |
| * carrying the ... arguments |
| */ |
| Type* |
| sigtype(Type *st) |
| { |
| Dcl *x; |
| Sym *s; |
| Type *t; |
| static int sigdddgen; |
| |
| dowidth(st); |
| |
| sigdddgen++; |
| snprint(namebuf, sizeof(namebuf), "dsigddd_%d", sigdddgen); |
| s = lookup(namebuf); |
| t = newtype(s); |
| t = dodcltype(t); |
| updatetype(t, st); |
| |
| // record internal type for signature generation |
| x = mal(sizeof(*x)); |
| x->op = OTYPE; |
| x->dsym = s; |
| x->dtype = s->otype; |
| x->forw = signatlist; |
| x->block = block; |
| signatlist = x; |
| |
| return s->otype; |
| } |
| |
| /* |
| * package all the arguments that |
| * match a ... parameter into an |
| * automatic structure. |
| * then call the ... arg (interface) |
| * with a pointer to the structure |
| */ |
| Node* |
| mkdotargs(Node *r, Node *rr, Iter *saver, Node *nn, Type *l, int fp) |
| { |
| Type *t, *st, *ft; |
| Node *a, *n, *var; |
| Iter saven; |
| |
| n = N; // list of assignments |
| |
| st = typ(TSTRUCT); // generated structure |
| ft = T; // last field |
| while(r != N) { |
| defaultlit(r); |
| |
| // generate the next structure field |
| t = typ(TFIELD); |
| t->type = r->type; |
| if(ft == T) |
| st->type = t; |
| else |
| ft->down = t; |
| ft = t; |
| |
| a = nod(OAS, N, r); |
| n = list(n, a); |
| if(rr != N) { |
| r = rr; |
| rr = N; |
| } else |
| r = listnext(saver); |
| } |
| |
| // make a named type for the struct |
| st = sigtype(st); |
| |
| // now we have the size, make the struct |
| var = nod(OXXX, N, N); |
| tempname(var, st); |
| |
| // assign the fields to the struct |
| n = rev(n); |
| r = listfirst(&saven, &n); |
| t = st->type; |
| while(r != N) { |
| r->left = nod(OXXX, N, N); |
| *r->left = *var; |
| r->left->type = r->right->type; |
| r->left->xoffset += t->width; |
| nn = list(r, nn); |
| r = listnext(&saven); |
| t = t->down; |
| } |
| |
| // last thing is to put assignment |
| // of a pointer to the structure to |
| // the DDD parameter |
| |
| a = nod(OADDR, var, N); |
| a->type = ptrto(st); |
| a = nod(OAS, nodarg(l, fp), a); |
| a = convas(a); |
| |
| nn = list(a, nn); |
| |
| return nn; |
| } |
| |
| Node* |
| ascompatte(int op, Type **nl, Node **nr, int fp) |
| { |
| Type *l, *ll; |
| Node *r, *rr, *nn, *a; |
| Iter savel, saver; |
| |
| /* |
| * check assign expression list to |
| * a type list. called in |
| * return expr-list |
| * func(expr-list) |
| */ |
| l = structfirst(&savel, nl); |
| r = listfirst(&saver, nr); |
| nn = N; |
| |
| loop: |
| if(l != T && isddd(l->type)) { |
| // the ddd parameter must be last |
| ll = structnext(&savel); |
| if(ll != T) |
| yyerror("... must be last argument"); |
| |
| // special case -- |
| // only if we are assigning a single ddd |
| // argument to a ddd parameter then it is |
| // passed thru unencapsulated |
| rr = listnext(&saver); |
| if(r != N && rr == N && isddd(r->type)) { |
| a = nod(OAS, nodarg(l, fp), r); |
| a = convas(a); |
| nn = list(a, nn); |
| return rev(nn); |
| } |
| |
| // normal case -- make a structure of all |
| // remaining arguments and pass a pointer to |
| // it to the ddd parameter (empty interface) |
| nn = mkdotargs(r, rr, &saver, nn, l, fp); |
| |
| return rev(nn); |
| } |
| |
| if(l == T || r == N) { |
| if(l != T || r != N) |
| yyerror("error in shape across %O", op); |
| return rev(nn); |
| } |
| convlit(r, l->type); |
| if(!ascompat(l->type, r->type)) { |
| badtype(op, l->type, r->type); |
| return N; |
| } |
| |
| |
| a = nod(OAS, nodarg(l, fp), r); |
| a = convas(a); |
| nn = list(a, nn); |
| |
| l = structnext(&savel); |
| r = listnext(&saver); |
| |
| goto loop; |
| } |
| |
| /* |
| * can we assign var of type t2 to var of type t1 |
| */ |
| int |
| ascompat(Type *t1, Type *t2) |
| { |
| if(eqtype(t1, t2, 0)) |
| return 1; |
| |
| // if(eqtype(t1, nilptr, 0)) |
| // return 1; |
| // if(eqtype(t2, nilptr, 0)) |
| // return 1; |
| |
| if(isnilinter(t1)) |
| return 1; |
| if(isinter(t1)) { |
| if(isinter(t2)) |
| return 1; |
| if(ismethod(t2)) |
| return 1; |
| } |
| |
| if(isnilinter(t2)) |
| return 1; |
| if(isinter(t2)) |
| if(ismethod(t1)) |
| return 1; |
| |
| if(isptrdarray(t1)) |
| if(isptrarray(t2)) |
| return 1; |
| |
| return 0; |
| } |
| |
| Node* |
| prcompat(Node *n, int fmt) |
| { |
| Node *l, *r; |
| Node *on; |
| Type *t; |
| Iter save; |
| int w, notfirst; |
| |
| r = N; |
| l = listfirst(&save, &n); |
| notfirst = 0; |
| |
| loop: |
| if(l == N) { |
| if(fmt) { |
| on = syslook("printnl", 0); |
| r = list(r, nod(OCALL, on, N)); |
| } |
| walktype(r, Etop); |
| return r; |
| } |
| |
| if(notfirst) { |
| on = syslook("printsp", 0); |
| r = list(r, nod(OCALL, on, N)); |
| } |
| |
| w = whatis(l); |
| switch(w) { |
| default: |
| if(l->type == T) |
| goto out; |
| if(isinter(l->type)) { |
| on = syslook("printinter", 1); |
| argtype(on, l->type); // any-1 |
| break; |
| } |
| if(isptr[l->type->etype]) { |
| on = syslook("printpointer", 1); |
| argtype(on, l->type->type); // any-1 |
| break; |
| } |
| badtype(n->op, l->type, T); |
| l = listnext(&save); |
| goto loop; |
| |
| case Wlitint: |
| case Wtint: |
| on = syslook("printint", 0); |
| break; |
| case Wlitfloat: |
| case Wtfloat: |
| on = syslook("printfloat", 0); |
| break; |
| case Wlitbool: |
| case Wtbool: |
| on = syslook("printbool", 0); |
| break; |
| case Wlitstr: |
| case Wtstr: |
| on = syslook("printstring", 0); |
| break; |
| } |
| |
| t = *getinarg(on->type); |
| if(t != nil) |
| t = t->type; |
| if(t != nil) |
| t = t->type; |
| |
| if(!eqtype(t, l->type, 0)) { |
| l = nod(OCONV, l, N); |
| l->type = t; |
| } |
| |
| r = list(r, nod(OCALL, on, l)); |
| |
| out: |
| notfirst = fmt; |
| l = listnext(&save); |
| goto loop; |
| } |
| |
| Node* |
| nodpanic(int32 lineno) |
| { |
| Node *n, *on; |
| |
| on = syslook("panicl", 0); |
| n = nodintconst(lineno); |
| n = nod(OCALL, on, n); |
| walktype(n, Etop); |
| return n; |
| } |
| |
| Node* |
| newcompat(Node *n) |
| { |
| Node *r, *on; |
| Type *t; |
| |
| t = n->type; |
| if(t == T || !isptr[t->etype] || t->type == T) |
| fatal("newcompat: type should be pointer %lT", t); |
| |
| t = t->type; |
| switch(t->etype) { |
| case TFUNC: |
| yyerror("cannot make new %T", t); |
| break; |
| |
| case TMAP: |
| r = mapop(n, Erv); |
| return r; |
| |
| case TCHAN: |
| r = chanop(n, Erv); |
| return r; |
| |
| case TARRAY: |
| r = arrayop(n, Erv); |
| return r; |
| } |
| |
| if(n->left != N) |
| yyerror("dont know what new(,e) means"); |
| |
| dowidth(t); |
| |
| on = syslook("mal", 1); |
| |
| argtype(on, t); |
| |
| r = nodintconst(t->width); |
| r = nod(OCALL, on, r); |
| walktype(r, Erv); |
| |
| // r = nod(OCONV, r, N); |
| r->type = n->type; |
| |
| return r; |
| } |
| |
| Node* |
| stringop(Node *n, int top) |
| { |
| Node *r, *c, *on; |
| |
| switch(n->op) { |
| default: |
| fatal("stringop: unknown op %O", n->op); |
| |
| case OEQ: |
| case ONE: |
| case OGE: |
| case OGT: |
| case OLE: |
| case OLT: |
| // sys_cmpstring(s1, s2) :: 0 |
| on = syslook("cmpstring", 0); |
| r = list(n->left, n->right); |
| r = nod(OCALL, on, r); |
| c = nodintconst(0); |
| r = nod(n->op, r, c); |
| break; |
| |
| case OADD: |
| // sys_catstring(s1, s2) |
| on = syslook("catstring", 0); |
| r = list(n->left, n->right); |
| r = nod(OCALL, on, r); |
| break; |
| |
| case OASOP: |
| // sys_catstring(s1, s2) |
| switch(n->etype) { |
| default: |
| fatal("stringop: unknown op %O-%O", n->op, n->etype); |
| |
| case OADD: |
| // s1 = sys_catstring(s1, s2) |
| if(n->etype != OADD) |
| fatal("stringop: not cat"); |
| r = list(n->left, n->right); |
| on = syslook("catstring", 0); |
| r = nod(OCALL, on, r); |
| r = nod(OAS, n->left, r); |
| break; |
| } |
| break; |
| |
| case OSLICE: |
| // sys_slicestring(s, lb, hb) |
| r = nod(OCONV, n->right->left, N); |
| r->type = types[TINT]; |
| |
| c = nod(OCONV, n->right->right, N); |
| c->type = types[TINT]; |
| |
| r = list(r, c); |
| r = list(n->left, r); |
| on = syslook("slicestring", 0); |
| r = nod(OCALL, on, r); |
| break; |
| |
| case OINDEX: |
| // sys_indexstring(s, i) |
| c = n->left; |
| if(isptrto(c->type->type, TSTRING)) { |
| // lhs is string or *string |
| c = nod(OIND, c, N); |
| c->type = c->left->type->type; |
| } |
| r = nod(OCONV, n->right, N); |
| r->type = types[TINT]; |
| r = list(c, r); |
| on = syslook("indexstring", 0); |
| r = nod(OCALL, on, r); |
| break; |
| |
| case OCONV: |
| // sys_intstring(v) |
| r = nod(OCONV, n->left, N); |
| r->type = types[TINT64]; |
| on = syslook("intstring", 0); |
| r = nod(OCALL, on, r); |
| break; |
| |
| case OARRAY: |
| // arraystring(*[]byte) string; |
| r = n->left; |
| if(!isptr[r->type->etype]) |
| r = nod(OADDR, r, N); |
| on = syslook("arraystring", 0); |
| r = nod(OCALL, on, r); |
| break; |
| } |
| |
| walktype(r, top); |
| return r; |
| } |
| |
| Type* |
| fixmap(Type *tm) |
| { |
| Type *t; |
| |
| t = tm->type; |
| if(t == T) |
| goto bad; |
| if(t->etype != TMAP) |
| goto bad; |
| if(t->down == T || t->type == T) |
| goto bad; |
| |
| dowidth(t->down); |
| dowidth(t->type); |
| |
| return t; |
| |
| bad: |
| yyerror("not a map: %lT", tm); |
| return T; |
| } |
| |
| Type* |
| fixchan(Type *tm) |
| { |
| Type *t; |
| |
| if(tm == T) |
| goto bad; |
| t = tm->type; |
| if(t == T) |
| goto bad; |
| if(t->etype != TCHAN) |
| goto bad; |
| if(t->type == T) |
| goto bad; |
| |
| dowidth(t->type); |
| |
| return t; |
| |
| bad: |
| yyerror("not a channel: %lT", tm); |
| return T; |
| } |
| |
| Node* |
| mapop(Node *n, int top) |
| { |
| Node *r, *a; |
| Type *t; |
| Node *on; |
| int cl, cr; |
| |
| //dump("mapop", n); |
| |
| r = n; |
| switch(n->op) { |
| default: |
| fatal("mapop: unknown op %O", n->op); |
| |
| case ONEW: |
| if(top != Erv) |
| goto nottop; |
| |
| // newmap(keysize int, valsize int, |
| // keyalg int, valalg int, |
| // hint int) (hmap *map[any-1]any-2); |
| |
| t = fixmap(n->type); |
| if(t == T) |
| break; |
| |
| a = n->left; // hint |
| if(n->left == N) |
| a = nodintconst(0); |
| r = a; |
| a = nodintconst(algtype(t->type)); // val algorithm |
| r = list(a, r); |
| a = nodintconst(algtype(t->down)); // key algorithm |
| r = list(a, r); |
| a = nodintconst(t->type->width); // val width |
| r = list(a, r); |
| a = nodintconst(t->down->width); // key width |
| r = list(a, r); |
| |
| on = syslook("newmap", 1); |
| |
| argtype(on, t->down); // any-1 |
| argtype(on, t->type); // any-2 |
| |
| r = nod(OCALL, on, r); |
| walktype(r, top); |
| r->type = n->type; |
| break; |
| |
| case OINDEX: |
| if(top != Erv) |
| goto nottop; |
| // mapaccess1(hmap *map[any]any, key any) (val any); |
| |
| t = fixmap(n->left->type); |
| if(t == T) |
| break; |
| |
| convlit(n->right, t->down); |
| |
| if(!eqtype(n->right->type, t->down, 0)) { |
| badtype(n->op, n->right->type, t->down); |
| break; |
| } |
| |
| a = n->right; // key |
| // if(!isptr[t->down->etype]) { |
| // a = nod(OADDR, a, N); |
| // a->type = ptrto(t); |
| // } |
| |
| r = a; |
| a = n->left; // map |
| r = list(a, r); |
| |
| on = syslook("mapaccess1", 1); |
| |
| argtype(on, t->down); // any-1 |
| argtype(on, t->type); // any-2 |
| argtype(on, t->down); // any-3 |
| argtype(on, t->type); // any-4 |
| |
| r = nod(OCALL, on, r); |
| walktype(r, Erv); |
| r->type = t->type; |
| break; |
| |
| case OAS: |
| cl = listcount(n->left); |
| cr = listcount(n->right); |
| |
| if(cl == 1 && cr == 2) |
| goto assign2; |
| if(cl == 2 && cr == 1) |
| goto access2; |
| if(cl != 1 || cr != 1) |
| goto shape; |
| |
| // mapassign1(hmap *map[any-1]any-2, key any-3, val any-4); |
| if(n->left->op != OINDEX) |
| goto shape; |
| |
| t = fixmap(n->left->left->type); |
| if(t == T) |
| break; |
| |
| a = n->right; // val |
| r = a; |
| a = n->left->right; // key |
| r = list(a, r); |
| a = n->left->left; // map |
| r = list(a, r); |
| |
| on = syslook("mapassign1", 1); |
| |
| argtype(on, t->down); // any-1 |
| argtype(on, t->type); // any-2 |
| argtype(on, t->down); // any-3 |
| argtype(on, t->type); // any-4 |
| |
| r = nod(OCALL, on, r); |
| walktype(r, Etop); |
| break; |
| |
| assign2: |
| // mapassign2(hmap *map[any]any, key any, val any, pres bool); |
| if(n->left->op != OINDEX) |
| goto shape; |
| |
| t = fixmap(n->left->left->type); |
| if(t == T) |
| break; |
| |
| a = n->right->right; // pres |
| r = a; |
| a = n->right->left; // val |
| r =list(a, r); |
| a = n->left->right; // key |
| r = list(a, r); |
| a = n->left->left; // map |
| r = list(a, r); |
| |
| on = syslook("mapassign2", 1); |
| |
| argtype(on, t->down); // any-1 |
| argtype(on, t->type); // any-2 |
| argtype(on, t->down); // any-3 |
| argtype(on, t->type); // any-4 |
| |
| r = nod(OCALL, on, r); |
| walktype(r, Etop); |
| break; |
| |
| access2: |
| // mapaccess2(hmap *map[any-1]any-2, key any-3) (val-4 any, pres bool); |
| |
| //dump("access2", n); |
| if(n->right->op != OINDEX) |
| goto shape; |
| |
| t = fixmap(n->right->left->type); |
| if(t == T) |
| break; |
| |
| a = n->right->right; // key |
| r = a; |
| a = n->right->left; // map |
| r = list(a, r); |
| |
| on = syslook("mapaccess2", 1); |
| |
| argtype(on, t->down); // any-1 |
| argtype(on, t->type); // any-2 |
| argtype(on, t->down); // any-3 |
| argtype(on, t->type); // any-4 |
| |
| n->right = nod(OCALL, on, r); |
| walktype(n, Etop); |
| r = n; |
| break; |
| |
| case OASOP: |
| // rewrite map[index] op= right |
| // into tmpi := index; map[tmpi] = map[tmpi] op right |
| |
| t = n->left->left->type->type; |
| a = nod(OXXX, N, N); |
| tempname(a, t->down); // tmpi |
| r = nod(OAS, a, n->left->right); // tmpi := index |
| n->left->right = a; // m[tmpi] |
| |
| a = nod(OXXX, N, N); |
| indir(a, n->left); // copy of map[tmpi] |
| a = nod(n->etype, a, n->right); // m[tmpi] op right |
| a = nod(OAS, n->left, a); // map[tmpi] = map[tmpi] op right |
| r = nod(OLIST, r, a); |
| walktype(r, Etop); |
| break; |
| } |
| return r; |
| |
| shape: |
| dump("shape", n); |
| fatal("mapop: cl=%d cr=%d, %O", top, n->op); |
| return N; |
| |
| nottop: |
| dump("bad top", n); |
| fatal("mapop: top=%d %O", top, n->op); |
| return N; |
| } |
| |
| Node* |
| chanop(Node *n, int top) |
| { |
| Node *r, *a; |
| Type *t; |
| Node *on; |
| int cl, cr; |
| |
| //dump("chanop", n); |
| |
| r = n; |
| switch(n->op) { |
| default: |
| fatal("chanop: unknown op %O", n->op); |
| |
| case ONEW: |
| // newchan(elemsize int, elemalg int, |
| // hint int) (hmap *chan[any-1]); |
| |
| t = fixchan(n->type); |
| if(t == T) |
| break; |
| |
| if(n->left != N) { |
| // async buf size |
| a = nod(OCONV, n->left, N); |
| a->type = types[TINT]; |
| } else |
| a = nodintconst(0); |
| |
| r = a; |
| a = nodintconst(algtype(t->type)); // elem algorithm |
| r = list(a, r); |
| a = nodintconst(t->type->width); // elem width |
| r = list(a, r); |
| |
| on = syslook("newchan", 1); |
| argtype(on, t->type); // any-1 |
| |
| r = nod(OCALL, on, r); |
| walktype(r, top); |
| r->type = n->type; |
| break; |
| |
| case OAS: |
| cl = listcount(n->left); |
| cr = listcount(n->right); |
| |
| if(cl != 2 || cr != 1 || n->right->op != ORECV) |
| goto shape; |
| |
| // chanrecv2(hchan *chan any) (elem any, pres bool); |
| t = fixchan(n->right->left->type); |
| if(t == T) |
| break; |
| |
| a = n->right->left; // chan |
| r = a; |
| |
| on = syslook("chanrecv2", 1); |
| |
| argtype(on, t->type); // any-1 |
| argtype(on, t->type); // any-2 |
| r = nod(OCALL, on, r); |
| n->right = r; |
| r = n; |
| walktype(r, Etop); |
| break; |
| |
| case ORECV: |
| if(n->right != N) |
| goto recv2; |
| |
| // chanrecv1(hchan *chan any) (elem any); |
| |
| t = fixchan(n->left->type); |
| if(t == T) |
| break; |
| |
| a = n->left; // chan |
| r = a; |
| |
| on = syslook("chanrecv1", 1); |
| |
| argtype(on, t->type); // any-1 |
| argtype(on, t->type); // any-2 |
| r = nod(OCALL, on, r); |
| walktype(r, Erv); |
| break; |
| |
| recv2: |
| // chanrecv3(hchan *chan any, *elem any) (pres bool); |
| t = fixchan(n->right->type); |
| if(t == T) |
| break; |
| |
| a = n->right; // chan |
| r = a; |
| a = n->left; // elem |
| if(a == N) { |
| a = nod(OLITERAL, N, N); |
| a->val.ctype = CTNIL; |
| } else |
| a = nod(OADDR, a, N); |
| |
| on = syslook("chanrecv3", 1); |
| |
| argtype(on, t->type); // any-1 |
| argtype(on, t->type); // any-2 |
| |
| r = nod(OCALL, on, r); |
| n->right = r; |
| r = n; |
| walktype(r, Etop); |
| break; |
| |
| case OSEND: |
| t = fixchan(n->left->type); |
| if(t == T) |
| break; |
| if(top != Etop) |
| goto send2; |
| |
| // chansend1(hchan *chan any, elem any); |
| t = fixchan(n->left->type); |
| if(t == T) |
| break; |
| |
| a = n->right; // e |
| r = a; |
| a = n->left; // chan |
| r = list(a, r); |
| |
| on = syslook("chansend1", 1); |
| argtype(on, t->type); // any-1 |
| argtype(on, t->type); // any-2 |
| r = nod(OCALL, on, r); |
| walktype(r, Etop); |
| break; |
| |
| send2: |
| // chansend2(hchan *chan any, val any) (pres bool); |
| t = fixchan(n->left->type); |
| if(t == T) |
| break; |
| |
| a = n->right; // e |
| r = a; |
| a = n->left; // chan |
| r = list(a, r); |
| |
| on = syslook("chansend2", 1); |
| argtype(on, t->type); // any-1 |
| argtype(on, t->type); // any-2 |
| r = nod(OCALL, on, r); |
| walktype(r, Etop); |
| break; |
| } |
| return r; |
| |
| shape: |
| fatal("chanop: %O", n->op); |
| return N; |
| } |
| |
| Type* |
| fixarray(Type *tm) |
| { |
| Type *t; |
| |
| t = tm->type; |
| if(t == T) |
| goto bad; |
| if(t->etype != TARRAY) |
| goto bad; |
| if(t->type == T) |
| goto bad; |
| |
| dowidth(t->type); |
| |
| return t; |
| |
| bad: |
| yyerror("not an array: %lT", tm); |
| return T; |
| |
| } |
| |
| Node* |
| arrayop(Node *n, int top) |
| { |
| Node *r, *a; |
| Type *t; |
| Node *on; |
| Iter save; |
| |
| r = n; |
| switch(n->op) { |
| default: |
| fatal("darrayop: unknown op %O", n->op); |
| |
| case ONEW: |
| // newarray(nel int, max int, width int) (ary *[]any) |
| t = fixarray(n->type); |
| |
| a = nodintconst(t->type->width); // width |
| a = nod(OCONV, a, N); |
| a->type = types[TINT]; |
| r = a; |
| |
| a = listfirst(&save, &n->left); // max |
| a = listnext(&save); |
| if(a == N) |
| a = nodintconst(0); |
| a = nod(OCONV, a, N); |
| a->type = types[TINT]; |
| r = list(a, r); |
| |
| a = listfirst(&save, &n->left); // nel |
| if(a == N) { |
| if(t->bound < 0) |
| yyerror("new open array must have size"); |
| a = nodintconst(t->bound); |
| } |
| a = nod(OCONV, a, N); |
| a->type = types[TINT]; |
| r = list(a, r); |
| |
| on = syslook("newarray", 1); |
| argtype(on, t->type); // any-1 |
| r = nod(OCALL, on, r); |
| |
| walktype(r, top); |
| if(t->etype == TARRAY) { |
| // single case when we can convert a dynamic |
| // array pointer to a static array pointer |
| // saves making a sys function to alloc a static |
| r = nod(OCONV, r, N); |
| r->type = ptrto(t); |
| } |
| break; |
| |
| case OAS: |
| // arrays2d(old *any, nel int) (ary *[]any) |
| t = fixarray(n->right->type); |
| |
| a = nodintconst(t->bound); // nel |
| a = nod(OCONV, a, N); |
| a->type = types[TINT]; |
| r = a; |
| |
| a = n->right; // old |
| r = list(a, r); |
| |
| on = syslook("arrays2d", 1); |
| argtype(on, n->right->type->type); // any-1 |
| argtype(on, t->type); // any-2 |
| r = nod(OCALL, on, r); |
| |
| walktype(r, top); |
| n->right = r; |
| return n; |
| |
| case OSLICE: |
| if(isptrarray(n->left->type)) |
| goto slicestatic; |
| |
| // arrayslices(old *[]any, lb int, hb int, width int) (ary *[]any) |
| t = fixarray(n->left->type); |
| |
| a = nodintconst(t->type->width); // width |
| a = nod(OCONV, a, N); |
| a->type = types[TINT]; |
| r = a; |
| |
| a = nod(OCONV, n->right->right, N); // hb |
| a->type = types[TINT]; |
| r = list(a, r); |
| |
| a = nod(OCONV, n->right->left, N); // lb |
| a->type = types[TINT]; |
| r = list(a, r); |
| |
| a = n->left; // old |
| r = list(a, r); |
| |
| on = syslook("arraysliced", 1); |
| argtype(on, t->type); // any-1 |
| argtype(on, t->type); // any-2 |
| r = nod(OCALL, on, r); |
| walktype(r, top); |
| break; |
| |
| slicestatic: |
| // arrayslices(old *any, nel int, lb int, hb int, width int) (ary *[]any) |
| t = fixarray(n->left->type); |
| |
| a = nodintconst(t->type->width); // width |
| a = nod(OCONV, a, N); |
| a->type = types[TINT]; |
| r = a; |
| |
| a = nod(OCONV, n->right->right, N); // hb |
| a->type = types[TINT]; |
| r = list(a, r); |
| |
| a = nod(OCONV, n->right->left, N); // lb |
| a->type = types[TINT]; |
| r = list(a, r); |
| |
| a = nodintconst(t->bound); // nel |
| a = nod(OCONV, a, N); |
| a->type = types[TINT]; |
| r = list(a, r); |
| |
| a = n->left; // old |
| r = list(a, r); |
| |
| on = syslook("arrayslices", 1); |
| argtype(on, t); // any-1 |
| argtype(on, t->type); // any-2 |
| r = nod(OCALL, on, r); |
| walktype(r, top); |
| break; |
| } |
| return r; |
| } |
| |
| int |
| isandss(Type *lt, Node *r) |
| { |
| Type *rt; |
| |
| rt = r->type; |
| if(isinter(lt)) { |
| if(isinter(rt)) { |
| if(isnilinter(lt) && isnilinter(rt)) |
| return Inone; |
| if(!eqtype(rt, lt, 0)) |
| return I2I; |
| return Inone; |
| } |
| if(isnilinter(lt) || ismethod(rt) != T) |
| return T2I; |
| return Inone; |
| } |
| |
| if(isinter(rt)) { |
| if(isnilinter(rt) || ismethod(lt) != T) |
| return I2T; |
| return Inone; |
| } |
| |
| return Inone; |
| } |
| |
| static char* |
| ifacename[] = |
| { |
| [I2T] = "ifaceI2T", |
| [I2T2] = "ifaceI2T2", |
| [I2I] = "ifaceI2I", |
| [I2I2] = "ifaceI2I2", |
| }; |
| |
| Node* |
| ifaceop(Type *tl, Node *n, int op) |
| { |
| Type *tr; |
| Node *r, *a, *on; |
| Sym *s; |
| |
| tr = n->type; |
| |
| switch(op) { |
| default: |
| fatal("ifaceop: unknown op %O\n", op); |
| |
| case T2I: |
| // ifaceT2I(sigi *byte, sigt *byte, elem any) (ret any); |
| |
| a = n; // elem |
| r = a; |
| |
| s = signame(tr); // sigt |
| if(s == S) |
| fatal("ifaceop: signame-1 T2I: %lT", tr); |
| a = s->oname; |
| a = nod(OADDR, a, N); |
| r = list(a, r); |
| |
| s = signame(tl); // sigi |
| if(s == S) { |
| fatal("ifaceop: signame-2 T2I: %lT", tl); |
| } |
| a = s->oname; |
| a = nod(OADDR, a, N); |
| r = list(a, r); |
| |
| on = syslook("ifaceT2I", 1); |
| argtype(on, tr); |
| argtype(on, tl); |
| |
| break; |
| |
| case I2T: |
| case I2T2: |
| case I2I: |
| case I2I2: |
| // iface[IT]2[IT][2](sigt *byte, iface any) (ret any[, ok bool]); |
| |
| a = n; // interface |
| r = a; |
| |
| s = signame(tl); // sigi |
| if(s == S) |
| fatal("ifaceop: signame %d", op); |
| a = s->oname; |
| a = nod(OADDR, a, N); |
| r = list(a, r); |
| |
| on = syslook(ifacename[op], 1); |
| argtype(on, tr); |
| argtype(on, tl); |
| |
| break; |
| |
| case OEQ: |
| case ONE: |
| // ifaceeq(i1 any-1, i2 any-2) (ret bool); |
| a = n->right; // i2 |
| r = a; |
| |
| a = n->left; // i1 |
| r = list(a, r); |
| |
| on = syslook("ifaceeq", 1); |
| argtype(on, n->right->type); |
| argtype(on, n->left->type); |
| |
| r = nod(OCALL, on, r); |
| if(op == ONE) |
| r = nod(ONOT, r, N); |
| |
| walktype(r, Erv); |
| return r; |
| } |
| |
| r = nod(OCALL, on, r); |
| walktype(r, Erv); |
| return r; |
| } |
| |
| Node* |
| convas(Node *n) |
| { |
| Node *l, *r; |
| Type *lt, *rt; |
| int et; |
| |
| if(n->op != OAS) |
| fatal("convas: not OAS %O", n->op); |
| |
| lt = T; |
| rt = T; |
| |
| l = n->left; |
| r = n->right; |
| if(l == N || r == N) |
| goto out; |
| |
| lt = l->type; |
| rt = r->type; |
| if(lt == T || rt == T) |
| goto out; |
| |
| if(n->left->op == OINDEX) |
| if(isptrto(n->left->left->type, TMAP)) { |
| indir(n, mapop(n, Elv)); |
| goto out; |
| } |
| |
| if(n->left->op == OINDEXPTR) |
| if(n->left->left->type->etype == TMAP) { |
| indir(n, mapop(n, Elv)); |
| goto out; |
| } |
| |
| if(n->left->op == OSEND) |
| if(n->left->type != T) { |
| indir(n, chanop(n, Elv)); |
| goto out; |
| } |
| |
| if(eqtype(lt, rt, 0)) |
| goto out; |
| |
| et = isandss(lt, r); |
| if(et != Inone) { |
| n->right = ifaceop(lt, r, et); |
| goto out; |
| } |
| |
| if(isptrdarray(lt) && isptrarray(rt)) { |
| if(!eqtype(lt->type->type, rt->type->type, 0)) |
| goto bad; |
| indir(n, arrayop(n, Etop)); |
| goto out; |
| } |
| |
| if(ascompat(lt, rt)) |
| goto out; |
| |
| bad: |
| badtype(n->op, lt, rt); |
| |
| out: |
| ullmancalc(n); |
| return n; |
| } |
| |
| Node* |
| old2new(Node *n, Type *t) |
| { |
| Node *l; |
| |
| if(n->op != ONAME && n->op != ONONAME) { |
| yyerror("left side of := must be a name"); |
| return n; |
| } |
| l = newname(n->sym); |
| dodclvar(l, t); |
| return l; |
| } |
| |
| Node* |
| colas(Node *nl, Node *nr) |
| { |
| Iter savel, saver; |
| Node *l, *r, *a, *n; |
| Type *t; |
| int cl, cr; |
| |
| /* nl is an expression list. |
| * nr is an expression list. |
| * return a newname-list from |
| * types derived from the rhs. |
| */ |
| n = N; |
| cr = listcount(nr); |
| cl = listcount(nl); |
| if(cl != cr) { |
| if(cr == 1) |
| goto multi; |
| goto badt; |
| } |
| |
| l = listfirst(&savel, &nl); |
| r = listfirst(&saver, &nr); |
| |
| while(l != N) { |
| walktype(r, Erv); |
| defaultlit(r); |
| a = old2new(l, r->type); |
| n = list(n, a); |
| |
| l = listnext(&savel); |
| r = listnext(&saver); |
| } |
| n = rev(n); |
| return n; |
| |
| multi: |
| /* |
| * there is a list on the left |
| * and a mono on the right. |
| * go into the right to get |
| * individual types for the left. |
| */ |
| switch(nr->op) { |
| default: |
| goto badt; |
| |
| case OCALLMETH: |
| case OCALLINTER: |
| case OCALL: |
| walktype(nr->left, Erv); |
| t = nr->left->type; |
| if(t != T && t->etype == tptr) |
| t = t->type; |
| if(t == T || t->etype != TFUNC) |
| goto badt; |
| if(t->outtuple != cl) |
| goto badt; |
| |
| l = listfirst(&savel, &nl); |
| t = structfirst(&saver, getoutarg(t)); |
| while(l != N) { |
| a = old2new(l, t->type); |
| n = list(n, a); |
| l = listnext(&savel); |
| t = structnext(&saver); |
| } |
| break; |
| |
| case OINDEX: |
| case OINDEXPTR: |
| // check if rhs is a map index. |
| // if so, types are bool,maptype |
| if(cl != 2) |
| goto badt; |
| walktype(nr->left, Elv); |
| t = nr->left->type; |
| if(t != T && isptr[t->etype]) |
| t = t->type; |
| if(t == T || t->etype != TMAP) |
| goto badt; |
| |
| a = old2new(nl->left, t->type); |
| n = a; |
| a = old2new(nl->right, types[TBOOL]); |
| n = list(n, a); |
| break; |
| |
| case OCONV: |
| // a,b := i.(T) |
| if(cl != 2) |
| goto badt; |
| walktype(nr->left, Erv); |
| if(!isinter(nr->left->type)) |
| goto badt; |
| // a,b = iface |
| a = old2new(nl->left, nr->type); |
| n = a; |
| a = old2new(nl->right, types[TBOOL]); |
| n = list(n, a); |
| break; |
| |
| case ORECV: |
| if(cl != 2) |
| goto badt; |
| walktype(nr->left, Erv); |
| t = nr->left->type; |
| if(!isptrto(t, TCHAN)) |
| goto badt; |
| a = old2new(nl->left, t->type->type); |
| n = a; |
| a = old2new(nl->right, types[TBOOL]); |
| n = list(n, a); |
| break; |
| } |
| n = rev(n); |
| return n; |
| |
| badt: |
| yyerror("shape error across :="); |
| return nl; |
| } |
| |
| /* |
| * from ascompat[te] |
| * evaluating actual function arguments. |
| * f(a,b) |
| * if there is exactly one function expr, |
| * then it is done first. otherwise must |
| * make temp variables |
| */ |
| Node* |
| reorder1(Node *n) |
| { |
| Iter save; |
| Node *l, *r, *f, *a, *g; |
| int c, t; |
| |
| l = listfirst(&save, &n); |
| c = 0; // function calls |
| t = 0; // total parameters |
| |
| loop1: |
| if(l == N) { |
| if(c == 0 || t == 1) |
| return n; |
| goto pass2; |
| } |
| if(l->op == OLIST) |
| fatal("reorder1 OLIST"); |
| |
| t++; |
| ullmancalc(l); |
| if(l->ullman >= UINF) |
| c++; |
| l = listnext(&save); |
| goto loop1; |
| |
| pass2: |
| l = listfirst(&save, &n); |
| g = N; // fncalls assigned to tempnames |
| f = N; // one fncall assigned to stack |
| r = N; // non fncalls and tempnames assigned to stack |
| |
| loop2: |
| if(l == N) { |
| r = rev(r); |
| g = rev(g); |
| f = list(g, f); |
| r = list(f, r); |
| return r; |
| } |
| ullmancalc(l); |
| if(l->ullman < UINF) { |
| r = list(l, r); |
| goto more; |
| } |
| if(f == N) { |
| f = l; |
| goto more; |
| } |
| |
| // make assignment of fncall to tempname |
| a = nod(OXXX, N, N); |
| tempname(a, l->right->type); |
| a = nod(OAS, a, l->right); |
| g = list(a, g); |
| |
| // put normal arg assignment on list |
| // with fncall replaced by tempname |
| l->right = a->left; |
| r = list(l, r); |
| |
| more: |
| l = listnext(&save); |
| goto loop2; |
| } |
| |
| /* |
| * from ascompat[et] |
| * a,b = f() |
| * return of a multi. |
| * there can be no function calls at all, |
| * or they will over-write the return values. |
| */ |
| Node* |
| reorder2(Node *n) |
| { |
| Iter save; |
| Node *l; |
| int c; |
| |
| l = listfirst(&save, &n); |
| c = 0; |
| |
| loop1: |
| if(l == N) { |
| if(c > 0) |
| yyerror("reorder2: too many funcation calls evaluating parameters"); |
| return n; |
| } |
| if(l->op == OLIST) |
| fatal("reorder2 OLIST"); |
| |
| ullmancalc(l); |
| if(l->ullman >= UINF) |
| c++; |
| l = listnext(&save); |
| goto loop1; |
| } |
| |
| /* |
| * from ascompat[ee] |
| * a,b = c,d |
| * simultaneous assignment. there cannot |
| * be later use of an earlier lvalue. |
| */ |
| int |
| vmatch2(Node *l, Node *r) |
| { |
| |
| loop: |
| /* |
| * isolate all right sides |
| */ |
| if(r == N) |
| return 0; |
| switch(r->op) { |
| case ONAME: |
| // match each right given left |
| if(l == r) |
| return 1; |
| case OLITERAL: |
| return 0; |
| } |
| if(vmatch2(l, r->right)) |
| return 1; |
| r = r->left; |
| goto loop; |
| } |
| |
| int |
| vmatch1(Node *l, Node *r) |
| { |
| |
| loop: |
| /* |
| * isolate all left sides |
| */ |
| if(l == N) |
| return 0; |
| switch(l->op) { |
| case ONAME: |
| // match each left with all rights |
| return vmatch2(l, r); |
| case OLITERAL: |
| return 0; |
| } |
| if(vmatch1(l->right, r)) |
| return 1; |
| l = l->left; |
| goto loop; |
| } |
| |
| Node* |
| reorder3(Node *n) |
| { |
| Iter save1, save2; |
| Node *l1, *l2, *q, *r; |
| int c1, c2; |
| |
| r = N; |
| |
| l1 = listfirst(&save1, &n); |
| c1 = 0; |
| |
| while(l1 != N) { |
| l2 = listfirst(&save2, &n); |
| c2 = 0; |
| while(l2 != N) { |
| if(c2 > c1) { |
| if(vmatch1(l1->left, l2->right)) { |
| q = nod(OXXX, N, N); |
| tempname(q, l1->right->type); |
| q = nod(OAS, l1->left, q); |
| l1->left = q->right; |
| r = list(r, q); |
| break; |
| } |
| } |
| l2 = listnext(&save2); |
| c2++; |
| } |
| l1 = listnext(&save1); |
| c1++; |
| } |
| if(r == N) |
| return n; |
| |
| q = N; |
| l1 = listfirst(&save1, &n); |
| while(l1 != N) { |
| q = list(q, l1); |
| l1 = listnext(&save1); |
| } |
| |
| r = rev(r); |
| l1 = listfirst(&save1, &r); |
| while(l1 != N) { |
| q = list(q, l1); |
| l1 = listnext(&save1); |
| } |
| |
| q = rev(q); |
| //dump("res", q); |
| return q; |
| } |
| |
| Node* |
| reorder4(Node *n) |
| { |
| /* |
| * from ascompat[te] |
| * return c,d |
| * return expression assigned to output |
| * parameters. there may be no problems. |
| */ |
| return n; |
| } |
| |
| Node* |
| structlit(Node *n) |
| { |
| Iter savel, saver; |
| Type *l, *t; |
| Node *var, *r, *a; |
| |
| t = n->type; |
| if(t->etype != TSTRUCT) |
| fatal("structlit: not struct"); |
| |
| var = nod(OXXX, N, N); |
| tempname(var, t); |
| |
| l = structfirst(&savel, &n->type); |
| r = listfirst(&saver, &n->left); |
| if(r != N && r->op == OEMPTY) |
| r = N; |
| |
| loop: |
| if(l == T || r == N) { |
| if(l != T) |
| yyerror("struct literal expect expr of type %T", l); |
| if(r != N) |
| yyerror("struct literal too many expressions"); |
| return var; |
| } |
| |
| // build list of var.field = expr |
| |
| a = nod(ODOT, var, newname(l->sym)); |
| a = nod(OAS, a, r); |
| addtop = list(addtop, a); |
| |
| l = structnext(&savel); |
| r = listnext(&saver); |
| goto loop; |
| } |
| |
| Node* |
| arraylit(Node *n) |
| { |
| Iter saver; |
| Type *t; |
| Node *var, *r, *a; |
| int idx; |
| |
| t = n->type; |
| if(t->etype != TARRAY) |
| fatal("arraylit: not array"); |
| |
| if(t->bound < 0) { |
| // make a shallow copy |
| t = typ(0); |
| *t = *n->type; |
| n->type = t; |
| |
| // make it a closed array |
| r = listfirst(&saver, &n->left); |
| if(r != N && r->op == OEMPTY) |
| r = N; |
| for(idx=0; r!=N; idx++) |
| r = listnext(&saver); |
| t->bound = idx; |
| } |
| |
| var = nod(OXXX, N, N); |
| tempname(var, t); |
| |
| idx = 0; |
| r = listfirst(&saver, &n->left); |
| if(r != N && r->op == OEMPTY) |
| r = N; |
| |
| loop: |
| if(r == N) |
| return var; |
| |
| // build list of var[c] = expr |
| |
| a = nodintconst(idx); |
| a = nod(OINDEX, var, a); |
| a = nod(OAS, a, r); |
| addtop = list(addtop, a); |
| idx++; |
| |
| r = listnext(&saver); |
| goto loop; |
| } |
| |
| Node* |
| maplit(Node *n) |
| { |
| Iter saver; |
| Type *t; |
| Node *var, *r, *a; |
| |
| t = n->type; |
| if(t->etype != TMAP) |
| fatal("maplit: not array"); |
| t = ptrto(t); |
| |
| var = nod(OXXX, N, N); |
| tempname(var, t); |
| |
| a = nod(ONEW, N, N); |
| a->type = t; |
| a = nod(OAS, var, a); |
| addtop = list(addtop, a); |
| |
| r = listfirst(&saver, &n->left); |
| if(r != N && r->op == OEMPTY) |
| r = N; |
| |
| loop: |
| if(r == N) { |
| return var; |
| } |
| |
| if(r->op != OKEY) { |
| yyerror("map literal must have key:value pairs"); |
| return var; |
| } |
| |
| // build list of var[c] = expr |
| |
| a = nod(OINDEX, var, r->left); |
| a = nod(OAS, a, r->right); |
| addtop = list(addtop, a); |
| |
| r = listnext(&saver); |
| goto loop; |
| } |