// 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 "runtime.h" #include "type.h" #define MAXALIGN 7 #define NOSELGEN 1 static int32 debug = 0; typedef struct WaitQ WaitQ; typedef struct SudoG SudoG; typedef struct Select Select; typedef struct Scase Scase; struct SudoG { G* g; // g and selgen constitute uint32 selgen; // a weak pointer to g SudoG* link; byte* elem; // data element }; struct WaitQ { SudoG* first; SudoG* last; }; struct Hchan { uint32 qcount; // total data in the q uint32 dataqsiz; // size of the circular q uint16 elemsize; bool closed; uint8 elemalign; Alg* elemalg; // interface for element type uint32 sendx; // send index uint32 recvx; // receive index WaitQ recvq; // list of recv waiters WaitQ sendq; // list of send waiters Lock; }; // Buffer follows Hchan immediately in memory. // chanbuf(c, i) is pointer to the i'th slot in the buffer. #define chanbuf(c, i) ((byte*)((c)+1)+(uintptr)(c)->elemsize*(i)) enum { // Scase.kind CaseRecv, CaseSend, CaseDefault, }; struct Scase { SudoG sg; // must be first member (cast to Scase) Hchan* chan; // chan byte* pc; // return pc uint16 kind; uint16 so; // vararg of selected bool bool* receivedp; // pointer to received bool (recv2) }; struct Select { uint16 tcase; // total count of scase[] uint16 ncase; // currently filled scase[] uint16* pollorder; // case poll order Hchan** lockorder; // channel lock order Scase scase[1]; // one per case (in order of appearance) }; static void dequeueg(WaitQ*); static SudoG* dequeue(WaitQ*); static void enqueue(WaitQ*, SudoG*); static void destroychan(Hchan*); Hchan* runtime·makechan_c(ChanType *t, int64 hint) { Hchan *c; int32 n; Type *elem; elem = t->elem; if(hint < 0 || (int32)hint != hint || (elem->size > 0 && hint > ((uintptr)-1) / elem->size)) runtime·panicstring("makechan: size out of range"); if(elem->alg >= nelem(runtime·algarray)) { runtime·printf("chan(alg=%d)\n", elem->alg); runtime·throw("runtime.makechan: unsupported elem type"); } // calculate rounded size of Hchan n = sizeof(*c); while(n & MAXALIGN) n++; // allocate memory in one call c = (Hchan*)runtime·mal(n + hint*elem->size); c->elemsize = elem->size; c->elemalg = &runtime·algarray[elem->alg]; c->elemalign = elem->align; c->dataqsiz = hint; if(debug) runtime·printf("makechan: chan=%p; elemsize=%D; elemalg=%d; elemalign=%d; dataqsiz=%d\n", c, (int64)elem->size, elem->alg, elem->align, c->dataqsiz); return c; } // For reflect // func makechan(typ *ChanType, size uint32) (chan) void reflect·makechan(ChanType *t, uint32 size, Hchan *c) { c = runtime·makechan_c(t, size); FLUSH(&c); } // makechan(t *ChanType, hint int64) (hchan *chan any); void runtime·makechan(ChanType *t, int64 hint, Hchan *ret) { ret = runtime·makechan_c(t, hint); FLUSH(&ret); } /* * generic single channel send/recv * if the bool pointer is nil, * then the full exchange will * occur. if pres is not nil, * then the protocol will not * sleep but return if it could * not complete. * * sleep can wake up with g->param == nil * when a channel involved in the sleep has * been closed. it is easiest to loop and re-run * the operation; we'll see that it's now closed. */ void runtime·chansend(ChanType *t, Hchan *c, byte *ep, bool *pres) { SudoG *sg; SudoG mysg; G* gp; if(c == nil) { USED(t); if(pres != nil) { *pres = false; return; } g->status = Gwaiting; g->waitreason = "chan send (nil chan)"; runtime·gosched(); return; // not reached } if(runtime·gcwaiting) runtime·gosched(); if(debug) { runtime·printf("chansend: chan=%p; elem=", c); c->elemalg->print(c->elemsize, ep); runtime·prints("\n"); } runtime·lock(c); if(c->closed) goto closed; if(c->dataqsiz > 0) goto asynch; sg = dequeue(&c->recvq); if(sg != nil) { runtime·unlock(c); gp = sg->g; gp->param = sg; if(sg->elem != nil) c->elemalg->copy(c->elemsize, sg->elem, ep); runtime·ready(gp); if(pres != nil) *pres = true; return; } if(pres != nil) { runtime·unlock(c); *pres = false; return; } mysg.elem = ep; mysg.g = g; mysg.selgen = NOSELGEN; g->param = nil; g->status = Gwaiting; g->waitreason = "chan send"; enqueue(&c->sendq, &mysg); runtime·unlock(c); runtime·gosched(); if(g->param == nil) { runtime·lock(c); if(!c->closed) runtime·throw("chansend: spurious wakeup"); goto closed; } return; asynch: if(c->closed) goto closed; if(c->qcount >= c->dataqsiz) { if(pres != nil) { runtime·unlock(c); *pres = false; return; } mysg.g = g; mysg.elem = nil; mysg.selgen = NOSELGEN; g->status = Gwaiting; g->waitreason = "chan send"; enqueue(&c->sendq, &mysg); runtime·unlock(c); runtime·gosched(); runtime·lock(c); goto asynch; } c->elemalg->copy(c->elemsize, chanbuf(c, c->sendx), ep); if(++c->sendx == c->dataqsiz) c->sendx = 0; c->qcount++; sg = dequeue(&c->recvq); if(sg != nil) { gp = sg->g; runtime·unlock(c); runtime·ready(gp); } else runtime·unlock(c); if(pres != nil) *pres = true; return; closed: runtime·unlock(c); runtime·panicstring("send on closed channel"); } void runtime·chanrecv(ChanType *t, Hchan* c, byte *ep, bool *selected, bool *received) { SudoG *sg; SudoG mysg; G *gp; if(runtime·gcwaiting) runtime·gosched(); if(debug) runtime·printf("chanrecv: chan=%p\n", c); if(c == nil) { USED(t); if(selected != nil) { *selected = false; return; } g->status = Gwaiting; g->waitreason = "chan receive (nil chan)"; runtime·gosched(); return; // not reached } runtime·lock(c); if(c->dataqsiz > 0) goto asynch; if(c->closed) goto closed; sg = dequeue(&c->sendq); if(sg != nil) { runtime·unlock(c); if(ep != nil) c->elemalg->copy(c->elemsize, ep, sg->elem); gp = sg->g; gp->param = sg; runtime·ready(gp); if(selected != nil) *selected = true; if(received != nil) *received = true; return; } if(selected != nil) { runtime·unlock(c); *selected = false; return; } mysg.elem = ep; mysg.g = g; mysg.selgen = NOSELGEN; g->param = nil; g->status = Gwaiting; g->waitreason = "chan receive"; enqueue(&c->recvq, &mysg); runtime·unlock(c); runtime·gosched(); if(g->param == nil) { runtime·lock(c); if(!c->closed) runtime·throw("chanrecv: spurious wakeup"); goto closed; } if(received != nil) *received = true; return; asynch: if(c->qcount <= 0) { if(c->closed) goto closed; if(selected != nil) { runtime·unlock(c); *selected = false; if(received != nil) *received = false; return; } mysg.g = g; mysg.elem = nil; mysg.selgen = NOSELGEN; g->status = Gwaiting; g->waitreason = "chan receive"; enqueue(&c->recvq, &mysg); runtime·unlock(c); runtime·gosched(); runtime·lock(c); goto asynch; } if(ep != nil) c->elemalg->copy(c->elemsize, ep, chanbuf(c, c->recvx)); c->elemalg->copy(c->elemsize, chanbuf(c, c->recvx), nil); if(++c->recvx == c->dataqsiz) c->recvx = 0; c->qcount--; sg = dequeue(&c->sendq); if(sg != nil) { gp = sg->g; runtime·unlock(c); runtime·ready(gp); } else runtime·unlock(c); if(selected != nil) *selected = true; if(received != nil) *received = true; return; closed: if(ep != nil) c->elemalg->copy(c->elemsize, ep, nil); if(selected != nil) *selected = true; if(received != nil) *received = false; runtime·unlock(c); } // chansend1(hchan *chan any, elem any); #pragma textflag 7 void runtime·chansend1(ChanType *t, Hchan* c, ...) { runtime·chansend(t, c, (byte*)(&c+1), nil); } // chanrecv1(hchan *chan any) (elem any); #pragma textflag 7 void runtime·chanrecv1(ChanType *t, Hchan* c, ...) { runtime·chanrecv(t, c, (byte*)(&c+1), nil, nil); } // chanrecv2(hchan *chan any) (elem any, received bool); #pragma textflag 7 void runtime·chanrecv2(ChanType *t, Hchan* c, ...) { byte *ae, *ap; ae = (byte*)(&c+1); ap = ae + t->elem->size; runtime·chanrecv(t, c, ae, nil, ap); } // func selectnbsend(c chan any, elem any) bool // // compiler implements // // select { // case c <- v: // ... foo // default: // ... bar // } // // as // // if selectnbsend(c, v) { // ... foo // } else { // ... bar // } // #pragma textflag 7 void runtime·selectnbsend(ChanType *t, Hchan *c, ...) { byte *ae, *ap; ae = (byte*)(&c + 1); ap = ae + runtime·rnd(t->elem->size, Structrnd); runtime·chansend(t, c, ae, ap); } // func selectnbrecv(elem *any, c chan any) bool // // compiler implements // // select { // case v = <-c: // ... foo // default: // ... bar // } // // as // // if selectnbrecv(&v, c) { // ... foo // } else { // ... bar // } // #pragma textflag 7 void runtime·selectnbrecv(ChanType *t, byte *v, Hchan *c, bool selected) { runtime·chanrecv(t, c, v, &selected, nil); } // func selectnbrecv2(elem *any, ok *bool, c chan any) bool // // compiler implements // // select { // case v, ok = <-c: // ... foo // default: // ... bar // } // // as // // if c != nil && selectnbrecv2(&v, &ok, c) { // ... foo // } else { // ... bar // } // #pragma textflag 7 void runtime·selectnbrecv2(ChanType *t, byte *v, bool *received, Hchan *c, bool selected) { runtime·chanrecv(t, c, v, &selected, received); } // For reflect: // func chansend(c chan, val iword, nb bool) (selected bool) // where an iword is the same word an interface value would use: // the actual data if it fits, or else a pointer to the data. // // The "uintptr selected" is really "bool selected" but saying // uintptr gets us the right alignment for the output parameter block. void reflect·chansend(ChanType *t, Hchan *c, uintptr val, bool nb, uintptr selected) { bool *sp; byte *vp; if(nb) { selected = false; sp = (bool*)&selected; } else { *(bool*)&selected = true; FLUSH(&selected); sp = nil; } if(t->elem->size <= sizeof(val)) vp = (byte*)&val; else vp = (byte*)val; runtime·chansend(t, c, vp, sp); } // For reflect: // func chanrecv(c chan, nb bool) (val iword, selected, received bool) // where an iword is the same word an interface value would use: // the actual data if it fits, or else a pointer to the data. void reflect·chanrecv(ChanType *t, Hchan *c, bool nb, uintptr val, bool selected, bool received) { byte *vp; bool *sp; if(nb) { selected = false; sp = &selected; } else { selected = true; FLUSH(&selected); sp = nil; } received = false; FLUSH(&received); if(t->elem->size <= sizeof(val)) { val = 0; vp = (byte*)&val; } else { vp = runtime·mal(t->elem->size); val = (uintptr)vp; FLUSH(&val); } runtime·chanrecv(t, c, vp, sp, &received); } static void newselect(int32, Select**); // newselect(size uint32) (sel *byte); #pragma textflag 7 void runtime·newselect(int32 size, ...) { int32 o; Select **selp; o = runtime·rnd(sizeof(size), Structrnd); selp = (Select**)((byte*)&size + o); newselect(size, selp); } static void newselect(int32 size, Select **selp) { int32 n; Select *sel; n = 0; if(size > 1) n = size-1; sel = runtime·mal(sizeof(*sel) + n*sizeof(sel->scase[0]) + size*sizeof(sel->lockorder[0]) + size*sizeof(sel->pollorder[0])); sel->tcase = size; sel->ncase = 0; sel->pollorder = (void*)(sel->scase + size); sel->lockorder = (void*)(sel->pollorder + size); *selp = sel; if(debug) runtime·printf("newselect s=%p size=%d\n", sel, size); } // cut in half to give stack a chance to split static void selectsend(Select *sel, Hchan *c, void *pc, void *elem, int32 so); // selectsend(sel *byte, hchan *chan any, elem *any) (selected bool); #pragma textflag 7 void runtime·selectsend(Select *sel, Hchan *c, void *elem, bool selected) { selected = false; FLUSH(&selected); // nil cases do not compete if(c == nil) return; selectsend(sel, c, runtime·getcallerpc(&sel), elem, (byte*)&selected - (byte*)&sel); } static void selectsend(Select *sel, Hchan *c, void *pc, void *elem, int32 so) { int32 i; Scase *cas; i = sel->ncase; if(i >= sel->tcase) runtime·throw("selectsend: too many cases"); sel->ncase = i+1; cas = &sel->scase[i]; cas->pc = pc; cas->chan = c; cas->so = so; cas->kind = CaseSend; cas->sg.elem = elem; if(debug) runtime·printf("selectsend s=%p pc=%p chan=%p so=%d\n", sel, cas->pc, cas->chan, cas->so); } // cut in half to give stack a chance to split static void selectrecv(Select *sel, Hchan *c, void *pc, void *elem, bool*, int32 so); // selectrecv(sel *byte, hchan *chan any, elem *any) (selected bool); #pragma textflag 7 void runtime·selectrecv(Select *sel, Hchan *c, void *elem, bool selected) { selected = false; FLUSH(&selected); // nil cases do not compete if(c == nil) return; selectrecv(sel, c, runtime·getcallerpc(&sel), elem, nil, (byte*)&selected - (byte*)&sel); } // selectrecv2(sel *byte, hchan *chan any, elem *any, received *bool) (selected bool); #pragma textflag 7 void runtime·selectrecv2(Select *sel, Hchan *c, void *elem, bool *received, bool selected) { selected = false; FLUSH(&selected); // nil cases do not compete if(c == nil) return; selectrecv(sel, c, runtime·getcallerpc(&sel), elem, received, (byte*)&selected - (byte*)&sel); } static void selectrecv(Select *sel, Hchan *c, void *pc, void *elem, bool *received, int32 so) { int32 i; Scase *cas; i = sel->ncase; if(i >= sel->tcase) runtime·throw("selectrecv: too many cases"); sel->ncase = i+1; cas = &sel->scase[i]; cas->pc = pc; cas->chan = c; cas->so = so; cas->kind = CaseRecv; cas->sg.elem = elem; cas->receivedp = received; if(debug) runtime·printf("selectrecv s=%p pc=%p chan=%p so=%d\n", sel, cas->pc, cas->chan, cas->so); } // cut in half to give stack a chance to split static void selectdefault(Select*, void*, int32); // selectdefault(sel *byte) (selected bool); #pragma textflag 7 void runtime·selectdefault(Select *sel, bool selected) { selected = false; FLUSH(&selected); selectdefault(sel, runtime·getcallerpc(&sel), (byte*)&selected - (byte*)&sel); } static void selectdefault(Select *sel, void *callerpc, int32 so) { int32 i; Scase *cas; i = sel->ncase; if(i >= sel->tcase) runtime·throw("selectdefault: too many cases"); sel->ncase = i+1; cas = &sel->scase[i]; cas->pc = callerpc; cas->chan = nil; cas->so = so; cas->kind = CaseDefault; if(debug) runtime·printf("selectdefault s=%p pc=%p so=%d\n", sel, cas->pc, cas->so); } static void sellock(Select *sel) { uint32 i; Hchan *c, *c0; c = nil; for(i=0; incase; i++) { c0 = sel->lockorder[i]; if(c0 && c0 != c) { c = sel->lockorder[i]; runtime·lock(c); } } } static void selunlock(Select *sel) { uint32 i; Hchan *c, *c0; c = nil; for(i=sel->ncase; i-->0;) { c0 = sel->lockorder[i]; if(c0 && c0 != c) { c = c0; runtime·unlock(c); } } } void runtime·block(void) { g->status = Gwaiting; // forever g->waitreason = "select (no cases)"; runtime·gosched(); } static void* selectgo(Select**); // selectgo(sel *byte); // // overwrites return pc on stack to signal which case of the select // to run, so cannot appear at the top of a split stack. #pragma textflag 7 void runtime·selectgo(Select *sel) { runtime·setcallerpc(&sel, selectgo(&sel)); } static void* selectgo(Select **selp) { Select *sel; uint32 o, i, j; Scase *cas, *dfl; Hchan *c; SudoG *sg; G *gp; byte *as; void *pc; sel = *selp; if(runtime·gcwaiting) runtime·gosched(); if(debug) runtime·printf("select: sel=%p\n", sel); // The compiler rewrites selects that statically have // only 0 or 1 cases plus default into simpler constructs. // The only way we can end up with such small sel->ncase // values here is for a larger select in which most channels // have been nilled out. The general code handles those // cases correctly, and they are rare enough not to bother // optimizing (and needing to test). // generate permuted order for(i=0; incase; i++) sel->pollorder[i] = i; for(i=1; incase; i++) { o = sel->pollorder[i]; j = runtime·fastrand1()%(i+1); sel->pollorder[i] = sel->pollorder[j]; sel->pollorder[j] = o; } // sort the cases by Hchan address to get the locking order. for(i=0; incase; i++) { c = sel->scase[i].chan; for(j=i; j>0 && sel->lockorder[j-1] >= c; j--) sel->lockorder[j] = sel->lockorder[j-1]; sel->lockorder[j] = c; } sellock(sel); loop: // pass 1 - look for something already waiting dfl = nil; for(i=0; incase; i++) { o = sel->pollorder[i]; cas = &sel->scase[o]; c = cas->chan; switch(cas->kind) { case CaseRecv: if(c->dataqsiz > 0) { if(c->qcount > 0) goto asyncrecv; } else { sg = dequeue(&c->sendq); if(sg != nil) goto syncrecv; } if(c->closed) goto rclose; break; case CaseSend: if(c->closed) goto sclose; if(c->dataqsiz > 0) { if(c->qcount < c->dataqsiz) goto asyncsend; } else { sg = dequeue(&c->recvq); if(sg != nil) goto syncsend; } break; case CaseDefault: dfl = cas; break; } } if(dfl != nil) { selunlock(sel); cas = dfl; goto retc; } // pass 2 - enqueue on all chans for(i=0; incase; i++) { o = sel->pollorder[i]; cas = &sel->scase[o]; c = cas->chan; sg = &cas->sg; sg->g = g; sg->selgen = g->selgen; switch(cas->kind) { case CaseRecv: enqueue(&c->recvq, sg); break; case CaseSend: enqueue(&c->sendq, sg); break; } } g->param = nil; g->status = Gwaiting; g->waitreason = "select"; selunlock(sel); runtime·gosched(); sellock(sel); sg = g->param; // pass 3 - dequeue from unsuccessful chans // otherwise they stack up on quiet channels for(i=0; incase; i++) { cas = &sel->scase[i]; if(cas != (Scase*)sg) { c = cas->chan; if(cas->kind == CaseSend) dequeueg(&c->sendq); else dequeueg(&c->recvq); } } if(sg == nil) goto loop; cas = (Scase*)sg; c = cas->chan; if(c->dataqsiz > 0) runtime·throw("selectgo: shouldnt happen"); if(debug) runtime·printf("wait-return: sel=%p c=%p cas=%p kind=%d\n", sel, c, cas, cas->kind); if(cas->kind == CaseRecv) { if(cas->receivedp != nil) *cas->receivedp = true; } selunlock(sel); goto retc; asyncrecv: // can receive from buffer if(cas->receivedp != nil) *cas->receivedp = true; if(cas->sg.elem != nil) c->elemalg->copy(c->elemsize, cas->sg.elem, chanbuf(c, c->recvx)); c->elemalg->copy(c->elemsize, chanbuf(c, c->recvx), nil); if(++c->recvx == c->dataqsiz) c->recvx = 0; c->qcount--; sg = dequeue(&c->sendq); if(sg != nil) { gp = sg->g; selunlock(sel); runtime·ready(gp); } else { selunlock(sel); } goto retc; asyncsend: // can send to buffer c->elemalg->copy(c->elemsize, chanbuf(c, c->sendx), cas->sg.elem); if(++c->sendx == c->dataqsiz) c->sendx = 0; c->qcount++; sg = dequeue(&c->recvq); if(sg != nil) { gp = sg->g; selunlock(sel); runtime·ready(gp); } else { selunlock(sel); } goto retc; syncrecv: // can receive from sleeping sender (sg) selunlock(sel); if(debug) runtime·printf("syncrecv: sel=%p c=%p o=%d\n", sel, c, o); if(cas->receivedp != nil) *cas->receivedp = true; if(cas->sg.elem != nil) c->elemalg->copy(c->elemsize, cas->sg.elem, sg->elem); gp = sg->g; gp->param = sg; runtime·ready(gp); goto retc; rclose: // read at end of closed channel selunlock(sel); if(cas->receivedp != nil) *cas->receivedp = false; if(cas->sg.elem != nil) c->elemalg->copy(c->elemsize, cas->sg.elem, nil); goto retc; syncsend: // can send to sleeping receiver (sg) selunlock(sel); if(debug) runtime·printf("syncsend: sel=%p c=%p o=%d\n", sel, c, o); if(sg->elem != nil) c->elemalg->copy(c->elemsize, sg->elem, cas->sg.elem); gp = sg->g; gp->param = sg; runtime·ready(gp); retc: // return to pc corresponding to chosen case pc = cas->pc; as = (byte*)selp + cas->so; runtime·free(sel); *as = true; return pc; sclose: // send on closed channel selunlock(sel); runtime·panicstring("send on closed channel"); return nil; // not reached } // closechan(sel *byte); void runtime·closechan(Hchan *c) { SudoG *sg; G* gp; if(c == nil) runtime·panicstring("close of nil channel"); if(runtime·gcwaiting) runtime·gosched(); runtime·lock(c); if(c->closed) { runtime·unlock(c); runtime·panicstring("close of closed channel"); } c->closed = true; // release all readers for(;;) { sg = dequeue(&c->recvq); if(sg == nil) break; gp = sg->g; gp->param = nil; runtime·ready(gp); } // release all writers for(;;) { sg = dequeue(&c->sendq); if(sg == nil) break; gp = sg->g; gp->param = nil; runtime·ready(gp); } runtime·unlock(c); } // For reflect // func chanclose(c chan) void reflect·chanclose(Hchan *c) { runtime·closechan(c); } // For reflect // func chanlen(c chan) (len int32) void reflect·chanlen(Hchan *c, int32 len) { if(c == nil) len = 0; else len = c->qcount; FLUSH(&len); } // For reflect // func chancap(c chan) (cap int32) void reflect·chancap(Hchan *c, int32 cap) { if(c == nil) cap = 0; else cap = c->dataqsiz; FLUSH(&cap); } static SudoG* dequeue(WaitQ *q) { SudoG *sgp; loop: sgp = q->first; if(sgp == nil) return nil; q->first = sgp->link; // if sgp is stale, ignore it if(sgp->selgen != NOSELGEN && (sgp->selgen != sgp->g->selgen || !runtime·cas(&sgp->g->selgen, sgp->selgen, sgp->selgen + 2))) { //prints("INVALID PSEUDOG POINTER\n"); goto loop; } return sgp; } static void dequeueg(WaitQ *q) { SudoG **l, *sgp, *prevsgp; prevsgp = nil; for(l=&q->first; (sgp=*l) != nil; l=&sgp->link, prevsgp=sgp) { if(sgp->g == g) { *l = sgp->link; if(q->last == sgp) q->last = prevsgp; break; } } } static void enqueue(WaitQ *q, SudoG *sgp) { sgp->link = nil; if(q->first == nil) { q->first = sgp; q->last = sgp; return; } q->last->link = sgp; q->last = sgp; }