Logo Search packages:      
Sourcecode: hamlib version File versions  Download package

rotctld.c

/*
 * rotctld.c - (C) Stephane Fillod 2000-2008
 *
 * This program test/control a rotator using Hamlib.
 * It takes commands from network connection.
 *
 *    $Id: rotctld.c,v 1.6 2008/11/02 12:39:36 fillods Exp $  
 *
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 * 
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <ctype.h>

#include <getopt.h>
#include <errno.h>

#include <sys/types.h>          /* See NOTES */

#ifdef HAVE_NETINET_IN_H
#include <netinet/in.h>
#endif
#ifdef HAVE_ARPA_INET_H
#include <arpa/inet.h>
#endif
#ifdef HAVE_SYS_SOCKET_H
#include <sys/socket.h>
#elif HAVE_WS2TCPIP_H
#include <ws2tcpip.h>
#endif

#ifdef HAVE_PTHREAD
#include <pthread.h>
#endif

#include <hamlib/rotator.h>
#include "misc.h"

#include "rotctl_parse.h"

struct handle_data {
      ROT *rot;
      int sock;
      struct sockaddr_in cli_addr;
      socklen_t clilen;
};

void * handle_socket(void * arg);

void usage();

/*
 * Reminder: when adding long options, 
 * keep up to date SHORT_OPTIONS, usage()'s output and man page. thanks.
 * NB: do NOT use -W since it's reserved by POSIX.
 * TODO: add an option to read from a file
 */
#define SHORT_OPTIONS "m:r:s:C:t:T:LevhVl"
static struct option long_options[] =
{
      {"model",    1, 0, 'm'},
      {"rot-file", 1, 0, 'r'},
      {"serial-speed", 1, 0, 's'},
      {"port",  1, 0, 't'},
      {"listen-addr",  1, 0, 'T'},
      {"list",     0, 0, 'l'},
      {"set-conf", 1, 0, 'C'},
      {"show-conf",0, 0, 'L'},
      {"end-marker", 0, 0, 'e'},
      {"verbose",  0, 0, 'v'},
      {"help",     0, 0, 'h'},
      {"version",  0, 0, 'V'},
      {0, 0, 0, 0}
};

int interactive = 1;    /* no cmd because of daemon */
int prompt= 0 ;         /* Daemon mode for rigparse return string */
int opt_end= 0 ;        /* END marker for rotctld */

int portno = 4533;
uint32_t src_addr = INADDR_ANY;

#define MAXCONFLEN 128


