X-Git-Url: http://git.dolda2000.com/gitweb/?a=blobdiff_plain;f=daemon%2Ffnet-dc.c;h=2e490eac6b393913ad406307a63e6486158c121a;hb=7bb3e8c17cf3da0c9911dc6d21795403317f3636;hp=f48d63d34896f281608bec5807a1f538f72b73d6;hpb=8d7a1e880011bbb660895f022a9f250f3cfda391;p=doldaconnect.git

diff --git a/daemon/fnet-dc.c b/daemon/fnet-dc.c
index f48d63d..2e490ea 100644
--- a/daemon/fnet-dc.c
+++ b/daemon/fnet-dc.c
@@ -872,7 +872,7 @@ static void cmd_hubname(struct socket *sk, struct fnetnode *fn, char *cmd, char
     struct dchub *hub;
     
     hub = fn->data;
-    if(hub->nativename == NULL)
+    if(hub->nativename != NULL)
 	free(hub->nativename);
     hub->nativename = sstrdup(args);
     buf = icmbstowcs(args, hub->charset);
@@ -1714,6 +1714,7 @@ static void startul(struct dcpeer *peer)
     peer->state = PEER_TRNS;
     transferstartul(peer->transfer, peer->sk);
     peer->sk->writecb = (void (*)(struct socket *, void *))transwrite;
+    transwrite(peer->sk, peer);
 }
 
 static void cmd_filelength(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
@@ -2117,6 +2118,7 @@ static void cmd_adcget(struct socket *sk, struct dcpeer *peer, char *cmd, char *
     } else if(fd >= 0) {
 	if((wbuf = adc2path(argv[1])) != NULL)
 	    transfersetpath(peer->transfer, wbuf);
+	free(wbuf);
 	peer->transfer->flags.b.minislot = 1;
     }
     if(fd < 0)
@@ -2688,28 +2690,24 @@ static void dctransgotdata(struct transfer *transfer, struct dcpeer *peer)
 	{
 	    if((buf = sockgetinbuf(peer->trpipe, &bufsize)) != NULL)
 	    {
-		if((transfer->endpos >= 0) && (transfer->curpos + bufsize >= transfer->endpos))
-		    bufsize = transfer->endpos - transfer->curpos;
-		if(bufsize > 0) {
-		    if(peer->compress == CPRS_NONE)
+		if(peer->compress == CPRS_NONE)
+		{
+		    sockqueue(peer->sk, buf, bufsize);
+		} else if(peer->compress == CPRS_ZLIB) {
+		    cstr = peer->cprsdata;
+		    cstr->next_in = buf;
+		    cstr->avail_in = bufsize;
+		    while(cstr->avail_in > 0)
 		    {
-			sockqueue(peer->sk, buf, bufsize);
-		    } else if(peer->compress == CPRS_ZLIB) {
-			cstr = peer->cprsdata;
-			cstr->next_in = buf;
-			cstr->avail_in = bufsize;
-			while(cstr->avail_in > 0)
+			cstr->next_out = outbuf;
+			cstr->avail_out = sizeof(outbuf);
+			if((ret = deflate(cstr, 0)) != Z_OK)
 			{
-			    cstr->next_out = outbuf;
-			    cstr->avail_out = sizeof(outbuf);
-			    if((ret = deflate(cstr, 0)) != Z_OK)
-			    {
-				flog(LOG_WARNING, "bug? deflate() did not return Z_OK (but rather %i)", ret);
-				freedcpeer(peer);
-				return;
-			    }
-			    sockqueue(peer->sk, outbuf, sizeof(outbuf) - cstr->avail_out);
+			    flog(LOG_WARNING, "bug? deflate() did not return Z_OK (but rather %i)", ret);
+			    freedcpeer(peer);
+			    return;
 			}
+			sockqueue(peer->sk, outbuf, sizeof(outbuf) - cstr->avail_out);
 		    }
 		}
 		free(buf);
@@ -2753,34 +2751,6 @@ static void dctransgotdata(struct transfer *transfer, struct dcpeer *peer)
     }
 }
 
-void trpiperead(struct socket *sk, struct dcpeer *peer)
-{
-    dctransgotdata(peer->transfer, peer);
-}
-
-void trpipewrite(struct socket *sk, struct dcpeer *peer)
-{
-}
-
-void trpipeerr(struct socket *sk, int errno, struct dcpeer *peer)
-{
-    peer->state = PEER_SYNC;
-    dctransgotdata(peer->transfer, peer);
-    CBUNREG(peer->transfer, trans_filterout, peer);
-}
-
-static struct socket *mktrpipe(struct dcpeer *peer)
-{
-    struct socket *sk;
-    
-    sk = netsockpipe();
-    sk->data = peer;
-    sk->readcb = (void (*)(struct socket *, void *))trpiperead;
-    sk->writecb = (void (*)(struct socket *, void *))trpipewrite;
-    sk->errcb = (void (*)(struct socket *, int, void *))trpipeerr;
-    return(sk);
-}
-
 static void transread(struct socket *sk, struct dcpeer *peer)
 {
     void *buf;
@@ -2788,20 +2758,23 @@ static void transread(struct socket *sk, struct dcpeer *peer)
     
     if(sockqueueleft(peer->trpipe) < 0)
 	return;
-    if((buf = sockgetinbuf(sk, &bufsize)) == NULL)
-	return;
-    if(peer->transfer == NULL)
+    if((buf = sockgetinbuf(sk, &bufsize)) != NULL)
     {
+	if(peer->transfer == NULL)
+	{
+	    free(buf);
+	    freedcpeer(peer);
+	    return;
+	}
+	sockqueue(peer->trpipe, buf, bufsize);
 	free(buf);
-	freedcpeer(peer);
-	return;
     }
-    sockqueue(peer->trpipe, buf, bufsize);
-    free(buf);
     if(peer->transfer->curpos >= peer->transfer->size)
     {
 	closesock(peer->trpipe);
 	quitsock(peer->trpipe);
+	peer->trpipe = NULL;
+	peer->transfer = NULL;
 	peer->close = 1;
 	return;
     }
@@ -2818,6 +2791,8 @@ static void transerr(struct socket *sk, int err, struct dcpeer *peer)
     }
     closesock(peer->trpipe);
     quitsock(peer->trpipe);
+    peer->trpipe = NULL;
+    peer->transfer = NULL;
     peer->close = 1;
 }
 
@@ -2833,6 +2808,35 @@ static void transwrite(struct socket *sk, struct dcpeer *peer)
     dctransgotdata(peer->transfer, peer);
 }
 
