1ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 2ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--------------------------------------------------------------------*/ 3ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--- A simple program to listen for valgrind logfile data. ---*/ 4ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--- valgrind-listener.c ---*/ 5ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--------------------------------------------------------------------*/ 6ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 7ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* 8ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown This file is part of Valgrind, a dynamic binary instrumentation 9ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown framework. 10ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 11436e89c602e787e7a27dd6624b09beed41a0da8aDmitriy Ivanov Copyright (C) 2000-2013 Julian Seward 12ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown jseward@acm.org 13ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 14ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown This program is free software; you can redistribute it and/or 15ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown modify it under the terms of the GNU General Public License as 16ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown published by the Free Software Foundation; either version 2 of the 17ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown License, or (at your option) any later version. 18ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 19ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown This program is distributed in the hope that it will be useful, but 20ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown WITHOUT ANY WARRANTY; without even the implied warranty of 21ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 22ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown General Public License for more details. 23ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 24ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown You should have received a copy of the GNU General Public License 25ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown along with this program; if not, write to the Free Software 26ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 27ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 02111-1307, USA. 28ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 29ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown The GNU General Public License is contained in the file COPYING. 30ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown*/ 31ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 32ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 33ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*---------------------------------------------------------------*/ 34ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 35ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* Include valgrind headers before system headers to avoid problems 36ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown with the system headers #defining things which are used as names 37ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown of structure members in vki headers. */ 38ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 39ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_core_basics.h" 40ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_core_libcassert.h" // For VG_BUGS_TO 41ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_core_vki.h" // Avoids warnings from 42ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // pub_core_libcfile.h 43ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "pub_core_libcfile.h" // For VG_CLO_DEFAULT_LOGPORT 44ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 45ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <stdio.h> 46ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <unistd.h> 47ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <string.h> 48ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <time.h> 49ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <fcntl.h> 50ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <stdlib.h> 51ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <signal.h> 52ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <sys/poll.h> 53ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <sys/types.h> 54ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <sys/socket.h> 55ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <netinet/in.h> 56ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 57ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 58ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*---------------------------------------------------------------*/ 59ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 60ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* The maximum allowable number concurrent connections. */ 61ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#define M_CONNECTIONS 50 62ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 63ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 64ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*---------------------------------------------------------------*/ 65ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 66ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown__attribute__ ((noreturn)) 67436e89c602e787e7a27dd6624b09beed41a0da8aDmitriy Ivanovstatic void panic ( const char* str ) 68ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 69ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fprintf(stderr, 70ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "\nvalgrind-listener: the " 71ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "'impossible' happened:\n %s\n", str); 72ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fprintf(stderr, 73ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Please report this bug at: %s\n\n", VG_BUGS_TO); 74ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown exit(1); 75ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 76ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 77ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown__attribute__ ((noreturn)) 78436e89c602e787e7a27dd6624b09beed41a0da8aDmitriy Ivanovstatic void my_assert_fail ( const char* expr, const char* file, int line, const char* fn ) 79ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 80ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fprintf(stderr, 81ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "\nvalgrind-listener: %s:%d (%s): Assertion '%s' failed.\n", 82ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown file, line, fn, expr ); 83ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fprintf(stderr, 84ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Please report this bug at: %s\n\n", VG_BUGS_TO); 85ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown exit(1); 86ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 87ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 88ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#undef assert 89ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 90ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#define assert(expr) \ 91ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown ((void) ((expr) ? 0 : \ 92ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown (my_assert_fail (VG_STRINGIFY(expr), \ 93ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown __FILE__, __LINE__, \ 94ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown __PRETTY_FUNCTION__), 0))) 95ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 96ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 97ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*---------------------------------------------------------------*/ 98ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 99ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* holds the fds for connections; zero if slot not in use. */ 100ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownint conn_count = 0; 101ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownint conn_fd[M_CONNECTIONS]; 102ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstruct pollfd conn_pollfd[M_CONNECTIONS]; 103ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 104ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 105ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void set_nonblocking ( int sd ) 106ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 107ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int res; 108ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = fcntl(sd, F_GETFL); 109ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = fcntl(sd, F_SETFL, res | O_NONBLOCK); 110ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (res != 0) { 111ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("fcntl failed"); 112ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("set_nonblocking"); 113ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 114ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 115ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 116ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void set_blocking ( int sd ) 117ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 118ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int res; 119ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = fcntl(sd, F_GETFL); 120ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = fcntl(sd, F_SETFL, res & ~O_NONBLOCK); 121ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (res != 0) { 122ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("fcntl failed"); 123ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("set_blocking"); 124ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 125ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 126ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 127ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 128ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void copyout ( char* buf, int nbuf ) 129ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 130ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int i; 131ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < nbuf; i++) { 132ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (buf[i] == '\n') { 133ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fprintf(stdout, "\n(%d) ", conn_count); 134ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } else { 135ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown __attribute__((unused)) size_t ignored 136ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown = fwrite(&buf[i], 1, 1, stdout); 137ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 138ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 139ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fflush(stdout); 140ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 141ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 142ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic int read_from_sd ( int sd ) 143ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 144ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown char buf[100]; 145ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int n; 146ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 147ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown set_blocking(sd); 148ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown n = read(sd, buf, 99); 149ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (n <= 0) return 0; /* closed */ 150ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown copyout(buf, n); 151ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 152ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown set_nonblocking(sd); 153ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown while (1) { 154ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown n = read(sd, buf, 100); 155ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (n <= 0) return 1; /* not closed */ 156ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown copyout(buf, n); 157ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 158ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 159ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 160ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 161ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void snooze ( void ) 162ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 163ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown struct timespec req; 164ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown req.tv_sec = 0; 165ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown req.tv_nsec = 200 * 1000 * 1000; 166ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown nanosleep(&req,NULL); 167ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 168ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 169ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 170ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* returns 0 if invalid, else port # */ 171436e89c602e787e7a27dd6624b09beed41a0da8aDmitriy Ivanovstatic int atoi_portno ( const char* str ) 172ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 173ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int n = 0; 174ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown while (1) { 175ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (*str == 0) 176ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown break; 177ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (*str < '0' || *str > '9') 178ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return 0; 179ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown n = 10*n + (int)(*str - '0'); 180ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown str++; 181ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (n >= 65536) 182ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return 0; 183ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 184ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (n < 1024) 185ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return 0; 186ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return n; 187ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 188ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 189ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 190ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void usage ( void ) 191ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 192ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fprintf(stderr, 193ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "\n" 194ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "usage is:\n" 195ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "\n" 196ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown " valgrind-listener [--exit-at-zero|-e] [port-number]\n" 197ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "\n" 198ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown " where --exit-at-zero or -e causes the listener to exit\n" 199ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown " when the number of connections falls back to zero\n" 200ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown " (the default is to keep listening forever)\n" 201ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "\n" 202ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown " port-number is the default port on which to listen for\n" 203ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown " connections. It must be between 1024 and 65535.\n" 204ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown " Current default is %d.\n" 205ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "\n" 206ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown , 207ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown VG_CLO_DEFAULT_LOGPORT 208ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown ); 209ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown exit(1); 210ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 211ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 212ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 213436e89c602e787e7a27dd6624b09beed41a0da8aDmitriy Ivanovstatic void banner ( const char* str ) 214ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 215ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown time_t t; 216ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown t = time(NULL); 217ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("valgrind-listener %s at %s", str, ctime(&t)); 218ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fflush(stdout); 219ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 220ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 221ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 222ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void exit_routine ( void ) 223ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 224ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown banner("exited"); 225ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown exit(0); 226ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 227ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 228ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 229ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void sigint_handler ( int signo ) 230ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 231ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown exit_routine(); 232ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 233ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 234ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 235ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownint main (int argc, char** argv) 236ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 237ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int i, j, k, res, one; 238ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int main_sd, new_sd; 239ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown socklen_t client_len; 240ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown struct sockaddr_in client_addr, server_addr; 241ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 242ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown char /*bool*/ exit_when_zero = 0; 243ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int port = VG_CLO_DEFAULT_LOGPORT; 244ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 245ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 1; i < argc; i++) { 246ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (0==strcmp(argv[i], "--exit-at-zero") 247ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown || 0==strcmp(argv[i], "-e")) { 248ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown exit_when_zero = 1; 249ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 250ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown else 251ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (atoi_portno(argv[i]) > 0) { 252ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown port = atoi_portno(argv[i]); 253ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 254ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown else 255ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown usage(); 256ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 257ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 258ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown banner("started"); 259ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown signal(SIGINT, sigint_handler); 260ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 261ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_count = 0; 262ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < M_CONNECTIONS; i++) 263ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_fd[i] = 0; 264ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 265ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* create socket */ 266ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown main_sd = socket(AF_INET, SOCK_STREAM, 0); 267ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (main_sd < 0) { 268ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("cannot open socket "); 269ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("main -- create socket"); 270ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 271ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 272ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* allow address reuse to avoid "address already in use" errors */ 273ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 274ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown one = 1; 275ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (setsockopt(main_sd, SOL_SOCKET, SO_REUSEADDR, 276ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown &one, sizeof(int)) < 0) { 277ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("cannot enable address reuse "); 278ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("main -- enable address reuse"); 279ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 280ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 281ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* bind server port */ 282ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown server_addr.sin_family = AF_INET; 283ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown server_addr.sin_addr.s_addr = htonl(INADDR_ANY); 284ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown server_addr.sin_port = htons(port); 285ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 286ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (bind(main_sd, (struct sockaddr *) &server_addr, 287ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown sizeof(server_addr) ) < 0) { 288ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("cannot bind port "); 289ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("main -- bind port"); 290ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 291ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 292ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = listen(main_sd,M_CONNECTIONS); 293ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (res != 0) { 294ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("listen failed "); 295ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("main -- listen"); 296ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 297ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 298ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown while (1) { 299ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 300ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown snooze(); 301ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 302ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* enquire, using poll, whether there is any activity available on 303ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown the main socket descriptor. If so, someone is trying to 304ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown connect; get the fd and add it to our table thereof. */ 305ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown { struct pollfd ufd; 306ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown while (1) { 307ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown ufd.fd = main_sd; 308ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown ufd.events = POLLIN; 309ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown ufd.revents = 0; 310ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = poll(&ufd, 1, 0); 311ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (res == 0) break; 312ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 313ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* ok, we have someone waiting to connect. Get the sd. */ 314ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown client_len = sizeof(client_addr); 315ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown new_sd = accept(main_sd, (struct sockaddr *)&client_addr, 316ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown &client_len); 317ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (new_sd < 0) { 318ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("cannot accept connection "); 319ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("main -- accept connection"); 320ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 321ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 322ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* find a place to put it. */ 323ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(new_sd > 0); 324ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < M_CONNECTIONS; i++) 325ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (conn_fd[i] == 0) 326ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown break; 327ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 328ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (i >= M_CONNECTIONS) { 329ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fprintf(stderr, "Too many concurrent connections. " 330ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Increase M_CONNECTIONS and recompile.\n"); 331ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("main -- too many concurrent connections"); 332ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 333ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 334ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_fd[i] = new_sd; 335ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_count++; 336ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("\n(%d) -------------------- CONNECT " 337ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "--------------------\n(%d)\n(%d) ", 338ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_count, conn_count, conn_count); 339ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fflush(stdout); 340ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } /* while (1) */ 341ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 342ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 343ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* We've processed all new connect requests. Listen for changes 344ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown to the current set of fds. */ 345ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown j = 0; 346ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < M_CONNECTIONS; i++) { 347ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (conn_fd[i] == 0) 348ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown continue; 349ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_pollfd[j].fd = conn_fd[i]; 350ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_pollfd[j].events = POLLIN /* | POLLHUP | POLLNVAL */; 351ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_pollfd[j].revents = 0; 352ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown j++; 353ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 354ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 355ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = poll(conn_pollfd, j, 0 /* return immediately. */ ); 356ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (res < 0) { 357ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown perror("poll(main) failed"); 358ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown panic("poll(main) failed"); 359ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 360ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 361ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* nothing happened. go round again. */ 362ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (res == 0) { 363ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown continue; 364ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 365ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 366ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* inspect the fds. */ 367ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < j; i++) { 368ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 369ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (conn_pollfd[i].revents & POLLIN) { 370ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* data is available on this fd */ 371ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown res = read_from_sd(conn_pollfd[i].fd); 372ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 373ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (res == 0) { 374ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* the connection has been closed. */ 375ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown close(conn_pollfd[i].fd); 376ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* this fd has been closed or otherwise gone bad; forget 377ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown about it. */ 378ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (k = 0; k < M_CONNECTIONS; k++) 379ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (conn_fd[k] == conn_pollfd[i].fd) 380ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown break; 381ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(k < M_CONNECTIONS); 382ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_fd[k] = 0; 383ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_count--; 384ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("\n(%d) ------------------- DISCONNECT " 385ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "-------------------\n(%d)\n(%d) ", 386ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown conn_count, conn_count, conn_count); 387ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fflush(stdout); 388ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (conn_count == 0 && exit_when_zero) { 389ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("\n"); 390ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown fflush(stdout); 391ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown exit_routine(); 392ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 393ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 394ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 395ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 396ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } /* for (i = 0; i < j; i++) */ 397ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 398ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } /* while (1) */ 399ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 400ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* NOTREACHED */ 401ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 402ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 403ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 404ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--------------------------------------------------------------------*/ 405ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--- end valgrind-listener.c ---*/ 406ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*--------------------------------------------------------------------*/ 407