| 1 | /* INDI Server for protocol version 1.7.
 | 
|---|
| 2 |  * Copyright (C) 2007 Elwood C. Downey ecdowney@clearskyinstitute.com
 | 
|---|
| 3 | 
 | 
|---|
| 4 |     This library is free software; you can redistribute it and/or
 | 
|---|
| 5 |     modify it under the terms of the GNU Lesser General Public
 | 
|---|
| 6 |     License as published by the Free Software Foundation; either
 | 
|---|
| 7 |     version 2.1 of the License, or (at your option) any later version.
 | 
|---|
| 8 | 
 | 
|---|
| 9 |     This library is distributed in the hope that it will be useful,
 | 
|---|
| 10 |     but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
|---|
| 11 |     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 | 
|---|
| 12 |     Lesser General Public License for more details.
 | 
|---|
| 13 | 
 | 
|---|
| 14 |     You should have received a copy of the GNU Lesser General Public
 | 
|---|
| 15 |     License along with this library; if not, write to the Free Software
 | 
|---|
| 16 |     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 | 
|---|
| 17 | 
 | 
|---|
| 18 |  * argv lists names of Driver programs to run or sockets to connect for Devices.
 | 
|---|
| 19 |  * Drivers are restarted if they exit or connection closes.
 | 
|---|
| 20 |  * Each local Driver's stdin/out are assumed to provide INDI traffic and are
 | 
|---|
| 21 |  *   connected here via pipes. Local Drivers' stderr are connected to our
 | 
|---|
| 22 |  *   stderr with date stamp and driver name prepended.
 | 
|---|
| 23 |  * We only support Drivers that advertise support for one Device. The problem
 | 
|---|
| 24 |  *   with multiple Devices in one Driver is without a way to know what they
 | 
|---|
| 25 |  *   _all_ are there is no way to avoid sending all messages to all Drivers.
 | 
|---|
| 26 |  * Outbound messages are limited to Devices and Properties seen inbound.
 | 
|---|
| 27 |  *   Messages to Devices on sockets always include Device so the chained
 | 
|---|
| 28 |  *   indiserver will only pass back info from that Device.
 | 
|---|
| 29 |  * All newXXX() received from one Client are echoed to all other Clients who
 | 
|---|
| 30 |  *   have shown an interest in the same Device and property.
 | 
|---|
| 31 |  *
 | 
|---|
| 32 |  * Implementation notes:
 | 
|---|
| 33 |  *
 | 
|---|
| 34 |  * We fork each driver and open a server socket listening for INDI clients.
 | 
|---|
| 35 |  * Then forever we listen for new clients and pass traffic between clients and
 | 
|---|
| 36 |  * drivers, subject to optimizations based on sniffing messages for matching
 | 
|---|
| 37 |  * Devices and Properties. Since one message might be destined to more than
 | 
|---|
| 38 |  * one client or device, they are queued and only removed after the last
 | 
|---|
| 39 |  * consumer is finished. XMLEle are converted to linear strings before being
 | 
|---|
| 40 |  * sent to optimize write system calls and avoid blocking to slow clients.
 | 
|---|
| 41 |  * Clients that get more than maxqsiz bytes behind are shut down.
 | 
|---|
| 42 |  */
 | 
|---|
| 43 | 
 | 
|---|
| 44 | #include <stdio.h>
 | 
|---|
| 45 | #include <stdlib.h>
 | 
|---|
| 46 | #include <stdarg.h>
 | 
|---|
| 47 | #include <signal.h>
 | 
|---|
| 48 | #include <string.h>
 | 
|---|
| 49 | #include <errno.h>
 | 
|---|
| 50 | #include <unistd.h>
 | 
|---|
| 51 | #include <time.h>
 | 
|---|
| 52 | #include <fcntl.h>
 | 
|---|
| 53 | #include <sys/time.h>
 | 
|---|
| 54 | #include <sys/types.h>
 | 
|---|
| 55 | #include <sys/stat.h>
 | 
|---|
| 56 | #include <sys/socket.h>
 | 
|---|
| 57 | #include <netinet/in.h>
 | 
|---|
| 58 | #include <netdb.h>
 | 
|---|
| 59 | #include <arpa/inet.h>
 | 
|---|
| 60 | 
 | 
|---|
| 61 | #include "lilxml.h"
 | 
|---|
| 62 | #include "indiapi.h"
 | 
|---|
| 63 | #include "fq.h"
 | 
|---|
| 64 | 
 | 
|---|
| 65 | #define INDIPORT        7624            /* default TCP/IP port to listen */
 | 
|---|
| 66 | #define REMOTEDVR       (-1234)         /* invalid PID to flag remote drivers */
 | 
|---|
| 67 | #define MAXRBUF         4096            /* max read buffering here */
 | 
|---|
| 68 | #define MAXWSIZ         4096            /* max bytes/write */
 | 
|---|
| 69 | #define DEFMAXQSIZ      10              /* default max q behind, MB */
 | 
|---|
| 70 | 
 | 
|---|
| 71 | /* associate a usage count with queuded client or device message */
 | 
|---|
| 72 | typedef struct {
 | 
|---|
| 73 |     int count;                          /* number of consumers left */
 | 
|---|
| 74 |     unsigned int cl;                    /* content length */
 | 
|---|
| 75 |     char *cp;                           /* content: buf or malloced */
 | 
|---|
| 76 |     char buf[MAXWSIZ];          /* local buf for most messages */
 | 
|---|
| 77 | } Msg;
 | 
|---|
| 78 | 
 | 
|---|
| 79 | /* BLOB handling, NEVER is the default */
 | 
|---|
| 80 | typedef enum {B_NEVER=0, B_ALSO, B_ONLY} BLOBHandling;
 | 
|---|
| 81 | 
 | 
|---|
| 82 | /* device + property name */
 | 
|---|
| 83 | typedef struct {
 | 
|---|
| 84 |     char dev[MAXINDIDEVICE];
 | 
|---|
| 85 |     char name[MAXINDINAME];
 | 
|---|
| 86 | } Property;
 | 
|---|
| 87 | 
 | 
|---|
| 88 | /* record of each snooped property */
 | 
|---|
| 89 | typedef struct {
 | 
|---|
| 90 |     Property prop;
 | 
|---|
| 91 |     BLOBHandling blob;                  /* when to snoop BLOBs */
 | 
|---|
| 92 | } Snoopee;
 | 
|---|
| 93 | 
 | 
|---|
| 94 | /* info for each connected client */
 | 
|---|
| 95 | typedef struct {
 | 
|---|
| 96 |     int active;                         /* 1 when this record is in use */
 | 
|---|
| 97 |     Property *props;                    /* malloced array of props we want */
 | 
|---|
| 98 |     int nprops;                         /* n entries in props[] */
 | 
|---|
| 99 |     int allprops;                       /* saw getProperties w/o device */
 | 
|---|
| 100 |     BLOBHandling blob;                  /* when to send setBLOBs */
 | 
|---|
| 101 |     int s;                              /* socket for this client */
 | 
|---|
| 102 |     LilXML *lp;                         /* XML parsing context */
 | 
|---|
| 103 |     FQ *msgq;                           /* Msg queue */
 | 
|---|
| 104 |     unsigned int nsent;                         /* bytes of current Msg sent so far */
 | 
|---|
| 105 | } ClInfo;
 | 
|---|
| 106 | static ClInfo *clinfo;                  /*  malloced pool of clients */
 | 
|---|
| 107 | static int nclinfo;                     /* n total (not active) */
 | 
|---|
| 108 | 
 | 
|---|
| 109 | /* info for each connected driver */
 | 
|---|
| 110 | typedef struct {
 | 
|---|
| 111 |     const char *name;                           /* persistent malloced name */
 | 
|---|
| 112 |     char dev[MAXINDIDEVICE];            /* device served by this driver */
 | 
|---|
| 113 |     Snoopee *sprops;                    /* malloced array of props we snoop */
 | 
|---|
| 114 |     int nsprops;                        /* n entries in sprops[] */
 | 
|---|
| 115 |     int pid;                            /* process id or REMOTEDVR if remote */
 | 
|---|
| 116 |     int rfd;                            /* read pipe fd */
 | 
|---|
| 117 |     int wfd;                            /* write pipe fd */
 | 
|---|
| 118 |     int efd;                            /* stderr from driver, if local */
 | 
|---|
| 119 |     int restarts;                       /* times process has been restarted */
 | 
|---|
| 120 |     LilXML *lp;                         /* XML parsing context */
 | 
|---|
| 121 |     FQ *msgq;                           /* Msg queue */
 | 
|---|
| 122 |     unsigned int nsent;                 /* bytes of current Msg sent so far */
 | 
|---|
| 123 | } DvrInfo;
 | 
|---|
| 124 | static DvrInfo *dvrinfo;                /* malloced array of drivers */
 | 
|---|
| 125 | static int ndvrinfo;                    /* n total */
 | 
|---|
| 126 | 
 | 
|---|
| 127 | static char *me;                        /* our name */
 | 
|---|
| 128 | static int port = INDIPORT;             /* public INDI port */
 | 
|---|
| 129 | static int verbose;                     /* chattiness */
 | 
|---|
| 130 | static int lsocket;                     /* listen socket */
 | 
|---|
| 131 | static char *ldir;                      /* where to log driver messages */
 | 
|---|
| 132 | static int maxqsiz = (DEFMAXQSIZ*1024*1024); /* kill if these bytes behind */
 | 
|---|
| 133 | 
 | 
|---|
| 134 | static void logStartup(int ac, char *av[]);
 | 
|---|
| 135 | static void usage (void);
 | 
|---|
| 136 | static void noZombies (void);
 | 
|---|
| 137 | static void noSIGPIPE (void);
 | 
|---|
| 138 | static void indiRun (void);
 | 
|---|
| 139 | static void indiListen (void);
 | 
|---|
| 140 | static void newClient (void);
 | 
|---|
| 141 | static int newClSocket (void);
 | 
|---|
| 142 | static void shutdownClient (ClInfo *cp);
 | 
|---|
| 143 | static int readFromClient (ClInfo *cp);
 | 
|---|
| 144 | static void startDvr (DvrInfo *dp);
 | 
|---|
| 145 | static void startLocalDvr (DvrInfo *dp);
 | 
|---|
| 146 | static void startRemoteDvr (DvrInfo *dp);
 | 
|---|
| 147 | static int openINDIServer (char host[], int indi_port);
 | 
|---|
| 148 | static void restartDvr (DvrInfo *dp);
 | 
|---|
| 149 | static void q2RDrivers (const char *dev, Msg *mp, XMLEle *root);
 | 
|---|
| 150 | static void q2SDrivers (int isblob, const char *dev, const char *name, Msg *mp,
 | 
|---|
| 151 |     XMLEle *root);
 | 
|---|
| 152 | static int q2Clients (ClInfo *notme, int isblob, const char *dev, const char *name,
 | 
|---|
| 153 |     Msg *mp, XMLEle *root);
 | 
|---|
| 154 | static void addSDevice (DvrInfo *dp, const char *dev, const char *name);
 | 
|---|
| 155 | static Snoopee *findSDevice (DvrInfo *dp, const char *dev, const char *name);
 | 
|---|
| 156 | static void addClDevice (ClInfo *cp, const char *dev, const char *name);
 | 
|---|
| 157 | static int findClDevice (ClInfo *cp, const char *dev, const char *name);
 | 
|---|
| 158 | static int readFromDriver (DvrInfo *dp);
 | 
|---|
| 159 | static int stderrFromDriver (DvrInfo *dp);
 | 
|---|
| 160 | static int msgQSize (FQ *q);
 | 
|---|
| 161 | static void setMsgXMLEle (Msg *mp, XMLEle *root);
 | 
|---|
| 162 | static void setMsgStr (Msg *mp, char *str);
 | 
|---|
| 163 | static void freeMsg (Msg *mp);
 | 
|---|
| 164 | static Msg *newMsg (void);
 | 
|---|
| 165 | static int sendClientMsg (ClInfo *cp);
 | 
|---|
| 166 | static int sendDriverMsg (DvrInfo *cp);
 | 
|---|
| 167 | static void crackBLOB (char *enableBLOB, BLOBHandling *bp);
 | 
|---|
| 168 | static void traceMsg (XMLEle *root);
 | 
|---|
| 169 | static char *indi_tstamp (char *s);
 | 
