79 lines
1.8 KiB
C
79 lines
1.8 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include "forge_socket.h"
|
|
#include <netinet/in.h>
|
|
#include <netinet/tcp.h>
|
|
|
|
// TODO: one day, we should actually add a libforge_socket to the forge_socket/userspace
|
|
|
|
// Fills in all but the src/dst ip/port and seq/ack numbers
|
|
// with some sane defaults
|
|
struct tcp_state *forge_socket_get_default_state()
|
|
{
|
|
struct tcp_state *st;
|
|
st = (struct tcp_state *)malloc(sizeof(struct tcp_state));
|
|
if (st == NULL) {
|
|
return NULL;
|
|
}
|
|
st->tstamp_ok = 0;
|
|
st->sack_ok = 0;
|
|
st->wscale_ok = 0;
|
|
st->ecn_ok = 0;
|
|
st->snd_wscale = 0;
|
|
st->rcv_wscale = 0;
|
|
st->snd_wnd = 0x1000;
|
|
st->rcv_wnd = 0x1000;
|
|
//make sure you set snd_una = seq (TODO: fix this in module)
|
|
|
|
return st;
|
|
}
|
|
|
|
|
|
int forge_socket_set_state(int sock, struct tcp_state *st)
|
|
{
|
|
struct sockaddr_in sin;
|
|
sin.sin_family = AF_INET;
|
|
sin.sin_addr.s_addr = st->src_ip;
|
|
sin.sin_port = st->sport;
|
|
|
|
// TODO: maybe not everyone wants this?
|
|
st->snd_una = st->seq;
|
|
|
|
/*
|
|
// TOOD: do we need/want this?
|
|
if (setsockopt(sock, SOL_SOCKET, SO_BINDTODEVICE, iface, strlen(iface)) < 0) {
|
|
perror("setsockopt SO_BINDTODEVICE");
|
|
return -1;
|
|
}
|
|
*/
|
|
|
|
int value = 1;
|
|
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &value, sizeof(value)) < 0) {
|
|
perror("setsockopt SO_REUSEADDR");
|
|
return -1;
|
|
}
|
|
|
|
if (setsockopt(sock, SOL_IP, IP_TRANSPARENT, &value, sizeof(value)) < 0) {
|
|
perror("setsockopt IP_TRANSPARENT");
|
|
return -1;
|
|
}
|
|
|
|
if (bind(sock, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
|
|
perror("bind");
|
|
return -1;
|
|
}
|
|
|
|
if (setsockopt(sock, IPPROTO_TCP, TCP_STATE, st, sizeof(struct tcp_state)) < 0) {
|
|
perror("setsockopt TCP_STATE");
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|