Commit | Line | Data |
---|---|---|
c1756f78 MW |
1 | /* -*-c-*- |
2 | * | |
3 | * Pyke: the Python Kit for Extensions, mLib integration | |
4 | * | |
5 | * (c) 2019 Straylight/Edgeware | |
6 | */ | |
7 | ||
8 | /*----- Licensing notice --------------------------------------------------* | |
9 | * | |
10 | * This file is part of Pyke: the Python Kit for Extensions. | |
11 | * | |
12 | * Pyke is free software: you can redistribute it and/or modify it under | |
13 | * the terms of the GNU General Public License as published by the Free | |
14 | * Software Foundation; either version 2 of the License, or (at your | |
15 | * option) any later version. | |
16 | * | |
17 | * Pyke is distributed in the hope that it will be useful, but WITHOUT | |
18 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
19 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
20 | * for more details. | |
21 | * | |
22 | * You should have received a copy of the GNU General Public License | |
23 | * along with Pyke. If not, write to the Free Software Foundation, Inc., | |
24 | * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | |
25 | */ | |
26 | ||
27 | /*----- Header files ------------------------------------------------------*/ | |
28 | ||
29 | #include "pyke-mLib.h" | |
30 | ||
31 | /* #undef HAVE_LONG_LONG */ | |
32 | ||
33 | /*----- Conversions -------------------------------------------------------*/ | |
34 | ||
35 | #ifndef HAVE_LONG_LONG | |
36 | static PyObject *i32 = 0; | |
37 | static int init_i32(void) | |
38 | { if (!i32 && (i32 = PyInt_FromLong(32)) == 0) return (-1); return (0); } | |
39 | #endif | |
40 | ||
41 | PyObject *getk64(kludge64 u) | |
42 | { | |
43 | #ifdef HAVE_LONG_LONG | |
44 | return (PyLong_FromUnsignedLongLong(GET64(unsigned PY_LONG_LONG, u))); | |
45 | #else | |
46 | PyObject *i = 0, *j = 0, *t; | |
47 | PyObject *rc = 0; | |
48 | ||
49 | if (init_i32()) goto end; | |
50 | if ((i = PyLong_FromUnsignedLong(HI64(u))) == 0) goto end; | |
51 | if ((t = PyNumber_InPlaceLshift(i, i32)) == 0) goto end; | |
52 | Py_DECREF(i); i = t; | |
53 | if ((j = PyLong_FromUnsignedLong(LO64(u))) == 0) goto end; | |
54 | if ((t = PyNumber_InPlaceOr(i, j)) == 0) goto end; | |
55 | Py_DECREF(i); i = t; | |
56 | if ((rc = PyNumber_Int(i)) == 0) goto end; | |
57 | end: | |
58 | Py_XDECREF(i); | |
59 | Py_XDECREF(j); | |
60 | return (rc); | |
61 | #endif | |
62 | } | |
63 | ||
64 | #ifdef HAVE_UINT64 | |
65 | # define CONVu64(n) do { \ | |
66 | kludge64 k; \ | |
67 | uint64 t; \ | |
68 | if (!convk64(o, &k)) goto end; \ | |
69 | t = GET64(uint64, k); \ | |
70 | if (t > MASK##n) VALERR("out of range"); \ | |
71 | *p = t; \ | |
72 | } while (0) | |
73 | #else | |
74 | # define CONVu64(n) assert(!"shouldn't be possible") | |
75 | #endif | |
76 | ||
77 | #define CONVU_(n) \ | |
78 | int convu##n(PyObject *o, void *pp) \ | |
79 | { \ | |
80 | unsigned long u; \ | |
81 | uint##n *p = pp; \ | |
82 | \ | |
83 | if (MASK##n > ULONG_MAX) \ | |
84 | CONVu64(n); \ | |
85 | else { \ | |
86 | if (!convulong(o, &u)) goto end; \ | |
87 | if (u > MASK##n) VALERR("out of range"); \ | |
88 | *p = u; \ | |
89 | } \ | |
90 | return (1); \ | |
91 | end: \ | |
92 | return (0); \ | |
93 | } | |
94 | DOUINTSZ(CONVU_) | |
95 | ||
96 | int convk64(PyObject *o, void *pp) | |
97 | { | |
98 | PyObject *i = 0; | |
99 | int rc = 0; | |
100 | #if HAVE_LONG_LONG | |
101 | unsigned PY_LONG_LONG t; | |
102 | #else | |
103 | PyObject *t; | |
104 | uint32 lo, hi; | |
105 | #endif | |
106 | ||
107 | #if HAVE_LONG_LONG | |
108 | if ((i = PyNumber_Long(o)) == 0) goto end; | |
109 | t = PyLong_AsUnsignedLongLong(i); | |
110 | if (t == (unsigned PY_LONG_LONG)-1 && PyErr_Occurred()) goto end; | |
111 | ASSIGN64(*(kludge64 *)pp, t); | |
112 | #else | |
113 | if (init_i32()) goto end; | |
114 | if ((i = PyNumber_Int(o)) == 0) goto end; | |
115 | lo = PyInt_AsUnsignedLongMask(i); | |
116 | if ((t = PyNumber_InPlaceRshift(i, i32)) == 0) goto end; | |
117 | Py_DECREF(i); i = t; | |
118 | hi = PyInt_AsUnsignedLongMask(i); | |
119 | if ((t = PyNumber_InPlaceRshift(i, i32)) == 0) goto end; | |
120 | Py_DECREF(i); i = t; | |
121 | if (PyObject_IsTrue(i)) VALERR("out of range"); | |
122 | SET64(*(kludge64 *)pp, hi, lo); | |
123 | #endif | |
124 | rc = 1; | |
125 | ||
126 | end: | |
127 | Py_XDECREF(i); | |
128 | return (rc); | |
129 | } | |
130 | ||
131 | /*----- That's all, folks -------------------------------------------------*/ |