|---|
| 170 | static void logDMsg (XMLEle *root, const char *dev);
 | 
|---|
| 171 | static void Bye(void);
 | 
|---|
| 172 | 
 | 
|---|
| 173 | int
 | 
|---|
| 174 | main (int ac, char *av[])
 | 
|---|
| 175 | {
 | 
|---|
| 176 |         /* log startup */
 | 
|---|
| 177 |         logStartup(ac, av);
 | 
|---|
| 178 | 
 | 
|---|
| 179 |         /* save our name */
 | 
|---|
| 180 |         me = av[0];
 | 
|---|
| 181 | 
 | 
|---|
| 182 |         /* crack args */
 | 
|---|
| 183 |         while ((--ac > 0) && ((*++av)[0] == '-')) {
 | 
|---|
| 184 |             char *s;
 | 
|---|
| 185 |             for (s = av[0]+1; *s != '\0'; s++)
 | 
|---|
| 186 |                 switch (*s) {
 | 
|---|
| 187 |                 case 'l':
 | 
|---|
| 188 |                     if (ac < 2) {
 | 
|---|
| 189 |                         fprintf (stderr, "-l requires log directory\n");
 | 
|---|
| 190 |                         usage();
 | 
|---|
| 191 |                     }
 | 
|---|
| 192 |                     ldir = *++av;
 | 
|---|
| 193 |                     ac--;
 | 
|---|
| 194 |                     break;
 | 
|---|
| 195 |                 case 'm':
 | 
|---|
| 196 |                     if (ac < 2) {
 | 
|---|
| 197 |                         fprintf (stderr, "-m requires max MB behind\n");
 | 
|---|
| 198 |                         usage();
 | 
|---|
| 199 |                     }
 | 
|---|
| 200 |                     maxqsiz = 1024*1024*atoi(*++av);
 | 
|---|
| 201 |                     ac--;
 | 
|---|
| 202 |                     break;
 | 
|---|
| 203 |                 case 'p':
 | 
|---|
| 204 |                     if (ac < 2) {
 | 
|---|
| 205 |                         fprintf (stderr, "-p requires port value\n");
 | 
|---|
| 206 |                         usage();
 | 
|---|
| 207 |                     }
 | 
|---|
| 208 |                     port = atoi(*++av);
 | 
|---|
| 209 |                     ac--;
 | 
|---|
| 210 |                     break;
 | 
|---|
| 211 |                 case 'v':
 | 
|---|
| 212 |                     verbose++;
 | 
|---|
| 213 |                     break;
 | 
|---|
| 214 |                 default:
 | 
|---|
| 215 |                     usage();
 | 
|---|
| 216 |                 }
 | 
|---|
| 217 |         }
 | 
|---|
| 218 | 
 | 
|---|
| 219 |         /* at this point there are ac args in av[] to name our drivers */
 | 
|---|
| 220 |         if (ac == 0)
 | 
|---|
| 221 |             usage();
 | 
|---|
| 222 | 
 | 
|---|
| 223 |         /* take care of some unixisms */
 | 
|---|
| 224 |         noZombies();
 | 
|---|
| 225 |         noSIGPIPE();
 | 
|---|
| 226 | 
 | 
|---|
| 227 |         /* realloc seed for client pool */
 | 
|---|
| 228 |         clinfo = (ClInfo *) malloc (1);
 | 
|---|
| 229 |         nclinfo = 0;
 | 
|---|
| 230 | 
 | 
|---|
| 231 |         /* create driver info array all at once since size never changes */
 | 
|---|
| 232 |         ndvrinfo = ac;
 | 
|---|
| 233 |         dvrinfo = (DvrInfo *) calloc (ndvrinfo, sizeof(DvrInfo));
 | 
|---|
| 234 | 
 | 
|---|
| 235 |         /* start each driver */
 | 
|---|
| 236 |         while (ac-- > 0) {
 | 
|---|
| 237 |             dvrinfo[ac].name = *av++;
 | 
|---|
| 238 |             startDvr (&dvrinfo[ac]);
 | 
|---|
| 239 |         }
 | 
|---|
| 240 | 
 | 
|---|
| 241 |         /* announce we are online */
 | 
|---|
| 242 |         indiListen();
 | 
|---|
| 243 | 
 | 
|---|
| 244 |         /* handle new clients and all io */
 | 
|---|
| 245 |         while (1)
 | 
|---|
| 246 |             indiRun();
 | 
|---|
| 247 | 
 | 
|---|
| 248 |         /* whoa! */
 | 
|---|
| 249 |         fprintf (stderr, "unexpected return from main\n");
 | 
|---|
| 250 |         return (1);
 | 
|---|
| 251 | }
 | 
|---|
| 252 | 
 | 
|---|
| 253 | /* record we have started and our args */
 | 
|---|
| 254 | static void
 | 
|---|
| 255 | logStartup(int ac, char *av[])
 | 
|---|
| 256 | {
 | 
|---|
| 257 |         int i;
 | 
|---|
| 258 | 
 | 
|---|
| 259 |         fprintf (stderr, "%s: startup: ", indi_tstamp(NULL));
 | 
|---|
| 260 |         for (i = 0; i < ac; i++)
 | 
|---|
| 261 |             fprintf (stderr, "%s ", av[i]);
 | 
|---|
| 262 |         fprintf (stderr, "\n");
 | 
|---|
| 263 | }
 | 
|---|
| 264 | 
 | 
|---|
| 265 | /* print usage message and exit (2) */
 | 
|---|
| 266 | static void
 | 
|---|
| 267 | usage(void)
 | 
|---|
| 268 | {
 | 
|---|
| 269 |         fprintf (stderr, "Usage: %s [options] driver [driver ...]\n", me);
 | 
|---|
| 270 |         fprintf (stderr, "Purpose: server for local and remote INDI drivers\n");
 | 
|---|
| 271 |         fprintf (stderr, "Code %s. Protocol %g.\n", "$Revision: 726523 $", INDIV);
 | 
|---|
| 272 |         fprintf (stderr, "Options:\n");
 | 
|---|
| 273 |         fprintf (stderr, " -l d  : log driver messages to <d>/YYYY-MM-DD.islog\n");
 | 
|---|
| 274 |         fprintf (stderr, " -m m  : kill client if gets more than this many MB behind, default %d\n", DEFMAXQSIZ);
 | 
|---|
| 275 |         fprintf (stderr, " -p p  : alternate IP port, default %d\n", INDIPORT);
 | 
|---|
| 276 |         fprintf (stderr, " -v    : show key events, no traffic\n");
 | 
|---|
| 277 |         fprintf (stderr, " -vv   : -v + key message content\n");
 | 
|---|
| 278 |         fprintf (stderr, " -vvv  : -vv + complete xml\n");
 | 
|---|
| 279 |         fprintf (stderr, "driver : executable or device@host[:port]\n");
 | 
|---|
| 280 | 
 | 
|---|
| 281 |         exit (2);
 | 
|---|
| 282 | }
 | 
|---|
| 283 | 
 | 
|---|
| 284 | /* arrange for no zombies if drivers die */
 | 
|---|
| 285 | static void
 | 
|---|
| 286 | noZombies()
 | 
|---|
| 287 | {
 | 
|---|
| 288 |         struct sigaction sa;
 | 
|---|
| 289 |         sa.sa_handler = SIG_IGN;
 | 
|---|
| 290 |         sigemptyset(&sa.sa_mask);
 | 
|---|
| 291 | #ifdef SA_NOCLDWAIT
 | 
|---|
| 292 |         sa.sa_flags = SA_NOCLDWAIT;
 | 
|---|
| 293 | #else
 | 
|---|
| 294 |         sa.sa_flags = 0;
 | 
|---|
| 295 | #endif
 | 
|---|
| 296 |         (void)sigaction(SIGCHLD, &sa, NULL);
 | 
|---|
| 297 | }
 | 
|---|
| 298 | 
 | 
|---|
| 299 | /* turn off SIGPIPE on bad write so we can handle it inline */
 | 
|---|
| 300 | static void
 | 
|---|
| 301 | noSIGPIPE()
 | 
|---|
| 302 | {
 | 
|---|
| 303 |         struct sigaction sa;
 | 
|---|
| 304 |         sa.sa_handler = SIG_IGN;
 | 
|---|
| 305 |         sigemptyset(&sa.sa_mask);
 | 
|---|
| 306 |         (void)sigaction(SIGPIPE, &sa, NULL);
 | 
|---|
| 307 | }
 | 
|---|
| 308 | 
 | 
|---|
| 309 | /* start the given INDI driver process or connection.
 | 
|---|
| 310 |  * exit if trouble.
 | 
|---|
| 311 |  */
 | 
|---|
| 312 | static void
 | 
|---|
| 313 | startDvr (DvrInfo *dp)
 | 
|---|
| 314 | {
 | 
|---|
| 315 |         if (strchr (dp->name, '@'))
 | 
|---|
| 316 |             startRemoteDvr (dp);
 | 
|---|
| 317 |         else
 | 
|---|
| 318 |             startLocalDvr (dp);
 | 
|---|
| 319 | }
 | 
|---|
| 320 | 
 | 
|---|
| 321 | /* start the given local INDI driver process.
 | 
|---|
| 322 |  * exit if trouble.
 | 
|---|
| 323 |  */
 | 
|---|
| 324 | static void
 | 
|---|
| 325 | startLocalDvr (DvrInfo *dp)
 | 
|---|
| 326 | {
 | 
|---|
| 327 |         Msg *mp;
 | 
|---|
| 328 |         char buf[1024];
 | 
|---|
| 329 |         int rp[2], wp[2], ep[2];
 | 
|---|
| 330 |         int pid;
 | 
|---|
| 331 | 
 | 
|---|
| 332 |         /* build three pipes: r, w and error*/
 | 
|---|
| 333 |         if (pipe (rp) < 0) {
 | 
|---|
| 334 |             fprintf (stderr, "%s: read pipe: %s\n", indi_tstamp(NULL),
 | 
|---|
| 335 |                                                             strerror(errno));
 | 
|---|
| 336 |             Bye();
 | 
|---|
| 337 |         }
 | 
|---|
| 338 |         if (pipe (wp) < 0) {
 | 
|---|
| 339 |             fprintf (stderr, "%s: write pipe: %s\n", indi_tstamp(NULL),
 | 
|---|
| 340 |                                                             strerror(errno));
 | 
|---|
| 341 |             Bye();
 | 
|---|
| 342 |         }
 | 
|---|
| 343 |         if (pipe (ep) < 0) {
 | 
|---|
| 344 |             fprintf (stderr, "%s: stderr pipe: %s\n", indi_tstamp(NULL),
 | 
|---|
| 345 |                                                             strerror(errno));
 | 
|---|
| 346 |             Bye();
 | 
|---|
| 347 |         }
 | 
|---|
| 348 | 
 | 
|---|
| 349 |         /* fork&exec new process */
 | 
|---|
| 350 |         pid = fork();
 | 
|---|
| 351 |         if (pid < 0) {
 | 
|---|
| 352 |             fprintf (stderr, "%s: fork: %s\n", indi_tstamp(NULL), strerror(errno));
 | 
|---|
| 353 |             Bye();
 | 
|---|
| 354 |         }
 | 
|---|
| 355 |         if (pid == 0) {
 | 
|---|
| 356 |             /* child: exec name */
 | 
|---|
| 357 |             int fd;
 | 
|---|
| 358 | 
 | 
|---|
| 359 |             /* rig up pipes */
 | 
|---|
| 360 |             dup2 (wp[0], 0);    /* driver stdin reads from wp[0] */
 | 
|---|
| 361 |             dup2 (rp[1], 1);    /* driver stdout writes to rp[1] */
 | 
|---|
| 362 |             dup2 (ep[1], 2);    /* driver stderr writes to e[]1] */
 | 
|---|
| 363 |             for (fd = 3; fd < 100; fd++)
 | 
|---|
| 364 |                 (void) close (fd);
 | 
|---|
| 365 | 
 | 
|---|
| 366 |             /* go -- should never return */
 | 
|---|
| 367 |             execlp (dp->name, dp->name, NULL);
 | 
|---|
| 368 |             fprintf (stderr, "%s: Driver %s: execlp: %s\n", indi_tstamp(NULL),
 | 
|---|
| 369 |                                                 dp->name, strerror(errno));
 | 
|---|
| 370 |             _exit (1);  /* parent will notice EOF shortly */
 | 
|---|
| 371 |         }
 | 
|---|
| 372 | 
 | 
|---|
| 373 |         /* don't need child's side of pipes */
 | 
|---|
| 374 |         close (wp[0]);
 | 
|---|
| 375 |         close (rp[1]);
 | 
|---|
| 376 |         close (ep[1]);
 | 
|---|
| 377 | 
 | 
|---|
| 378 |         /* record pid, io channels, init lp and snoop list */
 | 
|---|
| 379 |         dp->pid = pid;
 | 
|---|
| 380 |         dp->rfd = rp[0];
 | 
|---|
| 381 |         dp->wfd = wp[1];
 | 
|---|
| 382 |         dp->efd = ep[0];
 | 
|---|
| 383 |         dp->lp = newLilXML();
 | 
|---|
| 384 |         dp->msgq = newFQ(1);
 | 
|---|
| 385 |         dp->sprops = (Snoopee*) malloc (1);     /* seed for realloc */
 | 
|---|
| 386 |         dp->nsprops = 0;
 | 
|---|
| 387 |         dp->nsent = 0;
 | 
|---|
| 388 | 
 | 
|---|
| 389 |         /* first message primes driver to report its properties -- dev known
 | 
|---|
| 390 |          * if restarting
 | 
|---|
| 391 |          */
 | 
|---|
| 392 |         mp = newMsg();
 | 
|---|
| 393 |         pushFQ (dp->msgq, mp);
 | 
|---|
| 394 |         if (dp->dev[0])
 | 
|---|
| 395 |             sprintf (buf, "<getProperties device='%s' version='%g'/>\n",
 | 
|---|
| 396 |                                                             dp->dev, INDIV);
 | 
|---|
| 397 |         else
 | 
|---|
| 398 |             sprintf (buf, "<getProperties version='%g'/>\n", INDIV);
 | 
|---|
| 399 |         setMsgStr (mp, buf);
 | 
|---|
| 400 |         mp->count++;
 | 
|---|
| 401 | 
 | 
|---|
| 402 |         if (verbose > 0)
 | 
|---|
| 403 |             fprintf (stderr, "%s: Driver %s: pid=%d rfd=%d wfd=%d efd=%d\n",
 | 
|---|
| 404 |                     indi_tstamp(NULL), dp->name, dp->pid, dp->rfd, dp->wfd, dp->efd);
 | 
|---|
| 405 | }
 | 
