123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930 |
- /* Module definition and import implementation */
- #include "Python.h"
- #include "pycore_hashtable.h" // _Py_hashtable_new_full()
- #include "pycore_import.h" // _PyImport_BootstrapImp()
- #include "pycore_initconfig.h" // _PyStatus_OK()
- #include "pycore_interp.h" // struct _import_runtime_state
- #include "pycore_namespace.h" // _PyNamespace_Type
- #include "pycore_object.h" // _Py_SetImmortal()
- #include "pycore_pyerrors.h" // _PyErr_SetString()
- #include "pycore_pyhash.h" // _Py_KeyedHash()
- #include "pycore_pylifecycle.h"
- #include "pycore_pymem.h" // _PyMem_SetDefaultAllocator()
- #include "pycore_pystate.h" // _PyInterpreterState_GET()
- #include "pycore_sysmodule.h" // _PySys_Audit()
- #include "marshal.h" // PyMarshal_ReadObjectFromString()
- #include "importdl.h" // _PyImport_DynLoadFiletab
- #include "pydtrace.h" // PyDTrace_IMPORT_FIND_LOAD_START_ENABLED()
- #include <stdbool.h> // bool
- #ifdef HAVE_FCNTL_H
- #include <fcntl.h>
- #endif
- #ifdef __cplusplus
- extern "C" {
- #endif
- /*[clinic input]
- module _imp
- [clinic start generated code]*/
- /*[clinic end generated code: output=da39a3ee5e6b4b0d input=9c332475d8686284]*/
- #include "clinic/import.c.h"
- /*******************************/
- /* process-global import state */
- /*******************************/
- /* This table is defined in config.c: */
- extern struct _inittab _PyImport_Inittab[];
- // This is not used after Py_Initialize() is called.
- // (See _PyRuntimeState.imports.inittab.)
- struct _inittab *PyImport_Inittab = _PyImport_Inittab;
- // When we dynamically allocate a larger table for PyImport_ExtendInittab(),
- // we track the pointer here so we can deallocate it during finalization.
- static struct _inittab *inittab_copy = NULL;
- /*******************************/
- /* runtime-global import state */
- /*******************************/
- #define INITTAB _PyRuntime.imports.inittab
- #define LAST_MODULE_INDEX _PyRuntime.imports.last_module_index
- #define EXTENSIONS _PyRuntime.imports.extensions
- #define PKGCONTEXT (_PyRuntime.imports.pkgcontext)
- /*******************************/
- /* interpreter import state */
- /*******************************/
- #define MODULES(interp) \
- (interp)->imports.modules
- #define MODULES_BY_INDEX(interp) \
- (interp)->imports.modules_by_index
- #define IMPORTLIB(interp) \
- (interp)->imports.importlib
- #define OVERRIDE_MULTI_INTERP_EXTENSIONS_CHECK(interp) \
- (interp)->imports.override_multi_interp_extensions_check
- #define OVERRIDE_FROZEN_MODULES(interp) \
- (interp)->imports.override_frozen_modules
- #ifdef HAVE_DLOPEN
- # define DLOPENFLAGS(interp) \
- (interp)->imports.dlopenflags
- #endif
- #define IMPORT_FUNC(interp) \
- (interp)->imports.import_func
- #define IMPORT_LOCK(interp) \
- (interp)->imports.lock.mutex
- #define IMPORT_LOCK_THREAD(interp) \
- (interp)->imports.lock.thread
- #define IMPORT_LOCK_LEVEL(interp) \
- (interp)->imports.lock.level
- #define FIND_AND_LOAD(interp) \
- (interp)->imports.find_and_load
- /*******************/
- /* the import lock */
- /*******************/
- /* Locking primitives to prevent parallel imports of the same module
- in different threads to return with a partially loaded module.
- These calls are serialized by the global interpreter lock. */
- void
- _PyImport_AcquireLock(PyInterpreterState *interp)
- {
- unsigned long me = PyThread_get_thread_ident();
- if (me == PYTHREAD_INVALID_THREAD_ID)
- return; /* Too bad */
- if (IMPORT_LOCK(interp) == NULL) {
- IMPORT_LOCK(interp) = PyThread_allocate_lock();
- if (IMPORT_LOCK(interp) == NULL)
- return; /* Nothing much we can do. */
- }
- if (IMPORT_LOCK_THREAD(interp) == me) {
- IMPORT_LOCK_LEVEL(interp)++;
- return;
- }
- if (IMPORT_LOCK_THREAD(interp) != PYTHREAD_INVALID_THREAD_ID ||
- !PyThread_acquire_lock(IMPORT_LOCK(interp), 0))
- {
- PyThreadState *tstate = PyEval_SaveThread();
- PyThread_acquire_lock(IMPORT_LOCK(interp), WAIT_LOCK);
- PyEval_RestoreThread(tstate);
- }
- assert(IMPORT_LOCK_LEVEL(interp) == 0);
- IMPORT_LOCK_THREAD(interp) = me;
- IMPORT_LOCK_LEVEL(interp) = 1;
- }
- int
- _PyImport_ReleaseLock(PyInterpreterState *interp)
- {
- unsigned long me = PyThread_get_thread_ident();
- if (me == PYTHREAD_INVALID_THREAD_ID || IMPORT_LOCK(interp) == NULL)
- return 0; /* Too bad */
- if (IMPORT_LOCK_THREAD(interp) != me)
- return -1;
- IMPORT_LOCK_LEVEL(interp)--;
- assert(IMPORT_LOCK_LEVEL(interp) >= 0);
- if (IMPORT_LOCK_LEVEL(interp) == 0) {
- IMPORT_LOCK_THREAD(interp) = PYTHREAD_INVALID_THREAD_ID;
- PyThread_release_lock(IMPORT_LOCK(interp));
- }
- return 1;
- }
- #ifdef HAVE_FORK
- /* This function is called from PyOS_AfterFork_Child() to ensure that newly
- created child processes do not share locks with the parent.
- We now acquire the import lock around fork() calls but on some platforms
- (Solaris 9 and earlier? see isue7242) that still left us with problems. */
- PyStatus
- _PyImport_ReInitLock(PyInterpreterState *interp)
- {
- if (IMPORT_LOCK(interp) != NULL) {
- if (_PyThread_at_fork_reinit(&IMPORT_LOCK(interp)) < 0) {
- return _PyStatus_ERR("failed to create a new lock");
- }
- }
- if (IMPORT_LOCK_LEVEL(interp) > 1) {
- /* Forked as a side effect of import */
- unsigned long me = PyThread_get_thread_ident();
- PyThread_acquire_lock(IMPORT_LOCK(interp), WAIT_LOCK);
- IMPORT_LOCK_THREAD(interp) = me;
- IMPORT_LOCK_LEVEL(interp)--;
- } else {
- IMPORT_LOCK_THREAD(interp) = PYTHREAD_INVALID_THREAD_ID;
- IMPORT_LOCK_LEVEL(interp) = 0;
- }
- return _PyStatus_OK();
- }
- #endif
- /***************/
- /* sys.modules */
- /***************/
- PyObject *
- _PyImport_InitModules(PyInterpreterState *interp)
- {
- assert(MODULES(interp) == NULL);
- MODULES(interp) = PyDict_New();
- if (MODULES(interp) == NULL) {
- return NULL;
- }
- return MODULES(interp);
- }
- PyObject *
- _PyImport_GetModules(PyInterpreterState *interp)
- {
- return MODULES(interp);
- }
- void
- _PyImport_ClearModules(PyInterpreterState *interp)
- {
- Py_SETREF(MODULES(interp), NULL);
- }
- PyObject *
- PyImport_GetModuleDict(void)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (MODULES(interp) == NULL) {
- Py_FatalError("interpreter has no modules dictionary");
- }
- return MODULES(interp);
- }
- // This is only kept around for extensions that use _Py_IDENTIFIER.
- PyObject *
- _PyImport_GetModuleId(_Py_Identifier *nameid)
- {
- PyObject *name = _PyUnicode_FromId(nameid); /* borrowed */
- if (name == NULL) {
- return NULL;
- }
- return PyImport_GetModule(name);
- }
- int
- _PyImport_SetModule(PyObject *name, PyObject *m)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- PyObject *modules = MODULES(interp);
- return PyObject_SetItem(modules, name, m);
- }
- int
- _PyImport_SetModuleString(const char *name, PyObject *m)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- PyObject *modules = MODULES(interp);
- return PyMapping_SetItemString(modules, name, m);
- }
- static PyObject *
- import_get_module(PyThreadState *tstate, PyObject *name)
- {
- PyObject *modules = MODULES(tstate->interp);
- if (modules == NULL) {
- _PyErr_SetString(tstate, PyExc_RuntimeError,
- "unable to get sys.modules");
- return NULL;
- }
- PyObject *m;
- Py_INCREF(modules);
- if (PyDict_CheckExact(modules)) {
- m = PyDict_GetItemWithError(modules, name); /* borrowed */
- Py_XINCREF(m);
- }
- else {
- m = PyObject_GetItem(modules, name);
- if (m == NULL && _PyErr_ExceptionMatches(tstate, PyExc_KeyError)) {
- _PyErr_Clear(tstate);
- }
- }
- Py_DECREF(modules);
- return m;
- }
- static int
- import_ensure_initialized(PyInterpreterState *interp, PyObject *mod, PyObject *name)
- {
- PyObject *spec;
- /* Optimization: only call _bootstrap._lock_unlock_module() if
- __spec__._initializing is true.
- NOTE: because of this, initializing must be set *before*
- stuffing the new module in sys.modules.
- */
- spec = PyObject_GetAttr(mod, &_Py_ID(__spec__));
- int busy = _PyModuleSpec_IsInitializing(spec);
- Py_XDECREF(spec);
- if (busy) {
- /* Wait until module is done importing. */
- PyObject *value = _PyObject_CallMethodOneArg(
- IMPORTLIB(interp), &_Py_ID(_lock_unlock_module), name);
- if (value == NULL) {
- return -1;
- }
- Py_DECREF(value);
- }
- return 0;
- }
- static void remove_importlib_frames(PyThreadState *tstate);
- PyObject *
- PyImport_GetModule(PyObject *name)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *mod;
- mod = import_get_module(tstate, name);
- if (mod != NULL && mod != Py_None) {
- if (import_ensure_initialized(tstate->interp, mod, name) < 0) {
- Py_DECREF(mod);
- remove_importlib_frames(tstate);
- return NULL;
- }
- }
- return mod;
- }
- /* Get the module object corresponding to a module name.
- First check the modules dictionary if there's one there,
- if not, create a new one and insert it in the modules dictionary. */
- static PyObject *
- import_add_module(PyThreadState *tstate, PyObject *name)
- {
- PyObject *modules = MODULES(tstate->interp);
- if (modules == NULL) {
- _PyErr_SetString(tstate, PyExc_RuntimeError,
- "no import module dictionary");
- return NULL;
- }
- PyObject *m;
- if (PyDict_CheckExact(modules)) {
- m = Py_XNewRef(PyDict_GetItemWithError(modules, name));
- }
- else {
- m = PyObject_GetItem(modules, name);
- // For backward-compatibility we copy the behavior
- // of PyDict_GetItemWithError().
- if (_PyErr_ExceptionMatches(tstate, PyExc_KeyError)) {
- _PyErr_Clear(tstate);
- }
- }
- if (_PyErr_Occurred(tstate)) {
- return NULL;
- }
- if (m != NULL && PyModule_Check(m)) {
- return m;
- }
- Py_XDECREF(m);
- m = PyModule_NewObject(name);
- if (m == NULL)
- return NULL;
- if (PyObject_SetItem(modules, name, m) != 0) {
- Py_DECREF(m);
- return NULL;
- }
- return m;
- }
- PyObject *
- PyImport_AddModuleObject(PyObject *name)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *mod = import_add_module(tstate, name);
- if (mod) {
- PyObject *ref = PyWeakref_NewRef(mod, NULL);
- Py_DECREF(mod);
- if (ref == NULL) {
- return NULL;
- }
- mod = PyWeakref_GetObject(ref);
- Py_DECREF(ref);
- }
- return mod; /* borrowed reference */
- }
- PyObject *
- PyImport_AddModule(const char *name)
- {
- PyObject *nameobj = PyUnicode_FromString(name);
- if (nameobj == NULL) {
- return NULL;
- }
- PyObject *module = PyImport_AddModuleObject(nameobj);
- Py_DECREF(nameobj);
- return module;
- }
- /* Remove name from sys.modules, if it's there.
- * Can be called with an exception raised.
- * If fail to remove name a new exception will be chained with the old
- * exception, otherwise the old exception is preserved.
- */
- static void
- remove_module(PyThreadState *tstate, PyObject *name)
- {
- PyObject *exc = _PyErr_GetRaisedException(tstate);
- PyObject *modules = MODULES(tstate->interp);
- if (PyDict_CheckExact(modules)) {
- PyObject *mod = _PyDict_Pop(modules, name, Py_None);
- Py_XDECREF(mod);
- }
- else if (PyMapping_DelItem(modules, name) < 0) {
- if (_PyErr_ExceptionMatches(tstate, PyExc_KeyError)) {
- _PyErr_Clear(tstate);
- }
- }
- _PyErr_ChainExceptions1(exc);
- }
- /************************************/
- /* per-interpreter modules-by-index */
- /************************************/
- Py_ssize_t
- _PyImport_GetNextModuleIndex(void)
- {
- PyThread_acquire_lock(EXTENSIONS.mutex, WAIT_LOCK);
- LAST_MODULE_INDEX++;
- Py_ssize_t index = LAST_MODULE_INDEX;
- PyThread_release_lock(EXTENSIONS.mutex);
- return index;
- }
- static const char *
- _modules_by_index_check(PyInterpreterState *interp, Py_ssize_t index)
- {
- if (index == 0) {
- return "invalid module index";
- }
- if (MODULES_BY_INDEX(interp) == NULL) {
- return "Interpreters module-list not accessible.";
- }
- if (index > PyList_GET_SIZE(MODULES_BY_INDEX(interp))) {
- return "Module index out of bounds.";
- }
- return NULL;
- }
- static PyObject *
- _modules_by_index_get(PyInterpreterState *interp, PyModuleDef *def)
- {
- Py_ssize_t index = def->m_base.m_index;
- if (_modules_by_index_check(interp, index) != NULL) {
- return NULL;
- }
- PyObject *res = PyList_GET_ITEM(MODULES_BY_INDEX(interp), index);
- return res==Py_None ? NULL : res;
- }
- static int
- _modules_by_index_set(PyInterpreterState *interp,
- PyModuleDef *def, PyObject *module)
- {
- assert(def != NULL);
- assert(def->m_slots == NULL);
- assert(def->m_base.m_index > 0);
- if (MODULES_BY_INDEX(interp) == NULL) {
- MODULES_BY_INDEX(interp) = PyList_New(0);
- if (MODULES_BY_INDEX(interp) == NULL) {
- return -1;
- }
- }
- Py_ssize_t index = def->m_base.m_index;
- while (PyList_GET_SIZE(MODULES_BY_INDEX(interp)) <= index) {
- if (PyList_Append(MODULES_BY_INDEX(interp), Py_None) < 0) {
- return -1;
- }
- }
- return PyList_SetItem(MODULES_BY_INDEX(interp), index, Py_NewRef(module));
- }
- static int
- _modules_by_index_clear_one(PyInterpreterState *interp, PyModuleDef *def)
- {
- Py_ssize_t index = def->m_base.m_index;
- const char *err = _modules_by_index_check(interp, index);
- if (err != NULL) {
- Py_FatalError(err);
- return -1;
- }
- return PyList_SetItem(MODULES_BY_INDEX(interp), index, Py_NewRef(Py_None));
- }
- PyObject*
- PyState_FindModule(PyModuleDef* module)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (module->m_slots) {
- return NULL;
- }
- return _modules_by_index_get(interp, module);
- }
- /* _PyState_AddModule() has been completely removed from the C-API
- (and was removed from the limited API in 3.6). However, we're
- playing it safe and keeping it around for any stable ABI extensions
- built against 3.2-3.5. */
- int
- _PyState_AddModule(PyThreadState *tstate, PyObject* module, PyModuleDef* def)
- {
- if (!def) {
- assert(_PyErr_Occurred(tstate));
- return -1;
- }
- if (def->m_slots) {
- _PyErr_SetString(tstate,
- PyExc_SystemError,
- "PyState_AddModule called on module with slots");
- return -1;
- }
- return _modules_by_index_set(tstate->interp, def, module);
- }
- int
- PyState_AddModule(PyObject* module, PyModuleDef* def)
- {
- if (!def) {
- Py_FatalError("module definition is NULL");
- return -1;
- }
- PyThreadState *tstate = _PyThreadState_GET();
- if (def->m_slots) {
- _PyErr_SetString(tstate,
- PyExc_SystemError,
- "PyState_AddModule called on module with slots");
- return -1;
- }
- PyInterpreterState *interp = tstate->interp;
- Py_ssize_t index = def->m_base.m_index;
- if (MODULES_BY_INDEX(interp) &&
- index < PyList_GET_SIZE(MODULES_BY_INDEX(interp)) &&
- module == PyList_GET_ITEM(MODULES_BY_INDEX(interp), index))
- {
- _Py_FatalErrorFormat(__func__, "module %p already added", module);
- return -1;
- }
- return _modules_by_index_set(interp, def, module);
- }
- int
- PyState_RemoveModule(PyModuleDef* def)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- if (def->m_slots) {
- _PyErr_SetString(tstate,
- PyExc_SystemError,
- "PyState_RemoveModule called on module with slots");
- return -1;
- }
- return _modules_by_index_clear_one(tstate->interp, def);
- }
- // Used by finalize_modules()
- void
- _PyImport_ClearModulesByIndex(PyInterpreterState *interp)
- {
- if (!MODULES_BY_INDEX(interp)) {
- return;
- }
- Py_ssize_t i;
- for (i = 0; i < PyList_GET_SIZE(MODULES_BY_INDEX(interp)); i++) {
- PyObject *m = PyList_GET_ITEM(MODULES_BY_INDEX(interp), i);
- if (PyModule_Check(m)) {
- /* cleanup the saved copy of module dicts */
- PyModuleDef *md = PyModule_GetDef(m);
- if (md) {
- Py_CLEAR(md->m_base.m_copy);
- }
- }
- }
- /* Setting modules_by_index to NULL could be dangerous, so we
- clear the list instead. */
- if (PyList_SetSlice(MODULES_BY_INDEX(interp),
- 0, PyList_GET_SIZE(MODULES_BY_INDEX(interp)),
- NULL)) {
- PyErr_WriteUnraisable(MODULES_BY_INDEX(interp));
- }
- }
- /*********************/
- /* extension modules */
- /*********************/
- /*
- It may help to have a big picture view of what happens
- when an extension is loaded. This includes when it is imported
- for the first time.
- Here's a summary, using importlib._boostrap._load() as a starting point.
- 1. importlib._bootstrap._load()
- 2. _load(): acquire import lock
- 3. _load() -> importlib._bootstrap._load_unlocked()
- 4. _load_unlocked() -> importlib._bootstrap.module_from_spec()
- 5. module_from_spec() -> ExtensionFileLoader.create_module()
- 6. create_module() -> _imp.create_dynamic()
- (see below)
- 7. module_from_spec() -> importlib._bootstrap._init_module_attrs()
- 8. _load_unlocked(): sys.modules[name] = module
- 9. _load_unlocked() -> ExtensionFileLoader.exec_module()
- 10. exec_module() -> _imp.exec_dynamic()
- (see below)
- 11. _load(): release import lock
- ...for single-phase init modules, where m_size == -1:
- (6). first time (not found in _PyRuntime.imports.extensions):
- 1. _imp_create_dynamic_impl() -> import_find_extension()
- 2. _imp_create_dynamic_impl() -> _PyImport_LoadDynamicModuleWithSpec()
- 3. _PyImport_LoadDynamicModuleWithSpec(): load <module init func>
- 4. _PyImport_LoadDynamicModuleWithSpec(): call <module init func>
- 5. <module init func> -> PyModule_Create() -> PyModule_Create2() -> PyModule_CreateInitialized()
- 6. PyModule_CreateInitialized() -> PyModule_New()
- 7. PyModule_CreateInitialized(): allocate mod->md_state
- 8. PyModule_CreateInitialized() -> PyModule_AddFunctions()
- 9. PyModule_CreateInitialized() -> PyModule_SetDocString()
- 10. PyModule_CreateInitialized(): set mod->md_def
- 11. <module init func>: initialize the module
- 12. _PyImport_LoadDynamicModuleWithSpec() -> _PyImport_CheckSubinterpIncompatibleExtensionAllowed()
- 13. _PyImport_LoadDynamicModuleWithSpec(): set def->m_base.m_init
- 14. _PyImport_LoadDynamicModuleWithSpec(): set __file__
- 15. _PyImport_LoadDynamicModuleWithSpec() -> _PyImport_FixupExtensionObject()
- 16. _PyImport_FixupExtensionObject(): add it to interp->imports.modules_by_index
- 17. _PyImport_FixupExtensionObject(): copy __dict__ into def->m_base.m_copy
- 18. _PyImport_FixupExtensionObject(): add it to _PyRuntime.imports.extensions
- (6). subsequent times (found in _PyRuntime.imports.extensions):
- 1. _imp_create_dynamic_impl() -> import_find_extension()
- 2. import_find_extension() -> import_add_module()
- 3. if name in sys.modules: use that module
- 4. else:
- 1. import_add_module() -> PyModule_NewObject()
- 2. import_add_module(): set it on sys.modules
- 5. import_find_extension(): copy the "m_copy" dict into __dict__
- 6. _imp_create_dynamic_impl() -> _PyImport_CheckSubinterpIncompatibleExtensionAllowed()
- (10). (every time):
- 1. noop
- ...for single-phase init modules, where m_size >= 0:
- (6). not main interpreter and never loaded there - every time (not found in _PyRuntime.imports.extensions):
- 1-16. (same as for m_size == -1)
- (6). main interpreter - first time (not found in _PyRuntime.imports.extensions):
- 1-16. (same as for m_size == -1)
- 17. _PyImport_FixupExtensionObject(): add it to _PyRuntime.imports.extensions
- (6). previously loaded in main interpreter (found in _PyRuntime.imports.extensions):
- 1. _imp_create_dynamic_impl() -> import_find_extension()
- 2. import_find_extension(): call def->m_base.m_init
- 3. import_find_extension(): add the module to sys.modules
- (10). every time:
- 1. noop
- ...for multi-phase init modules:
- (6). every time:
- 1. _imp_create_dynamic_impl() -> import_find_extension() (not found)
- 2. _imp_create_dynamic_impl() -> _PyImport_LoadDynamicModuleWithSpec()
- 3. _PyImport_LoadDynamicModuleWithSpec(): load module init func
- 4. _PyImport_LoadDynamicModuleWithSpec(): call module init func
- 5. _PyImport_LoadDynamicModuleWithSpec() -> PyModule_FromDefAndSpec()
- 6. PyModule_FromDefAndSpec(): gather/check moduledef slots
- 7. if there's a Py_mod_create slot:
- 1. PyModule_FromDefAndSpec(): call its function
- 8. else:
- 1. PyModule_FromDefAndSpec() -> PyModule_NewObject()
- 9: PyModule_FromDefAndSpec(): set mod->md_def
- 10. PyModule_FromDefAndSpec() -> _add_methods_to_object()
- 11. PyModule_FromDefAndSpec() -> PyModule_SetDocString()
- (10). every time:
- 1. _imp_exec_dynamic_impl() -> exec_builtin_or_dynamic()
- 2. if mod->md_state == NULL (including if m_size == 0):
- 1. exec_builtin_or_dynamic() -> PyModule_ExecDef()
- 2. PyModule_ExecDef(): allocate mod->md_state
- 3. if there's a Py_mod_exec slot:
- 1. PyModule_ExecDef(): call its function
- */
- /* Make sure name is fully qualified.
- This is a bit of a hack: when the shared library is loaded,
- the module name is "package.module", but the module calls
- PyModule_Create*() with just "module" for the name. The shared
- library loader squirrels away the true name of the module in
- _PyRuntime.imports.pkgcontext, and PyModule_Create*() will
- substitute this (if the name actually matches).
- */
- #ifdef HAVE_THREAD_LOCAL
- _Py_thread_local const char *pkgcontext = NULL;
- # undef PKGCONTEXT
- # define PKGCONTEXT pkgcontext
- #endif
- const char *
- _PyImport_ResolveNameWithPackageContext(const char *name)
- {
- #ifndef HAVE_THREAD_LOCAL
- PyThread_acquire_lock(EXTENSIONS.mutex, WAIT_LOCK);
- #endif
- if (PKGCONTEXT != NULL) {
- const char *p = strrchr(PKGCONTEXT, '.');
- if (p != NULL && strcmp(name, p+1) == 0) {
- name = PKGCONTEXT;
- PKGCONTEXT = NULL;
- }
- }
- #ifndef HAVE_THREAD_LOCAL
- PyThread_release_lock(EXTENSIONS.mutex);
- #endif
- return name;
- }
- const char *
- _PyImport_SwapPackageContext(const char *newcontext)
- {
- #ifndef HAVE_THREAD_LOCAL
- PyThread_acquire_lock(EXTENSIONS.mutex, WAIT_LOCK);
- #endif
- const char *oldcontext = PKGCONTEXT;
- PKGCONTEXT = newcontext;
- #ifndef HAVE_THREAD_LOCAL
- PyThread_release_lock(EXTENSIONS.mutex);
- #endif
- return oldcontext;
- }
- #ifdef HAVE_DLOPEN
- int
- _PyImport_GetDLOpenFlags(PyInterpreterState *interp)
- {
- return DLOPENFLAGS(interp);
- }
- void
- _PyImport_SetDLOpenFlags(PyInterpreterState *interp, int new_val)
- {
- DLOPENFLAGS(interp) = new_val;
- }
- #endif // HAVE_DLOPEN
- /* Common implementation for _imp.exec_dynamic and _imp.exec_builtin */
- static int
- exec_builtin_or_dynamic(PyObject *mod) {
- PyModuleDef *def;
- void *state;
- if (!PyModule_Check(mod)) {
- return 0;
- }
- def = PyModule_GetDef(mod);
- if (def == NULL) {
- return 0;
- }
- state = PyModule_GetState(mod);
- if (state) {
- /* Already initialized; skip reload */
- return 0;
- }
- return PyModule_ExecDef(mod, def);
- }
- static int clear_singlephase_extension(PyInterpreterState *interp,
- PyObject *name, PyObject *filename);
- // Currently, this is only used for testing.
- // (See _testinternalcapi.clear_extension().)
- int
- _PyImport_ClearExtension(PyObject *name, PyObject *filename)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- /* Clearing a module's C globals is up to the module. */
- if (clear_singlephase_extension(interp, name, filename) < 0) {
- return -1;
- }
- // In the future we'll probably also make sure the extension's
- // file handle (and DL handle) is closed (requires saving it).
- return 0;
- }
- /*******************/
- #if defined(__EMSCRIPTEN__) && defined(PY_CALL_TRAMPOLINE)
- #error #include <emscripten.h>
- EM_JS(PyObject*, _PyImport_InitFunc_TrampolineCall, (PyModInitFunction func), {
- return wasmTable.get(func)();
- });
- #endif // __EMSCRIPTEN__ && PY_CALL_TRAMPOLINE
- /*****************************/
- /* single-phase init modules */
- /*****************************/
- /*
- We support a number of kinds of single-phase init builtin/extension modules:
- * "basic"
- * no module state (PyModuleDef.m_size == -1)
- * does not support repeated init (we use PyModuleDef.m_base.m_copy)
- * may have process-global state
- * the module's def is cached in _PyRuntime.imports.extensions,
- by (name, filename)
- * "reinit"
- * no module state (PyModuleDef.m_size == 0)
- * supports repeated init (m_copy is never used)
- * should not have any process-global state
- * its def is never cached in _PyRuntime.imports.extensions
- (except, currently, under the main interpreter, for some reason)
- * "with state" (almost the same as reinit)
- * has module state (PyModuleDef.m_size > 0)
- * supports repeated init (m_copy is never used)
- * should not have any process-global state
- * its def is never cached in _PyRuntime.imports.extensions
- (except, currently, under the main interpreter, for some reason)
- There are also variants within those classes:
- * two or more modules share a PyModuleDef
- * a module's init func uses another module's PyModuleDef
- * a module's init func calls another's module's init func
- * a module's init "func" is actually a variable statically initialized
- to another module's init func
- * two or modules share "methods"
- * a module's init func copies another module's PyModuleDef
- (with a different name)
- * (basic-only) two or modules share process-global state
- In the first case, where modules share a PyModuleDef, the following
- notable weirdness happens:
- * the module's __name__ matches the def, not the requested name
- * the last module (with the same def) to be imported for the first time wins
- * returned by PyState_Find_Module() (via interp->modules_by_index)
- * (non-basic-only) its init func is used when re-loading any of them
- (via the def's m_init)
- * (basic-only) the copy of its __dict__ is used when re-loading any of them
- (via the def's m_copy)
- However, the following happens as expected:
- * a new module object (with its own __dict__) is created for each request
- * the module's __spec__ has the requested name
- * the loaded module is cached in sys.modules under the requested name
- * the m_index field of the shared def is not changed,
- so at least PyState_FindModule() will always look in the same place
- For "basic" modules there are other quirks:
- * (whether sharing a def or not) when loaded the first time,
- m_copy is set before _init_module_attrs() is called
- in importlib._bootstrap.module_from_spec(),
- so when the module is re-loaded, the previous value
- for __wpec__ (and others) is reset, possibly unexpectedly.
- Generally, when multiple interpreters are involved, some of the above
- gets even messier.
- */
- static inline void
- extensions_lock_acquire(void)
- {
- PyThread_acquire_lock(_PyRuntime.imports.extensions.mutex, WAIT_LOCK);
- }
- static inline void
- extensions_lock_release(void)
- {
- PyThread_release_lock(_PyRuntime.imports.extensions.mutex);
- }
- /* Magic for extension modules (built-in as well as dynamically
- loaded). To prevent initializing an extension module more than
- once, we keep a static dictionary 'extensions' keyed by the tuple
- (module name, module name) (for built-in modules) or by
- (filename, module name) (for dynamically loaded modules), containing these
- modules. A copy of the module's dictionary is stored by calling
- _PyImport_FixupExtensionObject() immediately after the module initialization
- function succeeds. A copy can be retrieved from there by calling
- import_find_extension().
- Modules which do support multiple initialization set their m_size
- field to a non-negative number (indicating the size of the
- module-specific state). They are still recorded in the extensions
- dictionary, to avoid loading shared libraries twice.
- */
- static void *
- hashtable_key_from_2_strings(PyObject *str1, PyObject *str2, const char sep)
- {
- const char *str1_data = _PyUnicode_AsUTF8NoNUL(str1);
- const char *str2_data = _PyUnicode_AsUTF8NoNUL(str2);
- if (str1_data == NULL || str2_data == NULL) {
- return NULL;
- }
- Py_ssize_t str1_len = strlen(str1_data);
- Py_ssize_t str2_len = strlen(str2_data);
- /* Make sure sep and the NULL byte won't cause an overflow. */
- assert(SIZE_MAX - str1_len - str2_len > 2);
- size_t size = str1_len + 1 + str2_len + 1;
- char *key = PyMem_RawMalloc(size);
- if (key == NULL) {
- PyErr_NoMemory();
- return NULL;
- }
- strncpy(key, str1_data, str1_len);
- key[str1_len] = sep;
- strncpy(key + str1_len + 1, str2_data, str2_len + 1);
- assert(strlen(key) == size - 1);
- return key;
- }
- static Py_uhash_t
- hashtable_hash_str(const void *key)
- {
- return _Py_HashBytes(key, strlen((const char *)key));
- }
- static int
- hashtable_compare_str(const void *key1, const void *key2)
- {
- return strcmp((const char *)key1, (const char *)key2) == 0;
- }
- static void
- hashtable_destroy_str(void *ptr)
- {
- PyMem_RawFree(ptr);
- }
- #define HTSEP ':'
- static PyModuleDef *
- _extensions_cache_get(PyObject *filename, PyObject *name)
- {
- PyModuleDef *def = NULL;
- void *key = NULL;
- extensions_lock_acquire();
- if (EXTENSIONS.hashtable == NULL) {
- goto finally;
- }
- key = hashtable_key_from_2_strings(filename, name, HTSEP);
- if (key == NULL) {
- goto finally;
- }
- _Py_hashtable_entry_t *entry = _Py_hashtable_get_entry(
- EXTENSIONS.hashtable, key);
- if (entry == NULL) {
- goto finally;
- }
- def = (PyModuleDef *)entry->value;
- finally:
- extensions_lock_release();
- if (key != NULL) {
- PyMem_RawFree(key);
- }
- return def;
- }
- static int
- _extensions_cache_set(PyObject *filename, PyObject *name, PyModuleDef *def)
- {
- int res = -1;
- extensions_lock_acquire();
- if (EXTENSIONS.hashtable == NULL) {
- _Py_hashtable_allocator_t alloc = {PyMem_RawMalloc, PyMem_RawFree};
- EXTENSIONS.hashtable = _Py_hashtable_new_full(
- hashtable_hash_str,
- hashtable_compare_str,
- hashtable_destroy_str, // key
- /* There's no need to decref the def since it's immortal. */
- NULL, // value
- &alloc
- );
- if (EXTENSIONS.hashtable == NULL) {
- PyErr_NoMemory();
- goto finally;
- }
- }
- void *key = hashtable_key_from_2_strings(filename, name, HTSEP);
- if (key == NULL) {
- goto finally;
- }
- int already_set = 0;
- _Py_hashtable_entry_t *entry = _Py_hashtable_get_entry(
- EXTENSIONS.hashtable, key);
- if (entry == NULL) {
- if (_Py_hashtable_set(EXTENSIONS.hashtable, key, def) < 0) {
- PyMem_RawFree(key);
- PyErr_NoMemory();
- goto finally;
- }
- }
- else {
- if (entry->value == NULL) {
- entry->value = def;
- }
- else {
- /* We expect it to be static, so it must be the same pointer. */
- assert((PyModuleDef *)entry->value == def);
- already_set = 1;
- }
- PyMem_RawFree(key);
- }
- if (!already_set) {
- /* We assume that all module defs are statically allocated
- and will never be freed. Otherwise, we would incref here. */
- _Py_SetImmortal(def);
- }
- res = 0;
- finally:
- extensions_lock_release();
- return res;
- }
- static void
- _extensions_cache_delete(PyObject *filename, PyObject *name)
- {
- void *key = NULL;
- extensions_lock_acquire();
- if (EXTENSIONS.hashtable == NULL) {
- /* It was never added. */
- goto finally;
- }
- key = hashtable_key_from_2_strings(filename, name, HTSEP);
- if (key == NULL) {
- goto finally;
- }
- _Py_hashtable_entry_t *entry = _Py_hashtable_get_entry(
- EXTENSIONS.hashtable, key);
- if (entry == NULL) {
- /* It was never added. */
- goto finally;
- }
- if (entry->value == NULL) {
- /* It was already removed. */
- goto finally;
- }
- /* If we hadn't made the stored defs immortal, we would decref here.
- However, this decref would be problematic if the module def were
- dynamically allocated, it were the last ref, and this function
- were called with an interpreter other than the def's owner. */
- assert(_Py_IsImmortal(entry->value));
- entry->value = NULL;
- finally:
- extensions_lock_release();
- if (key != NULL) {
- PyMem_RawFree(key);
- }
- }
- static void
- _extensions_cache_clear_all(void)
- {
- /* The runtime (i.e. main interpreter) must be finalizing,
- so we don't need to worry about the lock. */
- _Py_hashtable_destroy(EXTENSIONS.hashtable);
- EXTENSIONS.hashtable = NULL;
- }
- #undef HTSEP
- static bool
- check_multi_interp_extensions(PyInterpreterState *interp)
- {
- int override = OVERRIDE_MULTI_INTERP_EXTENSIONS_CHECK(interp);
- if (override < 0) {
- return false;
- }
- else if (override > 0) {
- return true;
- }
- else if (_PyInterpreterState_HasFeature(
- interp, Py_RTFLAGS_MULTI_INTERP_EXTENSIONS)) {
- return true;
- }
- return false;
- }
- int
- _PyImport_CheckSubinterpIncompatibleExtensionAllowed(const char *name)
- {
- PyInterpreterState *interp = _PyInterpreterState_Get();
- if (check_multi_interp_extensions(interp)) {
- assert(!_Py_IsMainInterpreter(interp));
- PyErr_Format(PyExc_ImportError,
- "module %s does not support loading in subinterpreters",
- name);
- return -1;
- }
- return 0;
- }
- static PyObject *
- get_core_module_dict(PyInterpreterState *interp,
- PyObject *name, PyObject *filename)
- {
- /* Only builtin modules are core. */
- if (filename == name) {
- assert(!PyErr_Occurred());
- if (PyUnicode_CompareWithASCIIString(name, "sys") == 0) {
- return interp->sysdict_copy;
- }
- assert(!PyErr_Occurred());
- if (PyUnicode_CompareWithASCIIString(name, "builtins") == 0) {
- return interp->builtins_copy;
- }
- assert(!PyErr_Occurred());
- }
- return NULL;
- }
- static inline int
- is_core_module(PyInterpreterState *interp, PyObject *name, PyObject *filename)
- {
- /* This might be called before the core dict copies are in place,
- so we can't rely on get_core_module_dict() here. */
- if (filename == name) {
- if (PyUnicode_CompareWithASCIIString(name, "sys") == 0) {
- return 1;
- }
- if (PyUnicode_CompareWithASCIIString(name, "builtins") == 0) {
- return 1;
- }
- }
- return 0;
- }
- static int
- fix_up_extension(PyObject *mod, PyObject *name, PyObject *filename)
- {
- if (mod == NULL || !PyModule_Check(mod)) {
- PyErr_BadInternalCall();
- return -1;
- }
- struct PyModuleDef *def = PyModule_GetDef(mod);
- if (!def) {
- PyErr_BadInternalCall();
- return -1;
- }
- PyThreadState *tstate = _PyThreadState_GET();
- if (_modules_by_index_set(tstate->interp, def, mod) < 0) {
- return -1;
- }
- // bpo-44050: Extensions and def->m_base.m_copy can be updated
- // when the extension module doesn't support sub-interpreters.
- if (def->m_size == -1) {
- if (!is_core_module(tstate->interp, name, filename)) {
- assert(PyUnicode_CompareWithASCIIString(name, "sys") != 0);
- assert(PyUnicode_CompareWithASCIIString(name, "builtins") != 0);
- if (def->m_base.m_copy) {
- /* Somebody already imported the module,
- likely under a different name.
- XXX this should really not happen. */
- Py_CLEAR(def->m_base.m_copy);
- }
- PyObject *dict = PyModule_GetDict(mod);
- if (dict == NULL) {
- return -1;
- }
- def->m_base.m_copy = PyDict_Copy(dict);
- if (def->m_base.m_copy == NULL) {
- return -1;
- }
- }
- }
- // XXX Why special-case the main interpreter?
- if (_Py_IsMainInterpreter(tstate->interp) || def->m_size == -1) {
- if (_extensions_cache_set(filename, name, def) < 0) {
- return -1;
- }
- }
- return 0;
- }
- int
- _PyImport_FixupExtensionObject(PyObject *mod, PyObject *name,
- PyObject *filename, PyObject *modules)
- {
- if (PyObject_SetItem(modules, name, mod) < 0) {
- return -1;
- }
- if (fix_up_extension(mod, name, filename) < 0) {
- PyMapping_DelItem(modules, name);
- return -1;
- }
- return 0;
- }
- static PyObject *
- import_find_extension(PyThreadState *tstate, PyObject *name,
- PyObject *filename)
- {
- /* Only single-phase init modules will be in the cache. */
- PyModuleDef *def = _extensions_cache_get(filename, name);
- if (def == NULL) {
- return NULL;
- }
- /* It may have been successfully imported previously
- in an interpreter that allows legacy modules
- but is not allowed in the current interpreter. */
- const char *name_buf = PyUnicode_AsUTF8(name);
- assert(name_buf != NULL);
- if (_PyImport_CheckSubinterpIncompatibleExtensionAllowed(name_buf) < 0) {
- return NULL;
- }
- PyObject *mod, *mdict;
- PyObject *modules = MODULES(tstate->interp);
- if (def->m_size == -1) {
- PyObject *m_copy = def->m_base.m_copy;
- /* Module does not support repeated initialization */
- if (m_copy == NULL) {
- /* It might be a core module (e.g. sys & builtins),
- for which we don't set m_copy. */
- m_copy = get_core_module_dict(tstate->interp, name, filename);
- if (m_copy == NULL) {
- return NULL;
- }
- }
- mod = import_add_module(tstate, name);
- if (mod == NULL) {
- return NULL;
- }
- mdict = PyModule_GetDict(mod);
- if (mdict == NULL) {
- Py_DECREF(mod);
- return NULL;
- }
- if (PyDict_Update(mdict, m_copy)) {
- Py_DECREF(mod);
- return NULL;
- }
- }
- else {
- if (def->m_base.m_init == NULL)
- return NULL;
- mod = _PyImport_InitFunc_TrampolineCall(def->m_base.m_init);
- if (mod == NULL)
- return NULL;
- if (PyObject_SetItem(modules, name, mod) == -1) {
- Py_DECREF(mod);
- return NULL;
- }
- }
- if (_modules_by_index_set(tstate->interp, def, mod) < 0) {
- PyMapping_DelItem(modules, name);
- Py_DECREF(mod);
- return NULL;
- }
- int verbose = _PyInterpreterState_GetConfig(tstate->interp)->verbose;
- if (verbose) {
- PySys_FormatStderr("import %U # previously loaded (%R)\n",
- name, filename);
- }
- return mod;
- }
- static int
- clear_singlephase_extension(PyInterpreterState *interp,
- PyObject *name, PyObject *filename)
- {
- PyModuleDef *def = _extensions_cache_get(filename, name);
- if (def == NULL) {
- if (PyErr_Occurred()) {
- return -1;
- }
- return 0;
- }
- /* Clear data set when the module was initially loaded. */
- def->m_base.m_init = NULL;
- Py_CLEAR(def->m_base.m_copy);
- // We leave m_index alone since there's no reason to reset it.
- /* Clear the PyState_*Module() cache entry. */
- if (_modules_by_index_check(interp, def->m_base.m_index) == NULL) {
- if (_modules_by_index_clear_one(interp, def) < 0) {
- return -1;
- }
- }
- /* Clear the cached module def. */
- _extensions_cache_delete(filename, name);
- return 0;
- }
- /*******************/
- /* builtin modules */
- /*******************/
- int
- _PyImport_FixupBuiltin(PyObject *mod, const char *name, PyObject *modules)
- {
- int res = -1;
- PyObject *nameobj;
- nameobj = PyUnicode_InternFromString(name);
- if (nameobj == NULL) {
- return -1;
- }
- if (PyObject_SetItem(modules, nameobj, mod) < 0) {
- goto finally;
- }
- if (fix_up_extension(mod, nameobj, nameobj) < 0) {
- PyMapping_DelItem(modules, nameobj);
- goto finally;
- }
- res = 0;
- finally:
- Py_DECREF(nameobj);
- return res;
- }
- /* Helper to test for built-in module */
- static int
- is_builtin(PyObject *name)
- {
- int i;
- struct _inittab *inittab = INITTAB;
- for (i = 0; inittab[i].name != NULL; i++) {
- if (_PyUnicode_EqualToASCIIString(name, inittab[i].name)) {
- if (inittab[i].initfunc == NULL)
- return -1;
- else
- return 1;
- }
- }
- return 0;
- }
- static PyObject*
- create_builtin(PyThreadState *tstate, PyObject *name, PyObject *spec)
- {
- PyObject *mod = import_find_extension(tstate, name, name);
- if (mod || _PyErr_Occurred(tstate)) {
- return mod;
- }
- PyObject *modules = MODULES(tstate->interp);
- for (struct _inittab *p = INITTAB; p->name != NULL; p++) {
- if (_PyUnicode_EqualToASCIIString(name, p->name)) {
- if (p->initfunc == NULL) {
- /* Cannot re-init internal module ("sys" or "builtins") */
- mod = PyImport_AddModuleObject(name);
- return Py_XNewRef(mod);
- }
- mod = _PyImport_InitFunc_TrampolineCall(*p->initfunc);
- if (mod == NULL) {
- return NULL;
- }
- if (PyObject_TypeCheck(mod, &PyModuleDef_Type)) {
- return PyModule_FromDefAndSpec((PyModuleDef*)mod, spec);
- }
- else {
- /* Remember pointer to module init function. */
- PyModuleDef *def = PyModule_GetDef(mod);
- if (def == NULL) {
- return NULL;
- }
- def->m_base.m_init = p->initfunc;
- if (_PyImport_FixupExtensionObject(mod, name, name,
- modules) < 0) {
- return NULL;
- }
- return mod;
- }
- }
- }
- // not found
- Py_RETURN_NONE;
- }
- /*****************************/
- /* the builtin modules table */
- /*****************************/
- /* API for embedding applications that want to add their own entries
- to the table of built-in modules. This should normally be called
- *before* Py_Initialize(). When the table resize fails, -1 is
- returned and the existing table is unchanged.
- After a similar function by Just van Rossum. */
- int
- PyImport_ExtendInittab(struct _inittab *newtab)
- {
- struct _inittab *p;
- size_t i, n;
- int res = 0;
- if (INITTAB != NULL) {
- Py_FatalError("PyImport_ExtendInittab() may not be called after Py_Initialize()");
- }
- /* Count the number of entries in both tables */
- for (n = 0; newtab[n].name != NULL; n++)
- ;
- if (n == 0)
- return 0; /* Nothing to do */
- for (i = 0; PyImport_Inittab[i].name != NULL; i++)
- ;
- /* Force default raw memory allocator to get a known allocator to be able
- to release the memory in _PyImport_Fini2() */
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- /* Allocate new memory for the combined table */
- p = NULL;
- if (i + n <= SIZE_MAX / sizeof(struct _inittab) - 1) {
- size_t size = sizeof(struct _inittab) * (i + n + 1);
- p = PyMem_RawRealloc(inittab_copy, size);
- }
- if (p == NULL) {
- res = -1;
- goto done;
- }
- /* Copy the tables into the new memory at the first call
- to PyImport_ExtendInittab(). */
- if (inittab_copy != PyImport_Inittab) {
- memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
- }
- memcpy(p + i, newtab, (n + 1) * sizeof(struct _inittab));
- PyImport_Inittab = inittab_copy = p;
- done:
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- return res;
- }
- /* Shorthand to add a single entry given a name and a function */
- int
- PyImport_AppendInittab(const char *name, PyObject* (*initfunc)(void))
- {
- struct _inittab newtab[2];
- if (INITTAB != NULL) {
- Py_FatalError("PyImport_AppendInittab() may not be called after Py_Initialize()");
- }
- memset(newtab, '\0', sizeof newtab);
- newtab[0].name = name;
- newtab[0].initfunc = initfunc;
- return PyImport_ExtendInittab(newtab);
- }
- /* the internal table */
- static int
- init_builtin_modules_table(void)
- {
- size_t size;
- for (size = 0; PyImport_Inittab[size].name != NULL; size++)
- ;
- size++;
- /* Make the copy. */
- struct _inittab *copied = PyMem_RawMalloc(size * sizeof(struct _inittab));
- if (copied == NULL) {
- return -1;
- }
- memcpy(copied, PyImport_Inittab, size * sizeof(struct _inittab));
- INITTAB = copied;
- return 0;
- }
- static void
- fini_builtin_modules_table(void)
- {
- struct _inittab *inittab = INITTAB;
- INITTAB = NULL;
- PyMem_RawFree(inittab);
- }
- PyObject *
- _PyImport_GetBuiltinModuleNames(void)
- {
- PyObject *list = PyList_New(0);
- if (list == NULL) {
- return NULL;
- }
- struct _inittab *inittab = INITTAB;
- for (Py_ssize_t i = 0; inittab[i].name != NULL; i++) {
- PyObject *name = PyUnicode_FromString(inittab[i].name);
- if (name == NULL) {
- Py_DECREF(list);
- return NULL;
- }
- if (PyList_Append(list, name) < 0) {
- Py_DECREF(name);
- Py_DECREF(list);
- return NULL;
- }
- Py_DECREF(name);
- }
- return list;
- }
- /********************/
- /* the magic number */
- /********************/
- /* Helper for pythonrun.c -- return magic number and tag. */
- long
- PyImport_GetMagicNumber(void)
- {
- long res;
- PyInterpreterState *interp = _PyInterpreterState_GET();
- PyObject *external, *pyc_magic;
- external = PyObject_GetAttrString(IMPORTLIB(interp), "_bootstrap_external");
- if (external == NULL)
- return -1;
- pyc_magic = PyObject_GetAttrString(external, "_RAW_MAGIC_NUMBER");
- Py_DECREF(external);
- if (pyc_magic == NULL)
- return -1;
- res = PyLong_AsLong(pyc_magic);
- Py_DECREF(pyc_magic);
- return res;
- }
- extern const char * _PySys_ImplCacheTag;
- const char *
- PyImport_GetMagicTag(void)
- {
- return _PySys_ImplCacheTag;
- }
- /*********************************/
- /* a Python module's code object */
- /*********************************/
- /* Execute a code object in a module and return the module object
- * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
- * removed from sys.modules, to avoid leaving damaged module objects
- * in sys.modules. The caller may wish to restore the original
- * module object (if any) in this case; PyImport_ReloadModule is an
- * example.
- *
- * Note that PyImport_ExecCodeModuleWithPathnames() is the preferred, richer
- * interface. The other two exist primarily for backward compatibility.
- */
- PyObject *
- PyImport_ExecCodeModule(const char *name, PyObject *co)
- {
- return PyImport_ExecCodeModuleWithPathnames(
- name, co, (char *)NULL, (char *)NULL);
- }
- PyObject *
- PyImport_ExecCodeModuleEx(const char *name, PyObject *co, const char *pathname)
- {
- return PyImport_ExecCodeModuleWithPathnames(
- name, co, pathname, (char *)NULL);
- }
- PyObject *
- PyImport_ExecCodeModuleWithPathnames(const char *name, PyObject *co,
- const char *pathname,
- const char *cpathname)
- {
- PyObject *m = NULL;
- PyObject *nameobj, *pathobj = NULL, *cpathobj = NULL, *external= NULL;
- nameobj = PyUnicode_FromString(name);
- if (nameobj == NULL)
- return NULL;
- if (cpathname != NULL) {
- cpathobj = PyUnicode_DecodeFSDefault(cpathname);
- if (cpathobj == NULL)
- goto error;
- }
- else
- cpathobj = NULL;
- if (pathname != NULL) {
- pathobj = PyUnicode_DecodeFSDefault(pathname);
- if (pathobj == NULL)
- goto error;
- }
- else if (cpathobj != NULL) {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (interp == NULL) {
- Py_FatalError("no current interpreter");
- }
- external= PyObject_GetAttrString(IMPORTLIB(interp),
- "_bootstrap_external");
- if (external != NULL) {
- pathobj = _PyObject_CallMethodOneArg(
- external, &_Py_ID(_get_sourcefile), cpathobj);
- Py_DECREF(external);
- }
- if (pathobj == NULL)
- PyErr_Clear();
- }
- else
- pathobj = NULL;
- m = PyImport_ExecCodeModuleObject(nameobj, co, pathobj, cpathobj);
- error:
- Py_DECREF(nameobj);
- Py_XDECREF(pathobj);
- Py_XDECREF(cpathobj);
- return m;
- }
- static PyObject *
- module_dict_for_exec(PyThreadState *tstate, PyObject *name)
- {
- PyObject *m, *d;
- m = import_add_module(tstate, name);
- if (m == NULL)
- return NULL;
- /* If the module is being reloaded, we get the old module back
- and re-use its dict to exec the new code. */
- d = PyModule_GetDict(m);
- int r = PyDict_Contains(d, &_Py_ID(__builtins__));
- if (r == 0) {
- r = PyDict_SetItem(d, &_Py_ID(__builtins__), PyEval_GetBuiltins());
- }
- if (r < 0) {
- remove_module(tstate, name);
- Py_DECREF(m);
- return NULL;
- }
- Py_INCREF(d);
- Py_DECREF(m);
- return d;
- }
- static PyObject *
- exec_code_in_module(PyThreadState *tstate, PyObject *name,
- PyObject *module_dict, PyObject *code_object)
- {
- PyObject *v, *m;
- v = PyEval_EvalCode(code_object, module_dict, module_dict);
- if (v == NULL) {
- remove_module(tstate, name);
- return NULL;
- }
- Py_DECREF(v);
- m = import_get_module(tstate, name);
- if (m == NULL && !_PyErr_Occurred(tstate)) {
- _PyErr_Format(tstate, PyExc_ImportError,
- "Loaded module %R not found in sys.modules",
- name);
- }
- return m;
- }
- PyObject*
- PyImport_ExecCodeModuleObject(PyObject *name, PyObject *co, PyObject *pathname,
- PyObject *cpathname)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *d, *external, *res;
- d = module_dict_for_exec(tstate, name);
- if (d == NULL) {
- return NULL;
- }
- if (pathname == NULL) {
- pathname = ((PyCodeObject *)co)->co_filename;
- }
- external = PyObject_GetAttrString(IMPORTLIB(tstate->interp),
- "_bootstrap_external");
- if (external == NULL) {
- Py_DECREF(d);
- return NULL;
- }
- res = PyObject_CallMethodObjArgs(external, &_Py_ID(_fix_up_module),
- d, name, pathname, cpathname, NULL);
- Py_DECREF(external);
- if (res != NULL) {
- Py_DECREF(res);
- res = exec_code_in_module(tstate, name, d, co);
- }
- Py_DECREF(d);
- return res;
- }
- static void
- update_code_filenames(PyCodeObject *co, PyObject *oldname, PyObject *newname)
- {
- PyObject *constants, *tmp;
- Py_ssize_t i, n;
- if (PyUnicode_Compare(co->co_filename, oldname))
- return;
- Py_XSETREF(co->co_filename, Py_NewRef(newname));
- constants = co->co_consts;
- n = PyTuple_GET_SIZE(constants);
- for (i = 0; i < n; i++) {
- tmp = PyTuple_GET_ITEM(constants, i);
- if (PyCode_Check(tmp))
- update_code_filenames((PyCodeObject *)tmp,
- oldname, newname);
- }
- }
- static void
- update_compiled_module(PyCodeObject *co, PyObject *newname)
- {
- PyObject *oldname;
- if (PyUnicode_Compare(co->co_filename, newname) == 0)
- return;
- oldname = co->co_filename;
- Py_INCREF(oldname);
- update_code_filenames(co, oldname, newname);
- Py_DECREF(oldname);
- }
- /******************/
- /* frozen modules */
- /******************/
- /* Return true if the name is an alias. In that case, "alias" is set
- to the original module name. If it is an alias but the original
- module isn't known then "alias" is set to NULL while true is returned. */
- static bool
- resolve_module_alias(const char *name, const struct _module_alias *aliases,
- const char **alias)
- {
- const struct _module_alias *entry;
- for (entry = aliases; ; entry++) {
- if (entry->name == NULL) {
- /* It isn't an alias. */
- return false;
- }
- if (strcmp(name, entry->name) == 0) {
- if (alias != NULL) {
- *alias = entry->orig;
- }
- return true;
- }
- }
- }
- static bool
- use_frozen(void)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- int override = OVERRIDE_FROZEN_MODULES(interp);
- if (override > 0) {
- return true;
- }
- else if (override < 0) {
- return false;
- }
- else {
- return interp->config.use_frozen_modules;
- }
- }
- static PyObject *
- list_frozen_module_names(void)
- {
- PyObject *names = PyList_New(0);
- if (names == NULL) {
- return NULL;
- }
- bool enabled = use_frozen();
- const struct _frozen *p;
- #define ADD_MODULE(name) \
- do { \
- PyObject *nameobj = PyUnicode_FromString(name); \
- if (nameobj == NULL) { \
- goto error; \
- } \
- int res = PyList_Append(names, nameobj); \
- Py_DECREF(nameobj); \
- if (res != 0) { \
- goto error; \
- } \
- } while(0)
- // We always use the bootstrap modules.
- for (p = _PyImport_FrozenBootstrap; ; p++) {
- if (p->name == NULL) {
- break;
- }
- ADD_MODULE(p->name);
- }
- // Frozen stdlib modules may be disabled.
- for (p = _PyImport_FrozenStdlib; ; p++) {
- if (p->name == NULL) {
- break;
- }
- if (enabled) {
- ADD_MODULE(p->name);
- }
- }
- for (p = _PyImport_FrozenTest; ; p++) {
- if (p->name == NULL) {
- break;
- }
- if (enabled) {
- ADD_MODULE(p->name);
- }
- }
- #undef ADD_MODULE
- // Add any custom modules.
- if (PyImport_FrozenModules != NULL) {
- for (p = PyImport_FrozenModules; ; p++) {
- if (p->name == NULL) {
- break;
- }
- PyObject *nameobj = PyUnicode_FromString(p->name);
- if (nameobj == NULL) {
- goto error;
- }
- int found = PySequence_Contains(names, nameobj);
- if (found < 0) {
- Py_DECREF(nameobj);
- goto error;
- }
- else if (found) {
- Py_DECREF(nameobj);
- }
- else {
- int res = PyList_Append(names, nameobj);
- Py_DECREF(nameobj);
- if (res != 0) {
- goto error;
- }
- }
- }
- }
- return names;
- error:
- Py_DECREF(names);
- return NULL;
- }
- typedef enum {
- FROZEN_OKAY,
- FROZEN_BAD_NAME, // The given module name wasn't valid.
- FROZEN_NOT_FOUND, // It wasn't in PyImport_FrozenModules.
- FROZEN_DISABLED, // -X frozen_modules=off (and not essential)
- FROZEN_EXCLUDED, /* The PyImport_FrozenModules entry has NULL "code"
- (module is present but marked as unimportable, stops search). */
- FROZEN_INVALID, /* The PyImport_FrozenModules entry is bogus
- (eg. does not contain executable code). */
- } frozen_status;
- static inline void
- set_frozen_error(frozen_status status, PyObject *modname)
- {
- const char *err = NULL;
- switch (status) {
- case FROZEN_BAD_NAME:
- case FROZEN_NOT_FOUND:
- err = "No such frozen object named %R";
- break;
- case FROZEN_DISABLED:
- err = "Frozen modules are disabled and the frozen object named %R is not essential";
- break;
- case FROZEN_EXCLUDED:
- err = "Excluded frozen object named %R";
- break;
- case FROZEN_INVALID:
- err = "Frozen object named %R is invalid";
- break;
- case FROZEN_OKAY:
- // There was no error.
- break;
- default:
- Py_UNREACHABLE();
- }
- if (err != NULL) {
- PyObject *msg = PyUnicode_FromFormat(err, modname);
- if (msg == NULL) {
- PyErr_Clear();
- }
- PyErr_SetImportError(msg, modname, NULL);
- Py_XDECREF(msg);
- }
- }
- static const struct _frozen *
- look_up_frozen(const char *name)
- {
- const struct _frozen *p;
- // We always use the bootstrap modules.
- for (p = _PyImport_FrozenBootstrap; ; p++) {
- if (p->name == NULL) {
- // We hit the end-of-list sentinel value.
- break;
- }
- if (strcmp(name, p->name) == 0) {
- return p;
- }
- }
- // Prefer custom modules, if any. Frozen stdlib modules can be
- // disabled here by setting "code" to NULL in the array entry.
- if (PyImport_FrozenModules != NULL) {
- for (p = PyImport_FrozenModules; ; p++) {
- if (p->name == NULL) {
- break;
- }
- if (strcmp(name, p->name) == 0) {
- return p;
- }
- }
- }
- // Frozen stdlib modules may be disabled.
- if (use_frozen()) {
- for (p = _PyImport_FrozenStdlib; ; p++) {
- if (p->name == NULL) {
- break;
- }
- if (strcmp(name, p->name) == 0) {
- return p;
- }
- }
- for (p = _PyImport_FrozenTest; ; p++) {
- if (p->name == NULL) {
- break;
- }
- if (strcmp(name, p->name) == 0) {
- return p;
- }
- }
- }
- return NULL;
- }
- struct frozen_info {
- PyObject *nameobj;
- const char *data;
- PyObject *(*get_code)(void);
- Py_ssize_t size;
- bool is_package;
- bool is_alias;
- const char *origname;
- };
- static frozen_status
- find_frozen(PyObject *nameobj, struct frozen_info *info)
- {
- if (info != NULL) {
- memset(info, 0, sizeof(*info));
- }
- if (nameobj == NULL || nameobj == Py_None) {
- return FROZEN_BAD_NAME;
- }
- const char *name = PyUnicode_AsUTF8(nameobj);
- if (name == NULL) {
- // Note that this function previously used
- // _PyUnicode_EqualToASCIIString(). We clear the error here
- // (instead of propagating it) to match the earlier behavior
- // more closely.
- PyErr_Clear();
- return FROZEN_BAD_NAME;
- }
- const struct _frozen *p = look_up_frozen(name);
- if (p == NULL) {
- return FROZEN_NOT_FOUND;
- }
- if (info != NULL) {
- info->nameobj = nameobj; // borrowed
- info->data = (const char *)p->code;
- info->get_code = p->get_code;
- info->size = p->size;
- info->is_package = p->is_package;
- if (p->size < 0) {
- // backward compatibility with negative size values
- info->size = -(p->size);
- info->is_package = true;
- }
- info->origname = name;
- info->is_alias = resolve_module_alias(name, _PyImport_FrozenAliases,
- &info->origname);
- }
- if (p->code == NULL && p->size == 0 && p->get_code != NULL) {
- /* It is only deepfrozen. */
- return FROZEN_OKAY;
- }
- if (p->code == NULL) {
- /* It is frozen but marked as un-importable. */
- return FROZEN_EXCLUDED;
- }
- if (p->code[0] == '\0' || p->size == 0) {
- /* Does not contain executable code. */
- return FROZEN_INVALID;
- }
- return FROZEN_OKAY;
- }
- static PyObject *
- unmarshal_frozen_code(PyInterpreterState *interp, struct frozen_info *info)
- {
- if (info->get_code && _Py_IsMainInterpreter(interp)) {
- PyObject *code = info->get_code();
- assert(code != NULL);
- return code;
- }
- PyObject *co = PyMarshal_ReadObjectFromString(info->data, info->size);
- if (co == NULL) {
- /* Does not contain executable code. */
- PyErr_Clear();
- set_frozen_error(FROZEN_INVALID, info->nameobj);
- return NULL;
- }
- if (!PyCode_Check(co)) {
- // We stick with TypeError for backward compatibility.
- PyErr_Format(PyExc_TypeError,
- "frozen object %R is not a code object",
- info->nameobj);
- Py_DECREF(co);
- return NULL;
- }
- return co;
- }
- /* Initialize a frozen module.
- Return 1 for success, 0 if the module is not found, and -1 with
- an exception set if the initialization failed.
- This function is also used from frozenmain.c */
- int
- PyImport_ImportFrozenModuleObject(PyObject *name)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *co, *m, *d = NULL;
- int err;
- struct frozen_info info;
- frozen_status status = find_frozen(name, &info);
- if (status == FROZEN_NOT_FOUND || status == FROZEN_DISABLED) {
- return 0;
- }
- else if (status == FROZEN_BAD_NAME) {
- return 0;
- }
- else if (status != FROZEN_OKAY) {
- set_frozen_error(status, name);
- return -1;
- }
- co = unmarshal_frozen_code(tstate->interp, &info);
- if (co == NULL) {
- return -1;
- }
- if (info.is_package) {
- /* Set __path__ to the empty list */
- PyObject *l;
- m = import_add_module(tstate, name);
- if (m == NULL)
- goto err_return;
- d = PyModule_GetDict(m);
- l = PyList_New(0);
- if (l == NULL) {
- Py_DECREF(m);
- goto err_return;
- }
- err = PyDict_SetItemString(d, "__path__", l);
- Py_DECREF(l);
- Py_DECREF(m);
- if (err != 0)
- goto err_return;
- }
- d = module_dict_for_exec(tstate, name);
- if (d == NULL) {
- goto err_return;
- }
- m = exec_code_in_module(tstate, name, d, co);
- if (m == NULL) {
- goto err_return;
- }
- Py_DECREF(m);
- /* Set __origname__ (consumed in FrozenImporter._setup_module()). */
- PyObject *origname;
- if (info.origname) {
- origname = PyUnicode_FromString(info.origname);
- if (origname == NULL) {
- goto err_return;
- }
- }
- else {
- origname = Py_NewRef(Py_None);
- }
- err = PyDict_SetItemString(d, "__origname__", origname);
- Py_DECREF(origname);
- if (err != 0) {
- goto err_return;
- }
- Py_DECREF(d);
- Py_DECREF(co);
- return 1;
- err_return:
- Py_XDECREF(d);
- Py_DECREF(co);
- return -1;
- }
- int
- PyImport_ImportFrozenModule(const char *name)
- {
- PyObject *nameobj;
- int ret;
- nameobj = PyUnicode_InternFromString(name);
- if (nameobj == NULL)
- return -1;
- ret = PyImport_ImportFrozenModuleObject(nameobj);
- Py_DECREF(nameobj);
- return ret;
- }
- /*************/
- /* importlib */
- /*************/
- /* Import the _imp extension by calling manually _imp.create_builtin() and
- _imp.exec_builtin() since importlib is not initialized yet. Initializing
- importlib requires the _imp module: this function fix the bootstrap issue.
- */
- static PyObject*
- bootstrap_imp(PyThreadState *tstate)
- {
- PyObject *name = PyUnicode_FromString("_imp");
- if (name == NULL) {
- return NULL;
- }
- // Mock a ModuleSpec object just good enough for PyModule_FromDefAndSpec():
- // an object with just a name attribute.
- //
- // _imp.__spec__ is overridden by importlib._bootstrap._instal() anyway.
- PyObject *attrs = Py_BuildValue("{sO}", "name", name);
- if (attrs == NULL) {
- goto error;
- }
- PyObject *spec = _PyNamespace_New(attrs);
- Py_DECREF(attrs);
- if (spec == NULL) {
- goto error;
- }
- // Create the _imp module from its definition.
- PyObject *mod = create_builtin(tstate, name, spec);
- Py_CLEAR(name);
- Py_DECREF(spec);
- if (mod == NULL) {
- goto error;
- }
- assert(mod != Py_None); // not found
- // Execute the _imp module: call imp_module_exec().
- if (exec_builtin_or_dynamic(mod) < 0) {
- Py_DECREF(mod);
- goto error;
- }
- return mod;
- error:
- Py_XDECREF(name);
- return NULL;
- }
- /* Global initializations. Can be undone by Py_FinalizeEx(). Don't
- call this twice without an intervening Py_FinalizeEx() call. When
- initializations fail, a fatal error is issued and the function does
- not return. On return, the first thread and interpreter state have
- been created.
- Locking: you must hold the interpreter lock while calling this.
- (If the lock has not yet been initialized, that's equivalent to
- having the lock, but you cannot use multiple threads.)
- */
- static int
- init_importlib(PyThreadState *tstate, PyObject *sysmod)
- {
- assert(!_PyErr_Occurred(tstate));
- PyInterpreterState *interp = tstate->interp;
- int verbose = _PyInterpreterState_GetConfig(interp)->verbose;
- // Import _importlib through its frozen version, _frozen_importlib.
- if (verbose) {
- PySys_FormatStderr("import _frozen_importlib # frozen\n");
- }
- if (PyImport_ImportFrozenModule("_frozen_importlib") <= 0) {
- return -1;
- }
- PyObject *importlib = PyImport_AddModule("_frozen_importlib"); // borrowed
- if (importlib == NULL) {
- return -1;
- }
- IMPORTLIB(interp) = Py_NewRef(importlib);
- // Import the _imp module
- if (verbose) {
- PySys_FormatStderr("import _imp # builtin\n");
- }
- PyObject *imp_mod = bootstrap_imp(tstate);
- if (imp_mod == NULL) {
- return -1;
- }
- if (_PyImport_SetModuleString("_imp", imp_mod) < 0) {
- Py_DECREF(imp_mod);
- return -1;
- }
- // Install importlib as the implementation of import
- PyObject *value = PyObject_CallMethod(importlib, "_install",
- "OO", sysmod, imp_mod);
- Py_DECREF(imp_mod);
- if (value == NULL) {
- return -1;
- }
- Py_DECREF(value);
- assert(!_PyErr_Occurred(tstate));
- return 0;
- }
- static int
- init_importlib_external(PyInterpreterState *interp)
- {
- PyObject *value;
- value = PyObject_CallMethod(IMPORTLIB(interp),
- "_install_external_importers", "");
- if (value == NULL) {
- return -1;
- }
- Py_DECREF(value);
- value = PyImport_ImportModule("__res");
- if (value == NULL) {
- return -1;
- }
- Py_DECREF(value);
- return 0;
- }
- PyObject *
- _PyImport_GetImportlibLoader(PyInterpreterState *interp,
- const char *loader_name)
- {
- return PyObject_GetAttrString(IMPORTLIB(interp), loader_name);
- }
- PyObject *
- _PyImport_GetImportlibExternalLoader(PyInterpreterState *interp,
- const char *loader_name)
- {
- PyObject *bootstrap = PyObject_GetAttrString(IMPORTLIB(interp),
- "_bootstrap_external");
- if (bootstrap == NULL) {
- return NULL;
- }
- PyObject *loader_type = PyObject_GetAttrString(bootstrap, loader_name);
- Py_DECREF(bootstrap);
- return loader_type;
- }
- PyObject *
- _PyImport_BlessMyLoader(PyInterpreterState *interp, PyObject *module_globals)
- {
- PyObject *external = PyObject_GetAttrString(IMPORTLIB(interp),
- "_bootstrap_external");
- if (external == NULL) {
- return NULL;
- }
- PyObject *loader = PyObject_CallMethod(external, "_bless_my_loader",
- "O", module_globals, NULL);
- Py_DECREF(external);
- return loader;
- }
- PyObject *
- _PyImport_ImportlibModuleRepr(PyInterpreterState *interp, PyObject *m)
- {
- return PyObject_CallMethod(IMPORTLIB(interp), "_module_repr", "O", m);
- }
- /*******************/
- /* Return a finder object for a sys.path/pkg.__path__ item 'p',
- possibly by fetching it from the path_importer_cache dict. If it
- wasn't yet cached, traverse path_hooks until a hook is found
- that can handle the path item. Return None if no hook could;
- this tells our caller that the path based finder could not find
- a finder for this path item. Cache the result in
- path_importer_cache. */
- static PyObject *
- get_path_importer(PyThreadState *tstate, PyObject *path_importer_cache,
- PyObject *path_hooks, PyObject *p)
- {
- PyObject *importer;
- Py_ssize_t j, nhooks;
- if (!PyList_Check(path_hooks)) {
- PyErr_SetString(PyExc_RuntimeError, "sys.path_hooks is not a list");
- return NULL;
- }
- if (!PyDict_Check(path_importer_cache)) {
- PyErr_SetString(PyExc_RuntimeError, "sys.path_importer_cache is not a dict");
- return NULL;
- }
- nhooks = PyList_Size(path_hooks);
- if (nhooks < 0)
- return NULL; /* Shouldn't happen */
- importer = PyDict_GetItemWithError(path_importer_cache, p);
- if (importer != NULL || _PyErr_Occurred(tstate)) {
- return Py_XNewRef(importer);
- }
- /* set path_importer_cache[p] to None to avoid recursion */
- if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
- return NULL;
- for (j = 0; j < nhooks; j++) {
- PyObject *hook = PyList_GetItem(path_hooks, j);
- if (hook == NULL)
- return NULL;
- importer = PyObject_CallOneArg(hook, p);
- if (importer != NULL)
- break;
- if (!_PyErr_ExceptionMatches(tstate, PyExc_ImportError)) {
- return NULL;
- }
- _PyErr_Clear(tstate);
- }
- if (importer == NULL) {
- Py_RETURN_NONE;
- }
- if (PyDict_SetItem(path_importer_cache, p, importer) < 0) {
- Py_DECREF(importer);
- return NULL;
- }
- return importer;
- }
- PyObject *
- PyImport_GetImporter(PyObject *path)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *path_importer_cache = PySys_GetObject("path_importer_cache");
- if (path_importer_cache == NULL) {
- PyErr_SetString(PyExc_RuntimeError, "lost sys.path_importer_cache");
- return NULL;
- }
- Py_INCREF(path_importer_cache);
- PyObject *path_hooks = PySys_GetObject("path_hooks");
- if (path_hooks == NULL) {
- PyErr_SetString(PyExc_RuntimeError, "lost sys.path_hooks");
- Py_DECREF(path_importer_cache);
- return NULL;
- }
- Py_INCREF(path_hooks);
- PyObject *importer = get_path_importer(tstate, path_importer_cache, path_hooks, path);
- Py_DECREF(path_hooks);
- Py_DECREF(path_importer_cache);
- return importer;
- }
- /*********************/
- /* importing modules */
- /*********************/
- int
- _PyImport_InitDefaultImportFunc(PyInterpreterState *interp)
- {
- // Get the __import__ function
- PyObject *import_func = _PyDict_GetItemStringWithError(interp->builtins,
- "__import__");
- if (import_func == NULL) {
- return -1;
- }
- IMPORT_FUNC(interp) = Py_NewRef(import_func);
- return 0;
- }
- int
- _PyImport_IsDefaultImportFunc(PyInterpreterState *interp, PyObject *func)
- {
- return func == IMPORT_FUNC(interp);
- }
- /* Import a module, either built-in, frozen, or external, and return
- its module object WITH INCREMENTED REFERENCE COUNT */
- PyObject *
- PyImport_ImportModule(const char *name)
- {
- PyObject *pname;
- PyObject *result;
- pname = PyUnicode_FromString(name);
- if (pname == NULL)
- return NULL;
- result = PyImport_Import(pname);
- Py_DECREF(pname);
- return result;
- }
- /* Import a module without blocking
- *
- * At first it tries to fetch the module from sys.modules. If the module was
- * never loaded before it loads it with PyImport_ImportModule() unless another
- * thread holds the import lock. In the latter case the function raises an
- * ImportError instead of blocking.
- *
- * Returns the module object with incremented ref count.
- */
- PyObject *
- PyImport_ImportModuleNoBlock(const char *name)
- {
- return PyImport_ImportModule(name);
- }
- /* Remove importlib frames from the traceback,
- * except in Verbose mode. */
- static void
- remove_importlib_frames(PyThreadState *tstate)
- {
- const char *importlib_filename = "<frozen importlib._bootstrap>";
- const char *external_filename = "<frozen importlib._bootstrap_external>";
- const char *importer_filename = "library/python/runtime_py3/importer.pxi";
- const char *remove_frames = "_call_with_frames_removed";
- int always_trim = 0;
- int in_importlib = 0;
- PyObject **prev_link, **outer_link = NULL;
- PyObject *base_tb = NULL;
- /* Synopsis: if it's an ImportError, we trim all importlib chunks
- from the traceback. We always trim chunks
- which end with a call to "_call_with_frames_removed". */
- PyObject *exc = _PyErr_GetRaisedException(tstate);
- if (exc == NULL || _PyInterpreterState_GetConfig(tstate->interp)->verbose) {
- goto done;
- }
- if (PyType_IsSubtype(Py_TYPE(exc), (PyTypeObject *) PyExc_ImportError)) {
- always_trim = 1;
- }
- assert(PyExceptionInstance_Check(exc));
- base_tb = PyException_GetTraceback(exc);
- prev_link = &base_tb;
- PyObject *tb = base_tb;
- while (tb != NULL) {
- assert(PyTraceBack_Check(tb));
- PyTracebackObject *traceback = (PyTracebackObject *)tb;
- PyObject *next = (PyObject *) traceback->tb_next;
- PyFrameObject *frame = traceback->tb_frame;
- PyCodeObject *code = PyFrame_GetCode(frame);
- int now_in_importlib;
- now_in_importlib = _PyUnicode_EqualToASCIIString(code->co_filename, importlib_filename) ||
- _PyUnicode_EqualToASCIIString(code->co_filename, external_filename) ||
- _PyUnicode_EqualToASCIIString(code->co_filename, importer_filename);
- if (now_in_importlib && !in_importlib) {
- /* This is the link to this chunk of importlib tracebacks */
- outer_link = prev_link;
- }
- in_importlib = now_in_importlib;
- if (in_importlib &&
- (always_trim ||
- _PyUnicode_EqualToASCIIString(code->co_name, remove_frames))) {
- Py_XSETREF(*outer_link, Py_XNewRef(next));
- prev_link = outer_link;
- }
- else {
- prev_link = (PyObject **) &traceback->tb_next;
- }
- Py_DECREF(code);
- tb = next;
- }
- if (base_tb == NULL) {
- base_tb = Py_None;
- Py_INCREF(Py_None);
- }
- PyException_SetTraceback(exc, base_tb);
- done:
- Py_XDECREF(base_tb);
- _PyErr_SetRaisedException(tstate, exc);
- }
- static PyObject *
- resolve_name(PyThreadState *tstate, PyObject *name, PyObject *globals, int level)
- {
- PyObject *abs_name;
- PyObject *package = NULL;
- PyObject *spec;
- Py_ssize_t last_dot;
- PyObject *base;
- int level_up;
- if (globals == NULL) {
- _PyErr_SetString(tstate, PyExc_KeyError, "'__name__' not in globals");
- goto error;
- }
- if (!PyDict_Check(globals)) {
- _PyErr_SetString(tstate, PyExc_TypeError, "globals must be a dict");
- goto error;
- }
- package = PyDict_GetItemWithError(globals, &_Py_ID(__package__));
- if (package == Py_None) {
- package = NULL;
- }
- else if (package == NULL && _PyErr_Occurred(tstate)) {
- goto error;
- }
- spec = PyDict_GetItemWithError(globals, &_Py_ID(__spec__));
- if (spec == NULL && _PyErr_Occurred(tstate)) {
- goto error;
- }
- if (package != NULL) {
- Py_INCREF(package);
- if (!PyUnicode_Check(package)) {
- _PyErr_SetString(tstate, PyExc_TypeError,
- "package must be a string");
- goto error;
- }
- else if (spec != NULL && spec != Py_None) {
- int equal;
- PyObject *parent = PyObject_GetAttr(spec, &_Py_ID(parent));
- if (parent == NULL) {
- goto error;
- }
- equal = PyObject_RichCompareBool(package, parent, Py_EQ);
- Py_DECREF(parent);
- if (equal < 0) {
- goto error;
- }
- else if (equal == 0) {
- if (PyErr_WarnEx(PyExc_DeprecationWarning,
- "__package__ != __spec__.parent", 1) < 0) {
- goto error;
- }
- }
- }
- }
- else if (spec != NULL && spec != Py_None) {
- package = PyObject_GetAttr(spec, &_Py_ID(parent));
- if (package == NULL) {
- goto error;
- }
- else if (!PyUnicode_Check(package)) {
- _PyErr_SetString(tstate, PyExc_TypeError,
- "__spec__.parent must be a string");
- goto error;
- }
- }
- else {
- if (PyErr_WarnEx(PyExc_ImportWarning,
- "can't resolve package from __spec__ or __package__, "
- "falling back on __name__ and __path__", 1) < 0) {
- goto error;
- }
- package = PyDict_GetItemWithError(globals, &_Py_ID(__name__));
- if (package == NULL) {
- if (!_PyErr_Occurred(tstate)) {
- _PyErr_SetString(tstate, PyExc_KeyError,
- "'__name__' not in globals");
- }
- goto error;
- }
- Py_INCREF(package);
- if (!PyUnicode_Check(package)) {
- _PyErr_SetString(tstate, PyExc_TypeError,
- "__name__ must be a string");
- goto error;
- }
- int haspath = PyDict_Contains(globals, &_Py_ID(__path__));
- if (haspath < 0) {
- goto error;
- }
- if (!haspath) {
- Py_ssize_t dot;
- if (PyUnicode_READY(package) < 0) {
- goto error;
- }
- dot = PyUnicode_FindChar(package, '.',
- 0, PyUnicode_GET_LENGTH(package), -1);
- if (dot == -2) {
- goto error;
- }
- else if (dot == -1) {
- goto no_parent_error;
- }
- PyObject *substr = PyUnicode_Substring(package, 0, dot);
- if (substr == NULL) {
- goto error;
- }
- Py_SETREF(package, substr);
- }
- }
- last_dot = PyUnicode_GET_LENGTH(package);
- if (last_dot == 0) {
- goto no_parent_error;
- }
- for (level_up = 1; level_up < level; level_up += 1) {
- last_dot = PyUnicode_FindChar(package, '.', 0, last_dot, -1);
- if (last_dot == -2) {
- goto error;
- }
- else if (last_dot == -1) {
- _PyErr_SetString(tstate, PyExc_ImportError,
- "attempted relative import beyond top-level "
- "package");
- goto error;
- }
- }
- base = PyUnicode_Substring(package, 0, last_dot);
- Py_DECREF(package);
- if (base == NULL || PyUnicode_GET_LENGTH(name) == 0) {
- return base;
- }
- abs_name = PyUnicode_FromFormat("%U.%U", base, name);
- Py_DECREF(base);
- return abs_name;
- no_parent_error:
- _PyErr_SetString(tstate, PyExc_ImportError,
- "attempted relative import "
- "with no known parent package");
- error:
- Py_XDECREF(package);
- return NULL;
- }
- static PyObject *
- import_find_and_load(PyThreadState *tstate, PyObject *abs_name)
- {
- PyObject *mod = NULL;
- PyInterpreterState *interp = tstate->interp;
- int import_time = _PyInterpreterState_GetConfig(interp)->import_time;
- #define import_level FIND_AND_LOAD(interp).import_level
- #define accumulated FIND_AND_LOAD(interp).accumulated
- _PyTime_t t1 = 0, accumulated_copy = accumulated;
- PyObject *sys_path = PySys_GetObject("path");
- PyObject *sys_meta_path = PySys_GetObject("meta_path");
- PyObject *sys_path_hooks = PySys_GetObject("path_hooks");
- if (_PySys_Audit(tstate, "import", "OOOOO",
- abs_name, Py_None, sys_path ? sys_path : Py_None,
- sys_meta_path ? sys_meta_path : Py_None,
- sys_path_hooks ? sys_path_hooks : Py_None) < 0) {
- return NULL;
- }
- /* XOptions is initialized after first some imports.
- * So we can't have negative cache before completed initialization.
- * Anyway, importlib._find_and_load is much slower than
- * _PyDict_GetItemIdWithError().
- */
- if (import_time) {
- #define header FIND_AND_LOAD(interp).header
- if (header) {
- fputs("import time: self [us] | cumulative | imported package\n",
- stderr);
- header = 0;
- }
- #undef header
- import_level++;
- t1 = _PyTime_GetPerfCounter();
- accumulated = 0;
- }
- if (PyDTrace_IMPORT_FIND_LOAD_START_ENABLED())
- PyDTrace_IMPORT_FIND_LOAD_START(PyUnicode_AsUTF8(abs_name));
- mod = PyObject_CallMethodObjArgs(IMPORTLIB(interp), &_Py_ID(_find_and_load),
- abs_name, IMPORT_FUNC(interp), NULL);
- if (PyDTrace_IMPORT_FIND_LOAD_DONE_ENABLED())
- PyDTrace_IMPORT_FIND_LOAD_DONE(PyUnicode_AsUTF8(abs_name),
- mod != NULL);
- if (import_time) {
- _PyTime_t cum = _PyTime_GetPerfCounter() - t1;
- import_level--;
- fprintf(stderr, "import time: %9ld | %10ld | %*s%s\n",
- (long)_PyTime_AsMicroseconds(cum - accumulated, _PyTime_ROUND_CEILING),
- (long)_PyTime_AsMicroseconds(cum, _PyTime_ROUND_CEILING),
- import_level*2, "", PyUnicode_AsUTF8(abs_name));
- accumulated = accumulated_copy + cum;
- }
- return mod;
- #undef import_level
- #undef accumulated
- }
- PyObject *
- PyImport_ImportModuleLevelObject(PyObject *name, PyObject *globals,
- PyObject *locals, PyObject *fromlist,
- int level)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *abs_name = NULL;
- PyObject *final_mod = NULL;
- PyObject *mod = NULL;
- PyObject *package = NULL;
- PyInterpreterState *interp = tstate->interp;
- int has_from;
- if (name == NULL) {
- _PyErr_SetString(tstate, PyExc_ValueError, "Empty module name");
- goto error;
- }
- /* The below code is importlib.__import__() & _gcd_import(), ported to C
- for added performance. */
- if (!PyUnicode_Check(name)) {
- _PyErr_SetString(tstate, PyExc_TypeError,
- "module name must be a string");
- goto error;
- }
- if (PyUnicode_READY(name) < 0) {
- goto error;
- }
- if (level < 0) {
- _PyErr_SetString(tstate, PyExc_ValueError, "level must be >= 0");
- goto error;
- }
- if (level > 0) {
- abs_name = resolve_name(tstate, name, globals, level);
- if (abs_name == NULL)
- goto error;
- }
- else { /* level == 0 */
- if (PyUnicode_GET_LENGTH(name) == 0) {
- _PyErr_SetString(tstate, PyExc_ValueError, "Empty module name");
- goto error;
- }
- abs_name = Py_NewRef(name);
- }
- mod = import_get_module(tstate, abs_name);
- if (mod == NULL && _PyErr_Occurred(tstate)) {
- goto error;
- }
- if (mod != NULL && mod != Py_None) {
- if (import_ensure_initialized(tstate->interp, mod, abs_name) < 0) {
- goto error;
- }
- }
- else {
- Py_XDECREF(mod);
- mod = import_find_and_load(tstate, abs_name);
- if (mod == NULL) {
- goto error;
- }
- }
- has_from = 0;
- if (fromlist != NULL && fromlist != Py_None) {
- has_from = PyObject_IsTrue(fromlist);
- if (has_from < 0)
- goto error;
- }
- if (!has_from) {
- Py_ssize_t len = PyUnicode_GET_LENGTH(name);
- if (level == 0 || len > 0) {
- Py_ssize_t dot;
- dot = PyUnicode_FindChar(name, '.', 0, len, 1);
- if (dot == -2) {
- goto error;
- }
- if (dot == -1) {
- /* No dot in module name, simple exit */
- final_mod = Py_NewRef(mod);
- goto error;
- }
- if (level == 0) {
- PyObject *front = PyUnicode_Substring(name, 0, dot);
- if (front == NULL) {
- goto error;
- }
- final_mod = PyImport_ImportModuleLevelObject(front, NULL, NULL, NULL, 0);
- Py_DECREF(front);
- }
- else {
- Py_ssize_t cut_off = len - dot;
- Py_ssize_t abs_name_len = PyUnicode_GET_LENGTH(abs_name);
- PyObject *to_return = PyUnicode_Substring(abs_name, 0,
- abs_name_len - cut_off);
- if (to_return == NULL) {
- goto error;
- }
- final_mod = import_get_module(tstate, to_return);
- Py_DECREF(to_return);
- if (final_mod == NULL) {
- if (!_PyErr_Occurred(tstate)) {
- _PyErr_Format(tstate, PyExc_KeyError,
- "%R not in sys.modules as expected",
- to_return);
- }
- goto error;
- }
- }
- }
- else {
- final_mod = Py_NewRef(mod);
- }
- }
- else {
- PyObject *path;
- if (_PyObject_LookupAttr(mod, &_Py_ID(__path__), &path) < 0) {
- goto error;
- }
- if (path) {
- Py_DECREF(path);
- final_mod = PyObject_CallMethodObjArgs(
- IMPORTLIB(interp), &_Py_ID(_handle_fromlist),
- mod, fromlist, IMPORT_FUNC(interp), NULL);
- }
- else {
- final_mod = Py_NewRef(mod);
- }
- }
- error:
- Py_XDECREF(abs_name);
- Py_XDECREF(mod);
- Py_XDECREF(package);
- if (final_mod == NULL) {
- remove_importlib_frames(tstate);
- }
- return final_mod;
- }
- PyObject *
- PyImport_ImportModuleLevel(const char *name, PyObject *globals, PyObject *locals,
- PyObject *fromlist, int level)
- {
- PyObject *nameobj, *mod;
- nameobj = PyUnicode_FromString(name);
- if (nameobj == NULL)
- return NULL;
- mod = PyImport_ImportModuleLevelObject(nameobj, globals, locals,
- fromlist, level);
- Py_DECREF(nameobj);
- return mod;
- }
- /* Re-import a module of any kind and return its module object, WITH
- INCREMENTED REFERENCE COUNT */
- PyObject *
- PyImport_ReloadModule(PyObject *m)
- {
- PyObject *reloaded_module = NULL;
- PyObject *importlib = PyImport_GetModule(&_Py_ID(importlib));
- if (importlib == NULL) {
- if (PyErr_Occurred()) {
- return NULL;
- }
- importlib = PyImport_ImportModule("importlib");
- if (importlib == NULL) {
- return NULL;
- }
- }
- reloaded_module = PyObject_CallMethodOneArg(importlib, &_Py_ID(reload), m);
- Py_DECREF(importlib);
- return reloaded_module;
- }
- /* Higher-level import emulator which emulates the "import" statement
- more accurately -- it invokes the __import__() function from the
- builtins of the current globals. This means that the import is
- done using whatever import hooks are installed in the current
- environment.
- A dummy list ["__doc__"] is passed as the 4th argument so that
- e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
- will return <module "gencache"> instead of <module "win32com">. */
- PyObject *
- PyImport_Import(PyObject *module_name)
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *globals = NULL;
- PyObject *import = NULL;
- PyObject *builtins = NULL;
- PyObject *r = NULL;
- PyObject *from_list = PyList_New(0);
- if (from_list == NULL) {
- goto err;
- }
- /* Get the builtins from current globals */
- globals = PyEval_GetGlobals();
- if (globals != NULL) {
- Py_INCREF(globals);
- builtins = PyObject_GetItem(globals, &_Py_ID(__builtins__));
- if (builtins == NULL)
- goto err;
- }
- else {
- /* No globals -- use standard builtins, and fake globals */
- builtins = PyImport_ImportModuleLevel("builtins",
- NULL, NULL, NULL, 0);
- if (builtins == NULL) {
- goto err;
- }
- globals = Py_BuildValue("{OO}", &_Py_ID(__builtins__), builtins);
- if (globals == NULL)
- goto err;
- }
- /* Get the __import__ function from the builtins */
- if (PyDict_Check(builtins)) {
- import = PyObject_GetItem(builtins, &_Py_ID(__import__));
- if (import == NULL) {
- _PyErr_SetObject(tstate, PyExc_KeyError, &_Py_ID(__import__));
- }
- }
- else
- import = PyObject_GetAttr(builtins, &_Py_ID(__import__));
- if (import == NULL)
- goto err;
- /* Call the __import__ function with the proper argument list
- Always use absolute import here.
- Calling for side-effect of import. */
- r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
- globals, from_list, 0, NULL);
- if (r == NULL)
- goto err;
- Py_DECREF(r);
- r = import_get_module(tstate, module_name);
- if (r == NULL && !_PyErr_Occurred(tstate)) {
- _PyErr_SetObject(tstate, PyExc_KeyError, module_name);
- }
- err:
- Py_XDECREF(globals);
- Py_XDECREF(builtins);
- Py_XDECREF(import);
- Py_XDECREF(from_list);
- return r;
- }
- /*********************/
- /* runtime lifecycle */
- /*********************/
- PyStatus
- _PyImport_Init(void)
- {
- if (INITTAB != NULL) {
- return _PyStatus_ERR("global import state already initialized");
- }
- PyStatus status = _PyStatus_OK();
- /* Force default raw memory allocator to get a known allocator to be able
- to release the memory in _PyImport_Fini() */
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- if (init_builtin_modules_table() != 0) {
- status = PyStatus_NoMemory();
- goto done;
- }
- done:
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- return status;
- }
- void
- _PyImport_Fini(void)
- {
- /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
- // XXX Should we actually leave them (mostly) intact, since we don't
- // ever dlclose() the module files?
- _extensions_cache_clear_all();
- /* Use the same memory allocator as _PyImport_Init(). */
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- /* Free memory allocated by _PyImport_Init() */
- fini_builtin_modules_table();
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- }
- void
- _PyImport_Fini2(void)
- {
- /* Use the same memory allocator than PyImport_ExtendInittab(). */
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- // Reset PyImport_Inittab
- PyImport_Inittab = _PyImport_Inittab;
- /* Free memory allocated by PyImport_ExtendInittab() */
- PyMem_RawFree(inittab_copy);
- inittab_copy = NULL;
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- }
- /*************************/
- /* interpreter lifecycle */
- /*************************/
- PyStatus
- _PyImport_InitCore(PyThreadState *tstate, PyObject *sysmod, int importlib)
- {
- // XXX Initialize here: interp->modules and interp->import_func.
- // XXX Initialize here: sys.modules and sys.meta_path.
- if (importlib) {
- /* This call sets up builtin and frozen import support */
- if (init_importlib(tstate, sysmod) < 0) {
- return _PyStatus_ERR("failed to initialize importlib");
- }
- }
- return _PyStatus_OK();
- }
- /* In some corner cases it is important to be sure that the import
- machinery has been initialized (or not cleaned up yet). For
- example, see issue #4236 and PyModule_Create2(). */
- int
- _PyImport_IsInitialized(PyInterpreterState *interp)
- {
- if (MODULES(interp) == NULL)
- return 0;
- return 1;
- }
- /* Clear the direct per-interpreter import state, if not cleared already. */
- void
- _PyImport_ClearCore(PyInterpreterState *interp)
- {
- /* interp->modules should have been cleaned up and cleared already
- by _PyImport_FiniCore(). */
- Py_CLEAR(MODULES(interp));
- Py_CLEAR(MODULES_BY_INDEX(interp));
- Py_CLEAR(IMPORTLIB(interp));
- Py_CLEAR(IMPORT_FUNC(interp));
- }
- void
- _PyImport_FiniCore(PyInterpreterState *interp)
- {
- int verbose = _PyInterpreterState_GetConfig(interp)->verbose;
- if (_PySys_ClearAttrString(interp, "meta_path", verbose) < 0) {
- PyErr_WriteUnraisable(NULL);
- }
- // XXX Pull in most of finalize_modules() in pylifecycle.c.
- if (_PySys_ClearAttrString(interp, "modules", verbose) < 0) {
- PyErr_WriteUnraisable(NULL);
- }
- if (IMPORT_LOCK(interp) != NULL) {
- PyThread_free_lock(IMPORT_LOCK(interp));
- IMPORT_LOCK(interp) = NULL;
- }
- _PyImport_ClearCore(interp);
- }
- // XXX Add something like _PyImport_Disable() for use early in interp fini?
- /* "external" imports */
- static int
- init_zipimport(PyThreadState *tstate, int verbose)
- {
- PyObject *path_hooks = PySys_GetObject("path_hooks");
- if (path_hooks == NULL) {
- _PyErr_SetString(tstate, PyExc_RuntimeError,
- "unable to get sys.path_hooks");
- return -1;
- }
- if (verbose) {
- PySys_WriteStderr("# installing zipimport hook\n");
- }
- PyObject *zipimporter = _PyImport_GetModuleAttrString("zipimport", "zipimporter");
- if (zipimporter == NULL) {
- _PyErr_Clear(tstate); /* No zipimporter object -- okay */
- if (verbose) {
- PySys_WriteStderr("# can't import zipimport.zipimporter\n");
- }
- }
- else {
- /* sys.path_hooks.insert(0, zipimporter) */
- int err = PyList_Insert(path_hooks, 0, zipimporter);
- Py_DECREF(zipimporter);
- if (err < 0) {
- return -1;
- }
- if (verbose) {
- PySys_WriteStderr("# installed zipimport hook\n");
- }
- }
- return 0;
- }
- PyStatus
- _PyImport_InitExternal(PyThreadState *tstate)
- {
- int verbose = _PyInterpreterState_GetConfig(tstate->interp)->verbose;
- // XXX Initialize here: sys.path_hooks and sys.path_importer_cache.
- if (init_importlib_external(tstate->interp) != 0) {
- _PyErr_Print(tstate);
- return _PyStatus_ERR("external importer setup failed");
- }
- if (init_zipimport(tstate, verbose) != 0) {
- PyErr_Print();
- return _PyStatus_ERR("initializing zipimport failed");
- }
- return _PyStatus_OK();
- }
- void
- _PyImport_FiniExternal(PyInterpreterState *interp)
- {
- int verbose = _PyInterpreterState_GetConfig(interp)->verbose;
- // XXX Uninstall importlib metapath importers here?
- if (_PySys_ClearAttrString(interp, "path_importer_cache", verbose) < 0) {
- PyErr_WriteUnraisable(NULL);
- }
- if (_PySys_ClearAttrString(interp, "path_hooks", verbose) < 0) {
- PyErr_WriteUnraisable(NULL);
- }
- }
- /******************/
- /* module helpers */
- /******************/
- PyObject *
- _PyImport_GetModuleAttr(PyObject *modname, PyObject *attrname)
- {
- PyObject *mod = PyImport_Import(modname);
- if (mod == NULL) {
- return NULL;
- }
- PyObject *result = PyObject_GetAttr(mod, attrname);
- Py_DECREF(mod);
- return result;
- }
- PyObject *
- _PyImport_GetModuleAttrString(const char *modname, const char *attrname)
- {
- PyObject *pmodname = PyUnicode_FromString(modname);
- if (pmodname == NULL) {
- return NULL;
- }
- PyObject *pattrname = PyUnicode_FromString(attrname);
- if (pattrname == NULL) {
- Py_DECREF(pmodname);
- return NULL;
- }
- PyObject *result = _PyImport_GetModuleAttr(pmodname, pattrname);
- Py_DECREF(pattrname);
- Py_DECREF(pmodname);
- return result;
- }
- /**************/
- /* the module */
- /**************/
- /*[clinic input]
- _imp.lock_held
- Return True if the import lock is currently held, else False.
- On platforms without threads, return False.
- [clinic start generated code]*/
- static PyObject *
- _imp_lock_held_impl(PyObject *module)
- /*[clinic end generated code: output=8b89384b5e1963fc input=9b088f9b217d9bdf]*/
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- return PyBool_FromLong(
- IMPORT_LOCK_THREAD(interp) != PYTHREAD_INVALID_THREAD_ID);
- }
- /*[clinic input]
- _imp.acquire_lock
- Acquires the interpreter's import lock for the current thread.
- This lock should be used by import hooks to ensure thread-safety when importing
- modules. On platforms without threads, this function does nothing.
- [clinic start generated code]*/
- static PyObject *
- _imp_acquire_lock_impl(PyObject *module)
- /*[clinic end generated code: output=1aff58cb0ee1b026 input=4a2d4381866d5fdc]*/
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- _PyImport_AcquireLock(interp);
- Py_RETURN_NONE;
- }
- /*[clinic input]
- _imp.release_lock
- Release the interpreter's import lock.
- On platforms without threads, this function does nothing.
- [clinic start generated code]*/
- static PyObject *
- _imp_release_lock_impl(PyObject *module)
- /*[clinic end generated code: output=7faab6d0be178b0a input=934fb11516dd778b]*/
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (_PyImport_ReleaseLock(interp) < 0) {
- PyErr_SetString(PyExc_RuntimeError,
- "not holding the import lock");
- return NULL;
- }
- Py_RETURN_NONE;
- }
- /*[clinic input]
- _imp._fix_co_filename
- code: object(type="PyCodeObject *", subclass_of="&PyCode_Type")
- Code object to change.
- path: unicode
- File path to use.
- /
- Changes code.co_filename to specify the passed-in file path.
- [clinic start generated code]*/
- static PyObject *
- _imp__fix_co_filename_impl(PyObject *module, PyCodeObject *code,
- PyObject *path)
- /*[clinic end generated code: output=1d002f100235587d input=895ba50e78b82f05]*/
- {
- update_compiled_module(code, path);
- Py_RETURN_NONE;
- }
- /*[clinic input]
- _imp.create_builtin
- spec: object
- /
- Create an extension module.
- [clinic start generated code]*/
- static PyObject *
- _imp_create_builtin(PyObject *module, PyObject *spec)
- /*[clinic end generated code: output=ace7ff22271e6f39 input=37f966f890384e47]*/
- {
- PyThreadState *tstate = _PyThreadState_GET();
- PyObject *name = PyObject_GetAttrString(spec, "name");
- if (name == NULL) {
- return NULL;
- }
- if (!PyUnicode_Check(name)) {
- PyErr_Format(PyExc_TypeError,
- "name must be string, not %.200s",
- Py_TYPE(name)->tp_name);
- Py_DECREF(name);
- return NULL;
- }
- PyObject *mod = create_builtin(tstate, name, spec);
- Py_DECREF(name);
- return mod;
- }
- /*[clinic input]
- _imp.extension_suffixes
- Returns the list of file suffixes used to identify extension modules.
- [clinic start generated code]*/
- static PyObject *
- _imp_extension_suffixes_impl(PyObject *module)
- /*[clinic end generated code: output=0bf346e25a8f0cd3 input=ecdeeecfcb6f839e]*/
- {
- PyObject *list;
- list = PyList_New(0);
- if (list == NULL)
- return NULL;
- #ifdef HAVE_DYNAMIC_LOADING
- const char *suffix;
- unsigned int index = 0;
- while ((suffix = _PyImport_DynLoadFiletab[index])) {
- PyObject *item = PyUnicode_FromString(suffix);
- if (item == NULL) {
- Py_DECREF(list);
- return NULL;
- }
- if (PyList_Append(list, item) < 0) {
- Py_DECREF(list);
- Py_DECREF(item);
- return NULL;
- }
- Py_DECREF(item);
- index += 1;
- }
- #endif
- return list;
- }
- /*[clinic input]
- _imp.init_frozen
- name: unicode
- /
- Initializes a frozen module.
- [clinic start generated code]*/
- static PyObject *
- _imp_init_frozen_impl(PyObject *module, PyObject *name)
- /*[clinic end generated code: output=fc0511ed869fd69c input=13019adfc04f3fb3]*/
- {
- PyThreadState *tstate = _PyThreadState_GET();
- int ret;
- ret = PyImport_ImportFrozenModuleObject(name);
- if (ret < 0)
- return NULL;
- if (ret == 0) {
- Py_RETURN_NONE;
- }
- return import_add_module(tstate, name);
- }
- /*[clinic input]
- _imp.find_frozen
- name: unicode
- /
- *
- withdata: bool = False
- Return info about the corresponding frozen module (if there is one) or None.
- The returned info (a 2-tuple):
- * data the raw marshalled bytes
- * is_package whether or not it is a package
- * origname the originally frozen module's name, or None if not
- a stdlib module (this will usually be the same as
- the module's current name)
- [clinic start generated code]*/
- static PyObject *
- _imp_find_frozen_impl(PyObject *module, PyObject *name, int withdata)
- /*[clinic end generated code: output=8c1c3c7f925397a5 input=22a8847c201542fd]*/
- {
- struct frozen_info info;
- frozen_status status = find_frozen(name, &info);
- if (status == FROZEN_NOT_FOUND || status == FROZEN_DISABLED) {
- Py_RETURN_NONE;
- }
- else if (status == FROZEN_BAD_NAME) {
- Py_RETURN_NONE;
- }
- else if (status != FROZEN_OKAY) {
- set_frozen_error(status, name);
- return NULL;
- }
- PyObject *data = NULL;
- if (withdata) {
- data = PyMemoryView_FromMemory((char *)info.data, info.size, PyBUF_READ);
- if (data == NULL) {
- return NULL;
- }
- }
- PyObject *origname = NULL;
- if (info.origname != NULL && info.origname[0] != '\0') {
- origname = PyUnicode_FromString(info.origname);
- if (origname == NULL) {
- Py_XDECREF(data);
- return NULL;
- }
- }
- PyObject *result = PyTuple_Pack(3, data ? data : Py_None,
- info.is_package ? Py_True : Py_False,
- origname ? origname : Py_None);
- Py_XDECREF(origname);
- Py_XDECREF(data);
- return result;
- }
- /*[clinic input]
- _imp.get_frozen_object
- name: unicode
- data as dataobj: object = None
- /
- Create a code object for a frozen module.
- [clinic start generated code]*/
- static PyObject *
- _imp_get_frozen_object_impl(PyObject *module, PyObject *name,
- PyObject *dataobj)
- /*[clinic end generated code: output=54368a673a35e745 input=034bdb88f6460b7b]*/
- {
- struct frozen_info info = {0};
- Py_buffer buf = {0};
- if (PyObject_CheckBuffer(dataobj)) {
- if (PyObject_GetBuffer(dataobj, &buf, PyBUF_SIMPLE) != 0) {
- return NULL;
- }
- info.data = (const char *)buf.buf;
- info.size = buf.len;
- }
- else if (dataobj != Py_None) {
- _PyArg_BadArgument("get_frozen_object", "argument 2", "bytes", dataobj);
- return NULL;
- }
- else {
- frozen_status status = find_frozen(name, &info);
- if (status != FROZEN_OKAY) {
- set_frozen_error(status, name);
- return NULL;
- }
- }
- if (info.nameobj == NULL) {
- info.nameobj = name;
- }
- if (info.size == 0 && info.get_code == NULL) {
- /* Does not contain executable code. */
- set_frozen_error(FROZEN_INVALID, name);
- return NULL;
- }
- PyInterpreterState *interp = _PyInterpreterState_GET();
- PyObject *codeobj = unmarshal_frozen_code(interp, &info);
- if (dataobj != Py_None) {
- PyBuffer_Release(&buf);
- }
- return codeobj;
- }
- /*[clinic input]
- _imp.is_frozen_package
- name: unicode
- /
- Returns True if the module name is of a frozen package.
- [clinic start generated code]*/
- static PyObject *
- _imp_is_frozen_package_impl(PyObject *module, PyObject *name)
- /*[clinic end generated code: output=e70cbdb45784a1c9 input=81b6cdecd080fbb8]*/
- {
- struct frozen_info info;
- frozen_status status = find_frozen(name, &info);
- if (status != FROZEN_OKAY && status != FROZEN_EXCLUDED) {
- set_frozen_error(status, name);
- return NULL;
- }
- return PyBool_FromLong(info.is_package);
- }
- /*[clinic input]
- _imp.is_builtin
- name: unicode
- /
- Returns True if the module name corresponds to a built-in module.
- [clinic start generated code]*/
- static PyObject *
- _imp_is_builtin_impl(PyObject *module, PyObject *name)
- /*[clinic end generated code: output=3bfd1162e2d3be82 input=86befdac021dd1c7]*/
- {
- return PyLong_FromLong(is_builtin(name));
- }
- /*[clinic input]
- _imp.is_frozen
- name: unicode
- /
- Returns True if the module name corresponds to a frozen module.
- [clinic start generated code]*/
- static PyObject *
- _imp_is_frozen_impl(PyObject *module, PyObject *name)
- /*[clinic end generated code: output=01f408f5ec0f2577 input=7301dbca1897d66b]*/
- {
- struct frozen_info info;
- frozen_status status = find_frozen(name, &info);
- if (status != FROZEN_OKAY) {
- Py_RETURN_FALSE;
- }
- Py_RETURN_TRUE;
- }
- /*[clinic input]
- _imp._frozen_module_names
- Returns the list of available frozen modules.
- [clinic start generated code]*/
- static PyObject *
- _imp__frozen_module_names_impl(PyObject *module)
- /*[clinic end generated code: output=80609ef6256310a8 input=76237fbfa94460d2]*/
- {
- return list_frozen_module_names();
- }
- /*[clinic input]
- _imp._override_frozen_modules_for_tests
- override: int
- /
- (internal-only) Override PyConfig.use_frozen_modules.
- (-1: "off", 1: "on", 0: no override)
- See frozen_modules() in Lib/test/support/import_helper.py.
- [clinic start generated code]*/
- static PyObject *
- _imp__override_frozen_modules_for_tests_impl(PyObject *module, int override)
- /*[clinic end generated code: output=36d5cb1594160811 input=8f1f95a3ef21aec3]*/
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- OVERRIDE_FROZEN_MODULES(interp) = override;
- Py_RETURN_NONE;
- }
- /*[clinic input]
- _imp._override_multi_interp_extensions_check
- override: int
- /
- (internal-only) Override PyInterpreterConfig.check_multi_interp_extensions.
- (-1: "never", 1: "always", 0: no override)
- [clinic start generated code]*/
- static PyObject *
- _imp__override_multi_interp_extensions_check_impl(PyObject *module,
- int override)
- /*[clinic end generated code: output=3ff043af52bbf280 input=e086a2ea181f92ae]*/
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (_Py_IsMainInterpreter(interp)) {
- PyErr_SetString(PyExc_RuntimeError,
- "_imp._override_multi_interp_extensions_check() "
- "cannot be used in the main interpreter");
- return NULL;
- }
- int oldvalue = OVERRIDE_MULTI_INTERP_EXTENSIONS_CHECK(interp);
- OVERRIDE_MULTI_INTERP_EXTENSIONS_CHECK(interp) = override;
- return PyLong_FromLong(oldvalue);
- }
- #ifdef HAVE_DYNAMIC_LOADING
- /*[clinic input]
- _imp.create_dynamic
- spec: object
- file: object = NULL
- /
- Create an extension module.
- [clinic start generated code]*/
- static PyObject *
- _imp_create_dynamic_impl(PyObject *module, PyObject *spec, PyObject *file)
- /*[clinic end generated code: output=83249b827a4fde77 input=c31b954f4cf4e09d]*/
- {
- PyObject *mod, *name, *path;
- FILE *fp;
- name = PyObject_GetAttrString(spec, "name");
- if (name == NULL) {
- return NULL;
- }
- path = PyObject_GetAttrString(spec, "origin");
- if (path == NULL) {
- Py_DECREF(name);
- return NULL;
- }
- PyThreadState *tstate = _PyThreadState_GET();
- mod = import_find_extension(tstate, name, path);
- if (mod != NULL || _PyErr_Occurred(tstate)) {
- assert(mod == NULL || !_PyErr_Occurred(tstate));
- goto finally;
- }
- if (file != NULL) {
- fp = _Py_fopen_obj(path, "r");
- if (fp == NULL) {
- goto finally;
- }
- }
- else
- fp = NULL;
- mod = _PyImport_LoadDynamicModuleWithSpec(spec, fp);
- if (fp)
- fclose(fp);
- finally:
- Py_DECREF(name);
- Py_DECREF(path);
- return mod;
- }
- /*[clinic input]
- _imp.exec_dynamic -> int
- mod: object
- /
- Initialize an extension module.
- [clinic start generated code]*/
- static int
- _imp_exec_dynamic_impl(PyObject *module, PyObject *mod)
- /*[clinic end generated code: output=f5720ac7b465877d input=9fdbfcb250280d3a]*/
- {
- return exec_builtin_or_dynamic(mod);
- }
- #endif /* HAVE_DYNAMIC_LOADING */
- /*[clinic input]
- _imp.exec_builtin -> int
- mod: object
- /
- Initialize a built-in module.
- [clinic start generated code]*/
- static int
- _imp_exec_builtin_impl(PyObject *module, PyObject *mod)
- /*[clinic end generated code: output=0262447b240c038e input=7beed5a2f12a60ca]*/
- {
- return exec_builtin_or_dynamic(mod);
- }
- /*[clinic input]
- _imp.source_hash
- key: long
- source: Py_buffer
- [clinic start generated code]*/
- static PyObject *
- _imp_source_hash_impl(PyObject *module, long key, Py_buffer *source)
- /*[clinic end generated code: output=edb292448cf399ea input=9aaad1e590089789]*/
- {
- union {
- uint64_t x;
- char data[sizeof(uint64_t)];
- } hash;
- hash.x = _Py_KeyedHash((uint64_t)key, source->buf, source->len);
- #if !PY_LITTLE_ENDIAN
- // Force to little-endian. There really ought to be a succinct standard way
- // to do this.
- for (size_t i = 0; i < sizeof(hash.data)/2; i++) {
- char tmp = hash.data[i];
- hash.data[i] = hash.data[sizeof(hash.data) - i - 1];
- hash.data[sizeof(hash.data) - i - 1] = tmp;
- }
- #endif
- return PyBytes_FromStringAndSize(hash.data, sizeof(hash.data));
- }
- PyDoc_STRVAR(doc_imp,
- "(Extremely) low-level import machinery bits as used by importlib.");
- static PyMethodDef imp_methods[] = {
- _IMP_EXTENSION_SUFFIXES_METHODDEF
- _IMP_LOCK_HELD_METHODDEF
- _IMP_ACQUIRE_LOCK_METHODDEF
- _IMP_RELEASE_LOCK_METHODDEF
- _IMP_FIND_FROZEN_METHODDEF
- _IMP_GET_FROZEN_OBJECT_METHODDEF
- _IMP_IS_FROZEN_PACKAGE_METHODDEF
- _IMP_CREATE_BUILTIN_METHODDEF
- _IMP_INIT_FROZEN_METHODDEF
- _IMP_IS_BUILTIN_METHODDEF
- _IMP_IS_FROZEN_METHODDEF
- _IMP__FROZEN_MODULE_NAMES_METHODDEF
- _IMP__OVERRIDE_FROZEN_MODULES_FOR_TESTS_METHODDEF
- _IMP__OVERRIDE_MULTI_INTERP_EXTENSIONS_CHECK_METHODDEF
- _IMP_CREATE_DYNAMIC_METHODDEF
- _IMP_EXEC_DYNAMIC_METHODDEF
- _IMP_EXEC_BUILTIN_METHODDEF
- _IMP__FIX_CO_FILENAME_METHODDEF
- _IMP_SOURCE_HASH_METHODDEF
- {NULL, NULL} /* sentinel */
- };
- static int
- imp_module_exec(PyObject *module)
- {
- const wchar_t *mode = _Py_GetConfig()->check_hash_pycs_mode;
- PyObject *pyc_mode = PyUnicode_FromWideChar(mode, -1);
- if (pyc_mode == NULL) {
- return -1;
- }
- if (PyModule_AddObjectRef(module, "check_hash_based_pycs", pyc_mode) < 0) {
- Py_DECREF(pyc_mode);
- return -1;
- }
- Py_DECREF(pyc_mode);
- return 0;
- }
- static PyModuleDef_Slot imp_slots[] = {
- {Py_mod_exec, imp_module_exec},
- {Py_mod_multiple_interpreters, Py_MOD_PER_INTERPRETER_GIL_SUPPORTED},
- {0, NULL}
- };
- static struct PyModuleDef imp_module = {
- PyModuleDef_HEAD_INIT,
- .m_name = "_imp",
- .m_doc = doc_imp,
- .m_size = 0,
- .m_methods = imp_methods,
- .m_slots = imp_slots,
- };
- PyMODINIT_FUNC
- PyInit__imp(void)
- {
- return PyModuleDef_Init(&imp_module);
- }
- #ifdef __cplusplus
- }
- #endif
|