/* -*- c-indent-level: 2; c-continued-statement-offset: 2 -*- */ /* * %CopyrightBegin% * * Copyright Ericsson AB 1998-2010. All Rights Reserved. * * The contents of this file are subject to the Erlang Public License, * Version 1.1, (the "License"); you may not use this file except in * compliance with the License. You should have received a copy of the * Erlang Public License along with this software. If not, it can be * retrieved online at http://www.erlang.org/. * * Software distributed under the License is distributed on an "AS IS" * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See * the License for the specific language governing rights and limitations * under the License. * * %CopyrightEnd% */ /* * This file is for internal use within epmd. */ /* This file don't depend on "sys.h" so we have to do some target definitions ourselves */ #ifdef __WIN32__ #define NO_SYSLOG #define NO_SYSCONF #define NO_DAEMON #endif #ifdef VXWORKS #define NO_SYSLOG #define NO_SYSCONF #define NO_DAEMON #define NO_FCNTL #define DONT_USE_MAIN #endif /* ************************************************************************ */ /* Standard includes */ #include <stdio.h> #include <stdlib.h> #include <string.h> #ifdef __WIN32__ # ifndef WINDOWS_H_INCLUDES_WINSOCK2_H # include <winsock2.h> # endif # include <windows.h> # include <process.h> #endif #include <sys/types.h> #include <fcntl.h> #ifdef VXWORKS # include <sys/times.h> # include <time.h> # include <selectLib.h> # include <sysLib.h> # include <sockLib.h> # include <ioLib.h> # include <taskLib.h> # include <rpc/rpc.h> #else /* ! VXWORKS */ #ifndef __WIN32__ # ifdef TIME_WITH_SYS_TIME # include <sys/time.h> # include <time.h> # else # ifdef HAVE_SYS_TIME_H # include <sys/time.h> # else # include <time.h> # endif # endif #endif #endif /* ! VXWORKS */ #if !defined(__WIN32__) # include <netinet/in.h> # include <sys/socket.h> # include <sys/stat.h> # ifdef DEF_INADDR_LOOPBACK_IN_RPC_TYPES_H # include <rpc/types.h> # endif # include <arpa/inet.h> # include <netinet/tcp.h> #endif /* ! WIN32 */ #include <ctype.h> #include <signal.h> #include <errno.h> #ifndef NO_SYSLOG # include <syslog.h> #endif #ifdef SYS_SELECT_H # include <sys/select.h> #endif #ifdef HAVE_UNISTD_H # include <unistd.h> #endif #include <stdarg.h> /* ************************************************************************ */ /* Replace some functions by others by making the function name a macro */ #ifdef __WIN32__ # define close(s) closesocket((s)) # define write(a,b,c) send((a),(b),(c),0) # define read(a,b,c) recv((a),(char *)(b),(c),0) # define sleep(s) Sleep((s) * 1000) # define ioctl(s,r,o) ioctlsocket((s),(r),(o)) #endif /* WIN32 */ #ifdef VXWORKS #define sleep(n) taskDelay((n) * sysClkRateGet()) #endif /* VXWORKS */ #ifdef USE_BCOPY # define memcpy(a, b, c) bcopy((b), (a), (c)) # define memcmp(a, b, c) bcmp((a), (b), (c)) # define memzero(buf, len) bzero((buf), (len)) #else # define memzero(buf, len) memset((buf), '\0', (len)) #endif /* ************************************************************************ */ /* Try to find replacement values for undefined system parameters */ #if defined(__WIN32__) && !defined(EADDRINUSE) # define EADDRINUSE WSAEADDRINUSE #endif #if defined(__WIN32__) && !defined(ECONNABORTED) # define ECONNABORTED WSAECONNABORTED #endif #ifndef SOMAXCONN # define SOMAXCONN 128 #endif /* * How to get max no of file descriptors? We used to use NOFILE from * <sys/param.h>, but that tends to have little relation to reality. * Best is to use sysconf() (POSIX), but we'll just punt if that isn't * available. Start out with a high value because it will also be * used as the number of file descriptors given to select() (it's is * a terrible bug not to have all file descriptors included in the select()). * The value will be adjusted down if FD_SETSIZE is smaller. */ #define MAX_FILES 2048 /* if sysconf() isn't available, or fails */ /* ************************************************************************ */ /* Macros that let us use IPv6 */ #if defined(HAVE_IN6) && defined(AF_INET6) && defined(EPMD6) #define EPMD_SOCKADDR_IN sockaddr_in6 #define FAMILY AF_INET6 #define SET_ADDR_LOOPBACK(addr, af, port) do { \ memset((char*)&(addr), 0, sizeof(addr)); \ (addr).sin6_family = (af); \ (addr).sin6_flowinfo = 0; \ (addr).sin6_addr = in6addr_loopback; \ (addr).sin6_port = htons(port); \ } while(0) #define SET_ADDR_ANY(addr, af, port) do { \ memset((char*)&(addr), 0, sizeof(addr)); \ (addr).sin6_family = (af); \ (addr).sin6_flowinfo = 0; \ (addr).sin6_addr = in6addr_any; \ (addr).sin6_port = htons(port); \ } while(0) #else /* Not IP v6 */ #define EPMD_SOCKADDR_IN sockaddr_in #define FAMILY AF_INET #define SET_ADDR_LOOPBACK(addr, af, port) do { \ memset((char*)&(addr), 0, sizeof(addr)); \ (addr).sin_family = (af); \ (addr).sin_addr.s_addr = htonl(INADDR_LOOPBACK); \ (addr).sin_port = htons(port); \ } while(0) #define SET_ADDR_ANY(addr, af, port) do { \ memset((char*)&(addr), 0, sizeof(addr)); \ (addr).sin_family = (af); \ (addr).sin_addr.s_addr = htonl(INADDR_ANY); \ (addr).sin_port = htons(port); \ } while(0) #endif /* Not IP v6 */ /* ************************************************************************ */ /* Our own definitions */ #define EPMD_FALSE 0 #define EPMD_TRUE 1 /* If no activity we let select() return every IDLE_TIMEOUT second A file descriptor that are idle for CLOSE_TIMEOUT seconds and isn't a ALIVE socket is probably hanging and we close it */ #define IDLE_TIMEOUT 5 #define CLOSE_TIMEOUT 60 /* We save the name of nodes that are unregistered. If a new node register the name we want to increment the "creation", a constant 1..3. But we put an limit to this saving to keep the lookup fast and not to leak memory. */ #define MAX_UNREG_COUNT 1000 #define DEBUG_MAX_UNREG_COUNT 5 /* Maximum length of a node name == atom name */ #define MAXSYMLEN 255 #define INBUF_SIZE 1024 #define OUTBUF_SIZE 1024 #define get_int16(s) ((((unsigned char*) (s))[0] << 8) | \ (((unsigned char*) (s))[1])) #define put_int16(i, s) {((unsigned char*)(s))[0] = ((i) >> 8) & 0xff; \ ((unsigned char*)(s))[1] = (i) & 0xff;} /* ************************************************************************ */ /* Stuctures used by server */ typedef struct { int fd; /* File descriptor */ unsigned char open; /* TRUE if open */ unsigned char keep; /* Don't close when sent reply */ unsigned char local_peer; /* The peer of this connection is via loopback interface */ unsigned got; /* # of bytes we have got */ unsigned want; /* Number of bytes we want */ char *buf; /* The remaining buffer */ time_t mod_time; /* Last activity on this socket */ } Connection; struct enode { struct enode *next; int fd; /* The socket in use */ unsigned short port; /* Port number of Erlang node */ char symname[MAXSYMLEN+1]; /* Name of the Erlang node */ short creation; /* Started as a random number 1..3 */ char nodetype; /* 77 = normal erlang node 72 = hidden (c-node */ char protocol; /* 0 = tcp/ipv4 */ unsigned short highvsn; /* 0 = OTP-R3 erts-4.6.x, 1 = OTP-R4 erts-4.7.x*/ unsigned short lowvsn; int extralen; char extra[MAXSYMLEN+1]; }; typedef struct enode Node; typedef struct { Node *reg; Node *unreg; Node *unreg_tail; int unreg_count; } Nodes; /* This is the structure with all variables needed to pass on to all functions. This makes this program reentrant */ typedef struct { int port; int debug; int silent; int is_daemon; int brutal_kill; unsigned packet_timeout; unsigned delay_accept; unsigned delay_write; int max_conn; int active_conn; char *progname; Connection *conn; Nodes nodes; fd_set orig_read_mask; int listenfd; char **argv; } EpmdVars; void dbg_printf(EpmdVars*,int,const char*,...); void dbg_tty_printf(EpmdVars*,int,const char*,...); void dbg_perror(EpmdVars*,const char*,...); void kill_epmd(EpmdVars*); void epmd_call(EpmdVars*,int); void run(EpmdVars*); void epmd_cleanup_exit(EpmdVars*, int); int epmd_conn_close(EpmdVars*,Connection*); void stop_cli(EpmdVars *g, char *name); #ifdef DONT_USE_MAIN int start_epmd(char *,char *,char *,char *,char *,char *,char *,char *,char *,char *); int epmd(int,char **); int epmd_dbg(int,int); #endif