|---|
| 406 | 
 | 
|---|
| 407 | /* start the given remote INDI driver connection.
 | 
|---|
| 408 |  * exit if trouble.
 | 
|---|
| 409 |  */
 | 
|---|
| 410 | static void
 | 
|---|
| 411 | startRemoteDvr (DvrInfo *dp)
 | 
|---|
| 412 | {
 | 
|---|
| 413 |         Msg *mp;
 | 
|---|
| 414 |         char dev[1024];
 | 
|---|
| 415 |         char host[1024];
 | 
|---|
| 416 |         char buf[1024];
 | 
|---|
| 417 |         int indi_port, sockfd;
 | 
|---|
| 418 | 
 | 
|---|
| 419 |         /* extract host and port */
 | 
|---|
| 420 |         indi_port = INDIPORT;
 | 
|---|
| 421 |         if (sscanf (dp->name, "%[^@]@%[^:]:%d", dev, host, &indi_port) < 2) {
 | 
|---|
| 422 |             fprintf (stderr, "Bad remote device syntax: %s\n", dp->name);
 | 
|---|
| 423 |             Bye();
 | 
|---|
| 424 |         }
 | 
|---|
| 425 | 
 | 
|---|
| 426 |         /* connect */
 | 
|---|
| 427 |         sockfd = openINDIServer (host, indi_port);
 | 
|---|
| 428 | 
 | 
|---|
| 429 |         /* record flag pid, io channels, init lp and snoop list */
 | 
|---|
| 430 |         dp->pid = REMOTEDVR;
 | 
|---|
| 431 |         dp->rfd = sockfd;
 | 
|---|
| 432 |         dp->wfd = sockfd;
 | 
|---|
| 433 |         dp->lp = newLilXML();
 | 
|---|
| 434 |         dp->msgq = newFQ(1);
 | 
|---|
| 435 |         dp->sprops = (Snoopee*) malloc (1);     /* seed for realloc */
 | 
|---|
| 436 |         dp->nsprops = 0;
 | 
|---|
| 437 |         dp->nsent = 0;
 | 
|---|
| 438 | 
 | 
|---|
| 439 |         /* N.B. storing name now is key to limiting outbound traffic to this
 | 
|---|
| 440 |          * dev.
 | 
|---|
| 441 |          */
 | 
|---|
| 442 |         strncpy (dp->dev, dev, MAXINDIDEVICE-1);
 | 
|---|
| 443 |         dp->dev[MAXINDIDEVICE-1] = '\0';
 | 
|---|
| 444 | 
 | 
|---|
| 445 |         /* Sending getProperties with device lets remote server limit its
 | 
|---|
| 446 |          * outbound (and our inbound) traffic on this socket to this device.
 | 
|---|
| 447 |          */
 | 
|---|
| 448 |         mp = newMsg();
 | 
|---|
| 449 |         pushFQ (dp->msgq, mp);
 | 
|---|
| 450 |         sprintf (buf, "<getProperties device='%s' version='%g'/>\n",
 | 
|---|
| 451 |                                                             dp->dev, INDIV);
 | 
|---|
| 452 |         setMsgStr (mp, buf);
 | 
|---|
| 453 |         mp->count++;
 | 
|---|
| 454 | 
 | 
|---|
| 455 |         if (verbose > 0)
 | 
|---|
| 456 |             fprintf (stderr, "%s: Driver %s: socket=%d\n", indi_tstamp(NULL),
 | 
|---|
| 457 |                                                             dp->name, sockfd);
 | 
|---|
| 458 | }
 | 
|---|
| 459 | 
 | 
|---|
| 460 | /* open a connection to the given host and port or die.
 | 
|---|
| 461 |  * return socket fd.
 | 
|---|
| 462 |  */
 | 
|---|
| 463 | static int
 | 
|---|
| 464 | openINDIServer (char host[], int indi_port)
 | 
|---|
| 465 | {
 | 
|---|
| 466 |         struct sockaddr_in serv_addr;
 | 
|---|
| 467 |         struct hostent *hp;
 | 
|---|
| 468 |         int sockfd;
 | 
|---|
| 469 | 
 | 
|---|
| 470 |         /* lookup host address */
 | 
|---|
| 471 |         hp = gethostbyname (host);
 | 
|---|
| 472 |         if (!hp) {
 | 
|---|
| 473 |             fprintf (stderr, "gethostbyname(%s): %s\n", host, strerror(errno));
 | 
|---|
| 474 |             Bye();
 | 
|---|
| 475 |         }
 | 
|---|
| 476 | 
 | 
|---|
| 477 |         /* create a socket to the INDI server */
 | 
|---|
| 478 |         (void) memset ((char *)&serv_addr, 0, sizeof(serv_addr));
 | 
|---|
| 479 |         serv_addr.sin_family = AF_INET;
 | 
|---|
| 480 |         serv_addr.sin_addr.s_addr =
 | 
|---|
| 481 |                             ((struct in_addr *)(hp->h_addr_list[0]))->s_addr;
 | 
|---|
| 482 |         serv_addr.sin_port = htons(indi_port);
 | 
|---|
| 483 |         if ((sockfd = socket (AF_INET, SOCK_STREAM, 0)) < 0) {
 | 
|---|
| 484 |             fprintf (stderr, "socket(%s,%d): %s\n", host, indi_port,strerror(errno));
 | 
|---|
| 485 |             Bye();
 | 
|---|
| 486 |         }
 | 
|---|
| 487 | 
 | 
|---|
| 488 |         /* connect */
 | 
|---|
| 489 |         if (connect (sockfd,(struct sockaddr *)&serv_addr,sizeof(serv_addr))<0){
 | 
|---|
| 490 |             fprintf (stderr, "connect(%s,%d): %s\n", host,indi_port,strerror(errno));
 | 
|---|
| 491 |             Bye();
 | 
|---|
| 492 |         }
 | 
|---|
| 493 | 
 | 
|---|
| 494 |         /* ok */
 | 
|---|
| 495 |         return (sockfd);
 | 
|---|
| 496 | }
 | 
|---|
| 497 | 
 | 
|---|
| 498 | /* create the public INDI Driver endpoint lsocket on port.
 | 
|---|
| 499 |  * return server socket else exit.
 | 
|---|
| 500 |  */
 | 
|---|
| 501 | static void
 | 
|---|
| 502 | indiListen ()
 | 
|---|
| 503 | {
 | 
|---|
| 504 |         struct sockaddr_in serv_socket;
 | 
|---|
| 505 |         int sfd;
 | 
|---|
| 506 |         int reuse = 1;
 | 
|---|
| 507 | 
 | 
|---|
| 508 |         /* make socket endpoint */
 | 
|---|
| 509 |         if ((sfd = socket (AF_INET, SOCK_STREAM, 0)) < 0) {
 | 
|---|
| 510 |             fprintf (stderr, "%s: socket: %s\n", indi_tstamp(NULL), strerror(errno));
 | 
|---|
| 511 |             Bye();
 | 
|---|
| 512 |         }
 | 
|---|
| 513 | 
 | 
|---|
| 514 |         /* bind to given port for any IP address */
 | 
|---|
| 515 |         memset (&serv_socket, 0, sizeof(serv_socket));
 | 
|---|
| 516 |         serv_socket.sin_family = AF_INET;
 | 
|---|
| 517 |         #ifdef SSH_TUNNEL
 | 
|---|
| 518 |         serv_socket.sin_addr.s_addr = htonl (INADDR_LOOPBACK);
 | 
|---|
| 519 |         #else
 | 
|---|
| 520 |         serv_socket.sin_addr.s_addr = htonl (INADDR_ANY);
 | 
|---|
| 521 |         #endif
 | 
|---|
| 522 |         serv_socket.sin_port = htons ((unsigned short)port);
 | 
|---|
| 523 |         if (setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0){
 | 
|---|
| 524 |             fprintf (stderr, "%s: setsockopt: %s\n", indi_tstamp(NULL),
 | 
|---|
| 525 |                                                             strerror(errno));
 | 
|---|
| 526 |             Bye();
 | 
|---|
| 527 |         }
 | 
|---|
| 528 |         if (bind(sfd,(struct sockaddr*)&serv_socket,sizeof(serv_socket)) < 0){
 | 
|---|
| 529 |             fprintf (stderr, "%s: bind: %s\n", indi_tstamp(NULL), strerror(errno));
 | 
|---|
| 530 |             Bye();
 | 
|---|
| 531 |         }
 | 
|---|
| 532 | 
 | 
|---|
| 533 |         /* willing to accept connections with a backlog of 5 pending */
 | 
|---|
| 534 |         if (listen (sfd, 5) < 0) {
 | 
|---|
| 535 |             fprintf (stderr, "%s: listen: %s\n", indi_tstamp(NULL), strerror(errno));
 | 
|---|
| 536 |             Bye();
 | 
|---|
| 537 |         }
 | 
|---|
| 538 | 
 | 
|---|
| 539 |         /* ok */
 | 
|---|
| 540 |         lsocket = sfd;
 | 
|---|
| 541 |         if (verbose > 0)
 | 
|---|
| 542 |             fprintf (stderr, "%s: listening to port %d on fd %d\n",
 | 
|---|
| 543 |                                                 indi_tstamp(NULL), port, sfd);
 | 
|---|
| 544 | }
 | 
|---|
| 545 | 
 | 
|---|
| 546 | /* service traffic from clients and drivers */
 | 
|---|
| 547 | static void
 | 
|---|
| 548 | indiRun(void)
 | 
