N-sim Emulation and simulation of Wireless Sensor Networks    Home    Project Page    Download    CVS    Installation    Configuration    Plug-ins  Hosted by

/home/brennan/n-sim/Vaike/linux/system-addons/system/soap/sensor_ipc.c Go to the documentation of this file. /* * Copyright 2007. Los Alamos National Security, LLC. This material * was produced under U.S. Government contract DE-AC52-06NA25396 for * Los Alamos National Laboratory (LANL), which is operated by Los * Alamos National Security, LLC, for the Department of Energy. The * U.S. Government has rights to use, reproduce, and distribute this * software. NEITHER THE GOVERNMENT NOR LOS ALAMOS NATIONAL SECURITY, * LLC, MAKES ANY WARRANTY, EXPRESS OR IMPLIED, OR ASSUMES ANY LEGAL * LIABILITY FOR THE USE OF THIS SOFTWARE. If software is modified to * produce derivative works, such modified software should be clearly * marked, so as not to confuse it with the version available from LANL. * * Additionally, this program is free software; you can redistribute * it and/or modify it under the terms of the GNU General Public * License as published by the Free Software Foundation; version 2 of * the License. Accordingly, this program is distributed in the hope * it will be useful, but WITHOUT ANY WARRANTY; without even the * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR * PURPOSE. See the GNU General Public License for more details. */ #define _ipc_impl_ #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <errno.h> #include <string.h> #include <sys/time.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <netinet/sctp.h> #include <arpa/inet.h> #include <netdb.h> #include <limits.h> #include <signal.h> #include <pthread.h> #include "minisoap.h" #include "sensor_ipc.h" enum msgtype { DATA, CMD }; pthread_t data_thread, cmd_thread; static int data_send_fd, data_recv_fd; struct sockaddr_in data_send_addr; static unsigned int data_recv_port; static int cmd_send_fd, cmd_recv_fd; struct sockaddr_in cmd_send_addr; static unsigned int cmd_recv_port = 0; static unsigned int initialized = 0; volatile static unsigned int __shutdown__; static void sigint_handler(int s) { __shutdown__ = 1; } int quitting(void) { return __shutdown__ >= 1; } long long get_time(void) { struct timeval now; gettimeofday(&now, NULL); return (now.tv_sec * 1000000) + now.tv_usec; } int initialize_ipc(int port) { int error, yes = 1; struct hostent *he; if (!initialized) { __shutdown__ = 0; signal(SIGINT, sigint_handler); initialized = 1; } /* socket for sending data */ if ((data_send_fd = socket(PF_INET, SOCK_RDM, IPPROTO_SCTP)) < 0) { perror("IPC: socket"); return data_send_fd; } if ((error = setsockopt(data_send_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int))) < 0) { perror("IPC: setsockopt (SO_REUSEADDR)"); return error; } if ((he = gethostbyname("127.0.0.1")) == NULL) { herror("gethostbyname"); return -1; } data_send_addr.sin_family = AF_INET; data_send_addr.sin_port = htons(IPC_DATA_PORT); data_send_addr.sin_addr = *((struct in_addr *)he->h_addr); memset(data_send_addr.sin_zero, '\0', sizeof(data_send_addr.sin_zero)); if ((error = bind(data_send_fd, (struct sockaddr *)&data_send_addr, sizeof(data_send_addr))) < 0) { perror("IPC: bind"); return error; } /* socket for sending commands */ if ((cmd_send_fd = socket(PF_INET, SOCK_RDM, IPPROTO_SCTP)) < 0) { perror("IPC: socket"); return cmd_send_fd; } if ((error = setsockopt(cmd_send_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int))) < 0) { perror("IPC: setsockopt (SO_REUSEADDR)"); return error; } cmd_send_addr.sin_family = AF_INET; cmd_send_addr.sin_port = htons(IPC_CMD_PORT); cmd_send_addr.sin_addr = *((struct in_addr *)he->h_addr); memset(cmd_send_addr.sin_zero, '\0', sizeof(cmd_send_addr.sin_zero)); if ((error = bind(cmd_send_fd, (struct sockaddr *)&cmd_send_addr, sizeof(cmd_send_addr))) < 0) { perror("IPC: bind"); return error; } if (port > 0) { data_recv_port = port; /* socket for receiving data */ if ((data_recv_fd = socket(PF_INET, SOCK_RDM, IPPROTO_SCTP)) < 0) { perror("IPC: socket"); return data_recv_fd; } } /* cmd recv socket setup only in register_cmd_handler() */ return 0; } static void disconnect(int fd) { struct sockaddr_in unspec; unspec.sin_family = AF_UNSPEC; unspec.sin_port = 0; unspec.sin_addr.s_addr = INADDR_ANY; memset(unspec.sin_zero, '\0', sizeof(unspec.sin_zero)); connect(fd, (struct sockaddr*)&unspec, sizeof(unspec)); } static int ipc_send(enum msgtype t, SoapEvent *evt) { int pkt_len; int error, total = 0; int fd, remaining = pkt_len; struct sockaddr_in their_addr; char buf[MAX_EVENT_LEN + 1]; if (evt == NULL) return -1; SoapEvent_string(evt, buf, MAX_EVENT_LEN); pkt_len = strlen(buf); if (t == DATA) { fd = data_send_fd; memcpy(&their_addr, &data_send_addr, sizeof(struct sockaddr_in)); } else if (t == CMD) { fd = cmd_send_fd; memcpy(&their_addr, &cmd_send_addr, sizeof(struct sockaddr_in)); } else return -1; if ((error = connect(fd, (struct sockaddr*)&their_addr, sizeof(their_addr))) < 0) return error; if ((error = send(fd, &pkt_len, sizeof(int), 0)) < 0) return error; while (total < pkt_len) { if ((error = send(fd, buf + total, remaining, 0)) < 0) break; total += error; remaining -= error; } disconnect(fd); return error; } inline int ipc_data_send(SoapEvent *evt) { return ipc_send(DATA, evt); } inline int ipc_cmd_send(SoapEvent *evt) { return ipc_send(CMD, evt); } static int ipc_recv(enum msgtype t, SoapEvent *evt) { int len = MAX_EVENT_LEN + 1; char *buf = (char*) malloc(len); int fd, buf_len = len; int pkt_len = len; int error, total = 0; int remaining = pkt_len; struct sockaddr_in their_addr; socklen_t addr_len = sizeof(struct sockaddr); memset(buf, '\0', pkt_len); len = 0; if (t == DATA) fd = data_recv_fd; else if (t == CMD) fd = cmd_recv_fd; else return -1; if (fd == 0 || evt == NULL) return -1; if ((error = recvfrom(fd, &pkt_len, sizeof(int), 0, (struct sockaddr*)&their_addr, &addr_len)) < 0) return error; /* recv only from this address until complete */ if ((error = connect(fd, (struct sockaddr*)&their_addr, sizeof(their_addr))) < 0) return error; if (pkt_len >= buf_len) { if ((buf = realloc(buf, pkt_len + 1)) == NULL) return -1; memset(buf, '\0', pkt_len + 1); } remaining = pkt_len; while (total < pkt_len) { if ((error = recv(fd, buf + total, remaining, 0)) <= 0) break; total += error; remaining -= error; } buf[total] = '\0'; len = total + 1; string_SoapEvent(buf, evt); disconnect(fd); return error; } inline int ipc_data_recv(SoapEvent *evt) { return ipc_recv(DATA, evt); } inline int ipc_cmd_recv(SoapEvent *evt) { return ipc_recv(CMD, evt); } static void *cmd_listener(void *callback) { while (1) { SoapEvent evt; ipc_recv(CMD, &evt); ((ipc_recv_t) callback)(&evt); } return NULL; } int register_cmd_handler(int port, ipc_recv_t handler) { cmd_recv_port = port; /* socket for receiving commands */ if ((cmd_recv_fd = socket(PF_INET, SOCK_RDM, IPPROTO_SCTP)) < 0) { perror("IPC: socket"); return cmd_recv_fd; } return pthread_create(&cmd_thread, NULL, cmd_listener, (void*)handler); } static void *data_listener(void *callback) { while (1) { SoapEvent evt; ipc_recv(DATA, &evt); ((ipc_recv_t) callback)(&evt); } return NULL; } int register_data_handler(int port, ipc_recv_t handler) { data_recv_port = port; /* socket for receiving data */ if ((data_recv_fd = socket(PF_INET, SOCK_RDM, IPPROTO_SCTP)) < 0) { perror("IPC: socket"); return cmd_recv_fd; } return pthread_create(&data_thread, NULL, data_listener, (void*)handler); }

© 2007, Los Alamos National Security, LLC.