/RISCOS/Modules/riscosmodule.c
C | 603 lines | 521 code | 59 blank | 23 comment | 108 complexity | ea1a258a4c8401dd4aaacbb0edd8a85c MD5 | raw file
1/* RISCOS module implementation */ 2 3#include "oslib/osfscontrol.h" 4#include "oslib/osgbpb.h" 5#include "oslib/os.h" 6#include "oslib/osfile.h" 7#include "unixstuff.h" 8 9#include <sys/fcntl.h> 10 11#include "Python.h" 12#include "structseq.h" 13 14#include <errno.h> 15 16static os_error *e; 17 18/*static PyObject *RiscosError;*/ /* Exception riscos.error */ 19 20static PyObject *riscos_error(char *s) 21{ 22 PyErr_SetString(PyExc_OSError, s); 23 return NULL; 24} 25 26static PyObject *riscos_oserror(void) 27{ 28 return riscos_error(e->errmess); 29} 30 31 32/* RISCOS file commands */ 33 34static PyObject * 35riscos_remove(PyObject *self, PyObject *args) 36{ 37 char *path1; 38 if (!PyArg_ParseTuple(args, "s:remove", &path1)) return NULL; 39 if (remove(path1)) return PyErr_SetFromErrno(PyExc_OSError); 40 Py_INCREF(Py_None); 41 return Py_None; 42} 43 44static PyObject * 45riscos_rename(PyObject *self, PyObject *args) 46{ 47 char *path1, *path2; 48 if (!PyArg_ParseTuple(args, "ss:rename", &path1, &path2)) 49 return NULL; 50 if (rename(path1,path2)) return PyErr_SetFromErrno(PyExc_OSError); 51 Py_INCREF(Py_None); 52 return Py_None; 53} 54 55static PyObject * 56riscos_system(PyObject *self, PyObject *args) 57{ 58 char *command; 59 if (!PyArg_ParseTuple(args, "s:system", &command)) return NULL; 60 return PyInt_FromLong(system(command)); 61} 62 63static PyObject * 64riscos_chdir(PyObject *self, PyObject *args) 65{ 66 char *path; 67 if (!PyArg_ParseTuple(args, "s:chdir", &path)) return NULL; 68 e=xosfscontrol_dir(path); 69 if(e) return riscos_oserror(); 70 Py_INCREF(Py_None); 71 return Py_None; 72} 73 74static PyObject * 75canon(char *path) 76{ 77 int len; 78 PyObject *obj; 79 char *buf; 80 e=xosfscontrol_canonicalise_path(path,0,0,0,0,&len); 81 if(e) return riscos_oserror(); 82 obj=PyString_FromStringAndSize(NULL,-len); 83 if(obj==NULL) return NULL; 84 buf=PyString_AsString(obj); 85 e=xosfscontrol_canonicalise_path(path,buf,0,0,1-len,&len); 86 if(len!=1) return riscos_error("Error expanding path"); 87 if(!e) return obj; 88 Py_DECREF(obj); 89 return riscos_oserror(); 90} 91 92static PyObject * 93riscos_getcwd(PyObject *self, PyObject *unused) 94{ 95 return canon("@"); 96} 97 98static PyObject * 99riscos_expand(PyObject *self, PyObject *args) 100{ 101 char *path; 102 if (!PyArg_ParseTuple(args, "s:expand", &path)) return NULL; 103 return canon(path); 104} 105 106static PyObject * 107riscos_mkdir(PyObject *self, PyObject *args) 108{ 109 char *path; 110 int mode; 111 if (!PyArg_ParseTuple(args, "s|i:mkdir", &path, &mode)) return NULL; 112 e=xosfile_create_dir(path,0); 113 if(e) return riscos_oserror(); 114 Py_INCREF(Py_None); 115 return Py_None; 116} 117 118static PyObject * 119riscos_listdir(PyObject *self, PyObject *args) 120{ 121 char *path,buf[256]; 122 PyObject *d, *v; 123 int c=0,count; 124 if (!PyArg_ParseTuple(args, "s:listdir", &path)) return NULL; 125 d=PyList_New(0); 126 if(!d) return NULL; 127 for(;;) 128 { e=xosgbpb_dir_entries(path,(osgbpb_string_list*)buf, 129 1,c,256,0,&count,&c); 130 if(e) 131 { Py_DECREF(d);return riscos_oserror(); 132 } 133 if(count) 134 { v=PyString_FromString(buf); 135 if(!v) { Py_DECREF(d);return 0;} 136 if(PyList_Append(d,v)) {Py_DECREF(d);Py_DECREF(v);return 0;} 137 } 138 if(c==-1) break; 139 } 140 return d; 141} 142 143PyDoc_STRVAR(stat_result__doc__, 144"stat_result: Result from stat or lstat.\n\n\ 145This object may be accessed either as a tuple of\n\ 146 (mode,ino,dev,nlink,uid,gid,size,atime,mtime,ctime)\n\ 147or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\ 148\n\ 149RiscOS: The fields st_ftype, st_attrs, and st_obtype are also available.\n\ 150\n\ 151See os.stat for more information."); 152 153static PyStructSequence_Field stat_result_fields[] = { 154 { "st_mode", "protection bits" }, 155 { "st_ino", "inode" }, 156 { "st_dev", "device" }, 157 { "st_nlink", "number of hard links" }, 158 { "st_uid", "user ID of owner" }, 159 { "st_gid", "group ID of owner" }, 160 { "st_size", "total size, in bytes" }, 161 { "st_atime", "time of last access" }, 162 { "st_mtime", "time of last modification" }, 163 { "st_ctime", "time of last change" }, 164 { "st_ftype", "file type" }, 165 { "st_attrs", "attributes" }, 166 { "st_obtype", "object type" }, 167 { 0 } 168}; 169 170static PyStructSequence_Desc stat_result_desc = { 171 "riscos.stat_result", 172 stat_result__doc__, 173 stat_result_fields, 174 13 175}; 176 177static PyTypeObject StatResultType; 178 179static PyObject * 180riscos_stat(PyObject *self, PyObject *args) 181{ 182 PyObject *v; 183 char *path; 184 int ob,len; 185 bits t=0; 186 bits ld,ex,at,ft,mode; 187 if (!PyArg_ParseTuple(args, "s:stat", &path)) return NULL; 188 e=xosfile_read_stamped_no_path(path,&ob,&ld,&ex,&len,&at,&ft); 189 if(e) return riscos_oserror(); 190 switch (ob) 191 { case osfile_IS_FILE:mode=0100000;break; /* OCTAL */ 192 case osfile_IS_DIR:mode=040000;break; 193 case osfile_IS_IMAGE:mode=0140000;break; 194 default:return riscos_error("Not found"); 195 } 196 if(ft!=-1) t=unixtime(ld,ex); 197 mode|=(at&7)<<6; 198 mode|=((at&112)*9)>>4; 199 200 v = PyStructSequence_New(&StatResultType); 201 202 PyStructSequence_SET_ITEM(v, 0, 203 PyInt_FromLong((long) mode)); /*st_mode*/ 204 PyStructSequence_SET_ITEM(v, 1, PyInt_FromLong((long) 0)); /*st_ino*/ 205 PyStructSequence_SET_ITEM(v, 2, PyInt_FromLong((long) 0)); /*st_dev*/ 206 PyStructSequence_SET_ITEM(v, 3, PyInt_FromLong((long) 0)); /*st_nlink*/ 207 PyStructSequence_SET_ITEM(v, 4, PyInt_FromLong((long) 0)); /*st_uid*/ 208 PyStructSequence_SET_ITEM(v, 5, PyInt_FromLong((long) 0)); /*st_gid*/ 209 PyStructSequence_SET_ITEM(v, 6, 210 PyInt_FromLong((long) len)); /*st_size*/ 211 PyStructSequence_SET_ITEM(v, 7, PyInt_FromLong((long) t)); /*st_atime*/ 212 PyStructSequence_SET_ITEM(v, 8, PyInt_FromLong((long) t)); /*st_mtime*/ 213 PyStructSequence_SET_ITEM(v, 9, PyInt_FromLong((long) t)); /*st_ctime*/ 214 PyStructSequence_SET_ITEM(v, 10, 215 PyInt_FromLong((long) ft)); /*file type*/ 216 PyStructSequence_SET_ITEM(v, 11, 217 PyInt_FromLong((long) at)); /*attributes*/ 218 PyStructSequence_SET_ITEM(v, 12, 219 PyInt_FromLong((long) ob)); /*object type*/ 220 221 if (PyErr_Occurred()) { 222 Py_DECREF(v); 223 return NULL; 224 } 225 226 return v; 227} 228 229static PyObject * 230riscos_chmod(PyObject *self,PyObject *args) 231{ 232 char *path; 233 bits mode; 234 bits attr; 235 attr=(mode&0x700)>>8; 236 attr|=(mode&7)<<4; 237 if (!PyArg_ParseTuple(args, "si:chmod", &path,(int*)&mode)) return NULL; 238 e=xosfile_write_attr(path,attr); 239 if(e) return riscos_oserror(); 240 Py_INCREF(Py_None); 241 return Py_None; 242} 243 244 245static PyObject * 246riscos_utime(PyObject *self, PyObject *args) 247{ 248 char *path; 249 long atime, mtime; 250 PyObject* arg; 251 252 if (!PyArg_ParseTuple(args, "sO:utime", &path, &arg)) 253 return NULL; 254 255 if (arg == Py_None) { 256 /* optional time values not given */ 257 Py_BEGIN_ALLOW_THREADS 258 e=xosfile_stamp(path); 259 Py_END_ALLOW_THREADS 260 if(e) return riscos_oserror(); 261 } 262 else if (!PyArg_Parse(arg, "(ll)", &atime, &mtime)) { 263 PyErr_SetString(PyExc_TypeError, 264 "utime() arg 2 must be a tuple (atime, mtime)"); 265 return NULL; 266 } 267 else { 268 /* catalogue info*/ 269 fileswitch_object_type obj_type; 270 bits load_addr, exec_addr; 271 int size; 272 fileswitch_attr attr; 273 274 /* read old catalogue info */ 275 Py_BEGIN_ALLOW_THREADS 276 e=xosfile_read_no_path(path, &obj_type, &load_addr, &exec_addr, &size, &attr); 277 Py_END_ALLOW_THREADS 278 if(e) return riscos_oserror(); 279 280 /* check if load and exec address really contain filetype and date */ 281 if ( (load_addr & 0xFFF00000U) != 0xFFF00000U) 282 return riscos_error("can't set date for object with load and exec addresses"); 283 284 /* convert argument mtime to RISC OS load and exec address */ 285 if(acorntime(&exec_addr, &load_addr, (time_t) mtime)) 286 return riscos_oserror(); 287 288 /* write new load and exec address */ 289 Py_BEGIN_ALLOW_THREADS 290 e = xosfile_write(path, load_addr, exec_addr, attr); 291 Py_END_ALLOW_THREADS 292 if(e) return riscos_oserror(); 293 } 294 295 Py_INCREF(Py_None); 296 return Py_None; 297} 298 299static PyObject * 300riscos_settype(PyObject *self, PyObject *args) 301{ 302 char *path,*name; 303 int type; 304 if (!PyArg_ParseTuple(args, "si:settype", &path,&type)) 305 { 306 PyErr_Clear(); 307 if (!PyArg_ParseTuple(args, "ss:settype", &path,&name)) return NULL; 308 e=xosfscontrol_file_type_from_string(name,(bits*)&type); 309 if(e) return riscos_oserror(); 310 } 311 e=xosfile_set_type(path,type); 312 if(e) return riscos_oserror(); 313 Py_INCREF(Py_None); 314 return Py_None; 315} 316 317static PyObject * 318riscos_getenv(PyObject *self, PyObject *args) 319{ 320 char *name,*value; 321 if(!PyArg_ParseTuple(args,"s:getenv",&name)) return NULL; 322 value=getenv(name); 323 if(value) return PyString_FromString(value); 324 Py_INCREF(Py_None); 325 return Py_None; 326} 327 328static PyObject * 329riscos_putenv(PyObject *self, PyObject *args) 330{ 331 char *name,*value; 332 int len; 333 os_var_type type=os_VARTYPE_LITERAL_STRING; 334 if(!PyArg_ParseTuple(args,"ss|i:putenv",&name,&value,&type)) return NULL; 335 if(type!=os_VARTYPE_STRING&&type!=os_VARTYPE_MACRO&&type!=os_VARTYPE_EXPANDED 336 &&type!=os_VARTYPE_LITERAL_STRING) 337 return riscos_error("Bad putenv type"); 338 len=strlen(value); 339 if(type!=os_VARTYPE_LITERAL_STRING) len++; 340 /* Other types need null terminator! */ 341 e=xos_set_var_val(name,(byte*)value,len,0,type,0,0); 342 if(e) return riscos_oserror(); 343 Py_INCREF(Py_None); 344 return Py_None; 345} 346 347static PyObject * 348riscos_delenv(PyObject *self, PyObject *args) 349{ 350 char *name; 351 if(!PyArg_ParseTuple(args,"s:delenv",&name)) return NULL; 352 e=xos_set_var_val(name,NULL,-1,0,0,0,0); 353 if(e) return riscos_oserror(); 354 Py_INCREF(Py_None); 355 return Py_None; 356} 357 358static PyObject * 359riscos_getenvdict(PyObject *self, PyObject *args) 360{ 361 PyObject *dict; 362 char value[257]; 363 char *which="*"; 364 int size; 365 char *context=NULL; 366 if(!PyArg_ParseTuple(args,"|s:getenvdict",&which)) return NULL; 367 dict = PyDict_New(); 368 if (!dict) return NULL; 369 /* XXX This part ignores errors */ 370 while(!xos_read_var_val(which,value,sizeof(value)-1,(int)context, 371 os_VARTYPE_EXPANDED,&size,(int *)&context,0)) 372 { PyObject *v; 373 value[size]='\0'; 374 v = PyString_FromString(value); 375 if (v == NULL) continue; 376 PyDict_SetItemString(dict, context, v); 377 Py_DECREF(v); 378 } 379 return dict; 380} 381 382static PyMethodDef riscos_methods[] = { 383 384 {"unlink", riscos_remove, METH_VARARGS}, 385 {"remove", riscos_remove, METH_VARARGS}, 386 {"rename", riscos_rename, METH_VARARGS}, 387 {"system", riscos_system, METH_VARARGS}, 388 {"rmdir", riscos_remove, METH_VARARGS}, 389 {"chdir", riscos_chdir, METH_VARARGS}, 390 {"getcwd", riscos_getcwd, METH_NOARGS}, 391 {"expand", riscos_expand, METH_VARARGS}, 392 {"mkdir", riscos_mkdir, METH_VARARGS}, 393 {"listdir", riscos_listdir, METH_VARARGS}, 394 {"stat", riscos_stat, METH_VARARGS}, 395 {"lstat", riscos_stat, METH_VARARGS}, 396 {"chmod", riscos_chmod, METH_VARARGS}, 397 {"utime", riscos_utime, METH_VARARGS}, 398 {"settype", riscos_settype, METH_VARARGS}, 399 {"getenv", riscos_getenv, METH_VARARGS}, 400 {"putenv", riscos_putenv, METH_VARARGS}, 401 {"delenv", riscos_delenv, METH_VARARGS}, 402 {"getenvdict", riscos_getenvdict, METH_VARARGS}, 403 {NULL, NULL} /* Sentinel */ 404}; 405 406static int 407ins(PyObject *module, char *symbol, long value) 408{ 409 return PyModule_AddIntConstant(module, symbol, value); 410} 411 412 413static int 414all_ins(PyObject *d) 415{ 416#ifdef F_OK 417 if (ins(d, "F_OK", (long)F_OK)) return -1; 418#endif 419#ifdef R_OK 420 if (ins(d, "R_OK", (long)R_OK)) return -1; 421#endif 422#ifdef W_OK 423 if (ins(d, "W_OK", (long)W_OK)) return -1; 424#endif 425#ifdef X_OK 426 if (ins(d, "X_OK", (long)X_OK)) return -1; 427#endif 428#ifdef NGROUPS_MAX 429 if (ins(d, "NGROUPS_MAX", (long)NGROUPS_MAX)) return -1; 430#endif 431#ifdef TMP_MAX 432 if (ins(d, "TMP_MAX", (long)TMP_MAX)) return -1; 433#endif 434#ifdef WCONTINUED 435 if (ins(d, "WCONTINUED", (long)WCONTINUED)) return -1; 436#endif 437#ifdef WNOHANG 438 if (ins(d, "WNOHANG", (long)WNOHANG)) return -1; 439#endif 440#ifdef WUNTRACED 441 if (ins(d, "WUNTRACED", (long)WUNTRACED)) return -1; 442#endif 443#ifdef O_RDONLY 444 if (ins(d, "O_RDONLY", (long)O_RDONLY)) return -1; 445#endif 446#ifdef O_WRONLY 447 if (ins(d, "O_WRONLY", (long)O_WRONLY)) return -1; 448#endif 449#ifdef O_RDWR 450 if (ins(d, "O_RDWR", (long)O_RDWR)) return -1; 451#endif 452#ifdef O_NDELAY 453 if (ins(d, "O_NDELAY", (long)O_NDELAY)) return -1; 454#endif 455#ifdef O_NONBLOCK 456 if (ins(d, "O_NONBLOCK", (long)O_NONBLOCK)) return -1; 457#endif 458#ifdef O_APPEND 459 if (ins(d, "O_APPEND", (long)O_APPEND)) return -1; 460#endif 461#ifdef O_DSYNC 462 if (ins(d, "O_DSYNC", (long)O_DSYNC)) return -1; 463#endif 464#ifdef O_RSYNC 465 if (ins(d, "O_RSYNC", (long)O_RSYNC)) return -1; 466#endif 467#ifdef O_SYNC 468 if (ins(d, "O_SYNC", (long)O_SYNC)) return -1; 469#endif 470#ifdef O_NOCTTY 471 if (ins(d, "O_NOCTTY", (long)O_NOCTTY)) return -1; 472#endif 473#ifdef O_CREAT 474 if (ins(d, "O_CREAT", (long)O_CREAT)) return -1; 475#endif 476#ifdef O_EXCL 477 if (ins(d, "O_EXCL", (long)O_EXCL)) return -1; 478#endif 479#ifdef O_TRUNC 480 if (ins(d, "O_TRUNC", (long)O_TRUNC)) return -1; 481#endif 482#ifdef O_BINARY 483 if (ins(d, "O_BINARY", (long)O_BINARY)) return -1; 484#endif 485#ifdef O_TEXT 486 if (ins(d, "O_TEXT", (long)O_TEXT)) return -1; 487#endif 488#ifdef O_LARGEFILE 489 if (ins(d, "O_LARGEFILE", (long)O_LARGEFILE)) return -1; 490#endif 491 492/* MS Windows */ 493#ifdef O_NOINHERIT 494 /* Don't inherit in child processes. */ 495 if (ins(d, "O_NOINHERIT", (long)O_NOINHERIT)) return -1; 496#endif 497#ifdef _O_SHORT_LIVED 498 /* Optimize for short life (keep in memory). */ 499 /* MS forgot to define this one with a non-underscore form too. */ 500 if (ins(d, "O_SHORT_LIVED", (long)_O_SHORT_LIVED)) return -1; 501#endif 502#ifdef O_TEMPORARY 503 /* Automatically delete when last handle is closed. */ 504 if (ins(d, "O_TEMPORARY", (long)O_TEMPORARY)) return -1; 505#endif 506#ifdef O_RANDOM 507 /* Optimize for random access. */ 508 if (ins(d, "O_RANDOM", (long)O_RANDOM)) return -1; 509#endif 510#ifdef O_SEQUENTIAL 511 /* Optimize for sequential access. */ 512 if (ins(d, "O_SEQUENTIAL", (long)O_SEQUENTIAL)) return -1; 513#endif 514 515/* GNU extensions. */ 516#ifdef O_DIRECT 517 /* Direct disk access. */ 518 if (ins(d, "O_DIRECT", (long)O_DIRECT)) return -1; 519#endif 520#ifdef O_DIRECTORY 521 /* Must be a directory. */ 522 if (ins(d, "O_DIRECTORY", (long)O_DIRECTORY)) return -1; 523#endif 524#ifdef O_NOFOLLOW 525 /* Do not follow links. */ 526 if (ins(d, "O_NOFOLLOW", (long)O_NOFOLLOW)) return -1; 527#endif 528 529 /* These come from sysexits.h */ 530#ifdef EX_OK 531 if (ins(d, "EX_OK", (long)EX_OK)) return -1; 532#endif /* EX_OK */ 533#ifdef EX_USAGE 534 if (ins(d, "EX_USAGE", (long)EX_USAGE)) return -1; 535#endif /* EX_USAGE */ 536#ifdef EX_DATAERR 537 if (ins(d, "EX_DATAERR", (long)EX_DATAERR)) return -1; 538#endif /* EX_DATAERR */ 539#ifdef EX_NOINPUT 540 if (ins(d, "EX_NOINPUT", (long)EX_NOINPUT)) return -1; 541#endif /* EX_NOINPUT */ 542#ifdef EX_NOUSER 543 if (ins(d, "EX_NOUSER", (long)EX_NOUSER)) return -1; 544#endif /* EX_NOUSER */ 545#ifdef EX_NOHOST 546 if (ins(d, "EX_NOHOST", (long)EX_NOHOST)) return -1; 547#endif /* EX_NOHOST */ 548#ifdef EX_UNAVAILABLE 549 if (ins(d, "EX_UNAVAILABLE", (long)EX_UNAVAILABLE)) return -1; 550#endif /* EX_UNAVAILABLE */ 551#ifdef EX_SOFTWARE 552 if (ins(d, "EX_SOFTWARE", (long)EX_SOFTWARE)) return -1; 553#endif /* EX_SOFTWARE */ 554#ifdef EX_OSERR 555 if (ins(d, "EX_OSERR", (long)EX_OSERR)) return -1; 556#endif /* EX_OSERR */ 557#ifdef EX_OSFILE 558 if (ins(d, "EX_OSFILE", (long)EX_OSFILE)) return -1; 559#endif /* EX_OSFILE */ 560#ifdef EX_CANTCREAT 561 if (ins(d, "EX_CANTCREAT", (long)EX_CANTCREAT)) return -1; 562#endif /* EX_CANTCREAT */ 563#ifdef EX_IOERR 564 if (ins(d, "EX_IOERR", (long)EX_IOERR)) return -1; 565#endif /* EX_IOERR */ 566#ifdef EX_TEMPFAIL 567 if (ins(d, "EX_TEMPFAIL", (long)EX_TEMPFAIL)) return -1; 568#endif /* EX_TEMPFAIL */ 569#ifdef EX_PROTOCOL 570 if (ins(d, "EX_PROTOCOL", (long)EX_PROTOCOL)) return -1; 571#endif /* EX_PROTOCOL */ 572#ifdef EX_NOPERM 573 if (ins(d, "EX_NOPERM", (long)EX_NOPERM)) return -1; 574#endif /* EX_NOPERM */ 575#ifdef EX_CONFIG 576 if (ins(d, "EX_CONFIG", (long)EX_CONFIG)) return -1; 577#endif /* EX_CONFIG */ 578#ifdef EX_NOTFOUND 579 if (ins(d, "EX_NOTFOUND", (long)EX_NOTFOUND)) return -1; 580#endif /* EX_NOTFOUND */ 581 582 return 0; 583} 584 585 586void 587initriscos() 588{ 589 PyObject *m, *d, *stat_m; 590 591 m = Py_InitModule("riscos", riscos_methods); 592 593 if (all_ins(m)) 594 return; 595 596 d = PyModule_GetDict(m); 597 598 Py_INCREF(PyExc_OSError); 599 PyModule_AddObject(m, "error", PyExc_OSError); 600 601 PyStructSequence_InitType(&StatResultType, &stat_result_desc); 602 PyDict_SetItemString(d, "stat_result", (PyObject*) &StatResultType); 603}