tools/toollib/stdiocb.c

   1 /*
   2     Copyright © 1995-2001, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <string.h>
   7 #include <errno.h>
   8 #include <toollib/error.h>
   9 #include <toollib/stdiocb.h>
  10
  11 static int
  12 StdioGetCB (StdioStream * ss, int dummy, CBD data)
  13 {
  14     if (ss->in)
  15     {
  16         int c = getc (ss->in);
  17
  18         if (c == '\n')
  19             Str_NextLine (ss);
  20
  21         return c;
  22     }
  23
  24     errno = EINVAL;
  25     return -1;
  26 }
  27
  28 static int
  29 StdioUngetCB (StdioStream * ss, int c, CBD data)
  30 {
  31     if (ss->in)
  32         return ungetc (c, ss->in);
  33
  34     errno = EINVAL;
  35     return -1;
  36 }
  37
  38 static int
  39 StdioPutCB (StdioStream * ss, int c, CBD data)
  40 {
  41     if (ss->out)
  42         return putc (c, ss->out);
  43
  44     errno = EINVAL;
  45     return -1;
  46 }
  47
  48 static int
  49 StdioPutsCB (StdioStream * ss, const char * str, CBD data)
  50 {
  51     if (ss->out)
  52         return fputs (str, ss->out);
  53
  54     errno = EINVAL;
  55     return -1;
  56 }
  57
  58 StdioStream *
  59 StdStr_New (const char * path, const char * mode)
  60 {
  61     StdioStream * ss = new (StdioStream);
  62     FILE * fh = NULL;
  63
  64     if (strcmp (path, "-"))
  65     {
  66         fh = fopen (path, mode);
  67
  68         if (!fh)
  69         {
  70             PushStdError ("Can't open \"%s\" with mode \"%s\"\n", path, mode);
  71             return NULL;
  72         }
  73     }
  74
  75     Str_Init (&ss->stream, path);
  76
  77     ss->stream.get   = (CB) StdioGetCB;
  78     ss->stream.unget = (CB) StdioUngetCB;
  79     ss->stream.put   = (CB) StdioPutCB;
  80     ss->stream.puts  = (CB) StdioPutsCB;
  81
  82     if (strchr (mode, 'r'))
  83     {
  84         if (strcmp (path, "-"))
  85         {
  86             ss->in = fh;
  87             ss->closein = 1;
  88         }
  89         else
  90         {
  91             ss->in = stdin;
  92             ss->closein = 0;
  93         }
  94     }
  95     else
  96     {
  97         ss->in = NULL;
  98         ss->closein = 0;
  99     }
 100
 101     if (strchr (mode, 'w') || strchr (mode, 'a'))
 102     {
 103         if (strcmp (path, "-"))
 104         {
 105             ss->out = fh;
 106             ss->closeout = 1;
 107         }
 108         else
 109         {
 110             ss->out = stdout;
 111             ss->closeout = 0;
 112         }
 113     }
 114     else
 115     {
 116         ss->out = NULL;
 117         ss->closeout = 0;
 118     }
 119
 120     if (ss->in && ss->out)
 121     {
 122         ss->closeout = 0;
 123     }
 124
 125     return ss;
 126 }
 127
 128 void
 129 StdStr_Delete (StdioStream * ss)
 130 {
 131     if (ss->closein)
 132         fclose (ss->in);
 133
 134     if (ss->closeout)
 135         fclose (ss->out);
 136
 137     Str_Delete (&ss->stream);
 138
 139     xfree (ss);
 140 }
 141