Qlocal,
Qremote,
Qstatus,
+ Qlisten,
};
#define PATH(type, n) ((type)|((n)<<8))
{
Closed,
Dialing,
+ Listen,
Established,
Teardown,
Finished,
char *statestr[] = {
"Closed",
"Dialing",
+ "Listen",
"Established",
"Teardown",
"Finished",
int state;
int num;
int servernum;
- char *connect;
+
+ int rport, lport;
+ char *rhost;
+ char *lhost;
int sendpkt;
int sendwin;
};
enum {
+ MSG_GLOBAL_REQUEST = 80,
+
MSG_CHANNEL_OPEN = 90,
MSG_CHANNEL_OPEN_CONFIRMATION,
MSG_CHANNEL_OPEN_FAILURE,
Client **client;
char *mtpt;
int sshfd;
-int localport;
-char localip[] = "::";
int
vpack(uchar *p, int n, char *fmt, va_list a)
return client[num];
}
+Client*
+acceptclient(char *lhost, int lport, char *rhost, int rport)
+{
+ Client *c, *nc;
+ int i;
+
+ for(i = 0; i < nclient; i++){
+ c = client[i];
+ if(c->state == Listen && c->lport == lport && c->wq != nil){
+ nc = client[newclient()];
+ nc->wq = c->wq;
+ c->wq = nc->wq->aux;
+ nc->wq->aux = nil;
+ free(nc->lhost);
+ nc->lhost = lhost;
+ nc->lport = lport;
+ free(nc->rhost);
+ nc->rhost = rhost;
+ nc->rport = rport;
+ return nc;
+ }
+ }
+ return nil;
+}
+
void
adjustwin(Client *c, int len)
{
c->state = Closed;
sendmsg(pack(nil, "bu", MSG_CHANNEL_CLOSE, c->servernum));
break;
+ case Listen:
+ c->state = Closed;
+ sendmsg(pack(nil, "bsbsu", MSG_GLOBAL_REQUEST,
+ "cancel-tcpip-forward", 20,
+ 0,
+ c->lhost, strlen(c->lhost),
+ c->lport));
+ break;
}
while((m = c->mq) != nil){
c->mq = m->link;
"local", 0444,
"remote", 0444,
"status", 0444,
+ "listen", 0666,
};
static void
ctlwrite(Req *r, Client *c)
{
char *f[3], *s;
- int nf;
+ int nf, port;
s = emalloc9p(r->ifcall.count+1);
r->ofcall.count = r->ifcall.count;
teardownclient(c);
respond(r, nil);
}else if(strcmp(f[0], "connect") == 0){
- if(c->state != Closed)
+ if(nf != 2 || c->state != Closed)
goto Badarg;
- if(nf != 2)
+ if(getfields(f[1], f, nelem(f), 0, "!") != 2)
goto Badarg;
- free(c->connect);
- c->connect = estrdup9p(f[1]);
- nf = getfields(f[1], f, nelem(f), 0, "!");
- if(nf != 2)
+ if((port = ndbfindport(f[1])) < 0)
goto Badarg;
+ free(c->lhost);
+ c->lhost = estrdup9p("::");
+ c->lport = 0;
+ free(c->rhost);
+ c->rhost = estrdup9p(f[0]);
+ c->rport = port;
c->recvwin = WinPackets*MaxPacket;
c->recvacc = 0;
c->state = Dialing;
sendmsg(pack(nil, "bsuuususu", MSG_CHANNEL_OPEN,
"direct-tcpip", 12,
c->num, c->recvwin, MaxPacket,
- f[0], strlen(f[0]), ndbfindport(f[1]),
- localip, strlen(localip), localport));
+ c->rhost, strlen(c->rhost), c->rport,
+ c->lhost, strlen(c->lhost), c->lport));
+ }else if(strcmp(f[0], "announce") == 0){
+ if(nf != 2 || c->state != Closed)
+ goto Badarg;
+ if(getfields(f[1], f, nelem(f), 0, "!") != 2)
+ goto Badarg;
+ if((port = ndbfindport(f[1])) < 0)
+ goto Badarg;
+ if(strcmp(f[0], "*") == 0)
+ f[0] = "";
+ free(c->lhost);
+ c->lhost = estrdup9p(f[0]);
+ c->lport = port;
+ free(c->rhost);
+ c->rhost = estrdup9p("::");
+ c->rport = 0;
+ c->state = Listen;
+ sendmsg(pack(nil, "bsbsu", MSG_GLOBAL_REQUEST,
+ "tcpip-forward", 13, 0,
+ c->lhost, strlen(c->lhost), c->lport));
+ respond(r, nil);
}else{
Badarg:
respond(r, "bad or inappropriate tcp control message");
}
static void
-localread(Req *r)
+localread(Req *r, Client *c)
{
- char buf[128];
+ char buf[128], *s;
- snprint(buf, sizeof buf, "%s!%d\n", localip, localport);
+ s = c->lhost;
+ if(s == nil)
+ s = "::";
+ else if(*s == 0)
+ s = "*";
+ snprint(buf, sizeof buf, "%s!%d\n", s, c->lport);
readstr(r, buf);
respond(r, nil);
}
static void
remoteread(Req *r, Client *c)
{
- char *s;
- char buf[128];
+ char buf[128], *s;
- s = c->connect;
+ s = c->rhost;
if(s == nil)
- s = "::!0";
- snprint(buf, sizeof buf, "%s\n", s);
+ s = "::";
+ snprint(buf, sizeof buf, "%s!%d\n", s, c->rport);
readstr(r, buf);
respond(r, nil);
}
break;
case Qlocal:
- localread(r);
+ localread(r, client[NUM(path)]);
break;
case Qremote:
r->fid->aux = cs;
respond(r, nil);
break;
+ case Qlisten:
+ if(client[NUM(path)]->state != Listen){
+ respond(r, "no address set");
+ break;
+ }
+ queuewreq(client[NUM(path)], r);
+ break;
case Qclone:
n = newclient();
path = PATH(Qctl, n);
static void
handlemsg(Msg *m)
{
- int chan, win, pkt, n;
+ int chan, win, pkt, lport, rport, n, ln, rn;
+ char *s, *lhost, *rhost;
Client *c;
- char *s;
switch(m->rp[0]){
case MSG_CHANNEL_WINDOW_ADJUST:
}
free(s);
break;
+ case MSG_CHANNEL_OPEN:
+ if(unpack(m, "_suuususu", &s, &n, &chan,
+ &win, &pkt,
+ &lhost, &ln, &lport,
+ &rhost, &rn, &rport) < 0)
+ break;
+ if(n != 15 || strncmp(s, "forwarded-tcpip", 15) != 0){
+ n = 3, s = "unknown open type";
+ Reject:
+ sendmsg(pack(nil, "buus", MSG_CHANNEL_OPEN_FAILURE,
+ chan, n, s, strlen(s)));
+ break;
+ }
+ lhost = smprint("%.*s", utfnlen(lhost, ln), lhost);
+ rhost = smprint("%.*s", utfnlen(rhost, rn), rhost);
+ c = acceptclient(lhost, lport, rhost, rport);
+ if(c == nil){
+ free(lhost);
+ free(rhost);
+ n = 2, s = "connection refused";
+ goto Reject;
+ }
+ c->servernum = chan;
+ c->recvwin = WinPackets*MaxPacket;
+ c->recvacc = 0;
+ c->eof = 0;
+ c->sendpkt = pkt;
+ c->sendwin = win;
+ c->state = Established;
+
+ sendmsg(pack(nil, "buuuu", MSG_CHANNEL_OPEN_CONFIRMATION,
+ c->servernum, c->num, c->recvwin, MaxPacket));
+
+ c->ref++;
+ c->wq->fid->qid.path = PATH(Qctl, c->num);
+ c->wq->ofcall.qid.path = c->wq->fid->qid.path;
+ respond(c->wq, nil);
+ c->wq = nil;
+ break;
}
free(m);
}