/********** Client Side Code **********/
int getResponse(noPollConn * conn, int requestLen, int index) {
noPollMsg * msg;
int count = 0;
uint64_t beginTime = GetTime();
int msglen = 0;
while (true) {
count++;
msg = nopoll_conn_get_msg(conn);
if (msg != NULL) {
noPollOpCode code = nopoll_msg_opcode(msg);
if (!((code == NOPOLL_BINARY_FRAME) || (code == NOPOLL_CONTINUATION_FRAME))) {
LogPrintf("Received unexpected message type %d", code);
return -1;
}
msglen += nopoll_msg_get_payload_size(msg);
DebugPrintf("Retrieved message %d frag %d:%d of size %d (collected %d of %d)",
index, nopoll_msg_is_fragment(msg), nopoll_msg_is_final(msg),
nopoll_msg_get_payload_size(msg), msglen, requestLen);
if (nopoll_msg_is_final(msg) && (msglen != requestLen)) {
DebugPrintf("Final flag set arbitrarily??");
}
if (msglen == requestLen) {
if (!nopoll_msg_is_final(msg))
DebugPrintf("Final flag not set??");
break;
}
nopoll_msg_unref(msg);
}
if (!nopoll_conn_is_ok(conn)) {
LogPrintf("Client disconnected!!");
return -1;
}
nopoll_sleep(100);
}
DebugPrintf("Retrieved full message %d in %d ms looped: %d",
index, (int) (GetTime() - beginTime), count);
if (msglen != requestLen) {
LogPrintf("Received invalid response length %d != %d", msglen, requestLen);
return -1;
}
nopoll_msg_unref(msg);
return 0;
}
void listener_on_message(noPollCtx * ctx, noPollConn * conn, noPollMsg * msg, noPollPtr * user_data )
{
// print the message (for debugging purposes) and reply
printf ("Listener received (size: %d, ctx refs: %d): (first [removed] bytes, fragment: %d) '%s'\n",
nopoll_msg_get_payload_size (msg),
nopoll_ctx_ref_count (ctx), nopoll_msg_is_fragment (msg), nopoll_msg_get_payload(msg));
// reply to the message
nopoll_conn_send_text (conn, "Message received", 16);
return;
}
void listener_on_message(noPollCtx * ctx, noPollConn * conn, noPollMsg * msg, noPollPtr user_data)
{
noPollOpCode code = nopoll_msg_opcode(msg);
if (code == NOPOLL_TEXT_FRAME) {
int msglen = nopoll_msg_get_payload_size(msg);
char message[32];
if (msglen > 0 && msglen < 32) {
memcpy(message, nopoll_msg_get_payload(msg), msglen);
message[msglen] = 0;
handleRequest(conn, message);
}
else {
LogPrintf("Invalid request from client: Invalid message length.");
exit(1);
}
}
else {
LogPrintf("Invalid request from client: Expecting text message.");
exit(1);
}
}
void write_file_handler (noPollCtx * ctx, noPollConn * conn, noPollMsg * msg, noPollPtr user_data)
{
FILE * open_file_cmd = (FILE*) user_data;
const char * content = (const char *) nopoll_msg_get_payload (msg);
int value;
/* check for close operation */
if (nopoll_ncmp (content, "close-file", 10)) {
printf ("CLOSING FILE: opened..\n");
fclose (open_file_cmd);
open_file_cmd = NULL;
return;
} /* end if */
if (open_file_cmd) {
/* write content */
value = fwrite (content, 1, nopoll_msg_get_payload_size (msg), open_file_cmd);
if (value < 0)
return;
return;
} /* end if */
return;
}
void listener_on_message (noPollCtx * ctx, noPollConn * conn, noPollMsg * msg, noPollPtr user_data)
{
const char * content = (const char *) nopoll_msg_get_payload (msg);
FILE * file = NULL;
char buffer[1024];
int bytes;
int sent;
char example[100];
int shown;
noPollMsg * aux;
nopoll_bool dont_reply = nopoll_false;
FILE * open_file_cmd = NULL;
int iterator;
char * ref;
/* check for open file commands */
if (nopoll_ncmp (content, "open-file: ", 11)) {
#if defined(NOPOLL_OS_WIN32)
open_file_cmd = fopen (content + 11, "ab");
#else
open_file_cmd = fopen (content + 11, "a");
#endif
if (open_file_cmd == NULL) {
printf ("ERROR: unable to open file: %s\n", content + 11);
return;
} /* end if */
/* set handler */
nopoll_conn_set_on_msg (conn, write_file_handler, open_file_cmd);
return;
} /* end if */
/* printf ("Message received: %s\n", content); */
if (nopoll_ncmp (content, "close with message", 18)) {
printf ("Listener: RELEASING connection (closing it) with reason..\n");
nopoll_conn_close_ext (conn, 1048, "Hey, this is a very reasonable error message", 44);
return;
} /* end if */
if (nopoll_ncmp (content, "release-message", 15)) {
printf ("Listener: RELEASING previous message..\n");
nopoll_msg_unref (previous_msg);
previous_msg = NULL;
return;
} /* end if */
if (nopoll_ncmp (content, "get-cookie", 10)) {
printf ("Listener: reporting cookie: %s\n", nopoll_conn_get_cookie (conn));
nopoll_conn_send_text (conn, nopoll_conn_get_cookie (conn), strlen (nopoll_conn_get_cookie (conn)));
return;
}
/* printf ("Checking for set-broken socket: %s\n", content); */
if (nopoll_ncmp (content, "set-broken-socket", 17)) {
printf ("Listener: setting broken socket on conn: %p (socket=%d)\n",
conn, (int) nopoll_conn_socket (conn));
nopoll_conn_shutdown (conn);
return;
} /* end if */
if (nopoll_ncmp (content, "get-connection-close-count", 26)) {
printf ("Sending reply to report connection close...\n");
ref = nopoll_strdup_printf ("%d", connection_close_count);
nopoll_conn_send_text (conn, ref, strlen (ref));
nopoll_free (ref);
return;
} /* end if */
if (nopoll_ncmp (content, "1234-1) ", 8)) {
printf ("Listener: waiting a second to force buffer flooding..\n");
nopoll_sleep (100000);
dont_reply = nopoll_true;
} /* end if */
/* get initial bytes */
bytes = nopoll_msg_get_payload_size (msg);
shown = bytes > 100 ? 99 : bytes;
memset (example, 0, 100);
/* if (! nopoll_msg_is_fragment (msg)) */
memcpy (example, (const char *) nopoll_msg_get_payload (msg), shown);
printf ("Listener received (size: %d, ctx refs: %d): (first %d bytes, fragment: %d) '%s'\n",
nopoll_msg_get_payload_size (msg),
nopoll_ctx_ref_count (ctx), shown, nopoll_msg_is_fragment (msg), example);
if (nopoll_cmp (content, "ping")) {
/* send a ping */
nopoll_conn_send_ping (conn);
return;
} else if (nopoll_cmp (content, "get-file")) {
iterator = 0;
file = NULL;
while (nopoll_true) {
#if defined(NOPOLL_OS_WIN32)
file = fopen ("nopoll-regression-client.c", "rb");
#else
file = fopen ("nopoll-regression-client.c", "r");
#endif
printf ("LISTENER: file pointer (%p, errno=%d : %s)..\n", file, errno, strerror (errno));
if (file)
break;
iterator++;
if (iterator > 3) {
printf ("ERROR: failed to open nopoll-regression-client.c (fopen call failed)\n");
nopoll_conn_shutdown (conn);
return;
} /* end if */
} /* end while */
while (! feof (file)) {
/* read content */
bytes = fread (buffer, 1, 1024, file);
/* send content */
if (bytes > 0) {
/* send content and get the result */
/* printf ("Sending message with %d bytes..\n", bytes); */
/* nopoll_log_enable (ctx, nopoll_true); */
while (nopoll_true) {
/* try to send content */
sent = nopoll_conn_send_text (conn, buffer, bytes);
/* nopoll_log_enable (ctx, nopoll_false); */
if (sent != bytes) {
if (errno == NOPOLL_EWOULDBLOCK) {
nopoll_sleep (1000);
/* printf (" ..retrying..sending message with %d bytes..\n", bytes); */
continue;
} /* end if */
printf ("ERROR: expected to send %d bytes but sent different content size (%d bytes), errno=%d (%d)..\n",
bytes, sent, errno, NOPOLL_EWOULDBLOCK);
} /* end if */
break;
}
} /* end if */
/* next */
} /* end while */
/* now close the handle */
fclose (file);
return;
} /* end if */
/* check if we have to reply */
if (dont_reply)
return;
if (nopoll_msg_is_fragment (msg)) {
printf ("Found fragment, FIN = %d (%p)?..\n", nopoll_msg_is_final (msg), msg);
/* call to join this message */
aux = previous_msg;
previous_msg = nopoll_msg_join (previous_msg, msg);
nopoll_msg_unref (aux);
if (! nopoll_msg_is_final (msg)) {
printf ("Found fragment that is not final..\n");
printf ("Not replying because frame fragment received..\n");
return;
} /* end if */
printf ("Found final fragment, replying with complete content: %s (refs: %d)..\n",
(const char *) nopoll_msg_get_payload (previous_msg),
nopoll_msg_ref_count (previous_msg));
/* ok, now found final piece, replying */
nopoll_conn_send_text (conn, (const char *) nopoll_msg_get_payload (previous_msg),
nopoll_msg_get_payload_size (previous_msg));
/* release reference */
nopoll_msg_unref (previous_msg);
previous_msg = NULL;
return;
}
/* send reply as received */
printf ("Sending reply... (same message size: %d)\n", nopoll_msg_get_payload_size (msg));
nopoll_conn_send_text (conn, (const char *) nopoll_msg_get_payload (msg),
nopoll_msg_get_payload_size (msg));
return;
}
