+/* (c) 2008, Jack Lange <jarusl@cs.northwestern.edu> */
+/* (c) 2008, Lei Xia <xiaxlei@gmail.com>
+/* (c) 2008, The V3VEE Project <http://www.v3vee.org> */
+
#include <geekos/socket.h>
#include <geekos/malloc.h>
-#include <palacios/vmm_types.h>
#include <geekos/ne2k.h>
#include <uip/uip.h>
#include <uip/uip_arp.h>
#include <geekos/vmm_stubs.h>
-
-#define NULL (void *)0
+#include <geekos/debug.h>
+#include <geekos/timer.h>
#define BUF ((struct uip_eth_hdr *)&uip_buf[0])
struct socket sockets[MAX_SOCKS];
-extern void* memcpy(void *dst, const void* src, int n);
+void socket_appcall(void);
+#ifndef UIP_APPCALL
+#define UIP_APPCALL socket_appcall
+#endif /* UIP_APPCALL */
+
-int Packet_Received(struct NE2K_Packet_Info* info, uchar_t *pkt) ;
-void init_network() {
+static int Packet_Received(struct NE2K_Packet_Info* info, uchar_t *pkt);
+static void periodic_caller(int timer_id);
+
+void init_socket_layer() {
int i = 0;
+ bool iflag;
+
for (i = 0; i < MAX_SOCKS; i++) {
sockets[i].in_use = 0;
sockets[i].send_buf = NULL;
sockets[i].recv_buf = NULL;
+ sockets[i].state = CLOSED;
}
+
+
//initiate uIP
uip_init();
uip_arp_init();
//setup device driver
Init_Ne2k(&Packet_Received);
+ iflag = Begin_Int_Atomic();
+ Start_Timer(2, periodic_caller);
+ End_Int_Atomic(iflag);
}
+
+
+
+void set_ip_addr(uchar_t addr[4]) {
+ uip_ipaddr_t ipaddr;
+ uip_ipaddr(ipaddr, addr[0], addr[1], addr[2], addr[3]); /* Local IP address */
+ uip_sethostaddr(ipaddr);
+}
+
+
static int allocate_socket_fd() {
int i = 0;
free_ring_buffer(sockets[sockfd].recv_buf);
sockets[sockfd].send_buf = NULL;
sockets[sockfd].recv_buf = NULL;
+ sockets[sockfd].state = CLOSED;
}
return 0;
}
+
+static void periodic_caller(int timer_id) {
+ int i;
+ //handle the periodic calls of uIP
+
+ //PrintBoth("Timer CALLBACK handler\n");
+
+ for(i = 0; i < UIP_CONNS; ++i) {
+ uip_periodic(i);
+ if(uip_len > 0) {
+ //devicedriver_send();
+ PrintBoth("Sending Packet\n");
+ NE2K_Transmit(uip_len);
+ }
+ }
+ for(i = 0; i < UIP_UDP_CONNS; i++) {
+ uip_udp_periodic(i);
+ if(uip_len > 0) {
+ //devicedriver_send();
+ NE2K_Transmit(uip_len);
+ }
+ }
+}
+
+
int connect(const uchar_t ip_addr[4], ushort_t port) {
int sockfd = -1;
sockfd = allocate_socket_fd();
- uip_ipaddr_t ipaddr;
+ static uip_ipaddr_t ipaddr;
if (sockfd == -1) {
return -1;
uip_ipaddr(&ipaddr, ip_addr[0], ip_addr[1], ip_addr[2], ip_addr[3]);
- sockets[sockfd].con = uip_connect((uip_ipaddr_t *)&ip_addr, htons(port));
+ sockets[sockfd].con = uip_connect(&ipaddr, htons(port));
+
if (sockets[sockfd].con == NULL){
- release_socket_fd(sockfd);
- return -1;
+ release_socket_fd(sockfd);
+ return -1;
}
+
+ PrintBoth("Connection start\n");
+ Wait(&(sockets[sockfd].recv_wait_queue));
+
+
+ PrintBoth("Connected\n");
+
+
+
+
return sockfd;
}
struct socket *sock = get_socket_from_fd(sockfd);
+
// here we need some lock mechnism, just disable interrupt may not work properly because recv() will be run as a kernel thread
buf_read:
recvlen = rb_read(sock->recv_buf, buf, len);
return recvlen;
}
-void timer_int_Handler(struct Interrupt_State * state){
- int i;
- //handle the periodic calls of uIP
- for(i = 0; i < UIP_CONNS; ++i) {
- uip_periodic(i);
- if(uip_len > 0) {
- //devicedriver_send();
- NE2K_Transmit(uip_len);
- }
- }
- for(i = 0; i < UIP_UDP_CONNS; i++) {
- uip_udp_periodic(i);
- if(uip_len > 0) {
- //devicedriver_send();
- NE2K_Transmit(uip_len);
- }
- }
-}
+
// a series of utilities to handle conncetion states
-static void connected(int sockfd){
+static void connected(int sockfd) {
+ struct socket * sock = get_socket_from_fd(sockfd);
+
+ PrintBoth("Connected Interrupt\n");
+ sock->state = ESTABLISHED;
+
+ Wake_Up(&(sock->recv_wait_queue));
}
static void closed(int sockfd){
}
// not finished yet
-static void
-senddata(int sockfd){
+static void send_to_driver(int sockfd) {
+
+ PrintBoth("Sending data to driver\n");
+
+}
+
+
+
+int send(int sockfd, void * buf, uint_t len) {
struct socket * sock = get_socket_from_fd(sockfd);
- int mss = uip_mss();
- int pending_bytes = rb_data_len(sock->send_buf);
- int len = (mss < pending_bytes) ? mss: pending_bytes;
+ //int mss = uip_mss();
+ //int pending_bytes = rb_data_len(sock->send_buf);
+ //int len = (mss < pending_bytes) ? mss: pending_bytes;
int bytes_read = 0;
uchar_t * send_buf = uip_appdata;
if (bytes_read == 0) {
// no packet for send
- return;
+ return -1;
}
uip_send(send_buf, len);
+
+ return len;
}
//get the socket id by the local tcp port
-static int get_socket_from_port(ushort_t lport) {
+static int get_socket_from_port(ushort_t lport) {
int i;
- for (i = 0; i<MAX_SOCKS; i++){
- if (sockets[i].con->lport == lport)
- return i;
+
+ for (i = 0; i < MAX_SOCKS; i++){
+ if (sockets[i].con->lport == lport) {
+ return i;
+ }
}
return -1;
}
-void
-socket_appcall(void)
-{
- int sockfd;
+void socket_appcall(void) {
+
+ int sockfd;
+
sockfd = get_socket_from_port(uip_conn->lport);
- if (sockfd == -1) return;
-
- if(uip_connected()) {
- connected(sockfd);
+ PrintBoth("Appcall\n");
+
+
+ if (sockfd == -1) {
+ return;
+ }
+
+ if (uip_connected()) {
+ connected(sockfd);
+
}
- if(uip_closed() ||uip_aborted() ||uip_timedout()) {
+ if (uip_closed() ||uip_aborted() ||uip_timedout()) {
closed(sockfd);
return;
}
- if(uip_acked()) {
+ if (uip_acked()) {
acked(sockfd);
}
- if(uip_newdata()) {
+ if (uip_newdata()) {
newdata(sockfd);
}
- if(uip_rexmit() ||
- uip_newdata() ||
- uip_acked() ||
- uip_connected() ||
- uip_poll()) {
- senddata(sockfd);
+ if (uip_rexmit() ||
+ uip_newdata() ||
+ uip_acked() ||
+ uip_connected() ||
+ uip_poll()) {
+ send_to_driver(sockfd);
}
}
-int Packet_Received(struct NE2K_Packet_Info* info, uchar_t *pkt)
-{
- uip_len = info->size;
-
- // for (i = 0; i < info->size; i++) {
- // uip_buf[i] = *(pkt + i);
- //}
-
- memcpy(uip_buf, pkt, uip_len);
- Free(pkt);
- if(BUF->type == htons(UIP_ETHTYPE_ARP)) {
- uip_arp_arpin();
- if (uip_len > 0){
- //ethernet_devicedriver_send();
- NE2K_Transmit(uip_len);
- }
- } else {
- uip_arp_ipin();
- uip_input();
- if(uip_len > 0) {
- uip_arp_out();
- //ethernet_devicedriver_send();
- NE2K_Transmit(uip_len);
- }
- }
- return 0;
+
+static int Packet_Received(struct NE2K_Packet_Info * info, uchar_t * pkt) {
+ //int i;
+ uip_len = info->size;
+
+ // for (i = 0; i < info->size; i++) {
+ // uip_buf[i] = *(pkt + i);
+ //}
+
+ PrintBoth("Packet REceived\n");
+
+ memcpy(uip_buf, pkt, uip_len);
+
+
+ Free(pkt);
+
+ if (BUF->type == htons(UIP_ETHTYPE_ARP)) {
+ PrintBoth("ARP PACKET\n");
+
+ uip_arp_arpin();
+
+
+ if (uip_len > 0) {
+ PrintBoth("Transmitting\n");
+ NE2K_Transmit(uip_len);
+ }
+
+
+ } else {
+ PrintBoth("Data PACKET\n");
+ uip_arp_ipin();
+ uip_input();
+
+
+ if (uip_len > 0) {
+ PrintBoth("Transmitting\n");
+ uip_arp_out();
+ NE2K_Transmit(uip_len);
+ }
+
+ }
+
+ return 0;
}
