From: teodor <teodor>
Date: Thu, 29 Nov 2007 18:17:59 +0000 (+0000)
Subject: Add maxsize of packet
X-Git-Url: http://www.sigaev.ru/git/gitweb.cgi?p=remotetop.git;a=commitdiff_plain;h=0440f7f90885227829b0fdfa8f52b4f843423484

Add maxsize of packet
---

diff --git a/rtop.c b/rtop.c
index 85a0eb3..59b1664 100644
--- a/rtop.c
+++ b/rtop.c
@@ -68,11 +68,10 @@ main( int argc, char *argv[] ) {
 	int i;
 
 	MassiveUnit *mu;
-	TCMsg	*pmsg = (TCMsg*)malloc( TCMSGHDRSZ );
+	TCMsg	*pmsg;
 	TC_Connection  cs;
 
-	pmsg->type = TOPGETTYPE;
-	pmsg->len = TCMSGHDRSZ;
+	opentlog( TL_OPEN_STDERR, TL_INFO, NULL);
 
 	while( (ch=getopt(argc, argv, "Dh:p:"))!=-1) {
 		switch(ch) {
@@ -97,9 +96,14 @@ main( int argc, char *argv[] ) {
 	opentlog( TL_OPEN_STDERR,  (debug) ? TL_DEBUG : TL_INFO, NULL);
 
 	TC_fillConnection(&cs, host, port);
+
+	pmsg = (TCMsg*)tmalloc( TCMSGHDRSZ );
+	pmsg->type = TOPGETTYPE;
+	pmsg->len = TCMSGHDRSZ;
 	cs.buf = (char*)pmsg;
 	cs.len = pmsg->len;
-	if ( TC_Talk(&cs) == CS_OK ) {	
+
+	if ( TC_Talk(&cs, 0) == CS_OK ) {	
 		pmsg = (TCMsg*) cs.buf;
 		mu = (MassiveUnit*)pmsg->data;
 
diff --git a/td_lmsg.c b/td_lmsg.c
index 24c3b22..cd6e6ad 100644
--- a/td_lmsg.c
+++ b/td_lmsg.c
@@ -102,7 +102,7 @@ ConnectionWorks() {
 				break;
 			case CS_READ:
 				if ( conn->readyio ) {
-					TC_Read( conn );
+					TC_Read( conn, TCMSGHDRSZ );
 					if ( conn->state != CS_READ ) i--; /* check it one more time */ 
 				}
 				break;