PageRenderTime 34ms CodeModel.GetById 2ms app.highlight 27ms RepoModel.GetById 1ms app.codeStats 1ms

/RISCOS/Modules/riscosmodule.c

http://unladen-swallow.googlecode.com/
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}