int main (int argc, char *argv[])
{ 
      ROT *my_rot;            /* handle to rot (instance) */
      rot_model_t my_model = ROT_MODEL_DUMMY;

      int retcode;            /* generic return code from functions */

      int verbose = 0;
      int show_conf = 0;
      const char *rot_file=NULL;
      int serial_rate = 0;
      char conf_parms[MAXCONFLEN] = "";

      int sock_listen;
      struct sockaddr_in serv_addr;
      int reuseaddr = 1;
      int a0,a1,a2,a3;

      while(1) {
            int c;
            int option_index = 0;

            c = getopt_long (argc, argv, SHORT_OPTIONS,
                  long_options, &option_index);
            if (c == -1)
                  break;

            switch(c) {
                  case 'h':
                              usage();
                              exit(0);
                  case 'V':
                              version();
                              exit(0);
                  case 'm':
                              if (!optarg) {
                                    usage();    /* wrong arg count */
                                    exit(1);
                              }
                              my_model = atoi(optarg);
                              break;
                  case 'r':
                              if (!optarg) {
                                    usage();    /* wrong arg count */
                                    exit(1);
                              }
                              rot_file = optarg;
                              break;
                  case 's':
                              if (!optarg) {
                                    usage();    /* wrong arg count */
                                    exit(1);
                              }
                              serial_rate = atoi(optarg);
                              break;
                  case 'C':
                              if (!optarg) {
                                    usage();    /* wrong arg count */
                                    exit(1);
                              }
                              if (*conf_parms != '\0')
                                    strcat(conf_parms, ",");
                              strncat(conf_parms, optarg, MAXCONFLEN-strlen(conf_parms));
                              break;
                       case 't':
                              if (!optarg) {
                                    usage();        /* wrong arg count */
                                    exit(1);
                              }
                              portno = atoi(optarg);
                              break;
                  case 'T':
                              if (!optarg) {
                                    usage();    /* wrong arg count */
                                    exit(1);
                              }
                              if (4 != sscanf(optarg, "%d.%d.%d.%d", &a0,&a1,&a2,&a3)) {
                                    usage();    /* wrong arg count */
                                    exit(1);
                              }
                              src_addr = (a0<<24)|(a1<<16)|(a2<<8)|a3;
                              break;
                  case 'v':
                              verbose++;
                              break;
                  case 'L':
                              show_conf++;
                              break;
                  case 'l':
                              list_models();
                              exit(0);
                  case 'e':
                              opt_end = 1;
                              break;
                  default:
                              usage();    /* unknown option? */
                              exit(1);
            }
      }

      rig_set_debug(verbose<2 ? RIG_DEBUG_WARN: verbose);

      rig_debug(RIG_DEBUG_VERBOSE, "rotctld, %s\n", hamlib_version);
      rig_debug(RIG_DEBUG_VERBOSE, "Report bugs to "
                  "<hamlib-developer@lists.sourceforge.net>\n\n");

      my_rot = rot_init(my_model);

      if (!my_rot) {
            fprintf(stderr, "Unknown rot num %d, or initialization error.\n", 
                                    my_model);
            fprintf(stderr, "Please check with --list option.\n");
            exit(2);
      }

      retcode = set_conf(my_rot, conf_parms);
      if (retcode != RIG_OK) {
            fprintf(stderr, "Config parameter error: %s\n", rigerror(retcode));
            exit(2);
      }

      if (rot_file)
            strncpy(my_rot->state.rotport.pathname, rot_file, FILPATHLEN);

      /* FIXME: bound checking and port type == serial */
      if (serial_rate != 0)
            my_rot->state.rotport.parm.serial.rate = serial_rate;

      /*
       * print out conf parameters
       */
      if (show_conf) {
            rot_token_foreach(my_rot, print_conf_list, (rig_ptr_t)my_rot);
      }

      retcode = rot_open(my_rot);
      if (retcode != RIG_OK) {
            fprintf(stderr,"rot_open: error = %s \n", rigerror(retcode));
            exit(2);
      }

      if (verbose > 0)
            printf("Opened rot model %d, '%s'\n", my_rot->caps->rot_model,
                                    my_rot->caps->model_name);

      rig_debug(RIG_DEBUG_VERBOSE, "Backend version: %s, Status: %s\n",
                  my_rot->caps->version, rig_strstatus(my_rot->caps->status));

      /*
       * Prepare listening socket
       */
      sock_listen = socket(AF_INET, SOCK_STREAM, 0); 
      if (sock_listen < 0)  {
            perror("ERROR opening socket");
            exit(1);
      }
      memset((char *) &serv_addr, 0, sizeof(serv_addr));

      serv_addr.sin_family = AF_INET;
      serv_addr.sin_port = htons(portno);
      serv_addr.sin_addr.s_addr = htonl(src_addr);


      if (setsockopt(sock_listen, SOL_SOCKET, SO_REUSEADDR,
                        (char *)&reuseaddr,sizeof(reuseaddr)) < 0) {
            rig_debug(RIG_DEBUG_ERR, "setsockopt: %s\n", strerror(errno));
            exit (1);
      }
      if (bind(sock_listen, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
            rig_debug(RIG_DEBUG_ERR, "binding: %s\n", strerror(errno));
            exit (1);
      }
      if (listen(sock_listen,4) < 0) {
            rig_debug(RIG_DEBUG_ERR, "listening: %s\n", strerror(errno));
            exit (1);
      }

      /*
       * main loop accepting connections
       */
      do {
#ifdef HAVE_PTHREAD
            pthread_t thread;
            pthread_attr_t attr;
#endif
            struct handle_data *arg;

            arg = malloc(sizeof(struct handle_data));
            if (!arg) {
                  rig_debug(RIG_DEBUG_ERR, "malloc: %s\n", strerror(errno));
                  exit (1);
            }

            arg->rot = my_rot;
            arg->clilen = sizeof(arg->cli_addr);
            arg->sock = accept(sock_listen, (struct sockaddr *) &arg->cli_addr,
                        &arg->clilen);
            if (arg->sock < 0) {
                  rig_debug(RIG_DEBUG_ERR, "accept: %s\n", strerror(errno));
                  break;
            }

            rig_debug(RIG_DEBUG_VERBOSE, "Connection opened from %s:%d\n",
                        inet_ntoa(arg->cli_addr.sin_addr),
                        ntohs(arg->cli_addr.sin_port));

#ifdef HAVE_PTHREAD
            pthread_attr_init(&attr);
            pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);

            retcode = pthread_create(&thread, &attr, handle_socket, arg);
            if (retcode != 0) {
                  rig_debug(RIG_DEBUG_ERR, "pthread_create: %s\n", strerror(retcode));
                  break;
            }
#else
            handle_socket(arg);
#endif
      }
      while (retcode == 0);

      rot_close(my_rot); /* close port */
      rot_cleanup(my_rot); /* if you care about memory */

      return 0;
}


