static void write_to_sock (PseudoTcpSocket *sock)
{
gchar buf[1024];
gsize len;
gint wlen;
guint total = 0;
while (TRUE) {
len = fread (buf, 1, sizeof(buf), in);
if (len == 0) {
g_debug ("Done reading data from file");
g_assert (feof (in));
reading_done = TRUE;
pseudo_tcp_socket_close (sock, FALSE);
break;
} else {
wlen = pseudo_tcp_socket_send (sock, buf, len);
g_debug ("Sending %" G_GSIZE_FORMAT " bytes : %d", len, wlen);
total += wlen;
total_read += wlen;
if (wlen < (gint) len) {
g_debug ("seeking %ld from %lu", wlen - len, ftell (in));
fseek (in, wlen - len, SEEK_CUR);
g_assert (!feof (in));
g_debug ("Socket queue full after %d bytes written", total);
break;
}
}
}
adjust_clock (sock);
}
static gboolean notify_clock (gpointer data)
{
PseudoTcpSocket *sock = (PseudoTcpSocket *)data;
//g_debug ("Socket %p: Notifying clock", sock);
pseudo_tcp_socket_notify_clock (sock);
adjust_clock (sock);
return FALSE;
}
int main (int argc, char *argv[])
{
PseudoTcpCallbacks cbs = {
NULL, opened, readable, writable, closed, write_packet
};
setlocale (LC_ALL, "");
mainloop = g_main_loop_new (NULL, FALSE);
g_type_init ();
pseudo_tcp_set_debug_level (PSEUDO_TCP_DEBUG_VERBOSE);
left_closed = right_closed = FALSE;
left = pseudo_tcp_socket_new (0, &cbs);
right = pseudo_tcp_socket_new (0, &cbs);
g_debug ("Left: %p. Right: %p", left, right);
pseudo_tcp_socket_notify_mtu (left, 1496);
pseudo_tcp_socket_notify_mtu (right, 1496);
pseudo_tcp_socket_connect (left);
adjust_clock (left);
adjust_clock (right);
if (argc == 3) {
in = fopen (argv[1], "r");
out = fopen (argv[2], "w");
}
g_main_loop_run (mainloop);
g_object_unref (left);
g_object_unref (right);
if (in)
fclose (in);
if (out)
fclose (out);
return 0;
}
static gboolean notify_packet (gpointer user_data)
{
struct notify_data *data = (struct notify_data*) user_data;
pseudo_tcp_socket_notify_packet (data->sock, data->buffer, data->len);
adjust_clock (data->sock);
g_free (data);
return FALSE;
}
static gboolean
notify_packet (gpointer user_data)
{
struct notify_data *data = (struct notify_data*) user_data;
/* Fuzz the packet. */
data->len = fuzz_packet (data->buffer, data->len, data->stream_pos);
pseudo_tcp_socket_notify_packet (data->sock,
(gchar *) data->buffer, data->len);
adjust_clock (data->sock);
g_free (data);
return FALSE;
}
static void* read_callback_peer(void *context)
{
int return_value = 0;
sensor_context *sensor = NULL;
peer *client = (peer*)context;
message msg;
sensor = client->sensor;
int msg_logical_clock[CLOCK_SIZE];
return_value = read_message(client->comm_socket_fd, msg_logical_clock, &msg);
if(return_value != E_SUCCESS)
{
if(return_value == E_SOCKET_CONNECTION_CLOSED)
{
LOG_ERROR(("ERROR: Socket connection from server closed...\n"));
}
LOG_ERROR(("ERROR: Error in read message\n"));
return NULL;
}
if(msg.type == REGISTER)
{
client->ip_address = msg.u.s.ip_address;
client->port_no = msg.u.s.port_no;
return (NULL);
}
pthread_mutex_lock(&sensor->mutex_lock);
adjust_clock(sensor->logical_clock, msg_logical_clock);
pthread_mutex_unlock(&sensor->mutex_lock);
LOG_INFO(("INFO: Event Received: "));
print_logical_clock(sensor->logical_clock);
LOG_INFO((", timestamp: %lu, From %s:%s\n", msg.timestamp, client->ip_address, client->port_no));
return (NULL);
}
int main (int argc, char *argv[])
{
PseudoTcpCallbacks cbs = {
NULL, opened, readable, writable, closed, write_packet
};
GOptionContext *context;
GError *error = NULL;
setlocale (LC_ALL, "");
g_type_init ();
/* Configuration. */
context = g_option_context_new ("— fuzz-test the pseudotcp socket");
g_option_context_add_main_entries (context, entries, NULL);
if (!g_option_context_parse (context, &argc, &argv, &error)) {
g_printerr ("Option parsing failed: %s\n", error->message);
goto context_error;
}
if (n_changes_lambda == 0) {
g_printerr ("Option parsing failed: %s\n",
"Lambda values must be positive.");
goto context_error;
}
g_option_context_free (context);
/* Tweak the configuration. */
if (seed == 0) {
seed = g_get_real_time ();
}
/* Open the input and output files */
if (argc >= 3) {
in = fopen (argv[1], "r");
out = fopen (argv[2], "w");
}
/* Set up the main loop and sockets. */
main_loop = g_main_loop_new (NULL, FALSE);
g_print ("Using seed: %" G_GINT64_FORMAT ", start position: %u, λ: %u\n",
seed, fuzz_start_pos, n_changes_lambda);
prng = g_rand_new_with_seed (seed);
pseudo_tcp_set_debug_level (PSEUDO_TCP_DEBUG_VERBOSE);
left = pseudo_tcp_socket_new (0, &cbs);
right = pseudo_tcp_socket_new (0, &cbs);
g_debug ("Left: %p. Right: %p", left, right);
pseudo_tcp_socket_notify_mtu (left, 1496);
pseudo_tcp_socket_notify_mtu (right, 1496);
pseudo_tcp_socket_connect (left);
adjust_clock (left);
adjust_clock (right);
/* Run the main loop. */
g_main_loop_run (main_loop);
g_main_loop_unref (main_loop);
g_object_unref (left);
g_object_unref (right);
g_rand_free (prng);
if (in != NULL)
fclose (in);
if (out != NULL)
fclose (out);
return retval;
context_error:
g_printerr ("\n%s\n", g_option_context_get_help (context, TRUE, NULL));
g_option_context_free (context);
return 1;
}
