Usermode/Libraries/liburi.so_src/main.c

   1 /*
   2  * Acess2 - URI Parser and opener
   3  * By John Hodge (thePowersGang)
   4  */
   5 #include <acess/sys.h>
   6 #include <stdlib.h>
   7 #include <stdio.h>
   8 #include <string.h>
   9 #include <ctype.h>
  10 #include <uri.h>
  11
  12 // === STRUCTURES ===
  13 struct sURIFile
  14 {
  15          int    Handle;
  16          int    Mode;
  17         tURIHandler     *Handler;
  18          int    CurBlockOffset;
  19         char    Buffer[];
  20 };
  21
  22 // === PROTOTYPES ===
  23  int    SoMain(void);
  24 tURI    *URI_Parse(const char *String);
  25 tURIFile        *URI_MakeHandle(int Mode, int Handle, tURIHandler *Handler);
  26 tURIFile        *URI_Open(int Mode, tURI *URI);
  27 size_t  URI_Read(tURIFile *File, size_t Bytes, void *Buffer);
  28 size_t  URI_Write(tURIFile *File, size_t Bytes, void *Buffer);
  29 void    URI_Close(tURIFile *File);
  30 // --- file:/// handler
  31  int    URI_file_Open(char *Host, int Port, char *Path, int Mode);
  32 size_t  URI_file_Read(int Handle, size_t Bytes, void *Buffer);
  33 size_t  URI_file_Write(int Handle, size_t Bytes, void *Buffer);
  34 void    URI_file_Close(int Handle);
  35 size_t  URI_file_GetSize(int Handle);
  36
  37 // === CONSTANTS ===
  38 // Builtin URI protocol handlers
  39 tURIHandler     caBuiltinHandlers[] = {
  40         {"file", 0, URI_file_Open, URI_file_Close, URI_file_Read, URI_file_Write, URI_file_GetSize}
  41 };
  42 #define NUM_BUILTIN_HANDLERS    (sizeof(caBuiltinHandlers)/sizeof(caBuiltinHandlers[0]))
  43
  44 // === CODE ===
  45 int SoMain(void)
  46 {
  47         return 0;
  48 }
  49
  50 tURI *URI_Parse(const char *String)
  51 {
  52         const char      *tmp = String;
  53         tURI    *ret;
  54          int    protolen;
  55
  56         if(!String)     return NULL;
  57
  58         protolen = 0;
  59         while( isalpha(*tmp) || isdigit(*tmp) ) tmp++, protolen++;
  60
  61         // true URI
  62         if(tmp[0] == ':' && tmp[1] == '/' && tmp[2] == '/')
  63         {
  64                  int    hostlen, portlen, pathlen;
  65                 tmp += 3;       // Eat '://'
  66                 ret = malloc(sizeof(tURI) + strlen(String) - 2);
  67
  68                 ret->Proto = (char*)ret + sizeof(tURI);
  69                 memcpy(ret->Proto, String, protolen);
  70                 ret->Proto[protolen] = '\0';
  71
  72                 ret->Host = ret->Proto + protolen + 1;
  73                 hostlen = 0;
  74
  75                 // IPv6
  76                 if( *tmp == '[' )
  77                 {
  78                         tmp ++;
  79                         while( *tmp && *tmp != ']' ) {
  80                                 ret->Host[hostlen] = *tmp;
  81                                 tmp ++;
  82                                 hostlen ++;
  83                         }
  84                         tmp ++;
  85                         ret->Host[hostlen] = '\0';
  86                 }
  87                 // IPv4/DNS
  88                 else
  89                 {
  90                         while( *tmp && *tmp != '/' && *tmp != ':' )
  91                         {
  92                                 ret->Host[hostlen] = *tmp;
  93                                 tmp ++;
  94                                 hostlen ++;
  95                         }
  96                         ret->Host[hostlen] = '\0';
  97                 }
  98
  99                 // Port
 100                 if( *tmp == ':' ) {
 101                         ret->PortStr = ret->Host + hostlen + 1;
 102                         tmp ++;
 103                         portlen = 0;
 104                         while(isalpha(*tmp) || isdigit(*tmp))
 105                         {
 106                                 ret->PortStr[portlen] = *tmp;
 107                                 portlen ++;
 108                                 tmp ++;
 109                         }
 110                         ret->PortStr[portlen] = '\0';
 111
 112                         ret->PortNum = atoi(ret->PortStr);
 113                         if(!ret->PortNum && !(ret->PortStr[0] == '0' && portlen == 1) )
 114                         {
 115                                 // error!
 116                                 ret->Path = NULL;
 117                                 return ret;
 118                         }
 119                 }
 120                 else {
 121                         ret->PortStr = NULL;
 122                         ret->PortNum = -1;
 123                 }
 124
 125                 if(*tmp == '\0')
 126                 {
 127                         ret->Path = NULL;
 128                         return ret;
 129                 }
 130
 131                 // TODO: What to do on a parse error
 132                 if(*tmp != '/') {
 133                         ret->Path = NULL;
 134                         return ret;
 135                 }
 136
 137                 if(ret->PortStr)
 138                         ret->Path = ret->PortStr + portlen + 1;
 139                 else
 140                         ret->Path = ret->Host + hostlen + 1;
 141
 142                 tmp ++;
 143                 pathlen = 0;
 144                 while(*tmp)
 145                 {
 146                         ret->Path[pathlen] = *tmp;
 147                         tmp ++;
 148                         pathlen ++;
 149                 }
 150
 151                 return ret;
 152         }
 153         else
 154         {
 155 //              char    *cwd;
 156 //               int    retlen;
 157
 158                 // Path
 159                 // TODO: What to do?
 160                 // Probably return file:///<path>
 161                 // but should I get the CWD and use append that?
 162                 ret = malloc( sizeof(tURI) + strlen(String) + 1 );
 163                 ret->Path = (char*)ret + sizeof(tURI);
 164                 strcpy(ret->Path, String);
 165                 ret->Proto = "file";
 166                 ret->Host = NULL;
 167                 ret->PortNum = 0;
 168                 ret->PortStr = NULL;
 169                 return ret;
 170         }
 171 }
 172
 173 tURIFile *URI_MakeHandle(int Mode, int Handle, tURIHandler *Handler)
 174 {
 175         tURIFile        *ret;
 176
 177         ret = malloc(sizeof(tURIFile)+Handler->BlockSize);
 178         if(!ret)        return NULL;
 179
 180         ret->Handle = Handle;
 181         ret->Mode = Mode;
 182         ret->Handler = Handler;
 183         ret->CurBlockOffset = 0;
 184
 185         return ret;
 186 }
 187
 188 tURIFile *URI_Open(int Mode, tURI *URI)
 189 {
 190         tURIHandler     *handler;
 191         tURIFile        *ret;
 192          int    handle;
 193          int    i;
 194
 195         if(!URI)
 196                 return NULL;
 197
 198         for( i = 0; i < NUM_BUILTIN_HANDLERS; i ++ )
 199         {
 200                 if(strcmp(URI->Proto, caBuiltinHandlers[i].Name) == 0)
 201                         break;
 202         }
 203
 204         if( i == NUM_BUILTIN_HANDLERS )
 205         {
 206                 // TODO: Dynamics
 207                 printf("URI_Open: Warning - Unknown URI handler\n");
 208                 return NULL;
 209         }
 210         else
 211                 handler = &caBuiltinHandlers[i];
 212
 213         printf("URI_Open: handler->Open = %p\n", handler->Open);
 214
 215         handle = handler->Open(URI->Host, URI->PortNum, URI->Path, Mode);
 216         printf("URI_Open: handle = %i\n", handle);
 217         if(handle == -1)        return NULL;
 218
 219         printf("URI_MakeHandle(Mode=%i, handle=%i, handler=%p)\n",
 220                 Mode, handle, handler);
 221         ret = URI_MakeHandle(Mode, handle, handler);
 222         if(!ret) {
 223                 handler->Close( handle );
 224                 return NULL;
 225         }
 226         return ret;
 227 }
 228
 229 int URI_GetSize(tURIFile *File, size_t *Size)
 230 {
 231         if( !File || !Size )    return -1;
 232
 233         if( File->Handler->GetSize )
 234         {
 235                 *Size = File->Handler->GetSize(File->Handle);
 236                 return 0;       // Success
 237         }
 238
 239         return 1;       // Size not avaliable
 240 }
 241
 242 /**
 243  * \brief Read from a URI file
 244  */
 245 size_t URI_Read(tURIFile *File, size_t Bytes, void *Buffer)
 246 {
 247         size_t  rem = Bytes;
 248         void    *buf = Buffer;
 249         size_t  tmp;
 250
 251         printf("URI_Read(File=%p, Bytes=%u, Buffer=%p)\n",
 252                 File, Bytes, Buffer);
 253
 254         if(!File || !Buffer)    return -1;
 255         if(Bytes == 0)  return 0;
 256
 257         if( !(File->Mode & URI_MODE_READ) )     return -1;
 258
 259         // Read from cache if avaliable
 260         if(File->Handler->BlockSize && File->CurBlockOffset)
 261         {
 262                  int    avail;
 263
 264                 avail = File->Handler->BlockSize - File->CurBlockOffset;
 265
 266                 if(avail >= Bytes) {
 267                         memcpy(Buffer, File->Buffer, Bytes);
 268                         File->CurBlockOffset += Bytes;
 269                         File->CurBlockOffset %= File->Handler->BlockSize;
 270                         return Bytes;
 271                 }
 272
 273                 rem -= avail;
 274                 memcpy(Buffer, File->Buffer, avail);
 275                 File->CurBlockOffset = 0;
 276                 buf += avail;
 277         }
 278
 279
 280         if( File->Handler->BlockSize )
 281         {
 282                 // Read whole blocks
 283                 while( rem >= File->Handler->BlockSize )
 284                 {
 285                         tmp = File->Handler->Read( File->Handle, File->Handler->BlockSize, buf );
 286                         if(tmp < File->Handler->BlockSize)
 287                                 return Bytes - rem - tmp;
 288                         buf += File->Handler->BlockSize;
 289                 }
 290
 291                 // Read the trailing part
 292                 if(rem)
 293                 {
 294                         File->Handler->Read( File->Handle, File->Handler->BlockSize, File->Buffer );
 295                         memcpy( buf, File->Buffer, rem );
 296                         File->CurBlockOffset += rem;
 297                 }
 298                 return Bytes;
 299         }
 300
 301         return File->Handler->Read( File->Handle, Bytes, Buffer );
 302 }
 303
 304 /**
 305  * \brief Write to a URI file
 306  */
 307
 308
 309 // ====
 310 // Builtin Handlers
 311 // ====
 312 int URI_file_Open(char *Host, int Port, char *Path, int Mode)
 313 {
 314          int    smode = 0;
 315         if(Mode & URI_MODE_READ)        smode |= OPENFLAG_READ;
 316         if(Mode & URI_MODE_WRITE)       smode |= OPENFLAG_WRITE;
 317
 318         printf("URI_file_Open: open('%s', 0x%x)\n", Path, smode);
 319         {
 320                  int    ret;
 321                 ret = open(Path, smode);
 322                 return ret;
 323         }
 324 }
 325 size_t URI_file_Read(int Handle, size_t Bytes, void *Buffer)
 326 {
 327         printf("URI_file_Read: (Handle=%i, Buffer=%p, Bytes=%i)\n",
 328                 Handle, Buffer, Bytes);
 329         return read(Handle, Buffer, Bytes);
 330 }
 331 size_t URI_file_Write(int Handle, size_t Bytes, void *Buffer)
 332 {
 333         return write(Handle, Buffer, Bytes);
 334 }
 335 void URI_file_Close(int Handle)
 336 {
 337         close(Handle);
 338 }
 339 size_t URI_file_GetSize(int Handle)
 340 {
 341         uint64_t curpos = tell(Handle);
 342         size_t ret;
 343         seek(Handle, 0, SEEK_END);
 344         ret = tell(Handle);
 345         seek(Handle, curpos, SEEK_SET);
 346         return ret;
 347 }