/*
 * This is the function run by the threads
 */
void * handle_socket(void *arg)
{
      struct handle_data *handle_data_arg = (struct handle_data *)arg;
      FILE *fsockin;
      FILE *fsockout;
      int retcode;

      fsockin = fdopen(handle_data_arg->sock, "rb");
      if (!fsockin) {
            rig_debug(RIG_DEBUG_ERR, "fdopen in: %s\n", strerror(errno));
            free(arg);
            close(handle_data_arg->sock);
#ifdef HAVE_PTHREAD
            pthread_exit(NULL);
#endif
            return NULL;
      }

      fsockout = fdopen(handle_data_arg->sock, "wb");
      if (!fsockout) {
            rig_debug(RIG_DEBUG_ERR, "fdopen out: %s\n", strerror(errno));
            fclose(fsockin);
            close(handle_data_arg->sock);
            free(arg);
#ifdef HAVE_PTHREAD
            pthread_exit(NULL);
#endif
            return NULL;
      }

      do {
            retcode = rotctl_parse(handle_data_arg->rot, fsockin, fsockout, NULL, 0);
            if (ferror(fsockin) || ferror(fsockout))
                  retcode = 1;
      }
      while (retcode == 0);

      rig_debug(RIG_DEBUG_VERBOSE, "Connection closed from %s:%d\n",
                        inet_ntoa(handle_data_arg->cli_addr.sin_addr),
                        ntohs(handle_data_arg->cli_addr.sin_port));

      fclose(fsockin);
      fclose(fsockout);
      close(handle_data_arg->sock);
      free(arg);

#ifdef HAVE_PTHREAD
      pthread_exit(NULL);
#endif
      return NULL;
}

void usage()
{
      printf("Usage: rotctld [OPTION]... [COMMAND]...\n"
         "Daemon serving COMMANDs to a connected antenna rotator.\n\n");

      printf(
      "  -m, --model=ID             select rotator model number. See model list\n"
      "  -r, --rot-file=DEVICE      set device of the rotator to operate on\n"
      "  -s, --serial-speed=BAUD    set serial speed of the serial port\n"
      "  -t, --port=NUM             set TCP listening port, default %d\n"
      "  -T, --listen-addr=IPADDR   set listening IP address, default ANY\n"
      "  -C, --set-conf=PARM=VAL    set config parameters\n"
      "  -L, --show-conf            list all config parameters\n"
      "  -l, --list                 list all model numbers and exit\n"
      "  -e, --end-marker           use END marker in rotctld protocol\n"
      "  -v, --verbose              set verbose mode, cumulative\n"
      "  -h, --help                 display this help and exit\n"
      "  -V, --version              output version information and exit\n\n",
            portno);

      usage_rot(stdout);

      printf("\nReport bugs to <hamlib-developer@lists.sourceforge.net>.\n");
}


Generated by  Doxygen 1.6.0   Back to index