|---|
| 549 | {
 | 
|---|
| 550 |         fd_set rs, ws;
 | 
|---|
| 551 |         int maxfd;
 | 
|---|
| 552 |         int i, s;
 | 
|---|
| 553 | 
 | 
|---|
| 554 |         /* init with no writers or readers */
 | 
|---|
| 555 |         FD_ZERO(&ws);
 | 
|---|
| 556 |         FD_ZERO(&rs);
 | 
|---|
| 557 | 
 | 
|---|
| 558 |         /* always listen for new clients */
 | 
|---|
| 559 |         FD_SET(lsocket, &rs);
 | 
|---|
| 560 |         maxfd = lsocket;
 | 
|---|
| 561 | 
 | 
|---|
| 562 |         /* add all client readers and client writers with work to send */
 | 
|---|
| 563 |         for (i = 0; i < nclinfo; i++) {
 | 
|---|
| 564 |             ClInfo *cp = &clinfo[i];
 | 
|---|
| 565 |             if (cp->active) {
 | 
|---|
| 566 |                 FD_SET(cp->s, &rs);
 | 
|---|
| 567 |                 if (nFQ(cp->msgq) > 0)
 | 
|---|
| 568 |                     FD_SET(cp->s, &ws);
 | 
|---|
| 569 |                 if (cp->s > maxfd)
 | 
|---|
| 570 |                     maxfd = cp->s;
 | 
|---|
| 571 |             }
 | 
|---|
| 572 |         }
 | 
|---|
| 573 | 
 | 
|---|
| 574 |         /* add all driver readers and driver writers with work to send */
 | 
|---|
| 575 |         for (i = 0; i < ndvrinfo; i++) {
 | 
|---|
| 576 |             DvrInfo *dp = &dvrinfo[i];
 | 
|---|
| 577 |             FD_SET(dp->rfd, &rs);
 | 
|---|
| 578 |             if (dp->rfd > maxfd)
 | 
|---|
| 579 |                 maxfd = dp->rfd;
 | 
|---|
| 580 |             if (dp->pid != REMOTEDVR) {
 | 
|---|
| 581 |                 FD_SET(dp->efd, &rs);
 | 
|---|
| 582 |                 if (dp->efd > maxfd)
 | 
|---|
| 583 |                     maxfd = dp->efd;
 | 
|---|
| 584 |             }
 | 
|---|
| 585 |             if (nFQ(dp->msgq) > 0) {
 | 
|---|
| 586 |                 FD_SET(dp->wfd, &ws);
 | 
|---|
| 587 |                 if (dp->wfd > maxfd)
 | 
|---|
| 588 |                     maxfd = dp->wfd;
 | 
|---|
| 589 |             }
 | 
|---|
| 590 |         }
 | 
|---|
| 591 | 
 | 
|---|
| 592 |         /* wait for action */
 | 
|---|
| 593 |         s = select (maxfd+1, &rs, &ws, NULL, NULL);
 | 
|---|
| 594 |         if (s < 0) {
 | 
|---|
| 595 |             fprintf (stderr, "%s: select(%d): %s\n", indi_tstamp(NULL), maxfd+1,
 | 
|---|
| 596 |                                                             strerror(errno));
 | 
|---|
| 597 |             Bye();
 | 
|---|
| 598 |         }
 | 
|---|
| 599 | 
 | 
|---|
| 600 |         /* new client? */
 | 
|---|
| 601 |         if (s > 0 && FD_ISSET(lsocket, &rs)) {
 | 
|---|
| 602 |             newClient();
 | 
|---|
| 603 |             s--;
 | 
|---|
| 604 |         }
 | 
|---|
| 605 | 
 | 
|---|
| 606 |         /* message to/from client? */
 | 
|---|
| 607 |         for (i = 0; s > 0 && i < nclinfo; i++) {
 | 
|---|
| 608 |             ClInfo *cp = &clinfo[i];
 | 
|---|
| 609 |             if (cp->active) {
 | 
|---|
| 610 |                 if (FD_ISSET(cp->s, &rs)) {
 | 
|---|
| 611 |                     if (readFromClient(cp) < 0)
 | 
|---|
| 612 |                         return; /* fds effected */
 | 
|---|
| 613 |                     s--;
 | 
|---|
| 614 |                 }
 | 
|---|
| 615 |                 if (s > 0 && FD_ISSET(cp->s, &ws)) {
 | 
|---|
| 616 |                     if (sendClientMsg(cp) < 0)
 | 
|---|
| 617 |                         return; /* fds effected */
 | 
|---|
| 618 |                     s--;
 | 
|---|
| 619 |                 }
 | 
|---|
| 620 |             }
 | 
|---|
| 621 |         }
 | 
|---|
| 622 | 
 | 
|---|
| 623 |         /* message to/from driver? */
 | 
|---|
| 624 |         for (i = 0; s > 0 && i < ndvrinfo; i++) {
 | 
|---|
| 625 |             DvrInfo *dp = &dvrinfo[i];
 | 
|---|
| 626 |             if (dp->pid != REMOTEDVR && FD_ISSET(dp->efd, &rs)) {
 | 
|---|
| 627 |                 if (stderrFromDriver(dp) < 0)
 | 
|---|
| 628 |                     return;     /* fds effected */
 | 
|---|
| 629 |                 s--;
 | 
|---|
| 630 |             }
 | 
|---|
| 631 |             if (s > 0 && FD_ISSET(dp->rfd, &rs)) {
 | 
|---|
| 632 |                 if (readFromDriver(dp) < 0)
 | 
|---|
| 633 |                     return;     /* fds effected */
 | 
|---|
| 634 |                 s--;
 | 
|---|
| 635 |             }
 | 
|---|
| 636 |             if (s > 0 && FD_ISSET(dp->wfd, &ws) && nFQ(dp->msgq) > 0) {
 | 
|---|
| 637 |                 if (sendDriverMsg(dp) < 0)
 | 
|---|
| 638 |                     return;     /* fds effected */
 | 
|---|
| 639 |                 s--;
 | 
|---|
| 640 |             }
 | 
|---|
| 641 |         }
 | 
|---|
| 642 | }
 | 
|---|
| 643 | 
 | 
|---|
| 644 | /* prepare for new client arriving on lsocket.
 | 
|---|
| 645 |  * exit if trouble.
 | 
|---|
| 646 |  */
 | 
|---|
| 647 | static void
 | 
|---|
| 648 | newClient()
 | 
|---|
| 649 | {
 | 
|---|
| 650 |         ClInfo *cp = NULL;
 | 
|---|
| 651 |         int s, cli;
 | 
|---|
| 652 | 
 | 
|---|
| 653 |         /* assign new socket */
 | 
|---|
| 654 |         s = newClSocket ();
 | 
|---|
| 655 | 
 | 
|---|
| 656 |         /* try to reuse a clinfo slot, else add one */
 | 
|---|
| 657 |         for (cli = 0; cli < nclinfo; cli++)
 | 
|---|
| 658 |             if (!(cp = &clinfo[cli])->active)
 | 
|---|
| 659 |                 break;
 | 
|---|
| 660 |         if (cli == nclinfo) {
 | 
|---|
| 661 |             /* grow clinfo */
 | 
|---|
| 662 |             clinfo = (ClInfo *) realloc (clinfo, (nclinfo+1)*sizeof(ClInfo));
 | 
|---|
| 663 |             if (!clinfo) {
 | 
|---|
| 664 |                 fprintf (stderr, "no memory for new client\n");
 | 
|---|
| 665 |                 Bye();
 | 
|---|
| 666 |             }
 | 
|---|
| 667 |             cp = &clinfo[nclinfo++];
 | 
|---|
| 668 |         }
 | 
|---|
| 669 | 
 | 
|---|
| 670 |         /* rig up new clinfo entry */
 | 
|---|
| 671 |         memset (cp, 0, sizeof(*cp));
 | 
|---|
| 672 |         cp->active = 1;
 | 
|---|
| 673 |         cp->s = s;
 | 
|---|
| 674 |         cp->lp = newLilXML();
 | 
|---|
| 675 |         cp->msgq = newFQ(1);
 | 
|---|
| 676 |         cp->props = malloc (1);
 | 
|---|
| 677 |         cp->nsent = 0;
 | 
|---|
| 678 | 
 | 
|---|
| 679 |         if (verbose > 0) {
 | 
|---|
| 680 |             struct sockaddr_in addr;
 | 
|---|
| 681 |             socklen_t len = sizeof(addr);
 | 
|---|
| 682 |             getpeername(s, (struct sockaddr*)&addr, &len);
 | 
|---|
| 683 |             fprintf(stderr,"%s: Client %d: new arrival from %s:%d - welcome!\n",
 | 
|---|
| 684 |                             indi_tstamp(NULL), cp->s, inet_ntoa(addr.sin_addr),
 | 
|---|
| 685 |                                                         ntohs(addr.sin_port));
 | 
|---|
| 686 |         }
 | 
|---|
| 687 | }
 | 
|---|
| 688 | 
 | 
|---|
| 689 | /* read more from the given client, send to each appropriate driver when see
 | 
|---|
| 690 |  * xml closure. also send all newXXX() to all other interested clients.
 | 
|---|
| 691 |  * return -1 if had to shut down anything, else 0.
 | 
|---|
| 692 |  */
 | 
|---|
| 693 | static int
 | 
|---|
| 694 | readFromClient (ClInfo *cp)
 | 
|---|
| 695 | {
 | 
|---|
| 696 |         char buf[MAXRBUF];
 | 
|---|
| 697 |         int shutany = 0;
 | 
|---|
| 698 |         int i, nr;
 | 
|---|
| 699 | 
 | 
|---|
| 700 |         /* read client */
 | 
|---|
| 701 |         nr = read (cp->s, buf, sizeof(buf));
 | 
|---|
| 702 |         if (nr <= 0) {
 | 
|---|
| 703 |             if (nr < 0)
 | 
|---|
| 704 |                 fprintf (stderr, "%s: Client %d: read: %s\n", indi_tstamp(NULL),
 | 
|---|
| 705 |                                                         cp->s, strerror(errno));
 | 
|---|
| 706 |             else if (verbose > 0)
 | 
|---|
| 707 |                 fprintf (stderr, "%s: Client %d: read EOF\n", indi_tstamp(NULL),
 | 
|---|
| 708 |                                                                         cp->s);
 | 
|---|
| 709 |             shutdownClient (cp);
 | 
|---|
| 710 |             return (-1);
 | 
|---|
| 711 |         }
 | 
|---|
| 712 | 
 | 
|---|
| 713 |         /* process XML, sending when find closure */
 | 
|---|
| 714 |         for (i = 0; i < nr; i++) {
 | 
|---|
| 715 |             char err[1024];
 | 
|---|
| 716 |             XMLEle *root = readXMLEle (cp->lp, buf[i], err);
 | 
|---|
| 717 |             if (root) {
 | 
|---|
| 718 |                 char *roottag = tagXMLEle(root);
 | 
|---|
| 719 |                 const char *dev = findXMLAttValu (root, "device");
 | 
|---|
| 720 |                 const char *name = findXMLAttValu (root, "name");
 | 
|---|
| 721 |                 int isblob = !strcmp (tagXMLEle(root), "setBLOBVector");
 | 
|---|
| 722 |                 Msg *mp;
 | 
|---|
| 723 | 
 | 
|---|
| 724 |                 if (verbose > 2) {
 | 
|---|
| 725 |                     fprintf (stderr, "%s: Client %d: read ",indi_tstamp(NULL),cp->s);
 | 
|---|
| 726 |                     traceMsg (root);
 | 
|---|
| 727 |                 } else if (verbose > 1) {
 | 
|---|
| 728 |                     fprintf (stderr, "%s: Client %d: read <%s device='%s' name='%s'>\n",
 | 
|---|
| 729 |                                     indi_tstamp(NULL), cp->s, tagXMLEle(root),
 | 
|---|
| 730 |                                     findXMLAttValu (root, "device"),
 | 
|---|
| 731 |                                     findXMLAttValu (root, "name"));
 | 
|---|
| 732 |                 }
 | 
|---|
| 733 | 
 | 
|---|
| 734 |                 /* snag interested properties.
 | 
|---|
| 735 |                  * N.B. don't open to alldevs if seen specific dev already, else
 | 
|---|
| 736 |                  *   remote client connections start returning too much.
 | 
|---|
| 737 |                  */
 | 
|---|
| 738 |                 if (dev[0])
 | 
|---|
| 739 |                     addClDevice (cp, dev, name);
 | 
|---|
| 740 |                 else if (!strcmp (roottag, "getProperties") && !cp->nprops)
 | 
|---|
| 741 |                     cp->allprops = 1;
 | 
|---|
| 742 | 
 | 
|---|
| 743 |                 /* snag enableBLOB -- send to remote drivers too */
 | 
|---|
| 744 |                 if (!strcmp (roottag, "enableBLOB"))
 | 
|---|
| 745 |                     crackBLOB (pcdataXMLEle(root), &cp->blob);
 | 
|---|
| 746 | 
 | 
|---|
| 747 |                 /* build a new message -- set content iff anyone cares */
 | 
|---|
| 748 |                 mp = newMsg();
 | 
|---|
| 749 | 
 | 
|---|
| 750 |                 /* send message to driver(s) responsible for dev */
 | 
|---|
| 751 |                 q2RDrivers (dev, mp, root);
 | 
|---|
| 752 | 
 | 
|---|
| 753 |                 /* echo new* commands back to other clients */
 | 
|---|
| 754 |                 if (!strncmp (roottag, "new", 3)) {
 | 
|---|
| 755 |                     if (q2Clients (cp, isblob, dev, name, mp, root) < 0)
 | 
|---|
| 756 |                         shutany++;
 | 
|---|
| 757 |                 }
 | 
|---|
| 758 | 
 | 
|---|
| 759 |                 /* set message content if anyone cares else forget it */
 | 
|---|
| 760 |                 if (mp->count > 0)
 | 
|---|
| 761 |                     setMsgXMLEle (mp, root);
 | 
|---|
| 762 |                 else
 | 
|---|
| 763 |                     freeMsg (mp);
 | 
|---|
| 764 |                 delXMLEle (root);
 | 
|---|
| 765 | 
 | 
|---|
| 766 |             } else if (err[0]) {
 | 
|---|
| 767 |                 char *ts = indi_tstamp(NULL);
 | 
|---|
| 768 |                 fprintf (stderr, "%s: Client %d: XML error: %s\n", ts,
 | 
|---|
| 769 |                                                                 cp->s, err);
 | 
|---|
| 770 |                 fprintf (stderr, "%s: Client %d: XML read: %.*s\n", ts,
 | 
|---|
| 771 |                                                             cp->s, nr, buf);
 | 
|---|
| 772 |                 shutdownClient (cp);
 | 
|---|
| 773 |                 return (-1);
 | 
|---|
| 774 |             }
 | 
|---|
| 775 |         }
 | 
|---|
| 776 | 
 | 
|---|
| 777 |         return (shutany ? -1 : 0);
 | 
|---|
| 778 | }
 | 
