John Jones 2016-11-13 20:58:11 -05:00
commit d3d0b72506
6 changed files with 190 additions and 0 deletions

12
net/TODO Normal file
View File

@ -0,0 +1,12 @@
- SCTP datagram and sequenced packets implementations, just stream for now.
- UDP implementation, incomplete.
- Implement some kind of network events such as libutp does, so libutp can
be added and a more robust implementation for already implemented protocols,
it may also be easier to switch protocols or use multiples at once from
high-level calls.
- Implement UDT
- IPv6.

18
net/p2pnet.h Normal file
View File

@ -0,0 +1,18 @@
#ifndef P2PNET_H
#define P2PNET_H
int socket_bind4(int s, uint32_t ip, uint16_t port);
int socket_bind4_reuse(int s, uint32_t ip, uint16_t port);
int socket_accept4(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_listen(int s, uint32_t *localip, uint16_t *localport);
ssize_t socket_read(int s, char *buf, size_t len, int flags);
ssize_t socket_write(int s, char *buf, size_t len, int flags);
int socket_tcp4(void);
int socket_stream_sctp4(void);
int socket_udp4(void);
#endif // P2PNET_H

15
net/sctp.c Normal file
View File

@ -0,0 +1,15 @@
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include "p2pnet.h"
/* Create a SCTP socket.
*/
int socket_stream_sctp4(void)
{
int s;
s = socket(PF_INET, SOCK_STREAM, IPPROTO_SCTP);
if (s == -1) return -1;
return s;
}

114
net/socket.c Normal file
View File

@ -0,0 +1,114 @@
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <stdint.h>
#include <string.h>
#include <arpa/inet.h>
#include "p2pnet.h"
/* associate an IP address with an port to a socket.
* first param is the socket file description
* second is an array of four bytes IP address
* in binary format, this function return 0 on sucess
* or -1 on error setting errno apropriated.
*/
int socket_bind4(int s, uint32_t ip, uint16_t port)
{
struct sockaddr_in sa;
bzero(&sa, sizeof sa);
sa.sin_family = AF_INET;
sa.sin_port = htons(port);
sa.sin_addr.s_addr = ip;
return bind(s, (struct sockaddr *) &sa, sizeof sa);
}
/* Same as socket_bind4(), but set SO_REUSEADDR before
*/
int socket_bind4_reuse(int s, uint32_t ip, uint16_t port)
{
int opt = 1;
setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof opt);
return socket_bind4(s, ip, port);
}
/* Accept a connection in a socket and return ip and port of
* remote connection at pointers passed as parameters.
*/
int socket_accept4(int s, uint32_t *ip, uint16_t *port)
{
struct sockaddr_in sa;
socklen_t dummy = sizeof sa;
int fd;
fd = accept(s, (struct sockaddr *) &sa, &dummy);
if (fd == -1) return -1;
*ip = sa.sin_addr.s_addr;
*port = ntohs(sa.sin_port);
return fd;
}
/* retrieve local ip and port information from a socket.
*/
int socket_local4(int s, uint32_t *ip, uint16_t *port)
{
struct sockaddr_in sa;
socklen_t dummy = sizeof sa;
if (getsockname(s, (struct sockaddr *) &sa, &dummy) == -1) return -1;
*ip = sa.sin_addr.s_addr;
*port = ntohs(sa.sin_port);
return 0;
}
/* start a client connection.
*/
int socket_connect4(int s, uint32_t ip, uint16_t port)
{
struct sockaddr_in sa;
memset(&sa, 0, sizeof sa);
sa.sin_family = AF_INET;
sa.sin_port = htons(port);
sa.sin_addr.s_addr = ip;
return connect(s, (struct sockaddr *) &sa, sizeof sa);
}
/* bind and listen to a socket.
*/
int socket_listen(int s, uint32_t *localip, uint16_t *localport)
{
if (socket_bind4_reuse(s, *localip, *localport) == -1) {
close(s);
return -1;
}
if (socket_local4(s, localip, localport) == -1) {
close(s);
return -1;
}
if (listen(s, 1) == -1) {
close(s);
return -1;
}
return s;
}
/* 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.
*/
ssize_t socket_read(int s, char *buf, size_t len, int flags)
{
return recv(s, buf, len, flags);
}
/* Same reason as socket_read, but to send data instead of receive.
*/
ssize_t socket_write(int s, char *buf, size_t len, int flags)
{
return send(s, buf, len, flags);
}

15
net/tcp.c Normal file
View File

@ -0,0 +1,15 @@
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include "p2pnet.h"
/* Create a TCP socket.
*/
int socket_tcp4(void)
{
int s;
s = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
if (s == -1) return -1;
return s;
}

16
net/udp.c Normal file
View File

@ -0,0 +1,16 @@
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include "p2pnet.h"
/* Create a UDP socket.
*/
int socket_udp4(void)
{
int s;
s = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
if (s == -1) return -1;
return s;
}