123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180 |
- #include "Python.h"
- #include "pycore_fileutils.h" // _Py_HasFileSystemDefaultEncodeErrors
- #include "pycore_getopt.h" // _PyOS_GetOpt()
- #include "pycore_initconfig.h" // _PyStatus_OK()
- #include "pycore_interp.h" // _PyInterpreterState.runtime
- #include "pycore_long.h" // _PY_LONG_MAX_STR_DIGITS_THRESHOLD
- #include "pycore_pathconfig.h" // _Py_path_config
- #include "pycore_pyerrors.h" // _PyErr_GetRaisedException()
- #include "pycore_pylifecycle.h" // _Py_PreInitializeFromConfig()
- #include "pycore_pymem.h" // _PyMem_SetDefaultAllocator()
- #include "pycore_pystate.h" // _PyThreadState_GET()
- #include "osdefs.h" // DELIM
- #include <locale.h> // setlocale()
- #include <stdlib.h> // getenv()
- #if defined(MS_WINDOWS) || defined(__CYGWIN__)
- # ifdef HAVE_IO_H
- # include <io.h>
- # endif
- # ifdef HAVE_FCNTL_H
- # include <fcntl.h> // O_BINARY
- # endif
- #endif
- /* --- Command line options --------------------------------------- */
- /* Short usage message (with %s for argv0) */
- static const char usage_line[] =
- "usage: %ls [option] ... [-c cmd | -m mod | file | -] [arg] ...\n";
- /* Long help message */
- /* Lines sorted by option name; keep in sync with usage_envvars* below */
- static const char usage_help[] = "\
- Options (and corresponding environment variables):\n\
- -b : issue warnings about converting bytes/bytearray to str and comparing\n\
- bytes/bytearray with str or bytes with int. (-bb: issue errors)\n\
- -B : don't write .pyc files on import; also PYTHONDONTWRITEBYTECODE=x\n\
- -c cmd : program passed in as string (terminates option list)\n\
- -d : turn on parser debugging output (for experts only, only works on\n\
- debug builds); also PYTHONDEBUG=x\n\
- -E : ignore PYTHON* environment variables (such as PYTHONPATH)\n\
- -h : print this help message and exit (also -? or --help)\n\
- -i : inspect interactively after running script; forces a prompt even\n\
- if stdin does not appear to be a terminal; also PYTHONINSPECT=x\n\
- -I : isolate Python from the user's environment (implies -E and -s)\n\
- -m mod : run library module as a script (terminates option list)\n\
- -O : remove assert and __debug__-dependent statements; add .opt-1 before\n\
- .pyc extension; also PYTHONOPTIMIZE=x\n\
- -OO : do -O changes and also discard docstrings; add .opt-2 before\n\
- .pyc extension\n\
- -P : don't prepend a potentially unsafe path to sys.path; also\n\
- PYTHONSAFEPATH\n\
- -q : don't print version and copyright messages on interactive startup\n\
- -s : don't add user site directory to sys.path; also PYTHONNOUSERSITE=x\n\
- -S : don't imply 'import site' on initialization\n\
- -u : force the stdout and stderr streams to be unbuffered;\n\
- this option has no effect on stdin; also PYTHONUNBUFFERED=x\n\
- -v : verbose (trace import statements); also PYTHONVERBOSE=x\n\
- can be supplied multiple times to increase verbosity\n\
- -V : print the Python version number and exit (also --version)\n\
- when given twice, print more information about the build\n\
- -W arg : warning control; arg is action:message:category:module:lineno\n\
- also PYTHONWARNINGS=arg\n\
- -x : skip first line of source, allowing use of non-Unix forms of #!cmd\n\
- -X opt : set implementation-specific option\n\
- --check-hash-based-pycs always|default|never:\n\
- control how Python invalidates hash-based .pyc files\n\
- --help-env: print help about Python environment variables and exit\n\
- --help-xoptions: print help about implementation-specific -X options and exit\n\
- --help-all: print complete help information and exit\n\
- \n\
- Arguments:\n\
- file : program read from script file\n\
- - : program read from stdin (default; interactive mode if a tty)\n\
- arg ...: arguments passed to program in sys.argv[1:]\n\
- ";
- static const char usage_xoptions[] = "\
- The following implementation-specific options are available:\n\
- -X dev : enable Python Development Mode; also PYTHONDEVMODE\n\
- -X faulthandler: dump the Python traceback on fatal errors;\n\
- also PYTHONFAULTHANDLER\n\
- -X frozen_modules=[on|off]: whether to use frozen modules; the default is \"on\"\n\
- for installed Python and \"off\" for a local build\n\
- -X importtime: show how long each import takes; also PYTHONPROFILEIMPORTTIME\n\
- -X int_max_str_digits=N: limit the size of int<->str conversions;\n\
- 0 disables the limit; also PYTHONINTMAXSTRDIGITS\n\
- -X no_debug_ranges: don't include extra location information in code objects;\n\
- also PYTHONNODEBUGRANGES\n\
- -X perf: support the Linux \"perf\" profiler; also PYTHONPERFSUPPORT=1\n\
- "
- #ifdef Py_DEBUG
- "-X presite=MOD: import this module before site; also PYTHON_PRESITE\n"
- #endif
- "\
- -X pycache_prefix=PATH: write .pyc files to a parallel tree instead of to the\n\
- code tree; also PYTHONPYCACHEPREFIX\n\
- "
- #ifdef Py_STATS
- "-X pystats: enable pystats collection at startup; also PYTHONSTATS\n"
- #endif
- "\
- -X showrefcount: output the total reference count and number of used\n\
- memory blocks when the program finishes or after each statement in\n\
- the interactive interpreter; only works on debug builds\n\
- -X tracemalloc[=N]: trace Python memory allocations; N sets a traceback limit\n\
- of N frames (default: 1); also PYTHONTRACEMALLOC=N\n\
- -X utf8[=0|1]: enable (1) or disable (0) UTF-8 mode; also PYTHONUTF8\n\
- -X warn_default_encoding: enable opt-in EncodingWarning for 'encoding=None';\n\
- also PYTHONWARNDEFAULTENCODING\
- ";
- /* Envvars that don't have equivalent command-line options are listed first */
- static const char usage_envvars[] =
- "Environment variables that change behavior:\n"
- "PYTHONSTARTUP : file executed on interactive startup (no default)\n"
- "PYTHONPATH : '%lc'-separated list of directories prefixed to the\n"
- " default module search path. The result is sys.path.\n"
- "PYTHONHOME : alternate <prefix> directory (or <prefix>%lc<exec_prefix>).\n"
- " The default module search path uses %s.\n"
- "PYTHONPLATLIBDIR: override sys.platlibdir\n"
- "PYTHONCASEOK : ignore case in 'import' statements (Windows)\n"
- "PYTHONIOENCODING: encoding[:errors] used for stdin/stdout/stderr\n"
- "PYTHONHASHSEED : if this variable is set to 'random', a random value is used\n"
- " to seed the hashes of str and bytes objects. It can also be\n"
- " set to an integer in the range [0,4294967295] to get hash\n"
- " values with a predictable seed.\n"
- "PYTHONMALLOC : set the Python memory allocators and/or install debug hooks\n"
- " on Python memory allocators. Use PYTHONMALLOC=debug to\n"
- " install debug hooks.\n"
- "PYTHONCOERCECLOCALE: if this variable is set to 0, it disables the locale\n"
- " coercion behavior. Use PYTHONCOERCECLOCALE=warn to request\n"
- " display of locale coercion and locale compatibility warnings\n"
- " on stderr.\n"
- "PYTHONBREAKPOINT: if this variable is set to 0, it disables the default\n"
- " debugger. It can be set to the callable of your debugger of\n"
- " choice.\n"
- "\n"
- "These variables have equivalent command-line options (see --help for details):\n"
- "PYTHONDEBUG : enable parser debug mode (-d)\n"
- "PYTHONDEVMODE : enable Python Development Mode (-X dev)\n"
- "PYTHONDONTWRITEBYTECODE: don't write .pyc files (-B)\n"
- "PYTHONFAULTHANDLER: dump the Python traceback on fatal errors (-X faulthandler)\n"
- "PYTHONINSPECT : inspect interactively after running script (-i)\n"
- "PYTHONINTMAXSTRDIGITS: limit the size of int<->str conversions;\n"
- " 0 disables the limit (-X int_max_str_digits=N)\n"
- "PYTHONNODEBUGRANGES: don't include extra location information in code objects\n"
- " (-X no_debug_ranges)\n"
- "PYTHONNOUSERSITE: disable user site directory (-s)\n"
- "PYTHONOPTIMIZE : enable level 1 optimizations (-O)\n"
- "PYTHONPERFSUPPORT: support the Linux \"perf\" profiler (-X perf)\n"
- "PYTHONPROFILEIMPORTTIME: show how long each import takes (-X importtime)\n"
- "PYTHONPYCACHEPREFIX: root directory for bytecode cache (pyc) files\n"
- " (-X pycache_prefix)\n"
- "PYTHONSAFEPATH : don't prepend a potentially unsafe path to sys.path.\n"
- "PYTHONTRACEMALLOC: trace Python memory allocations (-X tracemalloc)\n"
- "PYTHONUNBUFFERED: disable stdout/stderr buffering (-u)\n"
- "PYTHONUTF8 : control the UTF-8 mode (-X utf8)\n"
- "PYTHONVERBOSE : trace import statements (-v)\n"
- "PYTHONWARNDEFAULTENCODING: enable opt-in EncodingWarning for 'encoding=None'\n"
- " (-X warn_default_encoding)\n"
- "PYTHONWARNINGS : warning control (-W)\n"
- ;
- #if defined(MS_WINDOWS)
- # define PYTHONHOMEHELP "<prefix>\\python{major}{minor}"
- #else
- # define PYTHONHOMEHELP "<prefix>/lib/pythonX.X"
- #endif
- /* --- Global configuration variables ----------------------------- */
- /* UTF-8 mode (PEP 540): if equals to 1, use the UTF-8 encoding, and change
- stdin and stdout error handler to "surrogateescape". */
- int Py_UTF8Mode = 0;
- int Py_DebugFlag = 0; /* Needed by parser.c */
- int Py_VerboseFlag = 0; /* Needed by import.c */
- int Py_QuietFlag = 0; /* Needed by sysmodule.c */
- int Py_InteractiveFlag = 0; /* Previously, was used by Py_FdIsInteractive() */
- int Py_InspectFlag = 0; /* Needed to determine whether to exit at SystemExit */
- int Py_OptimizeFlag = 0; /* Needed by compile.c */
- int Py_NoSiteFlag = 0; /* Suppress 'import site' */
- int Py_BytesWarningFlag = 0; /* Warn on str(bytes) and str(buffer) */
- int Py_FrozenFlag = 1; /* Needed by getpath.c */
- int Py_IgnoreEnvironmentFlag = 0; /* e.g. PYTHONPATH, PYTHONHOME */
- int Py_DontWriteBytecodeFlag = 0; /* Suppress writing bytecode files (*.pyc) */
- int Py_NoUserSiteDirectory = 0; /* for -s and site.py */
- int Py_UnbufferedStdioFlag = 0; /* Unbuffered binary std{in,out,err} */
- int Py_HashRandomizationFlag = 0; /* for -R and PYTHONHASHSEED */
- int Py_IsolatedFlag = 0; /* for -I, isolate from user's env */
- #ifdef MS_WINDOWS
- int Py_LegacyWindowsFSEncodingFlag = 0; /* Uses mbcs instead of utf-8 */
- int Py_LegacyWindowsStdioFlag = 0; /* Uses FileIO instead of WindowsConsoleIO */
- #endif
- static PyObject *
- _Py_GetGlobalVariablesAsDict(void)
- {
- _Py_COMP_DIAG_PUSH
- _Py_COMP_DIAG_IGNORE_DEPR_DECLS
- PyObject *dict, *obj;
- dict = PyDict_New();
- if (dict == NULL) {
- return NULL;
- }
- #define SET_ITEM(KEY, EXPR) \
- do { \
- obj = (EXPR); \
- if (obj == NULL) { \
- return NULL; \
- } \
- int res = PyDict_SetItemString(dict, (KEY), obj); \
- Py_DECREF(obj); \
- if (res < 0) { \
- goto fail; \
- } \
- } while (0)
- #define SET_ITEM_INT(VAR) \
- SET_ITEM(#VAR, PyLong_FromLong(VAR))
- #define FROM_STRING(STR) \
- ((STR != NULL) ? \
- PyUnicode_FromString(STR) \
- : Py_NewRef(Py_None))
- #define SET_ITEM_STR(VAR) \
- SET_ITEM(#VAR, FROM_STRING(VAR))
- SET_ITEM_STR(Py_FileSystemDefaultEncoding);
- SET_ITEM_INT(Py_HasFileSystemDefaultEncoding);
- SET_ITEM_STR(Py_FileSystemDefaultEncodeErrors);
- SET_ITEM_INT(_Py_HasFileSystemDefaultEncodeErrors);
- SET_ITEM_INT(Py_UTF8Mode);
- SET_ITEM_INT(Py_DebugFlag);
- SET_ITEM_INT(Py_VerboseFlag);
- SET_ITEM_INT(Py_QuietFlag);
- SET_ITEM_INT(Py_InteractiveFlag);
- SET_ITEM_INT(Py_InspectFlag);
- SET_ITEM_INT(Py_OptimizeFlag);
- SET_ITEM_INT(Py_NoSiteFlag);
- SET_ITEM_INT(Py_BytesWarningFlag);
- SET_ITEM_INT(Py_FrozenFlag);
- SET_ITEM_INT(Py_IgnoreEnvironmentFlag);
- SET_ITEM_INT(Py_DontWriteBytecodeFlag);
- SET_ITEM_INT(Py_NoUserSiteDirectory);
- SET_ITEM_INT(Py_UnbufferedStdioFlag);
- SET_ITEM_INT(Py_HashRandomizationFlag);
- SET_ITEM_INT(Py_IsolatedFlag);
- #ifdef MS_WINDOWS
- SET_ITEM_INT(Py_LegacyWindowsFSEncodingFlag);
- SET_ITEM_INT(Py_LegacyWindowsStdioFlag);
- #endif
- return dict;
- fail:
- Py_DECREF(dict);
- return NULL;
- #undef FROM_STRING
- #undef SET_ITEM
- #undef SET_ITEM_INT
- #undef SET_ITEM_STR
- _Py_COMP_DIAG_POP
- }
- char*
- Py_GETENV(const char *name)
- {
- _Py_COMP_DIAG_PUSH
- _Py_COMP_DIAG_IGNORE_DEPR_DECLS
- if (Py_IgnoreEnvironmentFlag) {
- return NULL;
- }
- return getenv(name);
- _Py_COMP_DIAG_POP
- }
- /* --- PyStatus ----------------------------------------------- */
- PyStatus PyStatus_Ok(void)
- { return _PyStatus_OK(); }
- PyStatus PyStatus_Error(const char *err_msg)
- {
- assert(err_msg != NULL);
- return (PyStatus){._type = _PyStatus_TYPE_ERROR,
- .err_msg = err_msg};
- }
- PyStatus PyStatus_NoMemory(void)
- { return PyStatus_Error("memory allocation failed"); }
- PyStatus PyStatus_Exit(int exitcode)
- { return _PyStatus_EXIT(exitcode); }
- int PyStatus_IsError(PyStatus status)
- { return _PyStatus_IS_ERROR(status); }
- int PyStatus_IsExit(PyStatus status)
- { return _PyStatus_IS_EXIT(status); }
- int PyStatus_Exception(PyStatus status)
- { return _PyStatus_EXCEPTION(status); }
- PyObject*
- _PyErr_SetFromPyStatus(PyStatus status)
- {
- if (!_PyStatus_IS_ERROR(status)) {
- PyErr_Format(PyExc_SystemError,
- "%s() expects an error PyStatus",
- _PyStatus_GET_FUNC());
- }
- else if (status.func) {
- PyErr_Format(PyExc_ValueError, "%s: %s", status.func, status.err_msg);
- }
- else {
- PyErr_Format(PyExc_ValueError, "%s", status.err_msg);
- }
- return NULL;
- }
- /* --- PyWideStringList ------------------------------------------------ */
- #ifndef NDEBUG
- int
- _PyWideStringList_CheckConsistency(const PyWideStringList *list)
- {
- assert(list->length >= 0);
- if (list->length != 0) {
- assert(list->items != NULL);
- }
- for (Py_ssize_t i = 0; i < list->length; i++) {
- assert(list->items[i] != NULL);
- }
- return 1;
- }
- #endif /* Py_DEBUG */
- void
- _PyWideStringList_Clear(PyWideStringList *list)
- {
- assert(_PyWideStringList_CheckConsistency(list));
- for (Py_ssize_t i=0; i < list->length; i++) {
- PyMem_RawFree(list->items[i]);
- }
- PyMem_RawFree(list->items);
- list->length = 0;
- list->items = NULL;
- }
- int
- _PyWideStringList_Copy(PyWideStringList *list, const PyWideStringList *list2)
- {
- assert(_PyWideStringList_CheckConsistency(list));
- assert(_PyWideStringList_CheckConsistency(list2));
- if (list2->length == 0) {
- _PyWideStringList_Clear(list);
- return 0;
- }
- PyWideStringList copy = _PyWideStringList_INIT;
- size_t size = list2->length * sizeof(list2->items[0]);
- copy.items = PyMem_RawMalloc(size);
- if (copy.items == NULL) {
- return -1;
- }
- for (Py_ssize_t i=0; i < list2->length; i++) {
- wchar_t *item = _PyMem_RawWcsdup(list2->items[i]);
- if (item == NULL) {
- _PyWideStringList_Clear(©);
- return -1;
- }
- copy.items[i] = item;
- copy.length = i + 1;
- }
- _PyWideStringList_Clear(list);
- *list = copy;
- return 0;
- }
- PyStatus
- PyWideStringList_Insert(PyWideStringList *list,
- Py_ssize_t index, const wchar_t *item)
- {
- Py_ssize_t len = list->length;
- if (len == PY_SSIZE_T_MAX) {
- /* length+1 would overflow */
- return _PyStatus_NO_MEMORY();
- }
- if (index < 0) {
- return _PyStatus_ERR("PyWideStringList_Insert index must be >= 0");
- }
- if (index > len) {
- index = len;
- }
- wchar_t *item2 = _PyMem_RawWcsdup(item);
- if (item2 == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- size_t size = (len + 1) * sizeof(list->items[0]);
- wchar_t **items2 = (wchar_t **)PyMem_RawRealloc(list->items, size);
- if (items2 == NULL) {
- PyMem_RawFree(item2);
- return _PyStatus_NO_MEMORY();
- }
- if (index < len) {
- memmove(&items2[index + 1],
- &items2[index],
- (len - index) * sizeof(items2[0]));
- }
- items2[index] = item2;
- list->items = items2;
- list->length++;
- return _PyStatus_OK();
- }
- PyStatus
- PyWideStringList_Append(PyWideStringList *list, const wchar_t *item)
- {
- return PyWideStringList_Insert(list, list->length, item);
- }
- PyStatus
- _PyWideStringList_Extend(PyWideStringList *list, const PyWideStringList *list2)
- {
- for (Py_ssize_t i = 0; i < list2->length; i++) {
- PyStatus status = PyWideStringList_Append(list, list2->items[i]);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- return _PyStatus_OK();
- }
- static int
- _PyWideStringList_Find(PyWideStringList *list, const wchar_t *item)
- {
- for (Py_ssize_t i = 0; i < list->length; i++) {
- if (wcscmp(list->items[i], item) == 0) {
- return 1;
- }
- }
- return 0;
- }
- PyObject*
- _PyWideStringList_AsList(const PyWideStringList *list)
- {
- assert(_PyWideStringList_CheckConsistency(list));
- PyObject *pylist = PyList_New(list->length);
- if (pylist == NULL) {
- return NULL;
- }
- for (Py_ssize_t i = 0; i < list->length; i++) {
- PyObject *item = PyUnicode_FromWideChar(list->items[i], -1);
- if (item == NULL) {
- Py_DECREF(pylist);
- return NULL;
- }
- PyList_SET_ITEM(pylist, i, item);
- }
- return pylist;
- }
- /* --- Py_SetStandardStreamEncoding() ----------------------------- */
- /* Helper to allow an embedding application to override the normal
- * mechanism that attempts to figure out an appropriate IO encoding
- */
- static char *_Py_StandardStreamEncoding = NULL;
- static char *_Py_StandardStreamErrors = NULL;
- int
- Py_SetStandardStreamEncoding(const char *encoding, const char *errors)
- {
- if (Py_IsInitialized()) {
- /* This is too late to have any effect */
- return -1;
- }
- int res = 0;
- /* Py_SetStandardStreamEncoding() can be called before Py_Initialize(),
- but Py_Initialize() can change the allocator. Use a known allocator
- to be able to release the memory later. */
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- /* Can't call PyErr_NoMemory() on errors, as Python hasn't been
- * initialised yet.
- *
- * However, the raw memory allocators are initialised appropriately
- * as C static variables, so _PyMem_RawStrdup is OK even though
- * Py_Initialize hasn't been called yet.
- */
- if (encoding) {
- PyMem_RawFree(_Py_StandardStreamEncoding);
- _Py_StandardStreamEncoding = _PyMem_RawStrdup(encoding);
- if (!_Py_StandardStreamEncoding) {
- res = -2;
- goto done;
- }
- }
- if (errors) {
- PyMem_RawFree(_Py_StandardStreamErrors);
- _Py_StandardStreamErrors = _PyMem_RawStrdup(errors);
- if (!_Py_StandardStreamErrors) {
- PyMem_RawFree(_Py_StandardStreamEncoding);
- _Py_StandardStreamEncoding = NULL;
- res = -3;
- goto done;
- }
- }
- #ifdef MS_WINDOWS
- if (_Py_StandardStreamEncoding) {
- _Py_COMP_DIAG_PUSH
- _Py_COMP_DIAG_IGNORE_DEPR_DECLS
- /* Overriding the stream encoding implies legacy streams */
- Py_LegacyWindowsStdioFlag = 1;
- _Py_COMP_DIAG_POP
- }
- #endif
- done:
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- return res;
- }
- void
- _Py_ClearStandardStreamEncoding(void)
- {
- /* Use the same allocator than Py_SetStandardStreamEncoding() */
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- /* We won't need them anymore. */
- if (_Py_StandardStreamEncoding) {
- PyMem_RawFree(_Py_StandardStreamEncoding);
- _Py_StandardStreamEncoding = NULL;
- }
- if (_Py_StandardStreamErrors) {
- PyMem_RawFree(_Py_StandardStreamErrors);
- _Py_StandardStreamErrors = NULL;
- }
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- }
- /* --- Py_GetArgcArgv() ------------------------------------------- */
- void
- _Py_ClearArgcArgv(void)
- {
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- _PyWideStringList_Clear(&_PyRuntime.orig_argv);
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- }
- static int
- _Py_SetArgcArgv(Py_ssize_t argc, wchar_t * const *argv)
- {
- const PyWideStringList argv_list = {.length = argc, .items = (wchar_t **)argv};
- int res;
- PyMemAllocatorEx old_alloc;
- _PyMem_SetDefaultAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- // XXX _PyRuntime.orig_argv only gets cleared by Py_Main(),
- // so it it currently leaks for embedders.
- res = _PyWideStringList_Copy(&_PyRuntime.orig_argv, &argv_list);
- PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &old_alloc);
- return res;
- }
- // _PyConfig_Write() calls _Py_SetArgcArgv() with PyConfig.orig_argv.
- void
- Py_GetArgcArgv(int *argc, wchar_t ***argv)
- {
- *argc = (int)_PyRuntime.orig_argv.length;
- *argv = _PyRuntime.orig_argv.items;
- }
- /* --- PyConfig ---------------------------------------------- */
- #define MAX_HASH_SEED 4294967295UL
- #ifndef NDEBUG
- static int
- config_check_consistency(const PyConfig *config)
- {
- /* Check config consistency */
- assert(config->isolated >= 0);
- assert(config->use_environment >= 0);
- assert(config->dev_mode >= 0);
- assert(config->install_signal_handlers >= 0);
- assert(config->use_hash_seed >= 0);
- assert(config->hash_seed <= MAX_HASH_SEED);
- assert(config->faulthandler >= 0);
- assert(config->tracemalloc >= 0);
- assert(config->import_time >= 0);
- assert(config->code_debug_ranges >= 0);
- assert(config->show_ref_count >= 0);
- assert(config->dump_refs >= 0);
- assert(config->malloc_stats >= 0);
- assert(config->site_import >= 0);
- assert(config->bytes_warning >= 0);
- assert(config->warn_default_encoding >= 0);
- assert(config->inspect >= 0);
- assert(config->interactive >= 0);
- assert(config->optimization_level >= 0);
- assert(config->parser_debug >= 0);
- assert(config->write_bytecode >= 0);
- assert(config->verbose >= 0);
- assert(config->quiet >= 0);
- assert(config->user_site_directory >= 0);
- assert(config->parse_argv >= 0);
- assert(config->configure_c_stdio >= 0);
- assert(config->buffered_stdio >= 0);
- assert(_PyWideStringList_CheckConsistency(&config->orig_argv));
- assert(_PyWideStringList_CheckConsistency(&config->argv));
- /* sys.argv must be non-empty: empty argv is replaced with [''] */
- assert(config->argv.length >= 1);
- assert(_PyWideStringList_CheckConsistency(&config->xoptions));
- assert(_PyWideStringList_CheckConsistency(&config->warnoptions));
- assert(_PyWideStringList_CheckConsistency(&config->module_search_paths));
- assert(config->module_search_paths_set >= 0);
- assert(config->filesystem_encoding != NULL);
- assert(config->filesystem_errors != NULL);
- assert(config->stdio_encoding != NULL);
- assert(config->stdio_errors != NULL);
- #ifdef MS_WINDOWS
- assert(config->legacy_windows_stdio >= 0);
- #endif
- /* -c and -m options are exclusive */
- assert(!(config->run_command != NULL && config->run_module != NULL));
- assert(config->check_hash_pycs_mode != NULL);
- assert(config->_install_importlib >= 0);
- assert(config->pathconfig_warnings >= 0);
- assert(config->_is_python_build >= 0);
- assert(config->safe_path >= 0);
- assert(config->int_max_str_digits >= 0);
- // config->use_frozen_modules is initialized later
- // by _PyConfig_InitImportConfig().
- return 1;
- }
- #endif
- /* Free memory allocated in config, but don't clear all attributes */
- void
- PyConfig_Clear(PyConfig *config)
- {
- #define CLEAR(ATTR) \
- do { \
- PyMem_RawFree(ATTR); \
- ATTR = NULL; \
- } while (0)
- CLEAR(config->pycache_prefix);
- CLEAR(config->pythonpath_env);
- CLEAR(config->home);
- CLEAR(config->program_name);
- _PyWideStringList_Clear(&config->argv);
- _PyWideStringList_Clear(&config->warnoptions);
- _PyWideStringList_Clear(&config->xoptions);
- _PyWideStringList_Clear(&config->module_search_paths);
- config->module_search_paths_set = 0;
- CLEAR(config->stdlib_dir);
- CLEAR(config->executable);
- CLEAR(config->base_executable);
- CLEAR(config->prefix);
- CLEAR(config->base_prefix);
- CLEAR(config->exec_prefix);
- CLEAR(config->base_exec_prefix);
- CLEAR(config->platlibdir);
- CLEAR(config->filesystem_encoding);
- CLEAR(config->filesystem_errors);
- CLEAR(config->stdio_encoding);
- CLEAR(config->stdio_errors);
- CLEAR(config->run_command);
- CLEAR(config->run_module);
- CLEAR(config->run_filename);
- CLEAR(config->check_hash_pycs_mode);
- _PyWideStringList_Clear(&config->orig_argv);
- #undef CLEAR
- }
- void
- _PyConfig_InitCompatConfig(PyConfig *config)
- {
- memset(config, 0, sizeof(*config));
- config->_config_init = (int)_PyConfig_INIT_COMPAT;
- config->isolated = -1;
- config->use_environment = -1;
- config->dev_mode = -1;
- config->install_signal_handlers = 1;
- config->use_hash_seed = -1;
- config->faulthandler = -1;
- config->tracemalloc = -1;
- config->perf_profiling = -1;
- config->module_search_paths_set = 0;
- config->parse_argv = 0;
- config->site_import = -1;
- config->bytes_warning = -1;
- config->warn_default_encoding = 0;
- config->inspect = -1;
- config->interactive = -1;
- config->optimization_level = -1;
- config->parser_debug= -1;
- config->write_bytecode = -1;
- config->verbose = -1;
- config->quiet = -1;
- config->user_site_directory = -1;
- config->configure_c_stdio = 0;
- config->buffered_stdio = -1;
- config->_install_importlib = 1;
- config->check_hash_pycs_mode = NULL;
- config->pathconfig_warnings = -1;
- config->_init_main = 1;
- #ifdef MS_WINDOWS
- config->legacy_windows_stdio = -1;
- #endif
- #ifdef Py_DEBUG
- config->use_frozen_modules = 0;
- #else
- config->use_frozen_modules = 1;
- #endif
- config->safe_path = 0;
- config->int_max_str_digits = -1;
- config->_is_python_build = 0;
- config->code_debug_ranges = 1;
- }
- static void
- config_init_defaults(PyConfig *config)
- {
- _PyConfig_InitCompatConfig(config);
- config->isolated = 0;
- config->use_environment = 1;
- config->site_import = 1;
- config->bytes_warning = 0;
- config->inspect = 0;
- config->interactive = 0;
- config->optimization_level = 0;
- config->parser_debug= 0;
- config->write_bytecode = 1;
- config->verbose = 0;
- config->quiet = 0;
- config->user_site_directory = 1;
- config->buffered_stdio = 1;
- config->pathconfig_warnings = 1;
- #ifdef MS_WINDOWS
- config->legacy_windows_stdio = 0;
- #endif
- }
- void
- PyConfig_InitPythonConfig(PyConfig *config)
- {
- config_init_defaults(config);
- config->_config_init = (int)_PyConfig_INIT_PYTHON;
- config->configure_c_stdio = 1;
- config->parse_argv = 1;
- }
- void
- PyConfig_InitIsolatedConfig(PyConfig *config)
- {
- config_init_defaults(config);
- config->_config_init = (int)_PyConfig_INIT_ISOLATED;
- config->isolated = 1;
- config->use_environment = 0;
- config->user_site_directory = 0;
- config->dev_mode = 0;
- config->install_signal_handlers = 0;
- config->use_hash_seed = 0;
- config->faulthandler = 0;
- config->tracemalloc = 0;
- config->perf_profiling = 0;
- config->int_max_str_digits = _PY_LONG_DEFAULT_MAX_STR_DIGITS;
- config->safe_path = 1;
- config->pathconfig_warnings = 0;
- #ifdef MS_WINDOWS
- config->legacy_windows_stdio = 0;
- #endif
- }
- /* Copy str into *config_str (duplicate the string) */
- PyStatus
- PyConfig_SetString(PyConfig *config, wchar_t **config_str, const wchar_t *str)
- {
- PyStatus status = _Py_PreInitializeFromConfig(config, NULL);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- wchar_t *str2;
- if (str != NULL) {
- str2 = _PyMem_RawWcsdup(str);
- if (str2 == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- }
- else {
- str2 = NULL;
- }
- PyMem_RawFree(*config_str);
- *config_str = str2;
- return _PyStatus_OK();
- }
- static PyStatus
- config_set_bytes_string(PyConfig *config, wchar_t **config_str,
- const char *str, const char *decode_err_msg)
- {
- PyStatus status = _Py_PreInitializeFromConfig(config, NULL);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- wchar_t *str2;
- if (str != NULL) {
- size_t len;
- str2 = Py_DecodeLocale(str, &len);
- if (str2 == NULL) {
- if (len == (size_t)-2) {
- return _PyStatus_ERR(decode_err_msg);
- }
- else {
- return _PyStatus_NO_MEMORY();
- }
- }
- }
- else {
- str2 = NULL;
- }
- PyMem_RawFree(*config_str);
- *config_str = str2;
- return _PyStatus_OK();
- }
- #define CONFIG_SET_BYTES_STR(config, config_str, str, NAME) \
- config_set_bytes_string(config, config_str, str, "cannot decode " NAME)
- /* Decode str using Py_DecodeLocale() and set the result into *config_str.
- Pre-initialize Python if needed to ensure that encodings are properly
- configured. */
- PyStatus
- PyConfig_SetBytesString(PyConfig *config, wchar_t **config_str,
- const char *str)
- {
- return CONFIG_SET_BYTES_STR(config, config_str, str, "string");
- }
- PyStatus
- _PyConfig_Copy(PyConfig *config, const PyConfig *config2)
- {
- PyStatus status;
- PyConfig_Clear(config);
- #define COPY_ATTR(ATTR) config->ATTR = config2->ATTR
- #define COPY_WSTR_ATTR(ATTR) \
- do { \
- status = PyConfig_SetString(config, &config->ATTR, config2->ATTR); \
- if (_PyStatus_EXCEPTION(status)) { \
- return status; \
- } \
- } while (0)
- #define COPY_WSTRLIST(LIST) \
- do { \
- if (_PyWideStringList_Copy(&config->LIST, &config2->LIST) < 0) { \
- return _PyStatus_NO_MEMORY(); \
- } \
- } while (0)
- COPY_ATTR(_config_init);
- COPY_ATTR(isolated);
- COPY_ATTR(use_environment);
- COPY_ATTR(dev_mode);
- COPY_ATTR(install_signal_handlers);
- COPY_ATTR(use_hash_seed);
- COPY_ATTR(hash_seed);
- COPY_ATTR(_install_importlib);
- COPY_ATTR(faulthandler);
- COPY_ATTR(tracemalloc);
- COPY_ATTR(perf_profiling);
- COPY_ATTR(import_time);
- COPY_ATTR(code_debug_ranges);
- COPY_ATTR(show_ref_count);
- COPY_ATTR(dump_refs);
- COPY_ATTR(dump_refs_file);
- COPY_ATTR(malloc_stats);
- COPY_WSTR_ATTR(pycache_prefix);
- COPY_WSTR_ATTR(pythonpath_env);
- COPY_WSTR_ATTR(home);
- COPY_WSTR_ATTR(program_name);
- COPY_ATTR(parse_argv);
- COPY_WSTRLIST(argv);
- COPY_WSTRLIST(warnoptions);
- COPY_WSTRLIST(xoptions);
- COPY_WSTRLIST(module_search_paths);
- COPY_ATTR(module_search_paths_set);
- COPY_WSTR_ATTR(stdlib_dir);
- COPY_WSTR_ATTR(executable);
- COPY_WSTR_ATTR(base_executable);
- COPY_WSTR_ATTR(prefix);
- COPY_WSTR_ATTR(base_prefix);
- COPY_WSTR_ATTR(exec_prefix);
- COPY_WSTR_ATTR(base_exec_prefix);
- COPY_WSTR_ATTR(platlibdir);
- COPY_ATTR(site_import);
- COPY_ATTR(bytes_warning);
- COPY_ATTR(warn_default_encoding);
- COPY_ATTR(inspect);
- COPY_ATTR(interactive);
- COPY_ATTR(optimization_level);
- COPY_ATTR(parser_debug);
- COPY_ATTR(write_bytecode);
- COPY_ATTR(verbose);
- COPY_ATTR(quiet);
- COPY_ATTR(user_site_directory);
- COPY_ATTR(configure_c_stdio);
- COPY_ATTR(buffered_stdio);
- COPY_WSTR_ATTR(filesystem_encoding);
- COPY_WSTR_ATTR(filesystem_errors);
- COPY_WSTR_ATTR(stdio_encoding);
- COPY_WSTR_ATTR(stdio_errors);
- #ifdef MS_WINDOWS
- COPY_ATTR(legacy_windows_stdio);
- #endif
- COPY_ATTR(skip_source_first_line);
- COPY_WSTR_ATTR(run_command);
- COPY_WSTR_ATTR(run_module);
- COPY_WSTR_ATTR(run_filename);
- COPY_WSTR_ATTR(check_hash_pycs_mode);
- COPY_ATTR(pathconfig_warnings);
- COPY_ATTR(_init_main);
- COPY_ATTR(use_frozen_modules);
- COPY_ATTR(safe_path);
- COPY_WSTRLIST(orig_argv);
- COPY_ATTR(_is_python_build);
- COPY_ATTR(int_max_str_digits);
- #undef COPY_ATTR
- #undef COPY_WSTR_ATTR
- #undef COPY_WSTRLIST
- return _PyStatus_OK();
- }
- PyObject *
- _PyConfig_AsDict(const PyConfig *config)
- {
- PyObject *dict = PyDict_New();
- if (dict == NULL) {
- return NULL;
- }
- #define SET_ITEM(KEY, EXPR) \
- do { \
- PyObject *obj = (EXPR); \
- if (obj == NULL) { \
- goto fail; \
- } \
- int res = PyDict_SetItemString(dict, (KEY), obj); \
- Py_DECREF(obj); \
- if (res < 0) { \
- goto fail; \
- } \
- } while (0)
- #define SET_ITEM_INT(ATTR) \
- SET_ITEM(#ATTR, PyLong_FromLong(config->ATTR))
- #define SET_ITEM_UINT(ATTR) \
- SET_ITEM(#ATTR, PyLong_FromUnsignedLong(config->ATTR))
- #define FROM_WSTRING(STR) \
- ((STR != NULL) ? \
- PyUnicode_FromWideChar(STR, -1) \
- : Py_NewRef(Py_None))
- #define SET_ITEM_WSTR(ATTR) \
- SET_ITEM(#ATTR, FROM_WSTRING(config->ATTR))
- #define SET_ITEM_WSTRLIST(LIST) \
- SET_ITEM(#LIST, _PyWideStringList_AsList(&config->LIST))
- SET_ITEM_INT(_config_init);
- SET_ITEM_INT(isolated);
- SET_ITEM_INT(use_environment);
- SET_ITEM_INT(dev_mode);
- SET_ITEM_INT(install_signal_handlers);
- SET_ITEM_INT(use_hash_seed);
- SET_ITEM_UINT(hash_seed);
- SET_ITEM_INT(faulthandler);
- SET_ITEM_INT(tracemalloc);
- SET_ITEM_INT(perf_profiling);
- SET_ITEM_INT(import_time);
- SET_ITEM_INT(code_debug_ranges);
- SET_ITEM_INT(show_ref_count);
- SET_ITEM_INT(dump_refs);
- SET_ITEM_INT(malloc_stats);
- SET_ITEM_WSTR(filesystem_encoding);
- SET_ITEM_WSTR(filesystem_errors);
- SET_ITEM_WSTR(pycache_prefix);
- SET_ITEM_WSTR(program_name);
- SET_ITEM_INT(parse_argv);
- SET_ITEM_WSTRLIST(argv);
- SET_ITEM_WSTRLIST(xoptions);
- SET_ITEM_WSTRLIST(warnoptions);
- SET_ITEM_WSTR(pythonpath_env);
- SET_ITEM_WSTR(home);
- SET_ITEM_INT(module_search_paths_set);
- SET_ITEM_WSTRLIST(module_search_paths);
- SET_ITEM_WSTR(stdlib_dir);
- SET_ITEM_WSTR(executable);
- SET_ITEM_WSTR(base_executable);
- SET_ITEM_WSTR(prefix);
- SET_ITEM_WSTR(base_prefix);
- SET_ITEM_WSTR(exec_prefix);
- SET_ITEM_WSTR(base_exec_prefix);
- SET_ITEM_WSTR(platlibdir);
- SET_ITEM_INT(site_import);
- SET_ITEM_INT(bytes_warning);
- SET_ITEM_INT(warn_default_encoding);
- SET_ITEM_INT(inspect);
- SET_ITEM_INT(interactive);
- SET_ITEM_INT(optimization_level);
- SET_ITEM_INT(parser_debug);
- SET_ITEM_INT(write_bytecode);
- SET_ITEM_INT(verbose);
- SET_ITEM_INT(quiet);
- SET_ITEM_INT(user_site_directory);
- SET_ITEM_INT(configure_c_stdio);
- SET_ITEM_INT(buffered_stdio);
- SET_ITEM_WSTR(stdio_encoding);
- SET_ITEM_WSTR(stdio_errors);
- #ifdef MS_WINDOWS
- SET_ITEM_INT(legacy_windows_stdio);
- #endif
- SET_ITEM_INT(skip_source_first_line);
- SET_ITEM_WSTR(run_command);
- SET_ITEM_WSTR(run_module);
- SET_ITEM_WSTR(run_filename);
- SET_ITEM_INT(_install_importlib);
- SET_ITEM_WSTR(check_hash_pycs_mode);
- SET_ITEM_INT(pathconfig_warnings);
- SET_ITEM_INT(_init_main);
- SET_ITEM_WSTRLIST(orig_argv);
- SET_ITEM_INT(use_frozen_modules);
- SET_ITEM_INT(safe_path);
- SET_ITEM_INT(_is_python_build);
- SET_ITEM_INT(int_max_str_digits);
- return dict;
- fail:
- Py_DECREF(dict);
- return NULL;
- #undef FROM_WSTRING
- #undef SET_ITEM
- #undef SET_ITEM_INT
- #undef SET_ITEM_UINT
- #undef SET_ITEM_WSTR
- #undef SET_ITEM_WSTRLIST
- }
- static PyObject*
- config_dict_get(PyObject *dict, const char *name)
- {
- PyObject *item = _PyDict_GetItemStringWithError(dict, name);
- if (item == NULL && !PyErr_Occurred()) {
- PyErr_Format(PyExc_ValueError, "missing config key: %s", name);
- return NULL;
- }
- return item;
- }
- static void
- config_dict_invalid_value(const char *name)
- {
- PyErr_Format(PyExc_ValueError, "invalid config value: %s", name);
- }
- static void
- config_dict_invalid_type(const char *name)
- {
- PyErr_Format(PyExc_TypeError, "invalid config type: %s", name);
- }
- static int
- config_dict_get_int(PyObject *dict, const char *name, int *result)
- {
- PyObject *item = config_dict_get(dict, name);
- if (item == NULL) {
- return -1;
- }
- int value = _PyLong_AsInt(item);
- if (value == -1 && PyErr_Occurred()) {
- if (PyErr_ExceptionMatches(PyExc_TypeError)) {
- config_dict_invalid_type(name);
- }
- else if (PyErr_ExceptionMatches(PyExc_OverflowError)) {
- config_dict_invalid_value(name);
- }
- return -1;
- }
- *result = value;
- return 0;
- }
- static int
- config_dict_get_ulong(PyObject *dict, const char *name, unsigned long *result)
- {
- PyObject *item = config_dict_get(dict, name);
- if (item == NULL) {
- return -1;
- }
- unsigned long value = PyLong_AsUnsignedLong(item);
- if (value == (unsigned long)-1 && PyErr_Occurred()) {
- if (PyErr_ExceptionMatches(PyExc_TypeError)) {
- config_dict_invalid_type(name);
- }
- else if (PyErr_ExceptionMatches(PyExc_OverflowError)) {
- config_dict_invalid_value(name);
- }
- return -1;
- }
- *result = value;
- return 0;
- }
- static int
- config_dict_get_wstr(PyObject *dict, const char *name, PyConfig *config,
- wchar_t **result)
- {
- PyObject *item = config_dict_get(dict, name);
- if (item == NULL) {
- return -1;
- }
- PyStatus status;
- if (item == Py_None) {
- status = PyConfig_SetString(config, result, NULL);
- }
- else if (!PyUnicode_Check(item)) {
- config_dict_invalid_type(name);
- return -1;
- }
- else {
- wchar_t *wstr = PyUnicode_AsWideCharString(item, NULL);
- if (wstr == NULL) {
- return -1;
- }
- status = PyConfig_SetString(config, result, wstr);
- PyMem_Free(wstr);
- }
- if (_PyStatus_EXCEPTION(status)) {
- PyErr_NoMemory();
- return -1;
- }
- return 0;
- }
- static int
- config_dict_get_wstrlist(PyObject *dict, const char *name, PyConfig *config,
- PyWideStringList *result)
- {
- PyObject *list = config_dict_get(dict, name);
- if (list == NULL) {
- return -1;
- }
- if (!PyList_CheckExact(list)) {
- config_dict_invalid_type(name);
- return -1;
- }
- PyWideStringList wstrlist = _PyWideStringList_INIT;
- for (Py_ssize_t i=0; i < PyList_GET_SIZE(list); i++) {
- PyObject *item = PyList_GET_ITEM(list, i);
- if (item == Py_None) {
- config_dict_invalid_value(name);
- goto error;
- }
- else if (!PyUnicode_Check(item)) {
- config_dict_invalid_type(name);
- goto error;
- }
- wchar_t *wstr = PyUnicode_AsWideCharString(item, NULL);
- if (wstr == NULL) {
- goto error;
- }
- PyStatus status = PyWideStringList_Append(&wstrlist, wstr);
- PyMem_Free(wstr);
- if (_PyStatus_EXCEPTION(status)) {
- PyErr_NoMemory();
- goto error;
- }
- }
- if (_PyWideStringList_Copy(result, &wstrlist) < 0) {
- PyErr_NoMemory();
- goto error;
- }
- _PyWideStringList_Clear(&wstrlist);
- return 0;
- error:
- _PyWideStringList_Clear(&wstrlist);
- return -1;
- }
- int
- _PyConfig_FromDict(PyConfig *config, PyObject *dict)
- {
- if (!PyDict_Check(dict)) {
- PyErr_SetString(PyExc_TypeError, "dict expected");
- return -1;
- }
- #define CHECK_VALUE(NAME, TEST) \
- if (!(TEST)) { \
- config_dict_invalid_value(NAME); \
- return -1; \
- }
- #define GET_UINT(KEY) \
- do { \
- if (config_dict_get_int(dict, #KEY, &config->KEY) < 0) { \
- return -1; \
- } \
- CHECK_VALUE(#KEY, config->KEY >= 0); \
- } while (0)
- #define GET_INT(KEY) \
- do { \
- if (config_dict_get_int(dict, #KEY, &config->KEY) < 0) { \
- return -1; \
- } \
- } while (0)
- #define GET_WSTR(KEY) \
- do { \
- if (config_dict_get_wstr(dict, #KEY, config, &config->KEY) < 0) { \
- return -1; \
- } \
- CHECK_VALUE(#KEY, config->KEY != NULL); \
- } while (0)
- #define GET_WSTR_OPT(KEY) \
- do { \
- if (config_dict_get_wstr(dict, #KEY, config, &config->KEY) < 0) { \
- return -1; \
- } \
- } while (0)
- #define GET_WSTRLIST(KEY) \
- do { \
- if (config_dict_get_wstrlist(dict, #KEY, config, &config->KEY) < 0) { \
- return -1; \
- } \
- } while (0)
- GET_UINT(_config_init);
- CHECK_VALUE("_config_init",
- config->_config_init == _PyConfig_INIT_COMPAT
- || config->_config_init == _PyConfig_INIT_PYTHON
- || config->_config_init == _PyConfig_INIT_ISOLATED);
- GET_UINT(isolated);
- GET_UINT(use_environment);
- GET_UINT(dev_mode);
- GET_UINT(install_signal_handlers);
- GET_UINT(use_hash_seed);
- if (config_dict_get_ulong(dict, "hash_seed", &config->hash_seed) < 0) {
- return -1;
- }
- CHECK_VALUE("hash_seed", config->hash_seed <= MAX_HASH_SEED);
- GET_UINT(faulthandler);
- GET_UINT(tracemalloc);
- GET_UINT(perf_profiling);
- GET_UINT(import_time);
- GET_UINT(code_debug_ranges);
- GET_UINT(show_ref_count);
- GET_UINT(dump_refs);
- GET_UINT(malloc_stats);
- GET_WSTR(filesystem_encoding);
- GET_WSTR(filesystem_errors);
- GET_WSTR_OPT(pycache_prefix);
- GET_UINT(parse_argv);
- GET_WSTRLIST(orig_argv);
- GET_WSTRLIST(argv);
- GET_WSTRLIST(xoptions);
- GET_WSTRLIST(warnoptions);
- GET_UINT(site_import);
- GET_UINT(bytes_warning);
- GET_UINT(warn_default_encoding);
- GET_UINT(inspect);
- GET_UINT(interactive);
- GET_UINT(optimization_level);
- GET_UINT(parser_debug);
- GET_UINT(write_bytecode);
- GET_UINT(verbose);
- GET_UINT(quiet);
- GET_UINT(user_site_directory);
- GET_UINT(configure_c_stdio);
- GET_UINT(buffered_stdio);
- GET_WSTR(stdio_encoding);
- GET_WSTR(stdio_errors);
- #ifdef MS_WINDOWS
- GET_UINT(legacy_windows_stdio);
- #endif
- GET_WSTR(check_hash_pycs_mode);
- GET_UINT(pathconfig_warnings);
- GET_WSTR(program_name);
- GET_WSTR_OPT(pythonpath_env);
- GET_WSTR_OPT(home);
- GET_WSTR(platlibdir);
- // Path configuration output
- GET_UINT(module_search_paths_set);
- GET_WSTRLIST(module_search_paths);
- GET_WSTR_OPT(stdlib_dir);
- GET_WSTR_OPT(executable);
- GET_WSTR_OPT(base_executable);
- GET_WSTR_OPT(prefix);
- GET_WSTR_OPT(base_prefix);
- GET_WSTR_OPT(exec_prefix);
- GET_WSTR_OPT(base_exec_prefix);
- GET_UINT(skip_source_first_line);
- GET_WSTR_OPT(run_command);
- GET_WSTR_OPT(run_module);
- GET_WSTR_OPT(run_filename);
- GET_UINT(_install_importlib);
- GET_UINT(_init_main);
- GET_UINT(use_frozen_modules);
- GET_UINT(safe_path);
- GET_UINT(_is_python_build);
- GET_INT(int_max_str_digits);
- #undef CHECK_VALUE
- #undef GET_UINT
- #undef GET_INT
- #undef GET_WSTR
- #undef GET_WSTR_OPT
- return 0;
- }
- static const char*
- config_get_env(const PyConfig *config, const char *name)
- {
- return _Py_GetEnv(config->use_environment, name);
- }
- /* Get a copy of the environment variable as wchar_t*.
- Return 0 on success, but *dest can be NULL.
- Return -1 on memory allocation failure. Return -2 on decoding error. */
- static PyStatus
- config_get_env_dup(PyConfig *config,
- wchar_t **dest,
- wchar_t *wname, char *name,
- const char *decode_err_msg)
- {
- assert(*dest == NULL);
- assert(config->use_environment >= 0);
- if (!config->use_environment) {
- *dest = NULL;
- return _PyStatus_OK();
- }
- #ifdef MS_WINDOWS
- const wchar_t *var = _wgetenv(wname);
- if (!var || var[0] == '\0') {
- *dest = NULL;
- return _PyStatus_OK();
- }
- return PyConfig_SetString(config, dest, var);
- #else
- const char *var = getenv(name);
- if (!var || var[0] == '\0') {
- *dest = NULL;
- return _PyStatus_OK();
- }
- return config_set_bytes_string(config, dest, var, decode_err_msg);
- #endif
- }
- #define CONFIG_GET_ENV_DUP(CONFIG, DEST, WNAME, NAME) \
- config_get_env_dup(CONFIG, DEST, WNAME, NAME, "cannot decode " NAME)
- static void
- config_get_global_vars(PyConfig *config)
- {
- _Py_COMP_DIAG_PUSH
- _Py_COMP_DIAG_IGNORE_DEPR_DECLS
- if (config->_config_init != _PyConfig_INIT_COMPAT) {
- /* Python and Isolated configuration ignore global variables */
- return;
- }
- #define COPY_FLAG(ATTR, VALUE) \
- if (config->ATTR == -1) { \
- config->ATTR = VALUE; \
- }
- #define COPY_NOT_FLAG(ATTR, VALUE) \
- if (config->ATTR == -1) { \
- config->ATTR = !(VALUE); \
- }
- COPY_FLAG(isolated, Py_IsolatedFlag);
- COPY_NOT_FLAG(use_environment, Py_IgnoreEnvironmentFlag);
- COPY_FLAG(bytes_warning, Py_BytesWarningFlag);
- COPY_FLAG(inspect, Py_InspectFlag);
- COPY_FLAG(interactive, Py_InteractiveFlag);
- COPY_FLAG(optimization_level, Py_OptimizeFlag);
- COPY_FLAG(parser_debug, Py_DebugFlag);
- COPY_FLAG(verbose, Py_VerboseFlag);
- COPY_FLAG(quiet, Py_QuietFlag);
- #ifdef MS_WINDOWS
- COPY_FLAG(legacy_windows_stdio, Py_LegacyWindowsStdioFlag);
- #endif
- COPY_NOT_FLAG(pathconfig_warnings, Py_FrozenFlag);
- COPY_NOT_FLAG(buffered_stdio, Py_UnbufferedStdioFlag);
- COPY_NOT_FLAG(site_import, Py_NoSiteFlag);
- COPY_NOT_FLAG(write_bytecode, Py_DontWriteBytecodeFlag);
- COPY_NOT_FLAG(user_site_directory, Py_NoUserSiteDirectory);
- #undef COPY_FLAG
- #undef COPY_NOT_FLAG
- _Py_COMP_DIAG_POP
- }
- /* Set Py_xxx global configuration variables from 'config' configuration. */
- static void
- config_set_global_vars(const PyConfig *config)
- {
- _Py_COMP_DIAG_PUSH
- _Py_COMP_DIAG_IGNORE_DEPR_DECLS
- #define COPY_FLAG(ATTR, VAR) \
- if (config->ATTR != -1) { \
- VAR = config->ATTR; \
- }
- #define COPY_NOT_FLAG(ATTR, VAR) \
- if (config->ATTR != -1) { \
- VAR = !config->ATTR; \
- }
- COPY_FLAG(isolated, Py_IsolatedFlag);
- COPY_NOT_FLAG(use_environment, Py_IgnoreEnvironmentFlag);
- COPY_FLAG(bytes_warning, Py_BytesWarningFlag);
- COPY_FLAG(inspect, Py_InspectFlag);
- COPY_FLAG(interactive, Py_InteractiveFlag);
- COPY_FLAG(optimization_level, Py_OptimizeFlag);
- COPY_FLAG(parser_debug, Py_DebugFlag);
- COPY_FLAG(verbose, Py_VerboseFlag);
- COPY_FLAG(quiet, Py_QuietFlag);
- #ifdef MS_WINDOWS
- COPY_FLAG(legacy_windows_stdio, Py_LegacyWindowsStdioFlag);
- #endif
- COPY_NOT_FLAG(pathconfig_warnings, Py_FrozenFlag);
- COPY_NOT_FLAG(buffered_stdio, Py_UnbufferedStdioFlag);
- COPY_NOT_FLAG(site_import, Py_NoSiteFlag);
- COPY_NOT_FLAG(write_bytecode, Py_DontWriteBytecodeFlag);
- COPY_NOT_FLAG(user_site_directory, Py_NoUserSiteDirectory);
- /* Random or non-zero hash seed */
- Py_HashRandomizationFlag = (config->use_hash_seed == 0 ||
- config->hash_seed != 0);
- #undef COPY_FLAG
- #undef COPY_NOT_FLAG
- _Py_COMP_DIAG_POP
- }
- static const wchar_t*
- config_get_xoption(const PyConfig *config, wchar_t *name)
- {
- return _Py_get_xoption(&config->xoptions, name);
- }
- static const wchar_t*
- config_get_xoption_value(const PyConfig *config, wchar_t *name)
- {
- const wchar_t *xoption = config_get_xoption(config, name);
- if (xoption == NULL) {
- return NULL;
- }
- const wchar_t *sep = wcschr(xoption, L'=');
- return sep ? sep + 1 : L"";
- }
- static PyStatus
- config_init_hash_seed(PyConfig *config)
- {
- static_assert(sizeof(_Py_HashSecret_t) == sizeof(_Py_HashSecret.uc),
- "_Py_HashSecret_t has wrong size");
- const char *seed_text = config_get_env(config, "PYTHONHASHSEED");
- /* Convert a text seed to a numeric one */
- if (seed_text && strcmp(seed_text, "random") != 0) {
- const char *endptr = seed_text;
- unsigned long seed;
- errno = 0;
- seed = strtoul(seed_text, (char **)&endptr, 10);
- if (*endptr != '\0'
- || seed > MAX_HASH_SEED
- || (errno == ERANGE && seed == ULONG_MAX))
- {
- return _PyStatus_ERR("PYTHONHASHSEED must be \"random\" "
- "or an integer in range [0; 4294967295]");
- }
- /* Use a specific hash */
- config->use_hash_seed = 1;
- config->hash_seed = seed;
- }
- else {
- /* Use a random hash */
- config->use_hash_seed = 0;
- config->hash_seed = 0;
- }
- return _PyStatus_OK();
- }
- static int
- config_wstr_to_int(const wchar_t *wstr, int *result)
- {
- const wchar_t *endptr = wstr;
- errno = 0;
- long value = wcstol(wstr, (wchar_t **)&endptr, 10);
- if (*endptr != '\0' || errno == ERANGE) {
- return -1;
- }
- if (value < INT_MIN || value > INT_MAX) {
- return -1;
- }
- *result = (int)value;
- return 0;
- }
- static PyStatus
- config_read_env_vars(PyConfig *config)
- {
- PyStatus status;
- int use_env = config->use_environment;
- /* Get environment variables */
- _Py_get_env_flag(use_env, &config->parser_debug, "PYTHONDEBUG");
- _Py_get_env_flag(use_env, &config->verbose, "PYTHONVERBOSE");
- _Py_get_env_flag(use_env, &config->optimization_level, "PYTHONOPTIMIZE");
- _Py_get_env_flag(use_env, &config->inspect, "PYTHONINSPECT");
- int dont_write_bytecode = 0;
- _Py_get_env_flag(use_env, &dont_write_bytecode, "PYTHONDONTWRITEBYTECODE");
- if (dont_write_bytecode) {
- config->write_bytecode = 0;
- }
- int no_user_site_directory = 0;
- _Py_get_env_flag(use_env, &no_user_site_directory, "PYTHONNOUSERSITE");
- if (no_user_site_directory) {
- config->user_site_directory = 0;
- }
- int unbuffered_stdio = 0;
- _Py_get_env_flag(use_env, &unbuffered_stdio, "PYTHONUNBUFFERED");
- if (unbuffered_stdio) {
- config->buffered_stdio = 0;
- }
- #ifdef MS_WINDOWS
- _Py_get_env_flag(use_env, &config->legacy_windows_stdio,
- "PYTHONLEGACYWINDOWSSTDIO");
- #endif
- if (config_get_env(config, "PYTHONDUMPREFS")) {
- config->dump_refs = 1;
- }
- if (config_get_env(config, "PYTHONMALLOCSTATS")) {
- config->malloc_stats = 1;
- }
- if (config->dump_refs_file == NULL) {
- status = CONFIG_GET_ENV_DUP(config, &config->dump_refs_file,
- L"PYTHONDUMPREFSFILE", "PYTHONDUMPREFSFILE");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->pythonpath_env == NULL) {
- status = CONFIG_GET_ENV_DUP(config, &config->pythonpath_env,
- L"PYTHONPATH", "PYTHONPATH");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if(config->platlibdir == NULL) {
- status = CONFIG_GET_ENV_DUP(config, &config->platlibdir,
- L"PYTHONPLATLIBDIR", "PYTHONPLATLIBDIR");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->use_hash_seed < 0) {
- status = config_init_hash_seed(config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config_get_env(config, "PYTHONSAFEPATH")) {
- config->safe_path = 1;
- }
- return _PyStatus_OK();
- }
- static PyStatus
- config_init_perf_profiling(PyConfig *config)
- {
- int active = 0;
- const char *env = config_get_env(config, "PYTHONPERFSUPPORT");
- if (env) {
- if (_Py_str_to_int(env, &active) != 0) {
- active = 0;
- }
- if (active) {
- config->perf_profiling = 1;
- }
- }
- const wchar_t *xoption = config_get_xoption(config, L"perf");
- if (xoption) {
- config->perf_profiling = 1;
- }
- return _PyStatus_OK();
- }
- static PyStatus
- config_init_tracemalloc(PyConfig *config)
- {
- int nframe;
- int valid;
- const char *env = config_get_env(config, "PYTHONTRACEMALLOC");
- if (env) {
- if (!_Py_str_to_int(env, &nframe)) {
- valid = (nframe >= 0);
- }
- else {
- valid = 0;
- }
- if (!valid) {
- return _PyStatus_ERR("PYTHONTRACEMALLOC: invalid number of frames");
- }
- config->tracemalloc = nframe;
- }
- const wchar_t *xoption = config_get_xoption(config, L"tracemalloc");
- if (xoption) {
- const wchar_t *sep = wcschr(xoption, L'=');
- if (sep) {
- if (!config_wstr_to_int(sep + 1, &nframe)) {
- valid = (nframe >= 0);
- }
- else {
- valid = 0;
- }
- if (!valid) {
- return _PyStatus_ERR("-X tracemalloc=NFRAME: "
- "invalid number of frames");
- }
- }
- else {
- /* -X tracemalloc behaves as -X tracemalloc=1 */
- nframe = 1;
- }
- config->tracemalloc = nframe;
- }
- return _PyStatus_OK();
- }
- static PyStatus
- config_init_int_max_str_digits(PyConfig *config)
- {
- int maxdigits;
- const char *env = config_get_env(config, "PYTHONINTMAXSTRDIGITS");
- if (env) {
- bool valid = 0;
- if (!_Py_str_to_int(env, &maxdigits)) {
- valid = ((maxdigits == 0) || (maxdigits >= _PY_LONG_MAX_STR_DIGITS_THRESHOLD));
- }
- if (!valid) {
- #define STRINGIFY(VAL) _STRINGIFY(VAL)
- #define _STRINGIFY(VAL) #VAL
- return _PyStatus_ERR(
- "PYTHONINTMAXSTRDIGITS: invalid limit; must be >= "
- STRINGIFY(_PY_LONG_MAX_STR_DIGITS_THRESHOLD)
- " or 0 for unlimited.");
- }
- config->int_max_str_digits = maxdigits;
- }
- const wchar_t *xoption = config_get_xoption(config, L"int_max_str_digits");
- if (xoption) {
- const wchar_t *sep = wcschr(xoption, L'=');
- bool valid = 0;
- if (sep) {
- if (!config_wstr_to_int(sep + 1, &maxdigits)) {
- valid = ((maxdigits == 0) || (maxdigits >= _PY_LONG_MAX_STR_DIGITS_THRESHOLD));
- }
- }
- if (!valid) {
- return _PyStatus_ERR(
- "-X int_max_str_digits: invalid limit; must be >= "
- STRINGIFY(_PY_LONG_MAX_STR_DIGITS_THRESHOLD)
- " or 0 for unlimited.");
- #undef _STRINGIFY
- #undef STRINGIFY
- }
- config->int_max_str_digits = maxdigits;
- }
- if (config->int_max_str_digits < 0) {
- config->int_max_str_digits = _PY_LONG_DEFAULT_MAX_STR_DIGITS;
- }
- return _PyStatus_OK();
- }
- static PyStatus
- config_init_pycache_prefix(PyConfig *config)
- {
- assert(config->pycache_prefix == NULL);
- const wchar_t *xoption = config_get_xoption(config, L"pycache_prefix");
- if (xoption) {
- const wchar_t *sep = wcschr(xoption, L'=');
- if (sep && wcslen(sep) > 1) {
- config->pycache_prefix = _PyMem_RawWcsdup(sep + 1);
- if (config->pycache_prefix == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- }
- else {
- // PYTHONPYCACHEPREFIX env var ignored
- // if "-X pycache_prefix=" option is used
- config->pycache_prefix = NULL;
- }
- return _PyStatus_OK();
- }
- return CONFIG_GET_ENV_DUP(config, &config->pycache_prefix,
- L"PYTHONPYCACHEPREFIX",
- "PYTHONPYCACHEPREFIX");
- }
- static PyStatus
- config_read_complex_options(PyConfig *config)
- {
- /* More complex options configured by env var and -X option */
- if (config->faulthandler < 0) {
- if (config_get_env(config, "PYTHONFAULTHANDLER")
- || config_get_xoption(config, L"faulthandler")) {
- config->faulthandler = 1;
- }
- }
- if (config_get_env(config, "PYTHONPROFILEIMPORTTIME")
- || config_get_xoption(config, L"importtime")) {
- config->import_time = 1;
- }
- if (config_get_env(config, "PYTHONNODEBUGRANGES")
- || config_get_xoption(config, L"no_debug_ranges")) {
- config->code_debug_ranges = 0;
- }
- PyStatus status;
- if (config->tracemalloc < 0) {
- status = config_init_tracemalloc(config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->perf_profiling < 0) {
- status = config_init_perf_profiling(config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->int_max_str_digits < 0) {
- status = config_init_int_max_str_digits(config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->pycache_prefix == NULL) {
- status = config_init_pycache_prefix(config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- return _PyStatus_OK();
- }
- static const wchar_t *
- config_get_stdio_errors(const PyPreConfig *preconfig)
- {
- if (preconfig->utf8_mode) {
- /* UTF-8 Mode uses UTF-8/surrogateescape */
- return L"surrogateescape";
- }
- #ifndef MS_WINDOWS
- const char *loc = setlocale(LC_CTYPE, NULL);
- if (loc != NULL) {
- /* surrogateescape is the default in the legacy C and POSIX locales */
- if (strcmp(loc, "C") == 0 || strcmp(loc, "POSIX") == 0) {
- return L"surrogateescape";
- }
- #ifdef PY_COERCE_C_LOCALE
- /* surrogateescape is the default in locale coercion target locales */
- if (_Py_IsLocaleCoercionTarget(loc)) {
- return L"surrogateescape";
- }
- #endif
- }
- return L"strict";
- #else
- /* On Windows, always use surrogateescape by default */
- return L"surrogateescape";
- #endif
- }
- // See also config_get_fs_encoding()
- static PyStatus
- config_get_locale_encoding(PyConfig *config, const PyPreConfig *preconfig,
- wchar_t **locale_encoding)
- {
- wchar_t *encoding;
- if (preconfig->utf8_mode) {
- encoding = _PyMem_RawWcsdup(L"utf-8");
- }
- else {
- encoding = _Py_GetLocaleEncoding();
- }
- if (encoding == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- PyStatus status = PyConfig_SetString(config, locale_encoding, encoding);
- PyMem_RawFree(encoding);
- return status;
- }
- static PyStatus
- config_init_stdio_encoding(PyConfig *config,
- const PyPreConfig *preconfig)
- {
- PyStatus status;
- /* If Py_SetStandardStreamEncoding() has been called, use its
- arguments if they are not NULL. */
- if (config->stdio_encoding == NULL && _Py_StandardStreamEncoding != NULL) {
- status = CONFIG_SET_BYTES_STR(config, &config->stdio_encoding,
- _Py_StandardStreamEncoding,
- "_Py_StandardStreamEncoding");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->stdio_errors == NULL && _Py_StandardStreamErrors != NULL) {
- status = CONFIG_SET_BYTES_STR(config, &config->stdio_errors,
- _Py_StandardStreamErrors,
- "_Py_StandardStreamErrors");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- // Exit if encoding and errors are defined
- if (config->stdio_encoding != NULL && config->stdio_errors != NULL) {
- return _PyStatus_OK();
- }
- /* PYTHONIOENCODING environment variable */
- const char *opt = config_get_env(config, "PYTHONIOENCODING");
- if (opt) {
- char *pythonioencoding = _PyMem_RawStrdup(opt);
- if (pythonioencoding == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- char *errors = strchr(pythonioencoding, ':');
- if (errors) {
- *errors = '\0';
- errors++;
- if (!errors[0]) {
- errors = NULL;
- }
- }
- /* Does PYTHONIOENCODING contain an encoding? */
- if (pythonioencoding[0]) {
- if (config->stdio_encoding == NULL) {
- status = CONFIG_SET_BYTES_STR(config, &config->stdio_encoding,
- pythonioencoding,
- "PYTHONIOENCODING environment variable");
- if (_PyStatus_EXCEPTION(status)) {
- PyMem_RawFree(pythonioencoding);
- return status;
- }
- }
- /* If the encoding is set but not the error handler,
- use "strict" error handler by default.
- PYTHONIOENCODING=latin1 behaves as
- PYTHONIOENCODING=latin1:strict. */
- if (!errors) {
- errors = "strict";
- }
- }
- if (config->stdio_errors == NULL && errors != NULL) {
- status = CONFIG_SET_BYTES_STR(config, &config->stdio_errors,
- errors,
- "PYTHONIOENCODING environment variable");
- if (_PyStatus_EXCEPTION(status)) {
- PyMem_RawFree(pythonioencoding);
- return status;
- }
- }
- PyMem_RawFree(pythonioencoding);
- }
- /* Choose the default error handler based on the current locale. */
- if (config->stdio_encoding == NULL) {
- status = config_get_locale_encoding(config, preconfig,
- &config->stdio_encoding);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->stdio_errors == NULL) {
- const wchar_t *errors = config_get_stdio_errors(preconfig);
- assert(errors != NULL);
- status = PyConfig_SetString(config, &config->stdio_errors, errors);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- return _PyStatus_OK();
- }
- // See also config_get_locale_encoding()
- static PyStatus
- config_get_fs_encoding(PyConfig *config, const PyPreConfig *preconfig,
- wchar_t **fs_encoding)
- {
- #ifdef _Py_FORCE_UTF8_FS_ENCODING
- return PyConfig_SetString(config, fs_encoding, L"utf-8");
- #elif defined(MS_WINDOWS)
- const wchar_t *encoding;
- if (preconfig->legacy_windows_fs_encoding) {
- // Legacy Windows filesystem encoding: mbcs/replace
- encoding = L"mbcs";
- }
- else {
- // Windows defaults to utf-8/surrogatepass (PEP 529)
- encoding = L"utf-8";
- }
- return PyConfig_SetString(config, fs_encoding, encoding);
- #else // !MS_WINDOWS
- if (preconfig->utf8_mode) {
- return PyConfig_SetString(config, fs_encoding, L"utf-8");
- }
- if (_Py_GetForceASCII()) {
- return PyConfig_SetString(config, fs_encoding, L"ascii");
- }
- return config_get_locale_encoding(config, preconfig, fs_encoding);
- #endif // !MS_WINDOWS
- }
- static PyStatus
- config_init_fs_encoding(PyConfig *config, const PyPreConfig *preconfig)
- {
- PyStatus status;
- if (config->filesystem_encoding == NULL) {
- status = config_get_fs_encoding(config, preconfig,
- &config->filesystem_encoding);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->filesystem_errors == NULL) {
- const wchar_t *errors;
- #ifdef MS_WINDOWS
- if (preconfig->legacy_windows_fs_encoding) {
- errors = L"replace";
- }
- else {
- errors = L"surrogatepass";
- }
- #else
- errors = L"surrogateescape";
- #endif
- status = PyConfig_SetString(config, &config->filesystem_errors, errors);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- return _PyStatus_OK();
- }
- static PyStatus
- config_init_import(PyConfig *config, int compute_path_config)
- {
- PyStatus status;
- status = _PyConfig_InitPathConfig(config, compute_path_config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- /* -X frozen_modules=[on|off] */
- const wchar_t *value = config_get_xoption_value(config, L"frozen_modules");
- if (value == NULL) {
- }
- else if (wcscmp(value, L"on") == 0) {
- config->use_frozen_modules = 1;
- }
- else if (wcscmp(value, L"off") == 0) {
- config->use_frozen_modules = 0;
- }
- else if (wcslen(value) == 0) {
- // "-X frozen_modules" and "-X frozen_modules=" both imply "on".
- config->use_frozen_modules = 1;
- }
- else {
- return PyStatus_Error("bad value for option -X frozen_modules "
- "(expected \"on\" or \"off\")");
- }
- assert(config->use_frozen_modules >= 0);
- return _PyStatus_OK();
- }
- PyStatus
- _PyConfig_InitImportConfig(PyConfig *config)
- {
- return config_init_import(config, 1);
- }
- static PyStatus
- config_read(PyConfig *config, int compute_path_config)
- {
- PyStatus status;
- const PyPreConfig *preconfig = &_PyRuntime.preconfig;
- if (config->use_environment) {
- status = config_read_env_vars(config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- /* -X options */
- if (config_get_xoption(config, L"showrefcount")) {
- config->show_ref_count = 1;
- }
- #ifdef Py_STATS
- if (config_get_xoption(config, L"pystats")) {
- _py_stats = &_py_stats_struct;
- }
- #endif
- status = config_read_complex_options(config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- if (config->_install_importlib) {
- status = config_init_import(config, compute_path_config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- /* default values */
- if (config->dev_mode) {
- if (config->faulthandler < 0) {
- config->faulthandler = 1;
- }
- }
- if (config->faulthandler < 0) {
- config->faulthandler = 0;
- }
- if (config->tracemalloc < 0) {
- config->tracemalloc = 0;
- }
- if (config->perf_profiling < 0) {
- config->perf_profiling = 0;
- }
- if (config->use_hash_seed < 0) {
- config->use_hash_seed = 0;
- config->hash_seed = 0;
- }
- if (config->filesystem_encoding == NULL || config->filesystem_errors == NULL) {
- status = config_init_fs_encoding(config, preconfig);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- status = config_init_stdio_encoding(config, preconfig);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- if (config->argv.length < 1) {
- /* Ensure at least one (empty) argument is seen */
- status = PyWideStringList_Append(&config->argv, L"");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->check_hash_pycs_mode == NULL) {
- status = PyConfig_SetString(config, &config->check_hash_pycs_mode,
- L"default");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- if (config->configure_c_stdio < 0) {
- config->configure_c_stdio = 1;
- }
- // Only parse arguments once.
- if (config->parse_argv == 1) {
- config->parse_argv = 2;
- }
- return _PyStatus_OK();
- }
- static void
- config_init_stdio(const PyConfig *config)
- {
- #if defined(MS_WINDOWS) || defined(__CYGWIN__)
- /* don't translate newlines (\r\n <=> \n) */
- _setmode(fileno(stdin), O_BINARY);
- _setmode(fileno(stdout), O_BINARY);
- _setmode(fileno(stderr), O_BINARY);
- #endif
- if (!config->buffered_stdio) {
- #ifdef HAVE_SETVBUF
- setvbuf(stdin, (char *)NULL, _IONBF, BUFSIZ);
- setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ);
- setvbuf(stderr, (char *)NULL, _IONBF, BUFSIZ);
- #else /* !HAVE_SETVBUF */
- setbuf(stdin, (char *)NULL);
- setbuf(stdout, (char *)NULL);
- setbuf(stderr, (char *)NULL);
- #endif /* !HAVE_SETVBUF */
- }
- else if (config->interactive) {
- #ifdef MS_WINDOWS
- /* Doesn't have to have line-buffered -- use unbuffered */
- /* Any set[v]buf(stdin, ...) screws up Tkinter :-( */
- setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ);
- #else /* !MS_WINDOWS */
- #ifdef HAVE_SETVBUF
- setvbuf(stdin, (char *)NULL, _IOLBF, BUFSIZ);
- setvbuf(stdout, (char *)NULL, _IOLBF, BUFSIZ);
- #endif /* HAVE_SETVBUF */
- #endif /* !MS_WINDOWS */
- /* Leave stderr alone - it should be unbuffered anyway. */
- }
- }
- /* Write the configuration:
- - set Py_xxx global configuration variables
- - initialize C standard streams (stdin, stdout, stderr) */
- PyStatus
- _PyConfig_Write(const PyConfig *config, _PyRuntimeState *runtime)
- {
- config_set_global_vars(config);
- if (config->configure_c_stdio) {
- config_init_stdio(config);
- }
- /* Write the new pre-configuration into _PyRuntime */
- PyPreConfig *preconfig = &runtime->preconfig;
- preconfig->isolated = config->isolated;
- preconfig->use_environment = config->use_environment;
- preconfig->dev_mode = config->dev_mode;
- if (_Py_SetArgcArgv(config->orig_argv.length,
- config->orig_argv.items) < 0)
- {
- return _PyStatus_NO_MEMORY();
- }
- return _PyStatus_OK();
- }
- /* --- PyConfig command line parser -------------------------- */
- static void
- config_usage(int error, const wchar_t* program)
- {
- FILE *f = error ? stderr : stdout;
- fprintf(f, usage_line, program);
- if (error)
- fprintf(f, "Try `python -h' for more information.\n");
- else {
- fputs(usage_help, f);
- }
- }
- static void
- config_envvars_usage(void)
- {
- printf(usage_envvars, (wint_t)DELIM, (wint_t)DELIM, PYTHONHOMEHELP);
- }
- static void
- config_xoptions_usage(void)
- {
- puts(usage_xoptions);
- }
- static void
- config_complete_usage(const wchar_t* program)
- {
- config_usage(0, program);
- putchar('\n');
- config_envvars_usage();
- putchar('\n');
- config_xoptions_usage();
- }
- /* Parse the command line arguments */
- static PyStatus
- config_parse_cmdline(PyConfig *config, PyWideStringList *warnoptions,
- Py_ssize_t *opt_index)
- {
- PyStatus status;
- const PyWideStringList *argv = &config->argv;
- int print_version = 0;
- const wchar_t* program = config->program_name;
- if (!program && argv->length >= 1) {
- program = argv->items[0];
- }
- _PyOS_ResetGetOpt();
- do {
- int longindex = -1;
- int c = _PyOS_GetOpt(argv->length, argv->items, &longindex);
- if (c == EOF) {
- break;
- }
- if (c == 'c') {
- if (config->run_command == NULL) {
- /* -c is the last option; following arguments
- that look like options are left for the
- command to interpret. */
- size_t len = wcslen(_PyOS_optarg) + 1 + 1;
- wchar_t *command = PyMem_RawMalloc(sizeof(wchar_t) * len);
- if (command == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- memcpy(command, _PyOS_optarg, (len - 2) * sizeof(wchar_t));
- command[len - 2] = '\n';
- command[len - 1] = 0;
- config->run_command = command;
- }
- break;
- }
- if (c == 'm') {
- /* -m is the last option; following arguments
- that look like options are left for the
- module to interpret. */
- if (config->run_module == NULL) {
- config->run_module = _PyMem_RawWcsdup(_PyOS_optarg);
- if (config->run_module == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- }
- break;
- }
- switch (c) {
- // Integers represent long options, see Python/getopt.c
- case 0:
- // check-hash-based-pycs
- if (wcscmp(_PyOS_optarg, L"always") == 0
- || wcscmp(_PyOS_optarg, L"never") == 0
- || wcscmp(_PyOS_optarg, L"default") == 0)
- {
- status = PyConfig_SetString(config, &config->check_hash_pycs_mode,
- _PyOS_optarg);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- } else {
- fprintf(stderr, "--check-hash-based-pycs must be one of "
- "'default', 'always', or 'never'\n");
- config_usage(1, program);
- return _PyStatus_EXIT(2);
- }
- break;
- case 1:
- // help-all
- config_complete_usage(program);
- return _PyStatus_EXIT(0);
- case 2:
- // help-env
- config_envvars_usage();
- return _PyStatus_EXIT(0);
- case 3:
- // help-xoptions
- config_xoptions_usage();
- return _PyStatus_EXIT(0);
- case 'b':
- config->bytes_warning++;
- break;
- case 'd':
- config->parser_debug++;
- break;
- case 'i':
- config->inspect++;
- config->interactive++;
- break;
- case 'E':
- case 'I':
- case 'X':
- /* option handled by _PyPreCmdline_Read() */
- break;
- /* case 'J': reserved for Jython */
- case 'O':
- config->optimization_level++;
- break;
- case 'P':
- config->safe_path = 1;
- break;
- case 'B':
- config->write_bytecode = 0;
- break;
- case 's':
- config->user_site_directory = 0;
- break;
- case 'S':
- config->site_import = 0;
- break;
- case 't':
- /* ignored for backwards compatibility */
- break;
- case 'u':
- config->buffered_stdio = 0;
- break;
- case 'v':
- config->verbose++;
- break;
- case 'x':
- config->skip_source_first_line = 1;
- break;
- case 'h':
- case '?':
- config_usage(0, program);
- return _PyStatus_EXIT(0);
- case 'V':
- print_version++;
- break;
- case 'W':
- status = PyWideStringList_Append(warnoptions, _PyOS_optarg);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- break;
- case 'q':
- config->quiet++;
- break;
- case 'R':
- config->use_hash_seed = 0;
- break;
- /* This space reserved for other options */
- default:
- /* unknown argument: parsing failed */
- config_usage(1, program);
- return _PyStatus_EXIT(2);
- }
- } while (1);
- if (print_version) {
- printf("Python %s\n",
- (print_version >= 2) ? Py_GetVersion() : PY_VERSION);
- return _PyStatus_EXIT(0);
- }
- if (config->run_command == NULL && config->run_module == NULL
- && _PyOS_optind < argv->length
- && wcscmp(argv->items[_PyOS_optind], L"-") != 0
- && config->run_filename == NULL)
- {
- config->run_filename = _PyMem_RawWcsdup(argv->items[_PyOS_optind]);
- if (config->run_filename == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- }
- if (config->run_command != NULL || config->run_module != NULL) {
- /* Backup _PyOS_optind */
- _PyOS_optind--;
- }
- *opt_index = _PyOS_optind;
- return _PyStatus_OK();
- }
- #ifdef MS_WINDOWS
- # define WCSTOK wcstok_s
- #else
- # define WCSTOK wcstok
- #endif
- /* Get warning options from PYTHONWARNINGS environment variable. */
- static PyStatus
- config_init_env_warnoptions(PyConfig *config, PyWideStringList *warnoptions)
- {
- PyStatus status;
- /* CONFIG_GET_ENV_DUP requires dest to be initialized to NULL */
- wchar_t *env = NULL;
- status = CONFIG_GET_ENV_DUP(config, &env,
- L"PYTHONWARNINGS", "PYTHONWARNINGS");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- /* env var is not set or is empty */
- if (env == NULL) {
- return _PyStatus_OK();
- }
- wchar_t *warning, *context = NULL;
- for (warning = WCSTOK(env, L",", &context);
- warning != NULL;
- warning = WCSTOK(NULL, L",", &context))
- {
- status = PyWideStringList_Append(warnoptions, warning);
- if (_PyStatus_EXCEPTION(status)) {
- PyMem_RawFree(env);
- return status;
- }
- }
- PyMem_RawFree(env);
- return _PyStatus_OK();
- }
- static PyStatus
- warnoptions_append(PyConfig *config, PyWideStringList *options,
- const wchar_t *option)
- {
- /* config_init_warnoptions() add existing config warnoptions at the end:
- ensure that the new option is not already present in this list to
- prevent change the options order when config_init_warnoptions() is
- called twice. */
- if (_PyWideStringList_Find(&config->warnoptions, option)) {
- /* Already present: do nothing */
- return _PyStatus_OK();
- }
- if (_PyWideStringList_Find(options, option)) {
- /* Already present: do nothing */
- return _PyStatus_OK();
- }
- return PyWideStringList_Append(options, option);
- }
- static PyStatus
- warnoptions_extend(PyConfig *config, PyWideStringList *options,
- const PyWideStringList *options2)
- {
- const Py_ssize_t len = options2->length;
- wchar_t *const *items = options2->items;
- for (Py_ssize_t i = 0; i < len; i++) {
- PyStatus status = warnoptions_append(config, options, items[i]);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- return _PyStatus_OK();
- }
- static PyStatus
- config_init_warnoptions(PyConfig *config,
- const PyWideStringList *cmdline_warnoptions,
- const PyWideStringList *env_warnoptions,
- const PyWideStringList *sys_warnoptions)
- {
- PyStatus status;
- PyWideStringList options = _PyWideStringList_INIT;
- /* Priority of warnings options, lowest to highest:
- *
- * - any implicit filters added by _warnings.c/warnings.py
- * - PyConfig.dev_mode: "default" filter
- * - PYTHONWARNINGS environment variable
- * - '-W' command line options
- * - PyConfig.bytes_warning ('-b' and '-bb' command line options):
- * "default::BytesWarning" or "error::BytesWarning" filter
- * - early PySys_AddWarnOption() calls
- * - PyConfig.warnoptions
- *
- * PyConfig.warnoptions is copied to sys.warnoptions. Since the warnings
- * module works on the basis of "the most recently added filter will be
- * checked first", we add the lowest precedence entries first so that later
- * entries override them.
- */
- if (config->dev_mode) {
- status = warnoptions_append(config, &options, L"default");
- if (_PyStatus_EXCEPTION(status)) {
- goto error;
- }
- }
- status = warnoptions_extend(config, &options, env_warnoptions);
- if (_PyStatus_EXCEPTION(status)) {
- goto error;
- }
- status = warnoptions_extend(config, &options, cmdline_warnoptions);
- if (_PyStatus_EXCEPTION(status)) {
- goto error;
- }
- /* If the bytes_warning_flag isn't set, bytesobject.c and bytearrayobject.c
- * don't even try to emit a warning, so we skip setting the filter in that
- * case.
- */
- if (config->bytes_warning) {
- const wchar_t *filter;
- if (config->bytes_warning> 1) {
- filter = L"error::BytesWarning";
- }
- else {
- filter = L"default::BytesWarning";
- }
- status = warnoptions_append(config, &options, filter);
- if (_PyStatus_EXCEPTION(status)) {
- goto error;
- }
- }
- status = warnoptions_extend(config, &options, sys_warnoptions);
- if (_PyStatus_EXCEPTION(status)) {
- goto error;
- }
- /* Always add all PyConfig.warnoptions options */
- status = _PyWideStringList_Extend(&options, &config->warnoptions);
- if (_PyStatus_EXCEPTION(status)) {
- goto error;
- }
- _PyWideStringList_Clear(&config->warnoptions);
- config->warnoptions = options;
- return _PyStatus_OK();
- error:
- _PyWideStringList_Clear(&options);
- return status;
- }
- static PyStatus
- config_update_argv(PyConfig *config, Py_ssize_t opt_index)
- {
- const PyWideStringList *cmdline_argv = &config->argv;
- PyWideStringList config_argv = _PyWideStringList_INIT;
- /* Copy argv to be able to modify it (to force -c/-m) */
- if (cmdline_argv->length <= opt_index) {
- /* Ensure at least one (empty) argument is seen */
- PyStatus status = PyWideStringList_Append(&config_argv, L"");
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- }
- else {
- PyWideStringList slice;
- slice.length = cmdline_argv->length - opt_index;
- slice.items = &cmdline_argv->items[opt_index];
- if (_PyWideStringList_Copy(&config_argv, &slice) < 0) {
- return _PyStatus_NO_MEMORY();
- }
- }
- assert(config_argv.length >= 1);
- wchar_t *arg0 = NULL;
- if (config->run_command != NULL) {
- /* Force sys.argv[0] = '-c' */
- arg0 = L"-c";
- }
- else if (config->run_module != NULL) {
- /* Force sys.argv[0] = '-m'*/
- arg0 = L"-m";
- }
- if (arg0 != NULL) {
- arg0 = _PyMem_RawWcsdup(arg0);
- if (arg0 == NULL) {
- _PyWideStringList_Clear(&config_argv);
- return _PyStatus_NO_MEMORY();
- }
- PyMem_RawFree(config_argv.items[0]);
- config_argv.items[0] = arg0;
- }
- _PyWideStringList_Clear(&config->argv);
- config->argv = config_argv;
- return _PyStatus_OK();
- }
- static PyStatus
- core_read_precmdline(PyConfig *config, _PyPreCmdline *precmdline)
- {
- PyStatus status;
- if (config->parse_argv == 1) {
- if (_PyWideStringList_Copy(&precmdline->argv, &config->argv) < 0) {
- return _PyStatus_NO_MEMORY();
- }
- }
- PyPreConfig preconfig;
- status = _PyPreConfig_InitFromPreConfig(&preconfig, &_PyRuntime.preconfig);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- _PyPreConfig_GetConfig(&preconfig, config);
- status = _PyPreCmdline_Read(precmdline, &preconfig);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- status = _PyPreCmdline_SetConfig(precmdline, config);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- return _PyStatus_OK();
- }
- /* Get run_filename absolute path */
- static PyStatus
- config_run_filename_abspath(PyConfig *config)
- {
- if (!config->run_filename) {
- return _PyStatus_OK();
- }
- #ifndef MS_WINDOWS
- if (_Py_isabs(config->run_filename)) {
- /* path is already absolute */
- return _PyStatus_OK();
- }
- #endif
- wchar_t *abs_filename;
- if (_Py_abspath(config->run_filename, &abs_filename) < 0) {
- /* failed to get the absolute path of the command line filename:
- ignore the error, keep the relative path */
- return _PyStatus_OK();
- }
- if (abs_filename == NULL) {
- return _PyStatus_NO_MEMORY();
- }
- PyMem_RawFree(config->run_filename);
- config->run_filename = abs_filename;
- return _PyStatus_OK();
- }
- static PyStatus
- config_read_cmdline(PyConfig *config)
- {
- PyStatus status;
- PyWideStringList cmdline_warnoptions = _PyWideStringList_INIT;
- PyWideStringList env_warnoptions = _PyWideStringList_INIT;
- PyWideStringList sys_warnoptions = _PyWideStringList_INIT;
- if (config->parse_argv < 0) {
- config->parse_argv = 1;
- }
- if (config->parse_argv == 1) {
- Py_ssize_t opt_index;
- status = config_parse_cmdline(config, &cmdline_warnoptions, &opt_index);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- status = config_run_filename_abspath(config);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- status = config_update_argv(config, opt_index);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- }
- else {
- status = config_run_filename_abspath(config);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- }
- if (config->use_environment) {
- status = config_init_env_warnoptions(config, &env_warnoptions);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- }
- /* Handle early PySys_AddWarnOption() calls */
- status = _PySys_ReadPreinitWarnOptions(&sys_warnoptions);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- status = config_init_warnoptions(config,
- &cmdline_warnoptions,
- &env_warnoptions,
- &sys_warnoptions);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- status = _PyStatus_OK();
- done:
- _PyWideStringList_Clear(&cmdline_warnoptions);
- _PyWideStringList_Clear(&env_warnoptions);
- _PyWideStringList_Clear(&sys_warnoptions);
- return status;
- }
- PyStatus
- _PyConfig_SetPyArgv(PyConfig *config, const _PyArgv *args)
- {
- PyStatus status = _Py_PreInitializeFromConfig(config, args);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- return _PyArgv_AsWstrList(args, &config->argv);
- }
- /* Set config.argv: decode argv using Py_DecodeLocale(). Pre-initialize Python
- if needed to ensure that encodings are properly configured. */
- PyStatus
- PyConfig_SetBytesArgv(PyConfig *config, Py_ssize_t argc, char * const *argv)
- {
- _PyArgv args = {
- .argc = argc,
- .use_bytes_argv = 1,
- .bytes_argv = argv,
- .wchar_argv = NULL};
- return _PyConfig_SetPyArgv(config, &args);
- }
- PyStatus
- PyConfig_SetArgv(PyConfig *config, Py_ssize_t argc, wchar_t * const *argv)
- {
- _PyArgv args = {
- .argc = argc,
- .use_bytes_argv = 0,
- .bytes_argv = NULL,
- .wchar_argv = argv};
- return _PyConfig_SetPyArgv(config, &args);
- }
- PyStatus
- PyConfig_SetWideStringList(PyConfig *config, PyWideStringList *list,
- Py_ssize_t length, wchar_t **items)
- {
- PyStatus status = _Py_PreInitializeFromConfig(config, NULL);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- PyWideStringList list2 = {.length = length, .items = items};
- if (_PyWideStringList_Copy(list, &list2) < 0) {
- return _PyStatus_NO_MEMORY();
- }
- return _PyStatus_OK();
- }
- /* Read the configuration into PyConfig from:
- * Command line arguments
- * Environment variables
- * Py_xxx global configuration variables
- The only side effects are to modify config and to call _Py_SetArgcArgv(). */
- PyStatus
- _PyConfig_Read(PyConfig *config, int compute_path_config)
- {
- PyStatus status;
- status = _Py_PreInitializeFromConfig(config, NULL);
- if (_PyStatus_EXCEPTION(status)) {
- return status;
- }
- config_get_global_vars(config);
- if (config->orig_argv.length == 0
- && !(config->argv.length == 1
- && wcscmp(config->argv.items[0], L"") == 0))
- {
- if (_PyWideStringList_Copy(&config->orig_argv, &config->argv) < 0) {
- return _PyStatus_NO_MEMORY();
- }
- }
- _PyPreCmdline precmdline = _PyPreCmdline_INIT;
- status = core_read_precmdline(config, &precmdline);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- assert(config->isolated >= 0);
- if (config->isolated) {
- config->safe_path = 1;
- config->use_environment = 0;
- config->user_site_directory = 0;
- }
- status = config_read_cmdline(config);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- /* Handle early PySys_AddXOption() calls */
- status = _PySys_ReadPreinitXOptions(config);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- status = config_read(config, compute_path_config);
- if (_PyStatus_EXCEPTION(status)) {
- goto done;
- }
- assert(config_check_consistency(config));
- status = _PyStatus_OK();
- done:
- _PyPreCmdline_Clear(&precmdline);
- return status;
- }
- PyStatus
- PyConfig_Read(PyConfig *config)
- {
- return _PyConfig_Read(config, 0);
- }
- PyObject*
- _Py_GetConfigsAsDict(void)
- {
- PyObject *result = NULL;
- PyObject *dict = NULL;
- result = PyDict_New();
- if (result == NULL) {
- goto error;
- }
- /* global result */
- dict = _Py_GetGlobalVariablesAsDict();
- if (dict == NULL) {
- goto error;
- }
- if (PyDict_SetItemString(result, "global_config", dict) < 0) {
- goto error;
- }
- Py_CLEAR(dict);
- /* pre config */
- PyInterpreterState *interp = _PyInterpreterState_GET();
- const PyPreConfig *pre_config = &interp->runtime->preconfig;
- dict = _PyPreConfig_AsDict(pre_config);
- if (dict == NULL) {
- goto error;
- }
- if (PyDict_SetItemString(result, "pre_config", dict) < 0) {
- goto error;
- }
- Py_CLEAR(dict);
- /* core config */
- const PyConfig *config = _PyInterpreterState_GetConfig(interp);
- dict = _PyConfig_AsDict(config);
- if (dict == NULL) {
- goto error;
- }
- if (PyDict_SetItemString(result, "config", dict) < 0) {
- goto error;
- }
- Py_CLEAR(dict);
- return result;
- error:
- Py_XDECREF(result);
- Py_XDECREF(dict);
- return NULL;
- }
- static void
- init_dump_ascii_wstr(const wchar_t *str)
- {
- if (str == NULL) {
- PySys_WriteStderr("(not set)");
- return;
- }
- PySys_WriteStderr("'");
- for (; *str != L'\0'; str++) {
- unsigned int ch = (unsigned int)*str;
- if (ch == L'\'') {
- PySys_WriteStderr("\\'");
- } else if (0x20 <= ch && ch < 0x7f) {
- PySys_WriteStderr("%c", ch);
- }
- else if (ch <= 0xff) {
- PySys_WriteStderr("\\x%02x", ch);
- }
- #if SIZEOF_WCHAR_T > 2
- else if (ch > 0xffff) {
- PySys_WriteStderr("\\U%08x", ch);
- }
- #endif
- else {
- PySys_WriteStderr("\\u%04x", ch);
- }
- }
- PySys_WriteStderr("'");
- }
- /* Dump the Python path configuration into sys.stderr */
- void
- _Py_DumpPathConfig(PyThreadState *tstate)
- {
- PyObject *exc = _PyErr_GetRaisedException(tstate);
- PySys_WriteStderr("Python path configuration:\n");
- #define DUMP_CONFIG(NAME, FIELD) \
- do { \
- PySys_WriteStderr(" " NAME " = "); \
- init_dump_ascii_wstr(config->FIELD); \
- PySys_WriteStderr("\n"); \
- } while (0)
- const PyConfig *config = _PyInterpreterState_GetConfig(tstate->interp);
- DUMP_CONFIG("PYTHONHOME", home);
- DUMP_CONFIG("PYTHONPATH", pythonpath_env);
- DUMP_CONFIG("program name", program_name);
- PySys_WriteStderr(" isolated = %i\n", config->isolated);
- PySys_WriteStderr(" environment = %i\n", config->use_environment);
- PySys_WriteStderr(" user site = %i\n", config->user_site_directory);
- PySys_WriteStderr(" safe_path = %i\n", config->safe_path);
- PySys_WriteStderr(" import site = %i\n", config->site_import);
- PySys_WriteStderr(" is in build tree = %i\n", config->_is_python_build);
- DUMP_CONFIG("stdlib dir", stdlib_dir);
- #undef DUMP_CONFIG
- #define DUMP_SYS(NAME) \
- do { \
- obj = PySys_GetObject(#NAME); \
- PySys_FormatStderr(" sys.%s = ", #NAME); \
- if (obj != NULL) { \
- PySys_FormatStderr("%A", obj); \
- } \
- else { \
- PySys_WriteStderr("(not set)"); \
- } \
- PySys_FormatStderr("\n"); \
- } while (0)
- PyObject *obj;
- DUMP_SYS(_base_executable);
- DUMP_SYS(base_prefix);
- DUMP_SYS(base_exec_prefix);
- DUMP_SYS(platlibdir);
- DUMP_SYS(executable);
- DUMP_SYS(prefix);
- DUMP_SYS(exec_prefix);
- #undef DUMP_SYS
- PyObject *sys_path = PySys_GetObject("path"); /* borrowed reference */
- if (sys_path != NULL && PyList_Check(sys_path)) {
- PySys_WriteStderr(" sys.path = [\n");
- Py_ssize_t len = PyList_GET_SIZE(sys_path);
- for (Py_ssize_t i=0; i < len; i++) {
- PyObject *path = PyList_GET_ITEM(sys_path, i);
- PySys_FormatStderr(" %A,\n", path);
- }
- PySys_WriteStderr(" ]\n");
- }
- _PyErr_SetRaisedException(tstate, exc);
- }
|