+static void trpiperead(struct socket *sk, struct dcpeer *peer)
+{
+    dctransgotdata(peer->transfer, peer);
+}
+
+static void trpipewrite(struct socket *sk, struct dcpeer *peer)
+{
+    transread(peer->sk, peer);
+}
+
+static void trpipeerr(struct socket *sk, int errno, struct dcpeer *peer)
+{
+    peer->state = PEER_SYNC;
+    dctransgotdata(peer->transfer, peer);
+    CBUNREG(peer->transfer, trans_filterout, peer);
+}
+
+static struct socket *mktrpipe(struct dcpeer *peer)
+{
+    struct socket *sk;
+    
+    sk = netsockpipe();
+    sk->data = peer;
+    sk->readcb = (void (*)(struct socket *, void *))trpiperead;
+    sk->writecb = (void (*)(struct socket *, void *))trpipewrite;
+    sk->errcb = (void (*)(struct socket *, int, void *))trpipeerr;
+    return(sk);
+}
+
 static void udpread(struct socket *sk, void *data)
 {
     char *buf, *p, *p2, *hashbuf;
@@ -3247,6 +3251,9 @@ static void peerread(struct socket *sk, struct dcpeer *peer)
     if(peer->state == PEER_CMD) {
 	if((peer->queue.size > 50) || (peer->inbufdata > 65536))
 	    return;
+    } else if(peer->state == PEER_TTHL) {
+    } else {
+	return;
     }
     if((newbuf = sockgetinbuf(sk, &datalen)) == NULL)
 	return;