workbench/c/WaitForPort.c

   1 /*
   2     Copyright (C) 2004, The AROS Development Team. All rights reserved.
   3     $Id$
   4
   5     Desc: Waits up to 10 seconds for a user specified Port to become available
   6     Lang: English
   7 */
   8
   9 #include <dos/rdargs.h>
  10 #include <proto/exec.h>
  11 #include <proto/dos.h>
  12 #include <devices/timer.h>
  13 #include <stdio.h>
  14 #include <proto/alib.h>
  15
  16 #include <aros/debug.h>
  17
  18 const char version[] = "$VER: WaitForPort 0.1 (26.12.2005)";
  19
  20 const char WaitForPort_ArgTemplate[] = "P=PORT/A";
  21 IPTR       WaitForPort_Arguments[2];
  22
  23 struct RDArgs *WFP_rda = NULL;
  24
  25 struct Device              *TimerBase = NULL;
  26 static struct MsgPort      *timerport = NULL;         /* Timer message reply port */
  27 static struct timerequest  *timerIORequest = NULL;    /* template IORequest */
  28 ULONG wait_time;
  29 ULONG wait_limit;
  30 int
  31 main(int argc, char *argv[])
  32 {
  33    struct MsgPort *AROSTCP_Port = NULL;
  34
  35    wait_time = 0;
  36
  37    if ((WFP_rda = ReadArgs(WaitForPort_ArgTemplate, WaitForPort_Arguments, NULL)))
  38    {
  39
  40       if (WaitForPort_Arguments[0])
  41       {
  42 D(bug("[WaitForPort] Waiting for '%s' port\n",WaitForPort_Arguments[0]));
  43       }
  44
  45       timerport = CreateMsgPort();
  46       if (timerport != NULL)
  47       {
  48          /* allocate and initialize the template message structure */
  49          timerIORequest = (struct timerequest *) CreateIORequest(timerport, sizeof(struct timerequest));
  50
  51          if (timerIORequest != NULL)
  52          {
  53             if (!(OpenDevice(TIMERNAME, UNIT_VBLANK,
  54                                               (struct IORequest *)timerIORequest, 0)))
  55             {
  56                 /* Make sure that we got at least V36 timer, since we use some
  57                  * functions defined only in V36 and later. */
  58
  59                 if ((timerIORequest->tr_node.io_Device)->dd_Library.lib_Version >= 36)
  60                 {
  61                    /* initialize TimerBase from timerIORequest */
  62                   TimerBase = timerIORequest->tr_node.io_Device;
  63
  64                    /* Initialize some fields of the IO request to common values */
  65                    timerIORequest->tr_node.io_Command = TR_ADDREQUEST;
  66
  67                    /* NT_UNKNOWN means unused, too (see note on exec/nodes.h) */
  68                    timerIORequest->tr_node.io_Message.mn_Node.ln_Type = NT_UNKNOWN;
  69
  70                   timerIORequest->tr_time.tv_micro = 1000000;
  71                   wait_limit = timerIORequest->tr_time.tv_micro * 10; /* Default to a 10 second wait */
  72
  73                   BeginIO((struct IORequest *)timerIORequest);
  74
  75 /* MAIN LOOP */
  76                   while(1)
  77                   {
  78 D(bug("[WaitForPort] In Wait Loop ..\n"));
  79                      ULONG mask = SIGBREAKF_CTRL_C | SIGBREAKF_CTRL_D | (1 << timerport->mp_SigBit);
  80                                         mask = Wait(mask);
  81                      if (mask & SIGBREAKF_CTRL_C) break;
  82                                         if (mask & SIGBREAKF_CTRL_D) break;
  83                      if (mask & (1 << timerport->mp_SigBit))
  84                      {
  85 D(bug("[WaitForPort] Recieved timer signal? ..\n"));
  86                         timerIORequest = (struct timerequest *)GetMsg(timerport);
  87                         if (timerIORequest)
  88                         {
  89                            AROSTCP_Port = FindPort((char *)WaitForPort_Arguments[0]);
  90                            wait_time += 1000000;
  91                            if (!(AROSTCP_Port))
  92                            {
  93                               if (wait_time > wait_limit)
  94                               {
  95 D(bug("[WaitForPort] Timeout Reached\n"));
  96                                   break;
  97                               }
  98 D(bug("[WaitForPort] Port not found .. secs=%d\n",wait_time/1000000));
  99                            }
 100                            else
 101                            {
 102 D(bug("[WaitForPort] Port found ... escaping from wait loop\n"));
 103                               break;
 104                            }
 105                            timerIORequest->tr_node.io_Command = TR_ADDREQUEST;
 106                            timerIORequest->tr_time.tv_micro = 1000000;
 107                            BeginIO((struct IORequest *)timerIORequest);
 108                         }
 109                      }
 110                   }
 111                }
 112             }
 113          }
 114
 115 /* CLEANUP */
 116
 117          if (timerIORequest)
 118          {
 119             TimerBase = NULL;
 120
 121             if (timerIORequest->tr_node.io_Device != NULL) CloseDevice((struct IORequest *)timerIORequest);
 122
 123             DeleteIORequest((struct IORequest *)timerIORequest);
 124             timerIORequest = NULL;
 125          }
 126
 127          if (timerport)
 128          {
 129             DeleteMsgPort(timerport);
 130             timerport = NULL;
 131          }
 132       }
 133
 134       FreeArgs(WFP_rda);
 135    }
 136    else
 137    {
 138       printf("WaitForPort: Bad Arguments .. Use 'WaitForPort ?' for correct useage\n");
 139    }
 140
 141    if (AROSTCP_Port) return RETURN_OK;
 142
 143    return RETURN_WARN;
 144 }