|---|
| 779 | 
 | 
|---|
| 780 | /* read more from the given driver, send to each interested client when see
 | 
|---|
| 781 |  * xml closure. if driver dies, try restarting.
 | 
|---|
| 782 |  * return 0 if ok else -1 if had to shut down anything.
 | 
|---|
| 783 |  */
 | 
|---|
| 784 | static int
 | 
|---|
| 785 | readFromDriver (DvrInfo *dp)
 | 
|---|
| 786 | {
 | 
|---|
| 787 |         char buf[MAXRBUF];
 | 
|---|
| 788 |         int shutany = 0;
 | 
|---|
| 789 |         int i, nr;
 | 
|---|
| 790 | 
 | 
|---|
| 791 |         /* read driver */
 | 
|---|
| 792 |         nr = read (dp->rfd, buf, sizeof(buf));
 | 
|---|
| 793 |         if (nr <= 0) {
 | 
|---|
| 794 |             if (nr < 0)
 | 
|---|
| 795 |                 fprintf (stderr, "%s: Driver %s: stdin %s\n", indi_tstamp(NULL),
 | 
|---|
| 796 |                                                     dp->name, strerror(errno));
 | 
|---|
| 797 |             else
 | 
|---|
| 798 |                 fprintf (stderr, "%s: Driver %s: stdin EOF\n",
 | 
|---|
| 799 |                                                         indi_tstamp(NULL), dp->name);
 | 
|---|
| 800 |             restartDvr (dp);
 | 
|---|
| 801 |             return (-1);
 | 
|---|
| 802 |         }
 | 
|---|
| 803 | 
 | 
|---|
| 804 |         /* process XML, sending when find closure */
 | 
|---|
| 805 |         for (i = 0; i < nr; i++) {
 | 
|---|
| 806 |             char err[1024];
 | 
|---|
| 807 |             XMLEle *root = readXMLEle (dp->lp, buf[i], err);
 | 
|---|
| 808 |             if (root) {
 | 
|---|
| 809 |                 char *roottag = tagXMLEle(root);
 | 
|---|
| 810 |                 const char *dev = findXMLAttValu (root, "device");
 | 
|---|
| 811 |                 const char *name = findXMLAttValu (root, "name");
 | 
|---|
| 812 |                 int isblob = !strcmp (tagXMLEle(root), "setBLOBVector");
 | 
|---|
| 813 |                 Msg *mp;
 | 
|---|
| 814 | 
 | 
|---|
| 815 |                 if (verbose > 2) {
 | 
|---|
| 816 |                     fprintf(stderr, "%s: Driver %s: read ", indi_tstamp(0),dp->name);
 | 
|---|
| 817 |                     traceMsg (root);
 | 
|---|
| 818 |                 } else if (verbose > 1) {
 | 
|---|
| 819 |                     fprintf (stderr, "%s: Driver %s: read <%s device='%s' name='%s'>\n",
 | 
|---|
| 820 |                                     indi_tstamp(NULL), dp->name, tagXMLEle(root),
 | 
|---|
| 821 |                                     findXMLAttValu (root, "device"),
 | 
|---|
| 822 |                                     findXMLAttValu (root, "name"));
 | 
|---|
| 823 |                 }
 | 
|---|
| 824 | 
 | 
|---|
| 825 |                 /* that's all if driver is just registering a snoop */
 | 
|---|
| 826 |                 if (!strcmp (roottag, "getProperties")) {
 | 
|---|
| 827 |                     addSDevice (dp, dev, name);
 | 
|---|
| 828 |                     delXMLEle (root);
 | 
|---|
| 829 |                     continue;
 | 
|---|
| 830 |                 }
 | 
|---|
| 831 | 
 | 
|---|
| 832 |                 /* that's all if driver is just registering a BLOB mode */
 | 
|---|
| 833 |                 if (!strcmp (roottag, "enableBLOB")) {
 | 
|---|
| 834 |                     Snoopee *sp = findSDevice (dp, dev, name);
 | 
|---|
| 835 |                     if (sp)
 | 
|---|
| 836 |                         crackBLOB (pcdataXMLEle (root), &sp->blob);
 | 
|---|
| 837 |                     delXMLEle (root);
 | 
|---|
| 838 |                     continue;
 | 
|---|
| 839 |                 }
 | 
|---|
| 840 | 
 | 
|---|
| 841 |                 /* snag device name if not known yet */
 | 
|---|
| 842 |                 if (!dp->dev[0] && dev[0]) {
 | 
|---|
| 843 |                     strncpy (dp->dev, dev, MAXINDIDEVICE-1);
 | 
|---|
| 844 |                     dp->dev[MAXINDIDEVICE-1] = '\0';
 | 
|---|
| 845 |                 }
 | 
|---|
| 846 | 
 | 
|---|
| 847 |                 /* log messages if any and wanted */
 | 
|---|
| 848 |                 if (ldir)
 | 
|---|
| 849 |                     logDMsg (root, dev);
 | 
|---|
| 850 | 
 | 
|---|
| 851 |                 /* build a new message -- set content iff anyone cares */
 | 
|---|
| 852 |                 mp = newMsg();
 | 
|---|
| 853 | 
 | 
|---|
| 854 |                 /* send to interested clients */
 | 
|---|
| 855 |                 if (q2Clients (NULL, isblob, dev, name, mp, root) < 0)
 | 
|---|
| 856 |                     shutany++;
 | 
|---|
| 857 | 
 | 
|---|
| 858 |                 /* send to snooping drivers */
 | 
|---|
| 859 |                 q2SDrivers (isblob, dev, name, mp, root);
 | 
|---|
| 860 | 
 | 
|---|
| 861 |                 /* set message content if anyone cares else forget it */
 | 
|---|
| 862 |                 if (mp->count > 0)
 | 
|---|
| 863 |                     setMsgXMLEle (mp, root);
 | 
|---|
| 864 |                 else
 | 
|---|
| 865 |                     freeMsg (mp);
 | 
|---|
| 866 |                 delXMLEle (root);
 | 
|---|
| 867 | 
 | 
|---|
| 868 |             } else if (err[0]) {
 | 
|---|
| 869 |                 char *ts = indi_tstamp(NULL);
 | 
|---|
| 870 |                 fprintf (stderr, "%s: Driver %s: XML error: %s\n", ts,
 | 
|---|
| 871 |                                                                 dp->name, err);
 | 
|---|
| 872 |                 fprintf (stderr, "%s: Driver %s: XML read: %.*s\n", ts,
 | 
|---|
| 873 |                                                             dp->name, nr, buf);
 | 
|---|
| 874 |                 restartDvr (dp);
 | 
|---|
| 875 |                 return (-1);
 | 
|---|
| 876 |             }
 | 
|---|
| 877 |         }
 | 
|---|
| 878 | 
 | 
|---|
| 879 |         return (shutany ? -1 : 0);
 | 
|---|
| 880 | }
 | 
|---|
| 881 | 
 | 
|---|
| 882 | /* read more from the given driver stderr, add prefix and send to our stderr.
 | 
|---|
| 883 |  * return 0 if ok else -1 if had to restart.
 | 
|---|
| 884 |  */
 | 
|---|
| 885 | static int
 | 
|---|
| 886 | stderrFromDriver (DvrInfo *dp)
 | 
|---|
| 887 | {
 | 
|---|
| 888 |         static char exbuf[MAXRBUF];
 | 
|---|
| 889 |         static int nexbuf;
 | 
|---|
| 890 |         int i, nr;
 | 
|---|
| 891 | 
 | 
|---|
| 892 |         /* read more */
 | 
|---|
| 893 |         nr = read (dp->efd, exbuf+nexbuf, sizeof(exbuf)-nexbuf);
 | 
|---|
| 894 |         if (nr <= 0) {
 | 
|---|
| 895 |             if (nr < 0)
 | 
|---|
| 896 |                 fprintf (stderr, "%s: Driver %s: stderr %s\n", indi_tstamp(NULL),
 | 
|---|
| 897 |                                                     dp->name, strerror(errno));
 | 
|---|
| 898 |             else
 | 
|---|
| 899 |                 fprintf (stderr, "%s: Driver %s: stderr EOF\n",
 | 
|---|
| 900 |                                                         indi_tstamp(NULL), dp->name);
 | 
|---|
| 901 |             restartDvr (dp);
 | 
|---|
| 902 |             return (-1);
 | 
|---|
| 903 |         }
 | 
|---|
| 904 |         nexbuf += nr;
 | 
|---|
| 905 | 
 | 
|---|
| 906 |         /* prefix each whole line to our stderr, save extra for next time */
 | 
|---|
| 907 |         for (i = 0; i < nexbuf; i++) {
 | 
|---|
| 908 |             if (exbuf[i] == '\n') {
 | 
|---|
| 909 |                 fprintf (stderr, "%s: Driver %s: %.*s\n", indi_tstamp(NULL),
 | 
|---|
| 910 |                                                             dp->name, i, exbuf);
 | 
|---|
| 911 |                 i++;                              /* count including nl */
 | 
|---|
| 912 |                 nexbuf -= i;                      /* remove from nexbuf */
 | 
|---|
| 913 |                 memmove (exbuf, exbuf+i, nexbuf); /* slide remaining to front */
 | 
|---|
| 914 |                 i = -1;                           /* restart for loop scan */
 | 
|---|
| 915 |             }
 | 
|---|
| 916 |         }
 | 
|---|
| 917 | 
 | 
|---|
| 918 |         return (0);
 | 
|---|
| 919 | }
 | 
|---|
| 920 | 
 | 
|---|
| 921 | /* close down the given client */
 | 
|---|
| 922 | static void
 | 
|---|
| 923 | shutdownClient (ClInfo *cp)
 | 
