--- mqtt/src/client.c 2011/12/13 15:23:43 1.1.2.5 +++ mqtt/src/client.c 2011/12/20 15:03:42 1.1.2.8 @@ -24,10 +24,11 @@ int try2Connect(int sock) { int siz = 0; + struct pollfd pfd; - siz = mqtt_msgCONNECT(args->msg, AIT_GET_STR(&args->ConnID), args->ka, - AIT_GET_STR(&args->User), AIT_GET_STR(&args->Pass), - args->Will.Topic.val.string, args->Will.Msg.val.string, + siz = mqtt_msgCONNECT(args->msg, (char*) AIT_GET_STR(&args->ConnID), args->ka, + (char*) AIT_GET_STR(&args->User), (char*) AIT_GET_STR(&args->Pass), + (char*) args->Will.Topic.val.string, (char*) args->Will.Msg.val.string, !args->notClear, args->QoS, args->Retain); if (siz == -1) { printf("Error:: msgCONNECT #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); @@ -39,11 +40,27 @@ try2Connect(int sock) printf("Error:: send() #%d - %s\n", errno, strerror(errno)); return -1; } else - VERB(3) printf("Sended CONNECT %d bytes\n", siz); + ioVERBOSE(3) printf("Sended CONNECT %d bytes\n", siz); - register int i; - for (i = 0; i < siz; i++) - printf("%02x ", ((u_char*)args->msg->msg_base)[i]); - printf("\n"); - return siz; + pfd.fd = sock; + pfd.events = POLLIN; + switch (poll(&pfd, 1, args->ka * 1000)) { + case -1: + printf("Error:: poll() #%d - %s\n", errno, strerror(errno)); + return -1; + case 0: + ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); + return -1; + } + if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) + return -1; + + siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0); + if (siz == -1) { + printf("Error:: recv() #%d - %s\n", errno, strerror(errno)); + return -1; + } else + ioVERBOSE(3) printf("Received %d bytes\n", siz); + + return (u_char) mqtt_readCONNACK(args->msg); }