/RISCOS/Modules/swimodule.c

http://unladen-swallow.googlecode.com/ · C · 575 lines · 500 code · 55 blank · 20 comment · 107 complexity · 41c0e08b1a7f80dbeb3fd9b9582c6f99 MD5 · raw file

  1. /* swi
  2. RISC OS swi functions
  3. 1.00 Chris Stretch
  4. 1.01 12 May 1999 Laurence Tratt
  5. * Changed swi.error to be a class based exception rather than string based
  6. * Added swi.ArgError which is generated for errors when the user passes invalid arguments to
  7. functions etc
  8. * Added "errnum" attribute to swi.error, so one can now check to see what the error number was
  9. 1.02 03 March 2002 Dietmar Schwertberger
  10. * Added string, integer, integers, tuple and tuples
  11. */
  12. #include "oslib/os.h"
  13. #include <kernel.h>
  14. #include "Python.h"
  15. #define PyBlock_Check(op) ((op)->ob_type == &PyBlockType)
  16. static PyObject *SwiError; /* Exception swi.error */
  17. static PyObject *ArgError; /* Exception swi.ArgError */
  18. static os_error *e;
  19. static PyObject *swi_oserror(void)
  20. { PyErr_SetString(SwiError,e->errmess);
  21. PyObject_SetAttrString(PyErr_Occurred(), "errnum", PyInt_FromLong(e->errnum));
  22. return 0;
  23. }
  24. static PyObject *swi_error(char *s)
  25. { PyErr_SetString(ArgError,s);
  26. return 0;
  27. }
  28. typedef struct
  29. { PyObject_HEAD
  30. void *block;
  31. int length; /*length in bytes*/
  32. int heap;
  33. } PyBlockObject;
  34. static PyTypeObject PyBlockType;
  35. /* block commands */
  36. static PyObject *PyBlock_New(PyObject *self,PyObject *args)
  37. { int size;
  38. PyBlockObject *b;
  39. PyObject *init=0;
  40. if(!PyArg_ParseTuple(args,"i|O",&size,&init)) return NULL;
  41. if(size<1) size=1;
  42. b=PyObject_NEW(PyBlockObject,&PyBlockType);
  43. if(!b) return NULL;
  44. b->block=malloc(4*size);
  45. if(!b->block)
  46. { Py_DECREF(b);
  47. return PyErr_NoMemory();
  48. }
  49. b->length=4*size;
  50. b->heap=1;
  51. if(init)
  52. { if(PyString_Check(init))
  53. { int n=PyString_Size(init);
  54. if (n>4*size) n=4*size;
  55. memcpy(b->block,PyString_AsString(init),n);
  56. memset((char*)b->block+n,0,4*size-n);
  57. }
  58. else
  59. { int n,k;
  60. long *p=(long*)b->block;
  61. if(!PyList_Check(init)) goto fail;
  62. n=PyList_Size(init);
  63. if (n>size) n=size;
  64. for(k=0;k<n;k++)
  65. { PyObject *q=PyList_GetItem(init,k);
  66. if(!PyInt_Check(q)) goto fail;
  67. p[k]=PyInt_AsLong(q);
  68. }
  69. for(;k<size;k++) p[k]=0;
  70. }
  71. }
  72. return (PyObject *)b;
  73. fail:PyErr_SetString(PyExc_TypeError,
  74. "block initialiser must be string or list of integers");
  75. Py_DECREF(b);
  76. return NULL;
  77. }
  78. static PyObject *PyRegister(PyObject *self,PyObject *args)
  79. { int size,ptr;
  80. PyBlockObject *b;
  81. if(!PyArg_ParseTuple(args,"ii",&size,&ptr)) return NULL;
  82. if(size<1) size=1;
  83. b=PyObject_NEW(PyBlockObject,&PyBlockType);
  84. if(!b) return NULL;
  85. b->block=(void*)ptr;
  86. b->length=4*size;
  87. b->heap=0;
  88. return (PyObject *)b;
  89. }
  90. static PyObject *PyBlock_ToString(PyBlockObject *self,PyObject *arg)
  91. { int s=0,e=self->length;
  92. if(!PyArg_ParseTuple(arg,"|ii",&s,&e)) return NULL;
  93. if(s<0||e>self->length||s>e)
  94. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  95. return NULL;
  96. }
  97. return PyString_FromStringAndSize((char*)self->block+s,e-s);
  98. }
  99. static PyObject *PyBlock_NullString(PyBlockObject *self,PyObject *arg)
  100. { int s=0,e=self->length,i;
  101. char *p=(char*)self->block;
  102. if(!PyArg_ParseTuple(arg,"|ii",&s,&e)) return NULL;
  103. if(s<0||e>self->length||s>e)
  104. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  105. return NULL;
  106. }
  107. for(i=s;i<e;i++) if(p[i]==0) break;
  108. return PyString_FromStringAndSize((char*)self->block+s,i-s);
  109. }
  110. static PyObject *PyBlock_CtrlString(PyBlockObject *self,PyObject *arg)
  111. { int s=0,e=self->length,i;
  112. char *p=(char*)self->block;
  113. if(!PyArg_ParseTuple(arg,"|ii",&s,&e)) return NULL;
  114. if(s<0||e>self->length||s>e)
  115. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  116. return NULL;
  117. }
  118. for(i=s;i<e;i++) if(p[i]<32) break;
  119. return PyString_FromStringAndSize((char*)self->block+s,i-s);
  120. }
  121. static PyObject *PyBlock_PadString(PyBlockObject *self,PyObject *arg)
  122. { int s=0,e=self->length,n,m;
  123. char *str;
  124. char c;
  125. char *p=(char*)self->block;
  126. if(!PyArg_ParseTuple(arg,"s#c|ii",&str,&n,&c,&s,&e)) return NULL;
  127. if(s<0||e>self->length||s>e)
  128. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  129. return NULL;
  130. }
  131. m=e-s;
  132. if(n>m) n=m;
  133. memcpy(p+s,str,n);memset(p+s+n,c,m-n);
  134. Py_INCREF(Py_None);return Py_None;
  135. }
  136. static PyObject *PyBlock_BitSet(PyBlockObject *self,PyObject *arg)
  137. { int i,x,y;
  138. int *p=(int*)self->block;
  139. if(!PyArg_ParseTuple(arg,"iii",&i,&x,&y)) return NULL;
  140. if(i<0||i>=self->length/4)
  141. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  142. return NULL;
  143. }
  144. p[i]=(p[i]&y)^x;
  145. Py_INCREF(Py_None);return Py_None;
  146. }
  147. static PyObject *PyBlock_Resize(PyBlockObject *self,PyObject *arg)
  148. { int n;
  149. if(!PyArg_ParseTuple(arg,"i",&n)) return NULL;
  150. if(n<1) n=1;
  151. if(self->heap)
  152. { void *v=realloc(self->block,4*n);
  153. if (!v) return PyErr_NoMemory();
  154. self->block=v;
  155. }
  156. self->length=4*n;
  157. Py_INCREF(Py_None);return Py_None;
  158. }
  159. static PyObject *PyBlock_ToFile(PyBlockObject *self,PyObject *arg)
  160. { int s=0,e=self->length/4;
  161. PyObject *f;
  162. FILE *fp;
  163. if(!PyArg_ParseTuple(arg,"O|ii",&f,&s,&e)) return NULL;
  164. fp=PyFile_AsFile(f);
  165. if (!fp)
  166. { PyErr_SetString(PyExc_TypeError, "arg must be open file");
  167. return NULL;
  168. }
  169. fwrite((int*)(self->block)+s,4,e-s,fp);
  170. Py_INCREF(Py_None);return Py_None;
  171. }
  172. static struct PyMethodDef PyBlock_Methods[]=
  173. { { "tostring",(PyCFunction)PyBlock_ToString,1},
  174. { "padstring",(PyCFunction)PyBlock_PadString,1},
  175. { "nullstring",(PyCFunction)PyBlock_NullString,1},
  176. { "ctrlstring",(PyCFunction)PyBlock_CtrlString,1},
  177. { "bitset",(PyCFunction)PyBlock_BitSet,1},
  178. { "resize",(PyCFunction)PyBlock_Resize,1},
  179. { "tofile",(PyCFunction)PyBlock_ToFile,1},
  180. { NULL,NULL} /* sentinel */
  181. };
  182. static int block_len(PyBlockObject *b)
  183. { return b->length/4;
  184. }
  185. static PyObject *block_concat(PyBlockObject *b,PyBlockObject *c)
  186. { PyErr_SetString(PyExc_IndexError,"block concatenation not implemented");
  187. return NULL;
  188. }
  189. static PyObject *block_repeat(PyBlockObject *b,Py_ssize_t i)
  190. { PyErr_SetString(PyExc_IndexError,"block repetition not implemented");
  191. return NULL;
  192. }
  193. static PyObject *block_item(PyBlockObject *b,Py_ssize_t i)
  194. { if(i<0||4*i>=b->length)
  195. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  196. return NULL;
  197. }
  198. return PyInt_FromLong(((long*)(b->block))[i]);
  199. }
  200. static PyObject *block_slice(PyBlockObject *b,Py_ssize_t i,Py_ssize_t j)
  201. { Py_ssize_t n,k;
  202. long *p=b->block;
  203. PyObject *result;
  204. if(j>b->length/4) j=b->length/4;
  205. if(i<0||i>j)
  206. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  207. return NULL;
  208. }
  209. n=j-i;
  210. result=PyList_New(n);
  211. for(k=0;k<n;k++) PyList_SetItem(result,k,PyInt_FromSsize_t(p[i+k]));
  212. return result;
  213. }
  214. static int block_ass_item(PyBlockObject *b,Py_ssize_t i,PyObject *v)
  215. { if(i<0||i>=b->length/4)
  216. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  217. return -1;
  218. }
  219. if(!PyInt_Check(v))
  220. { PyErr_SetString(PyExc_TypeError,"block item must be integer");
  221. return -1;
  222. }
  223. ((long*)(b->block))[i]=PyInt_AsLong(v);
  224. return 0;
  225. }
  226. static int block_ass_slice(PyBlockObject *b,Py_ssize_t i,Py_ssize_t j,PyObject *v)
  227. { Py_ssize_t n,k;
  228. long *p=b->block;
  229. if(j>b->length/4) j=b->length/4;
  230. if(i<0||i>j)
  231. { PyErr_SetString(PyExc_IndexError,"block index out of range");
  232. return -1;
  233. }
  234. if(!PyList_Check(v)) goto fail;
  235. n=PyList_Size(v);
  236. if(n>j-i) n=j-i;
  237. for(k=0;k<n;k++)
  238. { PyObject *q=PyList_GetItem(v,k);
  239. if(!PyInt_Check(q)) goto fail;
  240. p[i+k]=PyInt_AsLong(q);
  241. }
  242. for(;k<j-i;k++) p[i+k]=0;
  243. return 0;
  244. fail:PyErr_SetString(PyExc_TypeError,"block slice must be integer list");
  245. return -1;
  246. }
  247. static PySequenceMethods block_as_sequence=
  248. { (inquiry)block_len, /*sq_length*/
  249. (binaryfunc)block_concat, /*sq_concat*/
  250. (ssizeargfunc)block_repeat, /*sq_repeat*/
  251. (ssizeargfunc)block_item, /*sq_item*/
  252. (ssizessizeargfunc)block_slice, /*sq_slice*/
  253. (ssizeobjargproc)block_ass_item, /*sq_ass_item*/
  254. (ssizessizeobjargproc)block_ass_slice, /*sq_ass_slice*/
  255. };
  256. static PyObject *PyBlock_GetAttr(PyBlockObject *s,char *name)
  257. {
  258. if (!strcmp(name, "length")) return PyInt_FromLong((long)s->length);
  259. if (!strcmp(name, "start")) return PyInt_FromLong((long)s->block);
  260. if (!strcmp(name,"end")) return PyInt_FromLong(((long)(s->block)+s->length));
  261. if (!strcmp(name, "__members__"))
  262. { PyObject *list = PyList_New(3);
  263. if (list)
  264. { PyList_SetItem(list, 0, PyString_FromString("length"));
  265. PyList_SetItem(list, 1, PyString_FromString("start"));
  266. PyList_SetItem(list, 2, PyString_FromString("end"));
  267. if (PyErr_Occurred()) { Py_DECREF(list);list = NULL;}
  268. }
  269. return list;
  270. }
  271. return Py_FindMethod(PyBlock_Methods, (PyObject*) s,name);
  272. }
  273. static void PyBlock_Dealloc(PyBlockObject *b)
  274. {
  275. if(b->heap) {
  276. if (b->heap)
  277. ;
  278. else
  279. PyMem_DEL(b->block);
  280. }
  281. PyMem_DEL(b);
  282. }
  283. static PyTypeObject PyBlockType=
  284. { PyObject_HEAD_INIT(&PyType_Type)
  285. 0, /*ob_size*/
  286. "block", /*tp_name*/
  287. sizeof(PyBlockObject), /*tp_size*/
  288. 0, /*tp_itemsize*/
  289. /* methods */
  290. (destructor)PyBlock_Dealloc, /*tp_dealloc*/
  291. 0, /*tp_print*/
  292. (getattrfunc)PyBlock_GetAttr, /*tp_getattr*/
  293. 0, /*tp_setattr*/
  294. 0, /*tp_compare*/
  295. 0, /*tp_repr*/
  296. 0, /*tp_as_number*/
  297. &block_as_sequence, /*tp_as_sequence*/
  298. 0, /*tp_as_mapping*/
  299. 0, /*tp_hash*/
  300. };
  301. /* swi commands */
  302. static PyObject *swi_swi(PyObject *self,PyObject *args)
  303. { PyObject *name,*format,*result,*v;
  304. int swino,carry,rno=0,j,n;
  305. char *swiname,*fmt,*outfmt;
  306. _kernel_swi_regs r;
  307. PyBlockObject *ao;
  308. if(args==NULL||!PyTuple_Check(args)||(n=PyTuple_Size(args))<2)
  309. { PyErr_BadArgument(); return NULL;}
  310. name=PyTuple_GetItem(args,0);
  311. if(!PyArg_Parse(name,"i",&swino))
  312. { PyErr_Clear();
  313. if(!PyArg_Parse(name,"s",&swiname)) return NULL;
  314. e=xos_swi_number_from_string(swiname,&swino);
  315. if(e) return swi_oserror();
  316. }
  317. format=PyTuple_GetItem(args,1);
  318. if(!PyArg_Parse(format,"s",&fmt)) return NULL;
  319. j=2;
  320. for(;;fmt++)
  321. { switch(*fmt)
  322. { case '.': rno++;continue;
  323. case ';':case 0: goto swicall;
  324. case '0':case '1':case '2':case '3':case '4':
  325. case '5':case '6':case '7':case '8':case '9':
  326. r.r[rno++]=*fmt-'0';continue;
  327. case '-':r.r[rno++]=-1;continue;
  328. }
  329. if(j>=n) return swi_error("Too few arguments");
  330. v=PyTuple_GetItem(args,j++);
  331. switch(*fmt)
  332. { case 'i':if(!PyArg_Parse(v,"i",&r.r[rno])) return NULL;
  333. break;
  334. case 's':if(!PyArg_Parse(v,"s",(char**)(&r.r[rno]))) return NULL;
  335. break;
  336. case 'b':if(!PyArg_Parse(v,"O",(PyObject**)&ao)) return NULL;
  337. if(!PyBlock_Check(v)) return swi_error("Not a block");
  338. r.r[rno]=(int)(ao->block);
  339. break;
  340. case 'e':if(!PyArg_Parse(v,"O",(PyObject**)&ao)) return NULL;
  341. if(!PyBlock_Check(v)) return swi_error("Not a block");
  342. r.r[rno]=(int)(ao->block)+ao->length;
  343. break;
  344. default:return swi_error("Odd format character");
  345. }
  346. rno++;
  347. }
  348. swicall:e=(os_error*)_kernel_swi_c(swino,&r,&r,&carry);
  349. if(e) return swi_oserror();
  350. if(*fmt==0) { Py_INCREF(Py_None);return Py_None;}
  351. n=0;
  352. for(outfmt=++fmt;*outfmt;outfmt++) switch(*outfmt)
  353. { case 'i':case 's':case '*':n++;break;
  354. case '.':break;
  355. default:return swi_error("Odd format character");
  356. }
  357. if(n==0) { Py_INCREF(Py_None);return Py_None;}
  358. if(n!=1)
  359. { result=PyTuple_New(n);
  360. if(!result) return NULL;
  361. }
  362. rno=0;j=0;
  363. for(;*fmt;fmt++)
  364. { switch(*fmt)
  365. { case 'i':v=PyInt_FromLong((long)r.r[rno++]); break;
  366. case 's':v=PyString_FromString((char*)(r.r[rno++])); break;
  367. case '.':rno++; continue;
  368. case '*':v=PyInt_FromLong((long)carry); break;
  369. }
  370. if(!v) goto fail;
  371. if(n==1) return v;
  372. PyTuple_SetItem(result,j,v);
  373. j++;
  374. }
  375. return result;
  376. fail:Py_DECREF(result);return 0;
  377. }
  378. static PyObject *swi_string(PyObject *self, PyObject *arg)
  379. { char *s;
  380. int l=-1;
  381. if(!PyArg_ParseTuple(arg,"i|i",(unsigned int *)&s, &l)) return NULL;
  382. if (l==-1)
  383. l = strlen(s);
  384. return PyString_FromStringAndSize((char*)s, l);
  385. }
  386. static char swi_string__doc__[] =
  387. "string(address[, length]) -> string\n\
  388. Read a null terminated string from the given address.";
  389. static PyObject *swi_integer(PyObject *self, PyObject *arg)
  390. { int *i;
  391. if(!PyArg_ParseTuple(arg,"i",(unsigned int *)&i))
  392. return NULL;
  393. return PyInt_FromLong(*i);
  394. }
  395. static char swi_integer__doc__[] =
  396. "integer(address) -> string\n\
  397. Read an integer from the given address.";
  398. static PyObject *swi_integers(PyObject *self, PyObject *arg)
  399. { int *i;
  400. int c=-1;
  401. PyObject *result, *result1;
  402. if(!PyArg_ParseTuple(arg,"i|i",(unsigned int *)&i, &c)) return NULL;
  403. result=PyList_New(0);
  404. if (result) {
  405. while ( c>0 || (c==-1 && *i) ) {
  406. result1 = PyInt_FromLong((long)*i);
  407. if (!result1) {
  408. Py_DECREF(result);
  409. return NULL;
  410. }
  411. if (PyList_Append(result, result1)!=0) {
  412. Py_DECREF(result);
  413. Py_DECREF(result1);
  414. return NULL;
  415. };
  416. i++;
  417. if (c!=-1)
  418. c--;
  419. }
  420. }
  421. return result;
  422. }
  423. static char swi_integers__doc__[] =
  424. "integers(address[, count]) -> string\n\
  425. Either read a null terminated list of integers or\n\
  426. a list of given length from the given address.";
  427. static PyObject *swi_tuples(PyObject *self, PyObject *arg)
  428. {
  429. unsigned char *i; /* points to current */
  430. int c=-1, l=4, j, zero; /* count, length, index */
  431. PyObject *result, *result1, *result11;
  432. if(!PyArg_ParseTuple(arg,"i|ii",(unsigned int *)&i, &l, &c)) return NULL;
  433. result=PyList_New(0);
  434. if (result) {
  435. while (c) {
  436. result1 = PyTuple_New(l);
  437. if (!result1) {
  438. Py_DECREF(result);
  439. return NULL;
  440. }
  441. zero = (c==-1); /* check for zeros? */
  442. for(j=0;j<l;j++) {
  443. if (zero && *i)
  444. zero = 0; /* non-zero found */
  445. result11 = PyInt_FromLong((long)(*i));
  446. if (!result11) {
  447. Py_DECREF(result);
  448. return NULL;
  449. }
  450. PyTuple_SetItem(result1, j, result11);
  451. i++;
  452. }
  453. if (c==-1 && zero) {
  454. Py_DECREF(result1);
  455. c = 0;
  456. break;
  457. }
  458. if (PyList_Append(result, result1)!=0) {
  459. Py_DECREF(result1);
  460. Py_DECREF(result);
  461. return NULL;
  462. }
  463. if (c!=-1)
  464. c--;
  465. }
  466. }
  467. return result;
  468. }
  469. static char swi_tuples__doc__[] =
  470. "tuples(address[, length=4[, count]]) -> string\n\
  471. Either read a null terminated list of byte tuples or\n\
  472. a list of given length from the given address.";
  473. static PyObject *swi_tuple(PyObject *self, PyObject *arg)
  474. {
  475. unsigned char *i; /* points to current */
  476. int c=1, j;
  477. PyObject *result, *result1;
  478. if(!PyArg_ParseTuple(arg,"i|i",(unsigned int *)&i, &c)) return NULL;
  479. result = PyTuple_New(c);
  480. if (!result)
  481. return NULL;
  482. for(j=0;j<c;j++) {
  483. result1 = PyInt_FromLong((long)(i[j]));
  484. if (!result1) {
  485. Py_DECREF(result);
  486. return NULL;
  487. }
  488. PyTuple_SetItem(result, j, result1);
  489. }
  490. return result;
  491. }
  492. static char swi_tuple__doc__[] =
  493. "tuple(address[, count=1]]) -> tuple\n\
  494. Read count bytes from given address.";
  495. static PyMethodDef SwiMethods[]=
  496. { { "swi", swi_swi, METH_VARARGS},
  497. { "block", PyBlock_New, METH_VARARGS},
  498. { "register", PyRegister, METH_VARARGS},
  499. { "string", swi_string, METH_VARARGS, swi_string__doc__},
  500. { "integer", swi_integer, METH_VARARGS, swi_integer__doc__},
  501. { "integers", swi_integers, METH_VARARGS, swi_integers__doc__},
  502. { "tuples", swi_tuples, METH_VARARGS, swi_tuples__doc__},
  503. { "tuple", swi_tuple, METH_VARARGS, swi_tuple__doc__},
  504. { NULL,NULL,0,NULL} /* Sentinel */
  505. };
  506. void initswi()
  507. { PyObject *m, *d;
  508. m = Py_InitModule("swi", SwiMethods);
  509. d = PyModule_GetDict(m);
  510. SwiError=PyErr_NewException("swi.error", NULL, NULL);
  511. PyDict_SetItemString(d,"error",SwiError);
  512. ArgError=PyErr_NewException("swi.ArgError", NULL, NULL);
  513. PyDict_SetItemString(d,"ArgError",ArgError);
  514. }