|---|
| 924 | {
 | 
|---|
| 925 |         Msg *mp;
 | 
|---|
| 926 | 
 | 
|---|
| 927 |         /* close connection */
 | 
|---|
| 928 |         shutdown (cp->s, SHUT_RDWR);
 | 
|---|
| 929 |         close (cp->s);
 | 
|---|
| 930 | 
 | 
|---|
| 931 |         /* free memory */
 | 
|---|
| 932 |         delLilXML (cp->lp);
 | 
|---|
| 933 |         free (cp->props);
 | 
|---|
| 934 | 
 | 
|---|
| 935 |         /* decrement and possibly free any unsent messages for this client */
 | 
|---|
| 936 |         while ((mp = (Msg*) popFQ(cp->msgq)) != NULL)
 | 
|---|
| 937 |             if (--mp->count == 0)
 | 
|---|
| 938 |                 freeMsg (mp);
 | 
|---|
| 939 |         delFQ (cp->msgq);
 | 
|---|
| 940 | 
 | 
|---|
| 941 |         /* ok now to recycle */
 | 
|---|
| 942 |         cp->active = 0;
 | 
|---|
| 943 | 
 | 
|---|
| 944 |         if (verbose > 0)
 | 
|---|
| 945 |             fprintf (stderr, "%s: Client %d: shut down complete - bye!\n",
 | 
|---|
| 946 |                                                         indi_tstamp(NULL), cp->s);
 | 
|---|
| 947 | }
 | 
|---|
| 948 | 
 | 
|---|
| 949 | /* close down the given driver and restart */
 | 
|---|
| 950 | static void
 | 
|---|
| 951 | restartDvr (DvrInfo *dp)
 | 
|---|
| 952 | {
 | 
|---|
| 953 |         Msg *mp;
 | 
|---|
| 954 | 
 | 
|---|
| 955 |         /* make sure it's dead, reclaim resources */
 | 
|---|
| 956 |         if (dp->pid == REMOTEDVR) {
 | 
|---|
| 957 |             /* socket connection */
 | 
|---|
| 958 |             shutdown (dp->wfd, SHUT_RDWR);
 | 
|---|
| 959 |             close (dp->wfd);    /* same as rfd */
 | 
|---|
| 960 |         } else {
 | 
|---|
| 961 |             /* local pipe connection */
 | 
|---|
| 962 |             kill (dp->pid, SIGKILL);    /* we've insured there are no zombies */
 | 
|---|
| 963 |             close (dp->wfd);
 | 
|---|
| 964 |             close (dp->rfd);
 | 
|---|
| 965 |             close (dp->efd);
 | 
|---|
| 966 |         }
 | 
|---|
| 967 | 
 | 
|---|
| 968 |         /* free memory */
 | 
|---|
| 969 |         free (dp->sprops);
 | 
|---|
| 970 |         delLilXML (dp->lp);
 | 
|---|
| 971 | 
 | 
|---|
| 972 |         /* decrement and possibly free any unsent messages for this client */
 | 
|---|
| 973 |         while ((mp = (Msg*) popFQ(dp->msgq)) != NULL)
 | 
|---|
| 974 |             if (--mp->count == 0)
 | 
|---|
| 975 |                 freeMsg (mp);
 | 
|---|
| 976 |         delFQ (dp->msgq);
 | 
|---|
| 977 | 
 | 
|---|
| 978 |         fprintf (stderr, "%s: Driver %s: restart #%d\n", indi_tstamp(NULL),
 | 
|---|
| 979 |                                                     dp->name, ++dp->restarts);
 | 
|---|
| 980 |         startDvr (dp);
 | 
|---|
| 981 | }
 | 
|---|
| 982 | 
 | 
|---|
| 983 | /* put Msg mp on queue of each driver responsible for dev, or all drivers
 | 
|---|
| 984 |  * if dev not specified.
 | 
|---|
| 985 |  */
 | 
|---|
| 986 | static void
 | 
|---|
| 987 | q2RDrivers (const char *dev, Msg *mp, XMLEle *root)
 | 
|---|
| 988 | {
 | 
|---|
| 989 |         int sawremote = 0;
 | 
|---|
| 990 |         DvrInfo *dp;
 | 
|---|
| 991 | 
 | 
|---|
| 992 |         /* queue message to each interested driver.
 | 
|---|
| 993 |          * N.B. don't send generic getProps to more than one remote driver,
 | 
|---|
| 994 |          *   otherwise they all fan out and we get multiple responses back.
 | 
|---|
| 995 |          */
 | 
|---|
| 996 |         for (dp = dvrinfo; dp < &dvrinfo[ndvrinfo]; dp++) {
 | 
|---|
| 997 |             int isremote = (dp->pid == REMOTEDVR);
 | 
|---|
| 998 |             if (dev[0] && dp->dev[0] && strcmp (dev, dp->dev))
 | 
|---|
| 999 |                 continue;       /* driver known to not support this dev */
 | 
|---|
| 1000 |             if (!dev[0] && isremote && sawremote)
 | 
|---|
| 1001 |                 continue;       /* already sent generic to another remote */
 | 
|---|
| 1002 |             if (isremote)
 | 
|---|
| 1003 |                 sawremote = 1;
 | 
|---|
| 1004 | 
 | 
|---|
| 1005 |             /* ok: queue message to this driver */
 | 
|---|
| 1006 |             mp->count++;
 | 
|---|
| 1007 |             pushFQ (dp->msgq, mp);
 | 
|---|
| 1008 |             if (verbose > 1)
 | 
|---|
| 1009 |                 fprintf (stderr, "%s: Driver %s: queuing responsible for <%s device='%s' name='%s'>\n",
 | 
|---|
| 1010 |                                     indi_tstamp(NULL), dp->name, tagXMLEle(root),
 | 
|---|
| 1011 |                                     findXMLAttValu (root, "device"),
 | 
|---|
| 1012 |                                     findXMLAttValu (root, "name"));
 | 
|---|
| 1013 |         }
 | 
|---|
| 1014 | }
 | 
|---|
| 1015 | 
 | 
|---|
| 1016 | /* put Msg mp on queue of each driver snooping dev/name.
 | 
|---|
| 1017 |  * if BLOB always honor current mode.
 | 
|---|
| 1018 |  */
 | 
|---|
| 1019 | static void
 | 
|---|
| 1020 | q2SDrivers (int isblob, const char *dev, const char *name, Msg *mp, XMLEle *root)
 | 
|---|
| 1021 | {
 | 
|---|
| 1022 |         DvrInfo *dp;
 | 
|---|
| 1023 | 
 | 
|---|
| 1024 |         for (dp = dvrinfo; dp < &dvrinfo[ndvrinfo]; dp++) {
 | 
|---|
| 1025 |             Snoopee *sp = findSDevice (dp, dev, name);
 | 
|---|
| 1026 | 
 | 
|---|
| 1027 |             /* nothing for dp if not snooping for dev/name or wrong BLOB mode */
 | 
|---|
| 1028 |             if (!sp)
 | 
|---|
| 1029 |                 continue;
 | 
|---|
| 1030 |             if ((isblob && sp->blob==B_NEVER) || (!isblob && sp->blob==B_ONLY))
 | 
|---|
| 1031 |                 continue;
 | 
|---|
| 1032 | 
 | 
|---|
| 1033 |             /* ok: queue message to this device */
 | 
|---|
| 1034 |             mp->count++;
 | 
|---|
| 1035 |             pushFQ (dp->msgq, mp);
 | 
|---|
| 1036 |             if (verbose > 1) {
 | 
|---|
| 1037 |                 fprintf (stderr, "%s: Driver %s: queuing snooped <%s device='%s' name='%s'>\n",
 | 
|---|
| 1038 |                                     indi_tstamp(NULL), dp->name, tagXMLEle(root),
 | 
|---|
| 1039 |                                     findXMLAttValu (root, "device"),
 | 
|---|
| 1040 |                                     findXMLAttValu (root, "name"));
 | 
|---|
| 1041 |             }
 | 
|---|
| 1042 |         }
 | 
|---|
| 1043 | }
 | 
|---|
| 1044 | 
 | 
|---|
| 1045 | /* add dev/name to dp's snooping list.
 | 
|---|
| 1046 |  * init with blob mode set to B_NEVER.
 | 
|---|
| 1047 |  */
 | 
|---|
| 1048 | static void
 | 
|---|
| 1049 | addSDevice (DvrInfo *dp, const char *dev, const char *name)
 | 
|---|
| 1050 | {
 | 
|---|
| 1051 |         Snoopee *sp;
 | 
|---|
| 1052 |         char *ip;
 | 
|---|
| 1053 | 
 | 
|---|
| 1054 |         /* no dups */
 | 
|---|
| 1055 |         sp = findSDevice (dp, dev, name);
 | 
|---|
| 1056 |         if (sp)
 | 
|---|
| 1057 |             return;
 | 
|---|
| 1058 | 
 | 
|---|
| 1059 |         /* add dev to sdevs list */
 | 
|---|
| 1060 |         dp->sprops = (Snoopee*) realloc (dp->sprops,
 | 
|---|
| 1061 |                                             (dp->nsprops+1)*sizeof(Snoopee));
 | 
|---|
| 1062 |         sp = &dp->sprops[dp->nsprops++];
 | 
|---|
| 1063 | 
 | 
|---|
| 1064 |         ip = sp->prop.dev;
 | 
|---|
| 1065 |         strncpy (ip, dev, MAXINDIDEVICE-1);
 | 
|---|
| 1066 |         ip[MAXINDIDEVICE-1] = '\0';
 | 
|---|
| 1067 | 
 | 
|---|
| 1068 |         ip = sp->prop.name;
 | 
|---|
| 1069 |         strncpy (ip, name, MAXINDINAME-1);
 | 
|---|
| 1070 |         ip[MAXINDINAME-1] = '\0';
 | 
|---|
| 1071 | 
 | 
|---|
| 1072 |         sp->blob = B_NEVER;
 | 
|---|
| 1073 | 
 | 
|---|
| 1074 |         if (verbose)
 | 
|---|
| 1075 |             fprintf (stderr, "%s: Driver %s: snooping on %s.%s\n", indi_tstamp(NULL),
 | 
|---|
| 1076 |                                                         dp->name, dev, name);
 | 
|---|
| 1077 | }
 | 
|---|
| 1078 | 
 | 
|---|
| 1079 | /* return Snoopee if dp is snooping dev/name, else NULL.
 | 
|---|
| 1080 |  */
 | 
|---|
| 1081 | static Snoopee *
 | 
|---|
| 1082 | findSDevice (DvrInfo *dp, const char *dev, const char *name)
 | 
|---|
| 1083 | {
 | 
|---|
| 1084 |         int i;
 | 
|---|
| 1085 | 
 | 
|---|
| 1086 |         for (i = 0; i < dp->nsprops; i++) {
 | 
|---|
| 1087 |             Snoopee *sp = &dp->sprops[i];
 | 
|---|
| 1088 |             Property *pp = &sp->prop;
 | 
|---|
| 1089 |             if (!strcmp (pp->dev, dev) &&
 | 
|---|
| 1090 |                     (!pp->name[0] || !strcmp(pp->name, name)))
 | 
|---|
| 1091 |                 return (sp);
 | 
|---|
| 1092 |         }
 | 
|---|
| 1093 | 
 | 
|---|
| 1094 |         return (NULL);
 | 
|---|
| 1095 | }
 | 
|---|
| 1096 | 
 | 
|---|
| 1097 | /* put Msg mp on queue of each client interested in dev/name, except notme.
 | 
|---|
| 1098 |  * if BLOB always honor current mode.
 | 
|---|
| 1099 |  * return -1 if had to shut down any clients, else 0.
 | 
|---|
| 1100 |  */
 | 
|---|
| 1101 | static int
 | 
|---|
| 1102 | q2Clients (ClInfo *notme, int isblob, const char *dev, const char *name, Msg *mp, XMLEle *root)
 | 
