added comments

This commit is contained in:
John Jones 2017-07-27 15:13:35 -05:00
parent 1e49d8f7ab
commit 343748bc0f
3 changed files with 61 additions and 35 deletions

View file

@ -1,5 +1,4 @@
#ifndef P2PNET_H #pragma once
#define P2PNET_H
#include <stdint.h> #include <stdint.h>
#include <unistd.h> #include <unistd.h>
@ -12,6 +11,19 @@
int socket_local4(int s, uint32_t *ip, uint16_t *port); int socket_local4(int s, uint32_t *ip, uint16_t *port);
int socket_connect4(int s, uint32_t ip, uint16_t port); int socket_connect4(int s, uint32_t ip, uint16_t port);
int socket_listen(int s, uint32_t *localip, uint16_t *localport); int socket_listen(int s, uint32_t *localip, uint16_t *localport);
/***
* Reads data from a socket, used instead of recv so if a protocol needs
* to use something else before or after it can be done here instead of
* outside the lib.
*
* @param s the socket
* @param buf what to send
* @param len the length of buf
* @param flags network flags
* @param num_secs the number of seconds before a timeout
* @returns number of bytes, 0, or negative number on error (i.e. EAGAIN or EWOULDBLOCK)
*/
ssize_t socket_read(int s, char *buf, size_t len, int flags, int timeout_secs); ssize_t socket_read(int s, char *buf, size_t len, int flags, int timeout_secs);
ssize_t socket_write(int s, const char *buf, size_t len, int flags); ssize_t socket_write(int s, const char *buf, size_t len, int flags);
/** /**
@ -35,5 +47,3 @@
* @returns the ip address as an uint32_t * @returns the ip address as an uint32_t
*/ */
uint32_t hostname_to_ip(const char* hostname); uint32_t hostname_to_ip(const char* hostname);
#endif // P2PNET_H

View file

@ -94,6 +94,7 @@ int libp2p_net_multistream_read(void* stream_context, unsigned char** results, s
struct SessionContext* session_context = (struct SessionContext*)stream_context; struct SessionContext* session_context = (struct SessionContext*)stream_context;
struct Stream* stream = session_context->default_stream; struct Stream* stream = session_context->default_stream;
int bytes = 0; int bytes = 0;
// TODO: this is arbitrary, and should be dynamic // TODO: this is arbitrary, and should be dynamic
size_t buffer_size = 362144; size_t buffer_size = 362144;
char buffer[buffer_size]; char buffer[buffer_size];
@ -101,11 +102,15 @@ int libp2p_net_multistream_read(void* stream_context, unsigned char** results, s
size_t num_bytes_requested = 0, left = 0, already_read = 0; size_t num_bytes_requested = 0, left = 0, already_read = 0;
if (session_context->secure_stream == NULL) { if (session_context->secure_stream == NULL) {
int socketDescriptor = *( (int*) stream->socket_descriptor);
// first read the varint // first read the varint
while(1) { while(1) {
unsigned char c = '\0'; unsigned char c = '\0';
bytes = socket_read(*((int*)stream->socket_descriptor), (char*)&c, 1, 0, timeout_secs); bytes = socket_read(socketDescriptor, (char*)&c, 1, 0, timeout_secs);
if (bytes <= 0) { // timeout if (bytes <= 0) {
// possible error
if (bytes < 0)
libp2p_logger_error("multistream", "socket_read returned %d reading socket %d\n", bytes, socketDescriptor);
return 0; return 0;
} }
pos[0] = c; pos[0] = c;
@ -116,17 +121,20 @@ int libp2p_net_multistream_read(void* stream_context, unsigned char** results, s
} }
pos++; pos++;
} }
if (num_bytes_requested <= 0) if (num_bytes_requested <= 0) {
libp2p_logger_debug("multistream", "Reading the varint returned %d on socket %d\n", num_bytes_requested, socketDescriptor);
return 0; return 0;
}
left = num_bytes_requested; left = num_bytes_requested;
do { do {
bytes = socket_read(*((int*)stream->socket_descriptor), &buffer[already_read], left, 0, timeout_secs); bytes = socket_read(socketDescriptor, &buffer[already_read], left, 0, timeout_secs);
if (bytes < 0) { if (bytes < 0) {
bytes = 0; bytes = 0;
if ( errno == EAGAIN ) { if ( errno == EAGAIN ) {
// do something intelligent // do something intelligent
} else { } else {
libp2p_logger_error("multistream", "socket read returned error %d on socket descriptor %d.\n", errno, socketDescriptor);
return 0; return 0;
} }
} }

View file

@ -137,9 +137,17 @@ int socket_listen(int s, uint32_t *localip, uint16_t *localport)
return s; return s;
} }
/* Reads data from a socket, used instead of recv so if a protocol needs /***
* Reads data from a socket, used instead of recv so if a protocol needs
* to use something else before or after it can be done here instead of * to use something else before or after it can be done here instead of
* outside the lib. * outside the lib.
*
* @param s the socket
* @param buf what to send
* @param len the length of buf
* @param flags network flags
* @param num_secs the number of seconds before a timeout
* @returns number of bytes, 0, or negative number on error (i.e. EAGAIN or EWOULDBLOCK)
*/ */
ssize_t socket_read(int s, char *buf, size_t len, int flags, int num_secs) ssize_t socket_read(int s, char *buf, size_t len, int flags, int num_secs)
{ {