1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
1300
1301
1302
1303
1304
1305
1306
1307
1308
1309
1310
1311
1312
1313
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324
1325
1326
1327
1328
1329
1330
1331
1332
1333
1334
1335
1336
1337
1338
1339
1340
1341
1342
1343
1344
1345
1346
1347
1348
1349
1350
1351
1352
1353
1354
1355
1356
1357
1358
1359
1360
1361
1362
1363
1364
1365
1366
1367
1368
1369
1370
1371
1372
1373
1374
1375
1376
1377
1378
1379
1380
1381
1382
1383
1384
1385
1386
1387
1388
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398
1399
1400
1401
1402
1403
1404
1405
1406
1407
1408
1409
1410
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
1444
1445
1446
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
1461
1462
1463
1464
1465
1466
1467
1468
1469
1470
1471
1472
1473
1474
1475
1476
1477
1478
1479
1480
1481
1482
1483
1484
1485
1486
1487
1488
1489
1490
1491
1492
1493
1494
1495
1496
1497
1498
1499
1500
1501
1502
1503
1504
1505
1506
1507
1508
1509
1510
1511
1512
1513
1514
1515
1516
1517
1518
1519
1520
1521
1522
1523
1524
1525
1526
1527
1528
1529
1530
1531
1532
1533
1534
1535
1536
1537
1538
1539
1540
1541
1542
1543
1544
1545
1546
1547
1548
1549
1550
1551
1552
1553
1554
1555
1556
1557
1558
1559
1560
1561
1562
1563
1564
1565
1566
1567
1568
1569
1570
1571
1572
1573
1574
1575
1576
1577
1578
1579
1580
1581
1582
1583
1584
1585
1586
1587
1588
1589
1590
1591
1592
1593
1594
1595
1596
1597
1598
1599
1600
1601
1602
1603
1604
1605
1606
1607
1608
1609
1610
1611
1612
1613
1614
1615
1616
1617
1618
1619
1620
1621
1622
1623
1624
1625
1626
1627
1628
1629
1630
1631
1632
1633
1634
1635
1636
1637
1638
1639
1640
1641
1642
1643
1644
1645
1646
1647
1648
1649
1650
1651
1652
1653
1654
1655
1656
1657
1658
1659
1660
1661
1662
1663
1664
1665
1666
1667
1668
1669
1670
1671
1672
1673
1674
1675
1676
1677
1678
1679
1680
1681
1682
1683
1684
1685
1686
1687
1688
1689
1690
1691
1692
1693
1694
1695
1696
1697
1698
1699
1700
1701
1702
1703
1704
1705
1706
1707
1708
1709
1710
1711
1712
1713
1714
1715
1716
1717
1718
1719
1720
1721
1722
1723
1724
1725
1726
1727
1728
1729
1730
1731
1732
1733
1734
1735
1736
1737
1738
1739
1740
1741
1742
1743
1744
1745
1746
1747
1748
1749
1750
1751
1752
1753
1754
1755
1756
1757
1758
1759
1760
1761
1762
1763
1764
1765
1766
1767
1768
1769
1770
1771
1772
1773
1774
1775
1776
1777
1778
1779
1780
1781
1782
1783
1784
1785
1786
1787
1788
1789
1790
1791
1792
1793
1794
1795
1796
1797
1798
1799
1800
1801
1802
1803
1804
1805
1806
1807
1808
1809
1810
1811
1812
1813
1814
1815
1816
1817
1818
1819
1820
1821
1822
1823
1824
1825
1826
1827
1828
1829
1830
1831
1832
1833
1834
1835
1836
1837
1838
1839
1840
1841
1842
1843
1844
1845
1846
1847
1848
1849
1850
1851
1852
1853
1854
1855
1856
1857
1858
1859
1860
1861
1862
1863
1864
1865
1866
1867
1868
1869
1870
1871
1872
1873
1874
1875
1876
1877
1878
1879
1880
1881
1882
1883
1884
1885
1886
1887
1888
1889
1890
1891
1892
1893
1894
1895
1896
1897
1898
1899
1900
1901
1902
1903
1904
1905
1906
1907
1908
1909
1910
1911
1912
1913
1914
1915
1916
1917
1918
1919
1920
1921
1922
1923
1924
1925
1926
1927
1928
1929
1930
1931
1932
1933
1934
1935
1936
1937
1938
1939
1940
1941
1942
1943
1944
1945
1946
1947
1948
1949
1950
1951
1952
1953
1954
1955
1956
1957
1958
1959
1960
1961
1962
1963
1964
1965
1966
1967
1968
1969
1970
1971
1972
1973
1974
1975
1976
1977
1978
1979
1980
1981
1982
1983
1984
1985
1986
1987
1988
1989
1990
1991
1992
1993
1994
1995
1996
1997
1998
1999
2000
2001
2002
2003
2004
|
import ctypes
import sys, os
from collections import defaultdict
import py
from pypy import pypydir
from rpython.rtyper.lltypesystem import rffi, lltype
from rpython.rtyper.tool import rffi_platform
from rpython.rtyper.lltypesystem import ll2ctypes
from rpython.rtyper.annlowlevel import llhelper
from rpython.rlib.objectmodel import we_are_translated, keepalive_until_here
from rpython.rlib.objectmodel import dont_inline
from rpython.rlib.rfile import (FILEP, c_fread, c_fclose, c_fwrite,
c_fdopen, c_fileno, c_ferror,
c_fopen)# for tests
from rpython.rlib import jit, rutf8
from rpython.rlib.rarithmetic import widen
from rpython.translator import cdir
from rpython.translator.tool.cbuild import ExternalCompilationInfo
from rpython.translator.gensupp import NameManager
from rpython.tool.udir import udir
from pypy.module.cpyext.state import State
from pypy.interpreter.error import OperationError, oefmt, raise_import_error
from pypy.interpreter.baseobjspace import W_Root
from pypy.interpreter.gateway import unwrap_spec
from pypy.interpreter.nestedscope import Cell
from pypy.interpreter.module import Module
from pypy.interpreter.function import StaticMethod
from pypy.interpreter.pyparser import pygram
from pypy.objspace.std.sliceobject import W_SliceObject
from pypy.objspace.std.unicodeobject import encode_object
from pypy.module.__builtin__.descriptor import W_Property
#from pypy.module.micronumpy.base import W_NDimArray
from pypy.module.__pypy__.interp_buffer import W_Bufferable
from rpython.rlib.entrypoint import entrypoint_lowlevel
from rpython.rlib.rposix import SuppressIPH
from rpython.rlib.unroll import unrolling_iterable
from rpython.rlib.objectmodel import specialize
from pypy.module.exceptions import interp_exceptions
from rpython.tool.sourcetools import func_with_new_name
from rpython.rtyper.lltypesystem.lloperation import llop
from rpython.rlib import rawrefcount
from rpython.rlib import rthread
from rpython.rlib.debug import fatalerror_notb
from rpython.rlib import rstackovf
from pypy.objspace.std.typeobject import W_TypeObject, find_best_base
from rpython.tool.cparser import CTypeSpace
DEBUG_WRAPPER = True
if sys.platform == 'win32':
dash = '_'
WIN32 = True
else:
dash = ''
WIN32 = False
pypydir = py.path.local(pypydir)
include_dir = pypydir / 'module' / 'cpyext' / 'include'
pc_dir = pypydir / 'module' / 'cpyext' / 'PC'
parse_dir = pypydir / 'module' / 'cpyext' / 'parse'
source_dir = pypydir / 'module' / 'cpyext' / 'src'
translator_c_dir = py.path.local(cdir)
include_dirs = [
include_dir,
parse_dir,
translator_c_dir,
udir,
]
if WIN32:
include_dirs.insert(0, pc_dir)
configure_eci = ExternalCompilationInfo(
include_dirs=include_dirs,
includes=['Python.h', 'stdarg.h', 'structmember.h', 'marshal.h'],
compile_extra=['-DPy_BUILD_CORE'])
class CConfig:
_compilation_info_ = configure_eci
class CConfig2:
_compilation_info_ = configure_eci
class CConfig_constants:
_compilation_info_ = configure_eci
CONST_STRING = lltype.Ptr(lltype.Array(lltype.Char,
hints={'nolength': True}),
use_cache=False)
CONST_STRINGP = lltype.Ptr(lltype.Array(rffi.CCHARP,
hints={'nolength': True}),
use_cache=False)
CONST_WSTRING = lltype.Ptr(lltype.Array(lltype.UniChar,
hints={'nolength': True}),
use_cache=False)
assert CONST_STRING is not rffi.CCHARP
assert CONST_STRING == rffi.CCHARP
assert CONST_STRINGP is not rffi.CCHARPP
assert CONST_STRINGP == rffi.CCHARPP
assert CONST_WSTRING is not rffi.CWCHARP
assert CONST_WSTRING == rffi.CWCHARP
# FILE* interface
def fclose(fp):
try:
with SuppressIPH():
return c_fclose(fp)
except IOError:
return -1
def fwrite(buf, sz, n, fp):
with SuppressIPH():
return c_fwrite(buf, sz, n, fp)
def fread(buf, sz, n, fp):
with SuppressIPH():
return c_fread(buf, sz, n, fp)
_feof = rffi.llexternal('feof', [FILEP], rffi.INT)
def feof(fp):
with SuppressIPH():
return _feof(fp)
_ferror = rffi.llexternal('ferror', [FILEP], rffi.INT)
def ferror(fp):
with SuppressIPH():
return _ferror(fp)
pypy_decl = 'pypy_decl.h'
udir.join(pypy_decl).write("/* Will be filled later */\n")
udir.join('pypy_structmember_decl.h').write("/* Will be filled later */\n")
udir.join('pypy_marshal_decl.h').write("/* Will be filled later */\n")
udir.join('pypy_macros.h').write("/* Will be filled later */\n")
constant_names = """
Py_TPFLAGS_READY Py_TPFLAGS_READYING
METH_COEXIST METH_STATIC METH_CLASS Py_TPFLAGS_BASETYPE
METH_NOARGS METH_VARARGS METH_KEYWORDS METH_FASTCALL METH_O
Py_TPFLAGS_HEAPTYPE
Py_LT Py_LE Py_EQ Py_NE Py_GT Py_GE Py_MAX_NDIMS
Py_CLEANUP_SUPPORTED PyBUF_READ
PyBUF_FORMAT PyBUF_ND PyBUF_STRIDES PyBUF_WRITABLE PyBUF_SIMPLE PyBUF_WRITE
PY_SSIZE_T_MAX PY_SSIZE_T_MIN
""".split()
for name in ('LONG', 'LIST', 'TUPLE', 'UNICODE', 'DICT', 'BASE_EXC',
'TYPE', 'BYTES'):
constant_names.append('Py_TPFLAGS_%s_SUBCLASS' % name)
#pystrtod.h flags
for name in ('SIGN', 'ADD_DOT_0', 'ALT'):
constant_names.append('Py_DTSF_%s' % name)
# PyPy-specific flags
for name in ('FLOAT',):
constant_names.append('Py_TPPYPYFLAGS_%s_SUBCLASS' % name)
for name in constant_names:
setattr(CConfig_constants, name, rffi_platform.ConstantInteger(name))
globals().update(rffi_platform.configure(CConfig_constants))
def _copy_header_files(headers, dstdir):
for header in headers:
target = dstdir.join(header.basename)
try:
header.copy(dstdir)
except py.error.EACCES:
target.remove() # maybe it was a read-only file
header.copy(dstdir)
target.chmod(0444) # make the file read-only, to make sure that nobody
# edits it by mistake
def copy_header_files(cts, dstdir, copy_numpy_headers):
# XXX: 20 lines of code to recursively copy a directory, really??
assert dstdir.check(dir=True)
headers = include_dir.listdir('*.h') + include_dir.listdir('*.inl')
for name in ["pypy_macros.h", "graminit.h"] + FUNCTIONS_BY_HEADER.keys():
headers.append(udir.join(name))
for path in cts.parsed_headers:
headers.append(path)
_copy_header_files(headers, dstdir)
if copy_numpy_headers:
try:
dstdir.mkdir('_numpypy')
dstdir.mkdir('_numpypy/numpy')
except py.error.EEXIST:
pass
numpy_dstdir = dstdir / '_numpypy' / 'numpy'
numpy_include_dir = include_dir / '_numpypy' / 'numpy'
numpy_headers = numpy_include_dir.listdir('*.h') + numpy_include_dir.listdir('*.inl')
_copy_header_files(numpy_headers, numpy_dstdir)
if WIN32:
# Override pyconfig.h with the one for windows
PC_dir = pypydir / 'module' / 'cpyext' / 'PC'
headers = PC_dir.listdir('*.h')
_copy_header_files(headers, dstdir)
class NotSpecified(object):
pass
_NOT_SPECIFIED = NotSpecified()
class CannotFail(object):
pass
CANNOT_FAIL = CannotFail()
# The same function can be called in three different contexts:
# (1) from C code
# (2) in the test suite, though the "api" object
# (3) from RPython code, for example in the implementation of another function.
#
# In contexts (2) and (3), a function declaring a PyObject argument type will
# receive a wrapped pypy object if the parameter name starts with 'w_', a
# reference (= rffi pointer) otherwise; conversion is automatic. Context (2)
# only allows calls with a wrapped object.
#
# Functions with a PyObject return type should return a wrapped object.
#
# Functions may raise exceptions. In context (3), the exception flows normally
# through the calling function. In context (1) and (2), the exception is
# caught; if it is an OperationError, it is stored in the thread state; other
# exceptions generate a OperationError(w_SystemError); and the funtion returns
# the error value specifed in the API.
#
# Handling of the GIL
# -------------------
#
# **make_generic_cpy_call():** RPython to C, with the GIL held.
#
# **make_wrapper():** C to RPython; by default assume that the GIL is
# held, but accepts gil="acquire", "release", "around",
# "pygilstate_ensure", "pygilstate_release".
#
# When a wrapper() is called:
#
# * "acquire": assert that the GIL is not currently held (otherwise,
# deadlock!). Acquire the PyPy GIL.
#
# * gil=None: we should hold the GIL already. But check anyway, just
# in case. Do the acquire/release if it was not acquired before
# (workaround "_auto" case).
#
# * "pygilstate_ensure": if the GIL is already acquired,
# do nothing and set the extra arg to 0. Otherwise,
# do the "acquire" and set the extra arg to 1. Then we'll call
# pystate.py:PyGILState_Ensure() with this extra arg, which will do
# the rest of the logic.
#
# When a wrapper() returns:
#
# * "release": release the PyPy GIL.
#
# * gil=None: we keep holding the GIL in the normal case; we release it
# in the workaround "_auto" case.
#
# * "pygilstate_release": if the argument is PyGILState_UNLOCKED,
# release the PyPy GIL; otherwise, no-op. The rest of the logic of
# PyGILState_Release() should be done before, in pystate.py.
cpyext_namespace = NameManager('cpyext_')
class BaseApiFunction(object):
def __init__(self, argtypes, restype, callable):
self.argtypes = argtypes
self.restype = restype
self.functype = lltype.Ptr(lltype.FuncType(argtypes, restype))
self.callable = callable
self.cdecl = None # default
#
def get_llhelper(space):
return llhelper(self.functype, self.get_wrapper(space))
self.get_llhelper = get_llhelper
def get_api_decl(self, name, c_writer):
restype = self.get_c_restype(c_writer)
args = self.get_c_args(c_writer)
res = self.API_VISIBILITY % (restype,)
return "{res} {name}({args});".format(**locals())
def get_c_restype(self, c_writer):
if self.cdecl:
return self.cdecl.tp.result.get_c_name()
return c_writer.gettype(self.restype).replace('@', '').strip()
def get_c_args(self, c_writer):
if self.cdecl:
args = [tp.get_c_name('arg%d' % i) for i, tp in
enumerate(self.cdecl.tp.args)]
return ', '.join(args) or "void"
args = []
for i, argtype in enumerate(self.argtypes):
if argtype is CONST_STRING:
arg = 'const char *@'
elif argtype is CONST_STRINGP:
arg = 'const char **@'
elif argtype is CONST_WSTRING:
arg = 'const wchar_t *@'
else:
arg = c_writer.gettype(argtype)
arg = arg.replace('@', 'arg%d' % (i,)).strip()
args.append(arg)
args = ', '.join(args) or "void"
return args
def get_ptr_decl(self, name, c_writer):
restype = self.get_c_restype(c_writer)
args = self.get_c_args(c_writer)
return "{restype} (*{name})({args});".format(**locals())
def get_ctypes_impl(self, name, c_writer):
restype = self.get_c_restype(c_writer)
args = self.get_c_args(c_writer)
callargs = ', '.join('arg%d' % (i,)
for i in range(len(self.argtypes)))
if self.restype is lltype.Void:
body = "{ _pypyAPI.%s(%s); }" % (name, callargs)
else:
body = "{ return _pypyAPI.%s(%s); }" % (name, callargs)
return '%s %s(%s)\n%s' % (restype, name, args, body)
class ApiFunction(BaseApiFunction):
API_VISIBILITY = "PyAPI_FUNC(%s)"
def __init__(self, argtypes, restype, callable, error=CANNOT_FAIL,
c_name=None, cdecl=None, gil=None,
result_borrowed=False, result_is_ll=False):
BaseApiFunction.__init__(self, argtypes, restype, callable)
self.error_value = error
self.c_name = c_name
self.cdecl = cdecl
# extract the signature from the (CPython-level) code object
from pypy.interpreter import pycode
sig = pycode.cpython_code_signature(callable.func_code)
assert sig.argnames[0] == 'space'
self.argnames = sig.argnames[1:]
if gil == 'pygilstate_ensure':
assert self.argnames[-1] == 'previous_state'
del self.argnames[-1]
assert len(self.argnames) == len(self.argtypes)
self.gil = gil
self.result_borrowed = result_borrowed
self.result_is_ll = result_is_ll
def __repr__(self):
return "<cpyext function %s>" % (self.callable.__name__,)
def _freeze_(self):
return True
@specialize.memo()
def get_wrapper(self, space):
wrapper = getattr(self, '_wrapper', None)
if wrapper is None:
wrapper = self._wrapper = self._make_wrapper(space)
return wrapper
# Make the wrapper for the cases (1) and (2)
def _make_wrapper(self, space):
"NOT_RPYTHON"
# This logic is obscure, because we try to avoid creating one
# big wrapper() function for every callable. Instead we create
# only one per "signature".
argtypesw = zip(self.argtypes,
[_name.startswith("w_") for _name in self.argnames])
error_value = self.error_value
if (isinstance(self.restype, lltype.Ptr)
and error_value is not CANNOT_FAIL):
assert lltype.typeOf(error_value) == self.restype
assert not error_value # only support error=NULL
error_value = 0 # because NULL is not hashable
if self.result_is_ll:
result_kind = "L"
elif self.result_borrowed:
result_kind = "B" # note: 'result_borrowed' is ignored if we also
else: # say 'result_is_ll=True' (in this case it's
result_kind = "." # up to you to handle refcounting anyway)
signature = (tuple(argtypesw),
self.restype,
result_kind,
error_value,
self.gil)
cache = space.fromcache(WrapperCache)
try:
wrapper_gen = cache.wrapper_gens[signature]
except KeyError:
wrapper_gen = WrapperGen(space, signature)
cache.wrapper_gens[signature] = wrapper_gen
wrapper = wrapper_gen.make_wrapper(self.callable)
wrapper.relax_sig_check = True
if self.c_name is not None:
wrapper.c_name = cpyext_namespace.uniquename(self.c_name)
return wrapper
def get_unwrapper(self):
names = self.argnames
argtypesw = zip(self.argtypes,
[_name.startswith("w_") for _name in self.argnames])
types_names_enum_ui = unrolling_iterable(enumerate(argtypesw))
@specialize.ll()
def unwrapper(space, *args):
from pypy.module.cpyext.pyobject import is_pyobj
from pypy.module.cpyext.pyobject import from_ref, as_pyobj
newargs = ()
keepalives = ()
assert len(args) == len(self.argtypes)
for i, (ARG, is_wrapped) in types_names_enum_ui:
input_arg = args[i]
if is_PyObject(ARG) and not is_wrapped:
# build a 'PyObject *' (not holding a reference)
if not is_pyobj(input_arg):
keepalives += (input_arg,)
arg = rffi.cast(ARG, as_pyobj(space, input_arg))
else:
arg = rffi.cast(ARG, input_arg)
elif ARG == rffi.VOIDP and not is_wrapped:
# unlike is_PyObject case above, we allow any kind of
# argument -- just, if it's an object, we assume the
# caller meant for it to become a PyObject*.
if input_arg is None or isinstance(input_arg, W_Root):
keepalives += (input_arg,)
arg = rffi.cast(ARG, as_pyobj(space, input_arg))
else:
arg = rffi.cast(ARG, input_arg)
elif (is_PyObject(ARG) or ARG == rffi.VOIDP) and is_wrapped:
# build a W_Root, possibly from a 'PyObject *'
if is_pyobj(input_arg):
arg = from_ref(space, input_arg)
else:
arg = input_arg
else:
# arg is not declared as PyObject, no magic
arg = input_arg
newargs += (arg, )
try:
result = self.callable(space, *newargs)
finally:
keepalive_until_here(*keepalives)
#
# this is just a sanity check to ensure that we don't forget to
# specify result_is_ll=True
if self.restype == PyObject:
assert self.result_is_ll == is_pyobj(result)
return result
return unwrapper
DEFAULT_HEADER = 'pypy_decl.h'
def cpython_api(argtypes, restype, error=_NOT_SPECIFIED, header=DEFAULT_HEADER,
gil=None, result_borrowed=False, result_is_ll=False):
"""
Declares a function to be exported.
- `argtypes`, `restype` are lltypes and describe the function signature.
- `error` is the value returned when an applevel exception is raised. The
special value 'CANNOT_FAIL' (also when restype is Void) turns an eventual
exception into a wrapped SystemError. Unwrapped exceptions also cause a
SytemError.
- `header` is the header file to export the function in.
- set `gil` to "acquire", "release" or "around" to acquire the GIL,
release the GIL, or both
"""
assert header is not None
def decorate(func):
if func.__name__ in FUNCTIONS_BY_HEADER[header]:
raise ValueError("%s already registered" % func.__name__)
func._always_inline_ = 'try'
#
# XXX: should we @jit.dont_look_inside all the @cpython_api functions,
# or we should only disable some of them?
func._jit_look_inside_ = False
#
api_function = ApiFunction(
argtypes, restype, func,
error=_compute_error(error, restype), gil=gil,
result_borrowed=result_borrowed, result_is_ll=result_is_ll)
FUNCTIONS_BY_HEADER[header][func.__name__] = api_function
unwrapper = api_function.get_unwrapper()
unwrapper.func = func
unwrapper.api_func = api_function
INTERPLEVEL_API[func.__name__] = unwrapper # used in tests
return unwrapper
return decorate
class COnlyApiFunction(BaseApiFunction):
API_VISIBILITY = "extern %s"
def get_wrapper(self, space):
return self.callable
def __call__(self, *args):
raise TypeError("the function %s should not be directly "
"called from RPython, but only from C" % (self.func,))
def c_only(argtypes, restype):
def decorate(func):
header = DEFAULT_HEADER
if func.__name__ in FUNCTIONS_BY_HEADER[header]:
raise ValueError("%s already registered" % func.__name__)
func._revdb_c_only_ = True # hack for revdb
api_function = COnlyApiFunction(argtypes, restype, func)
FUNCTIONS_BY_HEADER[header][func.__name__] = api_function
return api_function
return decorate
def api_func_from_cdef(func, cdef, cts,
error=_NOT_SPECIFIED, header=DEFAULT_HEADER,
result_is_ll=False):
func._always_inline_ = 'try'
cdecl = cts.parse_func(cdef)
RESULT = cdecl.get_llresult(cts)
api_function = ApiFunction(
cdecl.get_llargs(cts), RESULT, func,
error=_compute_error(error, RESULT), cdecl=cdecl,
result_is_ll=result_is_ll)
FUNCTIONS_BY_HEADER[header][cdecl.name] = api_function
unwrapper = api_function.get_unwrapper()
unwrapper.func = func
unwrapper.api_func = api_function
return unwrapper
def api_decl(cdef, cts, error=_NOT_SPECIFIED, header=DEFAULT_HEADER):
def decorate(func):
return api_func_from_cdef(func, cdef, cts, error=error, header=header)
return decorate
def slot_function(argtypes, restype, error=_NOT_SPECIFIED):
def decorate(func):
func._always_inline_ = 'try'
api_function = ApiFunction(
argtypes, restype, func,
error=_compute_error(error, restype),
c_name=func.__name__)
unwrapper = api_function.get_unwrapper()
unwrapper.func = func
unwrapper.api_func = api_function
return unwrapper
return decorate
def _compute_error(error, restype):
"""Convert error specification to actual error value of type restype."""
if isinstance(restype, lltype.Typedef):
real_restype = restype.OF
else:
real_restype = restype
if error is _NOT_SPECIFIED:
if isinstance(real_restype, lltype.Ptr):
error = lltype.nullptr(real_restype.TO)
elif real_restype is lltype.Void:
error = CANNOT_FAIL
if type(error) is int:
error = rffi.cast(real_restype, error)
return error
def cpython_struct(name, fields, forward=None, level=1):
configname = name.replace(' ', '__')
if level == 1:
config = CConfig
else:
config = CConfig2
setattr(config, configname, rffi_platform.Struct(name, fields))
if forward is None:
forward = lltype.ForwardReference()
TYPES[configname] = forward
return forward
GLOBALS = {}
def register_global(name, typ, expr, header=None):
if header is not None:
name = '%s#%s' % (name, header)
GLOBALS[name] = (typ, expr)
INTERPLEVEL_API = {}
FUNCTIONS_BY_HEADER = defaultdict(dict)
# These are C symbols which cpyext will export, but which are defined in .c
# files somewhere in the implementation of cpyext (rather than being defined in
# RPython).
SYMBOLS_C = [
'Py_FatalError', 'PyOS_snprintf', 'PyOS_vsnprintf', 'PyArg_Parse',
'PyArg_ParseTuple', 'PyArg_UnpackTuple', 'PyArg_ParseTupleAndKeywords',
'PyArg_VaParse', 'PyArg_VaParseTupleAndKeywords', '_PyArg_NoKeywords',
'PyUnicode_FromFormat', 'PyUnicode_FromFormatV', 'PyUnicode_AsWideCharString',
'PyUnicode_GetSize', 'PyUnicode_GetLength', 'PyUnicode_FromWideChar',
'PyModule_AddObject', 'PyModule_AddIntConstant', 'PyModule_AddStringConstant',
'PyModule_GetDef', 'PyModuleDef_Init', 'PyModule_GetState',
'Py_BuildValue', 'Py_VaBuildValue', 'PyTuple_Pack',
'_PyArg_Parse_SizeT', '_PyArg_ParseTuple_SizeT',
'_PyArg_ParseTupleAndKeywords_SizeT', '_PyArg_VaParse_SizeT',
'_PyArg_VaParseTupleAndKeywords_SizeT',
'_Py_BuildValue_SizeT', '_Py_VaBuildValue_SizeT',
'PyErr_Format', 'PyErr_NewException', 'PyErr_NewExceptionWithDoc',
'PyErr_WarnFormat', '_PyErr_FormatFromCause',
'PySys_WriteStdout', 'PySys_WriteStderr',
'PyEval_CallFunction', 'PyEval_CallMethod', 'PyObject_CallFunction',
'PyObject_CallMethod', 'PyObject_CallFunctionObjArgs', 'PyObject_CallMethodObjArgs',
'_PyObject_CallFunction_SizeT', '_PyObject_CallMethod_SizeT',
'PyObject_DelItemString', 'PyObject_GetBuffer', 'PyBuffer_Release',
'_Py_setfilesystemdefaultencoding',
'PyCapsule_New', 'PyCapsule_IsValid', 'PyCapsule_GetPointer',
'PyCapsule_GetName', 'PyCapsule_GetDestructor', 'PyCapsule_GetContext',
'PyCapsule_SetPointer', 'PyCapsule_SetName', 'PyCapsule_SetDestructor',
'PyCapsule_SetContext', 'PyCapsule_Import', 'PyCapsule_Type', '_Py_get_capsule_type',
'PyComplex_AsCComplex', 'PyComplex_FromCComplex',
'PyObject_AsReadBuffer', 'PyObject_AsWriteBuffer', 'PyObject_CheckReadBuffer',
'PyBuffer_GetPointer', 'PyBuffer_ToContiguous', 'PyBuffer_FromContiguous',
'PyImport_ImportModuleLevel',
'PyOS_getsig', 'PyOS_setsig',
'_Py_RestoreSignals',
'PyThread_get_thread_ident', 'PyThread_allocate_lock', 'PyThread_free_lock',
'PyThread_acquire_lock', 'PyThread_release_lock',
'PyThread_create_key', 'PyThread_delete_key', 'PyThread_set_key_value',
'PyThread_get_key_value', 'PyThread_delete_key_value',
'PyThread_ReInitTLS', 'PyThread_init_thread',
'PyThread_start_new_thread',
'PyStructSequence_InitType', 'PyStructSequence_InitType2',
'PyStructSequence_New', 'PyStructSequence_UnnamedField',
'PyStructSequence_NewType',
'PyFunction_Type', 'PyMethod_Type', 'PyRange_Type', 'PyTraceBack_Type',
'Py_FrozenFlag', # not part of sys.flags
'Py_UnbufferedStdioFlag', # not part of sys.flags (python3)
'_Py_PackageContext', 'PyOS_InputHook',
'_Py_PackageContext',
'PyMem_RawMalloc', 'PyMem_RawCalloc', 'PyMem_RawRealloc', 'PyMem_RawFree',
'PyMem_Malloc', 'PyMem_Calloc', 'PyMem_Realloc', 'PyMem_Free',
'PyObject_CallFinalizerFromDealloc',
'PyTraceMalloc_Track', 'PyTraceMalloc_Untrack',
'PyBytes_FromFormat', 'PyBytes_FromFormatV',
'PyType_FromSpec',
'Py_IncRef', 'Py_DecRef', 'PyObject_Free', 'PyObject_GC_Del', 'PyType_GenericAlloc',
'_PyObject_New', '_PyObject_NewVar',
'_PyObject_GC_Malloc', '_PyObject_GC_New', '_PyObject_GC_NewVar',
'PyObject_Init', 'PyObject_InitVar',
'PyTuple_New', '_Py_Dealloc',
'PyVectorcall_Call',
]
if sys.platform == "win32":
SYMBOLS_C.append('Py_LegacyWindowsStdioFlag')
TYPES = {}
FORWARD_DECLS = []
INIT_FUNCTIONS = []
BOOTSTRAP_FUNCTIONS = []
# Keep synchronized with pypy.interpreter.app_main.sys_flags and
# module.sys.app.sysflags. Synchronized in an init_function
_flags = [
# c name, sys.flags name
('Py_DebugFlag', 'debug'),
('Py_InspectFlag', 'inspect'),
('Py_InteractiveFlag', 'interactive'),
('Py_OptimizeFlag', 'optimize'),
('Py_DontWriteBytecodeFlag', 'dont_write_bytecode'),
('Py_NoUserSiteDirectory', 'no_user_site'),
('Py_NoSiteFlag', 'no_site'),
('Py_IgnoreEnvironmentFlag', 'ignore_environment'),
('Py_VerboseFlag', 'verbose'),
('Py_BytesWarningFlag', 'bytes_warning'),
('Py_QuietFlag', 'quiet'),
('Py_HashRandomizationFlag', 'hash_randomization'),
('Py_IsolatedFlag', 'isolated'),
]
SYMBOLS_C += [c_name for c_name, _ in _flags]
# this needs to include all prebuilt pto, otherwise segfaults occur
register_global('_Py_NoneStruct',
'PyObject*', 'space.w_None', header=pypy_decl)
register_global('_Py_TrueStruct',
'PyObject*', 'space.w_True', header=pypy_decl)
register_global('_Py_FalseStruct',
'PyObject*', 'space.w_False', header=pypy_decl)
register_global('_Py_NotImplementedStruct',
'PyObject*', 'space.w_NotImplemented', header=pypy_decl)
register_global('_Py_EllipsisObject',
'PyObject*', 'space.w_Ellipsis', header=pypy_decl)
register_global('PyDateTimeAPI', 'PyDateTime_CAPI*', 'None')
def build_exported_objects():
# Standard exceptions
# PyExc_BaseException, PyExc_Exception, PyExc_ValueError, PyExc_KeyError,
# PyExc_IndexError, PyExc_IOError, PyExc_OSError, PyExc_TypeError,
# PyExc_AttributeError, PyExc_OverflowError, PyExc_ImportError,
# PyExc_NameError, PyExc_MemoryError, PyExc_RuntimeError,
# PyExc_UnicodeEncodeError, PyExc_UnicodeDecodeError, ...
from pypy.module.exceptions.moduledef import Module as ExcModule
all_exceptions = list(ExcModule.interpleveldefs)
for exc_name in all_exceptions:
if exc_name in ('EnvironmentError', 'IOError', 'WindowsError'):
# FIXME: aliases of OSError cause a clash of names via
# export_struct
continue
register_global('PyExc_' + exc_name,
'PyTypeObject*',
'space.gettypeobject(interp_exceptions.W_%s.typedef)'% (exc_name, ))
# Common types with their own struct
for cpyname, pypyexpr in {
"PyType_Type": "space.w_type",
"PyBytes_Type": "space.w_bytes",
"PyUnicode_Type": "space.w_unicode",
"PyDict_Type": "space.w_dict",
"PyDictProxy_Type": 'space.gettypeobject(cpyext.dictproxyobject.W_DictProxyObject.typedef)',
"PyTuple_Type": "space.w_tuple",
"PyList_Type": "space.w_list",
"PySet_Type": "space.w_set",
"PyFrozenSet_Type": "space.w_frozenset",
"PyBool_Type": "space.w_bool",
"PyFloat_Type": "space.w_float",
"PyLong_Type": "space.w_int",
"PyComplex_Type": "space.w_complex",
"PyByteArray_Type": "space.w_bytearray",
"PyMemoryView_Type": "space.w_memoryview",
"PyBaseObject_Type": "space.w_object",
'_PyNone_Type': 'space.type(space.w_None)',
'_PyNotImplemented_Type': 'space.type(space.w_NotImplemented)',
'PyCell_Type': 'space.gettypeobject(Cell.typedef)',
'PyModule_Type': 'space.gettypeobject(Module.typedef)',
'PyProperty_Type': 'space.gettypeobject(W_Property.typedef)',
'PySlice_Type': 'space.gettypeobject(W_SliceObject.typedef)',
'PyStaticMethod_Type': 'space.gettypeobject(StaticMethod.typedef)',
'PyCFunction_Type': 'space.gettypeobject(cpyext.methodobject.W_PyCFunctionObject.typedef)',
'PyClassMethodDescr_Type': 'space.gettypeobject(cpyext.methodobject.W_PyCClassMethodObject.typedef)',
'PyGetSetDescr_Type': 'space.gettypeobject(cpyext.typeobject.W_GetSetPropertyEx.typedef)',
'PyMemberDescr_Type': 'space.gettypeobject(cpyext.typeobject.W_MemberDescr.typedef)',
'PyMethodDescr_Type': 'space.gettypeobject(cpyext.methodobject.W_PyCMethodObject.typedef)',
'PyWrapperDescr_Type': 'space.gettypeobject(cpyext.methodobject.W_PyCWrapperObject.typedef)',
'PyInstanceMethod_Type': 'space.gettypeobject(cpyext.classobject.InstanceMethod.typedef)',
'PyBufferable_Type': 'space.gettypeobject(W_Bufferable.typedef)',
}.items():
register_global(cpyname, 'PyTypeObject*', pypyexpr, header=pypy_decl)
for cpyname in '''PyMethodObject PyListObject PyLongObject
PyBaseExceptionObject'''.split():
FORWARD_DECLS.append('typedef struct { PyObject_HEAD } %s'
% (cpyname, ))
build_exported_objects()
class CpyextTypeSpace(CTypeSpace):
def decl(self, cdef, error=_NOT_SPECIFIED, header=DEFAULT_HEADER,
result_is_ll=False):
def decorate(func):
return api_func_from_cdef(
func, cdef, self, error=error, header=header,
result_is_ll=result_is_ll)
return decorate
CPYEXT_BASE_HEADERS = ['sys/types.h', 'stdarg.h', 'stdio.h',
'stddef.h', 'pyport.h']
# Subtle. There are two pyconfig.h, one in PC (for windows, includes a pragma
# to link python*.lib), one in include. The dirs in include_dir purposely avoid
# the one in PC, since at this stage python*.lib may not exist.
# copy_header_files() will use the PC one on windows, which will then be used
# for all translated c-extension compilation
cts = CpyextTypeSpace(headers=CPYEXT_BASE_HEADERS, include_dirs = [include_dir])
# Ideally, we would parse pyport.h but that is beyond the parser.
cts.parse_source("""
#ifdef _WIN64
typedef long long Py_ssize_t;
typedef long long Py_hash_t;
#else
typedef long Py_ssize_t;
typedef long Py_hash_t;
#endif
""", configure=False)
cts.parse_header(parse_dir / 'cpyext_object.h', configure=False)
cts.parse_header(parse_dir / 'cpyext_descrobject.h', configure=False)
cts.configure_types()
Py_ssize_t = cts.gettype('Py_ssize_t')
Py_ssize_tP = cts.gettype('Py_ssize_t *')
size_t = lltype.Unsigned
ADDR = lltype.Signed
# Note: as a special case, "PyObject" is the pointer type in RPython,
# corresponding to "PyObject *" in C. We do that only for PyObject.
# For example, "PyTypeObject" is the struct type even in RPython.
PyTypeObject = cts.gettype('PyTypeObject')
PyTypeObjectPtr = cts.gettype('PyTypeObject *')
PyObjectStruct = cts.gettype('PyObject')
PyObject = cts.gettype('PyObject *')
PyObjectFields = (("ob_refcnt", lltype.Signed),
("ob_pypy_link", lltype.Signed),
("ob_type", PyTypeObjectPtr))
PyVarObjectFields = PyObjectFields + (("ob_size", Py_ssize_t), )
PyVarObjectStruct = cts.gettype('PyVarObject')
PyVarObject = cts.gettype('PyVarObject *')
Py_buffer = cts.gettype('Py_buffer')
Py_bufferP = cts.gettype('Py_buffer *')
@specialize.memo()
def is_PyObject(TYPE):
if not isinstance(TYPE, lltype.Ptr):
return False
if TYPE == PyObject:
return True
assert not isinstance(TYPE.TO, lltype.ForwardReference)
return hasattr(TYPE.TO, 'c_ob_refcnt') and hasattr(TYPE.TO, 'c_ob_type')
# a pointer to PyObject
PyObjectP = rffi.CArrayPtr(PyObject)
# int *
INTP_real = rffi.CArrayPtr(rffi.INT_real)
def configure_types():
for config in (CConfig, CConfig2):
for name, TYPE in rffi_platform.configure(config).iteritems():
if name in TYPES:
TYPES[name].become(TYPE)
def build_type_checkers(type_name, cls=None):
"""
Builds two api functions: Py_XxxCheck() and Py_XxxCheckExact().
- if `cls` is None, the type is space.w_[type].
- if `cls` is a string, it is the name of a space attribute, e.g. 'w_str'.
- else `cls` must be a W_Class with a typedef.
"""
if cls is None:
attrname = "w_" + type_name.lower()
def get_w_type(space):
return getattr(space, attrname)
elif isinstance(cls, str):
def get_w_type(space):
return getattr(space, cls)
else:
def get_w_type(space):
return space.gettypeobject(cls.typedef)
check_name = "Py" + type_name + "_Check"
@cts.decl("int %s(void * obj)" % check_name, error=CANNOT_FAIL)
def check(space, w_obj):
"Implements the Py_Xxx_Check function"
w_obj_type = space.type(w_obj)
w_type = get_w_type(space)
return (space.is_w(w_obj_type, w_type) or
space.issubtype_w(w_obj_type, w_type))
@cts.decl("int %sExact(void * obj)" % check_name, error=CANNOT_FAIL)
def check_exact(space, w_obj):
"Implements the Py_Xxx_CheckExact function"
w_obj_type = space.type(w_obj)
w_type = get_w_type(space)
return space.is_w(w_obj_type, w_type)
return check, check_exact
def build_type_checkers_flags(type_name, cls=None, flagsubstr=None):
"""
Builds two api functions: Py_XxxCheck() and Py_XxxCheckExact()
Does not export the functions, assumes they are macros in the *. files
check will try a fast path via pto flags
"""
if cls is None:
attrname = "w_" + type_name.lower()
def get_w_type(space):
return getattr(space, attrname)
else:
def get_w_type(space):
return getattr(space, cls)
if flagsubstr is None:
tp_flag_str = 'Py_TPFLAGS_%s_SUBCLASS' % type_name.upper()
else:
tp_flag_str = 'Py_TPFLAGS_%s_SUBCLASS' % flagsubstr
check_name = "Py" + type_name + "_Check"
tp_flag = globals()[tp_flag_str]
@specialize.argtype(1)
def check(space, pto):
from pypy.module.cpyext.pyobject import is_pyobj, as_pyobj
"Implements the Py_Xxx_Check function"
if is_pyobj(pto):
return (widen(pto.c_ob_type.c_tp_flags) & tp_flag) == tp_flag
w_obj_type = space.type(pto)
w_type = get_w_type(space)
return (space.is_w(w_obj_type, w_type) or
space.issubtype_w(w_obj_type, w_type))
def check_exact(space, w_obj):
"Implements the Py_Xxx_CheckExact function"
w_obj_type = space.type(w_obj)
w_type = get_w_type(space)
return space.is_w(w_obj_type, w_type)
return check, check_exact
pypy_debug_catch_fatal_exception = rffi.llexternal('pypy_debug_catch_fatal_exception', [], lltype.Void)
# ____________________________________________________________
class WrapperCache(object):
def __init__(self, space):
self.space = space
self.wrapper_gens = {} # {signature: WrapperGen()}
class WrapperGen(object):
wrapper_second_level = None
A = lltype.Array(lltype.Char)
def __init__(self, space, signature):
self.space = space
self.signature = signature
def make_wrapper(self, callable):
if self.wrapper_second_level is None:
self.wrapper_second_level = make_wrapper_second_level(
self.space, *self.signature)
wrapper_second_level = self.wrapper_second_level
name = callable.__name__
pname = lltype.malloc(self.A, len(name), flavor='raw', immortal=True)
for i in range(len(name)):
pname[i] = name[i]
def wrapper(*args):
# no GC here, not even any GC object
return wrapper_second_level(callable, pname, *args)
wrapper.__name__ = "wrapper for %r" % (callable, )
return wrapper
@dont_inline
def _unpack_name(pname):
return ''.join([pname[i] for i in range(len(pname))])
@dont_inline
def deadlock_error(funcname):
funcname = _unpack_name(funcname)
fatalerror_notb("GIL deadlock detected when a CPython C extension "
"module calls '%s'" % (funcname,))
@dont_inline
def no_gil_error(funcname):
funcname = _unpack_name(funcname)
fatalerror_notb("GIL not held when a CPython C extension "
"module calls '%s'" % (funcname,))
@dont_inline
def not_supposed_to_fail(funcname):
funcname = _unpack_name(funcname)
print "Error in cpyext, CPython compatibility layer:"
print "The function", funcname, "was not supposed to fail"
raise SystemError
@dont_inline
def unexpected_exception(funcname, e, tb):
funcname = _unpack_name(funcname)
print 'Fatal error in cpyext, CPython compatibility layer, calling',funcname
print 'Either report a bug or consider not using this particular extension'
if not we_are_translated():
if tb is None:
tb = sys.exc_info()[2]
import traceback
traceback.print_exc()
if sys.stdout == sys.__stdout__:
import pdb; pdb.post_mortem(tb)
# we can't do much here, since we're in ctypes, swallow
else:
print str(e)
pypy_debug_catch_fatal_exception()
assert False
def _restore_gil_state(pygilstate_release, gilstate, gil_release, _gil_auto):
from rpython.rlib import rgil
# see "Handling of the GIL" above
if pygilstate_release:
from pypy.module.cpyext import pystate
unlock = (gilstate == pystate.PyGILState_UNLOCKED)
else:
unlock = gil_release or _gil_auto
if unlock:
rgil.release()
def make_wrapper_second_level(space, argtypesw, restype,
result_kind, error_value, gil):
from rpython.rlib import rgil
argtypes_enum_ui = unrolling_iterable(enumerate(argtypesw))
fatal_value = restype._defl()
gil_auto_workaround = (gil is None) # automatically detect when we don't
# have the GIL, and acquire/release it
gil_acquire = (gil == "acquire" or gil == "around")
gil_release = (gil == "release" or gil == "around")
pygilstate_ensure = (gil == "pygilstate_ensure")
pygilstate_release = (gil == "pygilstate_release")
pygilstate_check = (gil == "pygilstate_check")
assert (gil is None or gil_acquire or gil_release
or pygilstate_ensure or pygilstate_release or pygilstate_check)
expected_nb_args = len(argtypesw) + pygilstate_ensure
if isinstance(restype, lltype.Ptr) and error_value == 0:
error_value = lltype.nullptr(restype.TO)
if error_value is not CANNOT_FAIL:
assert lltype.typeOf(error_value) == lltype.typeOf(fatal_value)
def invalid(err):
"NOT_RPYTHON: translation-time crash if this ends up being called"
raise ValueError(err)
def wrapper_second_level(callable, pname, *args):
from pypy.module.cpyext.pyobject import make_ref, from_ref, is_pyobj
from pypy.module.cpyext.pyobject import as_pyobj
from pypy.module.cpyext import pystate
# we hope that malloc removal removes the newtuple() that is
# inserted exactly here by the varargs specializer
# see "Handling of the GIL" above (careful, we don't have the GIL here)
_gil_auto = False
if gil_auto_workaround and not rgil.am_I_holding_the_GIL():
_gil_auto = True
if _gil_auto or gil_acquire:
if gil_acquire and rgil.am_I_holding_the_GIL():
deadlock_error(pname)
rgil.acquire()
if gil_auto_workaround:
# while we're in workaround-land, detect when a regular PyXxx()
# function is invoked at .so load-time, e.g. by a C++ global
# variable with an initializer, and in this case make sure we
# initialize things.
space.fromcache(State).make_sure_cpyext_is_imported()
elif pygilstate_ensure:
if rgil.am_I_holding_the_GIL():
args += (pystate.PyGILState_LOCKED,)
else:
rgil.acquire()
args += (pystate.PyGILState_UNLOCKED,)
elif pygilstate_check:
result = rgil.am_I_holding_the_GIL()
return rffi.cast(restype, result)
else:
if not rgil.am_I_holding_the_GIL():
no_gil_error(pname)
if pygilstate_release:
gilstate = rffi.cast(lltype.Signed, args[-1])
else:
gilstate = pystate.PyGILState_IGNORE
llop.gc_stack_bottom(lltype.Void) # marker to enter RPython from C
retval = fatal_value
boxed_args = ()
tb = None
state = space.fromcache(State)
try:
if not we_are_translated() and DEBUG_WRAPPER:
print >>sys.stderr, callable,
assert len(args) == expected_nb_args
for i, (typ, is_wrapped) in argtypes_enum_ui:
arg = args[i]
if is_PyObject(typ) and is_wrapped:
assert is_pyobj(arg)
arg_conv = from_ref(space, rffi.cast(PyObject, arg))
elif typ == rffi.VOIDP and is_wrapped:
# Many macros accept a void* so that one can pass a
# PyObject* or a PySomeSubtype*.
arg_conv = from_ref(space, rffi.cast(PyObject, arg))
else:
arg_conv = arg
boxed_args += (arg_conv, )
if pygilstate_ensure:
boxed_args += (args[-1], )
try:
result = callable(space, *boxed_args)
if not we_are_translated() and DEBUG_WRAPPER:
print >>sys.stderr, " DONE"
except OperationError as e:
failed = True
state.set_exception(e)
except BaseException as e:
failed = True
if not we_are_translated():
tb = sys.exc_info()[2]
message = repr(e)
import traceback
traceback.print_exc()
else:
message = str(e)
state.set_exception(OperationError(space.w_SystemError,
space.newtext(message)))
except rstackovf.StackOverflow as e:
rstackovf.check_stack_overflow()
failed = True
state.set_exception(OperationError(space.w_RuntimeError,
space.newtext("maximum recursion depth exceeded")))
else:
failed = False
if failed:
if error_value is CANNOT_FAIL:
raise not_supposed_to_fail(pname)
retval = error_value
elif is_PyObject(restype):
if is_pyobj(result):
if result_kind != "L":
raise invalid("missing result_is_ll=True")
else:
if result_kind == "L":
raise invalid("result_is_ll=True but not ll PyObject")
if result_kind == "B": # borrowed
result = as_pyobj(space, result)
else:
result = make_ref(space, result)
retval = rffi.cast(restype, result)
elif restype is not lltype.Void:
retval = rffi.cast(restype, result)
except Exception as e:
unexpected_exception(pname, e, tb)
_restore_gil_state(pygilstate_release, gilstate, gil_release, _gil_auto)
state.check_and_raise_exception(always=True)
return fatal_value
assert lltype.typeOf(retval) == restype
_restore_gil_state(pygilstate_release, gilstate, gil_release, _gil_auto)
return retval
wrapper_second_level._dont_inline_ = True
return wrapper_second_level
def setup_init_functions(eci, prefix):
# jump through hoops to avoid releasing the GIL during initialization
# of the cpyext module. The C functions are called with no wrapper,
# but must not do anything like calling back PyType_Ready(). We
# use them just to get a pointer to the PyTypeObjects defined in C.
get_capsule_type = rffi.llexternal('_%s_get_capsule_type' % prefix,
[], PyTypeObjectPtr,
compilation_info=eci, _nowrapper=True)
setdefenc = rffi.llexternal('_%s_setfilesystemdefaultencoding' % prefix,
[rffi.CCHARP], lltype.Void,
compilation_info=eci, _nowrapper=True)
@init_function
def init_types(space):
from pypy.module.cpyext.typeobject import py_type_ready
from pypy.module.sys.interp_encoding import getfilesystemencoding
py_type_ready(space, get_capsule_type())
s = space.text_w(getfilesystemencoding(space))
setdefenc(rffi.str2charp(s, track_allocation=False)) # "leaks"
from pypy.module.posix.interp_posix import add_fork_hook
global py_fatalerror
py_fatalerror = rffi.llexternal('%s_FatalError' % prefix,
[CONST_STRING], lltype.Void,
compilation_info=eci)
_reinit_tls = rffi.llexternal('%sThread_ReInitTLS' % prefix, [],
lltype.Void, compilation_info=eci)
def reinit_tls(space):
_reinit_tls()
add_fork_hook('child', reinit_tls)
def attach_c_functions(space, eci, prefix):
state = space.fromcache(State)
state.C._Py_Dealloc = rffi.llexternal(
mangle_name(prefix, '_Py_Dealloc'),
[PyObject], lltype.Void,
compilation_info=eci,
_nowrapper=True)
state.C.PyObject_Free = rffi.llexternal(
mangle_name(prefix, 'PyObject_Free'),
[rffi.VOIDP], lltype.Void,
compilation_info=eci,
_nowrapper=True)
state.C.PyType_GenericAlloc = rffi.llexternal(
mangle_name(prefix, 'PyType_GenericAlloc'),
[PyTypeObjectPtr, Py_ssize_t], PyObject,
compilation_info=eci,
_nowrapper=True)
state.C._PyPy_int_dealloc = rffi.llexternal(
'_PyPy_int_dealloc', [PyObject], lltype.Void,
compilation_info=eci, _nowrapper=True)
state.C.PyTuple_New = rffi.llexternal(
mangle_name(prefix, 'PyTuple_New'),
[Py_ssize_t], PyObject,
compilation_info=eci,
_nowrapper=True)
state.C._PyPy_tuple_dealloc = rffi.llexternal(
'_PyPy_tuple_dealloc', [PyObject], lltype.Void,
compilation_info=eci, _nowrapper=True)
_, state.C.set_marker = rffi.CExternVariable(
rffi.VOIDP, '_pypy_rawrefcount_w_marker_deallocating',
eci, _nowrapper=True, c_type='void *')
state.C._PyPy_subtype_dealloc = rffi.llexternal(
mangle_name(prefix, '_Py_subtype_dealloc'),
[PyObject], lltype.Void,
compilation_info=eci, _nowrapper=True)
state.C._PyPy_object_dealloc = rffi.llexternal(
'_PyPy_object_dealloc', [PyObject], lltype.Void,
compilation_info=eci, _nowrapper=True)
FUNCPTR = lltype.Ptr(lltype.FuncType([], rffi.INT))
state.C.get_pyos_inputhook = rffi.llexternal(
'_PyPy_get_PyOS_InputHook', [], FUNCPTR,
compilation_info=eci, _nowrapper=True)
state.C.tuple_new = rffi.llexternal(
mangle_name(prefix, '_Py_tuple_new'),
[PyTypeObjectPtr, PyObject, PyObject], PyObject,
compilation_info=eci, _nowrapper=True)
if we_are_translated():
eci_flags = eci
else:
# To get this to work in tests, we need a new eci to
# link to the pypyapi.so/dll. Note that all this linking
# will only happen for tests, when translating the link args here
# are irrelevant.
library_dirs = eci.library_dirs
link_extra = list(eci.link_extra)
link_files = eci.link_files
if sys.platform == "win32":
# since we include Python.h, we must disable linking with
# the regular import lib
from pypy.module.sys import version
ver = version.CPYTHON_VERSION[:2]
link_extra.append("/NODEFAULTLIB:Python%d%d.lib" % ver)
# for testing, make sure "pypyapi.lib" is linked in
link_extra += [x.replace('dll', 'lib') for x in eci.libraries]
eci_flags = ExternalCompilationInfo(
include_dirs=include_dirs,
includes=['Python.h'],
link_extra = link_extra,
link_files = link_files,
library_dirs = library_dirs,
)
state.C.flag_setters = {}
for c_name, attr in _flags:
_, setter = rffi.CExternVariable(rffi.SIGNED, c_name, eci_flags,
_nowrapper=True, c_type='int')
state.C.flag_setters[attr] = setter
def init_function(func):
INIT_FUNCTIONS.append(func)
return func
def bootstrap_function(func):
BOOTSTRAP_FUNCTIONS.append(func)
return func
def run_bootstrap_functions(space):
for func in BOOTSTRAP_FUNCTIONS:
func(space)
@init_function
def init_flags(space):
state = space.fromcache(State)
for _, attr in _flags:
f = state.C.flag_setters[attr]
f(space.sys.get_flag(attr))
#_____________________________________________________
# Build the bridge DLL, Allow extension DLLs to call
# back into Pypy space functions
# Do not call this more than once per process
def build_bridge(space):
"NOT_RPYTHON"
from rpython.translator.c.database import LowLevelDatabase
use_micronumpy = setup_micronumpy(space)
db = LowLevelDatabase()
prefix = 'cpyexttest'
generate_decls_and_callbacks(db, prefix=prefix)
# Structure declaration code
functions = []
members = []
structindex = {}
for header, header_functions in FUNCTIONS_BY_HEADER.iteritems():
for name, func in header_functions.iteritems():
functions.append(func.get_ctypes_impl(name, db))
members.append(func.get_ptr_decl(name, db))
structindex[name] = len(structindex)
structmembers = '\n'.join(members)
struct_declaration_code = """\
struct PyPyAPI {
%(members)s
} _pypyAPI;
RPY_EXTERN struct PyPyAPI* pypyAPI;
struct PyPyAPI* pypyAPI = &_pypyAPI;
""" % dict(members=structmembers)
prologue = ("#include <Python.h>\n" +
"#include <structmember.h>\n" +
"#include <marshal.h>\n" +
("#include <pypy_numpy.h>\n" if use_micronumpy else "") +
"#include <src/thread.c>\n")
code = (prologue +
struct_declaration_code +
'\n' +
'\n'.join(functions))
eci = build_eci(code, use_micronumpy, translating=False)
eci = eci.compile_shared_lib(
outputfilename=str(udir / "module_cache" / "pypyapi"))
space.fromcache(State).install_dll(eci)
modulename = py.path.local(eci.libraries[-1])
attach_c_functions(space, eci, prefix)
run_bootstrap_functions(space)
# load the bridge, and init structure
bridge = ctypes.CDLL(str(modulename), mode=ctypes.RTLD_GLOBAL)
# populate static data
builder = space.fromcache(State).builder = TestingObjBuilder()
for name, (typ, expr) in GLOBALS.iteritems():
if '#' in name:
name, header = name.split('#')
assert typ in ('PyObject*', 'PyTypeObject*', 'PyIntObject*')
isptr = False
elif name.startswith('PyExc_'):
isptr = False
elif typ == 'PyDateTime_CAPI*':
isptr = True
else:
raise ValueError("Unknown static data: %s %s" % (typ, name))
from pypy.module import cpyext # for the eval() below
w_obj = eval(expr)
INTERPLEVEL_API[name] = w_obj
mname = mangle_name(prefix, name)
if isptr:
assert typ == 'PyDateTime_CAPI*'
value = w_obj
ptr = ctypes.c_void_p.in_dll(bridge, mname)
ptr.value = ctypes.cast(ll2ctypes.lltype2ctypes(value),
ctypes.c_void_p).value
elif typ in ('PyObject*', 'PyTypeObject*'):
if name.startswith('PyExc_'):
# we already have the pointer
in_dll = ll2ctypes.get_ctypes_type(PyObject).in_dll(bridge, mname)
py_obj = ll2ctypes.ctypes2lltype(PyObject, in_dll)
else:
# we have a structure, get its address
in_dll = ll2ctypes.get_ctypes_type(PyObject.TO).in_dll(bridge, mname)
py_obj = ll2ctypes.ctypes2lltype(PyObject, ctypes.pointer(in_dll))
builder.prepare(py_obj, w_obj)
pypyAPI = ctypes.POINTER(ctypes.c_void_p).in_dll(bridge, 'pypyAPI')
# implement structure initialization code
for header, header_functions in FUNCTIONS_BY_HEADER.iteritems():
for name, func in header_functions.iteritems():
pypyAPI[structindex[name]] = ctypes.cast(
ll2ctypes.lltype2ctypes(func.get_llhelper(space)),
ctypes.c_void_p)
# we need to call this *after* the init code above, because it might
# indirectly call some functions which are attached to pypyAPI (e.g., we
# if do tuple_attach of the prebuilt empty tuple, we need to call
# _PyPy_Malloc)
builder.attach_all(space)
setup_init_functions(eci, prefix)
return modulename.new(ext='')
def attach_recursively(space, static_pyobjs, static_objs_w, attached_objs, i):
# Start at i but make sure all the base classes are already attached
from pypy.module.cpyext.pyobject import get_typedescr, make_ref
if i in attached_objs:
return
py_obj = static_pyobjs[i]
w_obj = static_objs_w[i]
w_base = None
# w_obj can be NotImplemented, which is not a W_TypeObject
if isinstance(w_obj, W_TypeObject):
bases_w = w_obj.bases_w
if bases_w:
w_base = find_best_base(bases_w)
if w_base:
try:
j = static_objs_w.index(w_base)
except ValueError:
j = -1
if j >=0 and j not in attached_objs:
attach_recursively(space, static_pyobjs, static_objs_w,
attached_objs, j)
w_type = space.type(w_obj)
typedescr = get_typedescr(w_type.layout.typedef)
py_obj.c_ob_type = rffi.cast(PyTypeObjectPtr,
make_ref(space, w_type))
typedescr.attach(space, py_obj, w_obj)
attached_objs.append(i)
class StaticObjectBuilder(object):
def __init__(self):
self.static_pyobjs = []
self.static_objs_w = []
self.cpyext_type_init = None
#
# add a "method" that is overridden in setup_library()
# ('self.static_pyobjs' is completely ignored in that case)
self.get_static_pyobjs = lambda: self.static_pyobjs
def prepare(self, py_obj, w_obj):
"NOT_RPYTHON"
if py_obj:
py_obj.c_ob_refcnt = 1 # 1 for kept immortal
self.static_pyobjs.append(py_obj)
self.static_objs_w.append(w_obj)
def attach_all(self, space):
# this is RPython, called once in pypy-c when it imports cpyext
from pypy.module.cpyext.typeobject import finish_type_1, finish_type_2
from pypy.module.cpyext.pyobject import track_reference
#
static_pyobjs = self.get_static_pyobjs()
static_objs_w = self.static_objs_w
for i in range(len(static_objs_w)):
track_reference(space, static_pyobjs[i], static_objs_w[i])
#
self.cpyext_type_init = []
attached_objs = []
for i in range(len(static_objs_w)):
attach_recursively(space, static_pyobjs, static_objs_w, attached_objs, i)
cpyext_type_init = self.cpyext_type_init
self.cpyext_type_init = None
for pto, w_type in cpyext_type_init:
finish_type_1(space, pto)
finish_type_2(space, pto, w_type)
class TestingObjBuilder(StaticObjectBuilder):
"""The StaticObjectBuilder used in tests."""
class TranslationObjBuilder(StaticObjectBuilder):
"""The StaticObjectBuilder used during translation."""
def mangle_name(prefix, name):
if name.startswith('PyPyUnicode'):
# for PyPyUnicode_Check, PyPyUnicode_CheckExact
return name
elif name.startswith('Py'):
return prefix + name[2:]
elif name.startswith('_Py'):
return '_' + prefix + name[3:]
else:
raise ValueError("Error converting '%s'" % name)
def write_header(header_name, decls):
lines = [
'#include "cpyext_object.h"',
'''
#ifdef _WIN64
#define Signed Py_ssize_t /* xxx temporary fix */
#define Unsigned unsigned long long /* xxx temporary fix */
#else
#define Signed Py_ssize_t /* xxx temporary fix */
#define Unsigned unsigned long /* xxx temporary fix */
#endif
'''] + decls + [
'',
'#undef Signed /* xxx temporary fix */',
'#undef Unsigned /* xxx temporary fix */',
'']
decl_h = udir.join(header_name)
decl_h.write('\n'.join(lines))
def generate_decls_and_callbacks(db, prefix=''):
"NOT_RPYTHON"
pypy_macros = []
for name in SYMBOLS_C:
newname = mangle_name(prefix, name)
pypy_macros.append('#define %s %s' % (name, newname))
# Generate defines
for macro_name, size in [
("SIZEOF_LONG_LONG", rffi.LONGLONG),
("SIZEOF_VOID_P", rffi.VOIDP),
("SIZEOF_SIZE_T", rffi.SIZE_T),
("SIZEOF_TIME_T", rffi.TIME_T),
("SIZEOF_LONG", rffi.LONG),
("SIZEOF_SHORT", rffi.SHORT),
("SIZEOF_INT", rffi.INT),
("SIZEOF_FLOAT", rffi.FLOAT),
("SIZEOF_DOUBLE", rffi.DOUBLE),
]:
pypy_macros.append("#define %s %s" % (macro_name, rffi.sizeof(size)))
pypy_macros.append('')
pypy_macros_h = udir.join('pypy_macros.h')
pypy_macros_h.write('\n'.join(pypy_macros))
# generate function decls
decls = defaultdict(list)
for decl in FORWARD_DECLS:
decls[pypy_decl].append("%s;" % (decl,))
decls[pypy_decl].append("""
/* hack for https://bugs.python.org/issue29943 */
PyAPI_FUNC(int) %s(PyObject *arg0,
Signed arg1, Signed *arg2,
Signed *arg3, Signed *arg4, Signed *arg5);
#ifdef __GNUC__
__attribute__((__unused__))
#endif
static int PySlice_GetIndicesEx(PyObject *arg0, Py_ssize_t arg1,
Py_ssize_t *arg2, Py_ssize_t *arg3, Py_ssize_t *arg4,
Py_ssize_t *arg5) {
return %s(arg0, arg1, arg2, arg3,
arg4, arg5);
}""" % ((mangle_name(prefix, 'PySlice_GetIndicesEx'),)*2))
for header_name, header_functions in FUNCTIONS_BY_HEADER.iteritems():
header = decls[header_name]
for name, func in sorted(header_functions.iteritems()):
_name = mangle_name(prefix, name)
header.append("#define %s %s" % (name, _name))
header.append(func.get_api_decl(name, db))
for name, (typ, expr) in GLOBALS.iteritems():
if '#' in name:
name, header = name.split("#")
typ = typ.replace("*", "")
elif name.startswith('PyExc_'):
typ = 'PyObject*'
header = pypy_decl
decls[header].append('#define %s %s' % (name, mangle_name(prefix, name)))
decls[header].append('PyAPI_DATA(%s) %s;' % (typ, name))
for header_name, header_decls in decls.iteritems():
write_header(header_name, header_decls)
# generate graminit.h
graminit_h = udir.join('graminit.h')
graminit_h.write('/* Generated from pypy.interpreter.pyparser.pygram.syms */')
for attr in dir(pygram.syms):
val = getattr(pygram.syms, attr)
graminit_h.write('#define {} {}'.format(attr, val))
separate_module_files = [source_dir / "varargwrapper.c",
source_dir / "pyerrors.c",
source_dir / "modsupport.c",
source_dir / "getargs.c",
source_dir / "abstract.c",
source_dir / "unicodeobject.c",
source_dir / "mysnprintf.c",
source_dir / "pythonrun.c",
source_dir / "sysmodule.c",
source_dir / "complexobject.c",
source_dir / "structseq.c",
source_dir / "capsule.c",
source_dir / "pysignals.c",
source_dir / "pythread.c",
source_dir / "missing.c",
source_dir / "pymem.c",
source_dir / "pytime.c",
source_dir / "bytesobject.c",
source_dir / "import.c",
source_dir / "_warnings.c",
source_dir / "pylifecycle.c",
source_dir / "object.c",
source_dir / "typeobject.c",
source_dir / "tupleobject.c",
source_dir / "sliceobject.c",
source_dir / "call.c",
]
if WIN32:
separate_module_files.append(source_dir / "pythread_nt.c")
else:
separate_module_files.append(source_dir / "pythread_posix.c")
def build_eci(code, use_micronumpy=False, translating=False):
"NOT_RPYTHON"
# Build code and get pointer to the structure
kwds = {}
compile_extra=['-DPy_BUILD_CORE']
if translating:
kwds["includes"] = ['Python.h'] # this is our Python.h
else:
if sys.platform == "win32":
# '%s' undefined; assuming extern returning int
compile_extra.append("/we4013")
# Sometimes the library is wrapped into another DLL, ensure that
# the correct bootstrap code is installed.
kwds["link_extra"] = ["msvcrt.lib"]
elif sys.platform.startswith('linux'):
compile_extra.append("-Werror=implicit-function-declaration")
compile_extra.append('-g')
compile_extra.append(
'-DCPYEXT_TESTS')
# Generate definitions for global structures
structs = ["#include <Python.h>"]
if use_micronumpy:
structs.append('#include <pypy_numpy.h> /* api.py line 1223 */')
for name, (typ, expr) in GLOBALS.iteritems():
if '#' in name:
structs.append('%s %s;' % (typ[:-1], name.split('#')[0]))
elif name.startswith('PyExc_'):
structs.append('PyTypeObject _%s;' % (name,))
structs.append('PyObject* %s = (PyObject*)&_%s;' % (name, name))
elif typ == 'PyDateTime_CAPI*':
structs.append('%s %s = NULL;' % (typ, name))
struct_source = '\n'.join(structs)
separate_module_sources = [code, struct_source]
if sys.platform == 'win32':
get_pythonapi_source = '''
RPY_EXTERN
HANDLE pypy_get_pythonapi_handle() {
MEMORY_BASIC_INFORMATION mi;
memset(&mi, 0, sizeof(mi));
if( !VirtualQueryEx(GetCurrentProcess(), &pypy_get_pythonapi_handle,
&mi, sizeof(mi)) )
return 0;
return (HMODULE)mi.AllocationBase;
}
'''
separate_module_sources.append(get_pythonapi_source)
kwds['post_include_bits'] = ['#include <windows.h>',
'RPY_EXTERN HANDLE pypy_get_pythonapi_handle();',
]
eci = ExternalCompilationInfo(
include_dirs=include_dirs,
separate_module_files= separate_module_files,
separate_module_sources=separate_module_sources,
compile_extra=compile_extra,
**kwds
)
return eci
def setup_micronumpy(space):
# py3k
return False
use_micronumpy = space.config.objspace.usemodules.micronumpy
if not use_micronumpy:
return use_micronumpy
# import registers api functions by side-effect, we also need HEADER
from pypy.module.cpyext.ndarrayobject import HEADER
register_global("PyArray_Type",
'PyTypeObject*', "space.gettypeobject(W_NDimArray.typedef)",
header=HEADER)
separate_module_files.append(source_dir / "ndarrayobject.c")
return use_micronumpy
def setup_library(space):
"NOT_RPYTHON"
from rpython.translator.c.database import LowLevelDatabase
use_micronumpy = setup_micronumpy(space)
db = LowLevelDatabase()
prefix = 'PyPy'
generate_decls_and_callbacks(db, prefix=prefix)
code = "#include <Python.h>\n"
if use_micronumpy:
code += "#include <pypy_numpy.h> /* api.py line 1290 */\n"
eci = build_eci(code, use_micronumpy, translating=True)
space.fromcache(State).install_dll(eci)
attach_c_functions(space, eci, prefix)
run_bootstrap_functions(space)
# emit uninitialized static data
builder = space.fromcache(State).builder = TranslationObjBuilder()
lines = ['PyObject *pypy_static_pyobjs[] = {\n']
include_lines = ['RPY_EXTERN PyObject *pypy_static_pyobjs[];\n']
for name, (typ, expr) in sorted(GLOBALS.items()):
if '#' in name:
name, header = name.split('#')
assert typ in ('PyObject*', 'PyTypeObject*')
typ = typ[:-1]
mname = mangle_name(prefix, name)
include_lines.append('#define %s %s\n' % (name, mname))
elif name.startswith('PyExc_'):
typ = 'PyTypeObject'
name = '_' + name
elif typ == 'PyDateTime_CAPI*':
continue
else:
raise ValueError("Unknown static data: %s %s" % (typ, name))
from pypy.module import cpyext # for the eval() below
w_obj = eval(expr)
builder.prepare(None, w_obj)
lines.append('\t(PyObject *)&%s,\n' % (name,))
include_lines.append('RPY_EXPORTED %s %s;\n' % (typ, name))
lines.append('};\n')
eci2 = configure_eci.merge(ExternalCompilationInfo(
separate_module_sources = [''.join(lines)],
post_include_bits = [''.join(include_lines)],
))
# override this method to return a pointer to this C array directly
builder.get_static_pyobjs = rffi.CExternVariable(
PyObjectP, 'pypy_static_pyobjs', eci2, c_type='PyObject **',
getter_only=True, declare_as_extern=False)
for header, header_functions in FUNCTIONS_BY_HEADER.iteritems():
for name, func in header_functions.iteritems():
newname = mangle_name(prefix, name)
deco = entrypoint_lowlevel("cpyext", func.argtypes, newname,
relax=True)
deco(func.get_wrapper(space))
setup_init_functions(eci, prefix)
if sys.platform == "win32":
trunk_include = pypydir.dirpath() / 'include'
else:
from pypy.module.sys import version
ver = version.CPYTHON_VERSION[:2]
trunk_include = pypydir.dirpath() / 'include' / 'pypy{}.{}'.format(*ver)
trunk_include.ensure(dir=True)
copy_header_files(cts, trunk_include, use_micronumpy)
def create_extension_module(space, w_spec):
# note: this is used both to load CPython-API-style C extension
# modules (cpyext) and to load CFFI-style extension modules
# (_cffi_backend). Any of the two can be disabled at translation
# time, though. For this reason, we need to be careful about the
# order of things here.
from rpython.rlib import rdynload
w_name = space.getattr(w_spec, space.newtext("name"))
w_path = space.getattr(w_spec, space.newtext("origin"))
name = space.text_w(w_name)
path = space.text_w(w_path)
if os.sep not in path:
path = os.curdir + os.sep + path # force a '/' in the path
try:
# XXX does this need a fsdecoder for utf8 paths?
ll_libname = rffi.str2charp(path)
try:
if WIN32:
from rpython.rlib import rwin32
# Allow other DLLs in the same directory
# use os.add_dll_directory for more locations
flags = (rwin32.LOAD_LIBRARY_SEARCH_DEFAULT_DIRS |
rwin32.LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR)
dll = rdynload.dlopenex(ll_libname, flags)
else:
dll = rdynload.dlopen(ll_libname, space.sys.dlopenflags)
finally:
lltype.free(ll_libname, flavor='raw')
except rdynload.DLOpenError as e:
raise raise_import_error(space,
space.newfilename(e.msg), w_name, w_path)
look_for = None
#
if space.config.objspace.usemodules._cffi_backend:
basename = name.split('.')[-1]
look_for = '_cffi_pypyinit_%s' % (basename,)
try:
initptr = rdynload.dlsym(dll, look_for)
except KeyError:
pass
else:
try:
from pypy.module._cffi_backend import cffi1_module
return cffi1_module.load_cffi1_module(space, name, path, initptr)
except:
rdynload.dlclose(dll)
raise
#
if space.config.objspace.usemodules.cpyext:
also_look_for = get_init_name(space, w_name)
try:
initptr = rdynload.dlsym(dll, also_look_for)
except KeyError:
pass
else:
return create_cpyext_module(space, w_spec, name, path, dll, initptr)
if look_for is not None:
look_for += ' or ' + also_look_for
else:
look_for = also_look_for
assert look_for is not None
msg = b"function %s not found in library %s" % (
look_for, space.utf8_w(space.newfilename(path)))
w_path = space.newfilename(path)
raise_import_error(space, space.newtext(msg), w_name, w_path)
def get_init_name(space, w_name):
name = space.utf8_w(w_name)
basename = name.split('.')[-1]
if rutf8.first_non_ascii_char(basename) == -1:
return 'PyInit_%s' % (basename,)
basename = space.bytes_w(encode_object(
space, space.newtext(basename), 'punycode', None))
basename = basename.replace('-', '_')
return 'PyInitU_%s' % (basename,)
initfunctype = lltype.Ptr(lltype.FuncType([], PyObject))
def create_cpyext_module(space, w_spec, name, path, dll, initptr):
from rpython.rlib import rdynload
from pypy.module.cpyext.pyobject import get_w_obj_and_decref
state = space.fromcache(State)
state.make_sure_cpyext_is_imported()
w_mod = state.find_extension(name, path)
if w_mod is not None:
rdynload.dlclose(dll)
return w_mod
old_context = state.package_context
state.package_context = name, path
try:
initfunc = rffi.cast(initfunctype, initptr)
initret = generic_cpy_call_dont_convert_result(space, initfunc)
if not initret:
state.check_and_raise_exception()
raise oefmt(space.w_SystemError,
"initialization of %s failed without raising an exception",
name)
else:
if state.clear_exception():
raise oefmt(space.w_SystemError,
"initialization of %s raised unreported exception",
name)
if not initret.c_ob_type:
raise oefmt(space.w_SystemError,
"init function of %s returned uninitialized object",
name)
# This should probably compare by identity with PyModuleDef_Type from
# modsupport.c, but I didn't find a way to do that.
tp_name_nonconst = rffi.cast(rffi.CCHARP, initret.c_ob_type.c_tp_name)
if rffi.charp2str(tp_name_nonconst) == "moduledef":
from pypy.module.cpyext.modsupport import \
create_module_from_def_and_spec
return create_module_from_def_and_spec(space, initret, w_spec,
name)
finally:
state.package_context = old_context
# XXX: should disable single-step init for non-ascii module names
w_mod = get_w_obj_and_decref(space, initret)
state.fixup_extension(w_mod, name, path)
return w_mod
@jit.dont_look_inside
def exec_extension_module(space, w_mod):
from pypy.module.cpyext.modsupport import exec_def, PyModuleObject
if not space.config.objspace.usemodules.cpyext:
return
if not isinstance(w_mod, Module):
return
space.getbuiltinmodule("cpyext")
mod = cts.cast('PyModuleObject*', rawrefcount.from_obj(PyObject, w_mod))
if mod:
if mod.c_md_state:
# already initialised
return
moddef = mod.c_md_def
return exec_def(space, mod, moddef)
def invoke_pyos_inputhook(space):
state = space.fromcache(State)
c_inputhook = state.C.get_pyos_inputhook()
if c_inputhook:
generic_cpy_call(space, c_inputhook)
@specialize.ll()
def generic_cpy_call(space, func, *args):
FT = lltype.typeOf(func).TO
return make_generic_cpy_call(FT, False, True)(space, func, *args)
@specialize.ll()
def generic_cpy_call_expect_null(space, func, *args):
FT = lltype.typeOf(func).TO
return make_generic_cpy_call(FT, True, True)(space, func, *args)
@specialize.ll()
def generic_cpy_call_dont_convert_result(space, func, *args):
FT = lltype.typeOf(func).TO
return make_generic_cpy_call(FT, False, False)(space, func, *args)
@specialize.memo()
def make_generic_cpy_call(FT, expect_null, convert_result):
from pypy.module.cpyext.pyobject import is_pyobj, make_ref, decref
from pypy.module.cpyext.pyobject import get_w_obj_and_decref
from pypy.module.cpyext.pyerrors import PyErr_Occurred
unrolling_arg_types = unrolling_iterable(enumerate(FT.ARGS))
RESULT_TYPE = FT.RESULT
# copied and modified from rffi.py
# We need tons of care to ensure that no GC operation and no
# exception checking occurs in call_external_function.
argnames = ', '.join(['a%d' % i for i in range(len(FT.ARGS))])
source = py.code.Source("""
def cpy_call_external(funcptr, %(argnames)s):
# NB. it is essential that no exception checking occurs here!
res = funcptr(%(argnames)s)
return res
""" % locals())
miniglobals = {'__name__': __name__, # for module name propagation
}
exec source.compile() in miniglobals
call_external_function = specialize.ll()(miniglobals['cpy_call_external'])
call_external_function._dont_inline_ = True
call_external_function._gctransformer_hint_close_stack_ = True
# don't inline, as a hack to guarantee that no GC pointer is alive
# anywhere in call_external_function
@specialize.ll()
def generic_cpy_call(space, func, *args):
boxed_args = ()
to_decref = ()
assert len(args) == len(FT.ARGS)
for i, ARG in unrolling_arg_types:
arg = args[i]
_pyobj = None
if is_PyObject(ARG):
if not is_pyobj(arg):
arg = make_ref(space, arg)
_pyobj = arg
boxed_args += (arg,)
to_decref += (_pyobj,)
if is_PyObject(RESULT_TYPE):
preexist_error = PyErr_Occurred(space)
else:
preexist_error = "this is not used"
try:
# Call the function
result = call_external_function(func, *boxed_args)
finally:
for i, ARG in unrolling_arg_types:
# note that this loop is nicely unrolled statically by RPython
_pyobj = to_decref[i]
if _pyobj is not None:
decref(space, _pyobj)
if convert_result and is_PyObject(RESULT_TYPE):
if not is_pyobj(result):
ret = result
else:
# The object reference returned from a C function
# that is called from Python must be an owned reference
# - ownership is transferred from the function to its caller.
if result:
ret = get_w_obj_and_decref(space, result)
else:
ret = None
# Check for exception consistency
# XXX best attempt, will miss preexisting error that is
# overwritten with a new error of the same type
error = PyErr_Occurred(space)
has_new_error = (error is not None) and (error is not preexist_error)
has_result = ret is not None
if not expect_null and has_new_error and has_result:
raise oefmt(space.w_SystemError,
"An exception was set, but function returned a "
"value")
elif not expect_null and not has_new_error and not has_result:
raise oefmt(space.w_SystemError,
"Function returned a NULL result without setting "
"an exception")
elif has_new_error:
state = space.fromcache(State)
state.check_and_raise_exception()
return ret
return result
return generic_cpy_call
|