|---|
| 1103 | {
 | 
|---|
| 1104 |         int shutany = 0;
 | 
|---|
| 1105 |         ClInfo *cp;
 | 
|---|
| 1106 |         int ql;
 | 
|---|
| 1107 | 
 | 
|---|
| 1108 |         /* queue message to each interested client */
 | 
|---|
| 1109 |         for (cp = clinfo; cp < &clinfo[nclinfo]; cp++) {
 | 
|---|
| 1110 |             /* cp in use? notme? want this dev/name? blob? */
 | 
|---|
| 1111 |             if (!cp->active || cp == notme)
 | 
|---|
| 1112 |                 continue;
 | 
|---|
| 1113 |             if (findClDevice (cp, dev, name) < 0)
 | 
|---|
| 1114 |                 continue;
 | 
|---|
| 1115 |             if ((isblob && cp->blob==B_NEVER) || (!isblob && cp->blob==B_ONLY))
 | 
|---|
| 1116 |                 continue;
 | 
|---|
| 1117 | 
 | 
|---|
| 1118 |             /* shut down this client if its q is already too large */
 | 
|---|
| 1119 |             ql = msgQSize(cp->msgq);
 | 
|---|
| 1120 |             if (ql > maxqsiz) {
 | 
|---|
| 1121 |                 if (verbose)
 | 
|---|
| 1122 |                     fprintf (stderr, "%s: Client %d: %d bytes behind, shutting down\n",
 | 
|---|
| 1123 |                                                     indi_tstamp(NULL), cp->s, ql);
 | 
|---|
| 1124 |                 shutdownClient (cp);
 | 
|---|
| 1125 |                 shutany++;
 | 
|---|
| 1126 |                 continue;
 | 
|---|
| 1127 |             }
 | 
|---|
| 1128 | 
 | 
|---|
| 1129 |             /* ok: queue message to this client */
 | 
|---|
| 1130 |             mp->count++;
 | 
|---|
| 1131 |             pushFQ (cp->msgq, mp);
 | 
|---|
| 1132 |             if (verbose > 1)
 | 
|---|
| 1133 |                 fprintf (stderr, "%s: Client %d: queuing <%s device='%s' name='%s'>\n",
 | 
|---|
| 1134 |                                     indi_tstamp(NULL), cp->s, tagXMLEle(root),
 | 
|---|
| 1135 |                                     findXMLAttValu (root, "device"),
 | 
|---|
| 1136 |                                     findXMLAttValu (root, "name"));
 | 
|---|
| 1137 |         }
 | 
|---|
| 1138 | 
 | 
|---|
| 1139 |         return (shutany ? -1 : 0);
 | 
|---|
| 1140 | }
 | 
|---|
| 1141 | 
 | 
|---|
| 1142 | /* return size of all Msqs on the given q */
 | 
|---|
| 1143 | static int
 | 
|---|
| 1144 | msgQSize (FQ *q)
 | 
|---|
| 1145 | {
 | 
|---|
| 1146 |         int i, l = 0;
 | 
|---|
| 1147 | 
 | 
|---|
| 1148 |         for (i = 0; i < nFQ(q); i++) {
 | 
|---|
| 1149 |             Msg *mp = (Msg *) peekiFQ(q,i);
 | 
|---|
| 1150 |             l += mp->cl;
 | 
|---|
| 1151 |         }
 | 
|---|
| 1152 | 
 | 
|---|
| 1153 |         return (l);
 | 
|---|
| 1154 | }
 | 
|---|
| 1155 | 
 | 
|---|
| 1156 | /* print root as content in Msg mp.
 | 
|---|
| 1157 |  */
 | 
|---|
| 1158 | static void
 | 
|---|
| 1159 | setMsgXMLEle (Msg *mp, XMLEle *root)
 | 
|---|
| 1160 | {
 | 
|---|
| 1161 |         /* want cl to only count content, but need room for final \0 */
 | 
|---|
| 1162 |         mp->cl = sprlXMLEle (root, 0);
 | 
|---|
| 1163 |         if (mp->cl < sizeof(mp->buf))
 | 
|---|
| 1164 |             mp->cp = mp->buf;
 | 
|---|
| 1165 |         else
 | 
|---|
| 1166 |             mp->cp = malloc (mp->cl+1);
 | 
|---|
| 1167 |         sprXMLEle (mp->cp, root, 0);
 | 
|---|
| 1168 | }
 | 
|---|
| 1169 | 
 | 
|---|
| 1170 | /* save str as content in Msg mp.
 | 
|---|
| 1171 |  */
 | 
|---|
| 1172 | static void
 | 
|---|
| 1173 | setMsgStr (Msg *mp, char *str)
 | 
|---|
| 1174 | {
 | 
|---|
| 1175 |         /* want cl to only count content, but need room for final \0 */
 | 
|---|
| 1176 |         mp->cl = strlen (str);
 | 
|---|
| 1177 |         if (mp->cl < sizeof(mp->buf))
 | 
|---|
| 1178 |             mp->cp = mp->buf;
 | 
|---|
| 1179 |         else
 | 
|---|
| 1180 |             mp->cp = malloc (mp->cl+1);
 | 
|---|
| 1181 |         strcpy (mp->cp, str);
 | 
|---|
| 1182 | }
 | 
|---|
| 1183 | 
 | 
|---|
| 1184 | /* return pointer to one new nulled Msg
 | 
|---|
| 1185 |  */
 | 
|---|
| 1186 | static Msg *
 | 
|---|
| 1187 | newMsg (void)
 | 
|---|
| 1188 | {
 | 
|---|
| 1189 |         return ((Msg *) calloc (1, sizeof(Msg)));
 | 
|---|
| 1190 | }
 | 
|---|
| 1191 | 
 | 
|---|
| 1192 | /* free Msg mp and everything it contains */
 | 
|---|
| 1193 | static void
 | 
|---|
| 1194 | freeMsg (Msg *mp)
 | 
|---|
| 1195 | {
 | 
|---|
| 1196 |         if (mp->cp && mp->cp != mp->buf)
 | 
|---|
| 1197 |             free (mp->cp);
 | 
|---|
| 1198 |         free (mp);
 | 
|---|
| 1199 | }
 | 
|---|
| 1200 | 
 | 
|---|
| 1201 | /* write the next chunk of the current message in the queue to the given
 | 
|---|
| 1202 |  * client. pop message from queue when complete and free the message if we are
 | 
|---|
| 1203 |  * the last one to use it. shut down this client if trouble.
 | 
|---|
| 1204 |  * N.B. we assume we will never be called with cp->msgq empty.
 | 
|---|
| 1205 |  * return 0 if ok else -1 if had to shut down.
 | 
|---|
| 1206 |  */
 | 
|---|
| 1207 | static int
 | 
|---|
| 1208 | sendClientMsg (ClInfo *cp)
 | 
|---|
| 1209 | {
 | 
|---|
| 1210 |         int nsend, nw;
 | 
|---|
| 1211 |         Msg *mp;
 | 
|---|
| 1212 | 
 | 
|---|
| 1213 |         /* get current message */
 | 
|---|
| 1214 |         mp = (Msg *) peekFQ (cp->msgq);
 | 
|---|
| 1215 | 
 | 
|---|
| 1216 |         /* send next chunk, never more than MAXWSIZ to reduce blocking */
 | 
|---|
| 1217 |         nsend = mp->cl - cp->nsent;
 | 
|---|
| 1218 |         if (nsend > MAXWSIZ)
 | 
|---|
| 1219 |             nsend = MAXWSIZ;
 | 
|---|
| 1220 |         nw = write (cp->s, &mp->cp[cp->nsent], nsend);
 | 
|---|
| 1221 | 
 | 
|---|
| 1222 |         /* shut down if trouble */
 | 
|---|
| 1223 |         if (nw <= 0) {
 | 
|---|
| 1224 |             if (nw == 0)
 | 
|---|
| 1225 |                 fprintf (stderr, "%s: Client %d: write returned 0\n",
 | 
|---|
| 1226 |                                                     indi_tstamp(NULL), cp->s);
 | 
|---|
| 1227 |             else
 | 
|---|
| 1228 |                 fprintf (stderr, "%s: Client %d: write: %s\n", indi_tstamp(NULL),
 | 
|---|
| 1229 |                                                     cp->s, strerror(errno));
 | 
|---|
| 1230 |             shutdownClient (cp);
 | 
|---|
| 1231 |             return (-1);
 | 
|---|
| 1232 |         }
 | 
|---|
| 1233 | 
 | 
|---|
| 1234 |         /* trace */
 | 
|---|
| 1235 |         if (verbose > 2) {
 | 
|---|
| 1236 |             fprintf(stderr, "%s: Client %d: sending msg copy %d nq %d:\n%.*s\n",
 | 
|---|
| 1237 |                                 indi_tstamp(NULL), cp->s, mp->count, nFQ(cp->msgq),
 | 
|---|
| 1238 |                                 nw, &mp->cp[cp->nsent]);
 | 
|---|
| 1239 |         } else if (verbose > 1) {
 | 
|---|
| 1240 |             fprintf(stderr, "%s: Client %d: sending %.50s\n", indi_tstamp(NULL),
 | 
|---|
| 1241 |                                                     cp->s, &mp->cp[cp->nsent]);
 | 
|---|
| 1242 |         }
 | 
|---|
| 1243 | 
 | 
|---|
| 1244 |         /* update amount sent. when complete: free message if we are the last
 | 
|---|
| 1245 |          * to use it and pop from our queue.
 | 
|---|
| 1246 |          */
 | 
|---|
| 1247 |         cp->nsent += nw;
 | 
|---|
| 1248 |         if (cp->nsent == mp->cl) {
 | 
|---|
| 1249 |             if (--mp->count == 0)
 | 
|---|
| 1250 |                 freeMsg (mp);
 | 
|---|
| 1251 |             popFQ (cp->msgq);
 | 
|---|
| 1252 |             cp->nsent = 0;
 | 
|---|
| 1253 |         }
 | 
|---|
| 1254 | 
 | 
|---|
| 1255 |         return (0);
 | 
|---|
| 1256 | }
 | 
|---|
| 1257 | 
 | 
|---|
| 1258 | /* write the next chunk of the current message in the queue to the given
 | 
|---|
| 1259 |  * driver. pop message from queue when complete and free the message if we are
 | 
|---|
| 1260 |  * the last one to use it. restart this driver if touble.
 | 
|---|
| 1261 |  * N.B. we assume we will never be called with dp->msgq empty.
 | 
|---|
| 1262 |  * return 0 if ok else -1 if had to shut down.
 | 
|---|
| 1263 |  */
 | 
|---|
| 1264 | static int
 | 
|---|
| 1265 | sendDriverMsg (DvrInfo *dp)
 | 
|---|
| 1266 | {
 | 
|---|
| 1267 |         int nsend, nw;
 | 
|---|
| 1268 |         Msg *mp;
 | 
|---|
| 1269 | 
 | 
|---|
| 1270 |         /* get current message */
 | 
|---|
| 1271 |         mp = (Msg *) peekFQ (dp->msgq);
 | 
|---|
| 1272 | 
 | 
|---|
| 1273 |         /* send next chunk, never more than MAXWSIZ to reduce blocking */
 | 
|---|
| 1274 |         nsend = mp->cl - dp->nsent;
 | 
|---|
| 1275 |         if (nsend > MAXWSIZ)
 | 
|---|
| 1276 |             nsend = MAXWSIZ;
 | 
|---|
| 1277 |         nw = write (dp->wfd, &mp->cp[dp->nsent], nsend);
 | 
|---|
| 1278 | 
 | 
|---|
| 1279 |         /* restart if trouble */
 | 
|---|
| 1280 |         if (nw <= 0) {
 | 
|---|
| 1281 |             if (nw == 0)
 | 
|---|
| 1282 |                 fprintf (stderr, "%s: Driver %s: write returned 0\n",
 | 
|---|
| 1283 |                                                     indi_tstamp(NULL), dp->name);
 | 
|---|
| 1284 |             else
 | 
|---|
| 1285 |                 fprintf (stderr, "%s: Driver %s: write: %s\n", indi_tstamp(NULL),
 | 
|---|
| 1286 |                                                     dp->name, strerror(errno));
 | 
|---|
| 1287 |             restartDvr (dp);
 | 
|---|
| 1288 |             return (-1);
 | 
|---|
| 1289 |         }
 | 
|---|
| 1290 | 
 | 
|---|
| 1291 |         /* trace */
 | 
|---|
| 1292 |         if (verbose > 2) {
 | 
|---|
| 1293 |             fprintf(stderr, "%s: Driver %s: sending msg copy %d nq %d:\n%.*s\n",
 | 
|---|
| 1294 |                             indi_tstamp(NULL), dp->name, mp->count, nFQ(dp->msgq),
 | 
|---|
| 1295 |                             nw, &mp->cp[dp->nsent]);
 | 
|---|
| 1296 |         } else if (verbose > 1) {
 | 
|---|
| 1297 |             fprintf(stderr, "%s: Driver %s: sending %.50s\n", indi_tstamp(NULL),
 | 
|---|
| 1298 |                                                 dp->name, &mp->cp[dp->nsent]);
 | 
|---|
| 1299 |         }
 | 
|---|
| 1300 | 
 | 
|---|
| 1301 |         /* update amount sent. when complete: free message if we are the last
 | 
|---|
| 1302 |          * to use it and pop from our queue.
 | 
|---|
| 1303 |          */
 | 
|---|
| 1304 |         dp->nsent += nw;
 | 
|---|
| 1305 |         if (dp->nsent == mp->cl) {
 | 
|---|
| 1306 |             if (--mp->count == 0)
 | 
|---|
| 1307 |                 freeMsg (mp);
 | 
|---|
| 1308 |             popFQ (dp->msgq);
 | 
|---|
| 1309 |             dp->nsent = 0;
 | 
|---|
| 1310 |         }
 | 
|---|
| 1311 | 
 | 
|---|
| 1312 |         return (0);
 | 
|---|
| 1313 | }
 | 
