go/door_daemon_zmq/main.go

   1 // (c) Bernhard Tittelbach, 2013
   2
   3 package main
   4
   5 import (
   6     "fmt"
   7     "os"
   8     "flag"
   9     //~ "log"
  10 )
  11
  12 //~ func StringArrayToByteArray(ss []string) [][]byte {
  13     //~ bb := make([][]byte, len(ss))
  14     //~ for index, s := range(ss) {
  15         //~ bb[index] = []byte(s)
  16     //~ }
  17     //~ return bb
  18 //~ }
  19
  20 // ---------- Main Code -------------
  21
  22 var (
  23     cmd_port_ string
  24     pub_port_ string
  25 )
  26
  27 func usage() {
  28     fmt.Fprintf(os.Stderr, "Usage: door_daemon_0mq <door tty device>\n")
  29     flag.PrintDefaults()
  30 }
  31
  32 func init() {
  33     flag.StringVar(&cmd_port_, "cmdport", "tcp://127.0.01:3232", "zmq command socket path")
  34     flag.StringVar(&pub_port_, "pubport", "pgm://233.252.1.42:4242", "zmq public/listen socket path")
  35     flag.Usage = usage
  36     flag.Parse()
  37 }
  38
  39 func main() {
  40     args := flag.Args()
  41     if len(args) < 1 {
  42         fmt.Fprintf(os.Stderr, "Input file is missing!\n");
  43         usage()
  44         os.Exit(1);
  45     }
  46
  47     zmqctx, cmd_chans, pub_chans := ZmqsInit(cmd_port_, pub_port_)
  48     defer cmd_chans.Close()
  49     defer pub_chans.Close()
  50     defer zmqctx.Close()
  51
  52     serial_wr, serial_rd, err := OpenAndHandleSerial(args[0])
  53     defer close(serial_wr)
  54     if err != nil {
  55         panic(err)
  56     }
  57
  58     //~ serial_wr <- "f"
  59     //~ firmware_version := <- serial_rd
  60     //~ log.Print("Firmware version:", firmware_version)
  61     var next_incoming_serial_is_client_reply bool
  62     for {
  63         select {
  64             case incoming_ser_line, is_notclosed := <- serial_rd:
  65                 if is_notclosed {
  66                     if next_incoming_serial_is_client_reply {
  67                         next_incoming_serial_is_client_reply = false
  68                         cmd_chans.Out() <- incoming_ser_line
  69                     }
  70                     pub_chans.Out() <- incoming_ser_line
  71                 } else {
  72                     os.Exit(1)
  73                 }
  74             case incoming_request, ic_notclosed := <- cmd_chans.In():
  75                 if ! ic_notclosed {os.Exit(2)}
  76                 //~ log.Print(incoming_request)
  77                  if err := HandleCommand(incoming_request, serial_wr, serial_rd); err != nil {
  78                     //~ log.Print(err)
  79                     out_msg := [][]byte{[]byte("ERROR"), []byte(err.Error())}
  80                     cmd_chans.Out() <- out_msg
  81                     //~ log.Print("sent error")
  82                  } else {
  83                     //~ log.Print(reply)
  84                     pub_chans.Out() <- incoming_request
  85                     next_incoming_serial_is_client_reply = true
  86                     //~ log.Print("sent reply")
  87                  }
  88         }
  89     }
  90 }