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