sources/FortuneServer.c

   1 /* FortuneServer.c
   2  *
   3  * Copyright (C) 2002 Simone Piccardi
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or (at
   8  * your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  */
  19 /****************************************************************
  20  *
  21  * Program fortuned
  22  * Fortune server
  23  *
  24  * Author: Simone Piccardi
  25  * Aug. 2002
  26  *
  27  * Usage: fortuned -h give all info
  28  *
  29  * $Id: FortuneServer.c,v 1.3 2002/08/18 23:24:44 piccardi Exp $
  30  *
  31  ****************************************************************/
  32 /*
  33  * Include needed headers
  34  */
  35 #include <sys/types.h>   /* predefined types */
  36 #include <sys/stat.h>    /*  */
  37 #include <unistd.h>      /* include unix standard library */
  38 #include <stdio.h>       /* include standard I/O library */
  39 #include <stdlib.h>      /* standard library */
  40 #include <string.h>      /* ANSI C standard string */
  41 #include <errno.h>       /* errorstring */
  42 #include <fcntl.h>       /*  */
  43
  44 #include "macros.h"
  45
  46 /* Subroutines declaration */
  47 void usage(void);
  48 int FortuneParse(char *file, char **fortune, int n);
  49
  50 /* name of well known fifo */
  51 char *fifoname = "/tmp/fortune.fifo";
  52 int main(int argc, char *argv[])
  53 {
  54 /* Variables definition */
  55     int i, n = 0;
  56     char *fortunefilename = "/usr/share/games/fortunes/kids";
  57     char **fortune;
  58     char line[80];
  59     int fifo_server, fifo_client;
  60     int nread;
  61     /*
  62      * Input section: decode parameters passed in the calling
  63      * Use getopt function
  64      */
  65     opterr = 0;                              /* don't want writing to stderr */
  66     while ( (i = getopt(argc, argv, "hn:f:")) != -1) {
  67         switch (i) {
  68         /*
  69          * Handling options
  70          */
  71         case 'h':
  72             printf("Wrong -h option use\n");
  73             usage();
  74             return(0);
  75             break;
  76         case 'f':
  77             fortunefilename = optarg;
  78             break;
  79         case 'n':
  80             n = strtol(optarg, NULL, 10);
  81             fortune = (char **) calloc(sizeof(*fortune), n);
  82             break;
  83         case '?':                                    /* unrecognized options */
  84             printf("Unrecognized options -%c\n",optopt);
  85             usage();
  86         default:                                       /* should not reached */
  87             usage();
  88         }
  89     }
  90     /* ***********************************************************
  91      *
  92      *           Options processing completed
  93      *
  94      *                Main code beginning
  95      *
  96      * ***********************************************************/
  97     Signal(SIGTERM, HandSIGTERM);
  98     Signal(SIGINT, HandSIGTERM);
  99     Signal(SIGQUIT, HandSIGTERM);
 100     if (n==0) usage();          /* if no pool depth exit printing usage info */
 101     i = FortuneParse(fortunefilename, fortune, n);          /* parse phrases */
 102     for (n=0; n<i; n++) debug("%s\n\n", fortune[n]);
 103     /*
 104      * Comunication section
 105      */
 106     if (mkfifo(fifoname, 0622)) {  /* create well known fifo if does't exist */
 107         if (errno!=EEXIST) {
 108             perror("Cannot create well known fifo");
 109             exit(-1);
 110         }
 111     }
 112     /* Main body: loop over requests */
 113     while (1) {
 114         fifo_server = open(fifoname, O_RDONLY);      /* open well known fifo */
 115         if (fifo_server < 0) {
 116             perror("Cannot open well known fifo");
 117             exit(-1);
 118         }
 119         nread = read(fifo_server, line, 79);                 /* read request */
 120         line[nread] = 0;
 121         debug("%s %d\n", line,nread);
 122         n = random() % i;                             /* select random value */
 123         debug("fortune[%d]=%s\n", n, fortune[n]);
 124         fifo_client = open(line, O_WRONLY);              /* open client fifo */
 125         nread = write(fifo_client,                           /* write phrase */
 126                       fortune[n], strlen(fortune[n])+1);
 127         close(fifo_client);                         /* close well known fifo */
 128         close(fifo_server);                             /* close client fifo */
 129     }
 130 }
 131 /*
 132  * routine to print usage info and exit
 133  */
 134 void usage(void) {
 135     printf("Elementary fortune server\n");
 136     printf("Usage:\n");
 137     printf("  fortuned [-h] [-f] -n XXX \n");
 138     printf("  -h   print this help\n");
 139     printf("  -f filename   set file for fortunes\n");
 140     printf("  -n XXX        set pool depth\n");
 141     exit(1);
 142 }
 143 /*
 144  * Signal Handler to manage termination
 145  */
 146 void HandSIGTERM(int signo) {
 147     unlink(fifoname);
 148     exit(0);
 149 }