workbench/c/Eject.c

   1 /*
   2     Copyright © 2009, The AROS Development Team. All rights reserved.
   3     $Id$
   4
   5     Desc: Eject/Load CLI command.
   6     Lang: English
   7 */
   8
   9 /******************************************************************************
  10
  11
  12     NAME
  13         Eject <device>
  14
  15     SYNOPSIS
  16         DEVICE/A
  17
  18     LOCATION
  19        C:
  20
  21     FUNCTION
  22         Ejects media from a device. This feature is not supported by all
  23         device types.
  24
  25     INPUTS
  26         DEVICE  --  Name of device to eject media from.
  27
  28     RESULT
  29
  30     NOTES
  31
  32     EXAMPLE
  33
  34     BUGS
  35
  36     SEE ALSO
  37
  38     INTERNALS
  39
  40 ******************************************************************************/
  41
  42 /******************************************************************************
  43
  44
  45     NAME
  46         Load <device>
  47
  48     SYNOPSIS
  49         DEVICE/A
  50
  51     LOCATION
  52        C:
  53
  54     FUNCTION
  55         Loads media into a device. This feature is not supported by all
  56         device types.
  57
  58     INPUTS
  59         DEVICE  --  Name of device to load media into.
  60
  61     RESULT
  62
  63     NOTES
  64
  65     EXAMPLE
  66
  67     BUGS
  68
  69     SEE ALSO
  70
  71     INTERNALS
  72
  73 ******************************************************************************/
  74
  75 int __nocommandline;
  76
  77 #include <exec/exec.h>
  78 #include <dos/dos.h>
  79 #include <dos/filehandler.h>
  80 #include <devices/trackdisk.h>
  81 #include <proto/exec.h>
  82 #include <proto/dos.h>
  83 #include <proto/utility.h>
  84
  85 #define NAME_BUFFER_SIZE 32
  86
  87 const TEXT version_string[] = "$VER: Eject 41.1 (26.9.2009)";
  88 const TEXT template[] = "DEVICE/A";
  89 const TEXT load_name[] = "Load";
  90
  91
  92 struct Args
  93 {
  94     TEXT *device;
  95 };
  96
  97
  98 int main(void)
  99 {
 100     struct RDArgs *read_args = NULL;
 101     LONG error = 0, result = RETURN_OK;
 102     struct Args args = {NULL};
 103     struct MsgPort *port = NULL;
 104     struct IOStdReq *request = NULL;
 105     ULONG signals, received;
 106     struct DosList *dos_list;
 107     TEXT node_name[NAME_BUFFER_SIZE];
 108     UWORD i;
 109     struct FileSysStartupMsg *fssm;
 110     struct DeviceNode *dev_node = NULL;
 111     BOOL load = FALSE;
 112
 113     port = CreateMsgPort();
 114     request =
 115         (struct IOStdReq *)CreateIORequest(port, sizeof(struct IOStdReq));
 116     if (port == NULL || request == NULL)
 117         error = IoErr();
 118
 119     if (error == 0)
 120     {
 121         /* Parse arguments */
 122
 123         read_args = ReadArgs(template, (APTR)&args, NULL);
 124         if (read_args == NULL)
 125         {
 126             error = IoErr();
 127             result = RETURN_ERROR;
 128         }
 129
 130         /* Find out if we should eject or load media based on name of command */
 131
 132         load = Strnicmp(FilePart(AROS_BSTR_ADDR(Cli()->cli_CommandName)),
 133             load_name, sizeof(load_name) - 1) == 0;
 134     }
 135
 136     if (error == 0)
 137     {
 138         /* Make copy of DOS device name without a colon */
 139
 140         for (i = 0; args.device[i] != '\0' && args.device[i] != ':'
 141             && i < NAME_BUFFER_SIZE; i++)
 142             node_name[i] = args.device[i];
 143         if (args.device[i] == ':' && i < NAME_BUFFER_SIZE)
 144             node_name[i] = '\0';
 145         else
 146             error = ERROR_INVALID_COMPONENT_NAME;
 147     }
 148
 149     /* Determine trackdisk-like device underlying DOS device */
 150
 151     if (error == 0)
 152     {
 153         dos_list = LockDosList(LDF_READ);
 154         if (dos_list != NULL)
 155         {
 156             dev_node = (struct DeviceNode *)FindDosEntry(dos_list, node_name,
 157                 LDF_DEVICES);
 158             if (dev_node == NULL)
 159                 error = IoErr();
 160             UnLockDosList(LDF_READ);
 161         }
 162         else
 163             error = IoErr();
 164     }
 165
 166     if (error == 0)
 167     {
 168         if (IsFileSystem(args.device))
 169             fssm = (struct FileSysStartupMsg *)BADDR(dev_node->dn_Startup);
 170         else
 171             error = ERROR_OBJECT_WRONG_TYPE;
 172     }
 173
 174     /* Open trackdisk-like device */
 175
 176     if (error == 0)
 177     {
 178         if (OpenDevice(AROS_BSTR_ADDR(fssm->fssm_Device), fssm->fssm_Unit,
 179             (APTR)request, 0) != 0)
 180             error = ERROR_UNKNOWN;
 181     }
 182
 183     if (error == 0)
 184     {
 185         /* Send command to eject or load media */
 186
 187         request->io_Command = TD_EJECT;
 188         request->io_Length = load ? 0 : 1;
 189         SendIO((APTR)request);
 190         signals = (1 << port->mp_SigBit) | SIGBREAKF_CTRL_C;
 191         received = 0;
 192
 193         /* Wait for command completion or user break */
 194
 195         while ((received & SIGBREAKF_CTRL_C) == 0 && GetMsg(port) == NULL)
 196             received = Wait(signals);
 197         if ((received & SIGBREAKF_CTRL_C) != 0)
 198         {
 199             AbortIO((APTR)request);
 200             WaitIO((APTR)request);
 201             GetMsg(port);
 202             error = ERROR_BREAK;
 203             result = RETURN_WARN;
 204         }
 205         else
 206         {
 207             if (request->io_Error != 0)
 208                 error = ERROR_UNKNOWN;
 209         }
 210     }
 211
 212     /* Deallocate resources */
 213
 214     if (request != NULL)
 215         CloseDevice((APTR)request);
 216     DeleteIORequest(request);
 217     DeleteMsgPort(port);
 218
 219     FreeArgs(read_args);
 220
 221     /* Print any error message and return */
 222
 223     SetIoErr(error);
 224     PrintFault(error, NULL);
 225
 226     if (error != 0 && result == RETURN_OK)
 227         result = RETURN_FAIL;
 228
 229     return result;
 230 }
 231