src/calculator_server/main.c

   1 /*
   2  * Copyright (c) 2009, Olivier MATZ <zer0@droids-corp.org>
   3  * All rights reserved.
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  *     * Redistributions of source code must retain the above copyright
   8  *       notice, this list of conditions and the following disclaimer.
   9  *     * Redistributions in binary form must reproduce the above copyright
  10  *       notice, this list of conditions and the following disclaimer in the
  11  *       documentation and/or other materials provided with the distribution.
  12  *     * Neither the name of the University of California, Berkeley nor the
  13  *       names of its contributors may be used to endorse or promote products
  14  *       derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <stdint.h>
  32 #include <stdlib.h>
  33 #include <stdarg.h>
  34 #include <inttypes.h>
  35 #include <ctype.h>
  36 #include <unistd.h>
  37 #include <netinet/in.h>
  38 #include <arpa/inet.h>
  39
  40 #include <getopt.h>
  41
  42 #include <cmdline_rdline.h>
  43 #include <cmdline_parse.h>
  44 #include <cmdline_socket.h>
  45 #include <cmdline.h>
  46
  47 extern cmdline_parse_ctx_t main_ctx[];
  48
  49 static struct sockaddr addr;
  50 static uint16_t port = 1234;
  51
  52 static void
  53 usage(const char *prgname)
  54 {
  55         printf("%s [-a ADDRESS] [-p PORT]\n"
  56                "\n"
  57                "Start a command line server on a TCP listen socket. Only\n"
  58                "one connection is accepted, then the program exits.\n"
  59                "The default behaviour is to listen on 0.0.0.0:1234\n"
  60                "\n"
  61                "  -a ADDR:  IPv4 or IPv6 local address to listen on (default\n"
  62                "            is 0.0.0.0).\n"
  63                "  -p PORT:  port to listen on (default is 1234).\n",
  64                prgname);
  65 }
  66
  67 /* Parse the argument given in the command line of the application */
  68 static int
  69 parse_args(int argc, char **argv)
  70 {
  71         int opt, ret;
  72         char **argvopt;
  73         int option_index;
  74         char *prgname = argv[0];
  75         char *end = NULL;
  76         static struct option lgopts[] = {
  77                 {0, 0, 0, 0}
  78         };
  79
  80         argvopt = argv;
  81
  82         while ((opt = getopt_long(argc, argvopt, "a:p:",
  83                                   lgopts, &option_index)) != EOF) {
  84
  85                 switch (opt) {
  86
  87                 case 'a':
  88                         if (strchr(optarg, '.') != NULL) {
  89                                 struct sockaddr_in *sin =
  90                                         (struct sockaddr_in *)&addr;
  91                                 addr.sa_family = AF_INET;
  92                                 ret = inet_pton(AF_INET, optarg,
  93                                                 &sin->sin_addr);
  94                         }
  95                         else {
  96                                 struct sockaddr_in6 *sin6 =
  97                                         (struct sockaddr_in6 *)&addr;
  98                                 addr.sa_family = AF_INET6;
  99                                 ret = inet_pton(AF_INET6, optarg,
 100                                                 &sin6->sin6_addr);
 101                         }
 102
 103                         if (ret != 1) {
 104                                 printf("Bad address\n");
 105                                 usage(prgname);
 106                                 return -1;
 107                         }
 108                         break;
 109
 110                 case 'p':
 111                         port = strtoul(optarg, &end, 0);
 112                         if ((optarg[0] == '\0') ||
 113                             (end == NULL) ||
 114                             (*end != '\0')) {
 115                                 printf("Invalid port\n");
 116                                 usage(prgname);
 117                                 return -1;
 118                         }
 119
 120                         break;
 121
 122                 /* long options */
 123                 case 0:
 124                         /* if (!strcmp(lgopts[option_index].name, "option")) */
 125                         break;
 126
 127                 default:
 128                         usage(prgname);
 129                         return -1;
 130                 }
 131         }
 132
 133         if (argc != optind) {
 134                 printf("Invalid argument\n");
 135                 usage(prgname);
 136                 return -1;
 137         }
 138
 139         ret = optind-1;
 140
 141         return ret;
 142 }
 143
 144 int main(int argc, char **argv)
 145 {
 146         struct sockaddr_in *sin = (struct sockaddr_in *)&addr;
 147         struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&addr;
 148         struct cmdline *cl;
 149         int s;
 150
 151         if (parse_args(argc, argv) < 0)
 152                 exit(1);
 153
 154         printf("Listening on port %d\n", port);
 155
 156         /* open a tcp server on specified port */
 157         if (addr.sa_family == AF_INET)
 158                 s = cmdline_tcpv4_listen(sin->sin_addr.s_addr, port);
 159         else
 160                 s = cmdline_tcpv6_listen(sin6->sin6_addr, port);
 161
 162         cl = cmdline_accept(main_ctx, "example> ", s);
 163         if (cl == NULL) {
 164                 printf("accept() failed\n");
 165                 return 1;
 166         }
 167         cmdline_interact(cl);
 168         cmdline_free(cl);
 169         return 0;
 170 }
 171
 172