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