static ssize_t ADC_client_recv(struct ADC_client* client) { static char buf[BIG_BUFSIZE]; struct ioq_recv* q = client->recv_queue; size_t buf_size = ioq_recv_get(q, buf, BIG_BUFSIZE); ssize_t size; ADC_TRACE; if (client->flags & cflag_choke) buf_size = 0; size = net_con_recv(client->con, buf + buf_size, BIG_BUFSIZE - buf_size); if (size > 0) buf_size += size; if (size < 0) return -1; else if (size == 0) return 0; else { char* lastPos = 0; char* start = buf; char* pos = 0; size_t remaining = buf_size; while ((pos = memchr(start, '\n', remaining))) { lastPos = pos+1; pos[0] = '\0'; #ifdef DEBUG_SENDQ LOG_DUMP("PROC: \"%s\" (%d)\n", start, (int) (pos - start)); #endif if (client->flags & cflag_choke) client->flags &= ~cflag_choke; else { if (((pos - start) > 0) && MAX_RECV_BUFFER > (pos - start)) { if (ADC_client_on_recv_line(client, start, pos - start) == -1) return -1; } } pos[0] = '\n'; /* FIXME: not needed */ pos ++; remaining -= (pos - start); start = pos; } if (lastPos || remaining) { if (remaining < (size_t) MAX_RECV_BUFFER) { ioq_recv_set(q, lastPos ? lastPos : buf, remaining); } else { ioq_recv_set(q, 0, 0); client->flags |= cflag_choke; LOG_WARN("Received message past MAX_RECV_BUFFER (%d), dropping message.", MAX_RECV_BUFFER); } } else { ioq_recv_set(q, 0, 0); } } return 0; }
int handle_net_read(struct hub_user* user) { static char buf[MAX_RECV_BUF]; struct ioq_recv* q = user->recv_queue; size_t buf_size = ioq_recv_get(q, buf, MAX_RECV_BUF); ssize_t size; if (user_flag_get(user, flag_maxbuf)) buf_size = 0; size = net_con_recv(user->connection, buf + buf_size, MAX_RECV_BUF - buf_size); if (size > 0) buf_size += size; if (size < 0) { if (size == -1) return quit_disconnected; else return quit_socket_error; } else if (size == 0) { return 0; } else { char* lastPos = 0; char* start = buf; char* pos = 0; size_t remaining = buf_size; while ((pos = memchr(start, '\n', remaining))) { lastPos = pos+1; pos[0] = '\0'; #ifdef DEBUG_SENDQ LOG_DUMP("PROC: \"%s\" (%d)\n", start, (int) (pos - start)); #endif if (user_flag_get(user, flag_maxbuf)) { user_flag_unset(user, flag_maxbuf); } else { if (((pos - start) > 0) && user->hub->config->max_recv_buffer > (pos - start)) { if (hub_handle_message(user->hub, user, start, (pos - start)) == -1) { return quit_protocol_error; } } } pos[0] = '\n'; /* FIXME: not needed */ pos ++; remaining -= (pos - start); start = pos; } if (lastPos || remaining) { if (remaining < (size_t) user->hub->config->max_recv_buffer) { ioq_recv_set(q, lastPos ? lastPos : buf, remaining); } else { ioq_recv_set(q, 0, 0); user_flag_set(user, flag_maxbuf); LOG_WARN("Received message past max_recv_buffer, dropping message."); } } else { ioq_recv_set(q, 0, 0); } } return 0; }