From 0440f7f90885227829b0fdfa8f52b4f843423484 Mon Sep 17 00:00:00 2001 From: teodor Date: Thu, 29 Nov 2007 18:17:59 +0000 Subject: [PATCH] Add maxsize of packet --- rtop.c | 12 ++++++++---- td_lmsg.c | 2 +- 2 files changed, 9 insertions(+), 5 deletions(-) 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; -- 2.46.1