1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205 |
- #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 str(bytes_instance), str(bytearray_instance)\n\
- and comparing bytes/bytearray with str. (-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 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\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\
- 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\
- \n\
- -X faulthandler: enable faulthandler\n\
- \n\
- -X showrefcount: output the total reference count and number of used\n\
- memory blocks when the program finishes or after each statement in the\n\
- interactive interpreter. This only works on debug builds\n\
- \n\
- -X tracemalloc: start tracing Python memory allocations using the\n\
- tracemalloc module. By default, only the most recent frame is stored in a\n\
- traceback of a trace. Use -X tracemalloc=NFRAME to start tracing with a\n\
- traceback limit of NFRAME frames\n\
- \n\
- -X importtime: show how long each import takes. It shows module name,\n\
- cumulative time (including nested imports) and self time (excluding\n\
- nested imports). Note that its output may be broken in multi-threaded\n\
- application. Typical usage is python3 -X importtime -c 'import asyncio'\n\
- \n\
- -X dev: enable CPython's \"development mode\", introducing additional runtime\n\
- checks which are too expensive to be enabled by default. Effect of the\n\
- developer mode:\n\
- * Add default warning filter, as -W default\n\
- * Install debug hooks on memory allocators: see the PyMem_SetupDebugHooks()\n\
- C function\n\
- * Enable the faulthandler module to dump the Python traceback on a crash\n\
- * Enable asyncio debug mode\n\
- * Set the dev_mode attribute of sys.flags to True\n\
- * io.IOBase destructor logs close() exceptions\n\
- \n\
- -X utf8: enable UTF-8 mode for operating system interfaces, overriding the default\n\
- locale-aware mode. -X utf8=0 explicitly disables UTF-8 mode (even when it would\n\
- otherwise activate automatically)\n\
- \n\
- -X pycache_prefix=PATH: enable writing .pyc files to a parallel tree rooted at the\n\
- given directory instead of to the code tree\n\
- \n\
- -X warn_default_encoding: enable opt-in EncodingWarning for 'encoding=None'\n\
- \n\
- -X no_debug_ranges: disable the inclusion of the tables mapping extra location \n\
- information (end line, start column offset and end column offset) to every \n\
- instruction in code objects. This is useful when smaller code objects and pyc \n\
- files are desired as well as suppressing the extra visual location indicators \n\
- when the interpreter displays tracebacks.\n\
- \n\
- -X perf: activate support for the Linux \"perf\" profiler by activating the \"perf\"\n\
- trampoline. When this option is activated, the Linux \"perf\" profiler will be \n\
- able to report Python calls. This option is only available on some platforms and will \n\
- do nothing if is not supported on the current system. The default value is \"off\".\n\
- \n\
- -X frozen_modules=[on|off]: whether or not frozen modules should be used.\n\
- The default is \"on\" (or \"off\" if you are running a local build).\n\
- \n\
- -X int_max_str_digits=number: limit the size of int<->str conversions.\n\
- This helps avoid denial of service attacks when parsing untrusted data.\n\
- The default is sys.int_info.default_max_str_digits. 0 disables."
- #ifdef Py_STATS
- "\n\
- \n\
- -X pystats: Enable pystats collection at startup."
- #endif
- ;
- /* 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"
- "PYTHONUTF8: if set to 1, enable the UTF-8 mode.\n"
- "PYTHONIOENCODING: Encoding[:errors] used for stdin/stdout/stderr.\n"
- "PYTHONFAULTHANDLER: dump the Python traceback on fatal errors.\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 set to an\n"
- " integer in the range [0,4294967295] to get hash values with a\n"
- " predictable seed.\n"
- "PYTHONINTMAXSTRDIGITS: limits the maximum digit characters in an int value\n"
- " when converting from a string and when converting an int back to a str.\n"
- " A value of 0 disables the limit. Conversions to or from bases 2, 4, 8,\n"
- " 16, and 32 are never limited.\n"
- "PYTHONMALLOC: set the Python memory allocators and/or install debug hooks\n"
- " on Python memory allocators. Use PYTHONMALLOC=debug to install debug\n"
- " hooks.\n"
- "PYTHONCOERCECLOCALE: if this variable is set to 0, it disables the locale\n"
- " coercion behavior. Use PYTHONCOERCECLOCALE=warn to request display of\n"
- " locale coercion and locale compatibility warnings 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 choice.\n"
- "PYTHONDEVMODE: enable the development mode.\n"
- "PYTHONPYCACHEPREFIX: root directory for bytecode cache (pyc) files.\n"
- "PYTHONWARNDEFAULTENCODING: enable opt-in EncodingWarning for 'encoding=None'.\n"
- "PYTHONNODEBUGRANGES: If this variable is set, it disables the inclusion of the \n"
- " tables mapping extra location information (end line, start column offset \n"
- " and end column offset) to every instruction in code objects. This is useful \n"
- " when smaller code objects and pyc files are desired as well as suppressing the \n"
- " extra visual location indicators when the interpreter displays tracebacks.\n"
- "These variables have equivalent command-line parameters (see --help for details):\n"
- "PYTHONDEBUG : enable parser debug mode (-d)\n"
- "PYTHONDONTWRITEBYTECODE : don't write .pyc files (-B)\n"
- "PYTHONINSPECT : inspect interactively after running script (-i)\n"
- "PYTHONINTMAXSTRDIGITS : limit max digit characters in an int value\n"
- " (-X int_max_str_digits=number)\n"
- "PYTHONNOUSERSITE : disable user site directory (-s)\n"
- "PYTHONOPTIMIZE : enable level 1 optimizations (-O)\n"
- "PYTHONSAFEPATH : don't prepend a potentially unsafe path to sys.path (-P)\n"
- "PYTHONUNBUFFERED : disable stdout/stderr buffering (-u)\n"
- "PYTHONVERBOSE : trace import statements (-v)\n"
- "PYTHONWARNINGS=arg : warning control (-W arg)\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);
- puts("\n");
- config_envvars_usage();
- puts("\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);
- }
|