|---|
| 1314 | 
 | 
|---|
| 1315 | /* return 0 if cp may be interested in dev/name else -1
 | 
|---|
| 1316 |  */
 | 
|---|
| 1317 | static int
 | 
|---|
| 1318 | findClDevice (ClInfo *cp, const char *dev, const char *name)
 | 
|---|
| 1319 | {
 | 
|---|
| 1320 |         int i;
 | 
|---|
| 1321 | 
 | 
|---|
| 1322 |         if (cp->allprops || !dev[0])
 | 
|---|
| 1323 |             return (0);
 | 
|---|
| 1324 |         for (i = 0; i < cp->nprops; i++) {
 | 
|---|
| 1325 |             Property *pp = &cp->props[i];
 | 
|---|
| 1326 |             if (!strcmp (pp->dev, dev) &&
 | 
|---|
| 1327 |                                     (!pp->name[0] || !strcmp(pp->name, name)))
 | 
|---|
| 1328 |                 return (0);
 | 
|---|
| 1329 |         }
 | 
|---|
| 1330 |         return (-1);
 | 
|---|
| 1331 | }
 | 
|---|
| 1332 | 
 | 
|---|
| 1333 | /* add the given device and property to the devs[] list of client if new.
 | 
|---|
| 1334 |  */
 | 
|---|
| 1335 | static void
 | 
|---|
| 1336 | addClDevice (ClInfo *cp, const char *dev, const char *name)
 | 
|---|
| 1337 | {
 | 
|---|
| 1338 |         Property *pp;
 | 
|---|
| 1339 |         char *ip;
 | 
|---|
| 1340 | 
 | 
|---|
| 1341 |         /* no dups */
 | 
|---|
| 1342 |         if (!findClDevice (cp, dev, name))
 | 
|---|
| 1343 |             return;
 | 
|---|
| 1344 | 
 | 
|---|
| 1345 |         /* add */
 | 
|---|
| 1346 |         cp->props = (Property *) realloc (cp->props,
 | 
|---|
| 1347 |                                             (cp->nprops+1)*sizeof(Property));
 | 
|---|
| 1348 |         pp = &cp->props[cp->nprops++];
 | 
|---|
| 1349 | 
 | 
|---|
| 1350 |         ip = pp->dev;
 | 
|---|
| 1351 |         strncpy (ip, dev, MAXINDIDEVICE-1);
 | 
|---|
| 1352 |         ip[MAXINDIDEVICE-1] = '\0';
 | 
|---|
| 1353 | 
 | 
|---|
| 1354 |         ip = pp->name;
 | 
|---|
| 1355 |         strncpy (ip, name, MAXINDINAME-1);
 | 
|---|
| 1356 |         ip[MAXINDINAME-1] = '\0';
 | 
|---|
| 1357 | }
 | 
|---|
| 1358 | 
 | 
|---|
| 1359 | 
 | 
|---|
| 1360 | /* block to accept a new client arriving on lsocket.
 | 
|---|
| 1361 |  * return private nonblocking socket or exit.
 | 
|---|
| 1362 |  */
 | 
|---|
| 1363 | static int
 | 
|---|
| 1364 | newClSocket ()
 | 
|---|
| 1365 | {
 | 
|---|
| 1366 |         struct sockaddr_in cli_socket;
 | 
|---|
| 1367 |         socklen_t cli_len;
 | 
|---|
| 1368 |         int cli_fd;
 | 
|---|
| 1369 | 
 | 
|---|
| 1370 |         /* get a private connection to new client */
 | 
|---|
| 1371 |         cli_len = sizeof(cli_socket);
 | 
|---|
| 1372 |         cli_fd = accept (lsocket, (struct sockaddr *)&cli_socket, &cli_len);
 | 
|---|
| 1373 |         if(cli_fd < 0) {
 | 
|---|
| 1374 |             fprintf (stderr, "accept: %s\n", strerror(errno));
 | 
|---|
| 1375 |             Bye();
 | 
|---|
| 1376 |         }
 | 
|---|
| 1377 | 
 | 
|---|
| 1378 |         /* ok */
 | 
|---|
| 1379 |         return (cli_fd);
 | 
|---|
| 1380 | }
 | 
|---|
| 1381 | 
 | 
|---|
| 1382 | /* convert the string value of enableBLOB to our B_ state value.
 | 
|---|
| 1383 |  * no change if unrecognized
 | 
|---|
| 1384 |  */
 | 
|---|
| 1385 | static void
 | 
|---|
| 1386 | crackBLOB (char *enableBLOB, BLOBHandling *bp)
 | 
|---|
| 1387 | {
 | 
|---|
| 1388 |         if (!strcmp (enableBLOB, "Also"))
 | 
|---|
| 1389 |             *bp = B_ALSO;
 | 
|---|
| 1390 |         else if (!strcmp (enableBLOB, "Only"))
 | 
|---|
| 1391 |             *bp = B_ONLY;
 | 
|---|
| 1392 |         else if (!strcmp (enableBLOB, "Never"))
 | 
|---|
| 1393 |             *bp = B_NEVER;
 | 
|---|
| 1394 | }
 | 
|---|
| 1395 | 
 | 
|---|
| 1396 | /* print key attributes and values of the given xml to stderr.
 | 
|---|
| 1397 |  */
 | 
|---|
| 1398 | static void
 | 
|---|
| 1399 | traceMsg (XMLEle *root)
 | 
|---|
| 1400 | {
 | 
|---|
| 1401 |         static const char *prtags[] = {
 | 
|---|
| 1402 |             "defNumber", "oneNumber",
 | 
|---|
| 1403 |             "defText",   "oneText",
 | 
|---|
| 1404 |             "defSwitch", "oneSwitch",
 | 
|---|
| 1405 |             "defLight",  "oneLight",
 | 
|---|
| 1406 |         };
 | 
|---|
| 1407 |         XMLEle *e;
 | 
|---|
| 1408 |         const char *msg, *perm, *pcd;
 | 
|---|
| 1409 |         unsigned int i;
 | 
|---|
| 1410 | 
 | 
|---|
| 1411 |         /* print tag header */
 | 
|---|
| 1412 |         fprintf (stderr, "%s %s %s %s", tagXMLEle(root),
 | 
|---|
| 1413 |                                                 findXMLAttValu(root,"device"),
 | 
|---|
| 1414 |                                                 findXMLAttValu(root,"name"),
 | 
|---|
| 1415 |                                                 findXMLAttValu(root,"state"));
 | 
|---|
| 1416 |         pcd = pcdataXMLEle (root);
 | 
|---|
| 1417 |         if (pcd[0])
 | 
|---|
| 1418 |             fprintf (stderr, " %s", pcd);
 | 
|---|
| 1419 |         perm = findXMLAttValu(root,"perm");
 | 
|---|
| 1420 |         if (perm[0])
 | 
|---|
| 1421 |             fprintf (stderr, " %s", perm);
 | 
|---|
| 1422 |         msg = findXMLAttValu(root,"message");
 | 
|---|
| 1423 |         if (msg[0])
 | 
|---|
| 1424 |             fprintf (stderr, " '%s'", msg);
 | 
|---|
| 1425 | 
 | 
|---|
| 1426 |         /* print each array value */
 | 
|---|
| 1427 |         for (e = nextXMLEle(root,1); e; e = nextXMLEle(root,0))
 | 
|---|
| 1428 |             for (i = 0; i < sizeof(prtags)/sizeof(prtags[0]); i++)
 | 
|---|
| 1429 |                 if (strcmp (prtags[i], tagXMLEle(e)) == 0)
 | 
|---|
| 1430 |                     fprintf (stderr, "\n %10s='%s'", findXMLAttValu(e,"name"),
 | 
|---|
| 1431 |                                                             pcdataXMLEle (e));
 | 
|---|
| 1432 | 
 | 
|---|
| 1433 |         fprintf (stderr, "\n");
 | 
|---|
| 1434 | }
 | 
|---|
| 1435 | 
 | 
|---|
| 1436 | /* fill s with current UT string.
 | 
|---|
| 1437 |  * if no s, use a static buffer
 | 
|---|
| 1438 |  * return s or buffer.
 | 
|---|
| 1439 |  * N.B. if use our buffer, be sure to use before calling again
 | 
|---|
| 1440 |  */
 | 
|---|
| 1441 | static char *
 | 
|---|
| 1442 | indi_tstamp (char *s)
 | 
|---|
| 1443 | {
 | 
|---|
| 1444 |         static char sbuf[64];
 | 
|---|
| 1445 |         struct tm *tp;
 | 
|---|
| 1446 |         time_t t;
 | 
|---|
| 1447 | 
 | 
|---|
| 1448 |         time (&t);
 | 
|---|
| 1449 |         tp = gmtime (&t);
 | 
|---|
| 1450 |         if (!s)
 | 
|---|
| 1451 |             s = sbuf;
 | 
|---|
| 1452 |         strftime (s, sizeof(sbuf), "%Y-%m-%dT%H:%M:%S", tp);
 | 
|---|
| 1453 |         return (s);
 | 
|---|
| 1454 | }
 | 
|---|
| 1455 | 
 | 
|---|
| 1456 | /* log message in root known to be from device dev to ldir, if any.
 | 
|---|
| 1457 |  */
 | 
|---|
| 1458 | static void
 | 
|---|
| 1459 | logDMsg (XMLEle *root, const char *dev)
 | 
|---|
| 1460 | {
 | 
|---|
| 1461 |         char stamp[64];
 | 
|---|
| 1462 |         char logfn[1024];
 | 
|---|
| 1463 |         const char *ts, *ms;
 | 
|---|
| 1464 |         FILE *fp;
 | 
|---|
| 1465 | 
 | 
|---|
| 1466 |         /* get message, if any */
 | 
|---|
| 1467 |         ms = findXMLAttValu (root, "message");
 | 
|---|
| 1468 |         if (!ms[0])
 | 
|---|
| 1469 |             return;
 | 
|---|
| 1470 | 
 | 
|---|
| 1471 |         /* get timestamp now if not provided */
 | 
|---|
| 1472 |         ts = findXMLAttValu (root, "timestamp");
 | 
|---|
| 1473 |         if (!ts[0])
 | 
|---|
| 1474 |         {
 | 
|---|
| 1475 |             indi_tstamp (stamp);
 | 
|---|
| 1476 |             ts = stamp;
 | 
|---|
| 1477 |         }
 | 
|---|
| 1478 | 
 | 
|---|
| 1479 |         /* append to log file, name is date portion of time stamp */
 | 
|---|
| 1480 |         sprintf (logfn, "%s/%.10s.islog", ldir, ts);
 | 
|---|
| 1481 |         fp = fopen (logfn, "a");
 | 
|---|
| 1482 |         if (!fp)
 | 
|---|
| 1483 |             return;     /* oh well */
 | 
|---|
| 1484 |         fprintf (fp, "%s: %s: %s\n", ts, dev, ms);
 | 
|---|
| 1485 |         fclose (fp);
 | 
|---|
| 1486 | }
 | 
|---|
| 1487 | 
 | 
|---|
| 1488 | /* log when then exit */
 | 
|---|
| 1489 | static void
 | 
|---|
| 1490 | Bye()
 | 
|---|
| 1491 | {
 | 
|---|
| 1492 |         fprintf (stderr, "%s: good bye\n", indi_tstamp(NULL));
 | 
|---|
| 1493 |         exit(1);
 | 
|---|
| 1494 | }
 | 
|---|
| 1495 | 
 | 
|---|