1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626562756285629563056315632563356345635563656375638563956405641564256435644564556465647564856495650565156525653565456555656565756585659566056615662566356645665566656675668566956705671567256735674567556765677567856795680568156825683568456855686568756885689569056915692569356945695569656975698569957005701570257035704570557065707570857095710571157125713571457155716571757185719572057215722572357245725572657275728572957305731573257335734573557365737573857395740574157425743574457455746574757485749575057515752575357545755575657575758575957605761576257635764576557665767576857695770577157725773577457755776577757785779578057815782578357845785578657875788578957905791579257935794579557965797579857995800580158025803580458055806580758085809581058115812581358145815581658175818581958205821582258235824582558265827582858295830583158325833583458355836583758385839584058415842584358445845584658475848584958505851585258535854585558565857585858595860586158625863586458655866586758685869587058715872587358745875587658775878587958805881588258835884588558865887588858895890589158925893589458955896589758985899590059015902590359045905590659075908590959105911591259135914591559165917591859195920592159225923592459255926592759285929593059315932593359345935593659375938593959405941594259435944594559465947594859495950595159525953595459555956595759585959596059615962596359645965596659675968596959705971597259735974597559765977597859795980598159825983598459855986598759885989599059915992599359945995599659975998599960006001600260036004600560066007600860096010601160126013601460156016601760186019602060216022602360246025602660276028602960306031603260336034603560366037603860396040604160426043604460456046604760486049605060516052605360546055605660576058605960606061606260636064606560666067606860696070607160726073607460756076607760786079608060816082608360846085608660876088608960906091609260936094609560966097609860996100610161026103610461056106610761086109611061116112611361146115611661176118611961206121612261236124612561266127612861296130613161326133613461356136613761386139614061416142614361446145614661476148614961506151615261536154615561566157615861596160616161626163616461656166616761686169617061716172617361746175617661776178617961806181618261836184618561866187618861896190619161926193619461956196619761986199620062016202620362046205620662076208620962106211621262136214621562166217621862196220622162226223622462256226622762286229623062316232623362346235623662376238623962406241624262436244624562466247624862496250625162526253625462556256625762586259626062616262626362646265626662676268626962706271627262736274627562766277627862796280628162826283628462856286628762886289629062916292629362946295629662976298629963006301630263036304630563066307630863096310631163126313631463156316631763186319632063216322632363246325632663276328632963306331633263336334633563366337633863396340634163426343634463456346634763486349635063516352635363546355635663576358635963606361636263636364636563666367636863696370637163726373637463756376637763786379638063816382638363846385638663876388638963906391639263936394639563966397639863996400640164026403640464056406640764086409641064116412641364146415641664176418641964206421642264236424642564266427642864296430643164326433643464356436643764386439644064416442644364446445644664476448644964506451645264536454645564566457645864596460646164626463646464656466646764686469647064716472647364746475647664776478647964806481648264836484648564866487648864896490649164926493649464956496649764986499650065016502650365046505650665076508650965106511651265136514651565166517651865196520652165226523652465256526652765286529653065316532653365346535653665376538653965406541654265436544654565466547654865496550655165526553655465556556655765586559656065616562656365646565656665676568656965706571657265736574657565766577657865796580658165826583658465856586658765886589659065916592659365946595659665976598659966006601660266036604660566066607660866096610661166126613661466156616661766186619662066216622662366246625662666276628662966306631663266336634663566366637663866396640664166426643664466456646664766486649665066516652665366546655665666576658665966606661666266636664666566666667666866696670667166726673667466756676667766786679668066816682668366846685668666876688668966906691669266936694669566966697669866996700670167026703670467056706670767086709671067116712671367146715671667176718671967206721672267236724672567266727672867296730673167326733673467356736673767386739674067416742674367446745674667476748674967506751675267536754675567566757675867596760676167626763676467656766676767686769677067716772677367746775677667776778677967806781678267836784678567866787678867896790679167926793679467956796679767986799680068016802680368046805680668076808680968106811681268136814681568166817681868196820682168226823682468256826682768286829683068316832683368346835683668376838683968406841684268436844684568466847684868496850685168526853685468556856685768586859686068616862686368646865686668676868686968706871687268736874687568766877687868796880688168826883688468856886688768886889689068916892689368946895689668976898689969006901690269036904690569066907690869096910691169126913691469156916691769186919692069216922692369246925692669276928692969306931693269336934693569366937693869396940694169426943694469456946694769486949695069516952695369546955695669576958695969606961696269636964696569666967696869696970697169726973697469756976697769786979698069816982698369846985698669876988698969906991699269936994699569966997699869997000700170027003700470057006700770087009701070117012701370147015701670177018701970207021702270237024702570267027702870297030703170327033703470357036703770387039704070417042704370447045704670477048704970507051705270537054705570567057705870597060706170627063706470657066706770687069707070717072707370747075707670777078707970807081708270837084708570867087708870897090709170927093709470957096709770987099710071017102710371047105710671077108710971107111711271137114711571167117711871197120712171227123712471257126712771287129713071317132713371347135713671377138713971407141714271437144714571467147714871497150715171527153715471557156715771587159716071617162716371647165716671677168716971707171717271737174717571767177717871797180718171827183718471857186718771887189719071917192719371947195719671977198719972007201720272037204720572067207720872097210721172127213721472157216721772187219722072217222722372247225722672277228722972307231723272337234723572367237723872397240724172427243724472457246724772487249725072517252725372547255725672577258725972607261726272637264726572667267726872697270727172727273727472757276727772787279728072817282728372847285728672877288728972907291729272937294729572967297729872997300730173027303730473057306730773087309731073117312731373147315731673177318731973207321732273237324732573267327732873297330733173327333733473357336733773387339734073417342734373447345734673477348734973507351735273537354735573567357735873597360736173627363736473657366736773687369737073717372737373747375737673777378737973807381738273837384738573867387738873897390739173927393739473957396739773987399740074017402740374047405740674077408740974107411741274137414741574167417741874197420742174227423742474257426742774287429743074317432743374347435743674377438743974407441744274437444744574467447744874497450745174527453745474557456745774587459746074617462746374647465746674677468746974707471747274737474747574767477747874797480748174827483748474857486748774887489749074917492749374947495749674977498749975007501750275037504750575067507750875097510751175127513751475157516751775187519752075217522752375247525752675277528752975307531753275337534753575367537753875397540754175427543754475457546754775487549755075517552755375547555755675577558755975607561756275637564756575667567756875697570757175727573757475757576757775787579758075817582758375847585758675877588758975907591759275937594759575967597759875997600760176027603760476057606760776087609761076117612761376147615761676177618761976207621762276237624762576267627762876297630763176327633763476357636763776387639764076417642764376447645764676477648764976507651765276537654765576567657765876597660766176627663766476657666766776687669767076717672767376747675767676777678767976807681768276837684768576867687768876897690769176927693769476957696769776987699770077017702770377047705770677077708770977107711771277137714771577167717771877197720772177227723772477257726772777287729773077317732773377347735773677377738773977407741774277437744774577467747774877497750775177527753775477557756775777587759776077617762776377647765776677677768776977707771777277737774777577767777777877797780778177827783778477857786778777887789779077917792779377947795779677977798779978007801780278037804780578067807780878097810781178127813781478157816781778187819782078217822782378247825782678277828782978307831783278337834783578367837783878397840784178427843784478457846784778487849785078517852785378547855785678577858785978607861786278637864786578667867786878697870787178727873787478757876787778787879788078817882788378847885788678877888788978907891789278937894789578967897789878997900790179027903790479057906790779087909791079117912791379147915791679177918791979207921792279237924792579267927792879297930793179327933793479357936793779387939794079417942794379447945794679477948794979507951795279537954795579567957795879597960796179627963796479657966796779687969797079717972797379747975797679777978797979807981798279837984798579867987798879897990799179927993799479957996799779987999800080018002800380048005800680078008800980108011801280138014801580168017801880198020802180228023802480258026802780288029803080318032803380348035803680378038803980408041804280438044804580468047804880498050805180528053805480558056805780588059806080618062806380648065806680678068806980708071807280738074807580768077807880798080808180828083808480858086808780888089809080918092809380948095809680978098809981008101810281038104810581068107810881098110811181128113811481158116811781188119812081218122812381248125812681278128812981308131813281338134813581368137813881398140814181428143814481458146814781488149815081518152815381548155815681578158815981608161816281638164816581668167816881698170817181728173817481758176817781788179818081818182818381848185818681878188818981908191819281938194819581968197819881998200820182028203820482058206820782088209821082118212821382148215821682178218821982208221822282238224822582268227822882298230823182328233823482358236823782388239824082418242824382448245824682478248824982508251825282538254825582568257825882598260826182628263826482658266826782688269827082718272827382748275827682778278827982808281828282838284828582868287828882898290829182928293829482958296829782988299830083018302830383048305830683078308830983108311831283138314831583168317831883198320832183228323832483258326832783288329833083318332833383348335833683378338833983408341834283438344834583468347834883498350835183528353835483558356835783588359836083618362836383648365836683678368836983708371837283738374837583768377837883798380838183828383838483858386838783888389839083918392839383948395839683978398839984008401840284038404840584068407840884098410841184128413841484158416841784188419842084218422842384248425842684278428842984308431843284338434843584368437843884398440844184428443844484458446844784488449845084518452845384548455845684578458845984608461846284638464846584668467846884698470847184728473847484758476847784788479848084818482848384848485848684878488848984908491849284938494849584968497849884998500850185028503850485058506850785088509851085118512851385148515851685178518851985208521852285238524852585268527852885298530853185328533853485358536853785388539854085418542854385448545854685478548854985508551855285538554855585568557855885598560856185628563856485658566856785688569857085718572857385748575857685778578857985808581858285838584858585868587858885898590859185928593859485958596859785988599860086018602860386048605860686078608860986108611861286138614861586168617861886198620862186228623862486258626862786288629863086318632863386348635863686378638863986408641864286438644864586468647864886498650865186528653865486558656865786588659866086618662866386648665866686678668866986708671867286738674867586768677867886798680868186828683868486858686868786888689869086918692869386948695869686978698869987008701870287038704870587068707870887098710871187128713871487158716871787188719872087218722872387248725872687278728872987308731873287338734873587368737873887398740874187428743874487458746874787488749875087518752875387548755875687578758875987608761876287638764876587668767876887698770877187728773877487758776877787788779878087818782878387848785878687878788878987908791879287938794879587968797879887998800880188028803880488058806880788088809881088118812881388148815881688178818881988208821882288238824882588268827882888298830883188328833883488358836883788388839884088418842884388448845884688478848884988508851885288538854885588568857885888598860886188628863886488658866886788688869887088718872887388748875887688778878887988808881888288838884888588868887888888898890889188928893889488958896889788988899890089018902890389048905890689078908890989108911891289138914891589168917891889198920892189228923892489258926892789288929893089318932893389348935893689378938893989408941894289438944894589468947894889498950895189528953895489558956895789588959896089618962896389648965896689678968896989708971897289738974897589768977897889798980898189828983898489858986898789888989899089918992899389948995899689978998899990009001900290039004900590069007900890099010901190129013901490159016901790189019902090219022902390249025902690279028902990309031903290339034903590369037903890399040904190429043904490459046904790489049905090519052905390549055905690579058905990609061906290639064906590669067906890699070907190729073907490759076907790789079908090819082908390849085908690879088908990909091909290939094909590969097909890999100910191029103910491059106910791089109911091119112911391149115911691179118911991209121912291239124912591269127912891299130913191329133913491359136913791389139914091419142914391449145914691479148914991509151915291539154915591569157915891599160916191629163916491659166916791689169917091719172917391749175917691779178917991809181918291839184918591869187918891899190919191929193919491959196919791989199920092019202920392049205920692079208920992109211921292139214921592169217921892199220922192229223922492259226922792289229923092319232923392349235923692379238923992409241924292439244924592469247924892499250925192529253925492559256925792589259926092619262926392649265926692679268926992709271927292739274927592769277927892799280928192829283928492859286928792889289929092919292929392949295929692979298929993009301930293039304930593069307930893099310931193129313931493159316931793189319932093219322932393249325932693279328932993309331933293339334933593369337933893399340934193429343934493459346934793489349935093519352935393549355935693579358935993609361936293639364936593669367936893699370937193729373937493759376937793789379938093819382938393849385938693879388938993909391939293939394939593969397939893999400940194029403940494059406940794089409941094119412941394149415941694179418941994209421942294239424942594269427942894299430943194329433943494359436943794389439944094419442944394449445944694479448944994509451945294539454945594569457945894599460946194629463946494659466946794689469947094719472947394749475947694779478947994809481948294839484948594869487948894899490949194929493949494959496949794989499950095019502950395049505950695079508950995109511951295139514951595169517951895199520952195229523952495259526952795289529953095319532953395349535953695379538953995409541954295439544954595469547954895499550955195529553955495559556955795589559956095619562956395649565956695679568956995709571957295739574957595769577957895799580958195829583958495859586958795889589959095919592959395949595959695979598959996009601960296039604960596069607960896099610961196129613961496159616961796189619962096219622962396249625962696279628962996309631963296339634963596369637963896399640964196429643964496459646964796489649965096519652965396549655965696579658965996609661966296639664966596669667966896699670967196729673967496759676967796789679968096819682968396849685968696879688968996909691969296939694969596969697969896999700970197029703970497059706970797089709971097119712971397149715971697179718971997209721972297239724972597269727972897299730973197329733973497359736973797389739974097419742974397449745974697479748974997509751975297539754975597569757975897599760976197629763976497659766976797689769977097719772977397749775977697779778977997809781978297839784978597869787978897899790979197929793979497959796979797989799980098019802980398049805980698079808980998109811981298139814981598169817981898199820982198229823982498259826982798289829983098319832983398349835983698379838983998409841984298439844984598469847984898499850985198529853985498559856985798589859986098619862986398649865986698679868986998709871987298739874987598769877987898799880988198829883988498859886988798889889989098919892989398949895989698979898989999009901990299039904990599069907990899099910991199129913991499159916991799189919992099219922992399249925992699279928992999309931993299339934993599369937993899399940994199429943994499459946994799489949995099519952995399549955995699579958995999609961996299639964996599669967996899699970997199729973997499759976997799789979998099819982998399849985998699879988998999909991999299939994999599969997999899991000010001100021000310004100051000610007100081000910010100111001210013100141001510016100171001810019100201002110022100231002410025100261002710028100291003010031100321003310034100351003610037100381003910040100411004210043100441004510046100471004810049100501005110052100531005410055100561005710058100591006010061100621006310064100651006610067100681006910070100711007210073100741007510076100771007810079100801008110082100831008410085100861008710088100891009010091100921009310094100951009610097100981009910100101011010210103101041010510106101071010810109101101011110112101131011410115101161011710118101191012010121101221012310124101251012610127101281012910130101311013210133101341013510136101371013810139101401014110142101431014410145101461014710148101491015010151101521015310154101551015610157101581015910160101611016210163101641016510166101671016810169101701017110172101731017410175101761017710178101791018010181101821018310184101851018610187101881018910190101911019210193101941019510196101971019810199102001020110202102031020410205102061020710208102091021010211102121021310214102151021610217102181021910220102211022210223102241022510226102271022810229102301023110232102331023410235102361023710238102391024010241102421024310244102451024610247102481024910250102511025210253102541025510256102571025810259102601026110262102631026410265102661026710268102691027010271102721027310274102751027610277102781027910280102811028210283102841028510286102871028810289102901029110292102931029410295102961029710298102991030010301103021030310304103051030610307103081030910310103111031210313103141031510316103171031810319103201032110322103231032410325103261032710328103291033010331103321033310334103351033610337103381033910340103411034210343103441034510346103471034810349103501035110352103531035410355103561035710358103591036010361103621036310364103651036610367103681036910370103711037210373103741037510376103771037810379103801038110382103831038410385103861038710388103891039010391103921039310394103951039610397103981039910400104011040210403104041040510406104071040810409104101041110412104131041410415104161041710418104191042010421104221042310424104251042610427104281042910430104311043210433104341043510436104371043810439104401044110442104431044410445104461044710448104491045010451104521045310454104551045610457104581045910460104611046210463104641046510466104671046810469104701047110472104731047410475104761047710478104791048010481104821048310484104851048610487104881048910490104911049210493104941049510496104971049810499105001050110502105031050410505105061050710508105091051010511105121051310514105151051610517105181051910520105211052210523105241052510526105271052810529105301053110532105331053410535105361053710538105391054010541105421054310544105451054610547105481054910550105511055210553105541055510556105571055810559105601056110562105631056410565105661056710568105691057010571105721057310574105751057610577105781057910580105811058210583105841058510586105871058810589105901059110592105931059410595105961059710598105991060010601106021060310604106051060610607106081060910610106111061210613106141061510616106171061810619106201062110622106231062410625106261062710628106291063010631106321063310634106351063610637106381063910640106411064210643106441064510646106471064810649106501065110652106531065410655106561065710658106591066010661106621066310664106651066610667106681066910670106711067210673106741067510676106771067810679106801068110682106831068410685106861068710688106891069010691106921069310694106951069610697106981069910700107011070210703107041070510706107071070810709107101071110712107131071410715107161071710718107191072010721107221072310724107251072610727107281072910730107311073210733107341073510736107371073810739107401074110742107431074410745107461074710748107491075010751107521075310754107551075610757107581075910760107611076210763107641076510766107671076810769107701077110772107731077410775107761077710778107791078010781107821078310784107851078610787107881078910790107911079210793107941079510796107971079810799108001080110802108031080410805108061080710808108091081010811108121081310814108151081610817108181081910820108211082210823108241082510826108271082810829108301083110832108331083410835108361083710838108391084010841108421084310844108451084610847108481084910850108511085210853108541085510856108571085810859108601086110862108631086410865108661086710868108691087010871108721087310874108751087610877108781087910880108811088210883108841088510886108871088810889108901089110892108931089410895108961089710898108991090010901109021090310904109051090610907109081090910910109111091210913109141091510916109171091810919109201092110922109231092410925109261092710928109291093010931109321093310934109351093610937109381093910940109411094210943109441094510946109471094810949109501095110952109531095410955109561095710958109591096010961109621096310964109651096610967109681096910970109711097210973109741097510976109771097810979109801098110982109831098410985109861098710988109891099010991109921099310994109951099610997109981099911000110011100211003110041100511006110071100811009110101101111012110131101411015110161101711018110191102011021110221102311024110251102611027110281102911030110311103211033110341103511036110371103811039110401104111042110431104411045110461104711048110491105011051110521105311054110551105611057110581105911060110611106211063110641106511066110671106811069110701107111072110731107411075110761107711078110791108011081110821108311084110851108611087110881108911090110911109211093110941109511096110971109811099111001110111102111031110411105111061110711108111091111011111111121111311114111151111611117111181111911120111211112211123111241112511126111271112811129111301113111132111331113411135111361113711138111391114011141111421114311144111451114611147111481114911150111511115211153111541115511156111571115811159111601116111162111631116411165111661116711168111691117011171111721117311174111751117611177111781117911180111811118211183111841118511186111871118811189111901119111192111931119411195111961119711198111991120011201112021120311204112051120611207112081120911210112111121211213112141121511216112171121811219112201122111222112231122411225112261122711228112291123011231112321123311234112351123611237112381123911240112411124211243112441124511246112471124811249112501125111252112531125411255112561125711258112591126011261112621126311264112651126611267112681126911270112711127211273112741127511276112771127811279112801128111282112831128411285112861128711288112891129011291112921129311294112951129611297112981129911300113011130211303113041130511306113071130811309113101131111312113131131411315113161131711318113191132011321113221132311324113251132611327113281132911330113311133211333113341133511336113371133811339113401134111342113431134411345113461134711348113491135011351113521135311354113551135611357113581135911360113611136211363113641136511366113671136811369113701137111372113731137411375113761137711378113791138011381113821138311384113851138611387113881138911390113911139211393113941139511396113971139811399114001140111402114031140411405114061140711408114091141011411114121141311414114151141611417114181141911420114211142211423114241142511426114271142811429114301143111432114331143411435114361143711438114391144011441114421144311444114451144611447114481144911450114511145211453114541145511456114571145811459114601146111462114631146411465114661146711468114691147011471114721147311474114751147611477114781147911480114811148211483114841148511486114871148811489114901149111492114931149411495114961149711498114991150011501115021150311504115051150611507115081150911510115111151211513115141151511516115171151811519115201152111522115231152411525115261152711528115291153011531115321153311534115351153611537115381153911540115411154211543115441154511546115471154811549115501155111552115531155411555115561155711558115591156011561115621156311564115651156611567115681156911570115711157211573115741157511576115771157811579115801158111582115831158411585115861158711588115891159011591115921159311594115951159611597115981159911600116011160211603116041160511606116071160811609116101161111612116131161411615116161161711618116191162011621116221162311624116251162611627116281162911630116311163211633116341163511636116371163811639116401164111642116431164411645116461164711648116491165011651116521165311654116551165611657116581165911660116611166211663116641166511666116671166811669116701167111672116731167411675116761167711678116791168011681116821168311684116851168611687116881168911690116911169211693116941169511696116971169811699117001170111702117031170411705117061170711708117091171011711117121171311714117151171611717117181171911720117211172211723117241172511726117271172811729117301173111732117331173411735117361173711738117391174011741117421174311744117451174611747117481174911750117511175211753117541175511756117571175811759117601176111762117631176411765117661176711768117691177011771117721177311774117751177611777117781177911780117811178211783117841178511786117871178811789117901179111792117931179411795117961179711798117991180011801118021180311804118051180611807118081180911810118111181211813118141181511816118171181811819118201182111822118231182411825118261182711828118291183011831118321183311834118351183611837118381183911840118411184211843118441184511846118471184811849118501185111852118531185411855118561185711858118591186011861118621186311864118651186611867118681186911870118711187211873118741187511876118771187811879118801188111882118831188411885118861188711888118891189011891118921189311894118951189611897118981189911900119011190211903119041190511906119071190811909119101191111912119131191411915119161191711918119191192011921119221192311924119251192611927119281192911930119311193211933119341193511936119371193811939119401194111942119431194411945119461194711948119491195011951119521195311954119551195611957119581195911960119611196211963119641196511966119671196811969119701197111972119731197411975119761197711978119791198011981119821198311984119851198611987119881198911990119911199211993119941199511996119971199811999120001200112002120031200412005120061200712008120091201012011120121201312014120151201612017120181201912020120211202212023120241202512026120271202812029120301203112032120331203412035120361203712038120391204012041120421204312044120451204612047120481204912050120511205212053120541205512056120571205812059120601206112062120631206412065120661206712068120691207012071120721207312074120751207612077120781207912080120811208212083120841208512086120871208812089120901209112092120931209412095120961209712098120991210012101121021210312104121051210612107121081210912110121111211212113121141211512116121171211812119121201212112122121231212412125121261212712128121291213012131121321213312134121351213612137121381213912140121411214212143121441214512146121471214812149121501215112152121531215412155121561215712158121591216012161121621216312164121651216612167121681216912170121711217212173121741217512176121771217812179121801218112182121831218412185121861218712188121891219012191121921219312194121951219612197121981219912200122011220212203122041220512206122071220812209122101221112212122131221412215122161221712218122191222012221122221222312224122251222612227122281222912230122311223212233122341223512236122371223812239122401224112242122431224412245122461224712248122491225012251122521225312254122551225612257122581225912260122611226212263122641226512266122671226812269122701227112272122731227412275122761227712278122791228012281122821228312284122851228612287122881228912290122911229212293122941229512296122971229812299123001230112302123031230412305123061230712308123091231012311123121231312314123151231612317123181231912320123211232212323123241232512326123271232812329123301233112332123331233412335123361233712338123391234012341123421234312344123451234612347123481234912350123511235212353123541235512356123571235812359123601236112362123631236412365123661236712368123691237012371123721237312374123751237612377123781237912380123811238212383123841238512386123871238812389123901239112392123931239412395123961239712398123991240012401124021240312404124051240612407124081240912410124111241212413124141241512416124171241812419124201242112422124231242412425124261242712428124291243012431124321243312434124351243612437124381243912440124411244212443124441244512446124471244812449124501245112452124531245412455124561245712458124591246012461124621246312464124651246612467124681246912470124711247212473124741247512476124771247812479124801248112482124831248412485124861248712488124891249012491124921249312494124951249612497124981249912500125011250212503125041250512506125071250812509125101251112512125131251412515125161251712518125191252012521125221252312524125251252612527125281252912530125311253212533125341253512536125371253812539125401254112542125431254412545125461254712548125491255012551125521255312554125551255612557125581255912560125611256212563125641256512566125671256812569125701257112572125731257412575125761257712578125791258012581125821258312584125851258612587125881258912590125911259212593125941259512596125971259812599126001260112602126031260412605126061260712608126091261012611126121261312614126151261612617126181261912620126211262212623126241262512626126271262812629126301263112632126331263412635126361263712638126391264012641126421264312644126451264612647126481264912650126511265212653126541265512656126571265812659126601266112662126631266412665126661266712668126691267012671126721267312674126751267612677126781267912680126811268212683126841268512686126871268812689126901269112692126931269412695126961269712698126991270012701127021270312704127051270612707127081270912710127111271212713127141271512716127171271812719127201272112722127231272412725127261272712728127291273012731127321273312734127351273612737127381273912740127411274212743127441274512746127471274812749127501275112752127531275412755127561275712758127591276012761127621276312764127651276612767127681276912770127711277212773127741277512776127771277812779127801278112782127831278412785127861278712788127891279012791127921279312794127951279612797127981279912800128011280212803128041280512806128071280812809128101281112812128131281412815128161281712818128191282012821128221282312824128251282612827128281282912830128311283212833128341283512836128371283812839128401284112842128431284412845128461284712848128491285012851128521285312854128551285612857128581285912860128611286212863128641286512866128671286812869128701287112872128731287412875128761287712878128791288012881128821288312884128851288612887128881288912890128911289212893128941289512896128971289812899129001290112902129031290412905129061290712908129091291012911129121291312914129151291612917129181291912920129211292212923129241292512926129271292812929129301293112932129331293412935129361293712938129391294012941129421294312944129451294612947129481294912950129511295212953129541295512956129571295812959129601296112962129631296412965129661296712968129691297012971129721297312974129751297612977129781297912980129811298212983129841298512986129871298812989129901299112992129931299412995129961299712998129991300013001130021300313004130051300613007130081300913010130111301213013130141301513016130171301813019130201302113022130231302413025130261302713028130291303013031130321303313034130351303613037130381303913040130411304213043130441304513046130471304813049130501305113052130531305413055130561305713058130591306013061130621306313064130651306613067130681306913070130711307213073130741307513076130771307813079130801308113082130831308413085130861308713088130891309013091130921309313094130951309613097130981309913100131011310213103131041310513106131071310813109131101311113112131131311413115131161311713118131191312013121131221312313124131251312613127131281312913130131311313213133131341313513136131371313813139131401314113142131431314413145131461314713148131491315013151131521315313154131551315613157131581315913160131611316213163131641316513166131671316813169131701317113172131731317413175131761317713178131791318013181131821318313184131851318613187131881318913190131911319213193131941319513196131971319813199132001320113202132031320413205132061320713208132091321013211132121321313214132151321613217132181321913220132211322213223132241322513226132271322813229132301323113232132331323413235132361323713238132391324013241132421324313244132451324613247132481324913250132511325213253132541325513256132571325813259132601326113262132631326413265132661326713268132691327013271132721327313274132751327613277132781327913280132811328213283132841328513286132871328813289132901329113292132931329413295132961329713298132991330013301133021330313304133051330613307133081330913310133111331213313133141331513316133171331813319133201332113322133231332413325133261332713328133291333013331133321333313334133351333613337133381333913340133411334213343133441334513346133471334813349133501335113352133531335413355133561335713358133591336013361133621336313364133651336613367133681336913370133711337213373133741337513376133771337813379133801338113382133831338413385133861338713388133891339013391133921339313394133951339613397133981339913400134011340213403134041340513406134071340813409134101341113412134131341413415134161341713418134191342013421134221342313424134251342613427134281342913430134311343213433134341343513436134371343813439134401344113442134431344413445134461344713448134491345013451134521345313454134551345613457134581345913460134611346213463134641346513466134671346813469134701347113472134731347413475134761347713478134791348013481134821348313484134851348613487134881348913490134911349213493134941349513496134971349813499135001350113502135031350413505135061350713508135091351013511135121351313514135151351613517135181351913520135211352213523135241352513526135271352813529135301353113532135331353413535135361353713538135391354013541135421354313544135451354613547135481354913550135511355213553135541355513556135571355813559135601356113562135631356413565135661356713568135691357013571135721357313574135751357613577135781357913580135811358213583135841358513586135871358813589135901359113592135931359413595135961359713598135991360013601136021360313604136051360613607136081360913610136111361213613136141361513616136171361813619136201362113622136231362413625136261362713628136291363013631136321363313634136351363613637136381363913640136411364213643136441364513646136471364813649136501365113652136531365413655136561365713658136591366013661136621366313664136651366613667136681366913670136711367213673136741367513676136771367813679136801368113682136831368413685136861368713688136891369013691136921369313694136951369613697136981369913700137011370213703137041370513706137071370813709137101371113712137131371413715137161371713718137191372013721137221372313724137251372613727137281372913730137311373213733137341373513736137371373813739137401374113742137431374413745137461374713748137491375013751137521375313754137551375613757137581375913760137611376213763137641376513766137671376813769137701377113772137731377413775137761377713778137791378013781137821378313784137851378613787137881378913790137911379213793137941379513796137971379813799138001380113802138031380413805138061380713808138091381013811138121381313814138151381613817138181381913820138211382213823138241382513826138271382813829138301383113832138331383413835138361383713838138391384013841138421384313844138451384613847138481384913850138511385213853138541385513856138571385813859138601386113862138631386413865138661386713868138691387013871138721387313874138751387613877138781387913880138811388213883138841388513886138871388813889138901389113892138931389413895138961389713898138991390013901139021390313904139051390613907139081390913910139111391213913139141391513916139171391813919139201392113922139231392413925139261392713928139291393013931139321393313934139351393613937139381393913940139411394213943139441394513946139471394813949139501395113952139531395413955139561395713958139591396013961139621396313964139651396613967139681396913970139711397213973139741397513976139771397813979139801398113982139831398413985139861398713988139891399013991139921399313994139951399613997139981399914000140011400214003140041400514006140071400814009140101401114012140131401414015140161401714018140191402014021140221402314024140251402614027140281402914030140311403214033140341403514036140371403814039140401404114042140431404414045140461404714048140491405014051140521405314054140551405614057140581405914060140611406214063140641406514066140671406814069140701407114072140731407414075140761407714078140791408014081140821408314084140851408614087140881408914090140911409214093140941409514096140971409814099141001410114102141031410414105141061410714108141091411014111141121411314114141151411614117141181411914120141211412214123141241412514126141271412814129141301413114132141331413414135141361413714138141391414014141141421414314144141451414614147141481414914150141511415214153141541415514156141571415814159141601416114162141631416414165141661416714168141691417014171141721417314174141751417614177141781417914180141811418214183141841418514186141871418814189141901419114192141931419414195141961419714198141991420014201142021420314204142051420614207142081420914210142111421214213142141421514216142171421814219142201422114222142231422414225142261422714228142291423014231142321423314234142351423614237142381423914240142411424214243142441424514246142471424814249142501425114252142531425414255142561425714258142591426014261142621426314264142651426614267142681426914270142711427214273142741427514276142771427814279142801428114282142831428414285142861428714288142891429014291142921429314294142951429614297142981429914300143011430214303143041430514306143071430814309143101431114312143131431414315143161431714318143191432014321143221432314324143251432614327143281432914330143311433214333143341433514336143371433814339143401434114342143431434414345143461434714348143491435014351143521435314354143551435614357143581435914360143611436214363143641436514366143671436814369143701437114372143731437414375143761437714378143791438014381143821438314384143851438614387143881438914390143911439214393143941439514396143971439814399144001440114402144031440414405144061440714408144091441014411144121441314414144151441614417144181441914420144211442214423144241442514426144271442814429144301443114432144331443414435144361443714438144391444014441144421444314444144451444614447144481444914450144511445214453144541445514456144571445814459144601446114462144631446414465144661446714468144691447014471144721447314474144751447614477144781447914480144811448214483144841448514486144871448814489144901449114492144931449414495144961449714498144991450014501145021450314504145051450614507145081450914510145111451214513145141451514516145171451814519145201452114522145231452414525145261452714528145291453014531145321453314534145351453614537145381453914540145411454214543145441454514546145471454814549145501455114552145531455414555145561455714558145591456014561145621456314564145651456614567145681456914570145711457214573145741457514576145771457814579145801458114582145831458414585145861458714588145891459014591145921459314594145951459614597145981459914600146011460214603146041460514606146071460814609146101461114612146131461414615146161461714618146191462014621146221462314624146251462614627146281462914630146311463214633146341463514636146371463814639146401464114642146431464414645146461464714648146491465014651146521465314654146551465614657146581465914660146611466214663146641466514666146671466814669146701467114672146731467414675146761467714678146791468014681146821468314684146851468614687146881468914690146911469214693146941469514696146971469814699147001470114702147031470414705147061470714708147091471014711147121471314714147151471614717147181471914720147211472214723147241472514726147271472814729147301473114732147331473414735147361473714738147391474014741147421474314744147451474614747147481474914750147511475214753147541475514756147571475814759147601476114762147631476414765147661476714768147691477014771147721477314774147751477614777147781477914780147811478214783147841478514786147871478814789147901479114792147931479414795147961479714798147991480014801148021480314804148051480614807148081480914810148111481214813148141481514816148171481814819148201482114822148231482414825148261482714828148291483014831148321483314834148351483614837148381483914840148411484214843148441484514846148471484814849148501485114852148531485414855148561485714858148591486014861148621486314864148651486614867148681486914870148711487214873148741487514876148771487814879148801488114882148831488414885148861488714888148891489014891148921489314894148951489614897148981489914900149011490214903149041490514906149071490814909149101491114912149131491414915149161491714918149191492014921149221492314924149251492614927149281492914930149311493214933149341493514936149371493814939149401494114942149431494414945149461494714948149491495014951149521495314954149551495614957149581495914960149611496214963149641496514966149671496814969149701497114972149731497414975149761497714978149791498014981149821498314984149851498614987149881498914990149911499214993149941499514996149971499814999150001500115002150031500415005150061500715008150091501015011150121501315014150151501615017150181501915020150211502215023150241502515026150271502815029150301503115032150331503415035150361503715038150391504015041150421504315044150451504615047150481504915050150511505215053150541505515056150571505815059150601506115062150631506415065150661506715068150691507015071150721507315074150751507615077150781507915080150811508215083150841508515086150871508815089150901509115092150931509415095150961509715098150991510015101151021510315104151051510615107151081510915110151111511215113151141511515116151171511815119151201512115122151231512415125151261512715128151291513015131151321513315134151351513615137151381513915140151411514215143151441514515146151471514815149151501515115152151531515415155151561515715158151591516015161151621516315164151651516615167151681516915170151711517215173151741517515176151771517815179151801518115182151831518415185151861518715188151891519015191151921519315194151951519615197151981519915200152011520215203152041520515206152071520815209152101521115212152131521415215152161521715218152191522015221152221522315224152251522615227152281522915230152311523215233152341523515236152371523815239152401524115242152431524415245152461524715248152491525015251152521525315254152551525615257152581525915260152611526215263152641526515266152671526815269152701527115272152731527415275152761527715278152791528015281152821528315284152851528615287152881528915290152911529215293152941529515296152971529815299153001530115302153031530415305153061530715308153091531015311153121531315314153151531615317153181531915320153211532215323153241532515326153271532815329153301533115332153331533415335153361533715338153391534015341153421534315344153451534615347153481534915350153511535215353153541535515356153571535815359153601536115362153631536415365153661536715368153691537015371153721537315374153751537615377153781537915380153811538215383153841538515386153871538815389153901539115392153931539415395153961539715398153991540015401154021540315404154051540615407154081540915410154111541215413154141541515416154171541815419154201542115422154231542415425154261542715428154291543015431154321543315434154351543615437154381543915440154411544215443154441544515446154471544815449154501545115452154531545415455154561545715458154591546015461154621546315464154651546615467154681546915470154711547215473154741547515476154771547815479154801548115482154831548415485154861548715488154891549015491154921549315494154951549615497154981549915500155011550215503155041550515506155071550815509155101551115512155131551415515155161551715518155191552015521155221552315524155251552615527155281552915530155311553215533155341553515536155371553815539155401554115542155431554415545155461554715548155491555015551155521555315554155551555615557155581555915560155611556215563155641556515566155671556815569155701557115572155731557415575155761557715578155791558015581155821558315584155851558615587155881558915590155911559215593155941559515596155971559815599156001560115602156031560415605156061560715608156091561015611156121561315614156151561615617156181561915620156211562215623156241562515626156271562815629156301563115632156331563415635156361563715638156391564015641156421564315644156451564615647156481564915650156511565215653156541565515656156571565815659156601566115662156631566415665156661566715668156691567015671156721567315674156751567615677156781567915680156811568215683156841568515686156871568815689156901569115692156931569415695156961569715698156991570015701157021570315704157051570615707157081570915710157111571215713157141571515716157171571815719157201572115722157231572415725157261572715728157291573015731157321573315734157351573615737157381573915740157411574215743157441574515746157471574815749157501575115752157531575415755157561575715758157591576015761157621576315764157651576615767157681576915770157711577215773157741577515776157771577815779157801578115782157831578415785157861578715788157891579015791157921579315794157951579615797157981579915800158011580215803158041580515806158071580815809158101581115812158131581415815158161581715818158191582015821158221582315824158251582615827158281582915830158311583215833158341583515836158371583815839158401584115842158431584415845158461584715848158491585015851158521585315854158551585615857158581585915860158611586215863158641586515866158671586815869158701587115872158731587415875158761587715878158791588015881158821588315884158851588615887158881588915890158911589215893158941589515896158971589815899159001590115902159031590415905159061590715908159091591015911159121591315914159151591615917159181591915920159211592215923159241592515926159271592815929159301593115932159331593415935159361593715938159391594015941159421594315944159451594615947159481594915950159511595215953159541595515956159571595815959159601596115962159631596415965159661596715968159691597015971159721597315974159751597615977159781597915980159811598215983159841598515986159871598815989159901599115992159931599415995159961599715998159991600016001160021600316004160051600616007160081600916010160111601216013160141601516016160171601816019160201602116022160231602416025160261602716028160291603016031160321603316034160351603616037160381603916040160411604216043160441604516046160471604816049160501605116052160531605416055160561605716058160591606016061160621606316064160651606616067160681606916070160711607216073160741607516076160771607816079160801608116082160831608416085160861608716088160891609016091160921609316094160951609616097160981609916100161011610216103161041610516106161071610816109161101611116112161131611416115161161611716118161191612016121161221612316124161251612616127161281612916130161311613216133161341613516136161371613816139161401614116142161431614416145161461614716148161491615016151161521615316154161551615616157161581615916160161611616216163161641616516166161671616816169161701617116172161731617416175161761617716178161791618016181161821618316184161851618616187161881618916190161911619216193161941619516196161971619816199162001620116202162031620416205162061620716208162091621016211162121621316214162151621616217162181621916220162211622216223162241622516226162271622816229162301623116232162331623416235162361623716238162391624016241162421624316244162451624616247162481624916250162511625216253162541625516256162571625816259162601626116262162631626416265162661626716268162691627016271162721627316274162751627616277162781627916280162811628216283162841628516286162871628816289162901629116292162931629416295162961629716298162991630016301163021630316304163051630616307163081630916310163111631216313163141631516316163171631816319163201632116322163231632416325163261632716328163291633016331163321633316334163351633616337163381633916340163411634216343163441634516346163471634816349163501635116352163531635416355163561635716358163591636016361163621636316364163651636616367163681636916370163711637216373163741637516376163771637816379163801638116382163831638416385163861638716388163891639016391163921639316394163951639616397163981639916400164011640216403164041640516406164071640816409164101641116412164131641416415164161641716418164191642016421164221642316424164251642616427164281642916430164311643216433164341643516436164371643816439164401644116442164431644416445164461644716448164491645016451164521645316454164551645616457164581645916460164611646216463164641646516466164671646816469164701647116472164731647416475164761647716478164791648016481164821648316484164851648616487164881648916490164911649216493164941649516496164971649816499165001650116502165031650416505165061650716508165091651016511165121651316514165151651616517165181651916520165211652216523165241652516526165271652816529165301653116532165331653416535165361653716538165391654016541165421654316544165451654616547165481654916550165511655216553165541655516556165571655816559165601656116562165631656416565165661656716568165691657016571165721657316574165751657616577165781657916580165811658216583165841658516586165871658816589165901659116592165931659416595165961659716598165991660016601166021660316604166051660616607166081660916610166111661216613166141661516616166171661816619166201662116622166231662416625166261662716628166291663016631166321663316634166351663616637166381663916640166411664216643166441664516646166471664816649166501665116652166531665416655166561665716658166591666016661166621666316664166651666616667166681666916670166711667216673166741667516676166771667816679166801668116682166831668416685166861668716688166891669016691166921669316694166951669616697166981669916700167011670216703167041670516706167071670816709167101671116712167131671416715167161671716718167191672016721167221672316724167251672616727167281672916730167311673216733167341673516736167371673816739167401674116742167431674416745167461674716748167491675016751167521675316754167551675616757167581675916760167611676216763167641676516766167671676816769167701677116772167731677416775167761677716778167791678016781167821678316784167851678616787167881678916790167911679216793167941679516796167971679816799168001680116802168031680416805168061680716808168091681016811168121681316814168151681616817168181681916820168211682216823168241682516826168271682816829168301683116832168331683416835168361683716838168391684016841168421684316844168451684616847168481684916850168511685216853168541685516856168571685816859168601686116862168631686416865168661686716868168691687016871168721687316874168751687616877168781687916880168811688216883168841688516886168871688816889168901689116892168931689416895168961689716898168991690016901169021690316904169051690616907169081690916910169111691216913169141691516916169171691816919169201692116922169231692416925169261692716928169291693016931169321693316934169351693616937169381693916940169411694216943169441694516946169471694816949169501695116952169531695416955169561695716958169591696016961169621696316964169651696616967169681696916970169711697216973169741697516976169771697816979169801698116982169831698416985169861698716988169891699016991169921699316994169951699616997169981699917000170011700217003170041700517006170071700817009170101701117012170131701417015170161701717018170191702017021170221702317024170251702617027170281702917030170311703217033170341703517036170371703817039170401704117042170431704417045170461704717048170491705017051170521705317054170551705617057170581705917060170611706217063170641706517066170671706817069170701707117072170731707417075170761707717078170791708017081170821708317084170851708617087170881708917090170911709217093170941709517096170971709817099171001710117102171031710417105171061710717108171091711017111171121711317114171151711617117171181711917120171211712217123171241712517126 |
- /* POSIX module implementation */
- /* This file is also used for Windows NT/MS-Win. In that case the
- module actually calls itself 'nt', not 'posix', and a few
- functions are either unimplemented or implemented differently. The source
- assumes that for Windows NT, the macro 'MS_WINDOWS' is defined independent
- of the compiler used. Different compilers define their own feature
- test macro, e.g. '_MSC_VER'. */
- #define PY_SSIZE_T_CLEAN
- #include "Python.h"
- #ifdef __VXWORKS__
- # include "pycore_bitutils.h" // _Py_popcount32()
- #endif
- #include "pycore_call.h" // _PyObject_CallNoArgs()
- #include "pycore_ceval.h" // _PyEval_ReInitThreads()
- #include "pycore_fileutils.h" // _Py_closerange()
- #include "pycore_import.h" // _PyImport_ReInitLock()
- #include "pycore_initconfig.h" // _PyStatus_EXCEPTION()
- #include "pycore_long.h" // _PyLong_IsNegative()
- #include "pycore_moduleobject.h" // _PyModule_GetState()
- #include "pycore_object.h" // _PyObject_LookupSpecial()
- #include "pycore_pystate.h" // _PyInterpreterState_GET()
- #include "pycore_signal.h" // Py_NSIG
- #include "pycore_typeobject.h" // _PyType_AddMethod()
- #ifdef MS_WINDOWS
- # include <windows.h>
- # if !defined(MS_WINDOWS_GAMES) || defined(MS_WINDOWS_DESKTOP)
- # include <pathcch.h>
- # endif
- # include <winioctl.h>
- # include <lmcons.h> // UNLEN
- # include "osdefs.h" // SEP
- # include <aclapi.h> // SetEntriesInAcl
- # include <sddl.h> // SDDL_REVISION_1
- # if defined(MS_WINDOWS_DESKTOP) || defined(MS_WINDOWS_SYSTEM)
- # define HAVE_SYMLINK
- # endif /* MS_WINDOWS_DESKTOP | MS_WINDOWS_SYSTEM */
- #endif
- #include "structmember.h" // PyMemberDef
- #ifndef MS_WINDOWS
- # include "posixmodule.h"
- #else
- # include "pycore_fileutils_windows.h"
- # include "winreparse.h"
- #endif
- #if !defined(EX_OK) && defined(EXIT_SUCCESS)
- # define EX_OK EXIT_SUCCESS
- #endif
- /* On android API level 21, 'AT_EACCESS' is not declared although
- * HAVE_FACCESSAT is defined. */
- #ifdef __ANDROID__
- # undef HAVE_FACCESSAT
- #endif
- #include <stdio.h> // ctermid()
- #include <stdlib.h> // system()
- /*
- * A number of APIs are available on macOS from a certain macOS version.
- * To support building with a new SDK while deploying to older versions
- * the availability test is split into two:
- * - HAVE_<FUNCTION>: The configure check for compile time availability
- * - HAVE_<FUNCTION>_RUNTIME: Runtime check for availability
- *
- * The latter is always true when not on macOS, or when using a compiler
- * that does not support __has_builtin (older versions of Xcode).
- *
- * Due to compiler restrictions there is one valid use of HAVE_<FUNCTION>_RUNTIME:
- * if (HAVE_<FUNCTION>_RUNTIME) { ... }
- *
- * In mixing the test with other tests or using negations will result in compile
- * errors.
- */
- #if defined(__APPLE__)
- #include <mach/mach.h>
- #if defined(__has_builtin)
- #if __has_builtin(__builtin_available)
- #define HAVE_BUILTIN_AVAILABLE 1
- #endif
- #endif
- #ifdef HAVE_BUILTIN_AVAILABLE
- # define HAVE_FSTATAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_FACCESSAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_FCHMODAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_FCHOWNAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_LINKAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_FDOPENDIR_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_MKDIRAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_RENAMEAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_UNLINKAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_OPENAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_READLINKAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_SYMLINKAT_RUNTIME __builtin_available(macOS 10.10, iOS 8.0, *)
- # define HAVE_FUTIMENS_RUNTIME __builtin_available(macOS 10.13, iOS 11.0, tvOS 11.0, watchOS 4.0, *)
- # define HAVE_UTIMENSAT_RUNTIME __builtin_available(macOS 10.13, iOS 11.0, tvOS 11.0, watchOS 4.0, *)
- # define HAVE_PWRITEV_RUNTIME __builtin_available(macOS 11.0, iOS 14.0, tvOS 14.0, watchOS 7.0, *)
- # define HAVE_MKFIFOAT_RUNTIME __builtin_available(macOS 13.0, iOS 16.0, tvOS 16.0, watchOS 9.0, *)
- # define HAVE_MKNODAT_RUNTIME __builtin_available(macOS 13.0, iOS 16.0, tvOS 16.0, watchOS 9.0, *)
- # define HAVE_POSIX_SPAWN_SETSID_RUNTIME __builtin_available(macOS 10.15, *)
- #else /* Xcode 8 or earlier */
- /* __builtin_available is not present in these compilers, but
- * some of the symbols might be weak linked (10.10 SDK or later
- * deploying on 10.9.
- *
- * Fall back to the older style of availability checking for
- * symbols introduced in macOS 10.10.
- */
- # ifdef HAVE_FSTATAT
- # define HAVE_FSTATAT_RUNTIME (fstatat != NULL)
- # endif
- # ifdef HAVE_FACCESSAT
- # define HAVE_FACCESSAT_RUNTIME (faccessat != NULL)
- # endif
- # ifdef HAVE_FCHMODAT
- # define HAVE_FCHMODAT_RUNTIME (fchmodat != NULL)
- # endif
- # ifdef HAVE_FCHOWNAT
- # define HAVE_FCHOWNAT_RUNTIME (fchownat != NULL)
- # endif
- # ifdef HAVE_LINKAT
- # define HAVE_LINKAT_RUNTIME (linkat != NULL)
- # endif
- # ifdef HAVE_FDOPENDIR
- # define HAVE_FDOPENDIR_RUNTIME (fdopendir != NULL)
- # endif
- # ifdef HAVE_MKDIRAT
- # define HAVE_MKDIRAT_RUNTIME (mkdirat != NULL)
- # endif
- # ifdef HAVE_RENAMEAT
- # define HAVE_RENAMEAT_RUNTIME (renameat != NULL)
- # endif
- # ifdef HAVE_UNLINKAT
- # define HAVE_UNLINKAT_RUNTIME (unlinkat != NULL)
- # endif
- # ifdef HAVE_OPENAT
- # define HAVE_OPENAT_RUNTIME (openat != NULL)
- # endif
- # ifdef HAVE_READLINKAT
- # define HAVE_READLINKAT_RUNTIME (readlinkat != NULL)
- # endif
- # ifdef HAVE_SYMLINKAT
- # define HAVE_SYMLINKAT_RUNTIME (symlinkat != NULL)
- # endif
- # ifdef HAVE_UTIMENSAT
- # define HAVE_UTIMENSAT_RUNTIME (utimensat != NULL)
- # endif
- # ifdef HAVE_FUTIMENS
- # define HAVE_FUTIMENS_RUNTIME (futimens != NULL)
- # endif
- # ifdef HAVE_PWRITEV
- # define HAVE_PWRITEV_RUNTIME (pwritev != NULL)
- # endif
- # ifdef HAVE_MKFIFOAT
- # define HAVE_MKFIFOAT_RUNTIME (mkfifoat != NULL)
- # endif
- # ifdef HAVE_MKNODAT
- # define HAVE_MKNODAT_RUNTIME (mknodat != NULL)
- # endif
- #endif
- #ifdef HAVE_FUTIMESAT
- /* Some of the logic for weak linking depends on this assertion */
- # error "HAVE_FUTIMESAT unexpectedly defined"
- #endif
- #else
- # define HAVE_FSTATAT_RUNTIME 1
- # define HAVE_FACCESSAT_RUNTIME 1
- # define HAVE_FCHMODAT_RUNTIME 1
- # define HAVE_FCHOWNAT_RUNTIME 1
- # define HAVE_LINKAT_RUNTIME 1
- # define HAVE_FDOPENDIR_RUNTIME 1
- # define HAVE_MKDIRAT_RUNTIME 1
- # define HAVE_RENAMEAT_RUNTIME 1
- # define HAVE_UNLINKAT_RUNTIME 1
- # define HAVE_OPENAT_RUNTIME 1
- # define HAVE_READLINKAT_RUNTIME 1
- # define HAVE_SYMLINKAT_RUNTIME 1
- # define HAVE_FUTIMENS_RUNTIME 1
- # define HAVE_UTIMENSAT_RUNTIME 1
- # define HAVE_PWRITEV_RUNTIME 1
- # define HAVE_MKFIFOAT_RUNTIME 1
- # define HAVE_MKNODAT_RUNTIME 1
- #endif
- #ifdef __cplusplus
- extern "C" {
- #endif
- PyDoc_STRVAR(posix__doc__,
- "This module provides access to operating system functionality that is\n\
- standardized by the C Standard and the POSIX standard (a thinly\n\
- disguised Unix interface). Refer to the library manual and\n\
- corresponding Unix manual entries for more information on calls.");
- #ifdef HAVE_SYS_UIO_H
- # include <sys/uio.h>
- #endif
- #ifdef HAVE_SYS_TYPES_H
- /* Should be included before <sys/sysmacros.h> on HP-UX v3 */
- # include <sys/types.h>
- #endif /* HAVE_SYS_TYPES_H */
- #ifdef HAVE_SYS_SYSMACROS_H
- /* GNU C Library: major(), minor(), makedev() */
- # include <sys/sysmacros.h>
- #endif
- #ifdef HAVE_SYS_STAT_H
- # include <sys/stat.h>
- #endif /* HAVE_SYS_STAT_H */
- #ifdef HAVE_SYS_WAIT_H
- # include <sys/wait.h> // WNOHANG
- #endif
- #ifdef HAVE_LINUX_WAIT_H
- # include <linux/wait.h> // P_PIDFD
- #endif
- #ifdef HAVE_SIGNAL_H
- # include <signal.h>
- #endif
- #ifdef HAVE_FCNTL_H
- # include <fcntl.h>
- #endif
- #ifdef HAVE_GRP_H
- # include <grp.h>
- #endif
- #ifdef HAVE_SYSEXITS_H
- # include <sysexits.h>
- #endif
- #ifdef HAVE_SYS_LOADAVG_H
- # include <sys/loadavg.h>
- #endif
- #ifdef HAVE_SYS_SENDFILE_H
- # include <sys/sendfile.h>
- #endif
- #if defined(__APPLE__)
- # include <copyfile.h>
- #endif
- #ifdef HAVE_SCHED_H
- # include <sched.h>
- #endif
- #ifdef HAVE_COPY_FILE_RANGE
- # include <unistd.h>
- #endif
- #if !defined(CPU_ALLOC) && defined(HAVE_SCHED_SETAFFINITY)
- # undef HAVE_SCHED_SETAFFINITY
- #endif
- #if defined(HAVE_SYS_XATTR_H) && defined(HAVE_LINUX_LIMITS_H) && !defined(__FreeBSD_kernel__) && !defined(__GNU__)
- # define USE_XATTRS
- # include <linux/limits.h> // Needed for XATTR_SIZE_MAX on musl libc.
- #endif
- #ifdef USE_XATTRS
- # include <sys/xattr.h>
- #endif
- #if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
- # ifdef HAVE_SYS_SOCKET_H
- # include <sys/socket.h>
- # endif
- #endif
- #ifdef HAVE_DLFCN_H
- # include <dlfcn.h>
- #endif
- #ifdef __hpux
- # include <sys/mpctl.h>
- #endif
- #if defined(__DragonFly__) || \
- defined(__OpenBSD__) || \
- defined(__FreeBSD__) || \
- defined(__NetBSD__) || \
- defined(__APPLE__)
- # include <sys/sysctl.h>
- #endif
- #ifdef HAVE_LINUX_RANDOM_H
- # include <linux/random.h>
- #endif
- #ifdef HAVE_GETRANDOM_SYSCALL
- # include <sys/syscall.h>
- # include <sys/random.h>
- #endif
- #ifdef HAVE_WINDOWS_CONSOLE_IO
- # define TERMSIZE_USE_CONIO
- #elif defined(HAVE_SYS_IOCTL_H)
- # include <sys/ioctl.h>
- # if defined(HAVE_TERMIOS_H)
- # include <termios.h>
- # endif
- # if defined(TIOCGWINSZ)
- # define TERMSIZE_USE_IOCTL
- # endif
- #endif /* HAVE_WINDOWS_CONSOLE_IO */
- /* Various compilers have only certain posix functions */
- /* XXX Gosh I wish these were all moved into pyconfig.h */
- #if defined(__WATCOMC__) && !defined(__QNX__) /* Watcom compiler */
- # define HAVE_OPENDIR 1
- # define HAVE_SYSTEM 1
- # include <process.h>
- #elif defined( _MSC_VER)
- /* Microsoft compiler */
- # if defined(MS_WINDOWS_DESKTOP) || defined(MS_WINDOWS_APP) || defined(MS_WINDOWS_SYSTEM)
- # define HAVE_GETPPID 1
- # endif /* MS_WINDOWS_DESKTOP | MS_WINDOWS_APP | MS_WINDOWS_SYSTEM */
- # if defined(MS_WINDOWS_DESKTOP)
- # define HAVE_GETLOGIN 1
- # endif /* MS_WINDOWS_DESKTOP */
- # if defined(MS_WINDOWS_DESKTOP) || defined(MS_WINDOWS_SYSTEM)
- # define HAVE_SPAWNV 1
- # define HAVE_EXECV 1
- # define HAVE_WSPAWNV 1
- # define HAVE_WEXECV 1
- # define HAVE_SYSTEM 1
- # define HAVE_CWAIT 1
- # endif /* MS_WINDOWS_DESKTOP | MS_WINDOWS_SYSTEM */
- # define HAVE_PIPE 1
- # define HAVE_FSYNC 1
- # define fsync _commit
- #endif /* ! __WATCOMC__ || __QNX__ */
- /*[clinic input]
- # one of the few times we lie about this name!
- module os
- [clinic start generated code]*/
- /*[clinic end generated code: output=da39a3ee5e6b4b0d input=94a0f0f978acae17]*/
- #ifndef _MSC_VER
- #if defined(__sgi)&&_COMPILER_VERSION>=700
- /* declare ctermid_r if compiling with MIPSPro 7.x in ANSI C mode
- (default) */
- extern char *ctermid_r(char *);
- #endif
- #endif /* !_MSC_VER */
- #if defined(__VXWORKS__)
- # error #include <vxCpuLib.h>
- # error #include <rtpLib.h>
- # include <wait.h>
- # error #include <taskLib.h>
- # ifndef _P_WAIT
- # define _P_WAIT 0
- # define _P_NOWAIT 1
- # define _P_NOWAITO 1
- # endif
- #endif /* __VXWORKS__ */
- #ifdef HAVE_POSIX_SPAWN
- # include <spawn.h>
- #endif
- #ifdef HAVE_UTIME_H
- # include <utime.h>
- #endif /* HAVE_UTIME_H */
- #ifdef HAVE_SYS_UTIME_H
- # include <sys/utime.h>
- # define HAVE_UTIME_H /* pretend we do for the rest of this file */
- #endif /* HAVE_SYS_UTIME_H */
- #ifdef HAVE_SYS_TIMES_H
- # include <sys/times.h>
- #endif /* HAVE_SYS_TIMES_H */
- #ifdef HAVE_SYS_PARAM_H
- # include <sys/param.h>
- #endif /* HAVE_SYS_PARAM_H */
- #ifdef HAVE_SYS_UTSNAME_H
- # include <sys/utsname.h>
- #endif /* HAVE_SYS_UTSNAME_H */
- #ifdef HAVE_DIRENT_H
- # include <dirent.h>
- # define NAMLEN(dirent) strlen((dirent)->d_name)
- #else
- # if defined(__WATCOMC__) && !defined(__QNX__)
- # include <direct.h>
- # define NAMLEN(dirent) strlen((dirent)->d_name)
- # else
- # define dirent direct
- # define NAMLEN(dirent) (dirent)->d_namlen
- # endif
- # ifdef HAVE_SYS_NDIR_H
- # include <sys/ndir.h>
- # endif
- # ifdef HAVE_SYS_DIR_H
- # include <sys/dir.h>
- # endif
- # ifdef HAVE_NDIR_H
- # include <ndir.h>
- # endif
- #endif
- #ifdef _MSC_VER
- # ifdef HAVE_DIRECT_H
- # include <direct.h>
- # endif
- # ifdef HAVE_IO_H
- # include <io.h>
- # endif
- # ifdef HAVE_PROCESS_H
- # include <process.h>
- # endif
- # include <malloc.h>
- #endif /* _MSC_VER */
- #ifndef MAXPATHLEN
- # if defined(PATH_MAX) && PATH_MAX > 1024
- # define MAXPATHLEN PATH_MAX
- # else
- # define MAXPATHLEN 1024
- # endif
- #endif /* MAXPATHLEN */
- #ifdef UNION_WAIT
- /* Emulate some macros on systems that have a union instead of macros */
- # ifndef WIFEXITED
- # define WIFEXITED(u_wait) (!(u_wait).w_termsig && !(u_wait).w_coredump)
- # endif
- # ifndef WEXITSTATUS
- # define WEXITSTATUS(u_wait) (WIFEXITED(u_wait)?((u_wait).w_retcode):-1)
- # endif
- # ifndef WTERMSIG
- # define WTERMSIG(u_wait) ((u_wait).w_termsig)
- # endif
- # define WAIT_TYPE union wait
- # define WAIT_STATUS_INT(s) (s.w_status)
- #else
- /* !UNION_WAIT */
- # define WAIT_TYPE int
- # define WAIT_STATUS_INT(s) (s)
- #endif /* UNION_WAIT */
- /* Don't use the "_r" form if we don't need it (also, won't have a
- prototype for it, at least on Solaris -- maybe others as well?). */
- #if defined(HAVE_CTERMID_R)
- # define USE_CTERMID_R
- #endif
- /* choose the appropriate stat and fstat functions and return structs */
- #undef STAT
- #undef FSTAT
- #undef STRUCT_STAT
- #ifdef MS_WINDOWS
- # define STAT win32_stat
- # define LSTAT win32_lstat
- # define FSTAT _Py_fstat_noraise
- # define STRUCT_STAT struct _Py_stat_struct
- #else
- # define STAT stat
- # define LSTAT lstat
- # define FSTAT fstat
- # define STRUCT_STAT struct stat
- #endif
- #if defined(MAJOR_IN_MKDEV)
- # include <sys/mkdev.h>
- #else
- # if defined(MAJOR_IN_SYSMACROS)
- # include <sys/sysmacros.h>
- # endif
- # if defined(HAVE_MKNOD) && defined(HAVE_SYS_MKDEV_H)
- # include <sys/mkdev.h>
- # endif
- #endif
- #ifdef MS_WINDOWS
- # define INITFUNC PyInit_nt
- # define MODNAME "nt"
- # define MODNAME_OBJ &_Py_ID(nt)
- #else
- # define INITFUNC PyInit_posix
- # define MODNAME "posix"
- # define MODNAME_OBJ &_Py_ID(posix)
- #endif
- #if defined(__sun)
- /* Something to implement in autoconf, not present in autoconf 2.69 */
- # define HAVE_STRUCT_STAT_ST_FSTYPE 1
- #endif
- /* memfd_create is either defined in sys/mman.h or sys/memfd.h
- * linux/memfd.h defines additional flags
- */
- #ifdef HAVE_SYS_MMAN_H
- # include <sys/mman.h>
- #endif
- #ifdef HAVE_SYS_MEMFD_H
- # error #include <sys/memfd.h>
- #endif
- #ifdef HAVE_LINUX_MEMFD_H
- # include <linux/memfd.h>
- #endif
- /* eventfd() */
- #ifdef HAVE_SYS_EVENTFD_H
- # include <sys/eventfd.h>
- #endif
- #ifdef _Py_MEMORY_SANITIZER
- # include <sanitizer/msan_interface.h>
- #endif
- #ifdef HAVE_FORK
- static void
- run_at_forkers(PyObject *lst, int reverse)
- {
- Py_ssize_t i;
- PyObject *cpy;
- if (lst != NULL) {
- assert(PyList_CheckExact(lst));
- /* Use a list copy in case register_at_fork() is called from
- * one of the callbacks.
- */
- cpy = PyList_GetSlice(lst, 0, PyList_GET_SIZE(lst));
- if (cpy == NULL)
- PyErr_WriteUnraisable(lst);
- else {
- if (reverse)
- PyList_Reverse(cpy);
- for (i = 0; i < PyList_GET_SIZE(cpy); i++) {
- PyObject *func, *res;
- func = PyList_GET_ITEM(cpy, i);
- res = _PyObject_CallNoArgs(func);
- if (res == NULL)
- PyErr_WriteUnraisable(func);
- else
- Py_DECREF(res);
- }
- Py_DECREF(cpy);
- }
- }
- }
- void
- PyOS_BeforeFork(void)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- run_at_forkers(interp->before_forkers, 1);
- _PyImport_AcquireLock(interp);
- }
- void
- PyOS_AfterFork_Parent(void)
- {
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (_PyImport_ReleaseLock(interp) <= 0) {
- Py_FatalError("failed releasing import lock after fork");
- }
- run_at_forkers(interp->after_forkers_parent, 0);
- }
- void
- PyOS_AfterFork_Child(void)
- {
- PyStatus status;
- _PyRuntimeState *runtime = &_PyRuntime;
- status = _PyRuntimeState_ReInitThreads(runtime);
- if (_PyStatus_EXCEPTION(status)) {
- goto fatal_error;
- }
- PyThreadState *tstate = _PyThreadState_GET();
- _Py_EnsureTstateNotNULL(tstate);
- #ifdef PY_HAVE_THREAD_NATIVE_ID
- tstate->native_thread_id = PyThread_get_thread_native_id();
- #endif
- status = _PyEval_ReInitThreads(tstate);
- if (_PyStatus_EXCEPTION(status)) {
- goto fatal_error;
- }
- status = _PyImport_ReInitLock(tstate->interp);
- if (_PyStatus_EXCEPTION(status)) {
- goto fatal_error;
- }
- _PySignal_AfterFork();
- status = _PyInterpreterState_DeleteExceptMain(runtime);
- if (_PyStatus_EXCEPTION(status)) {
- goto fatal_error;
- }
- assert(_PyThreadState_GET() == tstate);
- status = _PyPerfTrampoline_AfterFork_Child();
- if (_PyStatus_EXCEPTION(status)) {
- goto fatal_error;
- }
- run_at_forkers(tstate->interp->after_forkers_child, 0);
- return;
- fatal_error:
- Py_ExitStatusException(status);
- }
- static int
- register_at_forker(PyObject **lst, PyObject *func)
- {
- if (func == NULL) /* nothing to register? do nothing. */
- return 0;
- if (*lst == NULL) {
- *lst = PyList_New(0);
- if (*lst == NULL)
- return -1;
- }
- return PyList_Append(*lst, func);
- }
- #endif /* HAVE_FORK */
- /* Legacy wrapper */
- void
- PyOS_AfterFork(void)
- {
- #ifdef HAVE_FORK
- PyOS_AfterFork_Child();
- #endif
- }
- #ifdef MS_WINDOWS
- /* defined in fileutils.c */
- void _Py_time_t_to_FILE_TIME(time_t, int, FILETIME *);
- void _Py_attribute_data_to_stat(BY_HANDLE_FILE_INFORMATION *, ULONG,
- FILE_BASIC_INFO *, FILE_ID_INFO *,
- struct _Py_stat_struct *);
- void _Py_stat_basic_info_to_stat(FILE_STAT_BASIC_INFORMATION *,
- struct _Py_stat_struct *);
- #endif
- #ifndef MS_WINDOWS
- PyObject *
- _PyLong_FromUid(uid_t uid)
- {
- if (uid == (uid_t)-1)
- return PyLong_FromLong(-1);
- return PyLong_FromUnsignedLong(uid);
- }
- PyObject *
- _PyLong_FromGid(gid_t gid)
- {
- if (gid == (gid_t)-1)
- return PyLong_FromLong(-1);
- return PyLong_FromUnsignedLong(gid);
- }
- int
- _Py_Uid_Converter(PyObject *obj, uid_t *p)
- {
- uid_t uid;
- PyObject *index;
- int overflow;
- long result;
- unsigned long uresult;
- index = _PyNumber_Index(obj);
- if (index == NULL) {
- PyErr_Format(PyExc_TypeError,
- "uid should be integer, not %.200s",
- _PyType_Name(Py_TYPE(obj)));
- return 0;
- }
- /*
- * Handling uid_t is complicated for two reasons:
- * * Although uid_t is (always?) unsigned, it still
- * accepts -1.
- * * We don't know its size in advance--it may be
- * bigger than an int, or it may be smaller than
- * a long.
- *
- * So a bit of defensive programming is in order.
- * Start with interpreting the value passed
- * in as a signed long and see if it works.
- */
- result = PyLong_AsLongAndOverflow(index, &overflow);
- if (!overflow) {
- uid = (uid_t)result;
- if (result == -1) {
- if (PyErr_Occurred())
- goto fail;
- /* It's a legitimate -1, we're done. */
- goto success;
- }
- /* Any other negative number is disallowed. */
- if (result < 0)
- goto underflow;
- /* Ensure the value wasn't truncated. */
- if (sizeof(uid_t) < sizeof(long) &&
- (long)uid != result)
- goto underflow;
- goto success;
- }
- if (overflow < 0)
- goto underflow;
- /*
- * Okay, the value overflowed a signed long. If it
- * fits in an *unsigned* long, it may still be okay,
- * as uid_t may be unsigned long on this platform.
- */
- uresult = PyLong_AsUnsignedLong(index);
- if (PyErr_Occurred()) {
- if (PyErr_ExceptionMatches(PyExc_OverflowError))
- goto overflow;
- goto fail;
- }
- uid = (uid_t)uresult;
- /*
- * If uid == (uid_t)-1, the user actually passed in ULONG_MAX,
- * but this value would get interpreted as (uid_t)-1 by chown
- * and its siblings. That's not what the user meant! So we
- * throw an overflow exception instead. (We already
- * handled a real -1 with PyLong_AsLongAndOverflow() above.)
- */
- if (uid == (uid_t)-1)
- goto overflow;
- /* Ensure the value wasn't truncated. */
- if (sizeof(uid_t) < sizeof(long) &&
- (unsigned long)uid != uresult)
- goto overflow;
- /* fallthrough */
- success:
- Py_DECREF(index);
- *p = uid;
- return 1;
- underflow:
- PyErr_SetString(PyExc_OverflowError,
- "uid is less than minimum");
- goto fail;
- overflow:
- PyErr_SetString(PyExc_OverflowError,
- "uid is greater than maximum");
- /* fallthrough */
- fail:
- Py_DECREF(index);
- return 0;
- }
- int
- _Py_Gid_Converter(PyObject *obj, gid_t *p)
- {
- gid_t gid;
- PyObject *index;
- int overflow;
- long result;
- unsigned long uresult;
- index = _PyNumber_Index(obj);
- if (index == NULL) {
- PyErr_Format(PyExc_TypeError,
- "gid should be integer, not %.200s",
- _PyType_Name(Py_TYPE(obj)));
- return 0;
- }
- /*
- * Handling gid_t is complicated for two reasons:
- * * Although gid_t is (always?) unsigned, it still
- * accepts -1.
- * * We don't know its size in advance--it may be
- * bigger than an int, or it may be smaller than
- * a long.
- *
- * So a bit of defensive programming is in order.
- * Start with interpreting the value passed
- * in as a signed long and see if it works.
- */
- result = PyLong_AsLongAndOverflow(index, &overflow);
- if (!overflow) {
- gid = (gid_t)result;
- if (result == -1) {
- if (PyErr_Occurred())
- goto fail;
- /* It's a legitimate -1, we're done. */
- goto success;
- }
- /* Any other negative number is disallowed. */
- if (result < 0) {
- goto underflow;
- }
- /* Ensure the value wasn't truncated. */
- if (sizeof(gid_t) < sizeof(long) &&
- (long)gid != result)
- goto underflow;
- goto success;
- }
- if (overflow < 0)
- goto underflow;
- /*
- * Okay, the value overflowed a signed long. If it
- * fits in an *unsigned* long, it may still be okay,
- * as gid_t may be unsigned long on this platform.
- */
- uresult = PyLong_AsUnsignedLong(index);
- if (PyErr_Occurred()) {
- if (PyErr_ExceptionMatches(PyExc_OverflowError))
- goto overflow;
- goto fail;
- }
- gid = (gid_t)uresult;
- /*
- * If gid == (gid_t)-1, the user actually passed in ULONG_MAX,
- * but this value would get interpreted as (gid_t)-1 by chown
- * and its siblings. That's not what the user meant! So we
- * throw an overflow exception instead. (We already
- * handled a real -1 with PyLong_AsLongAndOverflow() above.)
- */
- if (gid == (gid_t)-1)
- goto overflow;
- /* Ensure the value wasn't truncated. */
- if (sizeof(gid_t) < sizeof(long) &&
- (unsigned long)gid != uresult)
- goto overflow;
- /* fallthrough */
- success:
- Py_DECREF(index);
- *p = gid;
- return 1;
- underflow:
- PyErr_SetString(PyExc_OverflowError,
- "gid is less than minimum");
- goto fail;
- overflow:
- PyErr_SetString(PyExc_OverflowError,
- "gid is greater than maximum");
- /* fallthrough */
- fail:
- Py_DECREF(index);
- return 0;
- }
- #endif /* MS_WINDOWS */
- static PyObject *
- _PyLong_FromDev(dev_t dev)
- {
- #ifdef NODEV
- if (dev == NODEV) {
- return PyLong_FromLongLong((long long)dev);
- }
- #endif
- return PyLong_FromUnsignedLongLong((unsigned long long)dev);
- }
- #if (defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)) || defined(HAVE_DEVICE_MACROS)
- static int
- _Py_Dev_Converter(PyObject *obj, void *p)
- {
- #ifdef NODEV
- if (PyLong_Check(obj) && _PyLong_IsNegative((PyLongObject *)obj)) {
- int overflow;
- long long result = PyLong_AsLongLongAndOverflow(obj, &overflow);
- if (result == -1 && PyErr_Occurred()) {
- return 0;
- }
- if (!overflow && result == (long long)NODEV) {
- *((dev_t *)p) = NODEV;
- return 1;
- }
- }
- #endif
- unsigned long long result = PyLong_AsUnsignedLongLong(obj);
- if (result == (unsigned long long)-1 && PyErr_Occurred()) {
- return 0;
- }
- if ((unsigned long long)(dev_t)result != result) {
- PyErr_SetString(PyExc_OverflowError,
- "Python int too large to convert to C dev_t");
- return 0;
- }
- *((dev_t *)p) = (dev_t)result;
- return 1;
- }
- #endif /* (HAVE_MKNOD && HAVE_MAKEDEV) || HAVE_DEVICE_MACROS */
- #ifdef AT_FDCWD
- /*
- * Why the (int) cast? Solaris 10 defines AT_FDCWD as 0xffd19553 (-3041965);
- * without the int cast, the value gets interpreted as uint (4291925331),
- * which doesn't play nicely with all the initializer lines in this file that
- * look like this:
- * int dir_fd = DEFAULT_DIR_FD;
- */
- #define DEFAULT_DIR_FD (int)AT_FDCWD
- #else
- #define DEFAULT_DIR_FD (-100)
- #endif
- static int
- _fd_converter(PyObject *o, int *p)
- {
- int overflow;
- long long_value;
- PyObject *index = _PyNumber_Index(o);
- if (index == NULL) {
- return 0;
- }
- assert(PyLong_Check(index));
- long_value = PyLong_AsLongAndOverflow(index, &overflow);
- Py_DECREF(index);
- assert(!PyErr_Occurred());
- if (overflow > 0 || long_value > INT_MAX) {
- PyErr_SetString(PyExc_OverflowError,
- "fd is greater than maximum");
- return 0;
- }
- if (overflow < 0 || long_value < INT_MIN) {
- PyErr_SetString(PyExc_OverflowError,
- "fd is less than minimum");
- return 0;
- }
- *p = (int)long_value;
- return 1;
- }
- static int
- dir_fd_converter(PyObject *o, void *p)
- {
- if (o == Py_None) {
- *(int *)p = DEFAULT_DIR_FD;
- return 1;
- }
- else if (PyIndex_Check(o)) {
- return _fd_converter(o, (int *)p);
- }
- else {
- PyErr_Format(PyExc_TypeError,
- "argument should be integer or None, not %.200s",
- _PyType_Name(Py_TYPE(o)));
- return 0;
- }
- }
- typedef struct {
- PyObject *billion;
- PyObject *DirEntryType;
- PyObject *ScandirIteratorType;
- #if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM)
- PyObject *SchedParamType;
- #endif
- newfunc statresult_new_orig;
- PyObject *StatResultType;
- PyObject *StatVFSResultType;
- PyObject *TerminalSizeType;
- PyObject *TimesResultType;
- PyObject *UnameResultType;
- #if defined(HAVE_WAITID) && !defined(__APPLE__)
- PyObject *WaitidResultType;
- #endif
- #if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
- PyObject *struct_rusage;
- #endif
- PyObject *st_mode;
- } _posixstate;
- static inline _posixstate*
- get_posix_state(PyObject *module)
- {
- void *state = _PyModule_GetState(module);
- assert(state != NULL);
- return (_posixstate *)state;
- }
- /*
- * A PyArg_ParseTuple "converter" function
- * that handles filesystem paths in the manner
- * preferred by the os module.
- *
- * path_converter accepts (Unicode) strings and their
- * subclasses, and bytes and their subclasses. What
- * it does with the argument depends on path.make_wide:
- *
- * * If path.make_wide is nonzero, if we get a (Unicode)
- * string we extract the wchar_t * and return it; if we
- * get bytes we decode to wchar_t * and return that.
- *
- * * If path.make_wide is zero, if we get bytes we extract
- * the char_t * and return it; if we get a (Unicode)
- * string we encode to char_t * and return that.
- *
- * path_converter also optionally accepts signed
- * integers (representing open file descriptors) instead
- * of path strings.
- *
- * Input fields:
- * path.nullable
- * If nonzero, the path is permitted to be None.
- * path.nonstrict
- * If nonzero, the path is permitted to contain
- * embedded null characters and have any length.
- * path.make_wide
- * If nonzero, the converter always uses wide, decoding if necessary, else
- * it always uses narrow, encoding if necessary. The default value is
- * nonzero on Windows, else zero.
- * path.suppress_value_error
- * If nonzero, raising ValueError is suppressed.
- * path.allow_fd
- * If nonzero, the path is permitted to be a file handle
- * (a signed int) instead of a string.
- * path.function_name
- * If non-NULL, path_converter will use that as the name
- * of the function in error messages.
- * (If path.function_name is NULL it omits the function name.)
- * path.argument_name
- * If non-NULL, path_converter will use that as the name
- * of the parameter in error messages.
- * (If path.argument_name is NULL it uses "path".)
- *
- * Output fields:
- * path.wide
- * Points to the path if it was expressed as Unicode
- * or if it was bytes and decoded to Unicode.
- * path.narrow
- * Points to the path if it was expressed as bytes,
- * or if it was Unicode and encoded to bytes.
- * path.fd
- * Contains a file descriptor if path.accept_fd was true
- * and the caller provided a signed integer instead of any
- * sort of string.
- *
- * WARNING: if your "path" parameter is optional, and is
- * unspecified, path_converter will never get called.
- * So if you set allow_fd, you *MUST* initialize path.fd = -1
- * yourself!
- * path.value_error
- * If nonzero, then suppress_value_error was specified and a ValueError
- * occurred.
- * path.length
- * The length of the path in characters, if specified as
- * a string.
- * path.object
- * The original object passed in (if get a PathLike object,
- * the result of PyOS_FSPath() is treated as the original object).
- * Own a reference to the object.
- * path.cleanup
- * For internal use only. May point to a temporary object.
- * (Pay no attention to the man behind the curtain.)
- *
- * At most one of path.wide or path.narrow will be non-NULL.
- * If path was None and path.nullable was set,
- * or if path was an integer and path.allow_fd was set,
- * both path.wide and path.narrow will be NULL
- * and path.length will be 0.
- *
- * path_converter takes care to not write to the path_t
- * unless it's successful. However it must reset the
- * "cleanup" field each time it's called.
- *
- * Use as follows:
- * path_t path;
- * memset(&path, 0, sizeof(path));
- * PyArg_ParseTuple(args, "O&", path_converter, &path);
- * // ... use values from path ...
- * path_cleanup(&path);
- *
- * (Note that if PyArg_Parse fails you don't need to call
- * path_cleanup(). However it is safe to do so.)
- */
- typedef struct {
- // Input fields
- const char *function_name;
- const char *argument_name;
- int nullable;
- int nonstrict;
- int make_wide;
- int suppress_value_error;
- int allow_fd;
- // Output fields
- const wchar_t *wide;
- const char *narrow;
- int fd;
- int value_error;
- Py_ssize_t length;
- PyObject *object;
- PyObject *cleanup;
- } path_t;
- #define PATH_T_INITIALIZE(function_name, argument_name, nullable, nonstrict, \
- make_wide, suppress_value_error, allow_fd) \
- {function_name, argument_name, nullable, nonstrict, make_wide, \
- suppress_value_error, allow_fd, NULL, NULL, -1, 0, 0, NULL, NULL}
- #ifdef MS_WINDOWS
- #define PATH_T_INITIALIZE_P(function_name, argument_name, nullable, \
- nonstrict, suppress_value_error, allow_fd) \
- PATH_T_INITIALIZE(function_name, argument_name, nullable, nonstrict, 1, \
- suppress_value_error, allow_fd)
- #else
- #define PATH_T_INITIALIZE_P(function_name, argument_name, nullable, \
- nonstrict, suppress_value_error, allow_fd) \
- PATH_T_INITIALIZE(function_name, argument_name, nullable, nonstrict, 0, \
- suppress_value_error, allow_fd)
- #endif
- static void
- path_cleanup(path_t *path)
- {
- wchar_t *wide = (wchar_t *)path->wide;
- path->wide = NULL;
- PyMem_Free(wide);
- Py_CLEAR(path->object);
- Py_CLEAR(path->cleanup);
- }
- static int
- path_converter(PyObject *o, void *p)
- {
- path_t *path = (path_t *)p;
- PyObject *bytes = NULL;
- Py_ssize_t length = 0;
- int is_index, is_bytes, is_unicode;
- const char *narrow;
- PyObject *wo = NULL;
- wchar_t *wide = NULL;
- #define FORMAT_EXCEPTION(exc, fmt) \
- PyErr_Format(exc, "%s%s" fmt, \
- path->function_name ? path->function_name : "", \
- path->function_name ? ": " : "", \
- path->argument_name ? path->argument_name : "path")
- /* Py_CLEANUP_SUPPORTED support */
- if (o == NULL) {
- path_cleanup(path);
- return 1;
- }
- /* Ensure it's always safe to call path_cleanup(). */
- path->object = path->cleanup = NULL;
- /* path->object owns a reference to the original object */
- Py_INCREF(o);
- if ((o == Py_None) && path->nullable) {
- path->wide = NULL;
- path->narrow = NULL;
- path->fd = -1;
- goto success_exit;
- }
- /* Only call this here so that we don't treat the return value of
- os.fspath() as an fd or buffer. */
- is_index = path->allow_fd && PyIndex_Check(o);
- is_bytes = PyBytes_Check(o);
- is_unicode = PyUnicode_Check(o);
- if (!is_index && !is_unicode && !is_bytes) {
- /* Inline PyOS_FSPath() for better error messages. */
- PyObject *func, *res;
- func = _PyObject_LookupSpecial(o, &_Py_ID(__fspath__));
- if (NULL == func) {
- goto error_format;
- }
- res = _PyObject_CallNoArgs(func);
- Py_DECREF(func);
- if (NULL == res) {
- goto error_exit;
- }
- else if (PyUnicode_Check(res)) {
- is_unicode = 1;
- }
- else if (PyBytes_Check(res)) {
- is_bytes = 1;
- }
- else {
- PyErr_Format(PyExc_TypeError,
- "expected %.200s.__fspath__() to return str or bytes, "
- "not %.200s", _PyType_Name(Py_TYPE(o)),
- _PyType_Name(Py_TYPE(res)));
- Py_DECREF(res);
- goto error_exit;
- }
- /* still owns a reference to the original object */
- Py_SETREF(o, res);
- }
- if (is_unicode) {
- if (path->make_wide) {
- wide = PyUnicode_AsWideCharString(o, &length);
- if (!wide) {
- goto error_exit;
- }
- #ifdef MS_WINDOWS
- if (!path->nonstrict && length > 32767) {
- FORMAT_EXCEPTION(PyExc_ValueError, "%s too long for Windows");
- goto error_exit;
- }
- #endif
- if (!path->nonstrict && wcslen(wide) != (size_t)length) {
- FORMAT_EXCEPTION(PyExc_ValueError,
- "embedded null character in %s");
- goto error_exit;
- }
- path->wide = wide;
- path->narrow = NULL;
- path->fd = -1;
- wide = NULL;
- goto success_exit;
- }
- bytes = PyUnicode_EncodeFSDefault(o);
- if (!bytes) {
- goto error_exit;
- }
- }
- else if (is_bytes) {
- bytes = Py_NewRef(o);
- }
- else if (is_index) {
- if (!_fd_converter(o, &path->fd)) {
- goto error_exit;
- }
- path->wide = NULL;
- path->narrow = NULL;
- goto success_exit;
- }
- else {
- error_format:
- PyErr_Format(PyExc_TypeError, "%s%s%s should be %s, not %.200s",
- path->function_name ? path->function_name : "",
- path->function_name ? ": " : "",
- path->argument_name ? path->argument_name : "path",
- path->allow_fd && path->nullable ? "string, bytes, os.PathLike, "
- "integer or None" :
- path->allow_fd ? "string, bytes, os.PathLike or integer" :
- path->nullable ? "string, bytes, os.PathLike or None" :
- "string, bytes or os.PathLike",
- _PyType_Name(Py_TYPE(o)));
- goto error_exit;
- }
- length = PyBytes_GET_SIZE(bytes);
- narrow = PyBytes_AS_STRING(bytes);
- if (!path->nonstrict && strlen(narrow) != (size_t)length) {
- FORMAT_EXCEPTION(PyExc_ValueError, "embedded null character in %s");
- goto error_exit;
- }
- if (path->make_wide) {
- wo = PyUnicode_DecodeFSDefaultAndSize(narrow, length);
- if (!wo) {
- goto error_exit;
- }
- wide = PyUnicode_AsWideCharString(wo, &length);
- Py_DECREF(wo);
- if (!wide) {
- goto error_exit;
- }
- #ifdef MS_WINDOWS
- if (!path->nonstrict && length > 32767) {
- FORMAT_EXCEPTION(PyExc_ValueError, "%s too long for Windows");
- goto error_exit;
- }
- #endif
- if (!path->nonstrict && wcslen(wide) != (size_t)length) {
- FORMAT_EXCEPTION(PyExc_ValueError,
- "embedded null character in %s");
- goto error_exit;
- }
- path->wide = wide;
- path->narrow = NULL;
- Py_DECREF(bytes);
- wide = NULL;
- }
- else {
- path->wide = NULL;
- path->narrow = narrow;
- if (bytes == o) {
- /* Still a reference owned by path->object, don't have to
- worry about path->narrow is used after free. */
- Py_DECREF(bytes);
- }
- else {
- path->cleanup = bytes;
- }
- }
- path->fd = -1;
- success_exit:
- path->value_error = 0;
- path->length = length;
- path->object = o;
- return Py_CLEANUP_SUPPORTED;
- error_exit:
- Py_XDECREF(o);
- Py_XDECREF(bytes);
- PyMem_Free(wide);
- if (!path->suppress_value_error ||
- !PyErr_ExceptionMatches(PyExc_ValueError))
- {
- return 0;
- }
- PyErr_Clear();
- path->wide = NULL;
- path->narrow = NULL;
- path->fd = -1;
- path->value_error = 1;
- path->length = 0;
- path->object = NULL;
- return Py_CLEANUP_SUPPORTED;
- }
- static void
- argument_unavailable_error(const char *function_name, const char *argument_name)
- {
- PyErr_Format(PyExc_NotImplementedError,
- "%s%s%s unavailable on this platform",
- (function_name != NULL) ? function_name : "",
- (function_name != NULL) ? ": ": "",
- argument_name);
- }
- static int
- dir_fd_unavailable(PyObject *o, void *p)
- {
- int dir_fd;
- if (!dir_fd_converter(o, &dir_fd))
- return 0;
- if (dir_fd != DEFAULT_DIR_FD) {
- argument_unavailable_error(NULL, "dir_fd");
- return 0;
- }
- *(int *)p = dir_fd;
- return 1;
- }
- static int
- fd_specified(const char *function_name, int fd)
- {
- if (fd == -1)
- return 0;
- argument_unavailable_error(function_name, "fd");
- return 1;
- }
- static int
- follow_symlinks_specified(const char *function_name, int follow_symlinks)
- {
- if (follow_symlinks)
- return 0;
- argument_unavailable_error(function_name, "follow_symlinks");
- return 1;
- }
- static int
- path_and_dir_fd_invalid(const char *function_name, path_t *path, int dir_fd)
- {
- if (!path->wide && (dir_fd != DEFAULT_DIR_FD) && !path->narrow) {
- PyErr_Format(PyExc_ValueError,
- "%s: can't specify dir_fd without matching path",
- function_name);
- return 1;
- }
- return 0;
- }
- static int
- dir_fd_and_fd_invalid(const char *function_name, int dir_fd, int fd)
- {
- if ((dir_fd != DEFAULT_DIR_FD) && (fd != -1)) {
- PyErr_Format(PyExc_ValueError,
- "%s: can't specify both dir_fd and fd",
- function_name);
- return 1;
- }
- return 0;
- }
- static int
- fd_and_follow_symlinks_invalid(const char *function_name, int fd,
- int follow_symlinks)
- {
- if ((fd > 0) && (!follow_symlinks)) {
- PyErr_Format(PyExc_ValueError,
- "%s: cannot use fd and follow_symlinks together",
- function_name);
- return 1;
- }
- return 0;
- }
- static int
- dir_fd_and_follow_symlinks_invalid(const char *function_name, int dir_fd,
- int follow_symlinks)
- {
- if ((dir_fd != DEFAULT_DIR_FD) && (!follow_symlinks)) {
- PyErr_Format(PyExc_ValueError,
- "%s: cannot use dir_fd and follow_symlinks together",
- function_name);
- return 1;
- }
- return 0;
- }
- #ifdef MS_WINDOWS
- typedef long long Py_off_t;
- #else
- typedef off_t Py_off_t;
- #endif
- static int
- Py_off_t_converter(PyObject *arg, void *addr)
- {
- #ifdef HAVE_LARGEFILE_SUPPORT
- *((Py_off_t *)addr) = PyLong_AsLongLong(arg);
- #else
- *((Py_off_t *)addr) = PyLong_AsLong(arg);
- #endif
- if (PyErr_Occurred())
- return 0;
- return 1;
- }
- static PyObject *
- PyLong_FromPy_off_t(Py_off_t offset)
- {
- #ifdef HAVE_LARGEFILE_SUPPORT
- return PyLong_FromLongLong(offset);
- #else
- return PyLong_FromLong(offset);
- #endif
- }
- #ifdef HAVE_SIGSET_T
- /* Convert an iterable of integers to a sigset.
- Return 1 on success, return 0 and raise an exception on error. */
- int
- _Py_Sigset_Converter(PyObject *obj, void *addr)
- {
- sigset_t *mask = (sigset_t *)addr;
- PyObject *iterator, *item;
- long signum;
- int overflow;
- // The extra parens suppress the unreachable-code warning with clang on MacOS
- if (sigemptyset(mask) < (0)) {
- /* Probably only if mask == NULL. */
- PyErr_SetFromErrno(PyExc_OSError);
- return 0;
- }
- iterator = PyObject_GetIter(obj);
- if (iterator == NULL) {
- return 0;
- }
- while ((item = PyIter_Next(iterator)) != NULL) {
- signum = PyLong_AsLongAndOverflow(item, &overflow);
- Py_DECREF(item);
- if (signum <= 0 || signum >= Py_NSIG) {
- if (overflow || signum != -1 || !PyErr_Occurred()) {
- PyErr_Format(PyExc_ValueError,
- "signal number %ld out of range [1; %i]",
- signum, Py_NSIG - 1);
- }
- goto error;
- }
- if (sigaddset(mask, (int)signum)) {
- if (errno != EINVAL) {
- /* Probably impossible */
- PyErr_SetFromErrno(PyExc_OSError);
- goto error;
- }
- /* For backwards compatibility, allow idioms such as
- * `range(1, NSIG)` but warn about invalid signal numbers
- */
- const char msg[] =
- "invalid signal number %ld, please use valid_signals()";
- if (PyErr_WarnFormat(PyExc_RuntimeWarning, 1, msg, signum)) {
- goto error;
- }
- }
- }
- if (!PyErr_Occurred()) {
- Py_DECREF(iterator);
- return 1;
- }
- error:
- Py_DECREF(iterator);
- return 0;
- }
- #endif /* HAVE_SIGSET_T */
- /* Return a dictionary corresponding to the POSIX environment table */
- #if defined(WITH_NEXT_FRAMEWORK) || (defined(__APPLE__) && defined(Py_ENABLE_SHARED))
- /* On Darwin/MacOSX a shared library or framework has no access to
- ** environ directly, we must obtain it with _NSGetEnviron(). See also
- ** man environ(7).
- */
- #include <crt_externs.h>
- #elif !defined(_MSC_VER) && (!defined(__WATCOMC__) || defined(__QNX__) || defined(__VXWORKS__))
- extern char **environ;
- #endif /* !_MSC_VER */
- static PyObject *
- convertenviron(void)
- {
- PyObject *d;
- #ifdef MS_WINDOWS
- wchar_t **e;
- #else
- char **e;
- #endif
- d = PyDict_New();
- if (d == NULL)
- return NULL;
- #ifdef MS_WINDOWS
- /* _wenviron must be initialized in this way if the program is started
- through main() instead of wmain(). */
- (void)_wgetenv(L"");
- e = _wenviron;
- #elif defined(WITH_NEXT_FRAMEWORK) || (defined(__APPLE__) && defined(Py_ENABLE_SHARED))
- /* environ is not accessible as an extern in a shared object on OSX; use
- _NSGetEnviron to resolve it. The value changes if you add environment
- variables between calls to Py_Initialize, so don't cache the value. */
- e = *_NSGetEnviron();
- #else
- e = environ;
- #endif
- if (e == NULL)
- return d;
- for (; *e != NULL; e++) {
- PyObject *k;
- PyObject *v;
- #ifdef MS_WINDOWS
- const wchar_t *p = wcschr(*e, L'=');
- #else
- const char *p = strchr(*e, '=');
- #endif
- if (p == NULL)
- continue;
- #ifdef MS_WINDOWS
- k = PyUnicode_FromWideChar(*e, (Py_ssize_t)(p-*e));
- #else
- k = PyBytes_FromStringAndSize(*e, (int)(p-*e));
- #endif
- if (k == NULL) {
- Py_DECREF(d);
- return NULL;
- }
- #ifdef MS_WINDOWS
- v = PyUnicode_FromWideChar(p+1, wcslen(p+1));
- #else
- v = PyBytes_FromStringAndSize(p+1, strlen(p+1));
- #endif
- if (v == NULL) {
- Py_DECREF(k);
- Py_DECREF(d);
- return NULL;
- }
- if (PyDict_SetDefault(d, k, v) == NULL) {
- Py_DECREF(v);
- Py_DECREF(k);
- Py_DECREF(d);
- return NULL;
- }
- Py_DECREF(k);
- Py_DECREF(v);
- }
- return d;
- }
- /* Set a POSIX-specific error from errno, and return NULL */
- static PyObject *
- posix_error(void)
- {
- return PyErr_SetFromErrno(PyExc_OSError);
- }
- #ifdef MS_WINDOWS
- static PyObject *
- win32_error(const char* function, const char* filename)
- {
- /* XXX We should pass the function name along in the future.
- (winreg.c also wants to pass the function name.)
- This would however require an additional param to the
- Windows error object, which is non-trivial.
- */
- errno = GetLastError();
- if (filename)
- return PyErr_SetFromWindowsErrWithFilename(errno, filename);
- else
- return PyErr_SetFromWindowsErr(errno);
- }
- static PyObject *
- win32_error_object_err(const char* function, PyObject* filename, DWORD err)
- {
- /* XXX - see win32_error for comments on 'function' */
- if (filename)
- return PyErr_SetExcFromWindowsErrWithFilenameObject(
- PyExc_OSError,
- err,
- filename);
- else
- return PyErr_SetFromWindowsErr(err);
- }
- static PyObject *
- win32_error_object(const char* function, PyObject* filename)
- {
- errno = GetLastError();
- return win32_error_object_err(function, filename, errno);
- }
- #endif /* MS_WINDOWS */
- static PyObject *
- posix_path_object_error(PyObject *path)
- {
- return PyErr_SetFromErrnoWithFilenameObject(PyExc_OSError, path);
- }
- static PyObject *
- path_object_error(PyObject *path)
- {
- #ifdef MS_WINDOWS
- return PyErr_SetExcFromWindowsErrWithFilenameObject(
- PyExc_OSError, 0, path);
- #else
- return posix_path_object_error(path);
- #endif
- }
- static PyObject *
- path_object_error2(PyObject *path, PyObject *path2)
- {
- #ifdef MS_WINDOWS
- return PyErr_SetExcFromWindowsErrWithFilenameObjects(
- PyExc_OSError, 0, path, path2);
- #else
- return PyErr_SetFromErrnoWithFilenameObjects(PyExc_OSError, path, path2);
- #endif
- }
- static PyObject *
- path_error(path_t *path)
- {
- return path_object_error(path->object);
- }
- static PyObject *
- posix_path_error(path_t *path)
- {
- return posix_path_object_error(path->object);
- }
- static PyObject *
- path_error2(path_t *path, path_t *path2)
- {
- return path_object_error2(path->object, path2->object);
- }
- /* POSIX generic methods */
- static PyObject *
- posix_fildes_fd(int fd, int (*func)(int))
- {
- int res;
- int async_err = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- res = (*func)(fd);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (res != 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res != 0)
- return (!async_err) ? posix_error() : NULL;
- Py_RETURN_NONE;
- }
- #ifdef MS_WINDOWS
- /* This is a reimplementation of the C library's chdir function,
- but one that produces Win32 errors instead of DOS error codes.
- chdir is essentially a wrapper around SetCurrentDirectory; however,
- it also needs to set "magic" environment variables indicating
- the per-drive current directory, which are of the form =<drive>: */
- static BOOL __stdcall
- win32_wchdir(LPCWSTR path)
- {
- wchar_t path_buf[MAX_PATH], *new_path = path_buf;
- int result;
- wchar_t env[4] = L"=x:";
- if(!SetCurrentDirectoryW(path))
- return FALSE;
- result = GetCurrentDirectoryW(Py_ARRAY_LENGTH(path_buf), new_path);
- if (!result)
- return FALSE;
- if (result > Py_ARRAY_LENGTH(path_buf)) {
- new_path = PyMem_RawMalloc(result * sizeof(wchar_t));
- if (!new_path) {
- SetLastError(ERROR_OUTOFMEMORY);
- return FALSE;
- }
- result = GetCurrentDirectoryW(result, new_path);
- if (!result) {
- PyMem_RawFree(new_path);
- return FALSE;
- }
- }
- int is_unc_like_path = (wcsncmp(new_path, L"\\\\", 2) == 0 ||
- wcsncmp(new_path, L"//", 2) == 0);
- if (!is_unc_like_path) {
- env[1] = new_path[0];
- result = SetEnvironmentVariableW(env, new_path);
- }
- if (new_path != path_buf)
- PyMem_RawFree(new_path);
- return result ? TRUE : FALSE;
- }
- #endif
- #ifdef MS_WINDOWS
- /* The CRT of Windows has a number of flaws wrt. its stat() implementation:
- - time stamps are restricted to second resolution
- - file modification times suffer from forth-and-back conversions between
- UTC and local time
- Therefore, we implement our own stat, based on the Win32 API directly.
- */
- #define HAVE_STAT_NSEC 1
- #define HAVE_STRUCT_STAT_ST_FILE_ATTRIBUTES 1
- #define HAVE_STRUCT_STAT_ST_REPARSE_TAG 1
- static void
- find_data_to_file_info(WIN32_FIND_DATAW *pFileData,
- BY_HANDLE_FILE_INFORMATION *info,
- ULONG *reparse_tag)
- {
- memset(info, 0, sizeof(*info));
- info->dwFileAttributes = pFileData->dwFileAttributes;
- info->ftCreationTime = pFileData->ftCreationTime;
- info->ftLastAccessTime = pFileData->ftLastAccessTime;
- info->ftLastWriteTime = pFileData->ftLastWriteTime;
- info->nFileSizeHigh = pFileData->nFileSizeHigh;
- info->nFileSizeLow = pFileData->nFileSizeLow;
- /* info->nNumberOfLinks = 1; */
- if (pFileData->dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
- *reparse_tag = pFileData->dwReserved0;
- else
- *reparse_tag = 0;
- }
- static BOOL
- attributes_from_dir(LPCWSTR pszFile, BY_HANDLE_FILE_INFORMATION *info, ULONG *reparse_tag)
- {
- HANDLE hFindFile;
- WIN32_FIND_DATAW FileData;
- LPCWSTR filename = pszFile;
- size_t n = wcslen(pszFile);
- if (n && (pszFile[n - 1] == L'\\' || pszFile[n - 1] == L'/')) {
- // cannot use PyMem_Malloc here because we do not hold the GIL
- filename = (LPCWSTR)malloc((n + 1) * sizeof(filename[0]));
- if(!filename) {
- SetLastError(ERROR_NOT_ENOUGH_MEMORY);
- return FALSE;
- }
- wcsncpy_s((LPWSTR)filename, n + 1, pszFile, n);
- while (--n > 0 && (filename[n] == L'\\' || filename[n] == L'/')) {
- ((LPWSTR)filename)[n] = L'\0';
- }
- if (!n || (n == 1 && filename[1] == L':')) {
- // Nothing left to query
- free((void *)filename);
- return FALSE;
- }
- }
- hFindFile = FindFirstFileW(filename, &FileData);
- if (pszFile != filename) {
- free((void *)filename);
- }
- if (hFindFile == INVALID_HANDLE_VALUE) {
- return FALSE;
- }
- FindClose(hFindFile);
- find_data_to_file_info(&FileData, info, reparse_tag);
- return TRUE;
- }
- static void
- update_st_mode_from_path(const wchar_t *path, DWORD attr,
- struct _Py_stat_struct *result)
- {
- if (!(attr & FILE_ATTRIBUTE_DIRECTORY)) {
- /* Fix the file execute permissions. This hack sets S_IEXEC if
- the filename has an extension that is commonly used by files
- that CreateProcessW can execute. A real implementation calls
- GetSecurityInfo, OpenThreadToken/OpenProcessToken, and
- AccessCheck to check for generic read, write, and execute
- access. */
- const wchar_t *fileExtension = wcsrchr(path, '.');
- if (fileExtension) {
- if (_wcsicmp(fileExtension, L".exe") == 0 ||
- _wcsicmp(fileExtension, L".bat") == 0 ||
- _wcsicmp(fileExtension, L".cmd") == 0 ||
- _wcsicmp(fileExtension, L".com") == 0) {
- result->st_mode |= 0111;
- }
- }
- }
- }
- static int
- win32_xstat_slow_impl(const wchar_t *path, struct _Py_stat_struct *result,
- BOOL traverse)
- {
- HANDLE hFile;
- BY_HANDLE_FILE_INFORMATION fileInfo;
- FILE_BASIC_INFO basicInfo;
- FILE_BASIC_INFO *pBasicInfo = NULL;
- FILE_ID_INFO idInfo;
- FILE_ID_INFO *pIdInfo = NULL;
- FILE_ATTRIBUTE_TAG_INFO tagInfo = { 0 };
- DWORD fileType, error;
- BOOL isUnhandledTag = FALSE;
- int retval = 0;
- DWORD access = FILE_READ_ATTRIBUTES;
- DWORD flags = FILE_FLAG_BACKUP_SEMANTICS; /* Allow opening directories. */
- if (!traverse) {
- flags |= FILE_FLAG_OPEN_REPARSE_POINT;
- }
- hFile = CreateFileW(path, access, 0, NULL, OPEN_EXISTING, flags, NULL);
- if (hFile == INVALID_HANDLE_VALUE) {
- /* Either the path doesn't exist, or the caller lacks access. */
- error = GetLastError();
- switch (error) {
- case ERROR_ACCESS_DENIED: /* Cannot sync or read attributes. */
- case ERROR_SHARING_VIOLATION: /* It's a paging file. */
- /* Try reading the parent directory. */
- if (!attributes_from_dir(path, &fileInfo, &tagInfo.ReparseTag)) {
- /* Cannot read the parent directory. */
- switch (GetLastError()) {
- case ERROR_FILE_NOT_FOUND: /* File cannot be found */
- case ERROR_PATH_NOT_FOUND: /* File parent directory cannot be found */
- case ERROR_NOT_READY: /* Drive exists but unavailable */
- case ERROR_BAD_NET_NAME: /* Remote drive unavailable */
- break;
- /* Restore the error from CreateFileW(). */
- default:
- SetLastError(error);
- }
- return -1;
- }
- if (fileInfo.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
- if (traverse ||
- !IsReparseTagNameSurrogate(tagInfo.ReparseTag)) {
- /* The stat call has to traverse but cannot, so fail. */
- SetLastError(error);
- return -1;
- }
- }
- break;
- case ERROR_INVALID_PARAMETER:
- /* \\.\con requires read or write access. */
- hFile = CreateFileW(path, access | GENERIC_READ,
- FILE_SHARE_READ | FILE_SHARE_WRITE, NULL,
- OPEN_EXISTING, flags, NULL);
- if (hFile == INVALID_HANDLE_VALUE) {
- SetLastError(error);
- return -1;
- }
- break;
- case ERROR_CANT_ACCESS_FILE:
- /* bpo37834: open unhandled reparse points if traverse fails. */
- if (traverse) {
- traverse = FALSE;
- isUnhandledTag = TRUE;
- hFile = CreateFileW(path, access, 0, NULL, OPEN_EXISTING,
- flags | FILE_FLAG_OPEN_REPARSE_POINT, NULL);
- }
- if (hFile == INVALID_HANDLE_VALUE) {
- SetLastError(error);
- return -1;
- }
- break;
- default:
- return -1;
- }
- }
- if (hFile != INVALID_HANDLE_VALUE) {
- /* Handle types other than files on disk. */
- fileType = GetFileType(hFile);
- if (fileType != FILE_TYPE_DISK) {
- if (fileType == FILE_TYPE_UNKNOWN && GetLastError() != 0) {
- retval = -1;
- goto cleanup;
- }
- DWORD fileAttributes = GetFileAttributesW(path);
- memset(result, 0, sizeof(*result));
- if (fileAttributes != INVALID_FILE_ATTRIBUTES &&
- fileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
- /* \\.\pipe\ or \\.\mailslot\ */
- result->st_mode = _S_IFDIR;
- } else if (fileType == FILE_TYPE_CHAR) {
- /* \\.\nul */
- result->st_mode = _S_IFCHR;
- } else if (fileType == FILE_TYPE_PIPE) {
- /* \\.\pipe\spam */
- result->st_mode = _S_IFIFO;
- }
- /* FILE_TYPE_UNKNOWN, e.g. \\.\mailslot\waitfor.exe\spam */
- goto cleanup;
- }
- /* Query the reparse tag, and traverse a non-link. */
- if (!traverse) {
- if (!GetFileInformationByHandleEx(hFile, FileAttributeTagInfo,
- &tagInfo, sizeof(tagInfo))) {
- /* Allow devices that do not support FileAttributeTagInfo. */
- switch (GetLastError()) {
- case ERROR_INVALID_PARAMETER:
- case ERROR_INVALID_FUNCTION:
- case ERROR_NOT_SUPPORTED:
- tagInfo.FileAttributes = FILE_ATTRIBUTE_NORMAL;
- tagInfo.ReparseTag = 0;
- break;
- default:
- retval = -1;
- goto cleanup;
- }
- } else if (tagInfo.FileAttributes &
- FILE_ATTRIBUTE_REPARSE_POINT) {
- if (IsReparseTagNameSurrogate(tagInfo.ReparseTag)) {
- if (isUnhandledTag) {
- /* Traversing previously failed for either this link
- or its target. */
- SetLastError(ERROR_CANT_ACCESS_FILE);
- retval = -1;
- goto cleanup;
- }
- /* Traverse a non-link, but not if traversing already failed
- for an unhandled tag. */
- } else if (!isUnhandledTag) {
- CloseHandle(hFile);
- return win32_xstat_slow_impl(path, result, TRUE);
- }
- }
- }
- if (!GetFileInformationByHandle(hFile, &fileInfo) ||
- !GetFileInformationByHandleEx(hFile, FileBasicInfo,
- &basicInfo, sizeof(basicInfo))) {
- switch (GetLastError()) {
- case ERROR_INVALID_PARAMETER:
- case ERROR_INVALID_FUNCTION:
- case ERROR_NOT_SUPPORTED:
- /* Volumes and physical disks are block devices, e.g.
- \\.\C: and \\.\PhysicalDrive0. */
- memset(result, 0, sizeof(*result));
- result->st_mode = 0x6000; /* S_IFBLK */
- goto cleanup;
- }
- retval = -1;
- goto cleanup;
- }
- /* Successfully got FileBasicInfo, so we'll pass it along */
- pBasicInfo = &basicInfo;
- if (GetFileInformationByHandleEx(hFile, FileIdInfo, &idInfo, sizeof(idInfo))) {
- /* Successfully got FileIdInfo, so pass it along */
- pIdInfo = &idInfo;
- }
- }
- _Py_attribute_data_to_stat(&fileInfo, tagInfo.ReparseTag, pBasicInfo, pIdInfo, result);
- update_st_mode_from_path(path, fileInfo.dwFileAttributes, result);
- cleanup:
- if (hFile != INVALID_HANDLE_VALUE) {
- /* Preserve last error if we are failing */
- error = retval ? GetLastError() : 0;
- if (!CloseHandle(hFile)) {
- retval = -1;
- } else if (retval) {
- /* Restore last error */
- SetLastError(error);
- }
- }
- return retval;
- }
- static int
- win32_xstat_impl(const wchar_t *path, struct _Py_stat_struct *result,
- BOOL traverse)
- {
- FILE_STAT_BASIC_INFORMATION statInfo;
- if (_Py_GetFileInformationByName(path, FileStatBasicByNameInfo,
- &statInfo, sizeof(statInfo))) {
- if (// Cannot use fast path for reparse points ...
- !(statInfo.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
- // ... unless it's a name surrogate (symlink) and we're not following
- || (!traverse && IsReparseTagNameSurrogate(statInfo.ReparseTag))
- ) {
- _Py_stat_basic_info_to_stat(&statInfo, result);
- update_st_mode_from_path(path, statInfo.FileAttributes, result);
- return 0;
- }
- } else {
- switch(GetLastError()) {
- case ERROR_FILE_NOT_FOUND:
- case ERROR_PATH_NOT_FOUND:
- case ERROR_NOT_READY:
- case ERROR_BAD_NET_NAME:
- /* These errors aren't worth retrying with the slow path */
- return -1;
- case ERROR_NOT_SUPPORTED:
- /* indicates the API couldn't be loaded */
- break;
- }
- }
- return win32_xstat_slow_impl(path, result, traverse);
- }
- static int
- win32_xstat(const wchar_t *path, struct _Py_stat_struct *result, BOOL traverse)
- {
- /* Protocol violation: we explicitly clear errno, instead of
- setting it to a POSIX error. Callers should use GetLastError. */
- int code = win32_xstat_impl(path, result, traverse);
- errno = 0;
- /* ctime is only deprecated from 3.12, so we copy birthtime across */
- result->st_ctime = result->st_birthtime;
- result->st_ctime_nsec = result->st_birthtime_nsec;
- return code;
- }
- /* About the following functions: win32_lstat_w, win32_stat, win32_stat_w
- In Posix, stat automatically traverses symlinks and returns the stat
- structure for the target. In Windows, the equivalent GetFileAttributes by
- default does not traverse symlinks and instead returns attributes for
- the symlink.
- Instead, we will open the file (which *does* traverse symlinks by default)
- and GetFileInformationByHandle(). */
- static int
- win32_lstat(const wchar_t* path, struct _Py_stat_struct *result)
- {
- return win32_xstat(path, result, FALSE);
- }
- static int
- win32_stat(const wchar_t* path, struct _Py_stat_struct *result)
- {
- return win32_xstat(path, result, TRUE);
- }
- #endif /* MS_WINDOWS */
- PyDoc_STRVAR(stat_result__doc__,
- "stat_result: Result from stat, fstat, or lstat.\n\n\
- This object may be accessed either as a tuple of\n\
- (mode, ino, dev, nlink, uid, gid, size, atime, mtime, ctime)\n\
- or via the attributes st_mode, st_ino, st_dev, st_nlink, st_uid, and so on.\n\
- \n\
- Posix/windows: If your platform supports st_blksize, st_blocks, st_rdev,\n\
- or st_flags, they are available as attributes only.\n\
- \n\
- See os.stat for more information.");
- static PyStructSequence_Field stat_result_fields[] = {
- {"st_mode", "protection bits"},
- {"st_ino", "inode"},
- {"st_dev", "device"},
- {"st_nlink", "number of hard links"},
- {"st_uid", "user ID of owner"},
- {"st_gid", "group ID of owner"},
- {"st_size", "total size, in bytes"},
- /* The NULL is replaced with PyStructSequence_UnnamedField later. */
- {NULL, "integer time of last access"},
- {NULL, "integer time of last modification"},
- {NULL, "integer time of last change"},
- {"st_atime", "time of last access"},
- {"st_mtime", "time of last modification"},
- {"st_ctime", "time of last change"},
- {"st_atime_ns", "time of last access in nanoseconds"},
- {"st_mtime_ns", "time of last modification in nanoseconds"},
- {"st_ctime_ns", "time of last change in nanoseconds"},
- #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
- {"st_blksize", "blocksize for filesystem I/O"},
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
- {"st_blocks", "number of blocks allocated"},
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_RDEV
- {"st_rdev", "device type (if inode device)"},
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FLAGS
- {"st_flags", "user defined flags for file"},
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_GEN
- {"st_gen", "generation number"},
- #endif
- #if defined(HAVE_STRUCT_STAT_ST_BIRTHTIME) || defined(MS_WINDOWS)
- {"st_birthtime", "time of creation"},
- #endif
- #ifdef MS_WINDOWS
- {"st_birthtime_ns", "time of creation in nanoseconds"},
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FILE_ATTRIBUTES
- {"st_file_attributes", "Windows file attribute bits"},
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FSTYPE
- {"st_fstype", "Type of filesystem"},
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_REPARSE_TAG
- {"st_reparse_tag", "Windows reparse tag"},
- #endif
- {0}
- };
- #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
- #define ST_BLKSIZE_IDX 16
- #else
- #define ST_BLKSIZE_IDX 15
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
- #define ST_BLOCKS_IDX (ST_BLKSIZE_IDX+1)
- #else
- #define ST_BLOCKS_IDX ST_BLKSIZE_IDX
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_RDEV
- #define ST_RDEV_IDX (ST_BLOCKS_IDX+1)
- #else
- #define ST_RDEV_IDX ST_BLOCKS_IDX
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FLAGS
- #define ST_FLAGS_IDX (ST_RDEV_IDX+1)
- #else
- #define ST_FLAGS_IDX ST_RDEV_IDX
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_GEN
- #define ST_GEN_IDX (ST_FLAGS_IDX+1)
- #else
- #define ST_GEN_IDX ST_FLAGS_IDX
- #endif
- #if defined(HAVE_STRUCT_STAT_ST_BIRTHTIME) || defined(MS_WINDOWS)
- #define ST_BIRTHTIME_IDX (ST_GEN_IDX+1)
- #else
- #define ST_BIRTHTIME_IDX ST_GEN_IDX
- #endif
- #ifdef MS_WINDOWS
- #define ST_BIRTHTIME_NS_IDX (ST_BIRTHTIME_IDX+1)
- #else
- #define ST_BIRTHTIME_NS_IDX ST_BIRTHTIME_IDX
- #endif
- #if defined(HAVE_STRUCT_STAT_ST_FILE_ATTRIBUTES) || defined(MS_WINDOWS)
- #define ST_FILE_ATTRIBUTES_IDX (ST_BIRTHTIME_NS_IDX+1)
- #else
- #define ST_FILE_ATTRIBUTES_IDX ST_BIRTHTIME_NS_IDX
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FSTYPE
- #define ST_FSTYPE_IDX (ST_FILE_ATTRIBUTES_IDX+1)
- #else
- #define ST_FSTYPE_IDX ST_FILE_ATTRIBUTES_IDX
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_REPARSE_TAG
- #define ST_REPARSE_TAG_IDX (ST_FSTYPE_IDX+1)
- #else
- #define ST_REPARSE_TAG_IDX ST_FSTYPE_IDX
- #endif
- static PyStructSequence_Desc stat_result_desc = {
- "stat_result", /* name */
- stat_result__doc__, /* doc */
- stat_result_fields,
- 10
- };
- PyDoc_STRVAR(statvfs_result__doc__,
- "statvfs_result: Result from statvfs or fstatvfs.\n\n\
- This object may be accessed either as a tuple of\n\
- (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),\n\
- or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.\n\
- \n\
- See os.statvfs for more information.");
- static PyStructSequence_Field statvfs_result_fields[] = {
- {"f_bsize", },
- {"f_frsize", },
- {"f_blocks", },
- {"f_bfree", },
- {"f_bavail", },
- {"f_files", },
- {"f_ffree", },
- {"f_favail", },
- {"f_flag", },
- {"f_namemax",},
- {"f_fsid", },
- {0}
- };
- static PyStructSequence_Desc statvfs_result_desc = {
- "statvfs_result", /* name */
- statvfs_result__doc__, /* doc */
- statvfs_result_fields,
- 10
- };
- #if defined(HAVE_WAITID) && !defined(__APPLE__)
- PyDoc_STRVAR(waitid_result__doc__,
- "waitid_result: Result from waitid.\n\n\
- This object may be accessed either as a tuple of\n\
- (si_pid, si_uid, si_signo, si_status, si_code),\n\
- or via the attributes si_pid, si_uid, and so on.\n\
- \n\
- See os.waitid for more information.");
- static PyStructSequence_Field waitid_result_fields[] = {
- {"si_pid", },
- {"si_uid", },
- {"si_signo", },
- {"si_status", },
- {"si_code", },
- {0}
- };
- static PyStructSequence_Desc waitid_result_desc = {
- "waitid_result", /* name */
- waitid_result__doc__, /* doc */
- waitid_result_fields,
- 5
- };
- #endif
- static PyObject *
- statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
- {
- PyStructSequence *result;
- int i;
- // ht_module doesn't get set in PyStructSequence_NewType(),
- // so we can't use PyType_GetModule().
- PyObject *mod = PyImport_GetModule(MODNAME_OBJ);
- if (mod == NULL) {
- return NULL;
- }
- _posixstate *state = get_posix_state(mod);
- Py_DECREF(mod);
- if (state == NULL) {
- return NULL;
- }
- #define structseq_new state->statresult_new_orig
- result = (PyStructSequence*)structseq_new(type, args, kwds);
- if (!result)
- return NULL;
- /* If we have been initialized from a tuple,
- st_?time might be set to None. Initialize it
- from the int slots. */
- for (i = 7; i <= 9; i++) {
- if (result->ob_item[i+3] == Py_None) {
- Py_DECREF(Py_None);
- result->ob_item[i+3] = Py_NewRef(result->ob_item[i]);
- }
- }
- return (PyObject*)result;
- }
- static int
- _posix_clear(PyObject *module)
- {
- _posixstate *state = get_posix_state(module);
- Py_CLEAR(state->billion);
- Py_CLEAR(state->DirEntryType);
- Py_CLEAR(state->ScandirIteratorType);
- #if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM)
- Py_CLEAR(state->SchedParamType);
- #endif
- Py_CLEAR(state->StatResultType);
- Py_CLEAR(state->StatVFSResultType);
- Py_CLEAR(state->TerminalSizeType);
- Py_CLEAR(state->TimesResultType);
- Py_CLEAR(state->UnameResultType);
- #if defined(HAVE_WAITID) && !defined(__APPLE__)
- Py_CLEAR(state->WaitidResultType);
- #endif
- #if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
- Py_CLEAR(state->struct_rusage);
- #endif
- Py_CLEAR(state->st_mode);
- return 0;
- }
- static int
- _posix_traverse(PyObject *module, visitproc visit, void *arg)
- {
- _posixstate *state = get_posix_state(module);
- Py_VISIT(state->billion);
- Py_VISIT(state->DirEntryType);
- Py_VISIT(state->ScandirIteratorType);
- #if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM)
- Py_VISIT(state->SchedParamType);
- #endif
- Py_VISIT(state->StatResultType);
- Py_VISIT(state->StatVFSResultType);
- Py_VISIT(state->TerminalSizeType);
- Py_VISIT(state->TimesResultType);
- Py_VISIT(state->UnameResultType);
- #if defined(HAVE_WAITID) && !defined(__APPLE__)
- Py_VISIT(state->WaitidResultType);
- #endif
- #if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
- Py_VISIT(state->struct_rusage);
- #endif
- Py_VISIT(state->st_mode);
- return 0;
- }
- static void
- _posix_free(void *module)
- {
- _posix_clear((PyObject *)module);
- }
- static int
- fill_time(PyObject *module, PyObject *v, int s_index, int f_index, int ns_index, time_t sec, unsigned long nsec)
- {
- assert(!PyErr_Occurred());
- int res = -1;
- PyObject *s_in_ns = NULL;
- PyObject *ns_total = NULL;
- PyObject *float_s = NULL;
- PyObject *s = _PyLong_FromTime_t(sec);
- PyObject *ns_fractional = PyLong_FromUnsignedLong(nsec);
- if (!(s && ns_fractional)) {
- goto exit;
- }
- s_in_ns = PyNumber_Multiply(s, get_posix_state(module)->billion);
- if (!s_in_ns) {
- goto exit;
- }
- ns_total = PyNumber_Add(s_in_ns, ns_fractional);
- if (!ns_total)
- goto exit;
- float_s = PyFloat_FromDouble(sec + 1e-9*nsec);
- if (!float_s) {
- goto exit;
- }
- if (s_index >= 0) {
- PyStructSequence_SET_ITEM(v, s_index, s);
- s = NULL;
- }
- if (f_index >= 0) {
- PyStructSequence_SET_ITEM(v, f_index, float_s);
- float_s = NULL;
- }
- if (ns_index >= 0) {
- PyStructSequence_SET_ITEM(v, ns_index, ns_total);
- ns_total = NULL;
- }
- assert(!PyErr_Occurred());
- res = 0;
- exit:
- Py_XDECREF(s);
- Py_XDECREF(ns_fractional);
- Py_XDECREF(s_in_ns);
- Py_XDECREF(ns_total);
- Py_XDECREF(float_s);
- return res;
- }
- #ifdef MS_WINDOWS
- static PyObject*
- _pystat_l128_from_l64_l64(uint64_t low, uint64_t high)
- {
- PyObject *o_low = PyLong_FromUnsignedLongLong(low);
- if (!o_low || !high) {
- return o_low;
- }
- PyObject *o_high = PyLong_FromUnsignedLongLong(high);
- PyObject *l64 = o_high ? PyLong_FromLong(64) : NULL;
- if (!l64) {
- Py_XDECREF(o_high);
- Py_DECREF(o_low);
- return NULL;
- }
- Py_SETREF(o_high, PyNumber_Lshift(o_high, l64));
- Py_DECREF(l64);
- if (!o_high) {
- Py_DECREF(o_low);
- return NULL;
- }
- Py_SETREF(o_low, PyNumber_Add(o_low, o_high));
- Py_DECREF(o_high);
- return o_low;
- }
- #endif
- /* pack a system stat C structure into the Python stat tuple
- (used by posix_stat() and posix_fstat()) */
- static PyObject*
- _pystat_fromstructstat(PyObject *module, STRUCT_STAT *st)
- {
- assert(!PyErr_Occurred());
- PyObject *StatResultType = get_posix_state(module)->StatResultType;
- PyObject *v = PyStructSequence_New((PyTypeObject *)StatResultType);
- if (v == NULL) {
- return NULL;
- }
- #define SET_ITEM(pos, expr) \
- do { \
- PyObject *obj = (expr); \
- if (obj == NULL) { \
- goto error; \
- } \
- PyStructSequence_SET_ITEM(v, (pos), obj); \
- } while (0)
- SET_ITEM(0, PyLong_FromLong((long)st->st_mode));
- #ifdef MS_WINDOWS
- SET_ITEM(1, _pystat_l128_from_l64_l64(st->st_ino, st->st_ino_high));
- SET_ITEM(2, PyLong_FromUnsignedLongLong(st->st_dev));
- #else
- static_assert(sizeof(unsigned long long) >= sizeof(st->st_ino),
- "stat.st_ino is larger than unsigned long long");
- SET_ITEM(1, PyLong_FromUnsignedLongLong(st->st_ino));
- SET_ITEM(2, _PyLong_FromDev(st->st_dev));
- #endif
- SET_ITEM(3, PyLong_FromLong((long)st->st_nlink));
- #if defined(MS_WINDOWS)
- SET_ITEM(4, PyLong_FromLong(0));
- SET_ITEM(5, PyLong_FromLong(0));
- #else
- SET_ITEM(4, _PyLong_FromUid(st->st_uid));
- SET_ITEM(5, _PyLong_FromGid(st->st_gid));
- #endif
- static_assert(sizeof(long long) >= sizeof(st->st_size),
- "stat.st_size is larger than long long");
- SET_ITEM(6, PyLong_FromLongLong(st->st_size));
- // Set st_atime, st_mtime and st_ctime
- unsigned long ansec, mnsec, cnsec;
- #if defined(HAVE_STAT_TV_NSEC)
- ansec = st->st_atim.tv_nsec;
- mnsec = st->st_mtim.tv_nsec;
- cnsec = st->st_ctim.tv_nsec;
- #elif defined(HAVE_STAT_TV_NSEC2)
- ansec = st->st_atimespec.tv_nsec;
- mnsec = st->st_mtimespec.tv_nsec;
- cnsec = st->st_ctimespec.tv_nsec;
- #elif defined(HAVE_STAT_NSEC)
- ansec = st->st_atime_nsec;
- mnsec = st->st_mtime_nsec;
- cnsec = st->st_ctime_nsec;
- #else
- ansec = mnsec = cnsec = 0;
- #endif
- if (fill_time(module, v, 7, 10, 13, st->st_atime, ansec) < 0) {
- goto error;
- }
- if (fill_time(module, v, 8, 11, 14, st->st_mtime, mnsec) < 0) {
- goto error;
- }
- if (fill_time(module, v, 9, 12, 15, st->st_ctime, cnsec) < 0) {
- goto error;
- }
- #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
- SET_ITEM(ST_BLKSIZE_IDX, PyLong_FromLong((long)st->st_blksize));
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_BLOCKS
- SET_ITEM(ST_BLOCKS_IDX, PyLong_FromLong((long)st->st_blocks));
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_RDEV
- SET_ITEM(ST_RDEV_IDX, PyLong_FromLong((long)st->st_rdev));
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_GEN
- SET_ITEM(ST_GEN_IDX, PyLong_FromLong((long)st->st_gen));
- #endif
- #if defined(HAVE_STRUCT_STAT_ST_BIRTHTIME)
- {
- unsigned long bsec, bnsec;
- bsec = (long)st->st_birthtime;
- #ifdef HAVE_STAT_TV_NSEC2
- bnsec = st->st_birthtimespec.tv_nsec;
- #else
- bnsec = 0;
- #endif
- SET_ITEM(ST_BIRTHTIME_IDX, PyFloat_FromDouble(bsec + bnsec * 1e-9));
- }
- #elif defined(MS_WINDOWS)
- if (fill_time(module, v, -1, ST_BIRTHTIME_IDX, ST_BIRTHTIME_NS_IDX,
- st->st_birthtime, st->st_birthtime_nsec) < 0) {
- goto error;
- }
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FLAGS
- SET_ITEM(ST_FLAGS_IDX, PyLong_FromLong((long)st->st_flags));
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FILE_ATTRIBUTES
- SET_ITEM(ST_FILE_ATTRIBUTES_IDX,
- PyLong_FromUnsignedLong(st->st_file_attributes));
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_FSTYPE
- SET_ITEM(ST_FSTYPE_IDX, PyUnicode_FromString(st->st_fstype));
- #endif
- #ifdef HAVE_STRUCT_STAT_ST_REPARSE_TAG
- SET_ITEM(ST_REPARSE_TAG_IDX, PyLong_FromUnsignedLong(st->st_reparse_tag));
- #endif
- assert(!PyErr_Occurred());
- return v;
- error:
- Py_DECREF(v);
- return NULL;
- #undef SET_ITEM
- }
- /* POSIX methods */
- static PyObject *
- posix_do_stat(PyObject *module, const char *function_name, path_t *path,
- int dir_fd, int follow_symlinks)
- {
- STRUCT_STAT st;
- int result;
- #ifdef HAVE_FSTATAT
- int fstatat_unavailable = 0;
- #endif
- #if !defined(MS_WINDOWS) && !defined(HAVE_FSTATAT) && !defined(HAVE_LSTAT)
- if (follow_symlinks_specified(function_name, follow_symlinks))
- return NULL;
- #endif
- if (path_and_dir_fd_invalid("stat", path, dir_fd) ||
- dir_fd_and_fd_invalid("stat", dir_fd, path->fd) ||
- fd_and_follow_symlinks_invalid("stat", path->fd, follow_symlinks))
- return NULL;
- Py_BEGIN_ALLOW_THREADS
- if (path->fd != -1)
- result = FSTAT(path->fd, &st);
- #ifdef MS_WINDOWS
- else if (follow_symlinks)
- result = win32_stat(path->wide, &st);
- else
- result = win32_lstat(path->wide, &st);
- #else
- else
- #if defined(HAVE_LSTAT)
- if ((!follow_symlinks) && (dir_fd == DEFAULT_DIR_FD))
- result = LSTAT(path->narrow, &st);
- else
- #endif /* HAVE_LSTAT */
- #ifdef HAVE_FSTATAT
- if ((dir_fd != DEFAULT_DIR_FD) || !follow_symlinks) {
- if (HAVE_FSTATAT_RUNTIME) {
- result = fstatat(dir_fd, path->narrow, &st,
- follow_symlinks ? 0 : AT_SYMLINK_NOFOLLOW);
- } else {
- fstatat_unavailable = 1;
- }
- } else
- #endif /* HAVE_FSTATAT */
- result = STAT(path->narrow, &st);
- #endif /* MS_WINDOWS */
- Py_END_ALLOW_THREADS
- #ifdef HAVE_FSTATAT
- if (fstatat_unavailable) {
- argument_unavailable_error("stat", "dir_fd");
- return NULL;
- }
- #endif
- if (result != 0) {
- return path_error(path);
- }
- return _pystat_fromstructstat(module, &st);
- }
- /*[python input]
- for s in """
- FACCESSAT
- FCHMODAT
- FCHOWNAT
- FSTATAT
- LINKAT
- MKDIRAT
- MKFIFOAT
- MKNODAT
- OPENAT
- READLINKAT
- SYMLINKAT
- UNLINKAT
- """.strip().split():
- s = s.strip()
- print("""
- #ifdef HAVE_{s}
- #define {s}_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define {s}_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- """.rstrip().format(s=s))
- for s in """
- FCHDIR
- FCHMOD
- FCHOWN
- FDOPENDIR
- FEXECVE
- FPATHCONF
- FSTATVFS
- FTRUNCATE
- """.strip().split():
- s = s.strip()
- print("""
- #ifdef HAVE_{s}
- #define PATH_HAVE_{s} 1
- #else
- #define PATH_HAVE_{s} 0
- #endif
- """.rstrip().format(s=s))
- [python start generated code]*/
- #ifdef HAVE_FACCESSAT
- #define FACCESSAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define FACCESSAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_FCHMODAT
- #define FCHMODAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define FCHMODAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_FCHOWNAT
- #define FCHOWNAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define FCHOWNAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_FSTATAT
- #define FSTATAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define FSTATAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_LINKAT
- #define LINKAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define LINKAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_MKDIRAT
- #define MKDIRAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define MKDIRAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_MKFIFOAT
- #define MKFIFOAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define MKFIFOAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_MKNODAT
- #define MKNODAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define MKNODAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_OPENAT
- #define OPENAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define OPENAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_READLINKAT
- #define READLINKAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define READLINKAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_SYMLINKAT
- #define SYMLINKAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define SYMLINKAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_UNLINKAT
- #define UNLINKAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define UNLINKAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #ifdef HAVE_FCHDIR
- #define PATH_HAVE_FCHDIR 1
- #else
- #define PATH_HAVE_FCHDIR 0
- #endif
- #ifdef HAVE_FCHMOD
- #define PATH_HAVE_FCHMOD 1
- #else
- #define PATH_HAVE_FCHMOD 0
- #endif
- #ifdef HAVE_FCHOWN
- #define PATH_HAVE_FCHOWN 1
- #else
- #define PATH_HAVE_FCHOWN 0
- #endif
- #ifdef HAVE_FDOPENDIR
- #define PATH_HAVE_FDOPENDIR 1
- #else
- #define PATH_HAVE_FDOPENDIR 0
- #endif
- #ifdef HAVE_FEXECVE
- #define PATH_HAVE_FEXECVE 1
- #else
- #define PATH_HAVE_FEXECVE 0
- #endif
- #ifdef HAVE_FPATHCONF
- #define PATH_HAVE_FPATHCONF 1
- #else
- #define PATH_HAVE_FPATHCONF 0
- #endif
- #ifdef HAVE_FSTATVFS
- #define PATH_HAVE_FSTATVFS 1
- #else
- #define PATH_HAVE_FSTATVFS 0
- #endif
- #ifdef HAVE_FTRUNCATE
- #define PATH_HAVE_FTRUNCATE 1
- #else
- #define PATH_HAVE_FTRUNCATE 0
- #endif
- /*[python end generated code: output=4bd4f6f7d41267f1 input=80b4c890b6774ea5]*/
- #ifdef MS_WINDOWS
- #undef PATH_HAVE_FTRUNCATE
- #define PATH_HAVE_FTRUNCATE 1
- #endif
- /*[python input]
- class path_t_converter(CConverter):
- type = "path_t"
- impl_by_reference = True
- parse_by_reference = True
- converter = 'path_converter'
- def converter_init(self, *, allow_fd=False, make_wide=None,
- nonstrict=False, nullable=False,
- suppress_value_error=False):
- # right now path_t doesn't support default values.
- # to support a default value, you'll need to override initialize().
- if self.default not in (unspecified, None):
- fail("Can't specify a default to the path_t converter!")
- if self.c_default not in (None, 'Py_None'):
- raise RuntimeError("Can't specify a c_default to the path_t converter!")
- self.nullable = nullable
- self.nonstrict = nonstrict
- self.make_wide = make_wide
- self.suppress_value_error = suppress_value_error
- self.allow_fd = allow_fd
- def pre_render(self):
- def strify(value):
- if isinstance(value, str):
- return value
- return str(int(bool(value)))
- # add self.py_name here when merging with posixmodule conversion
- if self.make_wide is None:
- self.c_default = 'PATH_T_INITIALIZE_P("{}", "{}", {}, {}, {}, {})'.format(
- self.function.name,
- self.name,
- strify(self.nullable),
- strify(self.nonstrict),
- strify(self.suppress_value_error),
- strify(self.allow_fd),
- )
- else:
- self.c_default = 'PATH_T_INITIALIZE("{}", "{}", {}, {}, {}, {}, {})'.format(
- self.function.name,
- self.name,
- strify(self.nullable),
- strify(self.nonstrict),
- strify(self.make_wide),
- strify(self.suppress_value_error),
- strify(self.allow_fd),
- )
- def cleanup(self):
- return "path_cleanup(&" + self.name + ");\n"
- class dir_fd_converter(CConverter):
- type = 'int'
- def converter_init(self, requires=None):
- if self.default in (unspecified, None):
- self.c_default = 'DEFAULT_DIR_FD'
- if isinstance(requires, str):
- self.converter = requires.upper() + '_DIR_FD_CONVERTER'
- else:
- self.converter = 'dir_fd_converter'
- class uid_t_converter(CConverter):
- type = "uid_t"
- converter = '_Py_Uid_Converter'
- class gid_t_converter(CConverter):
- type = "gid_t"
- converter = '_Py_Gid_Converter'
- class dev_t_converter(CConverter):
- type = 'dev_t'
- converter = '_Py_Dev_Converter'
- class dev_t_return_converter(unsigned_long_return_converter):
- type = 'dev_t'
- conversion_fn = '_PyLong_FromDev'
- unsigned_cast = '(dev_t)'
- class FSConverter_converter(CConverter):
- type = 'PyObject *'
- converter = 'PyUnicode_FSConverter'
- def converter_init(self):
- if self.default is not unspecified:
- fail("FSConverter_converter does not support default values")
- self.c_default = 'NULL'
- def cleanup(self):
- return "Py_XDECREF(" + self.name + ");\n"
- class pid_t_converter(CConverter):
- type = 'pid_t'
- format_unit = '" _Py_PARSE_PID "'
- class idtype_t_converter(int_converter):
- type = 'idtype_t'
- class id_t_converter(CConverter):
- type = 'id_t'
- format_unit = '" _Py_PARSE_PID "'
- class intptr_t_converter(CConverter):
- type = 'intptr_t'
- format_unit = '" _Py_PARSE_INTPTR "'
- class Py_off_t_converter(CConverter):
- type = 'Py_off_t'
- converter = 'Py_off_t_converter'
- class Py_off_t_return_converter(long_return_converter):
- type = 'Py_off_t'
- conversion_fn = 'PyLong_FromPy_off_t'
- class path_confname_converter(CConverter):
- type="int"
- converter="conv_path_confname"
- class confstr_confname_converter(path_confname_converter):
- converter='conv_confstr_confname'
- class sysconf_confname_converter(path_confname_converter):
- converter="conv_sysconf_confname"
- [python start generated code]*/
- /*[python end generated code: output=da39a3ee5e6b4b0d input=577cb476e5d64960]*/
- /*[clinic input]
- os.stat
- path : path_t(allow_fd=True)
- Path to be examined; can be string, bytes, a path-like object or
- open-file-descriptor int.
- *
- dir_fd : dir_fd(requires='fstatat') = None
- If not None, it should be a file descriptor open to a directory,
- and path should be a relative string; path will then be relative to
- that directory.
- follow_symlinks: bool = True
- If False, and the last element of the path is a symbolic link,
- stat will examine the symbolic link itself instead of the file
- the link points to.
- Perform a stat system call on the given path.
- dir_fd and follow_symlinks may not be implemented
- on your platform. If they are unavailable, using them will raise a
- NotImplementedError.
- It's an error to use dir_fd or follow_symlinks when specifying path as
- an open file descriptor.
- [clinic start generated code]*/
- static PyObject *
- os_stat_impl(PyObject *module, path_t *path, int dir_fd, int follow_symlinks)
- /*[clinic end generated code: output=7d4976e6f18a59c5 input=01d362ebcc06996b]*/
- {
- return posix_do_stat(module, "stat", path, dir_fd, follow_symlinks);
- }
- /*[clinic input]
- os.lstat
- path : path_t
- *
- dir_fd : dir_fd(requires='fstatat') = None
- Perform a stat system call on the given path, without following symbolic links.
- Like stat(), but do not follow symbolic links.
- Equivalent to stat(path, follow_symlinks=False).
- [clinic start generated code]*/
- static PyObject *
- os_lstat_impl(PyObject *module, path_t *path, int dir_fd)
- /*[clinic end generated code: output=ef82a5d35ce8ab37 input=0b7474765927b925]*/
- {
- int follow_symlinks = 0;
- return posix_do_stat(module, "lstat", path, dir_fd, follow_symlinks);
- }
- /*[clinic input]
- os.access -> bool
- path: path_t
- Path to be tested; can be string, bytes, or a path-like object.
- mode: int
- Operating-system mode bitfield. Can be F_OK to test existence,
- or the inclusive-OR of R_OK, W_OK, and X_OK.
- *
- dir_fd : dir_fd(requires='faccessat') = None
- If not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that
- directory.
- effective_ids: bool = False
- If True, access will use the effective uid/gid instead of
- the real uid/gid.
- follow_symlinks: bool = True
- If False, and the last element of the path is a symbolic link,
- access will examine the symbolic link itself instead of the file
- the link points to.
- Use the real uid/gid to test for access to a path.
- {parameters}
- dir_fd, effective_ids, and follow_symlinks may not be implemented
- on your platform. If they are unavailable, using them will raise a
- NotImplementedError.
- Note that most operations will use the effective uid/gid, therefore this
- routine can be used in a suid/sgid environment to test if the invoking user
- has the specified access to the path.
- [clinic start generated code]*/
- static int
- os_access_impl(PyObject *module, path_t *path, int mode, int dir_fd,
- int effective_ids, int follow_symlinks)
- /*[clinic end generated code: output=cf84158bc90b1a77 input=3ffe4e650ee3bf20]*/
- {
- int return_value;
- #ifdef MS_WINDOWS
- DWORD attr;
- #else
- int result;
- #endif
- #ifdef HAVE_FACCESSAT
- int faccessat_unavailable = 0;
- #endif
- #ifndef HAVE_FACCESSAT
- if (follow_symlinks_specified("access", follow_symlinks))
- return -1;
- if (effective_ids) {
- argument_unavailable_error("access", "effective_ids");
- return -1;
- }
- #endif
- #ifdef MS_WINDOWS
- Py_BEGIN_ALLOW_THREADS
- attr = GetFileAttributesW(path->wide);
- Py_END_ALLOW_THREADS
- /*
- * Access is possible if
- * * we didn't get a -1, and
- * * write access wasn't requested,
- * * or the file isn't read-only,
- * * or it's a directory.
- * (Directories cannot be read-only on Windows.)
- */
- return_value = (attr != INVALID_FILE_ATTRIBUTES) &&
- (!(mode & 2) ||
- !(attr & FILE_ATTRIBUTE_READONLY) ||
- (attr & FILE_ATTRIBUTE_DIRECTORY));
- #else
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_FACCESSAT
- if ((dir_fd != DEFAULT_DIR_FD) ||
- effective_ids ||
- !follow_symlinks) {
- if (HAVE_FACCESSAT_RUNTIME) {
- int flags = 0;
- if (!follow_symlinks)
- flags |= AT_SYMLINK_NOFOLLOW;
- if (effective_ids)
- flags |= AT_EACCESS;
- result = faccessat(dir_fd, path->narrow, mode, flags);
- } else {
- faccessat_unavailable = 1;
- }
- }
- else
- #endif
- result = access(path->narrow, mode);
- Py_END_ALLOW_THREADS
- #ifdef HAVE_FACCESSAT
- if (faccessat_unavailable) {
- if (dir_fd != DEFAULT_DIR_FD) {
- argument_unavailable_error("access", "dir_fd");
- return -1;
- }
- if (follow_symlinks_specified("access", follow_symlinks))
- return -1;
- if (effective_ids) {
- argument_unavailable_error("access", "effective_ids");
- return -1;
- }
- /* should be unreachable */
- return -1;
- }
- #endif
- return_value = !result;
- #endif
- return return_value;
- }
- #ifndef F_OK
- #define F_OK 0
- #endif
- #ifndef R_OK
- #define R_OK 4
- #endif
- #ifndef W_OK
- #define W_OK 2
- #endif
- #ifndef X_OK
- #define X_OK 1
- #endif
- #ifdef HAVE_TTYNAME
- /*[clinic input]
- os.ttyname
- fd: int
- Integer file descriptor handle.
- /
- Return the name of the terminal device connected to 'fd'.
- [clinic start generated code]*/
- static PyObject *
- os_ttyname_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=c424d2e9d1cd636a input=9ff5a58b08115c55]*/
- {
- long size = sysconf(_SC_TTY_NAME_MAX);
- if (size == -1) {
- return posix_error();
- }
- char *buffer = (char *)PyMem_RawMalloc(size);
- if (buffer == NULL) {
- return PyErr_NoMemory();
- }
- int ret = ttyname_r(fd, buffer, size);
- if (ret != 0) {
- PyMem_RawFree(buffer);
- errno = ret;
- return posix_error();
- }
- PyObject *res = PyUnicode_DecodeFSDefault(buffer);
- PyMem_RawFree(buffer);
- return res;
- }
- #endif
- #ifdef HAVE_CTERMID
- /*[clinic input]
- os.ctermid
- Return the name of the controlling terminal for this process.
- [clinic start generated code]*/
- static PyObject *
- os_ctermid_impl(PyObject *module)
- /*[clinic end generated code: output=02f017e6c9e620db input=3b87fdd52556382d]*/
- {
- char *ret;
- char buffer[L_ctermid];
- #ifdef USE_CTERMID_R
- ret = ctermid_r(buffer);
- #else
- ret = ctermid(buffer);
- #endif
- if (ret == NULL)
- return posix_error();
- return PyUnicode_DecodeFSDefault(buffer);
- }
- #endif /* HAVE_CTERMID */
- /*[clinic input]
- os.chdir
- path: path_t(allow_fd='PATH_HAVE_FCHDIR')
- Change the current working directory to the specified path.
- path may always be specified as a string.
- On some platforms, path may also be specified as an open file descriptor.
- If this functionality is unavailable, using it raises an exception.
- [clinic start generated code]*/
- static PyObject *
- os_chdir_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=3be6400eee26eaae input=1a4a15b4d12cb15d]*/
- {
- int result;
- if (PySys_Audit("os.chdir", "(O)", path->object) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- #ifdef MS_WINDOWS
- /* on unix, success = 0, on windows, success = !0 */
- result = !win32_wchdir(path->wide);
- #else
- #ifdef HAVE_FCHDIR
- if (path->fd != -1)
- result = fchdir(path->fd);
- else
- #endif
- result = chdir(path->narrow);
- #endif
- Py_END_ALLOW_THREADS
- if (result) {
- return path_error(path);
- }
- Py_RETURN_NONE;
- }
- #ifdef HAVE_FCHDIR
- /*[clinic input]
- os.fchdir
- fd: fildes
- Change to the directory of the given file descriptor.
- fd must be opened on a directory, not a file.
- Equivalent to os.chdir(fd).
- [clinic start generated code]*/
- static PyObject *
- os_fchdir_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=42e064ec4dc00ab0 input=18e816479a2fa985]*/
- {
- if (PySys_Audit("os.chdir", "(i)", fd) < 0) {
- return NULL;
- }
- return posix_fildes_fd(fd, fchdir);
- }
- #endif /* HAVE_FCHDIR */
- /*[clinic input]
- os.chmod
- path: path_t(allow_fd='PATH_HAVE_FCHMOD')
- Path to be modified. May always be specified as a str, bytes, or a path-like object.
- On some platforms, path may also be specified as an open file descriptor.
- If this functionality is unavailable, using it raises an exception.
- mode: int
- Operating-system mode bitfield.
- Be careful when using number literals for *mode*. The conventional UNIX notation for
- numeric modes uses an octal base, which needs to be indicated with a ``0o`` prefix in
- Python.
- *
- dir_fd : dir_fd(requires='fchmodat') = None
- If not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that
- directory.
- follow_symlinks: bool = True
- If False, and the last element of the path is a symbolic link,
- chmod will modify the symbolic link itself instead of the file
- the link points to.
- Change the access permissions of a file.
- It is an error to use dir_fd or follow_symlinks when specifying path as
- an open file descriptor.
- dir_fd and follow_symlinks may not be implemented on your platform.
- If they are unavailable, using them will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_chmod_impl(PyObject *module, path_t *path, int mode, int dir_fd,
- int follow_symlinks)
- /*[clinic end generated code: output=5cf6a94915cc7bff input=674a14bc998de09d]*/
- {
- int result;
- #ifdef MS_WINDOWS
- DWORD attr;
- #endif
- #ifdef HAVE_FCHMODAT
- int fchmodat_nofollow_unsupported = 0;
- int fchmodat_unsupported = 0;
- #endif
- #if !(defined(HAVE_FCHMODAT) || defined(HAVE_LCHMOD))
- if (follow_symlinks_specified("chmod", follow_symlinks))
- return NULL;
- #endif
- if (PySys_Audit("os.chmod", "Oii", path->object, mode,
- dir_fd == DEFAULT_DIR_FD ? -1 : dir_fd) < 0) {
- return NULL;
- }
- #ifdef MS_WINDOWS
- Py_BEGIN_ALLOW_THREADS
- attr = GetFileAttributesW(path->wide);
- if (attr == INVALID_FILE_ATTRIBUTES)
- result = 0;
- else {
- if (mode & _S_IWRITE)
- attr &= ~FILE_ATTRIBUTE_READONLY;
- else
- attr |= FILE_ATTRIBUTE_READONLY;
- result = SetFileAttributesW(path->wide, attr);
- }
- Py_END_ALLOW_THREADS
- if (!result) {
- return path_error(path);
- }
- #else /* MS_WINDOWS */
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_FCHMOD
- if (path->fd != -1)
- result = fchmod(path->fd, mode);
- else
- #endif /* HAVE_CHMOD */
- #ifdef HAVE_LCHMOD
- if ((!follow_symlinks) && (dir_fd == DEFAULT_DIR_FD))
- result = lchmod(path->narrow, mode);
- else
- #endif /* HAVE_LCHMOD */
- #ifdef HAVE_FCHMODAT
- if ((dir_fd != DEFAULT_DIR_FD) || !follow_symlinks) {
- if (HAVE_FCHMODAT_RUNTIME) {
- /*
- * fchmodat() doesn't currently support AT_SYMLINK_NOFOLLOW!
- * The documentation specifically shows how to use it,
- * and then says it isn't implemented yet.
- * (true on linux with glibc 2.15, and openindiana 3.x)
- *
- * Once it is supported, os.chmod will automatically
- * support dir_fd and follow_symlinks=False. (Hopefully.)
- * Until then, we need to be careful what exception we raise.
- */
- result = fchmodat(dir_fd, path->narrow, mode,
- follow_symlinks ? 0 : AT_SYMLINK_NOFOLLOW);
- /*
- * But wait! We can't throw the exception without allowing threads,
- * and we can't do that in this nested scope. (Macro trickery, sigh.)
- */
- fchmodat_nofollow_unsupported =
- result &&
- ((errno == ENOTSUP) || (errno == EOPNOTSUPP)) &&
- !follow_symlinks;
- } else {
- fchmodat_unsupported = 1;
- fchmodat_nofollow_unsupported = 1;
- result = -1;
- }
- }
- else
- #endif /* HAVE_FHCMODAT */
- {
- #ifdef HAVE_CHMOD
- result = chmod(path->narrow, mode);
- #elif defined(__wasi__)
- // WASI SDK 15.0 does not support chmod.
- // Ignore missing syscall for now.
- result = 0;
- #else
- result = -1;
- errno = ENOSYS;
- #endif
- }
- Py_END_ALLOW_THREADS
- if (result) {
- #ifdef HAVE_FCHMODAT
- if (fchmodat_unsupported) {
- if (dir_fd != DEFAULT_DIR_FD) {
- argument_unavailable_error("chmod", "dir_fd");
- return NULL;
- }
- }
- if (fchmodat_nofollow_unsupported) {
- if (dir_fd != DEFAULT_DIR_FD)
- dir_fd_and_follow_symlinks_invalid("chmod",
- dir_fd, follow_symlinks);
- else
- follow_symlinks_specified("chmod", follow_symlinks);
- return NULL;
- }
- else
- #endif /* HAVE_FCHMODAT */
- return path_error(path);
- }
- #endif /* MS_WINDOWS */
- Py_RETURN_NONE;
- }
- #ifdef HAVE_FCHMOD
- /*[clinic input]
- os.fchmod
- fd: int
- The file descriptor of the file to be modified.
- mode: int
- Operating-system mode bitfield.
- Be careful when using number literals for *mode*. The conventional UNIX notation for
- numeric modes uses an octal base, which needs to be indicated with a ``0o`` prefix in
- Python.
- Change the access permissions of the file given by file descriptor fd.
- Equivalent to os.chmod(fd, mode).
- [clinic start generated code]*/
- static PyObject *
- os_fchmod_impl(PyObject *module, int fd, int mode)
- /*[clinic end generated code: output=afd9bc05b4e426b3 input=b5594618bbbc22df]*/
- {
- int res;
- int async_err = 0;
- if (PySys_Audit("os.chmod", "iii", fd, mode, -1) < 0) {
- return NULL;
- }
- do {
- Py_BEGIN_ALLOW_THREADS
- res = fchmod(fd, mode);
- Py_END_ALLOW_THREADS
- } while (res != 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res != 0)
- return (!async_err) ? posix_error() : NULL;
- Py_RETURN_NONE;
- }
- #endif /* HAVE_FCHMOD */
- #ifdef HAVE_LCHMOD
- /*[clinic input]
- os.lchmod
- path: path_t
- mode: int
- Change the access permissions of a file, without following symbolic links.
- If path is a symlink, this affects the link itself rather than the target.
- Equivalent to chmod(path, mode, follow_symlinks=False)."
- [clinic start generated code]*/
- static PyObject *
- os_lchmod_impl(PyObject *module, path_t *path, int mode)
- /*[clinic end generated code: output=082344022b51a1d5 input=90c5663c7465d24f]*/
- {
- int res;
- if (PySys_Audit("os.chmod", "Oii", path->object, mode, -1) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- res = lchmod(path->narrow, mode);
- Py_END_ALLOW_THREADS
- if (res < 0) {
- path_error(path);
- return NULL;
- }
- Py_RETURN_NONE;
- }
- #endif /* HAVE_LCHMOD */
- #ifdef HAVE_CHFLAGS
- /*[clinic input]
- os.chflags
- path: path_t
- flags: unsigned_long(bitwise=True)
- follow_symlinks: bool=True
- Set file flags.
- If follow_symlinks is False, and the last element of the path is a symbolic
- link, chflags will change flags on the symbolic link itself instead of the
- file the link points to.
- follow_symlinks may not be implemented on your platform. If it is
- unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_chflags_impl(PyObject *module, path_t *path, unsigned long flags,
- int follow_symlinks)
- /*[clinic end generated code: output=85571c6737661ce9 input=0327e29feb876236]*/
- {
- int result;
- #ifndef HAVE_LCHFLAGS
- if (follow_symlinks_specified("chflags", follow_symlinks))
- return NULL;
- #endif
- if (PySys_Audit("os.chflags", "Ok", path->object, flags) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_LCHFLAGS
- if (!follow_symlinks)
- result = lchflags(path->narrow, flags);
- else
- #endif
- result = chflags(path->narrow, flags);
- Py_END_ALLOW_THREADS
- if (result)
- return path_error(path);
- Py_RETURN_NONE;
- }
- #endif /* HAVE_CHFLAGS */
- #ifdef HAVE_LCHFLAGS
- /*[clinic input]
- os.lchflags
- path: path_t
- flags: unsigned_long(bitwise=True)
- Set file flags.
- This function will not follow symbolic links.
- Equivalent to chflags(path, flags, follow_symlinks=False).
- [clinic start generated code]*/
- static PyObject *
- os_lchflags_impl(PyObject *module, path_t *path, unsigned long flags)
- /*[clinic end generated code: output=30ae958695c07316 input=f9f82ea8b585ca9d]*/
- {
- int res;
- if (PySys_Audit("os.chflags", "Ok", path->object, flags) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- res = lchflags(path->narrow, flags);
- Py_END_ALLOW_THREADS
- if (res < 0) {
- return path_error(path);
- }
- Py_RETURN_NONE;
- }
- #endif /* HAVE_LCHFLAGS */
- #ifdef HAVE_CHROOT
- /*[clinic input]
- os.chroot
- path: path_t
- Change root directory to path.
- [clinic start generated code]*/
- static PyObject *
- os_chroot_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=de80befc763a4475 input=14822965652c3dc3]*/
- {
- int res;
- Py_BEGIN_ALLOW_THREADS
- res = chroot(path->narrow);
- Py_END_ALLOW_THREADS
- if (res < 0)
- return path_error(path);
- Py_RETURN_NONE;
- }
- #endif /* HAVE_CHROOT */
- #ifdef HAVE_FSYNC
- /*[clinic input]
- os.fsync
- fd: fildes
- Force write of fd to disk.
- [clinic start generated code]*/
- static PyObject *
- os_fsync_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=4a10d773f52b3584 input=21c3645c056967f2]*/
- {
- return posix_fildes_fd(fd, fsync);
- }
- #endif /* HAVE_FSYNC */
- #ifdef HAVE_SYNC
- /*[clinic input]
- os.sync
- Force write of everything to disk.
- [clinic start generated code]*/
- static PyObject *
- os_sync_impl(PyObject *module)
- /*[clinic end generated code: output=2796b1f0818cd71c input=84749fe5e9b404ff]*/
- {
- Py_BEGIN_ALLOW_THREADS
- sync();
- Py_END_ALLOW_THREADS
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SYNC */
- #ifdef HAVE_FDATASYNC
- #ifdef __hpux
- extern int fdatasync(int); /* On HP-UX, in libc but not in unistd.h */
- #endif
- /*[clinic input]
- os.fdatasync
- fd: fildes
- Force write of fd to disk without forcing update of metadata.
- [clinic start generated code]*/
- static PyObject *
- os_fdatasync_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=b4b9698b5d7e26dd input=bc74791ee54dd291]*/
- {
- return posix_fildes_fd(fd, fdatasync);
- }
- #endif /* HAVE_FDATASYNC */
- #ifdef HAVE_CHOWN
- /*[clinic input]
- os.chown
- path : path_t(allow_fd='PATH_HAVE_FCHOWN')
- Path to be examined; can be string, bytes, a path-like object, or open-file-descriptor int.
- uid: uid_t
- gid: gid_t
- *
- dir_fd : dir_fd(requires='fchownat') = None
- If not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that
- directory.
- follow_symlinks: bool = True
- If False, and the last element of the path is a symbolic link,
- stat will examine the symbolic link itself instead of the file
- the link points to.
- Change the owner and group id of path to the numeric uid and gid.\
- path may always be specified as a string.
- On some platforms, path may also be specified as an open file descriptor.
- If this functionality is unavailable, using it raises an exception.
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- If follow_symlinks is False, and the last element of the path is a symbolic
- link, chown will modify the symbolic link itself instead of the file the
- link points to.
- It is an error to use dir_fd or follow_symlinks when specifying path as
- an open file descriptor.
- dir_fd and follow_symlinks may not be implemented on your platform.
- If they are unavailable, using them will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_chown_impl(PyObject *module, path_t *path, uid_t uid, gid_t gid,
- int dir_fd, int follow_symlinks)
- /*[clinic end generated code: output=4beadab0db5f70cd input=b08c5ec67996a97d]*/
- {
- int result;
- #if defined(HAVE_FCHOWNAT)
- int fchownat_unsupported = 0;
- #endif
- #if !(defined(HAVE_LCHOWN) || defined(HAVE_FCHOWNAT))
- if (follow_symlinks_specified("chown", follow_symlinks))
- return NULL;
- #endif
- if (dir_fd_and_fd_invalid("chown", dir_fd, path->fd) ||
- fd_and_follow_symlinks_invalid("chown", path->fd, follow_symlinks))
- return NULL;
- if (PySys_Audit("os.chown", "OIIi", path->object, uid, gid,
- dir_fd == DEFAULT_DIR_FD ? -1 : dir_fd) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_FCHOWN
- if (path->fd != -1)
- result = fchown(path->fd, uid, gid);
- else
- #endif
- #ifdef HAVE_LCHOWN
- if ((!follow_symlinks) && (dir_fd == DEFAULT_DIR_FD))
- result = lchown(path->narrow, uid, gid);
- else
- #endif
- #ifdef HAVE_FCHOWNAT
- if ((dir_fd != DEFAULT_DIR_FD) || (!follow_symlinks)) {
- if (HAVE_FCHOWNAT_RUNTIME) {
- result = fchownat(dir_fd, path->narrow, uid, gid,
- follow_symlinks ? 0 : AT_SYMLINK_NOFOLLOW);
- } else {
- fchownat_unsupported = 1;
- }
- } else
- #endif
- result = chown(path->narrow, uid, gid);
- Py_END_ALLOW_THREADS
- #ifdef HAVE_FCHOWNAT
- if (fchownat_unsupported) {
- /* This would be incorrect if the current platform
- * doesn't support lchown.
- */
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (result)
- return path_error(path);
- Py_RETURN_NONE;
- }
- #endif /* HAVE_CHOWN */
- #ifdef HAVE_FCHOWN
- /*[clinic input]
- os.fchown
- fd: int
- uid: uid_t
- gid: gid_t
- Change the owner and group id of the file specified by file descriptor.
- Equivalent to os.chown(fd, uid, gid).
- [clinic start generated code]*/
- static PyObject *
- os_fchown_impl(PyObject *module, int fd, uid_t uid, gid_t gid)
- /*[clinic end generated code: output=97d21cbd5a4350a6 input=3af544ba1b13a0d7]*/
- {
- int res;
- int async_err = 0;
- if (PySys_Audit("os.chown", "iIIi", fd, uid, gid, -1) < 0) {
- return NULL;
- }
- do {
- Py_BEGIN_ALLOW_THREADS
- res = fchown(fd, uid, gid);
- Py_END_ALLOW_THREADS
- } while (res != 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res != 0)
- return (!async_err) ? posix_error() : NULL;
- Py_RETURN_NONE;
- }
- #endif /* HAVE_FCHOWN */
- #ifdef HAVE_LCHOWN
- /*[clinic input]
- os.lchown
- path : path_t
- uid: uid_t
- gid: gid_t
- Change the owner and group id of path to the numeric uid and gid.
- This function will not follow symbolic links.
- Equivalent to os.chown(path, uid, gid, follow_symlinks=False).
- [clinic start generated code]*/
- static PyObject *
- os_lchown_impl(PyObject *module, path_t *path, uid_t uid, gid_t gid)
- /*[clinic end generated code: output=25eaf6af412fdf2f input=b1c6014d563a7161]*/
- {
- int res;
- if (PySys_Audit("os.chown", "OIIi", path->object, uid, gid, -1) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- res = lchown(path->narrow, uid, gid);
- Py_END_ALLOW_THREADS
- if (res < 0) {
- return path_error(path);
- }
- Py_RETURN_NONE;
- }
- #endif /* HAVE_LCHOWN */
- static PyObject *
- posix_getcwd(int use_bytes)
- {
- #ifdef MS_WINDOWS
- wchar_t wbuf[MAXPATHLEN];
- wchar_t *wbuf2 = wbuf;
- DWORD len;
- Py_BEGIN_ALLOW_THREADS
- len = GetCurrentDirectoryW(Py_ARRAY_LENGTH(wbuf), wbuf);
- /* If the buffer is large enough, len does not include the
- terminating \0. If the buffer is too small, len includes
- the space needed for the terminator. */
- if (len >= Py_ARRAY_LENGTH(wbuf)) {
- if (len <= PY_SSIZE_T_MAX / sizeof(wchar_t)) {
- wbuf2 = PyMem_RawMalloc(len * sizeof(wchar_t));
- }
- else {
- wbuf2 = NULL;
- }
- if (wbuf2) {
- len = GetCurrentDirectoryW(len, wbuf2);
- }
- }
- Py_END_ALLOW_THREADS
- if (!wbuf2) {
- PyErr_NoMemory();
- return NULL;
- }
- if (!len) {
- PyErr_SetFromWindowsErr(0);
- if (wbuf2 != wbuf)
- PyMem_RawFree(wbuf2);
- return NULL;
- }
- PyObject *resobj = PyUnicode_FromWideChar(wbuf2, len);
- if (wbuf2 != wbuf) {
- PyMem_RawFree(wbuf2);
- }
- if (use_bytes) {
- if (resobj == NULL) {
- return NULL;
- }
- Py_SETREF(resobj, PyUnicode_EncodeFSDefault(resobj));
- }
- return resobj;
- #else
- const size_t chunk = 1024;
- char *buf = NULL;
- char *cwd = NULL;
- size_t buflen = 0;
- Py_BEGIN_ALLOW_THREADS
- do {
- char *newbuf;
- if (buflen <= PY_SSIZE_T_MAX - chunk) {
- buflen += chunk;
- newbuf = PyMem_RawRealloc(buf, buflen);
- }
- else {
- newbuf = NULL;
- }
- if (newbuf == NULL) {
- PyMem_RawFree(buf);
- buf = NULL;
- break;
- }
- buf = newbuf;
- cwd = getcwd(buf, buflen);
- } while (cwd == NULL && errno == ERANGE);
- Py_END_ALLOW_THREADS
- if (buf == NULL) {
- return PyErr_NoMemory();
- }
- if (cwd == NULL) {
- posix_error();
- PyMem_RawFree(buf);
- return NULL;
- }
- PyObject *obj;
- if (use_bytes) {
- obj = PyBytes_FromStringAndSize(buf, strlen(buf));
- }
- else {
- obj = PyUnicode_DecodeFSDefault(buf);
- }
- PyMem_RawFree(buf);
- return obj;
- #endif /* !MS_WINDOWS */
- }
- /*[clinic input]
- os.getcwd
- Return a unicode string representing the current working directory.
- [clinic start generated code]*/
- static PyObject *
- os_getcwd_impl(PyObject *module)
- /*[clinic end generated code: output=21badfae2ea99ddc input=f069211bb70e3d39]*/
- {
- return posix_getcwd(0);
- }
- /*[clinic input]
- os.getcwdb
- Return a bytes string representing the current working directory.
- [clinic start generated code]*/
- static PyObject *
- os_getcwdb_impl(PyObject *module)
- /*[clinic end generated code: output=3dd47909480e4824 input=f6f6a378dad3d9cb]*/
- {
- return posix_getcwd(1);
- }
- #if ((!defined(HAVE_LINK)) && defined(MS_WINDOWS))
- #define HAVE_LINK 1
- #endif
- #ifdef HAVE_LINK
- /*[clinic input]
- os.link
- src : path_t
- dst : path_t
- *
- src_dir_fd : dir_fd = None
- dst_dir_fd : dir_fd = None
- follow_symlinks: bool = True
- Create a hard link to a file.
- If either src_dir_fd or dst_dir_fd is not None, it should be a file
- descriptor open to a directory, and the respective path string (src or dst)
- should be relative; the path will then be relative to that directory.
- If follow_symlinks is False, and the last element of src is a symbolic
- link, link will create a link to the symbolic link itself instead of the
- file the link points to.
- src_dir_fd, dst_dir_fd, and follow_symlinks may not be implemented on your
- platform. If they are unavailable, using them will raise a
- NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_link_impl(PyObject *module, path_t *src, path_t *dst, int src_dir_fd,
- int dst_dir_fd, int follow_symlinks)
- /*[clinic end generated code: output=7f00f6007fd5269a input=b0095ebbcbaa7e04]*/
- {
- #ifdef MS_WINDOWS
- BOOL result = FALSE;
- #else
- int result;
- #endif
- #if defined(HAVE_LINKAT)
- int linkat_unavailable = 0;
- #endif
- #ifndef HAVE_LINKAT
- if ((src_dir_fd != DEFAULT_DIR_FD) || (dst_dir_fd != DEFAULT_DIR_FD)) {
- argument_unavailable_error("link", "src_dir_fd and dst_dir_fd");
- return NULL;
- }
- #endif
- #ifndef MS_WINDOWS
- if ((src->narrow && dst->wide) || (src->wide && dst->narrow)) {
- PyErr_SetString(PyExc_NotImplementedError,
- "link: src and dst must be the same type");
- return NULL;
- }
- #endif
- if (PySys_Audit("os.link", "OOii", src->object, dst->object,
- src_dir_fd == DEFAULT_DIR_FD ? -1 : src_dir_fd,
- dst_dir_fd == DEFAULT_DIR_FD ? -1 : dst_dir_fd) < 0) {
- return NULL;
- }
- #ifdef MS_WINDOWS
- Py_BEGIN_ALLOW_THREADS
- result = CreateHardLinkW(dst->wide, src->wide, NULL);
- Py_END_ALLOW_THREADS
- if (!result)
- return path_error2(src, dst);
- #else
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_LINKAT
- if ((src_dir_fd != DEFAULT_DIR_FD) ||
- (dst_dir_fd != DEFAULT_DIR_FD) ||
- (!follow_symlinks)) {
- if (HAVE_LINKAT_RUNTIME) {
- result = linkat(src_dir_fd, src->narrow,
- dst_dir_fd, dst->narrow,
- follow_symlinks ? AT_SYMLINK_FOLLOW : 0);
- }
- #ifdef __APPLE__
- else {
- if (src_dir_fd == DEFAULT_DIR_FD && dst_dir_fd == DEFAULT_DIR_FD) {
- /* See issue 41355: This matches the behaviour of !HAVE_LINKAT */
- result = link(src->narrow, dst->narrow);
- } else {
- linkat_unavailable = 1;
- }
- }
- #endif
- }
- else
- #endif /* HAVE_LINKAT */
- result = link(src->narrow, dst->narrow);
- Py_END_ALLOW_THREADS
- #ifdef HAVE_LINKAT
- if (linkat_unavailable) {
- /* Either or both dir_fd arguments were specified */
- if (src_dir_fd != DEFAULT_DIR_FD) {
- argument_unavailable_error("link", "src_dir_fd");
- } else {
- argument_unavailable_error("link", "dst_dir_fd");
- }
- return NULL;
- }
- #endif
- if (result)
- return path_error2(src, dst);
- #endif /* MS_WINDOWS */
- Py_RETURN_NONE;
- }
- #endif
- #if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
- static PyObject *
- _listdir_windows_no_opendir(path_t *path, PyObject *list)
- {
- PyObject *v;
- HANDLE hFindFile = INVALID_HANDLE_VALUE;
- BOOL result, return_bytes;
- wchar_t namebuf[MAX_PATH+4]; /* Overallocate for "\*.*" */
- /* only claim to have space for MAX_PATH */
- Py_ssize_t len = Py_ARRAY_LENGTH(namebuf)-4;
- wchar_t *wnamebuf = NULL;
- WIN32_FIND_DATAW wFileData;
- const wchar_t *po_wchars;
- if (!path->wide) { /* Default arg: "." */
- po_wchars = L".";
- len = 1;
- return_bytes = 0;
- } else {
- po_wchars = path->wide;
- len = wcslen(path->wide);
- return_bytes = PyBytes_Check(path->object);
- }
- /* The +5 is so we can append "\\*.*\0" */
- wnamebuf = PyMem_New(wchar_t, len + 5);
- if (!wnamebuf) {
- PyErr_NoMemory();
- goto exit;
- }
- wcscpy(wnamebuf, po_wchars);
- if (len > 0) {
- wchar_t wch = wnamebuf[len-1];
- if (wch != SEP && wch != ALTSEP && wch != L':')
- wnamebuf[len++] = SEP;
- wcscpy(wnamebuf + len, L"*.*");
- }
- if ((list = PyList_New(0)) == NULL) {
- goto exit;
- }
- Py_BEGIN_ALLOW_THREADS
- hFindFile = FindFirstFileW(wnamebuf, &wFileData);
- Py_END_ALLOW_THREADS
- if (hFindFile == INVALID_HANDLE_VALUE) {
- int error = GetLastError();
- if (error == ERROR_FILE_NOT_FOUND)
- goto exit;
- path_error(path);
- Py_CLEAR(list);
- goto exit;
- }
- do {
- /* Skip over . and .. */
- if (wcscmp(wFileData.cFileName, L".") != 0 &&
- wcscmp(wFileData.cFileName, L"..") != 0) {
- v = PyUnicode_FromWideChar(wFileData.cFileName,
- wcslen(wFileData.cFileName));
- if (return_bytes && v) {
- Py_SETREF(v, PyUnicode_EncodeFSDefault(v));
- }
- if (v == NULL) {
- Py_CLEAR(list);
- break;
- }
- if (PyList_Append(list, v) != 0) {
- Py_DECREF(v);
- Py_CLEAR(list);
- break;
- }
- Py_DECREF(v);
- }
- Py_BEGIN_ALLOW_THREADS
- result = FindNextFileW(hFindFile, &wFileData);
- Py_END_ALLOW_THREADS
- /* FindNextFile sets error to ERROR_NO_MORE_FILES if
- it got to the end of the directory. */
- if (!result && GetLastError() != ERROR_NO_MORE_FILES) {
- path_error(path);
- Py_CLEAR(list);
- goto exit;
- }
- } while (result == TRUE);
- exit:
- if (hFindFile != INVALID_HANDLE_VALUE) {
- if (FindClose(hFindFile) == FALSE) {
- if (list != NULL) {
- path_error(path);
- Py_CLEAR(list);
- }
- }
- }
- PyMem_Free(wnamebuf);
- return list;
- } /* end of _listdir_windows_no_opendir */
- #else /* thus POSIX, ie: not (MS_WINDOWS and not HAVE_OPENDIR) */
- static PyObject *
- _posix_listdir(path_t *path, PyObject *list)
- {
- PyObject *v;
- DIR *dirp = NULL;
- struct dirent *ep;
- int return_str; /* if false, return bytes */
- #ifdef HAVE_FDOPENDIR
- int fd = -1;
- #endif
- errno = 0;
- #ifdef HAVE_FDOPENDIR
- if (path->fd != -1) {
- if (HAVE_FDOPENDIR_RUNTIME) {
- /* closedir() closes the FD, so we duplicate it */
- fd = _Py_dup(path->fd);
- if (fd == -1)
- return NULL;
- return_str = 1;
- Py_BEGIN_ALLOW_THREADS
- dirp = fdopendir(fd);
- Py_END_ALLOW_THREADS
- } else {
- PyErr_SetString(PyExc_TypeError,
- "listdir: path should be string, bytes, os.PathLike or None, not int");
- return NULL;
- }
- }
- else
- #endif
- {
- const char *name;
- if (path->narrow) {
- name = path->narrow;
- /* only return bytes if they specified a bytes object */
- return_str = !PyBytes_Check(path->object);
- }
- else {
- name = ".";
- return_str = 1;
- }
- Py_BEGIN_ALLOW_THREADS
- dirp = opendir(name);
- Py_END_ALLOW_THREADS
- }
- if (dirp == NULL) {
- path_error(path);
- list = NULL;
- #ifdef HAVE_FDOPENDIR
- if (fd != -1) {
- Py_BEGIN_ALLOW_THREADS
- close(fd);
- Py_END_ALLOW_THREADS
- }
- #endif
- goto exit;
- }
- if ((list = PyList_New(0)) == NULL) {
- goto exit;
- }
- for (;;) {
- errno = 0;
- Py_BEGIN_ALLOW_THREADS
- ep = readdir(dirp);
- Py_END_ALLOW_THREADS
- if (ep == NULL) {
- if (errno == 0) {
- break;
- } else {
- path_error(path);
- Py_CLEAR(list);
- goto exit;
- }
- }
- if (ep->d_name[0] == '.' &&
- (NAMLEN(ep) == 1 ||
- (ep->d_name[1] == '.' && NAMLEN(ep) == 2)))
- continue;
- if (return_str)
- v = PyUnicode_DecodeFSDefaultAndSize(ep->d_name, NAMLEN(ep));
- else
- v = PyBytes_FromStringAndSize(ep->d_name, NAMLEN(ep));
- if (v == NULL) {
- Py_CLEAR(list);
- break;
- }
- if (PyList_Append(list, v) != 0) {
- Py_DECREF(v);
- Py_CLEAR(list);
- break;
- }
- Py_DECREF(v);
- }
- exit:
- if (dirp != NULL) {
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_FDOPENDIR
- if (fd > -1)
- rewinddir(dirp);
- #endif
- closedir(dirp);
- Py_END_ALLOW_THREADS
- }
- return list;
- } /* end of _posix_listdir */
- #endif /* which OS */
- /*[clinic input]
- os.listdir
- path : path_t(nullable=True, allow_fd='PATH_HAVE_FDOPENDIR') = None
- Return a list containing the names of the files in the directory.
- path can be specified as either str, bytes, or a path-like object. If path is bytes,
- the filenames returned will also be bytes; in all other circumstances
- the filenames returned will be str.
- If path is None, uses the path='.'.
- On some platforms, path may also be specified as an open file descriptor;\
- the file descriptor must refer to a directory.
- If this functionality is unavailable, using it raises NotImplementedError.
- The list is in arbitrary order. It does not include the special
- entries '.' and '..' even if they are present in the directory.
- [clinic start generated code]*/
- static PyObject *
- os_listdir_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=293045673fcd1a75 input=e3f58030f538295d]*/
- {
- if (PySys_Audit("os.listdir", "O",
- path->object ? path->object : Py_None) < 0) {
- return NULL;
- }
- #if defined(MS_WINDOWS) && !defined(HAVE_OPENDIR)
- return _listdir_windows_no_opendir(path, NULL);
- #else
- return _posix_listdir(path, NULL);
- #endif
- }
- #ifdef MS_WINDOWS
- /*[clinic input]
- os.listdrives
- Return a list containing the names of drives in the system.
- A drive name typically looks like 'C:\\'.
- [clinic start generated code]*/
- static PyObject *
- os_listdrives_impl(PyObject *module)
- /*[clinic end generated code: output=aaece9dacdf682b5 input=1af9ccc9e583798e]*/
- {
- /* Number of possible drives is limited, so 256 should always be enough.
- On the day when it is not, listmounts() will have to be used. */
- wchar_t buffer[256];
- DWORD buflen = Py_ARRAY_LENGTH(buffer);
- PyObject *result = NULL;
- if (PySys_Audit("os.listdrives", NULL) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS;
- buflen = GetLogicalDriveStringsW(buflen, buffer);
- Py_END_ALLOW_THREADS;
- if (!buflen) {
- PyErr_SetFromWindowsErr(0);
- return NULL;
- } else if (buflen >= Py_ARRAY_LENGTH(buffer)) {
- PyErr_SetFromWindowsErr(ERROR_MORE_DATA);
- return NULL;
- }
- /* buflen includes a null terminator, so remove it */
- PyObject *str = PyUnicode_FromWideChar(buffer, buflen - 1);
- if (str) {
- PyObject *nullchar = PyUnicode_FromStringAndSize("\0", 1);
- if (nullchar) {
- result = PyUnicode_Split(str, nullchar, -1);
- Py_DECREF(nullchar);
- }
- Py_DECREF(str);
- }
- return result;
- }
- /*[clinic input]
- os.listvolumes
- Return a list containing the volumes in the system.
- Volumes are typically represented as a GUID path.
- [clinic start generated code]*/
- static PyObject *
- os_listvolumes_impl(PyObject *module)
- /*[clinic end generated code: output=534e10ea2bf9d386 input=f6e4e70371f11e99]*/
- {
- PyObject *result = PyList_New(0);
- HANDLE find = INVALID_HANDLE_VALUE;
- wchar_t buffer[MAX_PATH + 1];
- if (!result) {
- return NULL;
- }
- if (PySys_Audit("os.listvolumes", NULL) < 0) {
- Py_DECREF(result);
- return NULL;
- }
- int err = 0;
- Py_BEGIN_ALLOW_THREADS;
- find = FindFirstVolumeW(buffer, Py_ARRAY_LENGTH(buffer));
- if (find == INVALID_HANDLE_VALUE) {
- err = GetLastError();
- }
- Py_END_ALLOW_THREADS;
- while (!err) {
- PyObject *s = PyUnicode_FromWideChar(buffer, -1);
- if (!s || PyList_Append(result, s) < 0) {
- Py_XDECREF(s);
- Py_CLEAR(result);
- break;
- }
- Py_DECREF(s);
- Py_BEGIN_ALLOW_THREADS;
- if (!FindNextVolumeW(find, buffer, Py_ARRAY_LENGTH(buffer))) {
- err = GetLastError();
- }
- Py_END_ALLOW_THREADS;
- }
- if (find != INVALID_HANDLE_VALUE) {
- Py_BEGIN_ALLOW_THREADS;
- FindVolumeClose(find);
- Py_END_ALLOW_THREADS;
- }
- if (err && err != ERROR_NO_MORE_FILES) {
- PyErr_SetFromWindowsErr(err);
- Py_XDECREF(result);
- result = NULL;
- }
- return result;
- }
- /*[clinic input]
- os.listmounts
- volume: path_t
- Return a list containing mount points for a particular volume.
- 'volume' should be a GUID path as returned from os.listvolumes.
- [clinic start generated code]*/
- static PyObject *
- os_listmounts_impl(PyObject *module, path_t *volume)
- /*[clinic end generated code: output=06da49679de4512e input=a8a27178e3f67845]*/
- {
- wchar_t default_buffer[MAX_PATH + 1];
- DWORD buflen = Py_ARRAY_LENGTH(default_buffer);
- LPWSTR buffer = default_buffer;
- DWORD attributes;
- PyObject *str = NULL;
- PyObject *nullchar = NULL;
- PyObject *result = NULL;
- /* Ensure we have a valid volume path before continuing */
- Py_BEGIN_ALLOW_THREADS
- attributes = GetFileAttributesW(volume->wide);
- Py_END_ALLOW_THREADS
- if (attributes == INVALID_FILE_ATTRIBUTES &&
- GetLastError() == ERROR_UNRECOGNIZED_VOLUME)
- {
- return PyErr_SetFromWindowsErr(ERROR_UNRECOGNIZED_VOLUME);
- }
- if (PySys_Audit("os.listmounts", "O", volume->object) < 0) {
- return NULL;
- }
- while (1) {
- BOOL success;
- Py_BEGIN_ALLOW_THREADS
- success = GetVolumePathNamesForVolumeNameW(volume->wide, buffer,
- buflen, &buflen);
- Py_END_ALLOW_THREADS
- if (success) {
- break;
- }
- if (GetLastError() != ERROR_MORE_DATA) {
- PyErr_SetFromWindowsErr(0);
- goto exit;
- }
- if (buffer != default_buffer) {
- PyMem_Free((void *)buffer);
- }
- buffer = (wchar_t*)PyMem_Malloc(sizeof(wchar_t) * buflen);
- if (!buffer) {
- PyErr_NoMemory();
- goto exit;
- }
- }
- if (buflen < 2) {
- result = PyList_New(0);
- goto exit;
- }
- // buflen includes two null terminators, one for the last string
- // and one for the array of strings.
- str = PyUnicode_FromWideChar(buffer, buflen - 2);
- nullchar = PyUnicode_FromStringAndSize("\0", 1);
- if (str && nullchar) {
- result = PyUnicode_Split(str, nullchar, -1);
- }
- exit:
- if (buffer != default_buffer) {
- PyMem_Free(buffer);
- }
- Py_XDECREF(nullchar);
- Py_XDECREF(str);
- return result;
- }
- /*[clinic input]
- os._path_isdevdrive
- path: path_t
- Determines whether the specified path is on a Windows Dev Drive.
- [clinic start generated code]*/
- static PyObject *
- os__path_isdevdrive_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=1f437ea6677433a2 input=ee83e4996a48e23d]*/
- {
- #ifndef PERSISTENT_VOLUME_STATE_DEV_VOLUME
- /* This flag will be documented at
- https://learn.microsoft.com/windows-hardware/drivers/ddi/ntifs/ns-ntifs-_file_fs_persistent_volume_information
- after release, and will be available in the latest WinSDK.
- We include the flag to avoid a specific version dependency
- on the latest WinSDK. */
- const int PERSISTENT_VOLUME_STATE_DEV_VOLUME = 0x00002000;
- #endif
- int err = 0;
- PyObject *r = NULL;
- wchar_t volume[MAX_PATH];
- Py_BEGIN_ALLOW_THREADS
- if (!GetVolumePathNameW(path->wide, volume, MAX_PATH)) {
- /* invalid path of some kind */
- /* Note that this also includes the case where a volume is mounted
- in a path longer than 260 characters. This is likely to be rare
- and problematic for other reasons, so a (soft) failure in this
- check seems okay. */
- err = GetLastError();
- } else if (GetDriveTypeW(volume) != DRIVE_FIXED) {
- /* only care about local dev drives */
- r = Py_False;
- } else {
- HANDLE hVolume = CreateFileW(
- volume,
- FILE_READ_ATTRIBUTES,
- FILE_SHARE_READ | FILE_SHARE_WRITE,
- NULL,
- OPEN_EXISTING,
- FILE_FLAG_BACKUP_SEMANTICS,
- NULL
- );
- if (hVolume == INVALID_HANDLE_VALUE) {
- err = GetLastError();
- } else {
- FILE_FS_PERSISTENT_VOLUME_INFORMATION volumeState = {0};
- volumeState.Version = 1;
- volumeState.FlagMask = PERSISTENT_VOLUME_STATE_DEV_VOLUME;
- if (!DeviceIoControl(
- hVolume,
- FSCTL_QUERY_PERSISTENT_VOLUME_STATE,
- &volumeState,
- sizeof(volumeState),
- &volumeState,
- sizeof(volumeState),
- NULL,
- NULL
- )) {
- err = GetLastError();
- }
- CloseHandle(hVolume);
- if (err == ERROR_INVALID_PARAMETER) {
- /* not supported on this platform */
- r = Py_False;
- } else if (!err) {
- r = (volumeState.VolumeFlags & PERSISTENT_VOLUME_STATE_DEV_VOLUME)
- ? Py_True : Py_False;
- }
- }
- }
- Py_END_ALLOW_THREADS
- if (err) {
- PyErr_SetFromWindowsErr(err);
- return NULL;
- }
- if (r) {
- return Py_NewRef(r);
- }
- return NULL;
- }
- int
- _PyOS_getfullpathname(const wchar_t *path, wchar_t **abspath_p)
- {
- wchar_t woutbuf[MAX_PATH], *woutbufp = woutbuf;
- DWORD result;
- result = GetFullPathNameW(path,
- Py_ARRAY_LENGTH(woutbuf), woutbuf,
- NULL);
- if (!result) {
- return -1;
- }
- if (result >= Py_ARRAY_LENGTH(woutbuf)) {
- if ((size_t)result <= (size_t)PY_SSIZE_T_MAX / sizeof(wchar_t)) {
- woutbufp = PyMem_RawMalloc((size_t)result * sizeof(wchar_t));
- }
- else {
- woutbufp = NULL;
- }
- if (!woutbufp) {
- *abspath_p = NULL;
- return 0;
- }
- result = GetFullPathNameW(path, result, woutbufp, NULL);
- if (!result) {
- PyMem_RawFree(woutbufp);
- return -1;
- }
- }
- if (woutbufp != woutbuf) {
- *abspath_p = woutbufp;
- return 0;
- }
- *abspath_p = _PyMem_RawWcsdup(woutbufp);
- return 0;
- }
- /* A helper function for abspath on win32 */
- /*[clinic input]
- os._getfullpathname
- path: path_t
- /
- [clinic start generated code]*/
- static PyObject *
- os__getfullpathname_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=bb8679d56845bc9b input=332ed537c29d0a3e]*/
- {
- wchar_t *abspath;
- if (_PyOS_getfullpathname(path->wide, &abspath) < 0) {
- return win32_error_object("GetFullPathNameW", path->object);
- }
- if (abspath == NULL) {
- return PyErr_NoMemory();
- }
- PyObject *str = PyUnicode_FromWideChar(abspath, wcslen(abspath));
- PyMem_RawFree(abspath);
- if (str == NULL) {
- return NULL;
- }
- if (PyBytes_Check(path->object)) {
- Py_SETREF(str, PyUnicode_EncodeFSDefault(str));
- }
- return str;
- }
- /*[clinic input]
- os._getfinalpathname
- path: path_t
- /
- A helper function for samepath on windows.
- [clinic start generated code]*/
- static PyObject *
- os__getfinalpathname_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=621a3c79bc29ebfa input=2b6b6c7cbad5fb84]*/
- {
- HANDLE hFile;
- wchar_t buf[MAXPATHLEN], *target_path = buf;
- int buf_size = Py_ARRAY_LENGTH(buf);
- int result_length;
- PyObject *result;
- Py_BEGIN_ALLOW_THREADS
- hFile = CreateFileW(
- path->wide,
- 0, /* desired access */
- 0, /* share mode */
- NULL, /* security attributes */
- OPEN_EXISTING,
- /* FILE_FLAG_BACKUP_SEMANTICS is required to open a directory */
- FILE_FLAG_BACKUP_SEMANTICS,
- NULL);
- Py_END_ALLOW_THREADS
- if (hFile == INVALID_HANDLE_VALUE) {
- return win32_error_object("CreateFileW", path->object);
- }
- /* We have a good handle to the target, use it to determine the
- target path name. */
- while (1) {
- Py_BEGIN_ALLOW_THREADS
- result_length = GetFinalPathNameByHandleW(hFile, target_path,
- buf_size, VOLUME_NAME_DOS);
- Py_END_ALLOW_THREADS
- if (!result_length) {
- result = win32_error_object("GetFinalPathNameByHandleW",
- path->object);
- goto cleanup;
- }
- if (result_length < buf_size) {
- break;
- }
- wchar_t *tmp;
- tmp = PyMem_Realloc(target_path != buf ? target_path : NULL,
- result_length * sizeof(*tmp));
- if (!tmp) {
- result = PyErr_NoMemory();
- goto cleanup;
- }
- buf_size = result_length;
- target_path = tmp;
- }
- result = PyUnicode_FromWideChar(target_path, result_length);
- if (result && PyBytes_Check(path->object)) {
- Py_SETREF(result, PyUnicode_EncodeFSDefault(result));
- }
- cleanup:
- if (target_path != buf) {
- PyMem_Free(target_path);
- }
- CloseHandle(hFile);
- return result;
- }
- /*[clinic input]
- os._getvolumepathname
- path: path_t
- A helper function for ismount on Win32.
- [clinic start generated code]*/
- static PyObject *
- os__getvolumepathname_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=804c63fd13a1330b input=722b40565fa21552]*/
- {
- PyObject *result;
- wchar_t *mountpath=NULL;
- size_t buflen;
- BOOL ret;
- /* Volume path should be shorter than entire path */
- buflen = Py_MAX(path->length, MAX_PATH);
- if (buflen > PY_DWORD_MAX) {
- PyErr_SetString(PyExc_OverflowError, "path too long");
- return NULL;
- }
- mountpath = PyMem_New(wchar_t, buflen);
- if (mountpath == NULL)
- return PyErr_NoMemory();
- Py_BEGIN_ALLOW_THREADS
- ret = GetVolumePathNameW(path->wide, mountpath,
- Py_SAFE_DOWNCAST(buflen, size_t, DWORD));
- Py_END_ALLOW_THREADS
- if (!ret) {
- result = win32_error_object("_getvolumepathname", path->object);
- goto exit;
- }
- result = PyUnicode_FromWideChar(mountpath, wcslen(mountpath));
- if (PyBytes_Check(path->object))
- Py_SETREF(result, PyUnicode_EncodeFSDefault(result));
- exit:
- PyMem_Free(mountpath);
- return result;
- }
- /*[clinic input]
- os._path_splitroot
- path: path_t
- Removes everything after the root on Win32.
- [clinic start generated code]*/
- static PyObject *
- os__path_splitroot_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=ab7f1a88b654581c input=dc93b1d3984cffb6]*/
- {
- wchar_t *buffer;
- wchar_t *end;
- PyObject *result = NULL;
- HRESULT ret;
- buffer = (wchar_t*)PyMem_Malloc(sizeof(wchar_t) * (wcslen(path->wide) + 1));
- if (!buffer) {
- return NULL;
- }
- wcscpy(buffer, path->wide);
- for (wchar_t *p = wcschr(buffer, L'/'); p; p = wcschr(p, L'/')) {
- *p = L'\\';
- }
- Py_BEGIN_ALLOW_THREADS
- ret = PathCchSkipRoot(buffer, &end);
- Py_END_ALLOW_THREADS
- if (FAILED(ret)) {
- result = Py_BuildValue("sO", "", path->object);
- } else if (end != buffer) {
- size_t rootLen = (size_t)(end - buffer);
- result = Py_BuildValue("NN",
- PyUnicode_FromWideChar(path->wide, rootLen),
- PyUnicode_FromWideChar(path->wide + rootLen, -1)
- );
- } else {
- result = Py_BuildValue("Os", path->object, "");
- }
- PyMem_Free(buffer);
- return result;
- }
- #define PY_IFREG 1 // Regular file
- #define PY_IFDIR 2 // Directory
- #define PY_IFLNK 4 // Symlink
- #define PY_IFMNT 8 // Mount Point (junction)
- #define PY_IFLRP 16 // Link Reparse Point (name-surrogate, symlink, junction)
- #define PY_IFRRP 32 // Regular Reparse Point
- static inline BOOL
- _testInfo(DWORD attributes, DWORD reparseTag, BOOL diskDevice, int testedType)
- {
- switch (testedType) {
- case PY_IFREG:
- return diskDevice && attributes &&
- !(attributes & FILE_ATTRIBUTE_DIRECTORY);
- case PY_IFDIR:
- return attributes & FILE_ATTRIBUTE_DIRECTORY;
- case PY_IFLNK:
- return (attributes & FILE_ATTRIBUTE_REPARSE_POINT) &&
- reparseTag == IO_REPARSE_TAG_SYMLINK;
- case PY_IFMNT:
- return (attributes & FILE_ATTRIBUTE_REPARSE_POINT) &&
- reparseTag == IO_REPARSE_TAG_MOUNT_POINT;
- case PY_IFLRP:
- return (attributes & FILE_ATTRIBUTE_REPARSE_POINT) &&
- IsReparseTagNameSurrogate(reparseTag);
- case PY_IFRRP:
- return (attributes & FILE_ATTRIBUTE_REPARSE_POINT) &&
- reparseTag && !IsReparseTagNameSurrogate(reparseTag);
- }
- return FALSE;
- }
- static BOOL
- _testFileTypeByHandle(HANDLE hfile, int testedType, BOOL diskOnly)
- {
- assert(testedType == PY_IFREG || testedType == PY_IFDIR ||
- testedType == PY_IFLNK || testedType == PY_IFMNT ||
- testedType == PY_IFLRP || testedType == PY_IFRRP);
- BOOL diskDevice = GetFileType(hfile) == FILE_TYPE_DISK;
- if (diskOnly && !diskDevice) {
- return FALSE;
- }
- if (testedType != PY_IFREG && testedType != PY_IFDIR) {
- FILE_ATTRIBUTE_TAG_INFO info;
- return GetFileInformationByHandleEx(hfile, FileAttributeTagInfo, &info,
- sizeof(info)) &&
- _testInfo(info.FileAttributes, info.ReparseTag, diskDevice,
- testedType);
- }
- FILE_BASIC_INFO info;
- return GetFileInformationByHandleEx(hfile, FileBasicInfo, &info,
- sizeof(info)) &&
- _testInfo(info.FileAttributes, 0, diskDevice, testedType);
- }
- static BOOL
- _testFileTypeByName(LPCWSTR path, int testedType)
- {
- assert(testedType == PY_IFREG || testedType == PY_IFDIR ||
- testedType == PY_IFLNK || testedType == PY_IFMNT ||
- testedType == PY_IFLRP || testedType == PY_IFRRP);
- FILE_STAT_BASIC_INFORMATION info;
- if (_Py_GetFileInformationByName(path, FileStatBasicByNameInfo, &info,
- sizeof(info)))
- {
- BOOL diskDevice = info.DeviceType == FILE_DEVICE_DISK ||
- info.DeviceType == FILE_DEVICE_VIRTUAL_DISK ||
- info.DeviceType == FILE_DEVICE_CD_ROM;
- BOOL result = _testInfo(info.FileAttributes, info.ReparseTag,
- diskDevice, testedType);
- if (!result || (testedType != PY_IFREG && testedType != PY_IFDIR) ||
- !(info.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT))
- {
- return result;
- }
- }
- else if (_Py_GetFileInformationByName_ErrorIsTrustworthy(
- GetLastError()))
- {
- return FALSE;
- }
- DWORD flags = FILE_FLAG_BACKUP_SEMANTICS;
- if (testedType != PY_IFREG && testedType != PY_IFDIR) {
- flags |= FILE_FLAG_OPEN_REPARSE_POINT;
- }
- HANDLE hfile = CreateFileW(path, FILE_READ_ATTRIBUTES, 0, NULL,
- OPEN_EXISTING, flags, NULL);
- if (hfile != INVALID_HANDLE_VALUE) {
- BOOL result = _testFileTypeByHandle(hfile, testedType, FALSE);
- CloseHandle(hfile);
- return result;
- }
- switch (GetLastError()) {
- case ERROR_ACCESS_DENIED:
- case ERROR_SHARING_VIOLATION:
- case ERROR_CANT_ACCESS_FILE:
- case ERROR_INVALID_PARAMETER:
- int rc;
- STRUCT_STAT st;
- if (testedType == PY_IFREG || testedType == PY_IFDIR) {
- rc = STAT(path, &st);
- }
- else {
- // PY_IFRRP is not generally supported in this case, except for
- // unhandled reparse points such as IO_REPARSE_TAG_APPEXECLINK.
- rc = LSTAT(path, &st);
- }
- if (!rc) {
- return _testInfo(st.st_file_attributes, st.st_reparse_tag,
- st.st_mode & S_IFREG, testedType);
- }
- }
- return FALSE;
- }
- static BOOL
- _testFileExistsByName(LPCWSTR path, BOOL followLinks)
- {
- FILE_STAT_BASIC_INFORMATION info;
- if (_Py_GetFileInformationByName(path, FileStatBasicByNameInfo, &info,
- sizeof(info)))
- {
- if (!(info.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) ||
- !followLinks && IsReparseTagNameSurrogate(info.ReparseTag))
- {
- return TRUE;
- }
- }
- else if (_Py_GetFileInformationByName_ErrorIsTrustworthy(
- GetLastError()))
- {
- return FALSE;
- }
- DWORD flags = FILE_FLAG_BACKUP_SEMANTICS;
- if (!followLinks) {
- flags |= FILE_FLAG_OPEN_REPARSE_POINT;
- }
- HANDLE hfile = CreateFileW(path, FILE_READ_ATTRIBUTES, 0, NULL,
- OPEN_EXISTING, flags, NULL);
- if (hfile != INVALID_HANDLE_VALUE) {
- if (followLinks) {
- CloseHandle(hfile);
- return TRUE;
- }
- // Regular Reparse Points (PY_IFRRP) have to be traversed.
- BOOL result = _testFileTypeByHandle(hfile, PY_IFRRP, FALSE);
- CloseHandle(hfile);
- if (!result) {
- return TRUE;
- }
- hfile = CreateFileW(path, FILE_READ_ATTRIBUTES, 0, NULL, OPEN_EXISTING,
- FILE_FLAG_BACKUP_SEMANTICS, NULL);
- if (hfile != INVALID_HANDLE_VALUE) {
- CloseHandle(hfile);
- return TRUE;
- }
- }
- switch (GetLastError()) {
- case ERROR_ACCESS_DENIED:
- case ERROR_SHARING_VIOLATION:
- case ERROR_CANT_ACCESS_FILE:
- case ERROR_INVALID_PARAMETER:
- STRUCT_STAT _st;
- return followLinks ? !STAT(path, &_st): !LSTAT(path, &_st);
- }
- return FALSE;
- }
- static BOOL
- _testFileExists(path_t *path, BOOL followLinks)
- {
- BOOL result = FALSE;
- if (path->value_error) {
- return FALSE;
- }
- Py_BEGIN_ALLOW_THREADS
- if (path->fd != -1) {
- HANDLE hfile = _Py_get_osfhandle_noraise(path->fd);
- if (hfile != INVALID_HANDLE_VALUE) {
- if (GetFileType(hfile) != FILE_TYPE_UNKNOWN || !GetLastError()) {
- result = TRUE;
- }
- }
- }
- else if (path->wide) {
- result = _testFileExistsByName(path->wide, followLinks);
- }
- Py_END_ALLOW_THREADS
- return result;
- }
- static BOOL
- _testFileType(path_t *path, int testedType)
- {
- BOOL result = FALSE;
- if (path->value_error) {
- return FALSE;
- }
- Py_BEGIN_ALLOW_THREADS
- if (path->fd != -1) {
- HANDLE hfile = _Py_get_osfhandle_noraise(path->fd);
- if (hfile != INVALID_HANDLE_VALUE) {
- result = _testFileTypeByHandle(hfile, testedType, TRUE);
- }
- }
- else if (path->wide) {
- result = _testFileTypeByName(path->wide, testedType);
- }
- Py_END_ALLOW_THREADS
- return result;
- }
- /*[clinic input]
- os._path_exists -> bool
- path: path_t(allow_fd=True, suppress_value_error=True)
- Test whether a path exists. Returns False for broken symbolic links.
- [clinic start generated code]*/
- static int
- os__path_exists_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=8da13acf666e16ba input=142beabfc66783eb]*/
- {
- return _testFileExists(path, TRUE);
- }
- /*[clinic input]
- os._path_isdir -> bool
- s as path: path_t(allow_fd=True, suppress_value_error=True)
- Return true if the pathname refers to an existing directory.
- [clinic start generated code]*/
- static int
- os__path_isdir_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=d5786196f9e2fa7a input=132a3b5301aecf79]*/
- {
- return _testFileType(path, PY_IFDIR);
- }
- /*[clinic input]
- os._path_isfile -> bool
- path: path_t(allow_fd=True, suppress_value_error=True)
- Test whether a path is a regular file
- [clinic start generated code]*/
- static int
- os__path_isfile_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=5c3073bc212b9863 input=4ac1fd350b30a39e]*/
- {
- return _testFileType(path, PY_IFREG);
- }
- /*[clinic input]
- os._path_islink -> bool
- path: path_t(allow_fd=True, suppress_value_error=True)
- Test whether a path is a symbolic link
- [clinic start generated code]*/
- static int
- os__path_islink_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=30da7bda8296adcc input=7510ce05b547debb]*/
- {
- return _testFileType(path, PY_IFLNK);
- }
- #undef PY_IFREG
- #undef PY_IFDIR
- #undef PY_IFLNK
- #undef PY_IFMNT
- #undef PY_IFLRP
- #undef PY_IFRRP
- #endif /* MS_WINDOWS */
- /*[clinic input]
- os._path_normpath
- path: path_t(make_wide=True, nonstrict=True)
- Normalize path, eliminating double slashes, etc.
- [clinic start generated code]*/
- static PyObject *
- os__path_normpath_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=d353e7ed9410c044 input=3d4ac23b06332dcb]*/
- {
- PyObject *result;
- Py_ssize_t norm_len;
- wchar_t *norm_path = _Py_normpath_and_size((wchar_t *)path->wide,
- path->length, &norm_len);
- if (!norm_len) {
- result = PyUnicode_FromOrdinal('.');
- }
- else {
- result = PyUnicode_FromWideChar(norm_path, norm_len);
- }
- if (PyBytes_Check(path->object)) {
- Py_SETREF(result, PyUnicode_EncodeFSDefault(result));
- }
- return result;
- }
- /*[clinic input]
- os.mkdir
- path : path_t
- mode: int = 0o777
- *
- dir_fd : dir_fd(requires='mkdirat') = None
- # "mkdir(path, mode=0o777, *, dir_fd=None)\n\n\
- Create a directory.
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- The mode argument is ignored on Windows. Where it is used, the current umask
- value is first masked out.
- [clinic start generated code]*/
- static PyObject *
- os_mkdir_impl(PyObject *module, path_t *path, int mode, int dir_fd)
- /*[clinic end generated code: output=a70446903abe821f input=a61722e1576fab03]*/
- {
- int result;
- #ifdef MS_WINDOWS
- int error = 0;
- int pathError = 0;
- SECURITY_ATTRIBUTES secAttr = { sizeof(secAttr) };
- SECURITY_ATTRIBUTES *pSecAttr = NULL;
- #endif
- #ifdef HAVE_MKDIRAT
- int mkdirat_unavailable = 0;
- #endif
- if (PySys_Audit("os.mkdir", "Oii", path->object, mode,
- dir_fd == DEFAULT_DIR_FD ? -1 : dir_fd) < 0) {
- return NULL;
- }
- #ifdef MS_WINDOWS
- Py_BEGIN_ALLOW_THREADS
- if (mode == 0700 /* 0o700 */) {
- ULONG sdSize;
- pSecAttr = &secAttr;
- // Set a discretionary ACL (D) that is protected (P) and includes
- // inheritable (OICI) entries that allow (A) full control (FA) to
- // SYSTEM (SY), Administrators (BA), and the owner (OW).
- if (!ConvertStringSecurityDescriptorToSecurityDescriptorW(
- L"D:P(A;OICI;FA;;;SY)(A;OICI;FA;;;BA)(A;OICI;FA;;;OW)",
- SDDL_REVISION_1,
- &secAttr.lpSecurityDescriptor,
- &sdSize
- )) {
- error = GetLastError();
- }
- }
- if (!error) {
- result = CreateDirectoryW(path->wide, pSecAttr);
- if (secAttr.lpSecurityDescriptor &&
- // uncommonly, LocalFree returns non-zero on error, but still uses
- // GetLastError() to see what the error code is
- LocalFree(secAttr.lpSecurityDescriptor)) {
- error = GetLastError();
- }
- }
- Py_END_ALLOW_THREADS
- if (error) {
- return PyErr_SetFromWindowsErr(error);
- }
- if (!result) {
- return path_error(path);
- }
- #else
- Py_BEGIN_ALLOW_THREADS
- #if HAVE_MKDIRAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_MKDIRAT_RUNTIME) {
- result = mkdirat(dir_fd, path->narrow, mode);
- } else {
- mkdirat_unavailable = 1;
- }
- } else
- #endif
- #if defined(__WATCOMC__) && !defined(__QNX__)
- result = mkdir(path->narrow);
- #else
- result = mkdir(path->narrow, mode);
- #endif
- Py_END_ALLOW_THREADS
- #if HAVE_MKDIRAT
- if (mkdirat_unavailable) {
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (result < 0)
- return path_error(path);
- #endif /* MS_WINDOWS */
- Py_RETURN_NONE;
- }
- /* sys/resource.h is needed for at least: wait3(), wait4(), broken nice. */
- #if defined(HAVE_SYS_RESOURCE_H)
- #include <sys/resource.h>
- #endif
- #ifdef HAVE_NICE
- /*[clinic input]
- os.nice
- increment: int
- /
- Add increment to the priority of process and return the new priority.
- [clinic start generated code]*/
- static PyObject *
- os_nice_impl(PyObject *module, int increment)
- /*[clinic end generated code: output=9dad8a9da8109943 input=864be2d402a21da2]*/
- {
- int value;
- /* There are two flavours of 'nice': one that returns the new
- priority (as required by almost all standards out there) and the
- Linux/FreeBSD one, which returns '0' on success and advices
- the use of getpriority() to get the new priority.
- If we are of the nice family that returns the new priority, we
- need to clear errno before the call, and check if errno is filled
- before calling posix_error() on a returnvalue of -1, because the
- -1 may be the actual new priority! */
- errno = 0;
- value = nice(increment);
- #if defined(HAVE_BROKEN_NICE) && defined(HAVE_GETPRIORITY)
- if (value == 0)
- value = getpriority(PRIO_PROCESS, 0);
- #endif
- if (value == -1 && errno != 0)
- /* either nice() or getpriority() returned an error */
- return posix_error();
- return PyLong_FromLong((long) value);
- }
- #endif /* HAVE_NICE */
- #ifdef HAVE_GETPRIORITY
- /*[clinic input]
- os.getpriority
- which: int
- who: int
- Return program scheduling priority.
- [clinic start generated code]*/
- static PyObject *
- os_getpriority_impl(PyObject *module, int which, int who)
- /*[clinic end generated code: output=c41b7b63c7420228 input=9be615d40e2544ef]*/
- {
- int retval;
- errno = 0;
- retval = getpriority(which, who);
- if (errno != 0)
- return posix_error();
- return PyLong_FromLong((long)retval);
- }
- #endif /* HAVE_GETPRIORITY */
- #ifdef HAVE_SETPRIORITY
- /*[clinic input]
- os.setpriority
- which: int
- who: int
- priority: int
- Set program scheduling priority.
- [clinic start generated code]*/
- static PyObject *
- os_setpriority_impl(PyObject *module, int which, int who, int priority)
- /*[clinic end generated code: output=3d910d95a7771eb2 input=710ccbf65b9dc513]*/
- {
- int retval;
- retval = setpriority(which, who, priority);
- if (retval == -1)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETPRIORITY */
- static PyObject *
- internal_rename(path_t *src, path_t *dst, int src_dir_fd, int dst_dir_fd, int is_replace)
- {
- const char *function_name = is_replace ? "replace" : "rename";
- int dir_fd_specified;
- #ifdef HAVE_RENAMEAT
- int renameat_unavailable = 0;
- #endif
- #ifdef MS_WINDOWS
- BOOL result;
- int flags = is_replace ? MOVEFILE_REPLACE_EXISTING : 0;
- #else
- int result;
- #endif
- dir_fd_specified = (src_dir_fd != DEFAULT_DIR_FD) ||
- (dst_dir_fd != DEFAULT_DIR_FD);
- #ifndef HAVE_RENAMEAT
- if (dir_fd_specified) {
- argument_unavailable_error(function_name, "src_dir_fd and dst_dir_fd");
- return NULL;
- }
- #endif
- if (PySys_Audit("os.rename", "OOii", src->object, dst->object,
- src_dir_fd == DEFAULT_DIR_FD ? -1 : src_dir_fd,
- dst_dir_fd == DEFAULT_DIR_FD ? -1 : dst_dir_fd) < 0) {
- return NULL;
- }
- #ifdef MS_WINDOWS
- Py_BEGIN_ALLOW_THREADS
- result = MoveFileExW(src->wide, dst->wide, flags);
- Py_END_ALLOW_THREADS
- if (!result)
- return path_error2(src, dst);
- #else
- if ((src->narrow && dst->wide) || (src->wide && dst->narrow)) {
- PyErr_Format(PyExc_ValueError,
- "%s: src and dst must be the same type", function_name);
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_RENAMEAT
- if (dir_fd_specified) {
- if (HAVE_RENAMEAT_RUNTIME) {
- result = renameat(src_dir_fd, src->narrow, dst_dir_fd, dst->narrow);
- } else {
- renameat_unavailable = 1;
- }
- } else
- #endif
- result = rename(src->narrow, dst->narrow);
- Py_END_ALLOW_THREADS
- #ifdef HAVE_RENAMEAT
- if (renameat_unavailable) {
- argument_unavailable_error(function_name, "src_dir_fd and dst_dir_fd");
- return NULL;
- }
- #endif
- if (result)
- return path_error2(src, dst);
- #endif
- Py_RETURN_NONE;
- }
- /*[clinic input]
- os.rename
- src : path_t
- dst : path_t
- *
- src_dir_fd : dir_fd = None
- dst_dir_fd : dir_fd = None
- Rename a file or directory.
- If either src_dir_fd or dst_dir_fd is not None, it should be a file
- descriptor open to a directory, and the respective path string (src or dst)
- should be relative; the path will then be relative to that directory.
- src_dir_fd and dst_dir_fd, may not be implemented on your platform.
- If they are unavailable, using them will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_rename_impl(PyObject *module, path_t *src, path_t *dst, int src_dir_fd,
- int dst_dir_fd)
- /*[clinic end generated code: output=59e803072cf41230 input=faa61c847912c850]*/
- {
- return internal_rename(src, dst, src_dir_fd, dst_dir_fd, 0);
- }
- /*[clinic input]
- os.replace = os.rename
- Rename a file or directory, overwriting the destination.
- If either src_dir_fd or dst_dir_fd is not None, it should be a file
- descriptor open to a directory, and the respective path string (src or dst)
- should be relative; the path will then be relative to that directory.
- src_dir_fd and dst_dir_fd, may not be implemented on your platform.
- If they are unavailable, using them will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_replace_impl(PyObject *module, path_t *src, path_t *dst, int src_dir_fd,
- int dst_dir_fd)
- /*[clinic end generated code: output=1968c02e7857422b input=c003f0def43378ef]*/
- {
- return internal_rename(src, dst, src_dir_fd, dst_dir_fd, 1);
- }
- /*[clinic input]
- os.rmdir
- path: path_t
- *
- dir_fd: dir_fd(requires='unlinkat') = None
- Remove a directory.
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_rmdir_impl(PyObject *module, path_t *path, int dir_fd)
- /*[clinic end generated code: output=080eb54f506e8301 input=38c8b375ca34a7e2]*/
- {
- int result;
- #ifdef HAVE_UNLINKAT
- int unlinkat_unavailable = 0;
- #endif
- if (PySys_Audit("os.rmdir", "Oi", path->object,
- dir_fd == DEFAULT_DIR_FD ? -1 : dir_fd) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- #ifdef MS_WINDOWS
- /* Windows, success=1, UNIX, success=0 */
- result = !RemoveDirectoryW(path->wide);
- #else
- #ifdef HAVE_UNLINKAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_UNLINKAT_RUNTIME) {
- result = unlinkat(dir_fd, path->narrow, AT_REMOVEDIR);
- } else {
- unlinkat_unavailable = 1;
- result = -1;
- }
- } else
- #endif
- result = rmdir(path->narrow);
- #endif
- Py_END_ALLOW_THREADS
- #ifdef HAVE_UNLINKAT
- if (unlinkat_unavailable) {
- argument_unavailable_error("rmdir", "dir_fd");
- return NULL;
- }
- #endif
- if (result)
- return path_error(path);
- Py_RETURN_NONE;
- }
- #ifdef HAVE_SYSTEM
- #ifdef MS_WINDOWS
- /*[clinic input]
- os.system -> long
- command: Py_UNICODE
- Execute the command in a subshell.
- [clinic start generated code]*/
- static long
- os_system_impl(PyObject *module, const Py_UNICODE *command)
- /*[clinic end generated code: output=5b7c3599c068ca42 input=303f5ce97df606b0]*/
- {
- long result;
- if (PySys_Audit("os.system", "(u)", command) < 0) {
- return -1;
- }
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- result = _wsystem(command);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- return result;
- }
- #else /* MS_WINDOWS */
- /*[clinic input]
- os.system -> long
- command: FSConverter
- Execute the command in a subshell.
- [clinic start generated code]*/
- static long
- os_system_impl(PyObject *module, PyObject *command)
- /*[clinic end generated code: output=290fc437dd4f33a0 input=86a58554ba6094af]*/
- {
- long result;
- const char *bytes = PyBytes_AsString(command);
- if (PySys_Audit("os.system", "(O)", command) < 0) {
- return -1;
- }
- Py_BEGIN_ALLOW_THREADS
- result = system(bytes);
- Py_END_ALLOW_THREADS
- return result;
- }
- #endif
- #endif /* HAVE_SYSTEM */
- #ifdef HAVE_UMASK
- /*[clinic input]
- os.umask
- mask: int
- /
- Set the current numeric umask and return the previous umask.
- [clinic start generated code]*/
- static PyObject *
- os_umask_impl(PyObject *module, int mask)
- /*[clinic end generated code: output=a2e33ce3bc1a6e33 input=ab6bfd9b24d8a7e8]*/
- {
- int i = (int)umask(mask);
- if (i < 0)
- return posix_error();
- return PyLong_FromLong((long)i);
- }
- #endif
- #ifdef MS_WINDOWS
- /* override the default DeleteFileW behavior so that directory
- symlinks can be removed with this function, the same as with
- Unix symlinks */
- BOOL WINAPI Py_DeleteFileW(LPCWSTR lpFileName)
- {
- WIN32_FILE_ATTRIBUTE_DATA info;
- WIN32_FIND_DATAW find_data;
- HANDLE find_data_handle;
- int is_directory = 0;
- int is_link = 0;
- if (GetFileAttributesExW(lpFileName, GetFileExInfoStandard, &info)) {
- is_directory = info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
- /* Get WIN32_FIND_DATA structure for the path to determine if
- it is a symlink */
- if(is_directory &&
- info.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
- find_data_handle = FindFirstFileW(lpFileName, &find_data);
- if(find_data_handle != INVALID_HANDLE_VALUE) {
- /* IO_REPARSE_TAG_SYMLINK if it is a symlink and
- IO_REPARSE_TAG_MOUNT_POINT if it is a junction point. */
- is_link = find_data.dwReserved0 == IO_REPARSE_TAG_SYMLINK ||
- find_data.dwReserved0 == IO_REPARSE_TAG_MOUNT_POINT;
- FindClose(find_data_handle);
- }
- }
- }
- if (is_directory && is_link)
- return RemoveDirectoryW(lpFileName);
- return DeleteFileW(lpFileName);
- }
- #endif /* MS_WINDOWS */
- /*[clinic input]
- os.unlink
- path: path_t
- *
- dir_fd: dir_fd(requires='unlinkat')=None
- Remove a file (same as remove()).
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_unlink_impl(PyObject *module, path_t *path, int dir_fd)
- /*[clinic end generated code: output=621797807b9963b1 input=d7bcde2b1b2a2552]*/
- {
- int result;
- #ifdef HAVE_UNLINKAT
- int unlinkat_unavailable = 0;
- #endif
- if (PySys_Audit("os.remove", "Oi", path->object,
- dir_fd == DEFAULT_DIR_FD ? -1 : dir_fd) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef MS_WINDOWS
- /* Windows, success=1, UNIX, success=0 */
- result = !Py_DeleteFileW(path->wide);
- #else
- #ifdef HAVE_UNLINKAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_UNLINKAT_RUNTIME) {
- result = unlinkat(dir_fd, path->narrow, 0);
- } else {
- unlinkat_unavailable = 1;
- }
- } else
- #endif /* HAVE_UNLINKAT */
- result = unlink(path->narrow);
- #endif
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- #ifdef HAVE_UNLINKAT
- if (unlinkat_unavailable) {
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (result)
- return path_error(path);
- Py_RETURN_NONE;
- }
- /*[clinic input]
- os.remove = os.unlink
- Remove a file (same as unlink()).
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_remove_impl(PyObject *module, path_t *path, int dir_fd)
- /*[clinic end generated code: output=a8535b28f0068883 input=e05c5ab55cd30983]*/
- {
- return os_unlink_impl(module, path, dir_fd);
- }
- static PyStructSequence_Field uname_result_fields[] = {
- {"sysname", "operating system name"},
- {"nodename", "name of machine on network (implementation-defined)"},
- {"release", "operating system release"},
- {"version", "operating system version"},
- {"machine", "hardware identifier"},
- {NULL}
- };
- PyDoc_STRVAR(uname_result__doc__,
- "uname_result: Result from os.uname().\n\n\
- This object may be accessed either as a tuple of\n\
- (sysname, nodename, release, version, machine),\n\
- or via the attributes sysname, nodename, release, version, and machine.\n\
- \n\
- See os.uname for more information.");
- static PyStructSequence_Desc uname_result_desc = {
- MODNAME ".uname_result", /* name */
- uname_result__doc__, /* doc */
- uname_result_fields,
- 5
- };
- #ifdef HAVE_UNAME
- /*[clinic input]
- os.uname
- Return an object identifying the current operating system.
- The object behaves like a named tuple with the following fields:
- (sysname, nodename, release, version, machine)
- [clinic start generated code]*/
- static PyObject *
- os_uname_impl(PyObject *module)
- /*[clinic end generated code: output=e6a49cf1a1508a19 input=e68bd246db3043ed]*/
- {
- struct utsname u;
- int res;
- PyObject *value;
- Py_BEGIN_ALLOW_THREADS
- res = uname(&u);
- Py_END_ALLOW_THREADS
- if (res < 0)
- return posix_error();
- PyObject *UnameResultType = get_posix_state(module)->UnameResultType;
- value = PyStructSequence_New((PyTypeObject *)UnameResultType);
- if (value == NULL)
- return NULL;
- #define SET(i, field) \
- { \
- PyObject *o = PyUnicode_DecodeFSDefault(field); \
- if (!o) { \
- Py_DECREF(value); \
- return NULL; \
- } \
- PyStructSequence_SET_ITEM(value, i, o); \
- } \
- SET(0, u.sysname);
- SET(1, u.nodename);
- SET(2, u.release);
- SET(3, u.version);
- SET(4, u.machine);
- #undef SET
- return value;
- }
- #endif /* HAVE_UNAME */
- typedef struct {
- int now;
- time_t atime_s;
- long atime_ns;
- time_t mtime_s;
- long mtime_ns;
- } utime_t;
- /*
- * these macros assume that "ut" is a pointer to a utime_t
- * they also intentionally leak the declaration of a pointer named "time"
- */
- #define UTIME_TO_TIMESPEC \
- struct timespec ts[2]; \
- struct timespec *time; \
- if (ut->now) \
- time = NULL; \
- else { \
- ts[0].tv_sec = ut->atime_s; \
- ts[0].tv_nsec = ut->atime_ns; \
- ts[1].tv_sec = ut->mtime_s; \
- ts[1].tv_nsec = ut->mtime_ns; \
- time = ts; \
- } \
- #define UTIME_TO_TIMEVAL \
- struct timeval tv[2]; \
- struct timeval *time; \
- if (ut->now) \
- time = NULL; \
- else { \
- tv[0].tv_sec = ut->atime_s; \
- tv[0].tv_usec = ut->atime_ns / 1000; \
- tv[1].tv_sec = ut->mtime_s; \
- tv[1].tv_usec = ut->mtime_ns / 1000; \
- time = tv; \
- } \
- #define UTIME_TO_UTIMBUF \
- struct utimbuf u; \
- struct utimbuf *time; \
- if (ut->now) \
- time = NULL; \
- else { \
- u.actime = ut->atime_s; \
- u.modtime = ut->mtime_s; \
- time = &u; \
- }
- #define UTIME_TO_TIME_T \
- time_t timet[2]; \
- time_t *time; \
- if (ut->now) \
- time = NULL; \
- else { \
- timet[0] = ut->atime_s; \
- timet[1] = ut->mtime_s; \
- time = timet; \
- } \
- #if defined(HAVE_FUTIMESAT) || defined(HAVE_UTIMENSAT)
- static int
- utime_dir_fd(utime_t *ut, int dir_fd, const char *path, int follow_symlinks)
- {
- #if defined(__APPLE__) && defined(HAVE_UTIMENSAT)
- if (HAVE_UTIMENSAT_RUNTIME) {
- int flags = follow_symlinks ? 0 : AT_SYMLINK_NOFOLLOW;
- UTIME_TO_TIMESPEC;
- return utimensat(dir_fd, path, time, flags);
- } else {
- errno = ENOSYS;
- return -1;
- }
- #elif defined(HAVE_UTIMENSAT)
- int flags = follow_symlinks ? 0 : AT_SYMLINK_NOFOLLOW;
- UTIME_TO_TIMESPEC;
- return utimensat(dir_fd, path, time, flags);
- #elif defined(HAVE_FUTIMESAT)
- UTIME_TO_TIMEVAL;
- /*
- * follow_symlinks will never be false here;
- * we only allow !follow_symlinks and dir_fd together
- * if we have utimensat()
- */
- assert(follow_symlinks);
- return futimesat(dir_fd, path, time);
- #endif
- }
- #define FUTIMENSAT_DIR_FD_CONVERTER dir_fd_converter
- #else
- #define FUTIMENSAT_DIR_FD_CONVERTER dir_fd_unavailable
- #endif
- #if defined(HAVE_FUTIMES) || defined(HAVE_FUTIMENS)
- static int
- utime_fd(utime_t *ut, int fd)
- {
- #ifdef HAVE_FUTIMENS
- if (HAVE_FUTIMENS_RUNTIME) {
- UTIME_TO_TIMESPEC;
- return futimens(fd, time);
- } else
- #ifndef HAVE_FUTIMES
- {
- /* Not sure if this can happen */
- PyErr_SetString(
- PyExc_RuntimeError,
- "neither futimens nor futimes are supported"
- " on this system");
- return -1;
- }
- #endif
- #endif
- #ifdef HAVE_FUTIMES
- {
- UTIME_TO_TIMEVAL;
- return futimes(fd, time);
- }
- #endif
- }
- #define PATH_UTIME_HAVE_FD 1
- #else
- #define PATH_UTIME_HAVE_FD 0
- #endif
- #if defined(HAVE_UTIMENSAT) || defined(HAVE_LUTIMES)
- # define UTIME_HAVE_NOFOLLOW_SYMLINKS
- #endif
- #ifdef UTIME_HAVE_NOFOLLOW_SYMLINKS
- static int
- utime_nofollow_symlinks(utime_t *ut, const char *path)
- {
- #ifdef HAVE_UTIMENSAT
- if (HAVE_UTIMENSAT_RUNTIME) {
- UTIME_TO_TIMESPEC;
- return utimensat(DEFAULT_DIR_FD, path, time, AT_SYMLINK_NOFOLLOW);
- } else
- #ifndef HAVE_LUTIMES
- {
- /* Not sure if this can happen */
- PyErr_SetString(
- PyExc_RuntimeError,
- "neither utimensat nor lutimes are supported"
- " on this system");
- return -1;
- }
- #endif
- #endif
- #ifdef HAVE_LUTIMES
- {
- UTIME_TO_TIMEVAL;
- return lutimes(path, time);
- }
- #endif
- }
- #endif
- #ifndef MS_WINDOWS
- static int
- utime_default(utime_t *ut, const char *path)
- {
- #if defined(__APPLE__) && defined(HAVE_UTIMENSAT)
- if (HAVE_UTIMENSAT_RUNTIME) {
- UTIME_TO_TIMESPEC;
- return utimensat(DEFAULT_DIR_FD, path, time, 0);
- } else {
- UTIME_TO_TIMEVAL;
- return utimes(path, time);
- }
- #elif defined(HAVE_UTIMENSAT)
- UTIME_TO_TIMESPEC;
- return utimensat(DEFAULT_DIR_FD, path, time, 0);
- #elif defined(HAVE_UTIMES)
- UTIME_TO_TIMEVAL;
- return utimes(path, time);
- #elif defined(HAVE_UTIME_H)
- UTIME_TO_UTIMBUF;
- return utime(path, time);
- #else
- UTIME_TO_TIME_T;
- return utime(path, time);
- #endif
- }
- #endif
- static int
- split_py_long_to_s_and_ns(PyObject *module, PyObject *py_long, time_t *s, long *ns)
- {
- int result = 0;
- PyObject *divmod;
- divmod = PyNumber_Divmod(py_long, get_posix_state(module)->billion);
- if (!divmod)
- goto exit;
- if (!PyTuple_Check(divmod) || PyTuple_GET_SIZE(divmod) != 2) {
- PyErr_Format(PyExc_TypeError,
- "%.200s.__divmod__() must return a 2-tuple, not %.200s",
- _PyType_Name(Py_TYPE(py_long)), _PyType_Name(Py_TYPE(divmod)));
- goto exit;
- }
- *s = _PyLong_AsTime_t(PyTuple_GET_ITEM(divmod, 0));
- if ((*s == -1) && PyErr_Occurred())
- goto exit;
- *ns = PyLong_AsLong(PyTuple_GET_ITEM(divmod, 1));
- if ((*ns == -1) && PyErr_Occurred())
- goto exit;
- result = 1;
- exit:
- Py_XDECREF(divmod);
- return result;
- }
- /*[clinic input]
- os.utime
- path: path_t(allow_fd='PATH_UTIME_HAVE_FD')
- times: object = None
- *
- ns: object = NULL
- dir_fd: dir_fd(requires='futimensat') = None
- follow_symlinks: bool=True
- # "utime(path, times=None, *[, ns], dir_fd=None, follow_symlinks=True)\n\
- Set the access and modified time of path.
- path may always be specified as a string.
- On some platforms, path may also be specified as an open file descriptor.
- If this functionality is unavailable, using it raises an exception.
- If times is not None, it must be a tuple (atime, mtime);
- atime and mtime should be expressed as float seconds since the epoch.
- If ns is specified, it must be a tuple (atime_ns, mtime_ns);
- atime_ns and mtime_ns should be expressed as integer nanoseconds
- since the epoch.
- If times is None and ns is unspecified, utime uses the current time.
- Specifying tuples for both times and ns is an error.
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- If follow_symlinks is False, and the last element of the path is a symbolic
- link, utime will modify the symbolic link itself instead of the file the
- link points to.
- It is an error to use dir_fd or follow_symlinks when specifying path
- as an open file descriptor.
- dir_fd and follow_symlinks may not be available on your platform.
- If they are unavailable, using them will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_utime_impl(PyObject *module, path_t *path, PyObject *times, PyObject *ns,
- int dir_fd, int follow_symlinks)
- /*[clinic end generated code: output=cfcac69d027b82cf input=2fbd62a2f228f8f4]*/
- {
- #ifdef MS_WINDOWS
- HANDLE hFile;
- FILETIME atime, mtime;
- #else
- int result;
- #endif
- utime_t utime;
- memset(&utime, 0, sizeof(utime_t));
- if (times != Py_None && ns) {
- PyErr_SetString(PyExc_ValueError,
- "utime: you may specify either 'times'"
- " or 'ns' but not both");
- return NULL;
- }
- if (times != Py_None) {
- time_t a_sec, m_sec;
- long a_nsec, m_nsec;
- if (!PyTuple_CheckExact(times) || (PyTuple_Size(times) != 2)) {
- PyErr_SetString(PyExc_TypeError,
- "utime: 'times' must be either"
- " a tuple of two ints or None");
- return NULL;
- }
- utime.now = 0;
- if (_PyTime_ObjectToTimespec(PyTuple_GET_ITEM(times, 0),
- &a_sec, &a_nsec, _PyTime_ROUND_FLOOR) == -1 ||
- _PyTime_ObjectToTimespec(PyTuple_GET_ITEM(times, 1),
- &m_sec, &m_nsec, _PyTime_ROUND_FLOOR) == -1) {
- return NULL;
- }
- utime.atime_s = a_sec;
- utime.atime_ns = a_nsec;
- utime.mtime_s = m_sec;
- utime.mtime_ns = m_nsec;
- }
- else if (ns) {
- if (!PyTuple_CheckExact(ns) || (PyTuple_Size(ns) != 2)) {
- PyErr_SetString(PyExc_TypeError,
- "utime: 'ns' must be a tuple of two ints");
- return NULL;
- }
- utime.now = 0;
- if (!split_py_long_to_s_and_ns(module, PyTuple_GET_ITEM(ns, 0),
- &utime.atime_s, &utime.atime_ns) ||
- !split_py_long_to_s_and_ns(module, PyTuple_GET_ITEM(ns, 1),
- &utime.mtime_s, &utime.mtime_ns)) {
- return NULL;
- }
- }
- else {
- /* times and ns are both None/unspecified. use "now". */
- utime.now = 1;
- }
- #if !defined(UTIME_HAVE_NOFOLLOW_SYMLINKS)
- if (follow_symlinks_specified("utime", follow_symlinks))
- return NULL;
- #endif
- if (path_and_dir_fd_invalid("utime", path, dir_fd) ||
- dir_fd_and_fd_invalid("utime", dir_fd, path->fd) ||
- fd_and_follow_symlinks_invalid("utime", path->fd, follow_symlinks))
- return NULL;
- #if !defined(HAVE_UTIMENSAT)
- if ((dir_fd != DEFAULT_DIR_FD) && (!follow_symlinks)) {
- PyErr_SetString(PyExc_ValueError,
- "utime: cannot use dir_fd and follow_symlinks "
- "together on this platform");
- return NULL;
- }
- #endif
- if (PySys_Audit("os.utime", "OOOi", path->object, times, ns ? ns : Py_None,
- dir_fd == DEFAULT_DIR_FD ? -1 : dir_fd) < 0) {
- return NULL;
- }
- #ifdef MS_WINDOWS
- Py_BEGIN_ALLOW_THREADS
- hFile = CreateFileW(path->wide, FILE_WRITE_ATTRIBUTES, 0,
- NULL, OPEN_EXISTING,
- FILE_FLAG_BACKUP_SEMANTICS, NULL);
- Py_END_ALLOW_THREADS
- if (hFile == INVALID_HANDLE_VALUE) {
- path_error(path);
- return NULL;
- }
- if (utime.now) {
- GetSystemTimeAsFileTime(&mtime);
- atime = mtime;
- }
- else {
- _Py_time_t_to_FILE_TIME(utime.atime_s, utime.atime_ns, &atime);
- _Py_time_t_to_FILE_TIME(utime.mtime_s, utime.mtime_ns, &mtime);
- }
- if (!SetFileTime(hFile, NULL, &atime, &mtime)) {
- /* Avoid putting the file name into the error here,
- as that may confuse the user into believing that
- something is wrong with the file, when it also
- could be the time stamp that gives a problem. */
- PyErr_SetFromWindowsErr(0);
- CloseHandle(hFile);
- return NULL;
- }
- CloseHandle(hFile);
- #else /* MS_WINDOWS */
- Py_BEGIN_ALLOW_THREADS
- #ifdef UTIME_HAVE_NOFOLLOW_SYMLINKS
- if ((!follow_symlinks) && (dir_fd == DEFAULT_DIR_FD))
- result = utime_nofollow_symlinks(&utime, path->narrow);
- else
- #endif
- #if defined(HAVE_FUTIMESAT) || defined(HAVE_UTIMENSAT)
- if ((dir_fd != DEFAULT_DIR_FD) || (!follow_symlinks)) {
- result = utime_dir_fd(&utime, dir_fd, path->narrow, follow_symlinks);
- } else
- #endif
- #if defined(HAVE_FUTIMES) || defined(HAVE_FUTIMENS)
- if (path->fd != -1)
- result = utime_fd(&utime, path->fd);
- else
- #endif
- result = utime_default(&utime, path->narrow);
- Py_END_ALLOW_THREADS
- #if defined(__APPLE__) && defined(HAVE_UTIMENSAT)
- /* See utime_dir_fd implementation */
- if (result == -1 && errno == ENOSYS) {
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (result < 0) {
- /* see previous comment about not putting filename in error here */
- posix_error();
- return NULL;
- }
- #endif /* MS_WINDOWS */
- Py_RETURN_NONE;
- }
- /* Process operations */
- /*[clinic input]
- os._exit
- status: int
- Exit to the system with specified status, without normal exit processing.
- [clinic start generated code]*/
- static PyObject *
- os__exit_impl(PyObject *module, int status)
- /*[clinic end generated code: output=116e52d9c2260d54 input=5e6d57556b0c4a62]*/
- {
- _exit(status);
- return NULL; /* Make gcc -Wall happy */
- }
- #if defined(HAVE_WEXECV) || defined(HAVE_WSPAWNV)
- #define EXECV_CHAR wchar_t
- #else
- #define EXECV_CHAR char
- #endif
- #if defined(HAVE_EXECV) || defined(HAVE_SPAWNV) || defined(HAVE_RTPSPAWN)
- static void
- free_string_array(EXECV_CHAR **array, Py_ssize_t count)
- {
- Py_ssize_t i;
- for (i = 0; i < count; i++)
- PyMem_Free(array[i]);
- PyMem_Free(array);
- }
- static int
- fsconvert_strdup(PyObject *o, EXECV_CHAR **out)
- {
- Py_ssize_t size;
- PyObject *ub;
- int result = 0;
- #if defined(HAVE_WEXECV) || defined(HAVE_WSPAWNV)
- if (!PyUnicode_FSDecoder(o, &ub))
- return 0;
- *out = PyUnicode_AsWideCharString(ub, &size);
- if (*out)
- result = 1;
- #else
- if (!PyUnicode_FSConverter(o, &ub))
- return 0;
- size = PyBytes_GET_SIZE(ub);
- *out = PyMem_Malloc(size + 1);
- if (*out) {
- memcpy(*out, PyBytes_AS_STRING(ub), size + 1);
- result = 1;
- } else
- PyErr_NoMemory();
- #endif
- Py_DECREF(ub);
- return result;
- }
- #endif
- #if defined(HAVE_EXECV) || defined (HAVE_FEXECVE) || defined(HAVE_RTPSPAWN)
- static EXECV_CHAR**
- parse_envlist(PyObject* env, Py_ssize_t *envc_ptr)
- {
- Py_ssize_t i, pos, envc;
- PyObject *keys=NULL, *vals=NULL;
- PyObject *key2, *val2, *keyval;
- EXECV_CHAR **envlist;
- i = PyMapping_Size(env);
- if (i < 0)
- return NULL;
- envlist = PyMem_NEW(EXECV_CHAR *, i + 1);
- if (envlist == NULL) {
- PyErr_NoMemory();
- return NULL;
- }
- envc = 0;
- keys = PyMapping_Keys(env);
- if (!keys)
- goto error;
- vals = PyMapping_Values(env);
- if (!vals)
- goto error;
- if (!PyList_Check(keys) || !PyList_Check(vals)) {
- PyErr_Format(PyExc_TypeError,
- "env.keys() or env.values() is not a list");
- goto error;
- }
- for (pos = 0; pos < i; pos++) {
- PyObject *key = PyList_GetItem(keys, pos); // Borrowed ref.
- if (key == NULL) {
- goto error;
- }
- PyObject *val = PyList_GetItem(vals, pos); // Borrowed ref.
- if (val == NULL) {
- goto error;
- }
- #if defined(HAVE_WEXECV) || defined(HAVE_WSPAWNV)
- if (!PyUnicode_FSDecoder(key, &key2))
- goto error;
- if (!PyUnicode_FSDecoder(val, &val2)) {
- Py_DECREF(key2);
- goto error;
- }
- /* Search from index 1 because on Windows starting '=' is allowed for
- defining hidden environment variables. */
- if (PyUnicode_GET_LENGTH(key2) == 0 ||
- PyUnicode_FindChar(key2, '=', 1, PyUnicode_GET_LENGTH(key2), 1) != -1)
- {
- PyErr_SetString(PyExc_ValueError, "illegal environment variable name");
- Py_DECREF(key2);
- Py_DECREF(val2);
- goto error;
- }
- keyval = PyUnicode_FromFormat("%U=%U", key2, val2);
- #else
- if (!PyUnicode_FSConverter(key, &key2))
- goto error;
- if (!PyUnicode_FSConverter(val, &val2)) {
- Py_DECREF(key2);
- goto error;
- }
- if (PyBytes_GET_SIZE(key2) == 0 ||
- strchr(PyBytes_AS_STRING(key2) + 1, '=') != NULL)
- {
- PyErr_SetString(PyExc_ValueError, "illegal environment variable name");
- Py_DECREF(key2);
- Py_DECREF(val2);
- goto error;
- }
- keyval = PyBytes_FromFormat("%s=%s", PyBytes_AS_STRING(key2),
- PyBytes_AS_STRING(val2));
- #endif
- Py_DECREF(key2);
- Py_DECREF(val2);
- if (!keyval)
- goto error;
- if (!fsconvert_strdup(keyval, &envlist[envc++])) {
- Py_DECREF(keyval);
- goto error;
- }
- Py_DECREF(keyval);
- }
- Py_DECREF(vals);
- Py_DECREF(keys);
- envlist[envc] = 0;
- *envc_ptr = envc;
- return envlist;
- error:
- Py_XDECREF(keys);
- Py_XDECREF(vals);
- free_string_array(envlist, envc);
- return NULL;
- }
- static EXECV_CHAR**
- parse_arglist(PyObject* argv, Py_ssize_t *argc)
- {
- int i;
- EXECV_CHAR **argvlist = PyMem_NEW(EXECV_CHAR *, *argc+1);
- if (argvlist == NULL) {
- PyErr_NoMemory();
- return NULL;
- }
- for (i = 0; i < *argc; i++) {
- PyObject* item = PySequence_ITEM(argv, i);
- if (item == NULL)
- goto fail;
- if (!fsconvert_strdup(item, &argvlist[i])) {
- Py_DECREF(item);
- goto fail;
- }
- Py_DECREF(item);
- }
- argvlist[*argc] = NULL;
- return argvlist;
- fail:
- *argc = i;
- free_string_array(argvlist, *argc);
- return NULL;
- }
- #endif
- #ifdef HAVE_EXECV
- /*[clinic input]
- os.execv
- path: path_t
- Path of executable file.
- argv: object
- Tuple or list of strings.
- /
- Execute an executable path with arguments, replacing current process.
- [clinic start generated code]*/
- static PyObject *
- os_execv_impl(PyObject *module, path_t *path, PyObject *argv)
- /*[clinic end generated code: output=3b52fec34cd0dafd input=9bac31efae07dac7]*/
- {
- EXECV_CHAR **argvlist;
- Py_ssize_t argc;
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (!_PyInterpreterState_HasFeature(interp, Py_RTFLAGS_EXEC)) {
- PyErr_SetString(PyExc_RuntimeError,
- "exec not supported for isolated subinterpreters");
- return NULL;
- }
- /* execv has two arguments: (path, argv), where
- argv is a list or tuple of strings. */
- if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
- PyErr_SetString(PyExc_TypeError,
- "execv() arg 2 must be a tuple or list");
- return NULL;
- }
- argc = PySequence_Size(argv);
- if (argc < 1) {
- PyErr_SetString(PyExc_ValueError, "execv() arg 2 must not be empty");
- return NULL;
- }
- argvlist = parse_arglist(argv, &argc);
- if (argvlist == NULL) {
- return NULL;
- }
- if (!argvlist[0][0]) {
- PyErr_SetString(PyExc_ValueError,
- "execv() arg 2 first element cannot be empty");
- free_string_array(argvlist, argc);
- return NULL;
- }
- if (PySys_Audit("os.exec", "OOO", path->object, argv, Py_None) < 0) {
- free_string_array(argvlist, argc);
- return NULL;
- }
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef HAVE_WEXECV
- _wexecv(path->wide, argvlist);
- #else
- execv(path->narrow, argvlist);
- #endif
- _Py_END_SUPPRESS_IPH
- /* If we get here it's definitely an error */
- posix_error();
- free_string_array(argvlist, argc);
- return NULL;
- }
- /*[clinic input]
- os.execve
- path: path_t(allow_fd='PATH_HAVE_FEXECVE')
- Path of executable file.
- argv: object
- Tuple or list of strings.
- env: object
- Dictionary of strings mapping to strings.
- Execute an executable path with arguments, replacing current process.
- [clinic start generated code]*/
- static PyObject *
- os_execve_impl(PyObject *module, path_t *path, PyObject *argv, PyObject *env)
- /*[clinic end generated code: output=ff9fa8e4da8bde58 input=626804fa092606d9]*/
- {
- EXECV_CHAR **argvlist = NULL;
- EXECV_CHAR **envlist;
- Py_ssize_t argc, envc;
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (!_PyInterpreterState_HasFeature(interp, Py_RTFLAGS_EXEC)) {
- PyErr_SetString(PyExc_RuntimeError,
- "exec not supported for isolated subinterpreters");
- return NULL;
- }
- /* execve has three arguments: (path, argv, env), where
- argv is a list or tuple of strings and env is a dictionary
- like posix.environ. */
- if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
- PyErr_SetString(PyExc_TypeError,
- "execve: argv must be a tuple or list");
- goto fail_0;
- }
- argc = PySequence_Size(argv);
- if (argc < 1) {
- PyErr_SetString(PyExc_ValueError, "execve: argv must not be empty");
- return NULL;
- }
- if (!PyMapping_Check(env)) {
- PyErr_SetString(PyExc_TypeError,
- "execve: environment must be a mapping object");
- goto fail_0;
- }
- argvlist = parse_arglist(argv, &argc);
- if (argvlist == NULL) {
- goto fail_0;
- }
- if (!argvlist[0][0]) {
- PyErr_SetString(PyExc_ValueError,
- "execve: argv first element cannot be empty");
- goto fail_0;
- }
- envlist = parse_envlist(env, &envc);
- if (envlist == NULL)
- goto fail_0;
- if (PySys_Audit("os.exec", "OOO", path->object, argv, env) < 0) {
- goto fail_1;
- }
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef HAVE_FEXECVE
- if (path->fd > -1)
- fexecve(path->fd, argvlist, envlist);
- else
- #endif
- #ifdef HAVE_WEXECV
- _wexecve(path->wide, argvlist, envlist);
- #else
- execve(path->narrow, argvlist, envlist);
- #endif
- _Py_END_SUPPRESS_IPH
- /* If we get here it's definitely an error */
- posix_path_error(path);
- fail_1:
- free_string_array(envlist, envc);
- fail_0:
- if (argvlist)
- free_string_array(argvlist, argc);
- return NULL;
- }
- #endif /* HAVE_EXECV */
- #ifdef HAVE_POSIX_SPAWN
- enum posix_spawn_file_actions_identifier {
- POSIX_SPAWN_OPEN,
- POSIX_SPAWN_CLOSE,
- POSIX_SPAWN_DUP2
- };
- #if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM)
- static int
- convert_sched_param(PyObject *module, PyObject *param, struct sched_param *res);
- #endif
- static int
- parse_posix_spawn_flags(PyObject *module, const char *func_name, PyObject *setpgroup,
- int resetids, int setsid, PyObject *setsigmask,
- PyObject *setsigdef, PyObject *scheduler,
- posix_spawnattr_t *attrp)
- {
- long all_flags = 0;
- errno = posix_spawnattr_init(attrp);
- if (errno) {
- posix_error();
- return -1;
- }
- if (setpgroup) {
- pid_t pgid = PyLong_AsPid(setpgroup);
- if (pgid == (pid_t)-1 && PyErr_Occurred()) {
- goto fail;
- }
- errno = posix_spawnattr_setpgroup(attrp, pgid);
- if (errno) {
- posix_error();
- goto fail;
- }
- all_flags |= POSIX_SPAWN_SETPGROUP;
- }
- if (resetids) {
- all_flags |= POSIX_SPAWN_RESETIDS;
- }
- if (setsid) {
- #ifdef HAVE_POSIX_SPAWN_SETSID_RUNTIME
- if (HAVE_POSIX_SPAWN_SETSID_RUNTIME) {
- #endif
- #ifdef POSIX_SPAWN_SETSID
- all_flags |= POSIX_SPAWN_SETSID;
- #elif defined(POSIX_SPAWN_SETSID_NP)
- all_flags |= POSIX_SPAWN_SETSID_NP;
- #else
- argument_unavailable_error(func_name, "setsid");
- return -1;
- #endif
- #ifdef HAVE_POSIX_SPAWN_SETSID_RUNTIME
- } else {
- argument_unavailable_error(func_name, "setsid");
- return -1;
- }
- #endif /* HAVE_POSIX_SPAWN_SETSID_RUNTIME */
- }
- #ifdef HAVE_SIGSET_T
- if (setsigmask) {
- sigset_t set;
- if (!_Py_Sigset_Converter(setsigmask, &set)) {
- goto fail;
- }
- errno = posix_spawnattr_setsigmask(attrp, &set);
- if (errno) {
- posix_error();
- goto fail;
- }
- all_flags |= POSIX_SPAWN_SETSIGMASK;
- }
- if (setsigdef) {
- sigset_t set;
- if (!_Py_Sigset_Converter(setsigdef, &set)) {
- goto fail;
- }
- errno = posix_spawnattr_setsigdefault(attrp, &set);
- if (errno) {
- posix_error();
- goto fail;
- }
- all_flags |= POSIX_SPAWN_SETSIGDEF;
- }
- #else
- if (setsigmask || setsigdef) {
- PyErr_SetString(PyExc_NotImplementedError,
- "sigset is not supported on this platform");
- goto fail;
- }
- #endif
- if (scheduler) {
- #ifdef POSIX_SPAWN_SETSCHEDULER
- PyObject *py_schedpolicy;
- PyObject *schedparam_obj;
- struct sched_param schedparam;
- if (!PyArg_ParseTuple(scheduler, "OO"
- ";A scheduler tuple must have two elements",
- &py_schedpolicy, &schedparam_obj)) {
- goto fail;
- }
- if (!convert_sched_param(module, schedparam_obj, &schedparam)) {
- goto fail;
- }
- if (py_schedpolicy != Py_None) {
- int schedpolicy = _PyLong_AsInt(py_schedpolicy);
- if (schedpolicy == -1 && PyErr_Occurred()) {
- goto fail;
- }
- errno = posix_spawnattr_setschedpolicy(attrp, schedpolicy);
- if (errno) {
- posix_error();
- goto fail;
- }
- all_flags |= POSIX_SPAWN_SETSCHEDULER;
- }
- errno = posix_spawnattr_setschedparam(attrp, &schedparam);
- if (errno) {
- posix_error();
- goto fail;
- }
- all_flags |= POSIX_SPAWN_SETSCHEDPARAM;
- #else
- PyErr_SetString(PyExc_NotImplementedError,
- "The scheduler option is not supported in this system.");
- goto fail;
- #endif
- }
- errno = posix_spawnattr_setflags(attrp, all_flags);
- if (errno) {
- posix_error();
- goto fail;
- }
- return 0;
- fail:
- (void)posix_spawnattr_destroy(attrp);
- return -1;
- }
- static int
- parse_file_actions(PyObject *file_actions,
- posix_spawn_file_actions_t *file_actionsp,
- PyObject *temp_buffer)
- {
- PyObject *seq;
- PyObject *file_action = NULL;
- PyObject *tag_obj;
- seq = PySequence_Fast(file_actions,
- "file_actions must be a sequence or None");
- if (seq == NULL) {
- return -1;
- }
- errno = posix_spawn_file_actions_init(file_actionsp);
- if (errno) {
- posix_error();
- Py_DECREF(seq);
- return -1;
- }
- for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(seq); ++i) {
- file_action = PySequence_Fast_GET_ITEM(seq, i);
- Py_INCREF(file_action);
- if (!PyTuple_Check(file_action) || !PyTuple_GET_SIZE(file_action)) {
- PyErr_SetString(PyExc_TypeError,
- "Each file_actions element must be a non-empty tuple");
- goto fail;
- }
- long tag = PyLong_AsLong(PyTuple_GET_ITEM(file_action, 0));
- if (tag == -1 && PyErr_Occurred()) {
- goto fail;
- }
- /* Populate the file_actions object */
- switch (tag) {
- case POSIX_SPAWN_OPEN: {
- int fd, oflag;
- PyObject *path;
- unsigned long mode;
- if (!PyArg_ParseTuple(file_action, "OiO&ik"
- ";A open file_action tuple must have 5 elements",
- &tag_obj, &fd, PyUnicode_FSConverter, &path,
- &oflag, &mode))
- {
- goto fail;
- }
- if (PyList_Append(temp_buffer, path)) {
- Py_DECREF(path);
- goto fail;
- }
- errno = posix_spawn_file_actions_addopen(file_actionsp,
- fd, PyBytes_AS_STRING(path), oflag, (mode_t)mode);
- if (errno) {
- posix_error();
- Py_DECREF(path);
- goto fail;
- }
- Py_DECREF(path);
- break;
- }
- case POSIX_SPAWN_CLOSE: {
- int fd;
- if (!PyArg_ParseTuple(file_action, "Oi"
- ";A close file_action tuple must have 2 elements",
- &tag_obj, &fd))
- {
- goto fail;
- }
- errno = posix_spawn_file_actions_addclose(file_actionsp, fd);
- if (errno) {
- posix_error();
- goto fail;
- }
- break;
- }
- case POSIX_SPAWN_DUP2: {
- int fd1, fd2;
- if (!PyArg_ParseTuple(file_action, "Oii"
- ";A dup2 file_action tuple must have 3 elements",
- &tag_obj, &fd1, &fd2))
- {
- goto fail;
- }
- errno = posix_spawn_file_actions_adddup2(file_actionsp,
- fd1, fd2);
- if (errno) {
- posix_error();
- goto fail;
- }
- break;
- }
- default: {
- PyErr_SetString(PyExc_TypeError,
- "Unknown file_actions identifier");
- goto fail;
- }
- }
- Py_DECREF(file_action);
- }
- Py_DECREF(seq);
- return 0;
- fail:
- Py_DECREF(seq);
- Py_DECREF(file_action);
- (void)posix_spawn_file_actions_destroy(file_actionsp);
- return -1;
- }
- static PyObject *
- py_posix_spawn(int use_posix_spawnp, PyObject *module, path_t *path, PyObject *argv,
- PyObject *env, PyObject *file_actions,
- PyObject *setpgroup, int resetids, int setsid, PyObject *setsigmask,
- PyObject *setsigdef, PyObject *scheduler)
- {
- const char *func_name = use_posix_spawnp ? "posix_spawnp" : "posix_spawn";
- EXECV_CHAR **argvlist = NULL;
- EXECV_CHAR **envlist = NULL;
- posix_spawn_file_actions_t file_actions_buf;
- posix_spawn_file_actions_t *file_actionsp = NULL;
- posix_spawnattr_t attr;
- posix_spawnattr_t *attrp = NULL;
- Py_ssize_t argc, envc;
- PyObject *result = NULL;
- PyObject *temp_buffer = NULL;
- pid_t pid;
- int err_code;
- /* posix_spawn and posix_spawnp have three arguments: (path, argv, env), where
- argv is a list or tuple of strings and env is a dictionary
- like posix.environ. */
- if (!PyList_Check(argv) && !PyTuple_Check(argv)) {
- PyErr_Format(PyExc_TypeError,
- "%s: argv must be a tuple or list", func_name);
- goto exit;
- }
- argc = PySequence_Size(argv);
- if (argc < 1) {
- PyErr_Format(PyExc_ValueError,
- "%s: argv must not be empty", func_name);
- return NULL;
- }
- if (!PyMapping_Check(env)) {
- PyErr_Format(PyExc_TypeError,
- "%s: environment must be a mapping object", func_name);
- goto exit;
- }
- argvlist = parse_arglist(argv, &argc);
- if (argvlist == NULL) {
- goto exit;
- }
- if (!argvlist[0][0]) {
- PyErr_Format(PyExc_ValueError,
- "%s: argv first element cannot be empty", func_name);
- goto exit;
- }
- envlist = parse_envlist(env, &envc);
- if (envlist == NULL) {
- goto exit;
- }
- if (file_actions != NULL && file_actions != Py_None) {
- /* There is a bug in old versions of glibc that makes some of the
- * helper functions for manipulating file actions not copy the provided
- * buffers. The problem is that posix_spawn_file_actions_addopen does not
- * copy the value of path for some old versions of glibc (<2.20).
- * The use of temp_buffer here is a workaround that keeps the
- * python objects that own the buffers alive until posix_spawn gets called.
- * Check https://bugs.python.org/issue33630 and
- * https://sourceware.org/bugzilla/show_bug.cgi?id=17048 for more info.*/
- temp_buffer = PyList_New(0);
- if (!temp_buffer) {
- goto exit;
- }
- if (parse_file_actions(file_actions, &file_actions_buf, temp_buffer)) {
- goto exit;
- }
- file_actionsp = &file_actions_buf;
- }
- if (parse_posix_spawn_flags(module, func_name, setpgroup, resetids, setsid,
- setsigmask, setsigdef, scheduler, &attr)) {
- goto exit;
- }
- attrp = &attr;
- if (PySys_Audit("os.posix_spawn", "OOO", path->object, argv, env) < 0) {
- goto exit;
- }
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef HAVE_POSIX_SPAWNP
- if (use_posix_spawnp) {
- err_code = posix_spawnp(&pid, path->narrow,
- file_actionsp, attrp, argvlist, envlist);
- }
- else
- #endif /* HAVE_POSIX_SPAWNP */
- {
- err_code = posix_spawn(&pid, path->narrow,
- file_actionsp, attrp, argvlist, envlist);
- }
- _Py_END_SUPPRESS_IPH
- if (err_code) {
- errno = err_code;
- PyErr_SetFromErrnoWithFilenameObject(PyExc_OSError, path->object);
- goto exit;
- }
- #ifdef _Py_MEMORY_SANITIZER
- __msan_unpoison(&pid, sizeof(pid));
- #endif
- result = PyLong_FromPid(pid);
- exit:
- if (file_actionsp) {
- (void)posix_spawn_file_actions_destroy(file_actionsp);
- }
- if (attrp) {
- (void)posix_spawnattr_destroy(attrp);
- }
- if (envlist) {
- free_string_array(envlist, envc);
- }
- if (argvlist) {
- free_string_array(argvlist, argc);
- }
- Py_XDECREF(temp_buffer);
- return result;
- }
- /*[clinic input]
- os.posix_spawn
- path: path_t
- Path of executable file.
- argv: object
- Tuple or list of strings.
- env: object
- Dictionary of strings mapping to strings.
- /
- *
- file_actions: object(c_default='NULL') = ()
- A sequence of file action tuples.
- setpgroup: object = NULL
- The pgroup to use with the POSIX_SPAWN_SETPGROUP flag.
- resetids: bool = False
- If the value is `true` the POSIX_SPAWN_RESETIDS will be activated.
- setsid: bool = False
- If the value is `true` the POSIX_SPAWN_SETSID or POSIX_SPAWN_SETSID_NP will be activated.
- setsigmask: object(c_default='NULL') = ()
- The sigmask to use with the POSIX_SPAWN_SETSIGMASK flag.
- setsigdef: object(c_default='NULL') = ()
- The sigmask to use with the POSIX_SPAWN_SETSIGDEF flag.
- scheduler: object = NULL
- A tuple with the scheduler policy (optional) and parameters.
- Execute the program specified by path in a new process.
- [clinic start generated code]*/
- static PyObject *
- os_posix_spawn_impl(PyObject *module, path_t *path, PyObject *argv,
- PyObject *env, PyObject *file_actions,
- PyObject *setpgroup, int resetids, int setsid,
- PyObject *setsigmask, PyObject *setsigdef,
- PyObject *scheduler)
- /*[clinic end generated code: output=14a1098c566bc675 input=808aed1090d84e33]*/
- {
- return py_posix_spawn(0, module, path, argv, env, file_actions,
- setpgroup, resetids, setsid, setsigmask, setsigdef,
- scheduler);
- }
- #endif /* HAVE_POSIX_SPAWN */
- #ifdef HAVE_POSIX_SPAWNP
- /*[clinic input]
- os.posix_spawnp
- path: path_t
- Path of executable file.
- argv: object
- Tuple or list of strings.
- env: object
- Dictionary of strings mapping to strings.
- /
- *
- file_actions: object(c_default='NULL') = ()
- A sequence of file action tuples.
- setpgroup: object = NULL
- The pgroup to use with the POSIX_SPAWN_SETPGROUP flag.
- resetids: bool = False
- If the value is `True` the POSIX_SPAWN_RESETIDS will be activated.
- setsid: bool = False
- If the value is `True` the POSIX_SPAWN_SETSID or POSIX_SPAWN_SETSID_NP will be activated.
- setsigmask: object(c_default='NULL') = ()
- The sigmask to use with the POSIX_SPAWN_SETSIGMASK flag.
- setsigdef: object(c_default='NULL') = ()
- The sigmask to use with the POSIX_SPAWN_SETSIGDEF flag.
- scheduler: object = NULL
- A tuple with the scheduler policy (optional) and parameters.
- Execute the program specified by path in a new process.
- [clinic start generated code]*/
- static PyObject *
- os_posix_spawnp_impl(PyObject *module, path_t *path, PyObject *argv,
- PyObject *env, PyObject *file_actions,
- PyObject *setpgroup, int resetids, int setsid,
- PyObject *setsigmask, PyObject *setsigdef,
- PyObject *scheduler)
- /*[clinic end generated code: output=7b9aaefe3031238d input=9e89e616116752a1]*/
- {
- return py_posix_spawn(1, module, path, argv, env, file_actions,
- setpgroup, resetids, setsid, setsigmask, setsigdef,
- scheduler);
- }
- #endif /* HAVE_POSIX_SPAWNP */
- #ifdef HAVE_RTPSPAWN
- static intptr_t
- _rtp_spawn(int mode, const char *rtpFileName, const char *argv[],
- const char *envp[])
- {
- RTP_ID rtpid;
- int status;
- pid_t res;
- int async_err = 0;
- /* Set priority=100 and uStackSize=16 MiB (0x1000000) for new processes.
- uStackSize=0 cannot be used, the default stack size is too small for
- Python. */
- if (envp) {
- rtpid = rtpSpawn(rtpFileName, argv, envp,
- 100, 0x1000000, 0, VX_FP_TASK);
- }
- else {
- rtpid = rtpSpawn(rtpFileName, argv, (const char **)environ,
- 100, 0x1000000, 0, VX_FP_TASK);
- }
- if ((rtpid != RTP_ID_ERROR) && (mode == _P_WAIT)) {
- do {
- res = waitpid((pid_t)rtpid, &status, 0);
- } while (res < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res < 0)
- return RTP_ID_ERROR;
- return ((intptr_t)status);
- }
- return ((intptr_t)rtpid);
- }
- #endif
- #if defined(HAVE_SPAWNV) || defined(HAVE_WSPAWNV) || defined(HAVE_RTPSPAWN)
- /*[clinic input]
- os.spawnv
- mode: int
- Mode of process creation.
- path: path_t
- Path of executable file.
- argv: object
- Tuple or list of strings.
- /
- Execute the program specified by path in a new process.
- [clinic start generated code]*/
- static PyObject *
- os_spawnv_impl(PyObject *module, int mode, path_t *path, PyObject *argv)
- /*[clinic end generated code: output=71cd037a9d96b816 input=43224242303291be]*/
- {
- EXECV_CHAR **argvlist;
- int i;
- Py_ssize_t argc;
- intptr_t spawnval;
- PyObject *(*getitem)(PyObject *, Py_ssize_t);
- /* spawnv has three arguments: (mode, path, argv), where
- argv is a list or tuple of strings. */
- if (PyList_Check(argv)) {
- argc = PyList_Size(argv);
- getitem = PyList_GetItem;
- }
- else if (PyTuple_Check(argv)) {
- argc = PyTuple_Size(argv);
- getitem = PyTuple_GetItem;
- }
- else {
- PyErr_SetString(PyExc_TypeError,
- "spawnv() arg 2 must be a tuple or list");
- return NULL;
- }
- if (argc == 0) {
- PyErr_SetString(PyExc_ValueError,
- "spawnv() arg 2 cannot be empty");
- return NULL;
- }
- argvlist = PyMem_NEW(EXECV_CHAR *, argc+1);
- if (argvlist == NULL) {
- return PyErr_NoMemory();
- }
- for (i = 0; i < argc; i++) {
- if (!fsconvert_strdup((*getitem)(argv, i),
- &argvlist[i])) {
- free_string_array(argvlist, i);
- PyErr_SetString(
- PyExc_TypeError,
- "spawnv() arg 2 must contain only strings");
- return NULL;
- }
- if (i == 0 && !argvlist[0][0]) {
- free_string_array(argvlist, i + 1);
- PyErr_SetString(
- PyExc_ValueError,
- "spawnv() arg 2 first element cannot be empty");
- return NULL;
- }
- }
- argvlist[argc] = NULL;
- #if !defined(HAVE_RTPSPAWN)
- if (mode == _OLD_P_OVERLAY)
- mode = _P_OVERLAY;
- #endif
- if (PySys_Audit("os.spawn", "iOOO", mode, path->object, argv,
- Py_None) < 0) {
- free_string_array(argvlist, argc);
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef HAVE_WSPAWNV
- spawnval = _wspawnv(mode, path->wide, argvlist);
- #elif defined(HAVE_RTPSPAWN)
- spawnval = _rtp_spawn(mode, path->narrow, (const char **)argvlist, NULL);
- #else
- spawnval = _spawnv(mode, path->narrow, argvlist);
- #endif
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- int saved_errno = errno;
- free_string_array(argvlist, argc);
- if (spawnval == -1) {
- errno = saved_errno;
- posix_error();
- return NULL;
- }
- return Py_BuildValue(_Py_PARSE_INTPTR, spawnval);
- }
- /*[clinic input]
- os.spawnve
- mode: int
- Mode of process creation.
- path: path_t
- Path of executable file.
- argv: object
- Tuple or list of strings.
- env: object
- Dictionary of strings mapping to strings.
- /
- Execute the program specified by path in a new process.
- [clinic start generated code]*/
- static PyObject *
- os_spawnve_impl(PyObject *module, int mode, path_t *path, PyObject *argv,
- PyObject *env)
- /*[clinic end generated code: output=30fe85be56fe37ad input=3e40803ee7c4c586]*/
- {
- EXECV_CHAR **argvlist;
- EXECV_CHAR **envlist;
- PyObject *res = NULL;
- Py_ssize_t argc, i, envc;
- intptr_t spawnval;
- PyObject *(*getitem)(PyObject *, Py_ssize_t);
- Py_ssize_t lastarg = 0;
- /* spawnve has four arguments: (mode, path, argv, env), where
- argv is a list or tuple of strings and env is a dictionary
- like posix.environ. */
- if (PyList_Check(argv)) {
- argc = PyList_Size(argv);
- getitem = PyList_GetItem;
- }
- else if (PyTuple_Check(argv)) {
- argc = PyTuple_Size(argv);
- getitem = PyTuple_GetItem;
- }
- else {
- PyErr_SetString(PyExc_TypeError,
- "spawnve() arg 2 must be a tuple or list");
- goto fail_0;
- }
- if (argc == 0) {
- PyErr_SetString(PyExc_ValueError,
- "spawnve() arg 2 cannot be empty");
- goto fail_0;
- }
- if (!PyMapping_Check(env)) {
- PyErr_SetString(PyExc_TypeError,
- "spawnve() arg 3 must be a mapping object");
- goto fail_0;
- }
- argvlist = PyMem_NEW(EXECV_CHAR *, argc+1);
- if (argvlist == NULL) {
- PyErr_NoMemory();
- goto fail_0;
- }
- for (i = 0; i < argc; i++) {
- if (!fsconvert_strdup((*getitem)(argv, i),
- &argvlist[i]))
- {
- lastarg = i;
- goto fail_1;
- }
- if (i == 0 && !argvlist[0][0]) {
- lastarg = i + 1;
- PyErr_SetString(
- PyExc_ValueError,
- "spawnv() arg 2 first element cannot be empty");
- goto fail_1;
- }
- }
- lastarg = argc;
- argvlist[argc] = NULL;
- envlist = parse_envlist(env, &envc);
- if (envlist == NULL)
- goto fail_1;
- #if !defined(HAVE_RTPSPAWN)
- if (mode == _OLD_P_OVERLAY)
- mode = _P_OVERLAY;
- #endif
- if (PySys_Audit("os.spawn", "iOOO", mode, path->object, argv, env) < 0) {
- goto fail_2;
- }
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef HAVE_WSPAWNV
- spawnval = _wspawnve(mode, path->wide, argvlist, envlist);
- #elif defined(HAVE_RTPSPAWN)
- spawnval = _rtp_spawn(mode, path->narrow, (const char **)argvlist,
- (const char **)envlist);
- #else
- spawnval = _spawnve(mode, path->narrow, argvlist, envlist);
- #endif
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- if (spawnval == -1)
- (void) posix_error();
- else
- res = Py_BuildValue(_Py_PARSE_INTPTR, spawnval);
- fail_2:
- while (--envc >= 0) {
- PyMem_Free(envlist[envc]);
- }
- PyMem_Free(envlist);
- fail_1:
- free_string_array(argvlist, lastarg);
- fail_0:
- return res;
- }
- #endif /* HAVE_SPAWNV */
- #ifdef HAVE_FORK
- /* Helper function to validate arguments.
- Returns 0 on success. non-zero on failure with a TypeError raised.
- If obj is non-NULL it must be callable. */
- static int
- check_null_or_callable(PyObject *obj, const char* obj_name)
- {
- if (obj && !PyCallable_Check(obj)) {
- PyErr_Format(PyExc_TypeError, "'%s' must be callable, not %s",
- obj_name, _PyType_Name(Py_TYPE(obj)));
- return -1;
- }
- return 0;
- }
- /*[clinic input]
- os.register_at_fork
- *
- before: object=NULL
- A callable to be called in the parent before the fork() syscall.
- after_in_child: object=NULL
- A callable to be called in the child after fork().
- after_in_parent: object=NULL
- A callable to be called in the parent after fork().
- Register callables to be called when forking a new process.
- 'before' callbacks are called in reverse order.
- 'after_in_child' and 'after_in_parent' callbacks are called in order.
- [clinic start generated code]*/
- static PyObject *
- os_register_at_fork_impl(PyObject *module, PyObject *before,
- PyObject *after_in_child, PyObject *after_in_parent)
- /*[clinic end generated code: output=5398ac75e8e97625 input=cd1187aa85d2312e]*/
- {
- PyInterpreterState *interp;
- if (!before && !after_in_child && !after_in_parent) {
- PyErr_SetString(PyExc_TypeError, "At least one argument is required.");
- return NULL;
- }
- if (check_null_or_callable(before, "before") ||
- check_null_or_callable(after_in_child, "after_in_child") ||
- check_null_or_callable(after_in_parent, "after_in_parent")) {
- return NULL;
- }
- interp = _PyInterpreterState_GET();
- if (register_at_forker(&interp->before_forkers, before)) {
- return NULL;
- }
- if (register_at_forker(&interp->after_forkers_child, after_in_child)) {
- return NULL;
- }
- if (register_at_forker(&interp->after_forkers_parent, after_in_parent)) {
- return NULL;
- }
- Py_RETURN_NONE;
- }
- #endif /* HAVE_FORK */
- #if defined(HAVE_FORK1) || defined(HAVE_FORKPTY) || defined(HAVE_FORK)
- // Common code to raise a warning if we detect there is more than one thread
- // running in the process. Best effort, silent if unable to count threads.
- // Constraint: Quick. Never overcounts. Never leaves an error set.
- //
- // This code might do an import, thus acquiring the import lock, which
- // PyOS_BeforeFork() also does. As this should only be called from
- // the parent process, it is in the same thread so that works.
- static void warn_about_fork_with_threads(const char* name) {
- // TODO: Consider making an `os` module API to return the current number
- // of threads in the process. That'd presumably use this platform code but
- // raise an error rather than using the inaccurate fallback.
- Py_ssize_t num_python_threads = 0;
- #if defined(__APPLE__) && defined(HAVE_GETPID)
- mach_port_t macos_self = mach_task_self();
- mach_port_t macos_task;
- if (task_for_pid(macos_self, getpid(), &macos_task) == KERN_SUCCESS) {
- thread_array_t macos_threads;
- mach_msg_type_number_t macos_n_threads;
- if (task_threads(macos_task, &macos_threads,
- &macos_n_threads) == KERN_SUCCESS) {
- num_python_threads = macos_n_threads;
- }
- }
- #elif defined(__linux__)
- // Linux /proc/self/stat 20th field is the number of threads.
- FILE* proc_stat = fopen("/proc/self/stat", "r");
- if (proc_stat) {
- size_t n;
- // Size chosen arbitrarily. ~60% more bytes than a 20th column index
- // observed on the author's workstation.
- char stat_line[160];
- n = fread(&stat_line, 1, 159, proc_stat);
- stat_line[n] = '\0';
- fclose(proc_stat);
- char *saveptr = NULL;
- char *field = strtok_r(stat_line, " ", &saveptr);
- unsigned int idx;
- for (idx = 19; idx && field; --idx) {
- field = strtok_r(NULL, " ", &saveptr);
- }
- if (idx == 0 && field) { // found the 20th field
- num_python_threads = atoi(field); // 0 on error
- }
- }
- #endif
- if (num_python_threads <= 0) {
- // Fall back to just the number our threading module knows about.
- // An incomplete view of the world, but better than nothing.
- PyObject *threading = PyImport_GetModule(&_Py_ID(threading));
- if (!threading) {
- PyErr_Clear();
- return;
- }
- PyObject *threading_active =
- PyObject_GetAttr(threading, &_Py_ID(_active));
- if (!threading_active) {
- PyErr_Clear();
- Py_DECREF(threading);
- return;
- }
- PyObject *threading_limbo =
- PyObject_GetAttr(threading, &_Py_ID(_limbo));
- if (!threading_limbo) {
- PyErr_Clear();
- Py_DECREF(threading);
- Py_DECREF(threading_active);
- return;
- }
- Py_DECREF(threading);
- // Duplicating what threading.active_count() does but without holding
- // threading._active_limbo_lock so our count could be inaccurate if
- // these dicts are mid-update from another thread. Not a big deal.
- // Worst case if someone replaced threading._active or threading._limbo
- // with non-dicts, we get -1 from *Length() below and undercount.
- // Nobody should, but we're best effort so we clear errors and move on.
- num_python_threads = (PyMapping_Length(threading_active)
- + PyMapping_Length(threading_limbo));
- PyErr_Clear();
- Py_DECREF(threading_active);
- Py_DECREF(threading_limbo);
- }
- if (num_python_threads > 1) {
- PyErr_WarnFormat(
- PyExc_DeprecationWarning, 1,
- #ifdef HAVE_GETPID
- "This process (pid=%d) is multi-threaded, "
- #else
- "This process is multi-threaded, "
- #endif
- "use of %s() may lead to deadlocks in the child.",
- #ifdef HAVE_GETPID
- getpid(),
- #endif
- name);
- PyErr_Clear();
- }
- }
- #endif // HAVE_FORK1 || HAVE_FORKPTY || HAVE_FORK
- #ifdef HAVE_FORK1
- /*[clinic input]
- os.fork1
- Fork a child process with a single multiplexed (i.e., not bound) thread.
- Return 0 to child process and PID of child to parent process.
- [clinic start generated code]*/
- static PyObject *
- os_fork1_impl(PyObject *module)
- /*[clinic end generated code: output=0de8e67ce2a310bc input=12db02167893926e]*/
- {
- pid_t pid;
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (_PyInterpreterState_GetFinalizing(interp) != NULL) {
- PyErr_SetString(PyExc_RuntimeError,
- "can't fork at interpreter shutdown");
- return NULL;
- }
- if (!_Py_IsMainInterpreter(interp)) {
- PyErr_SetString(PyExc_RuntimeError, "fork not supported for subinterpreters");
- return NULL;
- }
- PyOS_BeforeFork();
- pid = fork1();
- int saved_errno = errno;
- if (pid == 0) {
- /* child: this clobbers and resets the import lock. */
- PyOS_AfterFork_Child();
- } else {
- warn_about_fork_with_threads("fork1");
- /* parent: release the import lock. */
- PyOS_AfterFork_Parent();
- }
- if (pid == -1) {
- errno = saved_errno;
- return posix_error();
- }
- return PyLong_FromPid(pid);
- }
- #endif /* HAVE_FORK1 */
- #ifdef HAVE_FORK
- /*[clinic input]
- os.fork
- Fork a child process.
- Return 0 to child process and PID of child to parent process.
- [clinic start generated code]*/
- static PyObject *
- os_fork_impl(PyObject *module)
- /*[clinic end generated code: output=3626c81f98985d49 input=13c956413110eeaa]*/
- {
- pid_t pid;
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (_PyInterpreterState_GetFinalizing(interp) != NULL) {
- PyErr_SetString(PyExc_RuntimeError,
- "can't fork at interpreter shutdown");
- return NULL;
- }
- if (!_PyInterpreterState_HasFeature(interp, Py_RTFLAGS_FORK)) {
- PyErr_SetString(PyExc_RuntimeError,
- "fork not supported for isolated subinterpreters");
- return NULL;
- }
- if (PySys_Audit("os.fork", NULL) < 0) {
- return NULL;
- }
- PyOS_BeforeFork();
- pid = fork();
- int saved_errno = errno;
- if (pid == 0) {
- /* child: this clobbers and resets the import lock. */
- PyOS_AfterFork_Child();
- } else {
- warn_about_fork_with_threads("fork");
- /* parent: release the import lock. */
- PyOS_AfterFork_Parent();
- }
- if (pid == -1) {
- errno = saved_errno;
- return posix_error();
- }
- return PyLong_FromPid(pid);
- }
- #endif /* HAVE_FORK */
- #ifdef HAVE_SCHED_H
- #ifdef HAVE_SCHED_GET_PRIORITY_MAX
- /*[clinic input]
- os.sched_get_priority_max
- policy: int
- Get the maximum scheduling priority for policy.
- [clinic start generated code]*/
- static PyObject *
- os_sched_get_priority_max_impl(PyObject *module, int policy)
- /*[clinic end generated code: output=9e465c6e43130521 input=2097b7998eca6874]*/
- {
- int max;
- max = sched_get_priority_max(policy);
- if (max < 0)
- return posix_error();
- return PyLong_FromLong(max);
- }
- /*[clinic input]
- os.sched_get_priority_min
- policy: int
- Get the minimum scheduling priority for policy.
- [clinic start generated code]*/
- static PyObject *
- os_sched_get_priority_min_impl(PyObject *module, int policy)
- /*[clinic end generated code: output=7595c1138cc47a6d input=21bc8fa0d70983bf]*/
- {
- int min = sched_get_priority_min(policy);
- if (min < 0)
- return posix_error();
- return PyLong_FromLong(min);
- }
- #endif /* HAVE_SCHED_GET_PRIORITY_MAX */
- #ifdef HAVE_SCHED_SETSCHEDULER
- /*[clinic input]
- os.sched_getscheduler
- pid: pid_t
- /
- Get the scheduling policy for the process identified by pid.
- Passing 0 for pid returns the scheduling policy for the calling process.
- [clinic start generated code]*/
- static PyObject *
- os_sched_getscheduler_impl(PyObject *module, pid_t pid)
- /*[clinic end generated code: output=dce4c0bd3f1b34c8 input=8d99dac505485ac8]*/
- {
- int policy;
- policy = sched_getscheduler(pid);
- if (policy < 0)
- return posix_error();
- return PyLong_FromLong(policy);
- }
- #endif /* HAVE_SCHED_SETSCHEDULER */
- #if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM)
- /*[clinic input]
- class os.sched_param "PyObject *" "SchedParamType"
- @classmethod
- os.sched_param.__new__
- sched_priority: object
- A scheduling parameter.
- Currently has only one field: sched_priority
- [clinic start generated code]*/
- static PyObject *
- os_sched_param_impl(PyTypeObject *type, PyObject *sched_priority)
- /*[clinic end generated code: output=48f4067d60f48c13 input=eb42909a2c0e3e6c]*/
- {
- PyObject *res;
- res = PyStructSequence_New(type);
- if (!res)
- return NULL;
- PyStructSequence_SET_ITEM(res, 0, Py_NewRef(sched_priority));
- return res;
- }
- static PyObject *
- os_sched_param_reduce(PyObject *self, PyObject *Py_UNUSED(ignored))
- {
- return Py_BuildValue("(O(N))", Py_TYPE(self), PyStructSequence_GetItem(self, 0));
- }
- static PyMethodDef os_sched_param_reduce_method = {
- "__reduce__", (PyCFunction)os_sched_param_reduce, METH_NOARGS|METH_COEXIST, NULL,
- };
- PyDoc_VAR(os_sched_param__doc__);
- static PyStructSequence_Field sched_param_fields[] = {
- {"sched_priority", "the scheduling priority"},
- {0}
- };
- static PyStructSequence_Desc sched_param_desc = {
- "sched_param", /* name */
- os_sched_param__doc__, /* doc */
- sched_param_fields,
- 1
- };
- static int
- convert_sched_param(PyObject *module, PyObject *param, struct sched_param *res)
- {
- long priority;
- if (!Py_IS_TYPE(param, (PyTypeObject *)get_posix_state(module)->SchedParamType)) {
- PyErr_SetString(PyExc_TypeError, "must have a sched_param object");
- return 0;
- }
- priority = PyLong_AsLong(PyStructSequence_GET_ITEM(param, 0));
- if (priority == -1 && PyErr_Occurred())
- return 0;
- if (priority > INT_MAX || priority < INT_MIN) {
- PyErr_SetString(PyExc_OverflowError, "sched_priority out of range");
- return 0;
- }
- res->sched_priority = Py_SAFE_DOWNCAST(priority, long, int);
- return 1;
- }
- #endif /* defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM) */
- #ifdef HAVE_SCHED_SETSCHEDULER
- /*[clinic input]
- os.sched_setscheduler
- pid: pid_t
- policy: int
- param as param_obj: object
- /
- Set the scheduling policy for the process identified by pid.
- If pid is 0, the calling process is changed.
- param is an instance of sched_param.
- [clinic start generated code]*/
- static PyObject *
- os_sched_setscheduler_impl(PyObject *module, pid_t pid, int policy,
- PyObject *param_obj)
- /*[clinic end generated code: output=cde27faa55dc993e input=73013d731bd8fbe9]*/
- {
- struct sched_param param;
- if (!convert_sched_param(module, param_obj, ¶m)) {
- return NULL;
- }
- /*
- ** sched_setscheduler() returns 0 in Linux, but the previous
- ** scheduling policy under Solaris/Illumos, and others.
- ** On error, -1 is returned in all Operating Systems.
- */
- if (sched_setscheduler(pid, policy, ¶m) == -1)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SCHED_SETSCHEDULER*/
- #ifdef HAVE_SCHED_SETPARAM
- /*[clinic input]
- os.sched_getparam
- pid: pid_t
- /
- Returns scheduling parameters for the process identified by pid.
- If pid is 0, returns parameters for the calling process.
- Return value is an instance of sched_param.
- [clinic start generated code]*/
- static PyObject *
- os_sched_getparam_impl(PyObject *module, pid_t pid)
- /*[clinic end generated code: output=b194e8708dcf2db8 input=18a1ef9c2efae296]*/
- {
- struct sched_param param;
- PyObject *result;
- PyObject *priority;
- if (sched_getparam(pid, ¶m))
- return posix_error();
- PyObject *SchedParamType = get_posix_state(module)->SchedParamType;
- result = PyStructSequence_New((PyTypeObject *)SchedParamType);
- if (!result)
- return NULL;
- priority = PyLong_FromLong(param.sched_priority);
- if (!priority) {
- Py_DECREF(result);
- return NULL;
- }
- PyStructSequence_SET_ITEM(result, 0, priority);
- return result;
- }
- /*[clinic input]
- os.sched_setparam
- pid: pid_t
- param as param_obj: object
- /
- Set scheduling parameters for the process identified by pid.
- If pid is 0, sets parameters for the calling process.
- param should be an instance of sched_param.
- [clinic start generated code]*/
- static PyObject *
- os_sched_setparam_impl(PyObject *module, pid_t pid, PyObject *param_obj)
- /*[clinic end generated code: output=f19fe020a53741c1 input=27b98337c8b2dcc7]*/
- {
- struct sched_param param;
- if (!convert_sched_param(module, param_obj, ¶m)) {
- return NULL;
- }
- if (sched_setparam(pid, ¶m))
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SCHED_SETPARAM */
- #ifdef HAVE_SCHED_RR_GET_INTERVAL
- /*[clinic input]
- os.sched_rr_get_interval -> double
- pid: pid_t
- /
- Return the round-robin quantum for the process identified by pid, in seconds.
- Value returned is a float.
- [clinic start generated code]*/
- static double
- os_sched_rr_get_interval_impl(PyObject *module, pid_t pid)
- /*[clinic end generated code: output=7e2d935833ab47dc input=2a973da15cca6fae]*/
- {
- struct timespec interval;
- if (sched_rr_get_interval(pid, &interval)) {
- posix_error();
- return -1.0;
- }
- #ifdef _Py_MEMORY_SANITIZER
- __msan_unpoison(&interval, sizeof(interval));
- #endif
- return (double)interval.tv_sec + 1e-9*interval.tv_nsec;
- }
- #endif /* HAVE_SCHED_RR_GET_INTERVAL */
- /*[clinic input]
- os.sched_yield
- Voluntarily relinquish the CPU.
- [clinic start generated code]*/
- static PyObject *
- os_sched_yield_impl(PyObject *module)
- /*[clinic end generated code: output=902323500f222cac input=e54d6f98189391d4]*/
- {
- int result;
- Py_BEGIN_ALLOW_THREADS
- result = sched_yield();
- Py_END_ALLOW_THREADS
- if (result < 0) {
- return posix_error();
- }
- Py_RETURN_NONE;
- }
- #ifdef HAVE_SCHED_SETAFFINITY
- /* The minimum number of CPUs allocated in a cpu_set_t */
- static const int NCPUS_START = sizeof(unsigned long) * CHAR_BIT;
- /*[clinic input]
- os.sched_setaffinity
- pid: pid_t
- mask : object
- /
- Set the CPU affinity of the process identified by pid to mask.
- mask should be an iterable of integers identifying CPUs.
- [clinic start generated code]*/
- static PyObject *
- os_sched_setaffinity_impl(PyObject *module, pid_t pid, PyObject *mask)
- /*[clinic end generated code: output=882d7dd9a229335b input=a0791a597c7085ba]*/
- {
- int ncpus;
- size_t setsize;
- cpu_set_t *cpu_set = NULL;
- PyObject *iterator = NULL, *item;
- iterator = PyObject_GetIter(mask);
- if (iterator == NULL)
- return NULL;
- ncpus = NCPUS_START;
- setsize = CPU_ALLOC_SIZE(ncpus);
- cpu_set = CPU_ALLOC(ncpus);
- if (cpu_set == NULL) {
- PyErr_NoMemory();
- goto error;
- }
- CPU_ZERO_S(setsize, cpu_set);
- while ((item = PyIter_Next(iterator))) {
- long cpu;
- if (!PyLong_Check(item)) {
- PyErr_Format(PyExc_TypeError,
- "expected an iterator of ints, "
- "but iterator yielded %R",
- Py_TYPE(item));
- Py_DECREF(item);
- goto error;
- }
- cpu = PyLong_AsLong(item);
- Py_DECREF(item);
- if (cpu < 0) {
- if (!PyErr_Occurred())
- PyErr_SetString(PyExc_ValueError, "negative CPU number");
- goto error;
- }
- if (cpu > INT_MAX - 1) {
- PyErr_SetString(PyExc_OverflowError, "CPU number too large");
- goto error;
- }
- if (cpu >= ncpus) {
- /* Grow CPU mask to fit the CPU number */
- int newncpus = ncpus;
- cpu_set_t *newmask;
- size_t newsetsize;
- while (newncpus <= cpu) {
- if (newncpus > INT_MAX / 2)
- newncpus = cpu + 1;
- else
- newncpus = newncpus * 2;
- }
- newmask = CPU_ALLOC(newncpus);
- if (newmask == NULL) {
- PyErr_NoMemory();
- goto error;
- }
- newsetsize = CPU_ALLOC_SIZE(newncpus);
- CPU_ZERO_S(newsetsize, newmask);
- memcpy(newmask, cpu_set, setsize);
- CPU_FREE(cpu_set);
- setsize = newsetsize;
- cpu_set = newmask;
- ncpus = newncpus;
- }
- CPU_SET_S(cpu, setsize, cpu_set);
- }
- if (PyErr_Occurred()) {
- goto error;
- }
- Py_CLEAR(iterator);
- if (sched_setaffinity(pid, setsize, cpu_set)) {
- posix_error();
- goto error;
- }
- CPU_FREE(cpu_set);
- Py_RETURN_NONE;
- error:
- if (cpu_set)
- CPU_FREE(cpu_set);
- Py_XDECREF(iterator);
- return NULL;
- }
- /*[clinic input]
- os.sched_getaffinity
- pid: pid_t
- /
- Return the affinity of the process identified by pid (or the current process if zero).
- The affinity is returned as a set of CPU identifiers.
- [clinic start generated code]*/
- static PyObject *
- os_sched_getaffinity_impl(PyObject *module, pid_t pid)
- /*[clinic end generated code: output=f726f2c193c17a4f input=983ce7cb4a565980]*/
- {
- int cpu, ncpus, count;
- size_t setsize;
- cpu_set_t *mask = NULL;
- PyObject *res = NULL;
- ncpus = NCPUS_START;
- while (1) {
- setsize = CPU_ALLOC_SIZE(ncpus);
- mask = CPU_ALLOC(ncpus);
- if (mask == NULL)
- return PyErr_NoMemory();
- if (sched_getaffinity(pid, setsize, mask) == 0)
- break;
- CPU_FREE(mask);
- if (errno != EINVAL)
- return posix_error();
- if (ncpus > INT_MAX / 2) {
- PyErr_SetString(PyExc_OverflowError, "could not allocate "
- "a large enough CPU set");
- return NULL;
- }
- ncpus = ncpus * 2;
- }
- res = PySet_New(NULL);
- if (res == NULL)
- goto error;
- for (cpu = 0, count = CPU_COUNT_S(setsize, mask); count; cpu++) {
- if (CPU_ISSET_S(cpu, setsize, mask)) {
- PyObject *cpu_num = PyLong_FromLong(cpu);
- --count;
- if (cpu_num == NULL)
- goto error;
- if (PySet_Add(res, cpu_num)) {
- Py_DECREF(cpu_num);
- goto error;
- }
- Py_DECREF(cpu_num);
- }
- }
- CPU_FREE(mask);
- return res;
- error:
- if (mask)
- CPU_FREE(mask);
- Py_XDECREF(res);
- return NULL;
- }
- #endif /* HAVE_SCHED_SETAFFINITY */
- #endif /* HAVE_SCHED_H */
- /* AIX uses /dev/ptc but is otherwise the same as /dev/ptmx */
- #if defined(HAVE_DEV_PTC) && !defined(HAVE_DEV_PTMX)
- # define DEV_PTY_FILE "/dev/ptc"
- # define HAVE_DEV_PTMX
- #else
- # define DEV_PTY_FILE "/dev/ptmx"
- #endif
- #if defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_LOGIN_TTY) || defined(HAVE_DEV_PTMX)
- #ifdef HAVE_PTY_H
- #include <pty.h>
- #ifdef HAVE_UTMP_H
- #include <utmp.h>
- #endif /* HAVE_UTMP_H */
- #elif defined(HAVE_LIBUTIL_H)
- #include <libutil.h>
- #elif defined(HAVE_UTIL_H)
- #include <util.h>
- #endif /* HAVE_PTY_H */
- #ifdef HAVE_STROPTS_H
- #include <stropts.h>
- #else
- #define __SID ('S' << 8)
- #define I_PUSH (__SID | 2)
- #endif
- #endif /* defined(HAVE_OPENPTY) || defined(HAVE_FORKPTY) || defined(HAVE_LOGIN_TTY) || defined(HAVE_DEV_PTMX) */
- #if defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX)
- /*[clinic input]
- os.openpty
- Open a pseudo-terminal.
- Return a tuple of (master_fd, slave_fd) containing open file descriptors
- for both the master and slave ends.
- [clinic start generated code]*/
- static PyObject *
- os_openpty_impl(PyObject *module)
- /*[clinic end generated code: output=98841ce5ec9cef3c input=f3d99fd99e762907]*/
- {
- int master_fd = -1, slave_fd = -1;
- #ifndef HAVE_OPENPTY
- char * slave_name;
- #endif
- #if defined(HAVE_DEV_PTMX) && !defined(HAVE_OPENPTY) && !defined(HAVE__GETPTY)
- PyOS_sighandler_t sig_saved;
- #if defined(__sun) && defined(__SVR4)
- extern char *ptsname(int fildes);
- #endif
- #endif
- #ifdef HAVE_OPENPTY
- if (openpty(&master_fd, &slave_fd, NULL, NULL, NULL) != 0)
- goto posix_error;
- if (_Py_set_inheritable(master_fd, 0, NULL) < 0)
- goto error;
- if (_Py_set_inheritable(slave_fd, 0, NULL) < 0)
- goto error;
- #elif defined(HAVE__GETPTY)
- slave_name = _getpty(&master_fd, O_RDWR, 0666, 0);
- if (slave_name == NULL)
- goto posix_error;
- if (_Py_set_inheritable(master_fd, 0, NULL) < 0)
- goto error;
- slave_fd = _Py_open(slave_name, O_RDWR);
- if (slave_fd < 0)
- goto error;
- #else
- master_fd = open(DEV_PTY_FILE, O_RDWR | O_NOCTTY); /* open master */
- if (master_fd < 0)
- goto posix_error;
- sig_saved = PyOS_setsig(SIGCHLD, SIG_DFL);
- /* change permission of slave */
- if (grantpt(master_fd) < 0) {
- int saved_errno = errno;
- PyOS_setsig(SIGCHLD, sig_saved);
- errno = saved_errno;
- goto posix_error;
- }
- /* unlock slave */
- if (unlockpt(master_fd) < 0) {
- int saved_errno = errno;
- PyOS_setsig(SIGCHLD, sig_saved);
- errno = saved_errno;
- goto posix_error;
- }
- PyOS_setsig(SIGCHLD, sig_saved);
- slave_name = ptsname(master_fd); /* get name of slave */
- if (slave_name == NULL)
- goto posix_error;
- slave_fd = _Py_open(slave_name, O_RDWR | O_NOCTTY); /* open slave */
- if (slave_fd == -1)
- goto error;
- if (_Py_set_inheritable(master_fd, 0, NULL) < 0)
- goto posix_error;
- #if !defined(__CYGWIN__) && !defined(__ANDROID__) && !defined(HAVE_DEV_PTC)
- ioctl(slave_fd, I_PUSH, "ptem"); /* push ptem */
- ioctl(slave_fd, I_PUSH, "ldterm"); /* push ldterm */
- #ifndef __hpux
- ioctl(slave_fd, I_PUSH, "ttcompat"); /* push ttcompat */
- #endif /* __hpux */
- #endif /* HAVE_CYGWIN */
- #endif /* HAVE_OPENPTY */
- return Py_BuildValue("(ii)", master_fd, slave_fd);
- posix_error:
- posix_error();
- error:
- if (master_fd != -1)
- close(master_fd);
- if (slave_fd != -1)
- close(slave_fd);
- return NULL;
- }
- #endif /* defined(HAVE_OPENPTY) || defined(HAVE__GETPTY) || defined(HAVE_DEV_PTMX) */
- #if defined(HAVE_SETSID) && defined(TIOCSCTTY)
- #define HAVE_FALLBACK_LOGIN_TTY 1
- #endif /* defined(HAVE_SETSID) && defined(TIOCSCTTY) */
- #if defined(HAVE_LOGIN_TTY) || defined(HAVE_FALLBACK_LOGIN_TTY)
- /*[clinic input]
- os.login_tty
- fd: fildes
- /
- Prepare the tty of which fd is a file descriptor for a new login session.
- Make the calling process a session leader; make the tty the
- controlling tty, the stdin, the stdout, and the stderr of the
- calling process; close fd.
- [clinic start generated code]*/
- static PyObject *
- os_login_tty_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=495a79911b4cc1bc input=5f298565099903a2]*/
- {
- #ifdef HAVE_LOGIN_TTY
- if (login_tty(fd) == -1) {
- return posix_error();
- }
- #else /* defined(HAVE_FALLBACK_LOGIN_TTY) */
- /* Establish a new session. */
- if (setsid() == -1) {
- return posix_error();
- }
- /* The tty becomes the controlling terminal. */
- if (ioctl(fd, TIOCSCTTY, (char *)NULL) == -1) {
- return posix_error();
- }
- /* The tty becomes stdin/stdout/stderr */
- if (dup2(fd, 0) == -1 || dup2(fd, 1) == -1 || dup2(fd, 2) == -1) {
- return posix_error();
- }
- if (fd > 2) {
- close(fd);
- }
- #endif /* HAVE_LOGIN_TTY */
- Py_RETURN_NONE;
- }
- #endif /* defined(HAVE_LOGIN_TTY) || defined(HAVE_FALLBACK_LOGIN_TTY) */
- #ifdef HAVE_FORKPTY
- /*[clinic input]
- os.forkpty
- Fork a new process with a new pseudo-terminal as controlling tty.
- Returns a tuple of (pid, master_fd).
- Like fork(), return pid of 0 to the child process,
- and pid of child to the parent process.
- To both, return fd of newly opened pseudo-terminal.
- [clinic start generated code]*/
- static PyObject *
- os_forkpty_impl(PyObject *module)
- /*[clinic end generated code: output=60d0a5c7512e4087 input=f1f7f4bae3966010]*/
- {
- int master_fd = -1;
- pid_t pid;
- PyInterpreterState *interp = _PyInterpreterState_GET();
- if (_PyInterpreterState_GetFinalizing(interp) != NULL) {
- PyErr_SetString(PyExc_RuntimeError,
- "can't fork at interpreter shutdown");
- return NULL;
- }
- if (!_Py_IsMainInterpreter(interp)) {
- PyErr_SetString(PyExc_RuntimeError, "fork not supported for subinterpreters");
- return NULL;
- }
- if (PySys_Audit("os.forkpty", NULL) < 0) {
- return NULL;
- }
- PyOS_BeforeFork();
- pid = forkpty(&master_fd, NULL, NULL, NULL);
- if (pid == 0) {
- /* child: this clobbers and resets the import lock. */
- PyOS_AfterFork_Child();
- } else {
- warn_about_fork_with_threads("forkpty");
- /* parent: release the import lock. */
- PyOS_AfterFork_Parent();
- }
- if (pid == -1) {
- return posix_error();
- }
- return Py_BuildValue("(Ni)", PyLong_FromPid(pid), master_fd);
- }
- #endif /* HAVE_FORKPTY */
- #ifdef HAVE_GETEGID
- /*[clinic input]
- os.getegid
- Return the current process's effective group id.
- [clinic start generated code]*/
- static PyObject *
- os_getegid_impl(PyObject *module)
- /*[clinic end generated code: output=67d9be7ac68898a2 input=1596f79ad1107d5d]*/
- {
- return _PyLong_FromGid(getegid());
- }
- #endif /* HAVE_GETEGID */
- #ifdef HAVE_GETEUID
- /*[clinic input]
- os.geteuid
- Return the current process's effective user id.
- [clinic start generated code]*/
- static PyObject *
- os_geteuid_impl(PyObject *module)
- /*[clinic end generated code: output=ea1b60f0d6abb66e input=4644c662d3bd9f19]*/
- {
- return _PyLong_FromUid(geteuid());
- }
- #endif /* HAVE_GETEUID */
- #ifdef HAVE_GETGID
- /*[clinic input]
- os.getgid
- Return the current process's group id.
- [clinic start generated code]*/
- static PyObject *
- os_getgid_impl(PyObject *module)
- /*[clinic end generated code: output=4f28ebc9d3e5dfcf input=58796344cd87c0f6]*/
- {
- return _PyLong_FromGid(getgid());
- }
- #endif /* HAVE_GETGID */
- #if defined(HAVE_GETPID)
- /*[clinic input]
- os.getpid
- Return the current process id.
- [clinic start generated code]*/
- static PyObject *
- os_getpid_impl(PyObject *module)
- /*[clinic end generated code: output=9ea6fdac01ed2b3c input=5a9a00f0ab68aa00]*/
- {
- #if !defined(MS_WINDOWS) || defined(MS_WINDOWS_DESKTOP) || defined(MS_WINDOWS_SYSTEM)
- return PyLong_FromPid(getpid());
- #else
- return PyLong_FromUnsignedLong(GetCurrentProcessId());
- #endif
- }
- #endif /* defined(HAVE_GETPID) */
- #ifdef NGROUPS_MAX
- #define MAX_GROUPS NGROUPS_MAX
- #else
- /* defined to be 16 on Solaris7, so this should be a small number */
- #define MAX_GROUPS 64
- #endif
- #ifdef HAVE_GETGROUPLIST
- #ifdef __APPLE__
- /*[clinic input]
- os.getgrouplist
- user: str
- username to lookup
- group as basegid: int
- base group id of the user
- /
- Returns a list of groups to which a user belongs.
- [clinic start generated code]*/
- static PyObject *
- os_getgrouplist_impl(PyObject *module, const char *user, int basegid)
- /*[clinic end generated code: output=6e734697b8c26de0 input=f8d870374b09a490]*/
- #else
- /*[clinic input]
- os.getgrouplist
- user: str
- username to lookup
- group as basegid: gid_t
- base group id of the user
- /
- Returns a list of groups to which a user belongs.
- [clinic start generated code]*/
- static PyObject *
- os_getgrouplist_impl(PyObject *module, const char *user, gid_t basegid)
- /*[clinic end generated code: output=0ebd7fb70115575b input=cc61d5c20b08958d]*/
- #endif
- {
- int i, ngroups;
- PyObject *list;
- #ifdef __APPLE__
- int *groups;
- #else
- gid_t *groups;
- #endif
- /*
- * NGROUPS_MAX is defined by POSIX.1 as the maximum
- * number of supplimental groups a users can belong to.
- * We have to increment it by one because
- * getgrouplist() returns both the supplemental groups
- * and the primary group, i.e. all of the groups the
- * user belongs to.
- */
- ngroups = 1 + MAX_GROUPS;
- while (1) {
- #ifdef __APPLE__
- groups = PyMem_New(int, ngroups);
- #else
- groups = PyMem_New(gid_t, ngroups);
- #endif
- if (groups == NULL) {
- return PyErr_NoMemory();
- }
- int old_ngroups = ngroups;
- if (getgrouplist(user, basegid, groups, &ngroups) != -1) {
- /* Success */
- break;
- }
- /* getgrouplist() fails if the group list is too small */
- PyMem_Free(groups);
- if (ngroups > old_ngroups) {
- /* If the group list is too small, the glibc implementation of
- getgrouplist() sets ngroups to the total number of groups and
- returns -1. */
- }
- else {
- /* Double the group list size */
- if (ngroups > INT_MAX / 2) {
- return PyErr_NoMemory();
- }
- ngroups *= 2;
- }
- /* Retry getgrouplist() with a larger group list */
- }
- #ifdef _Py_MEMORY_SANITIZER
- /* Clang memory sanitizer libc intercepts don't know getgrouplist. */
- __msan_unpoison(&ngroups, sizeof(ngroups));
- __msan_unpoison(groups, ngroups*sizeof(*groups));
- #endif
- list = PyList_New(ngroups);
- if (list == NULL) {
- PyMem_Free(groups);
- return NULL;
- }
- for (i = 0; i < ngroups; i++) {
- #ifdef __APPLE__
- PyObject *o = PyLong_FromUnsignedLong((unsigned long)groups[i]);
- #else
- PyObject *o = _PyLong_FromGid(groups[i]);
- #endif
- if (o == NULL) {
- Py_DECREF(list);
- PyMem_Free(groups);
- return NULL;
- }
- PyList_SET_ITEM(list, i, o);
- }
- PyMem_Free(groups);
- return list;
- }
- #endif /* HAVE_GETGROUPLIST */
- #ifdef HAVE_GETGROUPS
- /*[clinic input]
- os.getgroups
- Return list of supplemental group IDs for the process.
- [clinic start generated code]*/
- static PyObject *
- os_getgroups_impl(PyObject *module)
- /*[clinic end generated code: output=42b0c17758561b56 input=d3f109412e6a155c]*/
- {
- // Call getgroups with length 0 to get the actual number of groups
- int n = getgroups(0, NULL);
- if (n < 0) {
- return posix_error();
- }
- if (n == 0) {
- return PyList_New(0);
- }
- gid_t *grouplist = PyMem_New(gid_t, n);
- if (grouplist == NULL) {
- return PyErr_NoMemory();
- }
- n = getgroups(n, grouplist);
- if (n == -1) {
- posix_error();
- PyMem_Free(grouplist);
- return NULL;
- }
- PyObject *result = PyList_New(n);
- if (result == NULL) {
- goto error;
- }
- for (int i = 0; i < n; ++i) {
- PyObject *group = _PyLong_FromGid(grouplist[i]);
- if (group == NULL) {
- goto error;
- }
- PyList_SET_ITEM(result, i, group);
- }
- PyMem_Free(grouplist);
- return result;
- error:
- PyMem_Free(grouplist);
- Py_XDECREF(result);
- return NULL;
- }
- #endif /* HAVE_GETGROUPS */
- #ifdef HAVE_INITGROUPS
- #ifdef __APPLE__
- /*[clinic input]
- os.initgroups
- username as oname: FSConverter
- gid: int
- /
- Initialize the group access list.
- Call the system initgroups() to initialize the group access list with all of
- the groups of which the specified username is a member, plus the specified
- group id.
- [clinic start generated code]*/
- static PyObject *
- os_initgroups_impl(PyObject *module, PyObject *oname, int gid)
- /*[clinic end generated code: output=7f074d30a425fd3a input=df3d54331b0af204]*/
- #else
- /*[clinic input]
- os.initgroups
- username as oname: FSConverter
- gid: gid_t
- /
- Initialize the group access list.
- Call the system initgroups() to initialize the group access list with all of
- the groups of which the specified username is a member, plus the specified
- group id.
- [clinic start generated code]*/
- static PyObject *
- os_initgroups_impl(PyObject *module, PyObject *oname, gid_t gid)
- /*[clinic end generated code: output=59341244521a9e3f input=0cb91bdc59a4c564]*/
- #endif
- {
- const char *username = PyBytes_AS_STRING(oname);
- if (initgroups(username, gid) == -1)
- return PyErr_SetFromErrno(PyExc_OSError);
- Py_RETURN_NONE;
- }
- #endif /* HAVE_INITGROUPS */
- #ifdef HAVE_GETPGID
- /*[clinic input]
- os.getpgid
- pid: pid_t
- Call the system call getpgid(), and return the result.
- [clinic start generated code]*/
- static PyObject *
- os_getpgid_impl(PyObject *module, pid_t pid)
- /*[clinic end generated code: output=1db95a97be205d18 input=39d710ae3baaf1c7]*/
- {
- pid_t pgid = getpgid(pid);
- if (pgid < 0)
- return posix_error();
- return PyLong_FromPid(pgid);
- }
- #endif /* HAVE_GETPGID */
- #ifdef HAVE_GETPGRP
- /*[clinic input]
- os.getpgrp
- Return the current process group id.
- [clinic start generated code]*/
- static PyObject *
- os_getpgrp_impl(PyObject *module)
- /*[clinic end generated code: output=c4fc381e51103cf3 input=6846fb2bb9a3705e]*/
- {
- #ifdef GETPGRP_HAVE_ARG
- return PyLong_FromPid(getpgrp(0));
- #else /* GETPGRP_HAVE_ARG */
- return PyLong_FromPid(getpgrp());
- #endif /* GETPGRP_HAVE_ARG */
- }
- #endif /* HAVE_GETPGRP */
- #ifdef HAVE_SETPGRP
- /*[clinic input]
- os.setpgrp
- Make the current process the leader of its process group.
- [clinic start generated code]*/
- static PyObject *
- os_setpgrp_impl(PyObject *module)
- /*[clinic end generated code: output=2554735b0a60f0a0 input=1f0619fcb5731e7e]*/
- {
- #ifdef SETPGRP_HAVE_ARG
- if (setpgrp(0, 0) < 0)
- #else /* SETPGRP_HAVE_ARG */
- if (setpgrp() < 0)
- #endif /* SETPGRP_HAVE_ARG */
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETPGRP */
- #ifdef HAVE_GETPPID
- #ifdef MS_WINDOWS
- #include <processsnapshot.h>
- static PyObject*
- win32_getppid(void)
- {
- DWORD error;
- PyObject* result = NULL;
- HANDLE process = GetCurrentProcess();
- HPSS snapshot = NULL;
- error = PssCaptureSnapshot(process, PSS_CAPTURE_NONE, 0, &snapshot);
- if (error != ERROR_SUCCESS) {
- return PyErr_SetFromWindowsErr(error);
- }
- PSS_PROCESS_INFORMATION info;
- error = PssQuerySnapshot(snapshot, PSS_QUERY_PROCESS_INFORMATION, &info,
- sizeof(info));
- if (error == ERROR_SUCCESS) {
- result = PyLong_FromUnsignedLong(info.ParentProcessId);
- }
- else {
- result = PyErr_SetFromWindowsErr(error);
- }
- PssFreeSnapshot(process, snapshot);
- return result;
- }
- #endif /*MS_WINDOWS*/
- /*[clinic input]
- os.getppid
- Return the parent's process id.
- If the parent process has already exited, Windows machines will still
- return its id; others systems will return the id of the 'init' process (1).
- [clinic start generated code]*/
- static PyObject *
- os_getppid_impl(PyObject *module)
- /*[clinic end generated code: output=43b2a946a8c603b4 input=e637cb87539c030e]*/
- {
- #ifdef MS_WINDOWS
- return win32_getppid();
- #else
- return PyLong_FromPid(getppid());
- #endif
- }
- #endif /* HAVE_GETPPID */
- #ifdef HAVE_GETLOGIN
- /*[clinic input]
- os.getlogin
- Return the actual login name.
- [clinic start generated code]*/
- static PyObject *
- os_getlogin_impl(PyObject *module)
- /*[clinic end generated code: output=a32e66a7e5715dac input=2a21ab1e917163df]*/
- {
- PyObject *result = NULL;
- #ifdef MS_WINDOWS
- wchar_t user_name[UNLEN + 1];
- DWORD num_chars = Py_ARRAY_LENGTH(user_name);
- if (GetUserNameW(user_name, &num_chars)) {
- /* num_chars is the number of unicode chars plus null terminator */
- result = PyUnicode_FromWideChar(user_name, num_chars - 1);
- }
- else
- result = PyErr_SetFromWindowsErr(GetLastError());
- #else
- char *name;
- int old_errno = errno;
- errno = 0;
- name = getlogin();
- if (name == NULL) {
- if (errno)
- posix_error();
- else
- PyErr_SetString(PyExc_OSError, "unable to determine login name");
- }
- else
- result = PyUnicode_DecodeFSDefault(name);
- errno = old_errno;
- #endif
- return result;
- }
- #endif /* HAVE_GETLOGIN */
- #ifdef HAVE_GETUID
- /*[clinic input]
- os.getuid
- Return the current process's user id.
- [clinic start generated code]*/
- static PyObject *
- os_getuid_impl(PyObject *module)
- /*[clinic end generated code: output=415c0b401ebed11a input=b53c8b35f110a516]*/
- {
- return _PyLong_FromUid(getuid());
- }
- #endif /* HAVE_GETUID */
- #ifdef MS_WINDOWS
- #define HAVE_KILL
- #endif /* MS_WINDOWS */
- #ifdef HAVE_KILL
- /*[clinic input]
- os.kill
- pid: pid_t
- signal: Py_ssize_t
- /
- Kill a process with a signal.
- [clinic start generated code]*/
- static PyObject *
- os_kill_impl(PyObject *module, pid_t pid, Py_ssize_t signal)
- /*[clinic end generated code: output=8e346a6701c88568 input=61a36b86ca275ab9]*/
- {
- if (PySys_Audit("os.kill", "in", pid, signal) < 0) {
- return NULL;
- }
- #ifndef MS_WINDOWS
- if (kill(pid, (int)signal) == -1) {
- return posix_error();
- }
- // Check immediately if the signal was sent to the current process.
- // Don't micro-optimize pid == getpid(), since PyErr_SetString() check
- // is cheap.
- if (PyErr_CheckSignals()) {
- return NULL;
- }
- Py_RETURN_NONE;
- #else /* !MS_WINDOWS */
- PyObject *result;
- DWORD sig = (DWORD)signal;
- DWORD err;
- HANDLE handle;
- #ifdef HAVE_WINDOWS_CONSOLE_IO
- /* Console processes which share a common console can be sent CTRL+C or
- CTRL+BREAK events, provided they handle said events. */
- if (sig == CTRL_C_EVENT || sig == CTRL_BREAK_EVENT) {
- if (GenerateConsoleCtrlEvent(sig, (DWORD)pid) == 0) {
- err = GetLastError();
- PyErr_SetFromWindowsErr(err);
- }
- else {
- Py_RETURN_NONE;
- }
- }
- #endif /* HAVE_WINDOWS_CONSOLE_IO */
- /* If the signal is outside of what GenerateConsoleCtrlEvent can use,
- attempt to open and terminate the process. */
- handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, (DWORD)pid);
- if (handle == NULL) {
- err = GetLastError();
- return PyErr_SetFromWindowsErr(err);
- }
- if (TerminateProcess(handle, sig) == 0) {
- err = GetLastError();
- result = PyErr_SetFromWindowsErr(err);
- } else {
- result = Py_NewRef(Py_None);
- }
- CloseHandle(handle);
- return result;
- #endif /* !MS_WINDOWS */
- }
- #endif /* HAVE_KILL */
- #ifdef HAVE_KILLPG
- /*[clinic input]
- os.killpg
- pgid: pid_t
- signal: int
- /
- Kill a process group with a signal.
- [clinic start generated code]*/
- static PyObject *
- os_killpg_impl(PyObject *module, pid_t pgid, int signal)
- /*[clinic end generated code: output=6dbcd2f1fdf5fdba input=38b5449eb8faec19]*/
- {
- if (PySys_Audit("os.killpg", "ii", pgid, signal) < 0) {
- return NULL;
- }
- /* XXX some man pages make the `pgid` parameter an int, others
- a pid_t. Since getpgrp() returns a pid_t, we assume killpg should
- take the same type. Moreover, pid_t is always at least as wide as
- int (else compilation of this module fails), which is safe. */
- if (killpg(pgid, signal) == -1)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_KILLPG */
- #ifdef HAVE_PLOCK
- #ifdef HAVE_SYS_LOCK_H
- #include <sys/lock.h>
- #endif
- /*[clinic input]
- os.plock
- op: int
- /
- Lock program segments into memory.");
- [clinic start generated code]*/
- static PyObject *
- os_plock_impl(PyObject *module, int op)
- /*[clinic end generated code: output=81424167033b168e input=e6e5e348e1525f60]*/
- {
- if (plock(op) == -1)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_PLOCK */
- #ifdef HAVE_SETUID
- /*[clinic input]
- os.setuid
- uid: uid_t
- /
- Set the current process's user id.
- [clinic start generated code]*/
- static PyObject *
- os_setuid_impl(PyObject *module, uid_t uid)
- /*[clinic end generated code: output=a0a41fd0d1ec555f input=c921a3285aa22256]*/
- {
- if (setuid(uid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETUID */
- #ifdef HAVE_SETEUID
- /*[clinic input]
- os.seteuid
- euid: uid_t
- /
- Set the current process's effective user id.
- [clinic start generated code]*/
- static PyObject *
- os_seteuid_impl(PyObject *module, uid_t euid)
- /*[clinic end generated code: output=102e3ad98361519a input=ba93d927e4781aa9]*/
- {
- if (seteuid(euid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETEUID */
- #ifdef HAVE_SETEGID
- /*[clinic input]
- os.setegid
- egid: gid_t
- /
- Set the current process's effective group id.
- [clinic start generated code]*/
- static PyObject *
- os_setegid_impl(PyObject *module, gid_t egid)
- /*[clinic end generated code: output=4e4b825a6a10258d input=4080526d0ccd6ce3]*/
- {
- if (setegid(egid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETEGID */
- #ifdef HAVE_SETREUID
- /*[clinic input]
- os.setreuid
- ruid: uid_t
- euid: uid_t
- /
- Set the current process's real and effective user ids.
- [clinic start generated code]*/
- static PyObject *
- os_setreuid_impl(PyObject *module, uid_t ruid, uid_t euid)
- /*[clinic end generated code: output=62d991210006530a input=0ca8978de663880c]*/
- {
- if (setreuid(ruid, euid) < 0) {
- return posix_error();
- } else {
- Py_RETURN_NONE;
- }
- }
- #endif /* HAVE_SETREUID */
- #ifdef HAVE_SETREGID
- /*[clinic input]
- os.setregid
- rgid: gid_t
- egid: gid_t
- /
- Set the current process's real and effective group ids.
- [clinic start generated code]*/
- static PyObject *
- os_setregid_impl(PyObject *module, gid_t rgid, gid_t egid)
- /*[clinic end generated code: output=aa803835cf5342f3 input=c59499f72846db78]*/
- {
- if (setregid(rgid, egid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETREGID */
- #ifdef HAVE_SETGID
- /*[clinic input]
- os.setgid
- gid: gid_t
- /
- Set the current process's group id.
- [clinic start generated code]*/
- static PyObject *
- os_setgid_impl(PyObject *module, gid_t gid)
- /*[clinic end generated code: output=bdccd7403f6ad8c3 input=27d30c4059045dc6]*/
- {
- if (setgid(gid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETGID */
- #ifdef HAVE_SETGROUPS
- /*[clinic input]
- os.setgroups
- groups: object
- /
- Set the groups of the current process to list.
- [clinic start generated code]*/
- static PyObject *
- os_setgroups(PyObject *module, PyObject *groups)
- /*[clinic end generated code: output=3fcb32aad58c5ecd input=fa742ca3daf85a7e]*/
- {
- if (!PySequence_Check(groups)) {
- PyErr_SetString(PyExc_TypeError, "setgroups argument must be a sequence");
- return NULL;
- }
- Py_ssize_t len = PySequence_Size(groups);
- if (len < 0) {
- return NULL;
- }
- if (len > MAX_GROUPS) {
- PyErr_SetString(PyExc_ValueError, "too many groups");
- return NULL;
- }
- gid_t *grouplist = PyMem_New(gid_t, len);
- for (Py_ssize_t i = 0; i < len; i++) {
- PyObject *elem;
- elem = PySequence_GetItem(groups, i);
- if (!elem) {
- PyMem_Free(grouplist);
- return NULL;
- }
- if (!PyLong_Check(elem)) {
- PyErr_SetString(PyExc_TypeError,
- "groups must be integers");
- Py_DECREF(elem);
- PyMem_Free(grouplist);
- return NULL;
- } else {
- if (!_Py_Gid_Converter(elem, &grouplist[i])) {
- Py_DECREF(elem);
- PyMem_Free(grouplist);
- return NULL;
- }
- }
- Py_DECREF(elem);
- }
- if (setgroups(len, grouplist) < 0) {
- posix_error();
- PyMem_Free(grouplist);
- return NULL;
- }
- PyMem_Free(grouplist);
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETGROUPS */
- #if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
- static PyObject *
- wait_helper(PyObject *module, pid_t pid, int status, struct rusage *ru)
- {
- PyObject *result;
- PyObject *struct_rusage;
- if (pid == -1)
- return posix_error();
- // If wait succeeded but no child was ready to report status, ru will not
- // have been populated.
- if (pid == 0) {
- memset(ru, 0, sizeof(*ru));
- }
- struct_rusage = _PyImport_GetModuleAttrString("resource", "struct_rusage");
- if (struct_rusage == NULL)
- return NULL;
- /* XXX(nnorwitz): Copied (w/mods) from resource.c, there should be only one. */
- result = PyStructSequence_New((PyTypeObject*) struct_rusage);
- Py_DECREF(struct_rusage);
- if (!result)
- return NULL;
- int pos = 0;
- #ifndef doubletime
- #define doubletime(TV) ((double)(TV).tv_sec + (TV).tv_usec * 0.000001)
- #endif
- #define SET_RESULT(CALL) \
- do { \
- PyObject *item = (CALL); \
- if (item == NULL) { \
- Py_DECREF(result); \
- return NULL; \
- } \
- PyStructSequence_SET_ITEM(result, pos++, item); \
- } while(0)
- SET_RESULT(PyFloat_FromDouble(doubletime(ru->ru_utime)));
- SET_RESULT(PyFloat_FromDouble(doubletime(ru->ru_stime)));
- SET_RESULT(PyLong_FromLong(ru->ru_maxrss));
- SET_RESULT(PyLong_FromLong(ru->ru_ixrss));
- SET_RESULT(PyLong_FromLong(ru->ru_idrss));
- SET_RESULT(PyLong_FromLong(ru->ru_isrss));
- SET_RESULT(PyLong_FromLong(ru->ru_minflt));
- SET_RESULT(PyLong_FromLong(ru->ru_majflt));
- SET_RESULT(PyLong_FromLong(ru->ru_nswap));
- SET_RESULT(PyLong_FromLong(ru->ru_inblock));
- SET_RESULT(PyLong_FromLong(ru->ru_oublock));
- SET_RESULT(PyLong_FromLong(ru->ru_msgsnd));
- SET_RESULT(PyLong_FromLong(ru->ru_msgrcv));
- SET_RESULT(PyLong_FromLong(ru->ru_nsignals));
- SET_RESULT(PyLong_FromLong(ru->ru_nvcsw));
- SET_RESULT(PyLong_FromLong(ru->ru_nivcsw));
- #undef SET_RESULT
- return Py_BuildValue("NiN", PyLong_FromPid(pid), status, result);
- }
- #endif /* HAVE_WAIT3 || HAVE_WAIT4 */
- #ifdef HAVE_WAIT3
- /*[clinic input]
- os.wait3
- options: int
- Wait for completion of a child process.
- Returns a tuple of information about the child process:
- (pid, status, rusage)
- [clinic start generated code]*/
- static PyObject *
- os_wait3_impl(PyObject *module, int options)
- /*[clinic end generated code: output=92c3224e6f28217a input=8ac4c56956b61710]*/
- {
- pid_t pid;
- struct rusage ru;
- int async_err = 0;
- WAIT_TYPE status;
- WAIT_STATUS_INT(status) = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- pid = wait3(&status, options, &ru);
- Py_END_ALLOW_THREADS
- } while (pid < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (pid < 0)
- return (!async_err) ? posix_error() : NULL;
- return wait_helper(module, pid, WAIT_STATUS_INT(status), &ru);
- }
- #endif /* HAVE_WAIT3 */
- #ifdef HAVE_WAIT4
- /*[clinic input]
- os.wait4
- pid: pid_t
- options: int
- Wait for completion of a specific child process.
- Returns a tuple of information about the child process:
- (pid, status, rusage)
- [clinic start generated code]*/
- static PyObject *
- os_wait4_impl(PyObject *module, pid_t pid, int options)
- /*[clinic end generated code: output=66195aa507b35f70 input=d11deed0750600ba]*/
- {
- pid_t res;
- struct rusage ru;
- int async_err = 0;
- WAIT_TYPE status;
- WAIT_STATUS_INT(status) = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- res = wait4(pid, &status, options, &ru);
- Py_END_ALLOW_THREADS
- } while (res < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res < 0)
- return (!async_err) ? posix_error() : NULL;
- return wait_helper(module, res, WAIT_STATUS_INT(status), &ru);
- }
- #endif /* HAVE_WAIT4 */
- #if defined(HAVE_WAITID) && !defined(__APPLE__)
- /*[clinic input]
- os.waitid
- idtype: idtype_t
- Must be one of be P_PID, P_PGID or P_ALL.
- id: id_t
- The id to wait on.
- options: int
- Constructed from the ORing of one or more of WEXITED, WSTOPPED
- or WCONTINUED and additionally may be ORed with WNOHANG or WNOWAIT.
- /
- Returns the result of waiting for a process or processes.
- Returns either waitid_result or None if WNOHANG is specified and there are
- no children in a waitable state.
- [clinic start generated code]*/
- static PyObject *
- os_waitid_impl(PyObject *module, idtype_t idtype, id_t id, int options)
- /*[clinic end generated code: output=5d2e1c0bde61f4d8 input=d8e7f76e052b7920]*/
- {
- PyObject *result;
- int res;
- int async_err = 0;
- siginfo_t si;
- si.si_pid = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- res = waitid(idtype, id, &si, options);
- Py_END_ALLOW_THREADS
- } while (res < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res < 0)
- return (!async_err) ? posix_error() : NULL;
- if (si.si_pid == 0)
- Py_RETURN_NONE;
- PyObject *WaitidResultType = get_posix_state(module)->WaitidResultType;
- result = PyStructSequence_New((PyTypeObject *)WaitidResultType);
- if (!result)
- return NULL;
- int pos = 0;
- #define SET_RESULT(CALL) \
- do { \
- PyObject *item = (CALL); \
- if (item == NULL) { \
- Py_DECREF(result); \
- return NULL; \
- } \
- PyStructSequence_SET_ITEM(result, pos++, item); \
- } while(0)
- SET_RESULT(PyLong_FromPid(si.si_pid));
- SET_RESULT(_PyLong_FromUid(si.si_uid));
- SET_RESULT(PyLong_FromLong((long)(si.si_signo)));
- SET_RESULT(PyLong_FromLong((long)(si.si_status)));
- SET_RESULT(PyLong_FromLong((long)(si.si_code)));
- #undef SET_RESULT
- return result;
- }
- #endif /* defined(HAVE_WAITID) && !defined(__APPLE__) */
- #if defined(HAVE_WAITPID)
- /*[clinic input]
- os.waitpid
- pid: pid_t
- options: int
- /
- Wait for completion of a given child process.
- Returns a tuple of information regarding the child process:
- (pid, status)
- The options argument is ignored on Windows.
- [clinic start generated code]*/
- static PyObject *
- os_waitpid_impl(PyObject *module, pid_t pid, int options)
- /*[clinic end generated code: output=5c37c06887a20270 input=0bf1666b8758fda3]*/
- {
- pid_t res;
- int async_err = 0;
- WAIT_TYPE status;
- WAIT_STATUS_INT(status) = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- res = waitpid(pid, &status, options);
- Py_END_ALLOW_THREADS
- } while (res < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res < 0)
- return (!async_err) ? posix_error() : NULL;
- return Py_BuildValue("Ni", PyLong_FromPid(res), WAIT_STATUS_INT(status));
- }
- #elif defined(HAVE_CWAIT)
- /* MS C has a variant of waitpid() that's usable for most purposes. */
- /*[clinic input]
- os.waitpid
- pid: intptr_t
- options: int
- /
- Wait for completion of a given process.
- Returns a tuple of information regarding the process:
- (pid, status << 8)
- The options argument is ignored on Windows.
- [clinic start generated code]*/
- static PyObject *
- os_waitpid_impl(PyObject *module, intptr_t pid, int options)
- /*[clinic end generated code: output=be836b221271d538 input=40f2440c515410f8]*/
- {
- int status;
- intptr_t res;
- int async_err = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- res = _cwait(&status, pid, options);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (res < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res < 0)
- return (!async_err) ? posix_error() : NULL;
- unsigned long long ustatus = (unsigned int)status;
- /* shift the status left a byte so this is more like the POSIX waitpid */
- return Py_BuildValue(_Py_PARSE_INTPTR "K", res, ustatus << 8);
- }
- #endif
- #ifdef HAVE_WAIT
- /*[clinic input]
- os.wait
- Wait for completion of a child process.
- Returns a tuple of information about the child process:
- (pid, status)
- [clinic start generated code]*/
- static PyObject *
- os_wait_impl(PyObject *module)
- /*[clinic end generated code: output=6bc419ac32fb364b input=03b0182d4a4700ce]*/
- {
- pid_t pid;
- int async_err = 0;
- WAIT_TYPE status;
- WAIT_STATUS_INT(status) = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- pid = wait(&status);
- Py_END_ALLOW_THREADS
- } while (pid < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (pid < 0)
- return (!async_err) ? posix_error() : NULL;
- return Py_BuildValue("Ni", PyLong_FromPid(pid), WAIT_STATUS_INT(status));
- }
- #endif /* HAVE_WAIT */
- #if defined(__linux__) && defined(__NR_pidfd_open)
- /*[clinic input]
- os.pidfd_open
- pid: pid_t
- flags: unsigned_int = 0
- Return a file descriptor referring to the process *pid*.
- The descriptor can be used to perform process management without races and
- signals.
- [clinic start generated code]*/
- static PyObject *
- os_pidfd_open_impl(PyObject *module, pid_t pid, unsigned int flags)
- /*[clinic end generated code: output=5c7252698947dc41 input=c3fd99ce947ccfef]*/
- {
- int fd = syscall(__NR_pidfd_open, pid, flags);
- if (fd < 0) {
- return posix_error();
- }
- return PyLong_FromLong(fd);
- }
- #endif
- #ifdef HAVE_SETNS
- /*[clinic input]
- os.setns
- fd: fildes
- A file descriptor to a namespace.
- nstype: int = 0
- Type of namespace.
- Move the calling thread into different namespaces.
- [clinic start generated code]*/
- static PyObject *
- os_setns_impl(PyObject *module, int fd, int nstype)
- /*[clinic end generated code: output=5dbd055bfb66ecd0 input=42787871226bf3ee]*/
- {
- int res;
- Py_BEGIN_ALLOW_THREADS
- res = setns(fd, nstype);
- Py_END_ALLOW_THREADS
- if (res != 0) {
- return posix_error();
- }
- Py_RETURN_NONE;
- }
- #endif
- #ifdef HAVE_UNSHARE
- /*[clinic input]
- os.unshare
- flags: int
- Namespaces to be unshared.
- Disassociate parts of a process (or thread) execution context.
- [clinic start generated code]*/
- static PyObject *
- os_unshare_impl(PyObject *module, int flags)
- /*[clinic end generated code: output=1b3177906dd237ee input=9e065db3232b8b1b]*/
- {
- int res;
- Py_BEGIN_ALLOW_THREADS
- res = unshare(flags);
- Py_END_ALLOW_THREADS
- if (res != 0) {
- return posix_error();
- }
- Py_RETURN_NONE;
- }
- #endif
- #if defined(HAVE_READLINK) || defined(MS_WINDOWS)
- /*[clinic input]
- os.readlink
- path: path_t
- *
- dir_fd: dir_fd(requires='readlinkat') = None
- Return a string representing the path to which the symbolic link points.
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform. If it is unavailable,
- using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_readlink_impl(PyObject *module, path_t *path, int dir_fd)
- /*[clinic end generated code: output=d21b732a2e814030 input=113c87e0db1ecaf2]*/
- {
- #if defined(HAVE_READLINK)
- char buffer[MAXPATHLEN+1];
- ssize_t length;
- #ifdef HAVE_READLINKAT
- int readlinkat_unavailable = 0;
- #endif
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_READLINKAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_READLINKAT_RUNTIME) {
- length = readlinkat(dir_fd, path->narrow, buffer, MAXPATHLEN);
- } else {
- readlinkat_unavailable = 1;
- }
- } else
- #endif
- length = readlink(path->narrow, buffer, MAXPATHLEN);
- Py_END_ALLOW_THREADS
- #ifdef HAVE_READLINKAT
- if (readlinkat_unavailable) {
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (length < 0) {
- return path_error(path);
- }
- buffer[length] = '\0';
- if (PyUnicode_Check(path->object))
- return PyUnicode_DecodeFSDefaultAndSize(buffer, length);
- else
- return PyBytes_FromStringAndSize(buffer, length);
- #elif defined(MS_WINDOWS)
- DWORD n_bytes_returned;
- DWORD io_result = 0;
- HANDLE reparse_point_handle;
- char target_buffer[_Py_MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
- _Py_REPARSE_DATA_BUFFER *rdb = (_Py_REPARSE_DATA_BUFFER *)target_buffer;
- PyObject *result = NULL;
- /* First get a handle to the reparse point */
- Py_BEGIN_ALLOW_THREADS
- reparse_point_handle = CreateFileW(
- path->wide,
- 0,
- 0,
- 0,
- OPEN_EXISTING,
- FILE_FLAG_OPEN_REPARSE_POINT|FILE_FLAG_BACKUP_SEMANTICS,
- 0);
- if (reparse_point_handle != INVALID_HANDLE_VALUE) {
- /* New call DeviceIoControl to read the reparse point */
- io_result = DeviceIoControl(
- reparse_point_handle,
- FSCTL_GET_REPARSE_POINT,
- 0, 0, /* in buffer */
- target_buffer, sizeof(target_buffer),
- &n_bytes_returned,
- 0 /* we're not using OVERLAPPED_IO */
- );
- CloseHandle(reparse_point_handle);
- }
- Py_END_ALLOW_THREADS
- if (io_result == 0) {
- return path_error(path);
- }
- wchar_t *name = NULL;
- Py_ssize_t nameLen = 0;
- if (rdb->ReparseTag == IO_REPARSE_TAG_SYMLINK)
- {
- name = (wchar_t *)((char*)rdb->SymbolicLinkReparseBuffer.PathBuffer +
- rdb->SymbolicLinkReparseBuffer.SubstituteNameOffset);
- nameLen = rdb->SymbolicLinkReparseBuffer.SubstituteNameLength / sizeof(wchar_t);
- }
- else if (rdb->ReparseTag == IO_REPARSE_TAG_MOUNT_POINT)
- {
- name = (wchar_t *)((char*)rdb->MountPointReparseBuffer.PathBuffer +
- rdb->MountPointReparseBuffer.SubstituteNameOffset);
- nameLen = rdb->MountPointReparseBuffer.SubstituteNameLength / sizeof(wchar_t);
- }
- else
- {
- PyErr_SetString(PyExc_ValueError, "not a symbolic link");
- }
- if (name) {
- if (nameLen > 4 && wcsncmp(name, L"\\??\\", 4) == 0) {
- /* Our buffer is mutable, so this is okay */
- name[1] = L'\\';
- }
- result = PyUnicode_FromWideChar(name, nameLen);
- if (result && PyBytes_Check(path->object)) {
- Py_SETREF(result, PyUnicode_EncodeFSDefault(result));
- }
- }
- return result;
- #endif
- }
- #endif /* defined(HAVE_READLINK) || defined(MS_WINDOWS) */
- #if defined(MS_WINDOWS)
- /* Remove the last portion of the path - return 0 on success */
- static int
- _dirnameW(WCHAR *path)
- {
- WCHAR *ptr;
- size_t length = wcsnlen_s(path, MAX_PATH);
- if (length == MAX_PATH) {
- return -1;
- }
- /* walk the path from the end until a backslash is encountered */
- for(ptr = path + length; ptr != path; ptr--) {
- if (*ptr == L'\\' || *ptr == L'/') {
- break;
- }
- }
- *ptr = 0;
- return 0;
- }
- #endif
- #ifdef HAVE_SYMLINK
- #if defined(MS_WINDOWS)
- /* Is this path absolute? */
- static int
- _is_absW(const WCHAR *path)
- {
- return path[0] == L'\\' || path[0] == L'/' ||
- (path[0] && path[1] == L':');
- }
- /* join root and rest with a backslash - return 0 on success */
- static int
- _joinW(WCHAR *dest_path, const WCHAR *root, const WCHAR *rest)
- {
- if (_is_absW(rest)) {
- return wcscpy_s(dest_path, MAX_PATH, rest);
- }
- if (wcscpy_s(dest_path, MAX_PATH, root)) {
- return -1;
- }
- if (dest_path[0] && wcscat_s(dest_path, MAX_PATH, L"\\")) {
- return -1;
- }
- return wcscat_s(dest_path, MAX_PATH, rest);
- }
- /* Return True if the path at src relative to dest is a directory */
- static int
- _check_dirW(LPCWSTR src, LPCWSTR dest)
- {
- WIN32_FILE_ATTRIBUTE_DATA src_info;
- WCHAR dest_parent[MAX_PATH];
- WCHAR src_resolved[MAX_PATH] = L"";
- /* dest_parent = os.path.dirname(dest) */
- if (wcscpy_s(dest_parent, MAX_PATH, dest) ||
- _dirnameW(dest_parent)) {
- return 0;
- }
- /* src_resolved = os.path.join(dest_parent, src) */
- if (_joinW(src_resolved, dest_parent, src)) {
- return 0;
- }
- return (
- GetFileAttributesExW(src_resolved, GetFileExInfoStandard, &src_info)
- && src_info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY
- );
- }
- #endif
- /*[clinic input]
- os.symlink
- src: path_t
- dst: path_t
- target_is_directory: bool = False
- *
- dir_fd: dir_fd(requires='symlinkat')=None
- # "symlink(src, dst, target_is_directory=False, *, dir_fd=None)\n\n\
- Create a symbolic link pointing to src named dst.
- target_is_directory is required on Windows if the target is to be
- interpreted as a directory. (On Windows, symlink requires
- Windows 6.0 or greater, and raises a NotImplementedError otherwise.)
- target_is_directory is ignored on non-Windows platforms.
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_symlink_impl(PyObject *module, path_t *src, path_t *dst,
- int target_is_directory, int dir_fd)
- /*[clinic end generated code: output=08ca9f3f3cf960f6 input=e820ec4472547bc3]*/
- {
- #ifdef MS_WINDOWS
- DWORD result;
- DWORD flags = 0;
- /* Assumed true, set to false if detected to not be available. */
- static int windows_has_symlink_unprivileged_flag = TRUE;
- #else
- int result;
- #ifdef HAVE_SYMLINKAT
- int symlinkat_unavailable = 0;
- #endif
- #endif
- if (PySys_Audit("os.symlink", "OOi", src->object, dst->object,
- dir_fd == DEFAULT_DIR_FD ? -1 : dir_fd) < 0) {
- return NULL;
- }
- #ifdef MS_WINDOWS
- if (windows_has_symlink_unprivileged_flag) {
- /* Allow non-admin symlinks if system allows it. */
- flags |= SYMBOLIC_LINK_FLAG_ALLOW_UNPRIVILEGED_CREATE;
- }
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- /* if src is a directory, ensure flags==1 (target_is_directory bit) */
- if (target_is_directory || _check_dirW(src->wide, dst->wide)) {
- flags |= SYMBOLIC_LINK_FLAG_DIRECTORY;
- }
- result = CreateSymbolicLinkW(dst->wide, src->wide, flags);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- if (windows_has_symlink_unprivileged_flag && !result &&
- ERROR_INVALID_PARAMETER == GetLastError()) {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- /* This error might be caused by
- SYMBOLIC_LINK_FLAG_ALLOW_UNPRIVILEGED_CREATE not being supported.
- Try again, and update windows_has_symlink_unprivileged_flag if we
- are successful this time.
- NOTE: There is a risk of a race condition here if there are other
- conditions than the flag causing ERROR_INVALID_PARAMETER, and
- another process (or thread) changes that condition in between our
- calls to CreateSymbolicLink.
- */
- flags &= ~(SYMBOLIC_LINK_FLAG_ALLOW_UNPRIVILEGED_CREATE);
- result = CreateSymbolicLinkW(dst->wide, src->wide, flags);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- if (result || ERROR_INVALID_PARAMETER != GetLastError()) {
- windows_has_symlink_unprivileged_flag = FALSE;
- }
- }
- if (!result)
- return path_error2(src, dst);
- #else
- if ((src->narrow && dst->wide) || (src->wide && dst->narrow)) {
- PyErr_SetString(PyExc_ValueError,
- "symlink: src and dst must be the same type");
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_SYMLINKAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_SYMLINKAT_RUNTIME) {
- result = symlinkat(src->narrow, dir_fd, dst->narrow);
- } else {
- symlinkat_unavailable = 1;
- }
- } else
- #endif
- result = symlink(src->narrow, dst->narrow);
- Py_END_ALLOW_THREADS
- #ifdef HAVE_SYMLINKAT
- if (symlinkat_unavailable) {
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (result)
- return path_error2(src, dst);
- #endif
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SYMLINK */
- static PyStructSequence_Field times_result_fields[] = {
- {"user", "user time"},
- {"system", "system time"},
- {"children_user", "user time of children"},
- {"children_system", "system time of children"},
- {"elapsed", "elapsed time since an arbitrary point in the past"},
- {NULL}
- };
- PyDoc_STRVAR(times_result__doc__,
- "times_result: Result from os.times().\n\n\
- This object may be accessed either as a tuple of\n\
- (user, system, children_user, children_system, elapsed),\n\
- or via the attributes user, system, children_user, children_system,\n\
- and elapsed.\n\
- \n\
- See os.times for more information.");
- static PyStructSequence_Desc times_result_desc = {
- "times_result", /* name */
- times_result__doc__, /* doc */
- times_result_fields,
- 5
- };
- #ifdef MS_WINDOWS
- #define HAVE_TIMES /* mandatory, for the method table */
- #endif
- #ifdef HAVE_TIMES
- static PyObject *
- build_times_result(PyObject *module, double user, double system,
- double children_user, double children_system,
- double elapsed)
- {
- PyObject *TimesResultType = get_posix_state(module)->TimesResultType;
- PyObject *value = PyStructSequence_New((PyTypeObject *)TimesResultType);
- if (value == NULL)
- return NULL;
- #define SET(i, field) \
- { \
- PyObject *o = PyFloat_FromDouble(field); \
- if (!o) { \
- Py_DECREF(value); \
- return NULL; \
- } \
- PyStructSequence_SET_ITEM(value, i, o); \
- } \
- SET(0, user);
- SET(1, system);
- SET(2, children_user);
- SET(3, children_system);
- SET(4, elapsed);
- #undef SET
- return value;
- }
- /*[clinic input]
- os.times
- Return a collection containing process timing information.
- The object returned behaves like a named tuple with these fields:
- (utime, stime, cutime, cstime, elapsed_time)
- All fields are floating-point numbers.
- [clinic start generated code]*/
- static PyObject *
- os_times_impl(PyObject *module)
- /*[clinic end generated code: output=35f640503557d32a input=8dbfe33a2dcc3df3]*/
- #ifdef MS_WINDOWS
- {
- FILETIME create, exit, kernel, user;
- HANDLE hProc;
- hProc = GetCurrentProcess();
- GetProcessTimes(hProc, &create, &exit, &kernel, &user);
- /* The fields of a FILETIME structure are the hi and lo part
- of a 64-bit value expressed in 100 nanosecond units.
- 1e7 is one second in such units; 1e-7 the inverse.
- 429.4967296 is 2**32 / 1e7 or 2**32 * 1e-7.
- */
- return build_times_result(module,
- (double)(user.dwHighDateTime*429.4967296 +
- user.dwLowDateTime*1e-7),
- (double)(kernel.dwHighDateTime*429.4967296 +
- kernel.dwLowDateTime*1e-7),
- (double)0,
- (double)0,
- (double)0);
- }
- #else /* MS_WINDOWS */
- {
- struct tms t;
- clock_t c;
- errno = 0;
- c = times(&t);
- if (c == (clock_t) -1) {
- return posix_error();
- }
- assert(_PyRuntime.time.ticks_per_second_initialized);
- #define ticks_per_second _PyRuntime.time.ticks_per_second
- return build_times_result(module,
- (double)t.tms_utime / ticks_per_second,
- (double)t.tms_stime / ticks_per_second,
- (double)t.tms_cutime / ticks_per_second,
- (double)t.tms_cstime / ticks_per_second,
- (double)c / ticks_per_second);
- #undef ticks_per_second
- }
- #endif /* MS_WINDOWS */
- #endif /* HAVE_TIMES */
- #ifdef HAVE_GETSID
- /*[clinic input]
- os.getsid
- pid: pid_t
- /
- Call the system call getsid(pid) and return the result.
- [clinic start generated code]*/
- static PyObject *
- os_getsid_impl(PyObject *module, pid_t pid)
- /*[clinic end generated code: output=112deae56b306460 input=eeb2b923a30ce04e]*/
- {
- int sid;
- sid = getsid(pid);
- if (sid < 0)
- return posix_error();
- return PyLong_FromLong((long)sid);
- }
- #endif /* HAVE_GETSID */
- #ifdef HAVE_SETSID
- /*[clinic input]
- os.setsid
- Call the system call setsid().
- [clinic start generated code]*/
- static PyObject *
- os_setsid_impl(PyObject *module)
- /*[clinic end generated code: output=e2ddedd517086d77 input=5fff45858e2f0776]*/
- {
- if (setsid() < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETSID */
- #ifdef HAVE_SETPGID
- /*[clinic input]
- os.setpgid
- pid: pid_t
- pgrp: pid_t
- /
- Call the system call setpgid(pid, pgrp).
- [clinic start generated code]*/
- static PyObject *
- os_setpgid_impl(PyObject *module, pid_t pid, pid_t pgrp)
- /*[clinic end generated code: output=6461160319a43d6a input=fceb395eca572e1a]*/
- {
- if (setpgid(pid, pgrp) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETPGID */
- #ifdef HAVE_TCGETPGRP
- /*[clinic input]
- os.tcgetpgrp
- fd: int
- /
- Return the process group associated with the terminal specified by fd.
- [clinic start generated code]*/
- static PyObject *
- os_tcgetpgrp_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=f865e88be86c272b input=7f6c18eac10ada86]*/
- {
- pid_t pgid = tcgetpgrp(fd);
- if (pgid < 0)
- return posix_error();
- return PyLong_FromPid(pgid);
- }
- #endif /* HAVE_TCGETPGRP */
- #ifdef HAVE_TCSETPGRP
- /*[clinic input]
- os.tcsetpgrp
- fd: int
- pgid: pid_t
- /
- Set the process group associated with the terminal specified by fd.
- [clinic start generated code]*/
- static PyObject *
- os_tcsetpgrp_impl(PyObject *module, int fd, pid_t pgid)
- /*[clinic end generated code: output=f1821a381b9daa39 input=5bdc997c6a619020]*/
- {
- if (tcsetpgrp(fd, pgid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_TCSETPGRP */
- /* Functions acting on file descriptors */
- #ifdef O_CLOEXEC
- extern int _Py_open_cloexec_works;
- #endif
- /*[clinic input]
- os.open -> int
- path: path_t
- flags: int
- mode: int = 0o777
- *
- dir_fd: dir_fd(requires='openat') = None
- # "open(path, flags, mode=0o777, *, dir_fd=None)\n\n\
- Open a file for low level IO. Returns a file descriptor (integer).
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static int
- os_open_impl(PyObject *module, path_t *path, int flags, int mode, int dir_fd)
- /*[clinic end generated code: output=abc7227888c8bc73 input=ad8623b29acd2934]*/
- {
- int fd;
- int async_err = 0;
- #ifdef HAVE_OPENAT
- int openat_unavailable = 0;
- #endif
- #ifdef O_CLOEXEC
- int *atomic_flag_works = &_Py_open_cloexec_works;
- #elif !defined(MS_WINDOWS)
- int *atomic_flag_works = NULL;
- #endif
- #ifdef MS_WINDOWS
- flags |= O_NOINHERIT;
- #elif defined(O_CLOEXEC)
- flags |= O_CLOEXEC;
- #endif
- if (PySys_Audit("open", "OOi", path->object, Py_None, flags) < 0) {
- return -1;
- }
- _Py_BEGIN_SUPPRESS_IPH
- do {
- Py_BEGIN_ALLOW_THREADS
- #ifdef MS_WINDOWS
- fd = _wopen(path->wide, flags, mode);
- #else
- #ifdef HAVE_OPENAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_OPENAT_RUNTIME) {
- fd = openat(dir_fd, path->narrow, flags, mode);
- } else {
- openat_unavailable = 1;
- fd = -1;
- }
- } else
- #endif /* HAVE_OPENAT */
- fd = open(path->narrow, flags, mode);
- #endif /* !MS_WINDOWS */
- Py_END_ALLOW_THREADS
- } while (fd < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- _Py_END_SUPPRESS_IPH
- #ifdef HAVE_OPENAT
- if (openat_unavailable) {
- argument_unavailable_error(NULL, "dir_fd");
- return -1;
- }
- #endif
- if (fd < 0) {
- if (!async_err)
- PyErr_SetFromErrnoWithFilenameObject(PyExc_OSError, path->object);
- return -1;
- }
- #ifndef MS_WINDOWS
- if (_Py_set_inheritable(fd, 0, atomic_flag_works) < 0) {
- close(fd);
- return -1;
- }
- #endif
- return fd;
- }
- /*[clinic input]
- os.close
- fd: int
- Close a file descriptor.
- [clinic start generated code]*/
- static PyObject *
- os_close_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=2fe4e93602822c14 input=2bc42451ca5c3223]*/
- {
- int res;
- /* We do not want to retry upon EINTR: see http://lwn.net/Articles/576478/
- * and http://linux.derkeiler.com/Mailing-Lists/Kernel/2005-09/3000.html
- * for more details.
- */
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- res = close(fd);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- if (res < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- /*[clinic input]
- os.closerange
- fd_low: int
- fd_high: int
- /
- Closes all file descriptors in [fd_low, fd_high), ignoring errors.
- [clinic start generated code]*/
- static PyObject *
- os_closerange_impl(PyObject *module, int fd_low, int fd_high)
- /*[clinic end generated code: output=0ce5c20fcda681c2 input=5855a3d053ebd4ec]*/
- {
- Py_BEGIN_ALLOW_THREADS
- _Py_closerange(fd_low, fd_high - 1);
- Py_END_ALLOW_THREADS
- Py_RETURN_NONE;
- }
- /*[clinic input]
- os.dup -> int
- fd: int
- /
- Return a duplicate of a file descriptor.
- [clinic start generated code]*/
- static int
- os_dup_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=486f4860636b2a9f input=6f10f7ea97f7852a]*/
- {
- return _Py_dup(fd);
- }
- // dup2() is either provided by libc or dup2.c with AC_REPLACE_FUNCS().
- // dup2.c provides working dup2() if and only if F_DUPFD is available.
- #if (defined(HAVE_DUP3) || defined(F_DUPFD) || defined(MS_WINDOWS))
- /*[clinic input]
- os.dup2 -> int
- fd: int
- fd2: int
- inheritable: bool=True
- Duplicate file descriptor.
- [clinic start generated code]*/
- static int
- os_dup2_impl(PyObject *module, int fd, int fd2, int inheritable)
- /*[clinic end generated code: output=bc059d34a73404d1 input=c3cddda8922b038d]*/
- {
- int res = 0;
- #if defined(HAVE_DUP3) && \
- !(defined(HAVE_FCNTL_H) && defined(F_DUP2FD_CLOEXEC))
- /* dup3() is available on Linux 2.6.27+ and glibc 2.9 */
- static int dup3_works = -1;
- #endif
- /* dup2() can fail with EINTR if the target FD is already open, because it
- * then has to be closed. See os_close_impl() for why we don't handle EINTR
- * upon close(), and therefore below.
- */
- #ifdef MS_WINDOWS
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- res = dup2(fd, fd2);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- if (res < 0) {
- posix_error();
- return -1;
- }
- res = fd2; // msvcrt dup2 returns 0 on success.
- /* Character files like console cannot be make non-inheritable */
- if (!inheritable && _Py_set_inheritable(fd2, 0, NULL) < 0) {
- close(fd2);
- return -1;
- }
- #elif defined(HAVE_FCNTL_H) && defined(F_DUP2FD_CLOEXEC)
- Py_BEGIN_ALLOW_THREADS
- if (!inheritable)
- res = fcntl(fd, F_DUP2FD_CLOEXEC, fd2);
- else
- res = dup2(fd, fd2);
- Py_END_ALLOW_THREADS
- if (res < 0) {
- posix_error();
- return -1;
- }
- #else
- #ifdef HAVE_DUP3
- if (!inheritable && dup3_works != 0) {
- Py_BEGIN_ALLOW_THREADS
- res = dup3(fd, fd2, O_CLOEXEC);
- Py_END_ALLOW_THREADS
- if (res < 0) {
- if (dup3_works == -1)
- dup3_works = (errno != ENOSYS);
- if (dup3_works) {
- posix_error();
- return -1;
- }
- }
- }
- if (inheritable || dup3_works == 0)
- {
- #endif
- Py_BEGIN_ALLOW_THREADS
- res = dup2(fd, fd2);
- Py_END_ALLOW_THREADS
- if (res < 0) {
- posix_error();
- return -1;
- }
- if (!inheritable && _Py_set_inheritable(fd2, 0, NULL) < 0) {
- close(fd2);
- return -1;
- }
- #ifdef HAVE_DUP3
- }
- #endif
- #endif
- return res;
- }
- #endif
- #ifdef HAVE_LOCKF
- /*[clinic input]
- os.lockf
- fd: int
- An open file descriptor.
- command: int
- One of F_LOCK, F_TLOCK, F_ULOCK or F_TEST.
- length: Py_off_t
- The number of bytes to lock, starting at the current position.
- /
- Apply, test or remove a POSIX lock on an open file descriptor.
- [clinic start generated code]*/
- static PyObject *
- os_lockf_impl(PyObject *module, int fd, int command, Py_off_t length)
- /*[clinic end generated code: output=af7051f3e7c29651 input=65da41d2106e9b79]*/
- {
- int res;
- if (PySys_Audit("os.lockf", "iiL", fd, command, length) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- res = lockf(fd, command, length);
- Py_END_ALLOW_THREADS
- if (res < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_LOCKF */
- /*[clinic input]
- os.lseek -> Py_off_t
- fd: int
- An open file descriptor, as returned by os.open().
- position: Py_off_t
- Position, interpreted relative to 'whence'.
- whence as how: int
- The relative position to seek from. Valid values are:
- - SEEK_SET: seek from the start of the file.
- - SEEK_CUR: seek from the current file position.
- - SEEK_END: seek from the end of the file.
- /
- Set the position of a file descriptor. Return the new position.
- The return value is the number of bytes relative to the beginning of the file.
- [clinic start generated code]*/
- static Py_off_t
- os_lseek_impl(PyObject *module, int fd, Py_off_t position, int how)
- /*[clinic end generated code: output=971e1efb6b30bd2f input=f096e754c5367504]*/
- {
- Py_off_t result;
- #ifdef SEEK_SET
- /* Turn 0, 1, 2 into SEEK_{SET,CUR,END} */
- switch (how) {
- case 0: how = SEEK_SET; break;
- case 1: how = SEEK_CUR; break;
- case 2: how = SEEK_END; break;
- }
- #endif /* SEEK_END */
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef MS_WINDOWS
- result = _lseeki64(fd, position, how);
- #else
- result = lseek(fd, position, how);
- #endif
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- if (result < 0)
- posix_error();
- return result;
- }
- /*[clinic input]
- os.read
- fd: int
- length: Py_ssize_t
- /
- Read from a file descriptor. Returns a bytes object.
- [clinic start generated code]*/
- static PyObject *
- os_read_impl(PyObject *module, int fd, Py_ssize_t length)
- /*[clinic end generated code: output=dafbe9a5cddb987b input=1df2eaa27c0bf1d3]*/
- {
- Py_ssize_t n;
- PyObject *buffer;
- if (length < 0) {
- errno = EINVAL;
- return posix_error();
- }
- length = Py_MIN(length, _PY_READ_MAX);
- buffer = PyBytes_FromStringAndSize((char *)NULL, length);
- if (buffer == NULL)
- return NULL;
- n = _Py_read(fd, PyBytes_AS_STRING(buffer), length);
- if (n == -1) {
- Py_DECREF(buffer);
- return NULL;
- }
- if (n != length)
- _PyBytes_Resize(&buffer, n);
- return buffer;
- }
- #if (defined(HAVE_SENDFILE) && (defined(__FreeBSD__) || defined(__DragonFly__) \
- || defined(__APPLE__))) \
- || defined(HAVE_READV) || defined(HAVE_PREADV) || defined (HAVE_PREADV2) \
- || defined(HAVE_WRITEV) || defined(HAVE_PWRITEV) || defined (HAVE_PWRITEV2)
- static int
- iov_setup(struct iovec **iov, Py_buffer **buf, PyObject *seq, Py_ssize_t cnt, int type)
- {
- Py_ssize_t i, j;
- *iov = PyMem_New(struct iovec, cnt);
- if (*iov == NULL) {
- PyErr_NoMemory();
- return -1;
- }
- *buf = PyMem_New(Py_buffer, cnt);
- if (*buf == NULL) {
- PyMem_Free(*iov);
- PyErr_NoMemory();
- return -1;
- }
- for (i = 0; i < cnt; i++) {
- PyObject *item = PySequence_GetItem(seq, i);
- if (item == NULL)
- goto fail;
- if (PyObject_GetBuffer(item, &(*buf)[i], type) == -1) {
- Py_DECREF(item);
- goto fail;
- }
- Py_DECREF(item);
- (*iov)[i].iov_base = (*buf)[i].buf;
- (*iov)[i].iov_len = (*buf)[i].len;
- }
- return 0;
- fail:
- PyMem_Free(*iov);
- for (j = 0; j < i; j++) {
- PyBuffer_Release(&(*buf)[j]);
- }
- PyMem_Free(*buf);
- return -1;
- }
- static void
- iov_cleanup(struct iovec *iov, Py_buffer *buf, int cnt)
- {
- int i;
- PyMem_Free(iov);
- for (i = 0; i < cnt; i++) {
- PyBuffer_Release(&buf[i]);
- }
- PyMem_Free(buf);
- }
- #endif
- #ifdef HAVE_READV
- /*[clinic input]
- os.readv -> Py_ssize_t
- fd: int
- buffers: object
- /
- Read from a file descriptor fd into an iterable of buffers.
- The buffers should be mutable buffers accepting bytes.
- readv will transfer data into each buffer until it is full
- and then move on to the next buffer in the sequence to hold
- the rest of the data.
- readv returns the total number of bytes read,
- which may be less than the total capacity of all the buffers.
- [clinic start generated code]*/
- static Py_ssize_t
- os_readv_impl(PyObject *module, int fd, PyObject *buffers)
- /*[clinic end generated code: output=792da062d3fcebdb input=e679eb5dbfa0357d]*/
- {
- Py_ssize_t cnt, n;
- int async_err = 0;
- struct iovec *iov;
- Py_buffer *buf;
- if (!PySequence_Check(buffers)) {
- PyErr_SetString(PyExc_TypeError,
- "readv() arg 2 must be a sequence");
- return -1;
- }
- cnt = PySequence_Size(buffers);
- if (cnt < 0)
- return -1;
- if (iov_setup(&iov, &buf, buffers, cnt, PyBUF_WRITABLE) < 0)
- return -1;
- do {
- Py_BEGIN_ALLOW_THREADS
- n = readv(fd, iov, cnt);
- Py_END_ALLOW_THREADS
- } while (n < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- int saved_errno = errno;
- iov_cleanup(iov, buf, cnt);
- if (n < 0) {
- if (!async_err) {
- errno = saved_errno;
- posix_error();
- }
- return -1;
- }
- return n;
- }
- #endif /* HAVE_READV */
- #ifdef HAVE_PREAD
- /*[clinic input]
- os.pread
- fd: int
- length: Py_ssize_t
- offset: Py_off_t
- /
- Read a number of bytes from a file descriptor starting at a particular offset.
- Read length bytes from file descriptor fd, starting at offset bytes from
- the beginning of the file. The file offset remains unchanged.
- [clinic start generated code]*/
- static PyObject *
- os_pread_impl(PyObject *module, int fd, Py_ssize_t length, Py_off_t offset)
- /*[clinic end generated code: output=3f875c1eef82e32f input=85cb4a5589627144]*/
- {
- Py_ssize_t n;
- int async_err = 0;
- PyObject *buffer;
- if (length < 0) {
- errno = EINVAL;
- return posix_error();
- }
- buffer = PyBytes_FromStringAndSize((char *)NULL, length);
- if (buffer == NULL)
- return NULL;
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- n = pread(fd, PyBytes_AS_STRING(buffer), length, offset);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (n < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (n < 0) {
- if (!async_err) {
- posix_error();
- }
- Py_DECREF(buffer);
- return NULL;
- }
- if (n != length)
- _PyBytes_Resize(&buffer, n);
- return buffer;
- }
- #endif /* HAVE_PREAD */
- #if defined(HAVE_PREADV) || defined (HAVE_PREADV2)
- /*[clinic input]
- os.preadv -> Py_ssize_t
- fd: int
- buffers: object
- offset: Py_off_t
- flags: int = 0
- /
- Reads from a file descriptor into a number of mutable bytes-like objects.
- Combines the functionality of readv() and pread(). As readv(), it will
- transfer data into each buffer until it is full and then move on to the next
- buffer in the sequence to hold the rest of the data. Its fourth argument,
- specifies the file offset at which the input operation is to be performed. It
- will return the total number of bytes read (which can be less than the total
- capacity of all the objects).
- The flags argument contains a bitwise OR of zero or more of the following flags:
- - RWF_HIPRI
- - RWF_NOWAIT
- Using non-zero flags requires Linux 4.6 or newer.
- [clinic start generated code]*/
- static Py_ssize_t
- os_preadv_impl(PyObject *module, int fd, PyObject *buffers, Py_off_t offset,
- int flags)
- /*[clinic end generated code: output=26fc9c6e58e7ada5 input=4173919dc1f7ed99]*/
- {
- Py_ssize_t cnt, n;
- int async_err = 0;
- struct iovec *iov;
- Py_buffer *buf;
- if (!PySequence_Check(buffers)) {
- PyErr_SetString(PyExc_TypeError,
- "preadv2() arg 2 must be a sequence");
- return -1;
- }
- cnt = PySequence_Size(buffers);
- if (cnt < 0) {
- return -1;
- }
- #ifndef HAVE_PREADV2
- if(flags != 0) {
- argument_unavailable_error("preadv2", "flags");
- return -1;
- }
- #endif
- if (iov_setup(&iov, &buf, buffers, cnt, PyBUF_WRITABLE) < 0) {
- return -1;
- }
- #ifdef HAVE_PREADV2
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- n = preadv2(fd, iov, cnt, offset, flags);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (n < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- #else
- do {
- #if defined(__APPLE__) && defined(__clang__)
- /* This entire function will be removed from the module dict when the API
- * is not available.
- */
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wunguarded-availability"
- #pragma clang diagnostic ignored "-Wunguarded-availability-new"
- #endif
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- n = preadv(fd, iov, cnt, offset);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (n < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- #if defined(__APPLE__) && defined(__clang__)
- #pragma clang diagnostic pop
- #endif
- #endif
- int saved_errno = errno;
- iov_cleanup(iov, buf, cnt);
- if (n < 0) {
- if (!async_err) {
- errno = saved_errno;
- posix_error();
- }
- return -1;
- }
- return n;
- }
- #endif /* HAVE_PREADV */
- /*[clinic input]
- os.write -> Py_ssize_t
- fd: int
- data: Py_buffer
- /
- Write a bytes object to a file descriptor.
- [clinic start generated code]*/
- static Py_ssize_t
- os_write_impl(PyObject *module, int fd, Py_buffer *data)
- /*[clinic end generated code: output=e4ef5bc904b58ef9 input=3207e28963234f3c]*/
- {
- return _Py_write(fd, data->buf, data->len);
- }
- #ifdef HAVE_SENDFILE
- #ifdef __APPLE__
- /*[clinic input]
- os.sendfile
- out_fd: int
- in_fd: int
- offset: Py_off_t
- count as sbytes: Py_off_t
- headers: object(c_default="NULL") = ()
- trailers: object(c_default="NULL") = ()
- flags: int = 0
- Copy count bytes from file descriptor in_fd to file descriptor out_fd.
- [clinic start generated code]*/
- static PyObject *
- os_sendfile_impl(PyObject *module, int out_fd, int in_fd, Py_off_t offset,
- Py_off_t sbytes, PyObject *headers, PyObject *trailers,
- int flags)
- /*[clinic end generated code: output=81c4bcd143f5c82b input=b0d72579d4c69afa]*/
- #elif defined(__FreeBSD__) || defined(__DragonFly__)
- /*[clinic input]
- os.sendfile
- out_fd: int
- in_fd: int
- offset: Py_off_t
- count: Py_ssize_t
- headers: object(c_default="NULL") = ()
- trailers: object(c_default="NULL") = ()
- flags: int = 0
- Copy count bytes from file descriptor in_fd to file descriptor out_fd.
- [clinic start generated code]*/
- static PyObject *
- os_sendfile_impl(PyObject *module, int out_fd, int in_fd, Py_off_t offset,
- Py_ssize_t count, PyObject *headers, PyObject *trailers,
- int flags)
- /*[clinic end generated code: output=329ea009bdd55afc input=338adb8ff84ae8cd]*/
- #else
- /*[clinic input]
- os.sendfile
- out_fd: int
- in_fd: int
- offset as offobj: object
- count: Py_ssize_t
- Copy count bytes from file descriptor in_fd to file descriptor out_fd.
- [clinic start generated code]*/
- static PyObject *
- os_sendfile_impl(PyObject *module, int out_fd, int in_fd, PyObject *offobj,
- Py_ssize_t count)
- /*[clinic end generated code: output=ae81216e40f167d8 input=76d64058c74477ba]*/
- #endif
- {
- Py_ssize_t ret;
- int async_err = 0;
- #if defined(__FreeBSD__) || defined(__DragonFly__) || defined(__APPLE__)
- #ifndef __APPLE__
- off_t sbytes;
- #endif
- Py_buffer *hbuf, *tbuf;
- struct sf_hdtr sf;
- sf.headers = NULL;
- sf.trailers = NULL;
- if (headers != NULL) {
- if (!PySequence_Check(headers)) {
- PyErr_SetString(PyExc_TypeError,
- "sendfile() headers must be a sequence");
- return NULL;
- } else {
- Py_ssize_t i = PySequence_Size(headers);
- if (i < 0)
- return NULL;
- if (i > INT_MAX) {
- PyErr_SetString(PyExc_OverflowError,
- "sendfile() header is too large");
- return NULL;
- }
- if (i > 0) {
- sf.hdr_cnt = (int)i;
- if (iov_setup(&(sf.headers), &hbuf,
- headers, sf.hdr_cnt, PyBUF_SIMPLE) < 0)
- return NULL;
- #ifdef __APPLE__
- for (i = 0; i < sf.hdr_cnt; i++) {
- Py_ssize_t blen = sf.headers[i].iov_len;
- # define OFF_T_MAX 0x7fffffffffffffff
- if (sbytes >= OFF_T_MAX - blen) {
- PyErr_SetString(PyExc_OverflowError,
- "sendfile() header is too large");
- return NULL;
- }
- sbytes += blen;
- }
- #endif
- }
- }
- }
- if (trailers != NULL) {
- if (!PySequence_Check(trailers)) {
- PyErr_SetString(PyExc_TypeError,
- "sendfile() trailers must be a sequence");
- return NULL;
- } else {
- Py_ssize_t i = PySequence_Size(trailers);
- if (i < 0)
- return NULL;
- if (i > INT_MAX) {
- PyErr_SetString(PyExc_OverflowError,
- "sendfile() trailer is too large");
- return NULL;
- }
- if (i > 0) {
- sf.trl_cnt = (int)i;
- if (iov_setup(&(sf.trailers), &tbuf,
- trailers, sf.trl_cnt, PyBUF_SIMPLE) < 0)
- return NULL;
- }
- }
- }
- _Py_BEGIN_SUPPRESS_IPH
- do {
- Py_BEGIN_ALLOW_THREADS
- #ifdef __APPLE__
- ret = sendfile(in_fd, out_fd, offset, &sbytes, &sf, flags);
- #else
- ret = sendfile(in_fd, out_fd, offset, count, &sf, &sbytes, flags);
- #endif
- Py_END_ALLOW_THREADS
- } while (ret < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- _Py_END_SUPPRESS_IPH
- int saved_errno = errno;
- if (sf.headers != NULL)
- iov_cleanup(sf.headers, hbuf, sf.hdr_cnt);
- if (sf.trailers != NULL)
- iov_cleanup(sf.trailers, tbuf, sf.trl_cnt);
- if (ret < 0) {
- if ((saved_errno == EAGAIN) || (saved_errno == EBUSY)) {
- if (sbytes != 0) {
- // some data has been sent
- goto done;
- }
- // no data has been sent; upper application is supposed
- // to retry on EAGAIN or EBUSY
- }
- if (!async_err) {
- errno = saved_errno;
- posix_error();
- }
- return NULL;
- }
- goto done;
- done:
- #if !defined(HAVE_LARGEFILE_SUPPORT)
- return Py_BuildValue("l", sbytes);
- #else
- return Py_BuildValue("L", sbytes);
- #endif
- #else
- #ifdef __linux__
- if (offobj == Py_None) {
- do {
- Py_BEGIN_ALLOW_THREADS
- ret = sendfile(out_fd, in_fd, NULL, count);
- Py_END_ALLOW_THREADS
- } while (ret < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (ret < 0)
- return (!async_err) ? posix_error() : NULL;
- return Py_BuildValue("n", ret);
- }
- #endif
- off_t offset;
- if (!Py_off_t_converter(offobj, &offset))
- return NULL;
- #if defined(__sun) && defined(__SVR4)
- // On Solaris, sendfile raises EINVAL rather than returning 0
- // when the offset is equal or bigger than the in_fd size.
- struct stat st;
- do {
- Py_BEGIN_ALLOW_THREADS
- ret = fstat(in_fd, &st);
- Py_END_ALLOW_THREADS
- } while (ret != 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (ret < 0)
- return (!async_err) ? posix_error() : NULL;
- if (offset >= st.st_size) {
- return Py_BuildValue("i", 0);
- }
- // On illumos specifically sendfile() may perform a partial write but
- // return -1/an error (in one confirmed case the destination socket
- // had a 5 second timeout set and errno was EAGAIN) and it's on the client
- // code to check if the offset parameter was modified by sendfile().
- //
- // We need this variable to track said change.
- off_t original_offset = offset;
- #endif
- do {
- Py_BEGIN_ALLOW_THREADS
- ret = sendfile(out_fd, in_fd, &offset, count);
- #if defined(__sun) && defined(__SVR4)
- // This handles illumos-specific sendfile() partial write behavior,
- // see a comment above for more details.
- if (ret < 0 && offset != original_offset) {
- ret = offset - original_offset;
- }
- #endif
- Py_END_ALLOW_THREADS
- } while (ret < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (ret < 0)
- return (!async_err) ? posix_error() : NULL;
- return Py_BuildValue("n", ret);
- #endif
- }
- #endif /* HAVE_SENDFILE */
- #if defined(__APPLE__)
- /*[clinic input]
- os._fcopyfile
- in_fd: int
- out_fd: int
- flags: int
- /
- Efficiently copy content or metadata of 2 regular file descriptors (macOS).
- [clinic start generated code]*/
- static PyObject *
- os__fcopyfile_impl(PyObject *module, int in_fd, int out_fd, int flags)
- /*[clinic end generated code: output=c9d1a35a992e401b input=1e34638a86948795]*/
- {
- int ret;
- Py_BEGIN_ALLOW_THREADS
- ret = fcopyfile(in_fd, out_fd, NULL, flags);
- Py_END_ALLOW_THREADS
- if (ret < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif
- /*[clinic input]
- os.fstat
- fd : int
- Perform a stat system call on the given file descriptor.
- Like stat(), but for an open file descriptor.
- Equivalent to os.stat(fd).
- [clinic start generated code]*/
- static PyObject *
- os_fstat_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=efc038cb5f654492 input=27e0e0ebbe5600c9]*/
- {
- STRUCT_STAT st;
- int res;
- int async_err = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- res = FSTAT(fd, &st);
- Py_END_ALLOW_THREADS
- } while (res != 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (res != 0) {
- #ifdef MS_WINDOWS
- return PyErr_SetFromWindowsErr(0);
- #else
- return (!async_err) ? posix_error() : NULL;
- #endif
- }
- return _pystat_fromstructstat(module, &st);
- }
- /*[clinic input]
- os.isatty -> bool
- fd: int
- /
- Return True if the fd is connected to a terminal.
- Return True if the file descriptor is an open file descriptor
- connected to the slave end of a terminal.
- [clinic start generated code]*/
- static int
- os_isatty_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=6a48c8b4e644ca00 input=08ce94aa1eaf7b5e]*/
- {
- int return_value;
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- return_value = isatty(fd);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- return return_value;
- }
- #ifdef HAVE_PIPE
- /*[clinic input]
- os.pipe
- Create a pipe.
- Returns a tuple of two file descriptors:
- (read_fd, write_fd)
- [clinic start generated code]*/
- static PyObject *
- os_pipe_impl(PyObject *module)
- /*[clinic end generated code: output=ff9b76255793b440 input=02535e8c8fa6c4d4]*/
- {
- int fds[2];
- #ifdef MS_WINDOWS
- HANDLE read, write;
- SECURITY_ATTRIBUTES attr;
- BOOL ok;
- #else
- int res;
- #endif
- #ifdef MS_WINDOWS
- attr.nLength = sizeof(attr);
- attr.lpSecurityDescriptor = NULL;
- attr.bInheritHandle = FALSE;
- Py_BEGIN_ALLOW_THREADS
- ok = CreatePipe(&read, &write, &attr, 0);
- if (ok) {
- fds[0] = _Py_open_osfhandle_noraise(read, _O_RDONLY);
- fds[1] = _Py_open_osfhandle_noraise(write, _O_WRONLY);
- if (fds[0] == -1 || fds[1] == -1) {
- CloseHandle(read);
- CloseHandle(write);
- ok = 0;
- }
- }
- Py_END_ALLOW_THREADS
- if (!ok)
- return PyErr_SetFromWindowsErr(0);
- #else
- #ifdef HAVE_PIPE2
- Py_BEGIN_ALLOW_THREADS
- res = pipe2(fds, O_CLOEXEC);
- Py_END_ALLOW_THREADS
- if (res != 0 && errno == ENOSYS)
- {
- #endif
- Py_BEGIN_ALLOW_THREADS
- res = pipe(fds);
- Py_END_ALLOW_THREADS
- if (res == 0) {
- if (_Py_set_inheritable(fds[0], 0, NULL) < 0) {
- close(fds[0]);
- close(fds[1]);
- return NULL;
- }
- if (_Py_set_inheritable(fds[1], 0, NULL) < 0) {
- close(fds[0]);
- close(fds[1]);
- return NULL;
- }
- }
- #ifdef HAVE_PIPE2
- }
- #endif
- if (res != 0)
- return PyErr_SetFromErrno(PyExc_OSError);
- #endif /* !MS_WINDOWS */
- return Py_BuildValue("(ii)", fds[0], fds[1]);
- }
- #endif /* HAVE_PIPE */
- #ifdef HAVE_PIPE2
- /*[clinic input]
- os.pipe2
- flags: int
- /
- Create a pipe with flags set atomically.
- Returns a tuple of two file descriptors:
- (read_fd, write_fd)
- flags can be constructed by ORing together one or more of these values:
- O_NONBLOCK, O_CLOEXEC.
- [clinic start generated code]*/
- static PyObject *
- os_pipe2_impl(PyObject *module, int flags)
- /*[clinic end generated code: output=25751fb43a45540f input=f261b6e7e63c6817]*/
- {
- int fds[2];
- int res;
- res = pipe2(fds, flags);
- if (res != 0)
- return posix_error();
- return Py_BuildValue("(ii)", fds[0], fds[1]);
- }
- #endif /* HAVE_PIPE2 */
- #ifdef HAVE_WRITEV
- /*[clinic input]
- os.writev -> Py_ssize_t
- fd: int
- buffers: object
- /
- Iterate over buffers, and write the contents of each to a file descriptor.
- Returns the total number of bytes written.
- buffers must be a sequence of bytes-like objects.
- [clinic start generated code]*/
- static Py_ssize_t
- os_writev_impl(PyObject *module, int fd, PyObject *buffers)
- /*[clinic end generated code: output=56565cfac3aac15b input=5b8d17fe4189d2fe]*/
- {
- Py_ssize_t cnt;
- Py_ssize_t result;
- int async_err = 0;
- struct iovec *iov;
- Py_buffer *buf;
- if (!PySequence_Check(buffers)) {
- PyErr_SetString(PyExc_TypeError,
- "writev() arg 2 must be a sequence");
- return -1;
- }
- cnt = PySequence_Size(buffers);
- if (cnt < 0)
- return -1;
- if (iov_setup(&iov, &buf, buffers, cnt, PyBUF_SIMPLE) < 0) {
- return -1;
- }
- do {
- Py_BEGIN_ALLOW_THREADS
- result = writev(fd, iov, cnt);
- Py_END_ALLOW_THREADS
- } while (result < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (result < 0 && !async_err)
- posix_error();
- iov_cleanup(iov, buf, cnt);
- return result;
- }
- #endif /* HAVE_WRITEV */
- #ifdef HAVE_PWRITE
- /*[clinic input]
- os.pwrite -> Py_ssize_t
- fd: int
- buffer: Py_buffer
- offset: Py_off_t
- /
- Write bytes to a file descriptor starting at a particular offset.
- Write buffer to fd, starting at offset bytes from the beginning of
- the file. Returns the number of bytes writte. Does not change the
- current file offset.
- [clinic start generated code]*/
- static Py_ssize_t
- os_pwrite_impl(PyObject *module, int fd, Py_buffer *buffer, Py_off_t offset)
- /*[clinic end generated code: output=c74da630758ee925 input=19903f1b3dd26377]*/
- {
- Py_ssize_t size;
- int async_err = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- size = pwrite(fd, buffer->buf, (size_t)buffer->len, offset);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (size < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (size < 0 && !async_err)
- posix_error();
- return size;
- }
- #endif /* HAVE_PWRITE */
- #if defined(HAVE_PWRITEV) || defined (HAVE_PWRITEV2)
- /*[clinic input]
- os.pwritev -> Py_ssize_t
- fd: int
- buffers: object
- offset: Py_off_t
- flags: int = 0
- /
- Writes the contents of bytes-like objects to a file descriptor at a given offset.
- Combines the functionality of writev() and pwrite(). All buffers must be a sequence
- of bytes-like objects. Buffers are processed in array order. Entire contents of first
- buffer is written before proceeding to second, and so on. The operating system may
- set a limit (sysconf() value SC_IOV_MAX) on the number of buffers that can be used.
- This function writes the contents of each object to the file descriptor and returns
- the total number of bytes written.
- The flags argument contains a bitwise OR of zero or more of the following flags:
- - RWF_DSYNC
- - RWF_SYNC
- - RWF_APPEND
- Using non-zero flags requires Linux 4.7 or newer.
- [clinic start generated code]*/
- static Py_ssize_t
- os_pwritev_impl(PyObject *module, int fd, PyObject *buffers, Py_off_t offset,
- int flags)
- /*[clinic end generated code: output=e3dd3e9d11a6a5c7 input=35358c327e1a2a8e]*/
- {
- Py_ssize_t cnt;
- Py_ssize_t result;
- int async_err = 0;
- struct iovec *iov;
- Py_buffer *buf;
- if (!PySequence_Check(buffers)) {
- PyErr_SetString(PyExc_TypeError,
- "pwritev() arg 2 must be a sequence");
- return -1;
- }
- cnt = PySequence_Size(buffers);
- if (cnt < 0) {
- return -1;
- }
- #ifndef HAVE_PWRITEV2
- if(flags != 0) {
- argument_unavailable_error("pwritev2", "flags");
- return -1;
- }
- #endif
- if (iov_setup(&iov, &buf, buffers, cnt, PyBUF_SIMPLE) < 0) {
- return -1;
- }
- #ifdef HAVE_PWRITEV2
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- result = pwritev2(fd, iov, cnt, offset, flags);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (result < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- #else
- #if defined(__APPLE__) && defined(__clang__)
- /* This entire function will be removed from the module dict when the API
- * is not available.
- */
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wunguarded-availability"
- #pragma clang diagnostic ignored "-Wunguarded-availability-new"
- #endif
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- result = pwritev(fd, iov, cnt, offset);
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (result < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- #if defined(__APPLE__) && defined(__clang__)
- #pragma clang diagnostic pop
- #endif
- #endif
- if (result < 0) {
- if (!async_err) {
- posix_error();
- }
- result = -1;
- }
- iov_cleanup(iov, buf, cnt);
- return result;
- }
- #endif /* HAVE_PWRITEV */
- #ifdef HAVE_COPY_FILE_RANGE
- /*[clinic input]
- os.copy_file_range
- src: int
- Source file descriptor.
- dst: int
- Destination file descriptor.
- count: Py_ssize_t
- Number of bytes to copy.
- offset_src: object = None
- Starting offset in src.
- offset_dst: object = None
- Starting offset in dst.
- Copy count bytes from one file descriptor to another.
- If offset_src is None, then src is read from the current position;
- respectively for offset_dst.
- [clinic start generated code]*/
- static PyObject *
- os_copy_file_range_impl(PyObject *module, int src, int dst, Py_ssize_t count,
- PyObject *offset_src, PyObject *offset_dst)
- /*[clinic end generated code: output=1a91713a1d99fc7a input=42fdce72681b25a9]*/
- {
- off_t offset_src_val, offset_dst_val;
- off_t *p_offset_src = NULL;
- off_t *p_offset_dst = NULL;
- Py_ssize_t ret;
- int async_err = 0;
- /* The flags argument is provided to allow
- * for future extensions and currently must be to 0. */
- int flags = 0;
- if (count < 0) {
- PyErr_SetString(PyExc_ValueError, "negative value for 'count' not allowed");
- return NULL;
- }
- if (offset_src != Py_None) {
- if (!Py_off_t_converter(offset_src, &offset_src_val)) {
- return NULL;
- }
- p_offset_src = &offset_src_val;
- }
- if (offset_dst != Py_None) {
- if (!Py_off_t_converter(offset_dst, &offset_dst_val)) {
- return NULL;
- }
- p_offset_dst = &offset_dst_val;
- }
- do {
- Py_BEGIN_ALLOW_THREADS
- ret = copy_file_range(src, p_offset_src, dst, p_offset_dst, count, flags);
- Py_END_ALLOW_THREADS
- } while (ret < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (ret < 0) {
- return (!async_err) ? posix_error() : NULL;
- }
- return PyLong_FromSsize_t(ret);
- }
- #endif /* HAVE_COPY_FILE_RANGE*/
- #if (defined(HAVE_SPLICE) && !defined(_AIX))
- /*[clinic input]
- os.splice
- src: int
- Source file descriptor.
- dst: int
- Destination file descriptor.
- count: Py_ssize_t
- Number of bytes to copy.
- offset_src: object = None
- Starting offset in src.
- offset_dst: object = None
- Starting offset in dst.
- flags: unsigned_int = 0
- Flags to modify the semantics of the call.
- Transfer count bytes from one pipe to a descriptor or vice versa.
- If offset_src is None, then src is read from the current position;
- respectively for offset_dst. The offset associated to the file
- descriptor that refers to a pipe must be None.
- [clinic start generated code]*/
- static PyObject *
- os_splice_impl(PyObject *module, int src, int dst, Py_ssize_t count,
- PyObject *offset_src, PyObject *offset_dst,
- unsigned int flags)
- /*[clinic end generated code: output=d0386f25a8519dc5 input=047527c66c6d2e0a]*/
- {
- off_t offset_src_val, offset_dst_val;
- off_t *p_offset_src = NULL;
- off_t *p_offset_dst = NULL;
- Py_ssize_t ret;
- int async_err = 0;
- if (count < 0) {
- PyErr_SetString(PyExc_ValueError, "negative value for 'count' not allowed");
- return NULL;
- }
- if (offset_src != Py_None) {
- if (!Py_off_t_converter(offset_src, &offset_src_val)) {
- return NULL;
- }
- p_offset_src = &offset_src_val;
- }
- if (offset_dst != Py_None) {
- if (!Py_off_t_converter(offset_dst, &offset_dst_val)) {
- return NULL;
- }
- p_offset_dst = &offset_dst_val;
- }
- do {
- Py_BEGIN_ALLOW_THREADS
- ret = splice(src, p_offset_src, dst, p_offset_dst, count, flags);
- Py_END_ALLOW_THREADS
- } while (ret < 0 && errno == EINTR && !(async_err = PyErr_CheckSignals()));
- if (ret < 0) {
- return (!async_err) ? posix_error() : NULL;
- }
- return PyLong_FromSsize_t(ret);
- }
- #endif /* HAVE_SPLICE*/
- #ifdef HAVE_MKFIFO
- /*[clinic input]
- os.mkfifo
- path: path_t
- mode: int=0o666
- *
- dir_fd: dir_fd(requires='mkfifoat')=None
- Create a "fifo" (a POSIX named pipe).
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_mkfifo_impl(PyObject *module, path_t *path, int mode, int dir_fd)
- /*[clinic end generated code: output=ce41cfad0e68c940 input=73032e98a36e0e19]*/
- {
- int result;
- int async_err = 0;
- #ifdef HAVE_MKFIFOAT
- int mkfifoat_unavailable = 0;
- #endif
- do {
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_MKFIFOAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_MKFIFOAT_RUNTIME) {
- result = mkfifoat(dir_fd, path->narrow, mode);
- } else {
- mkfifoat_unavailable = 1;
- result = 0;
- }
- } else
- #endif
- result = mkfifo(path->narrow, mode);
- Py_END_ALLOW_THREADS
- } while (result != 0 && errno == EINTR &&
- !(async_err = PyErr_CheckSignals()));
- #ifdef HAVE_MKFIFOAT
- if (mkfifoat_unavailable) {
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (result != 0)
- return (!async_err) ? posix_error() : NULL;
- Py_RETURN_NONE;
- }
- #endif /* HAVE_MKFIFO */
- #if defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)
- /*[clinic input]
- os.mknod
- path: path_t
- mode: int=0o600
- device: dev_t=0
- *
- dir_fd: dir_fd(requires='mknodat')=None
- Create a node in the file system.
- Create a node in the file system (file, device special file or named pipe)
- at path. mode specifies both the permissions to use and the
- type of node to be created, being combined (bitwise OR) with one of
- S_IFREG, S_IFCHR, S_IFBLK, and S_IFIFO. If S_IFCHR or S_IFBLK is set on mode,
- device defines the newly created device special file (probably using
- os.makedev()). Otherwise device is ignored.
- If dir_fd is not None, it should be a file descriptor open to a directory,
- and path should be relative; path will then be relative to that directory.
- dir_fd may not be implemented on your platform.
- If it is unavailable, using it will raise a NotImplementedError.
- [clinic start generated code]*/
- static PyObject *
- os_mknod_impl(PyObject *module, path_t *path, int mode, dev_t device,
- int dir_fd)
- /*[clinic end generated code: output=92e55d3ca8917461 input=ee44531551a4d83b]*/
- {
- int result;
- int async_err = 0;
- #ifdef HAVE_MKNODAT
- int mknodat_unavailable = 0;
- #endif
- do {
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_MKNODAT
- if (dir_fd != DEFAULT_DIR_FD) {
- if (HAVE_MKNODAT_RUNTIME) {
- result = mknodat(dir_fd, path->narrow, mode, device);
- } else {
- mknodat_unavailable = 1;
- result = 0;
- }
- } else
- #endif
- result = mknod(path->narrow, mode, device);
- Py_END_ALLOW_THREADS
- } while (result != 0 && errno == EINTR &&
- !(async_err = PyErr_CheckSignals()));
- #ifdef HAVE_MKNODAT
- if (mknodat_unavailable) {
- argument_unavailable_error(NULL, "dir_fd");
- return NULL;
- }
- #endif
- if (result != 0)
- return (!async_err) ? posix_error() : NULL;
- Py_RETURN_NONE;
- }
- #endif /* defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV) */
- #ifdef HAVE_DEVICE_MACROS
- static PyObject *
- major_minor_conv(unsigned int value)
- {
- #ifdef NODEV
- if (value == (unsigned int)NODEV) {
- return PyLong_FromLong((int)NODEV);
- }
- #endif
- return PyLong_FromUnsignedLong(value);
- }
- static int
- major_minor_check(dev_t value)
- {
- #ifdef NODEV
- if (value == NODEV) {
- return 1;
- }
- #endif
- return (dev_t)(unsigned int)value == value;
- }
- /*[clinic input]
- os.major
- device: dev_t
- /
- Extracts a device major number from a raw device number.
- [clinic start generated code]*/
- static PyObject *
- os_major_impl(PyObject *module, dev_t device)
- /*[clinic end generated code: output=4071ffee17647891 input=b1a0a14ec9448229]*/
- {
- return major_minor_conv(major(device));
- }
- /*[clinic input]
- os.minor
- device: dev_t
- /
- Extracts a device minor number from a raw device number.
- [clinic start generated code]*/
- static PyObject *
- os_minor_impl(PyObject *module, dev_t device)
- /*[clinic end generated code: output=306cb78e3bc5004f input=2f686e463682a9da]*/
- {
- return major_minor_conv(minor(device));
- }
- /*[clinic input]
- os.makedev -> dev_t
- major: dev_t
- minor: dev_t
- /
- Composes a raw device number from the major and minor device numbers.
- [clinic start generated code]*/
- static dev_t
- os_makedev_impl(PyObject *module, dev_t major, dev_t minor)
- /*[clinic end generated code: output=cad6125c51f5af80 input=2146126ec02e55c1]*/
- {
- if (!major_minor_check(major) || !major_minor_check(minor)) {
- PyErr_SetString(PyExc_OverflowError,
- "Python int too large to convert to C unsigned int");
- return (dev_t)-1;
- }
- return makedev(major, minor);
- }
- #endif /* HAVE_DEVICE_MACROS */
- #if defined HAVE_FTRUNCATE || defined MS_WINDOWS
- /*[clinic input]
- os.ftruncate
- fd: int
- length: Py_off_t
- /
- Truncate a file, specified by file descriptor, to a specific length.
- [clinic start generated code]*/
- static PyObject *
- os_ftruncate_impl(PyObject *module, int fd, Py_off_t length)
- /*[clinic end generated code: output=fba15523721be7e4 input=63b43641e52818f2]*/
- {
- int result;
- int async_err = 0;
- if (PySys_Audit("os.truncate", "in", fd, length) < 0) {
- return NULL;
- }
- do {
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef MS_WINDOWS
- result = _chsize_s(fd, length);
- #else
- result = ftruncate(fd, length);
- #endif
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- } while (result != 0 && errno == EINTR &&
- !(async_err = PyErr_CheckSignals()));
- if (result != 0)
- return (!async_err) ? posix_error() : NULL;
- Py_RETURN_NONE;
- }
- #endif /* HAVE_FTRUNCATE || MS_WINDOWS */
- #if defined HAVE_TRUNCATE || defined MS_WINDOWS
- /*[clinic input]
- os.truncate
- path: path_t(allow_fd='PATH_HAVE_FTRUNCATE')
- length: Py_off_t
- Truncate a file, specified by path, to a specific length.
- On some platforms, path may also be specified as an open file descriptor.
- If this functionality is unavailable, using it raises an exception.
- [clinic start generated code]*/
- static PyObject *
- os_truncate_impl(PyObject *module, path_t *path, Py_off_t length)
- /*[clinic end generated code: output=43009c8df5c0a12b input=77229cf0b50a9b77]*/
- {
- int result;
- #ifdef MS_WINDOWS
- int fd;
- #endif
- if (path->fd != -1)
- return os_ftruncate_impl(module, path->fd, length);
- if (PySys_Audit("os.truncate", "On", path->object, length) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- _Py_BEGIN_SUPPRESS_IPH
- #ifdef MS_WINDOWS
- fd = _wopen(path->wide, _O_WRONLY | _O_BINARY | _O_NOINHERIT);
- if (fd < 0)
- result = -1;
- else {
- result = _chsize_s(fd, length);
- close(fd);
- if (result < 0)
- errno = result;
- }
- #else
- result = truncate(path->narrow, length);
- #endif
- _Py_END_SUPPRESS_IPH
- Py_END_ALLOW_THREADS
- if (result < 0)
- return posix_path_error(path);
- Py_RETURN_NONE;
- }
- #endif /* HAVE_TRUNCATE || MS_WINDOWS */
- /* Issue #22396: On 32-bit AIX platform, the prototypes of os.posix_fadvise()
- and os.posix_fallocate() in system headers are wrong if _LARGE_FILES is
- defined, which is the case in Python on AIX. AIX bug report:
- http://www-01.ibm.com/support/docview.wss?uid=isg1IV56170 */
- #if defined(_AIX) && defined(_LARGE_FILES) && !defined(__64BIT__)
- # define POSIX_FADVISE_AIX_BUG
- #endif
- /* GH-111804: Due to posix_fallocate() not having consistent semantics across
- OSs, support was dropped in WASI preview2. */
- #if defined(HAVE_POSIX_FALLOCATE) && !defined(POSIX_FADVISE_AIX_BUG) && \
- !defined(__wasi__)
- /*[clinic input]
- os.posix_fallocate
- fd: int
- offset: Py_off_t
- length: Py_off_t
- /
- Ensure a file has allocated at least a particular number of bytes on disk.
- Ensure that the file specified by fd encompasses a range of bytes
- starting at offset bytes from the beginning and continuing for length bytes.
- [clinic start generated code]*/
- static PyObject *
- os_posix_fallocate_impl(PyObject *module, int fd, Py_off_t offset,
- Py_off_t length)
- /*[clinic end generated code: output=73f107139564aa9d input=d7a2ef0ab2ca52fb]*/
- {
- int result;
- int async_err = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- result = posix_fallocate(fd, offset, length);
- Py_END_ALLOW_THREADS
- } while (result == EINTR && !(async_err = PyErr_CheckSignals()));
- if (result == 0)
- Py_RETURN_NONE;
- if (async_err)
- return NULL;
- errno = result;
- return posix_error();
- }
- #endif /* HAVE_POSIX_FALLOCATE) && !POSIX_FADVISE_AIX_BUG && !defined(__wasi__) */
- #if defined(HAVE_POSIX_FADVISE) && !defined(POSIX_FADVISE_AIX_BUG)
- /*[clinic input]
- os.posix_fadvise
- fd: int
- offset: Py_off_t
- length: Py_off_t
- advice: int
- /
- Announce an intention to access data in a specific pattern.
- Announce an intention to access data in a specific pattern, thus allowing
- the kernel to make optimizations.
- The advice applies to the region of the file specified by fd starting at
- offset and continuing for length bytes.
- advice is one of POSIX_FADV_NORMAL, POSIX_FADV_SEQUENTIAL,
- POSIX_FADV_RANDOM, POSIX_FADV_NOREUSE, POSIX_FADV_WILLNEED, or
- POSIX_FADV_DONTNEED.
- [clinic start generated code]*/
- static PyObject *
- os_posix_fadvise_impl(PyObject *module, int fd, Py_off_t offset,
- Py_off_t length, int advice)
- /*[clinic end generated code: output=412ef4aa70c98642 input=0fbe554edc2f04b5]*/
- {
- int result;
- int async_err = 0;
- do {
- Py_BEGIN_ALLOW_THREADS
- result = posix_fadvise(fd, offset, length, advice);
- Py_END_ALLOW_THREADS
- } while (result == EINTR && !(async_err = PyErr_CheckSignals()));
- if (result == 0)
- Py_RETURN_NONE;
- if (async_err)
- return NULL;
- errno = result;
- return posix_error();
- }
- #endif /* HAVE_POSIX_FADVISE && !POSIX_FADVISE_AIX_BUG */
- #ifdef MS_WINDOWS
- static PyObject*
- win32_putenv(PyObject *name, PyObject *value)
- {
- /* Search from index 1 because on Windows starting '=' is allowed for
- defining hidden environment variables. */
- if (PyUnicode_GET_LENGTH(name) == 0 ||
- PyUnicode_FindChar(name, '=', 1, PyUnicode_GET_LENGTH(name), 1) != -1)
- {
- PyErr_SetString(PyExc_ValueError, "illegal environment variable name");
- return NULL;
- }
- PyObject *unicode;
- if (value != NULL) {
- unicode = PyUnicode_FromFormat("%U=%U", name, value);
- }
- else {
- unicode = PyUnicode_FromFormat("%U=", name);
- }
- if (unicode == NULL) {
- return NULL;
- }
- Py_ssize_t size;
- wchar_t *env = PyUnicode_AsWideCharString(unicode, &size);
- Py_DECREF(unicode);
- if (env == NULL) {
- return NULL;
- }
- if (size > _MAX_ENV) {
- PyErr_Format(PyExc_ValueError,
- "the environment variable is longer than %u characters",
- _MAX_ENV);
- PyMem_Free(env);
- return NULL;
- }
- if (wcslen(env) != (size_t)size) {
- PyErr_SetString(PyExc_ValueError,
- "embedded null character");
- PyMem_Free(env);
- return NULL;
- }
- /* _wputenv() and SetEnvironmentVariableW() update the environment in the
- Process Environment Block (PEB). _wputenv() also updates CRT 'environ'
- and '_wenviron' variables, whereas SetEnvironmentVariableW() does not.
- Prefer _wputenv() to be compatible with C libraries using CRT
- variables and CRT functions using these variables (ex: getenv()). */
- int err = _wputenv(env);
- if (err) {
- posix_error();
- PyMem_Free(env);
- return NULL;
- }
- PyMem_Free(env);
- Py_RETURN_NONE;
- }
- #endif
- #ifdef MS_WINDOWS
- /*[clinic input]
- os.putenv
- name: unicode
- value: unicode
- /
- Change or add an environment variable.
- [clinic start generated code]*/
- static PyObject *
- os_putenv_impl(PyObject *module, PyObject *name, PyObject *value)
- /*[clinic end generated code: output=d29a567d6b2327d2 input=ba586581c2e6105f]*/
- {
- if (PySys_Audit("os.putenv", "OO", name, value) < 0) {
- return NULL;
- }
- return win32_putenv(name, value);
- }
- #else
- /*[clinic input]
- os.putenv
- name: FSConverter
- value: FSConverter
- /
- Change or add an environment variable.
- [clinic start generated code]*/
- static PyObject *
- os_putenv_impl(PyObject *module, PyObject *name, PyObject *value)
- /*[clinic end generated code: output=d29a567d6b2327d2 input=a97bc6152f688d31]*/
- {
- const char *name_string = PyBytes_AS_STRING(name);
- const char *value_string = PyBytes_AS_STRING(value);
- if (strchr(name_string, '=') != NULL) {
- PyErr_SetString(PyExc_ValueError, "illegal environment variable name");
- return NULL;
- }
- if (PySys_Audit("os.putenv", "OO", name, value) < 0) {
- return NULL;
- }
- if (setenv(name_string, value_string, 1)) {
- return posix_error();
- }
- Py_RETURN_NONE;
- }
- #endif /* !defined(MS_WINDOWS) */
- #ifdef MS_WINDOWS
- /*[clinic input]
- os.unsetenv
- name: unicode
- /
- Delete an environment variable.
- [clinic start generated code]*/
- static PyObject *
- os_unsetenv_impl(PyObject *module, PyObject *name)
- /*[clinic end generated code: output=54c4137ab1834f02 input=4d6a1747cc526d2f]*/
- {
- if (PySys_Audit("os.unsetenv", "(O)", name) < 0) {
- return NULL;
- }
- return win32_putenv(name, NULL);
- }
- #else
- /*[clinic input]
- os.unsetenv
- name: FSConverter
- /
- Delete an environment variable.
- [clinic start generated code]*/
- static PyObject *
- os_unsetenv_impl(PyObject *module, PyObject *name)
- /*[clinic end generated code: output=54c4137ab1834f02 input=2bb5288a599c7107]*/
- {
- if (PySys_Audit("os.unsetenv", "(O)", name) < 0) {
- return NULL;
- }
- #ifdef HAVE_BROKEN_UNSETENV
- unsetenv(PyBytes_AS_STRING(name));
- #else
- int err = unsetenv(PyBytes_AS_STRING(name));
- if (err) {
- return posix_error();
- }
- #endif
- Py_RETURN_NONE;
- }
- #endif /* !MS_WINDOWS */
- /*[clinic input]
- os.strerror
- code: int
- /
- Translate an error code to a message string.
- [clinic start generated code]*/
- static PyObject *
- os_strerror_impl(PyObject *module, int code)
- /*[clinic end generated code: output=baebf09fa02a78f2 input=75a8673d97915a91]*/
- {
- char *message = strerror(code);
- if (message == NULL) {
- PyErr_SetString(PyExc_ValueError,
- "strerror() argument out of range");
- return NULL;
- }
- return PyUnicode_DecodeLocale(message, "surrogateescape");
- }
- #ifdef HAVE_SYS_WAIT_H
- #ifdef WCOREDUMP
- /*[clinic input]
- os.WCOREDUMP -> bool
- status: int
- /
- Return True if the process returning status was dumped to a core file.
- [clinic start generated code]*/
- static int
- os_WCOREDUMP_impl(PyObject *module, int status)
- /*[clinic end generated code: output=1a584b147b16bd18 input=8b05e7ab38528d04]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WCOREDUMP(wait_status);
- }
- #endif /* WCOREDUMP */
- #ifdef WIFCONTINUED
- /*[clinic input]
- os.WIFCONTINUED -> bool
- status: int
- Return True if a particular process was continued from a job control stop.
- Return True if the process returning status was continued from a
- job control stop.
- [clinic start generated code]*/
- static int
- os_WIFCONTINUED_impl(PyObject *module, int status)
- /*[clinic end generated code: output=1e35295d844364bd input=e777e7d38eb25bd9]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WIFCONTINUED(wait_status);
- }
- #endif /* WIFCONTINUED */
- #ifdef WIFSTOPPED
- /*[clinic input]
- os.WIFSTOPPED -> bool
- status: int
- Return True if the process returning status was stopped.
- [clinic start generated code]*/
- static int
- os_WIFSTOPPED_impl(PyObject *module, int status)
- /*[clinic end generated code: output=fdb57122a5c9b4cb input=043cb7f1289ef904]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WIFSTOPPED(wait_status);
- }
- #endif /* WIFSTOPPED */
- #ifdef WIFSIGNALED
- /*[clinic input]
- os.WIFSIGNALED -> bool
- status: int
- Return True if the process returning status was terminated by a signal.
- [clinic start generated code]*/
- static int
- os_WIFSIGNALED_impl(PyObject *module, int status)
- /*[clinic end generated code: output=d1dde4dcc819a5f5 input=d55ba7cc9ce5dc43]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WIFSIGNALED(wait_status);
- }
- #endif /* WIFSIGNALED */
- #ifdef WIFEXITED
- /*[clinic input]
- os.WIFEXITED -> bool
- status: int
- Return True if the process returning status exited via the exit() system call.
- [clinic start generated code]*/
- static int
- os_WIFEXITED_impl(PyObject *module, int status)
- /*[clinic end generated code: output=01c09d6ebfeea397 input=d63775a6791586c0]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WIFEXITED(wait_status);
- }
- #endif /* WIFEXITED */
- #ifdef WEXITSTATUS
- /*[clinic input]
- os.WEXITSTATUS -> int
- status: int
- Return the process return code from status.
- [clinic start generated code]*/
- static int
- os_WEXITSTATUS_impl(PyObject *module, int status)
- /*[clinic end generated code: output=6e3efbba11f6488d input=e1fb4944e377585b]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WEXITSTATUS(wait_status);
- }
- #endif /* WEXITSTATUS */
- #ifdef WTERMSIG
- /*[clinic input]
- os.WTERMSIG -> int
- status: int
- Return the signal that terminated the process that provided the status value.
- [clinic start generated code]*/
- static int
- os_WTERMSIG_impl(PyObject *module, int status)
- /*[clinic end generated code: output=172f7dfc8dcfc3ad input=727fd7f84ec3f243]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WTERMSIG(wait_status);
- }
- #endif /* WTERMSIG */
- #ifdef WSTOPSIG
- /*[clinic input]
- os.WSTOPSIG -> int
- status: int
- Return the signal that stopped the process that provided the status value.
- [clinic start generated code]*/
- static int
- os_WSTOPSIG_impl(PyObject *module, int status)
- /*[clinic end generated code: output=0ab7586396f5d82b input=46ebf1d1b293c5c1]*/
- {
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- return WSTOPSIG(wait_status);
- }
- #endif /* WSTOPSIG */
- #endif /* HAVE_SYS_WAIT_H */
- #if defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H)
- #ifdef _SCO_DS
- /* SCO OpenServer 5.0 and later requires _SVID3 before it reveals the
- needed definitions in sys/statvfs.h */
- #define _SVID3
- #endif
- #include <sys/statvfs.h>
- static PyObject*
- _pystatvfs_fromstructstatvfs(PyObject *module, struct statvfs st) {
- PyObject *StatVFSResultType = get_posix_state(module)->StatVFSResultType;
- PyObject *v = PyStructSequence_New((PyTypeObject *)StatVFSResultType);
- if (v == NULL)
- return NULL;
- int pos = 0;
- #define SET_RESULT(CALL) \
- do { \
- PyObject *item = (CALL); \
- if (item == NULL) { \
- Py_DECREF(v); \
- return NULL; \
- } \
- PyStructSequence_SET_ITEM(v, pos++, item); \
- } while(0)
- #if !defined(HAVE_LARGEFILE_SUPPORT)
- SET_RESULT(PyLong_FromLong((long) st.f_bsize));
- SET_RESULT(PyLong_FromLong((long) st.f_frsize));
- SET_RESULT(PyLong_FromLong((long) st.f_blocks));
- SET_RESULT(PyLong_FromLong((long) st.f_bfree));
- SET_RESULT(PyLong_FromLong((long) st.f_bavail));
- SET_RESULT(PyLong_FromLong((long) st.f_files));
- SET_RESULT(PyLong_FromLong((long) st.f_ffree));
- SET_RESULT(PyLong_FromLong((long) st.f_favail));
- SET_RESULT(PyLong_FromLong((long) st.f_flag));
- SET_RESULT(PyLong_FromLong((long) st.f_namemax));
- #else
- SET_RESULT(PyLong_FromLong((long) st.f_bsize));
- SET_RESULT(PyLong_FromLong((long) st.f_frsize));
- SET_RESULT(PyLong_FromLongLong((long long) st.f_blocks));
- SET_RESULT(PyLong_FromLongLong((long long) st.f_bfree));
- SET_RESULT(PyLong_FromLongLong((long long) st.f_bavail));
- SET_RESULT(PyLong_FromLongLong((long long) st.f_files));
- SET_RESULT(PyLong_FromLongLong((long long) st.f_ffree));
- SET_RESULT(PyLong_FromLongLong((long long) st.f_favail));
- SET_RESULT(PyLong_FromLong((long) st.f_flag));
- SET_RESULT(PyLong_FromLong((long) st.f_namemax));
- #endif
- /* The _ALL_SOURCE feature test macro defines f_fsid as a structure
- * (issue #32390). */
- #if defined(_AIX) && defined(_ALL_SOURCE)
- SET_RESULT(PyLong_FromUnsignedLong(st.f_fsid.val[0]));
- #else
- SET_RESULT(PyLong_FromUnsignedLong(st.f_fsid));
- #endif
- #undef SET_RESULT
- return v;
- }
- /*[clinic input]
- os.fstatvfs
- fd: int
- /
- Perform an fstatvfs system call on the given fd.
- Equivalent to statvfs(fd).
- [clinic start generated code]*/
- static PyObject *
- os_fstatvfs_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=53547cf0cc55e6c5 input=d8122243ac50975e]*/
- {
- int result;
- int async_err = 0;
- struct statvfs st;
- do {
- Py_BEGIN_ALLOW_THREADS
- result = fstatvfs(fd, &st);
- Py_END_ALLOW_THREADS
- } while (result != 0 && errno == EINTR &&
- !(async_err = PyErr_CheckSignals()));
- if (result != 0)
- return (!async_err) ? posix_error() : NULL;
- return _pystatvfs_fromstructstatvfs(module, st);
- }
- #endif /* defined(HAVE_FSTATVFS) && defined(HAVE_SYS_STATVFS_H) */
- #if defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H)
- #include <sys/statvfs.h>
- /*[clinic input]
- os.statvfs
- path: path_t(allow_fd='PATH_HAVE_FSTATVFS')
- Perform a statvfs system call on the given path.
- path may always be specified as a string.
- On some platforms, path may also be specified as an open file descriptor.
- If this functionality is unavailable, using it raises an exception.
- [clinic start generated code]*/
- static PyObject *
- os_statvfs_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=87106dd1beb8556e input=3f5c35791c669bd9]*/
- {
- int result;
- struct statvfs st;
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_FSTATVFS
- if (path->fd != -1) {
- result = fstatvfs(path->fd, &st);
- }
- else
- #endif
- result = statvfs(path->narrow, &st);
- Py_END_ALLOW_THREADS
- if (result) {
- return path_error(path);
- }
- return _pystatvfs_fromstructstatvfs(module, st);
- }
- #endif /* defined(HAVE_STATVFS) && defined(HAVE_SYS_STATVFS_H) */
- #ifdef MS_WINDOWS
- /*[clinic input]
- os._getdiskusage
- path: path_t
- Return disk usage statistics about the given path as a (total, free) tuple.
- [clinic start generated code]*/
- static PyObject *
- os__getdiskusage_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=3bd3991f5e5c5dfb input=6af8d1b7781cc042]*/
- {
- BOOL retval;
- ULARGE_INTEGER _, total, free;
- DWORD err = 0;
- Py_BEGIN_ALLOW_THREADS
- retval = GetDiskFreeSpaceExW(path->wide, &_, &total, &free);
- Py_END_ALLOW_THREADS
- if (retval == 0) {
- if (GetLastError() == ERROR_DIRECTORY) {
- wchar_t *dir_path = NULL;
- dir_path = PyMem_New(wchar_t, path->length + 1);
- if (dir_path == NULL) {
- return PyErr_NoMemory();
- }
- wcscpy_s(dir_path, path->length + 1, path->wide);
- if (_dirnameW(dir_path) != -1) {
- Py_BEGIN_ALLOW_THREADS
- retval = GetDiskFreeSpaceExW(dir_path, &_, &total, &free);
- Py_END_ALLOW_THREADS
- }
- /* Record the last error in case it's modified by PyMem_Free. */
- err = GetLastError();
- PyMem_Free(dir_path);
- if (retval) {
- goto success;
- }
- }
- return PyErr_SetFromWindowsErr(err);
- }
- success:
- return Py_BuildValue("(LL)", total.QuadPart, free.QuadPart);
- }
- #endif /* MS_WINDOWS */
- /* This is used for fpathconf(), pathconf(), confstr() and sysconf().
- * It maps strings representing configuration variable names to
- * integer values, allowing those functions to be called with the
- * magic names instead of polluting the module's namespace with tons of
- * rarely-used constants. There are three separate tables that use
- * these definitions.
- *
- * This code is always included, even if none of the interfaces that
- * need it are included. The #if hackery needed to avoid it would be
- * sufficiently pervasive that it's not worth the loss of readability.
- */
- struct constdef {
- const char *name;
- int value;
- };
- static int
- conv_confname(PyObject *arg, int *valuep, struct constdef *table,
- size_t tablesize)
- {
- if (PyLong_Check(arg)) {
- int value = _PyLong_AsInt(arg);
- if (value == -1 && PyErr_Occurred())
- return 0;
- *valuep = value;
- return 1;
- }
- else {
- /* look up the value in the table using a binary search */
- size_t lo = 0;
- size_t mid;
- size_t hi = tablesize;
- int cmp;
- const char *confname;
- if (!PyUnicode_Check(arg)) {
- PyErr_SetString(PyExc_TypeError,
- "configuration names must be strings or integers");
- return 0;
- }
- confname = PyUnicode_AsUTF8(arg);
- if (confname == NULL)
- return 0;
- while (lo < hi) {
- mid = (lo + hi) / 2;
- cmp = strcmp(confname, table[mid].name);
- if (cmp < 0)
- hi = mid;
- else if (cmp > 0)
- lo = mid + 1;
- else {
- *valuep = table[mid].value;
- return 1;
- }
- }
- PyErr_SetString(PyExc_ValueError, "unrecognized configuration name");
- return 0;
- }
- }
- #if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
- static struct constdef posix_constants_pathconf[] = {
- #ifdef _PC_ABI_AIO_XFER_MAX
- {"PC_ABI_AIO_XFER_MAX", _PC_ABI_AIO_XFER_MAX},
- #endif
- #ifdef _PC_ABI_ASYNC_IO
- {"PC_ABI_ASYNC_IO", _PC_ABI_ASYNC_IO},
- #endif
- #ifdef _PC_ASYNC_IO
- {"PC_ASYNC_IO", _PC_ASYNC_IO},
- #endif
- #ifdef _PC_CHOWN_RESTRICTED
- {"PC_CHOWN_RESTRICTED", _PC_CHOWN_RESTRICTED},
- #endif
- #ifdef _PC_FILESIZEBITS
- {"PC_FILESIZEBITS", _PC_FILESIZEBITS},
- #endif
- #ifdef _PC_LAST
- {"PC_LAST", _PC_LAST},
- #endif
- #ifdef _PC_LINK_MAX
- {"PC_LINK_MAX", _PC_LINK_MAX},
- #endif
- #ifdef _PC_MAX_CANON
- {"PC_MAX_CANON", _PC_MAX_CANON},
- #endif
- #ifdef _PC_MAX_INPUT
- {"PC_MAX_INPUT", _PC_MAX_INPUT},
- #endif
- #ifdef _PC_NAME_MAX
- {"PC_NAME_MAX", _PC_NAME_MAX},
- #endif
- #ifdef _PC_NO_TRUNC
- {"PC_NO_TRUNC", _PC_NO_TRUNC},
- #endif
- #ifdef _PC_PATH_MAX
- {"PC_PATH_MAX", _PC_PATH_MAX},
- #endif
- #ifdef _PC_PIPE_BUF
- {"PC_PIPE_BUF", _PC_PIPE_BUF},
- #endif
- #ifdef _PC_PRIO_IO
- {"PC_PRIO_IO", _PC_PRIO_IO},
- #endif
- #ifdef _PC_SOCK_MAXBUF
- {"PC_SOCK_MAXBUF", _PC_SOCK_MAXBUF},
- #endif
- #ifdef _PC_SYNC_IO
- {"PC_SYNC_IO", _PC_SYNC_IO},
- #endif
- #ifdef _PC_VDISABLE
- {"PC_VDISABLE", _PC_VDISABLE},
- #endif
- #ifdef _PC_ACL_ENABLED
- {"PC_ACL_ENABLED", _PC_ACL_ENABLED},
- #endif
- #ifdef _PC_MIN_HOLE_SIZE
- {"PC_MIN_HOLE_SIZE", _PC_MIN_HOLE_SIZE},
- #endif
- #ifdef _PC_ALLOC_SIZE_MIN
- {"PC_ALLOC_SIZE_MIN", _PC_ALLOC_SIZE_MIN},
- #endif
- #ifdef _PC_REC_INCR_XFER_SIZE
- {"PC_REC_INCR_XFER_SIZE", _PC_REC_INCR_XFER_SIZE},
- #endif
- #ifdef _PC_REC_MAX_XFER_SIZE
- {"PC_REC_MAX_XFER_SIZE", _PC_REC_MAX_XFER_SIZE},
- #endif
- #ifdef _PC_REC_MIN_XFER_SIZE
- {"PC_REC_MIN_XFER_SIZE", _PC_REC_MIN_XFER_SIZE},
- #endif
- #ifdef _PC_REC_XFER_ALIGN
- {"PC_REC_XFER_ALIGN", _PC_REC_XFER_ALIGN},
- #endif
- #ifdef _PC_SYMLINK_MAX
- {"PC_SYMLINK_MAX", _PC_SYMLINK_MAX},
- #endif
- #ifdef _PC_XATTR_ENABLED
- {"PC_XATTR_ENABLED", _PC_XATTR_ENABLED},
- #endif
- #ifdef _PC_XATTR_EXISTS
- {"PC_XATTR_EXISTS", _PC_XATTR_EXISTS},
- #endif
- #ifdef _PC_TIMESTAMP_RESOLUTION
- {"PC_TIMESTAMP_RESOLUTION", _PC_TIMESTAMP_RESOLUTION},
- #endif
- };
- static int
- conv_path_confname(PyObject *arg, int *valuep)
- {
- return conv_confname(arg, valuep, posix_constants_pathconf,
- sizeof(posix_constants_pathconf)
- / sizeof(struct constdef));
- }
- #endif
- #ifdef HAVE_FPATHCONF
- /*[clinic input]
- os.fpathconf -> long
- fd: fildes
- name: path_confname
- /
- Return the configuration limit name for the file descriptor fd.
- If there is no limit, return -1.
- [clinic start generated code]*/
- static long
- os_fpathconf_impl(PyObject *module, int fd, int name)
- /*[clinic end generated code: output=d5b7042425fc3e21 input=5b8d2471cfaae186]*/
- {
- long limit;
- errno = 0;
- limit = fpathconf(fd, name);
- if (limit == -1 && errno != 0)
- posix_error();
- return limit;
- }
- #endif /* HAVE_FPATHCONF */
- #ifdef HAVE_PATHCONF
- /*[clinic input]
- os.pathconf -> long
- path: path_t(allow_fd='PATH_HAVE_FPATHCONF')
- name: path_confname
- Return the configuration limit name for the file or directory path.
- If there is no limit, return -1.
- On some platforms, path may also be specified as an open file descriptor.
- If this functionality is unavailable, using it raises an exception.
- [clinic start generated code]*/
- static long
- os_pathconf_impl(PyObject *module, path_t *path, int name)
- /*[clinic end generated code: output=5bedee35b293a089 input=bc3e2a985af27e5e]*/
- {
- long limit;
- errno = 0;
- #ifdef HAVE_FPATHCONF
- if (path->fd != -1)
- limit = fpathconf(path->fd, name);
- else
- #endif
- limit = pathconf(path->narrow, name);
- if (limit == -1 && errno != 0) {
- if (errno == EINVAL)
- /* could be a path or name problem */
- posix_error();
- else
- path_error(path);
- }
- return limit;
- }
- #endif /* HAVE_PATHCONF */
- #ifdef HAVE_CONFSTR
- static struct constdef posix_constants_confstr[] = {
- #ifdef _CS_ARCHITECTURE
- {"CS_ARCHITECTURE", _CS_ARCHITECTURE},
- #endif
- #ifdef _CS_GNU_LIBC_VERSION
- {"CS_GNU_LIBC_VERSION", _CS_GNU_LIBC_VERSION},
- #endif
- #ifdef _CS_GNU_LIBPTHREAD_VERSION
- {"CS_GNU_LIBPTHREAD_VERSION", _CS_GNU_LIBPTHREAD_VERSION},
- #endif
- #ifdef _CS_HOSTNAME
- {"CS_HOSTNAME", _CS_HOSTNAME},
- #endif
- #ifdef _CS_HW_PROVIDER
- {"CS_HW_PROVIDER", _CS_HW_PROVIDER},
- #endif
- #ifdef _CS_HW_SERIAL
- {"CS_HW_SERIAL", _CS_HW_SERIAL},
- #endif
- #ifdef _CS_INITTAB_NAME
- {"CS_INITTAB_NAME", _CS_INITTAB_NAME},
- #endif
- #ifdef _CS_LFS64_CFLAGS
- {"CS_LFS64_CFLAGS", _CS_LFS64_CFLAGS},
- #endif
- #ifdef _CS_LFS64_LDFLAGS
- {"CS_LFS64_LDFLAGS", _CS_LFS64_LDFLAGS},
- #endif
- #ifdef _CS_LFS64_LIBS
- {"CS_LFS64_LIBS", _CS_LFS64_LIBS},
- #endif
- #ifdef _CS_LFS64_LINTFLAGS
- {"CS_LFS64_LINTFLAGS", _CS_LFS64_LINTFLAGS},
- #endif
- #ifdef _CS_LFS_CFLAGS
- {"CS_LFS_CFLAGS", _CS_LFS_CFLAGS},
- #endif
- #ifdef _CS_LFS_LDFLAGS
- {"CS_LFS_LDFLAGS", _CS_LFS_LDFLAGS},
- #endif
- #ifdef _CS_LFS_LIBS
- {"CS_LFS_LIBS", _CS_LFS_LIBS},
- #endif
- #ifdef _CS_LFS_LINTFLAGS
- {"CS_LFS_LINTFLAGS", _CS_LFS_LINTFLAGS},
- #endif
- #ifdef _CS_MACHINE
- {"CS_MACHINE", _CS_MACHINE},
- #endif
- #ifdef _CS_PATH
- {"CS_PATH", _CS_PATH},
- #endif
- #ifdef _CS_RELEASE
- {"CS_RELEASE", _CS_RELEASE},
- #endif
- #ifdef _CS_SRPC_DOMAIN
- {"CS_SRPC_DOMAIN", _CS_SRPC_DOMAIN},
- #endif
- #ifdef _CS_SYSNAME
- {"CS_SYSNAME", _CS_SYSNAME},
- #endif
- #ifdef _CS_VERSION
- {"CS_VERSION", _CS_VERSION},
- #endif
- #ifdef _CS_XBS5_ILP32_OFF32_CFLAGS
- {"CS_XBS5_ILP32_OFF32_CFLAGS", _CS_XBS5_ILP32_OFF32_CFLAGS},
- #endif
- #ifdef _CS_XBS5_ILP32_OFF32_LDFLAGS
- {"CS_XBS5_ILP32_OFF32_LDFLAGS", _CS_XBS5_ILP32_OFF32_LDFLAGS},
- #endif
- #ifdef _CS_XBS5_ILP32_OFF32_LIBS
- {"CS_XBS5_ILP32_OFF32_LIBS", _CS_XBS5_ILP32_OFF32_LIBS},
- #endif
- #ifdef _CS_XBS5_ILP32_OFF32_LINTFLAGS
- {"CS_XBS5_ILP32_OFF32_LINTFLAGS", _CS_XBS5_ILP32_OFF32_LINTFLAGS},
- #endif
- #ifdef _CS_XBS5_ILP32_OFFBIG_CFLAGS
- {"CS_XBS5_ILP32_OFFBIG_CFLAGS", _CS_XBS5_ILP32_OFFBIG_CFLAGS},
- #endif
- #ifdef _CS_XBS5_ILP32_OFFBIG_LDFLAGS
- {"CS_XBS5_ILP32_OFFBIG_LDFLAGS", _CS_XBS5_ILP32_OFFBIG_LDFLAGS},
- #endif
- #ifdef _CS_XBS5_ILP32_OFFBIG_LIBS
- {"CS_XBS5_ILP32_OFFBIG_LIBS", _CS_XBS5_ILP32_OFFBIG_LIBS},
- #endif
- #ifdef _CS_XBS5_ILP32_OFFBIG_LINTFLAGS
- {"CS_XBS5_ILP32_OFFBIG_LINTFLAGS", _CS_XBS5_ILP32_OFFBIG_LINTFLAGS},
- #endif
- #ifdef _CS_XBS5_LP64_OFF64_CFLAGS
- {"CS_XBS5_LP64_OFF64_CFLAGS", _CS_XBS5_LP64_OFF64_CFLAGS},
- #endif
- #ifdef _CS_XBS5_LP64_OFF64_LDFLAGS
- {"CS_XBS5_LP64_OFF64_LDFLAGS", _CS_XBS5_LP64_OFF64_LDFLAGS},
- #endif
- #ifdef _CS_XBS5_LP64_OFF64_LIBS
- {"CS_XBS5_LP64_OFF64_LIBS", _CS_XBS5_LP64_OFF64_LIBS},
- #endif
- #ifdef _CS_XBS5_LP64_OFF64_LINTFLAGS
- {"CS_XBS5_LP64_OFF64_LINTFLAGS", _CS_XBS5_LP64_OFF64_LINTFLAGS},
- #endif
- #ifdef _CS_XBS5_LPBIG_OFFBIG_CFLAGS
- {"CS_XBS5_LPBIG_OFFBIG_CFLAGS", _CS_XBS5_LPBIG_OFFBIG_CFLAGS},
- #endif
- #ifdef _CS_XBS5_LPBIG_OFFBIG_LDFLAGS
- {"CS_XBS5_LPBIG_OFFBIG_LDFLAGS", _CS_XBS5_LPBIG_OFFBIG_LDFLAGS},
- #endif
- #ifdef _CS_XBS5_LPBIG_OFFBIG_LIBS
- {"CS_XBS5_LPBIG_OFFBIG_LIBS", _CS_XBS5_LPBIG_OFFBIG_LIBS},
- #endif
- #ifdef _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS
- {"CS_XBS5_LPBIG_OFFBIG_LINTFLAGS", _CS_XBS5_LPBIG_OFFBIG_LINTFLAGS},
- #endif
- #ifdef _MIPS_CS_AVAIL_PROCESSORS
- {"MIPS_CS_AVAIL_PROCESSORS", _MIPS_CS_AVAIL_PROCESSORS},
- #endif
- #ifdef _MIPS_CS_BASE
- {"MIPS_CS_BASE", _MIPS_CS_BASE},
- #endif
- #ifdef _MIPS_CS_HOSTID
- {"MIPS_CS_HOSTID", _MIPS_CS_HOSTID},
- #endif
- #ifdef _MIPS_CS_HW_NAME
- {"MIPS_CS_HW_NAME", _MIPS_CS_HW_NAME},
- #endif
- #ifdef _MIPS_CS_NUM_PROCESSORS
- {"MIPS_CS_NUM_PROCESSORS", _MIPS_CS_NUM_PROCESSORS},
- #endif
- #ifdef _MIPS_CS_OSREL_MAJ
- {"MIPS_CS_OSREL_MAJ", _MIPS_CS_OSREL_MAJ},
- #endif
- #ifdef _MIPS_CS_OSREL_MIN
- {"MIPS_CS_OSREL_MIN", _MIPS_CS_OSREL_MIN},
- #endif
- #ifdef _MIPS_CS_OSREL_PATCH
- {"MIPS_CS_OSREL_PATCH", _MIPS_CS_OSREL_PATCH},
- #endif
- #ifdef _MIPS_CS_OS_NAME
- {"MIPS_CS_OS_NAME", _MIPS_CS_OS_NAME},
- #endif
- #ifdef _MIPS_CS_OS_PROVIDER
- {"MIPS_CS_OS_PROVIDER", _MIPS_CS_OS_PROVIDER},
- #endif
- #ifdef _MIPS_CS_PROCESSORS
- {"MIPS_CS_PROCESSORS", _MIPS_CS_PROCESSORS},
- #endif
- #ifdef _MIPS_CS_SERIAL
- {"MIPS_CS_SERIAL", _MIPS_CS_SERIAL},
- #endif
- #ifdef _MIPS_CS_VENDOR
- {"MIPS_CS_VENDOR", _MIPS_CS_VENDOR},
- #endif
- };
- static int
- conv_confstr_confname(PyObject *arg, int *valuep)
- {
- return conv_confname(arg, valuep, posix_constants_confstr,
- sizeof(posix_constants_confstr)
- / sizeof(struct constdef));
- }
- /*[clinic input]
- os.confstr
- name: confstr_confname
- /
- Return a string-valued system configuration variable.
- [clinic start generated code]*/
- static PyObject *
- os_confstr_impl(PyObject *module, int name)
- /*[clinic end generated code: output=bfb0b1b1e49b9383 input=18fb4d0567242e65]*/
- {
- PyObject *result = NULL;
- char buffer[255];
- size_t len;
- errno = 0;
- len = confstr(name, buffer, sizeof(buffer));
- if (len == 0) {
- if (errno) {
- posix_error();
- return NULL;
- }
- else {
- Py_RETURN_NONE;
- }
- }
- if (len >= sizeof(buffer)) {
- size_t len2;
- char *buf = PyMem_Malloc(len);
- if (buf == NULL)
- return PyErr_NoMemory();
- len2 = confstr(name, buf, len);
- assert(len == len2);
- result = PyUnicode_DecodeFSDefaultAndSize(buf, len2-1);
- PyMem_Free(buf);
- }
- else
- result = PyUnicode_DecodeFSDefaultAndSize(buffer, len-1);
- return result;
- }
- #endif /* HAVE_CONFSTR */
- #ifdef HAVE_SYSCONF
- static struct constdef posix_constants_sysconf[] = {
- #ifdef _SC_2_CHAR_TERM
- {"SC_2_CHAR_TERM", _SC_2_CHAR_TERM},
- #endif
- #ifdef _SC_2_C_BIND
- {"SC_2_C_BIND", _SC_2_C_BIND},
- #endif
- #ifdef _SC_2_C_DEV
- {"SC_2_C_DEV", _SC_2_C_DEV},
- #endif
- #ifdef _SC_2_C_VERSION
- {"SC_2_C_VERSION", _SC_2_C_VERSION},
- #endif
- #ifdef _SC_2_FORT_DEV
- {"SC_2_FORT_DEV", _SC_2_FORT_DEV},
- #endif
- #ifdef _SC_2_FORT_RUN
- {"SC_2_FORT_RUN", _SC_2_FORT_RUN},
- #endif
- #ifdef _SC_2_LOCALEDEF
- {"SC_2_LOCALEDEF", _SC_2_LOCALEDEF},
- #endif
- #ifdef _SC_2_SW_DEV
- {"SC_2_SW_DEV", _SC_2_SW_DEV},
- #endif
- #ifdef _SC_2_UPE
- {"SC_2_UPE", _SC_2_UPE},
- #endif
- #ifdef _SC_2_VERSION
- {"SC_2_VERSION", _SC_2_VERSION},
- #endif
- #ifdef _SC_ABI_ASYNCHRONOUS_IO
- {"SC_ABI_ASYNCHRONOUS_IO", _SC_ABI_ASYNCHRONOUS_IO},
- #endif
- #ifdef _SC_ACL
- {"SC_ACL", _SC_ACL},
- #endif
- #ifdef _SC_AIO_LISTIO_MAX
- {"SC_AIO_LISTIO_MAX", _SC_AIO_LISTIO_MAX},
- #endif
- #ifdef _SC_AIO_MAX
- {"SC_AIO_MAX", _SC_AIO_MAX},
- #endif
- #ifdef _SC_AIO_PRIO_DELTA_MAX
- {"SC_AIO_PRIO_DELTA_MAX", _SC_AIO_PRIO_DELTA_MAX},
- #endif
- #ifdef _SC_ARG_MAX
- {"SC_ARG_MAX", _SC_ARG_MAX},
- #endif
- #ifdef _SC_ASYNCHRONOUS_IO
- {"SC_ASYNCHRONOUS_IO", _SC_ASYNCHRONOUS_IO},
- #endif
- #ifdef _SC_ATEXIT_MAX
- {"SC_ATEXIT_MAX", _SC_ATEXIT_MAX},
- #endif
- #ifdef _SC_AUDIT
- {"SC_AUDIT", _SC_AUDIT},
- #endif
- #ifdef _SC_AVPHYS_PAGES
- {"SC_AVPHYS_PAGES", _SC_AVPHYS_PAGES},
- #endif
- #ifdef _SC_BC_BASE_MAX
- {"SC_BC_BASE_MAX", _SC_BC_BASE_MAX},
- #endif
- #ifdef _SC_BC_DIM_MAX
- {"SC_BC_DIM_MAX", _SC_BC_DIM_MAX},
- #endif
- #ifdef _SC_BC_SCALE_MAX
- {"SC_BC_SCALE_MAX", _SC_BC_SCALE_MAX},
- #endif
- #ifdef _SC_BC_STRING_MAX
- {"SC_BC_STRING_MAX", _SC_BC_STRING_MAX},
- #endif
- #ifdef _SC_CAP
- {"SC_CAP", _SC_CAP},
- #endif
- #ifdef _SC_CHARCLASS_NAME_MAX
- {"SC_CHARCLASS_NAME_MAX", _SC_CHARCLASS_NAME_MAX},
- #endif
- #ifdef _SC_CHAR_BIT
- {"SC_CHAR_BIT", _SC_CHAR_BIT},
- #endif
- #ifdef _SC_CHAR_MAX
- {"SC_CHAR_MAX", _SC_CHAR_MAX},
- #endif
- #ifdef _SC_CHAR_MIN
- {"SC_CHAR_MIN", _SC_CHAR_MIN},
- #endif
- #ifdef _SC_CHILD_MAX
- {"SC_CHILD_MAX", _SC_CHILD_MAX},
- #endif
- #ifdef _SC_CLK_TCK
- {"SC_CLK_TCK", _SC_CLK_TCK},
- #endif
- #ifdef _SC_COHER_BLKSZ
- {"SC_COHER_BLKSZ", _SC_COHER_BLKSZ},
- #endif
- #ifdef _SC_COLL_WEIGHTS_MAX
- {"SC_COLL_WEIGHTS_MAX", _SC_COLL_WEIGHTS_MAX},
- #endif
- #ifdef _SC_DCACHE_ASSOC
- {"SC_DCACHE_ASSOC", _SC_DCACHE_ASSOC},
- #endif
- #ifdef _SC_DCACHE_BLKSZ
- {"SC_DCACHE_BLKSZ", _SC_DCACHE_BLKSZ},
- #endif
- #ifdef _SC_DCACHE_LINESZ
- {"SC_DCACHE_LINESZ", _SC_DCACHE_LINESZ},
- #endif
- #ifdef _SC_DCACHE_SZ
- {"SC_DCACHE_SZ", _SC_DCACHE_SZ},
- #endif
- #ifdef _SC_DCACHE_TBLKSZ
- {"SC_DCACHE_TBLKSZ", _SC_DCACHE_TBLKSZ},
- #endif
- #ifdef _SC_DELAYTIMER_MAX
- {"SC_DELAYTIMER_MAX", _SC_DELAYTIMER_MAX},
- #endif
- #ifdef _SC_EQUIV_CLASS_MAX
- {"SC_EQUIV_CLASS_MAX", _SC_EQUIV_CLASS_MAX},
- #endif
- #ifdef _SC_EXPR_NEST_MAX
- {"SC_EXPR_NEST_MAX", _SC_EXPR_NEST_MAX},
- #endif
- #ifdef _SC_FSYNC
- {"SC_FSYNC", _SC_FSYNC},
- #endif
- #ifdef _SC_GETGR_R_SIZE_MAX
- {"SC_GETGR_R_SIZE_MAX", _SC_GETGR_R_SIZE_MAX},
- #endif
- #ifdef _SC_GETPW_R_SIZE_MAX
- {"SC_GETPW_R_SIZE_MAX", _SC_GETPW_R_SIZE_MAX},
- #endif
- #ifdef _SC_ICACHE_ASSOC
- {"SC_ICACHE_ASSOC", _SC_ICACHE_ASSOC},
- #endif
- #ifdef _SC_ICACHE_BLKSZ
- {"SC_ICACHE_BLKSZ", _SC_ICACHE_BLKSZ},
- #endif
- #ifdef _SC_ICACHE_LINESZ
- {"SC_ICACHE_LINESZ", _SC_ICACHE_LINESZ},
- #endif
- #ifdef _SC_ICACHE_SZ
- {"SC_ICACHE_SZ", _SC_ICACHE_SZ},
- #endif
- #ifdef _SC_INF
- {"SC_INF", _SC_INF},
- #endif
- #ifdef _SC_INT_MAX
- {"SC_INT_MAX", _SC_INT_MAX},
- #endif
- #ifdef _SC_INT_MIN
- {"SC_INT_MIN", _SC_INT_MIN},
- #endif
- #ifdef _SC_IOV_MAX
- {"SC_IOV_MAX", _SC_IOV_MAX},
- #endif
- #ifdef _SC_IP_SECOPTS
- {"SC_IP_SECOPTS", _SC_IP_SECOPTS},
- #endif
- #ifdef _SC_JOB_CONTROL
- {"SC_JOB_CONTROL", _SC_JOB_CONTROL},
- #endif
- #ifdef _SC_KERN_POINTERS
- {"SC_KERN_POINTERS", _SC_KERN_POINTERS},
- #endif
- #ifdef _SC_KERN_SIM
- {"SC_KERN_SIM", _SC_KERN_SIM},
- #endif
- #ifdef _SC_LINE_MAX
- {"SC_LINE_MAX", _SC_LINE_MAX},
- #endif
- #ifdef _SC_LOGIN_NAME_MAX
- {"SC_LOGIN_NAME_MAX", _SC_LOGIN_NAME_MAX},
- #endif
- #ifdef _SC_LOGNAME_MAX
- {"SC_LOGNAME_MAX", _SC_LOGNAME_MAX},
- #endif
- #ifdef _SC_LONG_BIT
- {"SC_LONG_BIT", _SC_LONG_BIT},
- #endif
- #ifdef _SC_MAC
- {"SC_MAC", _SC_MAC},
- #endif
- #ifdef _SC_MAPPED_FILES
- {"SC_MAPPED_FILES", _SC_MAPPED_FILES},
- #endif
- #ifdef _SC_MAXPID
- {"SC_MAXPID", _SC_MAXPID},
- #endif
- #ifdef _SC_MB_LEN_MAX
- {"SC_MB_LEN_MAX", _SC_MB_LEN_MAX},
- #endif
- #ifdef _SC_MEMLOCK
- {"SC_MEMLOCK", _SC_MEMLOCK},
- #endif
- #ifdef _SC_MEMLOCK_RANGE
- {"SC_MEMLOCK_RANGE", _SC_MEMLOCK_RANGE},
- #endif
- #ifdef _SC_MEMORY_PROTECTION
- {"SC_MEMORY_PROTECTION", _SC_MEMORY_PROTECTION},
- #endif
- #ifdef _SC_MESSAGE_PASSING
- {"SC_MESSAGE_PASSING", _SC_MESSAGE_PASSING},
- #endif
- #ifdef _SC_MMAP_FIXED_ALIGNMENT
- {"SC_MMAP_FIXED_ALIGNMENT", _SC_MMAP_FIXED_ALIGNMENT},
- #endif
- #ifdef _SC_MQ_OPEN_MAX
- {"SC_MQ_OPEN_MAX", _SC_MQ_OPEN_MAX},
- #endif
- #ifdef _SC_MQ_PRIO_MAX
- {"SC_MQ_PRIO_MAX", _SC_MQ_PRIO_MAX},
- #endif
- #ifdef _SC_NACLS_MAX
- {"SC_NACLS_MAX", _SC_NACLS_MAX},
- #endif
- #ifdef _SC_NGROUPS_MAX
- {"SC_NGROUPS_MAX", _SC_NGROUPS_MAX},
- #endif
- #ifdef _SC_NL_ARGMAX
- {"SC_NL_ARGMAX", _SC_NL_ARGMAX},
- #endif
- #ifdef _SC_NL_LANGMAX
- {"SC_NL_LANGMAX", _SC_NL_LANGMAX},
- #endif
- #ifdef _SC_NL_MSGMAX
- {"SC_NL_MSGMAX", _SC_NL_MSGMAX},
- #endif
- #ifdef _SC_NL_NMAX
- {"SC_NL_NMAX", _SC_NL_NMAX},
- #endif
- #ifdef _SC_NL_SETMAX
- {"SC_NL_SETMAX", _SC_NL_SETMAX},
- #endif
- #ifdef _SC_NL_TEXTMAX
- {"SC_NL_TEXTMAX", _SC_NL_TEXTMAX},
- #endif
- #ifdef _SC_NPROCESSORS_CONF
- {"SC_NPROCESSORS_CONF", _SC_NPROCESSORS_CONF},
- #endif
- #ifdef _SC_NPROCESSORS_ONLN
- {"SC_NPROCESSORS_ONLN", _SC_NPROCESSORS_ONLN},
- #endif
- #ifdef _SC_NPROC_CONF
- {"SC_NPROC_CONF", _SC_NPROC_CONF},
- #endif
- #ifdef _SC_NPROC_ONLN
- {"SC_NPROC_ONLN", _SC_NPROC_ONLN},
- #endif
- #ifdef _SC_NZERO
- {"SC_NZERO", _SC_NZERO},
- #endif
- #ifdef _SC_OPEN_MAX
- {"SC_OPEN_MAX", _SC_OPEN_MAX},
- #endif
- #ifdef _SC_PAGESIZE
- {"SC_PAGESIZE", _SC_PAGESIZE},
- #endif
- #ifdef _SC_PAGE_SIZE
- {"SC_PAGE_SIZE", _SC_PAGE_SIZE},
- #endif
- #ifdef _SC_AIX_REALMEM
- {"SC_AIX_REALMEM", _SC_AIX_REALMEM},
- #endif
- #ifdef _SC_PASS_MAX
- {"SC_PASS_MAX", _SC_PASS_MAX},
- #endif
- #ifdef _SC_PHYS_PAGES
- {"SC_PHYS_PAGES", _SC_PHYS_PAGES},
- #endif
- #ifdef _SC_PII
- {"SC_PII", _SC_PII},
- #endif
- #ifdef _SC_PII_INTERNET
- {"SC_PII_INTERNET", _SC_PII_INTERNET},
- #endif
- #ifdef _SC_PII_INTERNET_DGRAM
- {"SC_PII_INTERNET_DGRAM", _SC_PII_INTERNET_DGRAM},
- #endif
- #ifdef _SC_PII_INTERNET_STREAM
- {"SC_PII_INTERNET_STREAM", _SC_PII_INTERNET_STREAM},
- #endif
- #ifdef _SC_PII_OSI
- {"SC_PII_OSI", _SC_PII_OSI},
- #endif
- #ifdef _SC_PII_OSI_CLTS
- {"SC_PII_OSI_CLTS", _SC_PII_OSI_CLTS},
- #endif
- #ifdef _SC_PII_OSI_COTS
- {"SC_PII_OSI_COTS", _SC_PII_OSI_COTS},
- #endif
- #ifdef _SC_PII_OSI_M
- {"SC_PII_OSI_M", _SC_PII_OSI_M},
- #endif
- #ifdef _SC_PII_SOCKET
- {"SC_PII_SOCKET", _SC_PII_SOCKET},
- #endif
- #ifdef _SC_PII_XTI
- {"SC_PII_XTI", _SC_PII_XTI},
- #endif
- #ifdef _SC_POLL
- {"SC_POLL", _SC_POLL},
- #endif
- #ifdef _SC_PRIORITIZED_IO
- {"SC_PRIORITIZED_IO", _SC_PRIORITIZED_IO},
- #endif
- #ifdef _SC_PRIORITY_SCHEDULING
- {"SC_PRIORITY_SCHEDULING", _SC_PRIORITY_SCHEDULING},
- #endif
- #ifdef _SC_REALTIME_SIGNALS
- {"SC_REALTIME_SIGNALS", _SC_REALTIME_SIGNALS},
- #endif
- #ifdef _SC_RE_DUP_MAX
- {"SC_RE_DUP_MAX", _SC_RE_DUP_MAX},
- #endif
- #ifdef _SC_RTSIG_MAX
- {"SC_RTSIG_MAX", _SC_RTSIG_MAX},
- #endif
- #ifdef _SC_SAVED_IDS
- {"SC_SAVED_IDS", _SC_SAVED_IDS},
- #endif
- #ifdef _SC_SCHAR_MAX
- {"SC_SCHAR_MAX", _SC_SCHAR_MAX},
- #endif
- #ifdef _SC_SCHAR_MIN
- {"SC_SCHAR_MIN", _SC_SCHAR_MIN},
- #endif
- #ifdef _SC_SELECT
- {"SC_SELECT", _SC_SELECT},
- #endif
- #ifdef _SC_SEMAPHORES
- {"SC_SEMAPHORES", _SC_SEMAPHORES},
- #endif
- #ifdef _SC_SEM_NSEMS_MAX
- {"SC_SEM_NSEMS_MAX", _SC_SEM_NSEMS_MAX},
- #endif
- #ifdef _SC_SEM_VALUE_MAX
- {"SC_SEM_VALUE_MAX", _SC_SEM_VALUE_MAX},
- #endif
- #ifdef _SC_SHARED_MEMORY_OBJECTS
- {"SC_SHARED_MEMORY_OBJECTS", _SC_SHARED_MEMORY_OBJECTS},
- #endif
- #ifdef _SC_SHRT_MAX
- {"SC_SHRT_MAX", _SC_SHRT_MAX},
- #endif
- #ifdef _SC_SHRT_MIN
- {"SC_SHRT_MIN", _SC_SHRT_MIN},
- #endif
- #ifdef _SC_SIGQUEUE_MAX
- {"SC_SIGQUEUE_MAX", _SC_SIGQUEUE_MAX},
- #endif
- #ifdef _SC_SIGRT_MAX
- {"SC_SIGRT_MAX", _SC_SIGRT_MAX},
- #endif
- #ifdef _SC_SIGRT_MIN
- {"SC_SIGRT_MIN", _SC_SIGRT_MIN},
- #endif
- #ifdef _SC_SOFTPOWER
- {"SC_SOFTPOWER", _SC_SOFTPOWER},
- #endif
- #ifdef _SC_SPLIT_CACHE
- {"SC_SPLIT_CACHE", _SC_SPLIT_CACHE},
- #endif
- #ifdef _SC_SSIZE_MAX
- {"SC_SSIZE_MAX", _SC_SSIZE_MAX},
- #endif
- #ifdef _SC_STACK_PROT
- {"SC_STACK_PROT", _SC_STACK_PROT},
- #endif
- #ifdef _SC_STREAM_MAX
- {"SC_STREAM_MAX", _SC_STREAM_MAX},
- #endif
- #ifdef _SC_SYNCHRONIZED_IO
- {"SC_SYNCHRONIZED_IO", _SC_SYNCHRONIZED_IO},
- #endif
- #ifdef _SC_THREADS
- {"SC_THREADS", _SC_THREADS},
- #endif
- #ifdef _SC_THREAD_ATTR_STACKADDR
- {"SC_THREAD_ATTR_STACKADDR", _SC_THREAD_ATTR_STACKADDR},
- #endif
- #ifdef _SC_THREAD_ATTR_STACKSIZE
- {"SC_THREAD_ATTR_STACKSIZE", _SC_THREAD_ATTR_STACKSIZE},
- #endif
- #ifdef _SC_THREAD_DESTRUCTOR_ITERATIONS
- {"SC_THREAD_DESTRUCTOR_ITERATIONS", _SC_THREAD_DESTRUCTOR_ITERATIONS},
- #endif
- #ifdef _SC_THREAD_KEYS_MAX
- {"SC_THREAD_KEYS_MAX", _SC_THREAD_KEYS_MAX},
- #endif
- #ifdef _SC_THREAD_PRIORITY_SCHEDULING
- {"SC_THREAD_PRIORITY_SCHEDULING", _SC_THREAD_PRIORITY_SCHEDULING},
- #endif
- #ifdef _SC_THREAD_PRIO_INHERIT
- {"SC_THREAD_PRIO_INHERIT", _SC_THREAD_PRIO_INHERIT},
- #endif
- #ifdef _SC_THREAD_PRIO_PROTECT
- {"SC_THREAD_PRIO_PROTECT", _SC_THREAD_PRIO_PROTECT},
- #endif
- #ifdef _SC_THREAD_PROCESS_SHARED
- {"SC_THREAD_PROCESS_SHARED", _SC_THREAD_PROCESS_SHARED},
- #endif
- #ifdef _SC_THREAD_SAFE_FUNCTIONS
- {"SC_THREAD_SAFE_FUNCTIONS", _SC_THREAD_SAFE_FUNCTIONS},
- #endif
- #ifdef _SC_THREAD_STACK_MIN
- {"SC_THREAD_STACK_MIN", _SC_THREAD_STACK_MIN},
- #endif
- #ifdef _SC_THREAD_THREADS_MAX
- {"SC_THREAD_THREADS_MAX", _SC_THREAD_THREADS_MAX},
- #endif
- #ifdef _SC_TIMERS
- {"SC_TIMERS", _SC_TIMERS},
- #endif
- #ifdef _SC_TIMER_MAX
- {"SC_TIMER_MAX", _SC_TIMER_MAX},
- #endif
- #ifdef _SC_TTY_NAME_MAX
- {"SC_TTY_NAME_MAX", _SC_TTY_NAME_MAX},
- #endif
- #ifdef _SC_TZNAME_MAX
- {"SC_TZNAME_MAX", _SC_TZNAME_MAX},
- #endif
- #ifdef _SC_T_IOV_MAX
- {"SC_T_IOV_MAX", _SC_T_IOV_MAX},
- #endif
- #ifdef _SC_UCHAR_MAX
- {"SC_UCHAR_MAX", _SC_UCHAR_MAX},
- #endif
- #ifdef _SC_UINT_MAX
- {"SC_UINT_MAX", _SC_UINT_MAX},
- #endif
- #ifdef _SC_UIO_MAXIOV
- {"SC_UIO_MAXIOV", _SC_UIO_MAXIOV},
- #endif
- #ifdef _SC_ULONG_MAX
- {"SC_ULONG_MAX", _SC_ULONG_MAX},
- #endif
- #ifdef _SC_USHRT_MAX
- {"SC_USHRT_MAX", _SC_USHRT_MAX},
- #endif
- #ifdef _SC_VERSION
- {"SC_VERSION", _SC_VERSION},
- #endif
- #ifdef _SC_WORD_BIT
- {"SC_WORD_BIT", _SC_WORD_BIT},
- #endif
- #ifdef _SC_XBS5_ILP32_OFF32
- {"SC_XBS5_ILP32_OFF32", _SC_XBS5_ILP32_OFF32},
- #endif
- #ifdef _SC_XBS5_ILP32_OFFBIG
- {"SC_XBS5_ILP32_OFFBIG", _SC_XBS5_ILP32_OFFBIG},
- #endif
- #ifdef _SC_XBS5_LP64_OFF64
- {"SC_XBS5_LP64_OFF64", _SC_XBS5_LP64_OFF64},
- #endif
- #ifdef _SC_XBS5_LPBIG_OFFBIG
- {"SC_XBS5_LPBIG_OFFBIG", _SC_XBS5_LPBIG_OFFBIG},
- #endif
- #ifdef _SC_XOPEN_CRYPT
- {"SC_XOPEN_CRYPT", _SC_XOPEN_CRYPT},
- #endif
- #ifdef _SC_XOPEN_ENH_I18N
- {"SC_XOPEN_ENH_I18N", _SC_XOPEN_ENH_I18N},
- #endif
- #ifdef _SC_XOPEN_LEGACY
- {"SC_XOPEN_LEGACY", _SC_XOPEN_LEGACY},
- #endif
- #ifdef _SC_XOPEN_REALTIME
- {"SC_XOPEN_REALTIME", _SC_XOPEN_REALTIME},
- #endif
- #ifdef _SC_XOPEN_REALTIME_THREADS
- {"SC_XOPEN_REALTIME_THREADS", _SC_XOPEN_REALTIME_THREADS},
- #endif
- #ifdef _SC_XOPEN_SHM
- {"SC_XOPEN_SHM", _SC_XOPEN_SHM},
- #endif
- #ifdef _SC_XOPEN_UNIX
- {"SC_XOPEN_UNIX", _SC_XOPEN_UNIX},
- #endif
- #ifdef _SC_XOPEN_VERSION
- {"SC_XOPEN_VERSION", _SC_XOPEN_VERSION},
- #endif
- #ifdef _SC_XOPEN_XCU_VERSION
- {"SC_XOPEN_XCU_VERSION", _SC_XOPEN_XCU_VERSION},
- #endif
- #ifdef _SC_XOPEN_XPG2
- {"SC_XOPEN_XPG2", _SC_XOPEN_XPG2},
- #endif
- #ifdef _SC_XOPEN_XPG3
- {"SC_XOPEN_XPG3", _SC_XOPEN_XPG3},
- #endif
- #ifdef _SC_XOPEN_XPG4
- {"SC_XOPEN_XPG4", _SC_XOPEN_XPG4},
- #endif
- #ifdef _SC_MINSIGSTKSZ
- {"SC_MINSIGSTKSZ", _SC_MINSIGSTKSZ},
- #endif
- };
- static int
- conv_sysconf_confname(PyObject *arg, int *valuep)
- {
- return conv_confname(arg, valuep, posix_constants_sysconf,
- sizeof(posix_constants_sysconf)
- / sizeof(struct constdef));
- }
- /*[clinic input]
- os.sysconf -> long
- name: sysconf_confname
- /
- Return an integer-valued system configuration variable.
- [clinic start generated code]*/
- static long
- os_sysconf_impl(PyObject *module, int name)
- /*[clinic end generated code: output=3662f945fc0cc756 input=279e3430a33f29e4]*/
- {
- long value;
- errno = 0;
- value = sysconf(name);
- if (value == -1 && errno != 0)
- posix_error();
- return value;
- }
- #endif /* HAVE_SYSCONF */
- /* This code is used to ensure that the tables of configuration value names
- * are in sorted order as required by conv_confname(), and also to build
- * the exported dictionaries that are used to publish information about the
- * names available on the host platform.
- *
- * Sorting the table at runtime ensures that the table is properly ordered
- * when used, even for platforms we're not able to test on. It also makes
- * it easier to add additional entries to the tables.
- */
- static int
- cmp_constdefs(const void *v1, const void *v2)
- {
- const struct constdef *c1 =
- (const struct constdef *) v1;
- const struct constdef *c2 =
- (const struct constdef *) v2;
- return strcmp(c1->name, c2->name);
- }
- static int
- setup_confname_table(struct constdef *table, size_t tablesize,
- const char *tablename, PyObject *module)
- {
- PyObject *d = NULL;
- size_t i;
- qsort(table, tablesize, sizeof(struct constdef), cmp_constdefs);
- d = PyDict_New();
- if (d == NULL)
- return -1;
- for (i=0; i < tablesize; ++i) {
- PyObject *o = PyLong_FromLong(table[i].value);
- if (o == NULL || PyDict_SetItemString(d, table[i].name, o) == -1) {
- Py_XDECREF(o);
- Py_DECREF(d);
- return -1;
- }
- Py_DECREF(o);
- }
- return _PyModule_Add(module, tablename, d);
- }
- /* Return -1 on failure, 0 on success. */
- static int
- setup_confname_tables(PyObject *module)
- {
- #if defined(HAVE_FPATHCONF) || defined(HAVE_PATHCONF)
- if (setup_confname_table(posix_constants_pathconf,
- sizeof(posix_constants_pathconf)
- / sizeof(struct constdef),
- "pathconf_names", module))
- return -1;
- #endif
- #ifdef HAVE_CONFSTR
- if (setup_confname_table(posix_constants_confstr,
- sizeof(posix_constants_confstr)
- / sizeof(struct constdef),
- "confstr_names", module))
- return -1;
- #endif
- #ifdef HAVE_SYSCONF
- if (setup_confname_table(posix_constants_sysconf,
- sizeof(posix_constants_sysconf)
- / sizeof(struct constdef),
- "sysconf_names", module))
- return -1;
- #endif
- return 0;
- }
- /*[clinic input]
- os.abort
- Abort the interpreter immediately.
- This function 'dumps core' or otherwise fails in the hardest way possible
- on the hosting operating system. This function never returns.
- [clinic start generated code]*/
- static PyObject *
- os_abort_impl(PyObject *module)
- /*[clinic end generated code: output=dcf52586dad2467c input=cf2c7d98bc504047]*/
- {
- abort();
- /*NOTREACHED*/
- #ifndef __clang__
- /* Issue #28152: abort() is declared with __attribute__((__noreturn__)).
- GCC emits a warning without "return NULL;" (compiler bug?), but Clang
- is smarter and emits a warning on the return. */
- Py_FatalError("abort() called from Python code didn't abort!");
- return NULL;
- #endif
- }
- #ifdef MS_WINDOWS
- /* Grab ShellExecute dynamically from shell32 */
- static int has_ShellExecute = -1;
- static HINSTANCE (CALLBACK *Py_ShellExecuteW)(HWND, LPCWSTR, LPCWSTR, LPCWSTR,
- LPCWSTR, INT);
- static int
- check_ShellExecute(void)
- {
- HINSTANCE hShell32;
- /* only recheck */
- if (-1 == has_ShellExecute) {
- Py_BEGIN_ALLOW_THREADS
- /* Security note: this call is not vulnerable to "DLL hijacking".
- SHELL32 is part of "KnownDLLs" and so Windows always load
- the system SHELL32.DLL, even if there is another SHELL32.DLL
- in the DLL search path. */
- hShell32 = LoadLibraryW(L"SHELL32");
- if (hShell32) {
- *(FARPROC*)&Py_ShellExecuteW = GetProcAddress(hShell32,
- "ShellExecuteW");
- has_ShellExecute = Py_ShellExecuteW != NULL;
- } else {
- has_ShellExecute = 0;
- }
- Py_END_ALLOW_THREADS
- }
- return has_ShellExecute;
- }
- /*[clinic input]
- os.startfile
- filepath: path_t
- operation: Py_UNICODE = NULL
- arguments: Py_UNICODE = NULL
- cwd: path_t(nullable=True) = None
- show_cmd: int = 1
- Start a file with its associated application.
- When "operation" is not specified or "open", this acts like
- double-clicking the file in Explorer, or giving the file name as an
- argument to the DOS "start" command: the file is opened with whatever
- application (if any) its extension is associated.
- When another "operation" is given, it specifies what should be done with
- the file. A typical operation is "print".
- "arguments" is passed to the application, but should be omitted if the
- file is a document.
- "cwd" is the working directory for the operation. If "filepath" is
- relative, it will be resolved against this directory. This argument
- should usually be an absolute path.
- "show_cmd" can be used to override the recommended visibility option.
- See the Windows ShellExecute documentation for values.
- startfile returns as soon as the associated application is launched.
- There is no option to wait for the application to close, and no way
- to retrieve the application's exit status.
- The filepath is relative to the current directory. If you want to use
- an absolute path, make sure the first character is not a slash ("/");
- the underlying Win32 ShellExecute function doesn't work if it is.
- [clinic start generated code]*/
- static PyObject *
- os_startfile_impl(PyObject *module, path_t *filepath,
- const Py_UNICODE *operation, const Py_UNICODE *arguments,
- path_t *cwd, int show_cmd)
- /*[clinic end generated code: output=3baa4f9795841880 input=8248997b80669622]*/
- {
- HINSTANCE rc;
- if(!check_ShellExecute()) {
- /* If the OS doesn't have ShellExecute, return a
- NotImplementedError. */
- return PyErr_Format(PyExc_NotImplementedError,
- "startfile not available on this platform");
- }
- if (PySys_Audit("os.startfile", "Ou", filepath->object, operation) < 0) {
- return NULL;
- }
- if (PySys_Audit("os.startfile/2", "OuuOi", filepath->object, operation,
- arguments, cwd->object ? cwd->object : Py_None,
- show_cmd) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- rc = Py_ShellExecuteW((HWND)0, operation, filepath->wide,
- arguments, cwd->wide, show_cmd);
- Py_END_ALLOW_THREADS
- if (rc <= (HINSTANCE)32) {
- win32_error_object("startfile", filepath->object);
- return NULL;
- }
- Py_RETURN_NONE;
- }
- #endif /* MS_WINDOWS */
- #ifdef HAVE_GETLOADAVG
- /*[clinic input]
- os.getloadavg
- Return average recent system load information.
- Return the number of processes in the system run queue averaged over
- the last 1, 5, and 15 minutes as a tuple of three floats.
- Raises OSError if the load average was unobtainable.
- [clinic start generated code]*/
- static PyObject *
- os_getloadavg_impl(PyObject *module)
- /*[clinic end generated code: output=9ad3a11bfb4f4bd2 input=3d6d826b76d8a34e]*/
- {
- double loadavg[3];
- if (getloadavg(loadavg, 3)!=3) {
- PyErr_SetString(PyExc_OSError, "Load averages are unobtainable");
- return NULL;
- } else
- return Py_BuildValue("ddd", loadavg[0], loadavg[1], loadavg[2]);
- }
- #endif /* HAVE_GETLOADAVG */
- /*[clinic input]
- os.device_encoding
- fd: int
- Return a string describing the encoding of a terminal's file descriptor.
- The file descriptor must be attached to a terminal.
- If the device is not a terminal, return None.
- [clinic start generated code]*/
- static PyObject *
- os_device_encoding_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=e0d294bbab7e8c2b input=9e1d4a42b66df312]*/
- {
- return _Py_device_encoding(fd);
- }
- #ifdef HAVE_SETRESUID
- /*[clinic input]
- os.setresuid
- ruid: uid_t
- euid: uid_t
- suid: uid_t
- /
- Set the current process's real, effective, and saved user ids.
- [clinic start generated code]*/
- static PyObject *
- os_setresuid_impl(PyObject *module, uid_t ruid, uid_t euid, uid_t suid)
- /*[clinic end generated code: output=834a641e15373e97 input=9e33cb79a82792f3]*/
- {
- if (setresuid(ruid, euid, suid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETRESUID */
- #ifdef HAVE_SETRESGID
- /*[clinic input]
- os.setresgid
- rgid: gid_t
- egid: gid_t
- sgid: gid_t
- /
- Set the current process's real, effective, and saved group ids.
- [clinic start generated code]*/
- static PyObject *
- os_setresgid_impl(PyObject *module, gid_t rgid, gid_t egid, gid_t sgid)
- /*[clinic end generated code: output=6aa402f3d2e514a9 input=33e9e0785ef426b1]*/
- {
- if (setresgid(rgid, egid, sgid) < 0)
- return posix_error();
- Py_RETURN_NONE;
- }
- #endif /* HAVE_SETRESGID */
- #ifdef HAVE_GETRESUID
- /*[clinic input]
- os.getresuid
- Return a tuple of the current process's real, effective, and saved user ids.
- [clinic start generated code]*/
- static PyObject *
- os_getresuid_impl(PyObject *module)
- /*[clinic end generated code: output=8e0becff5dece5bf input=41ccfa8e1f6517ad]*/
- {
- uid_t ruid, euid, suid;
- if (getresuid(&ruid, &euid, &suid) < 0)
- return posix_error();
- return Py_BuildValue("(NNN)", _PyLong_FromUid(ruid),
- _PyLong_FromUid(euid),
- _PyLong_FromUid(suid));
- }
- #endif /* HAVE_GETRESUID */
- #ifdef HAVE_GETRESGID
- /*[clinic input]
- os.getresgid
- Return a tuple of the current process's real, effective, and saved group ids.
- [clinic start generated code]*/
- static PyObject *
- os_getresgid_impl(PyObject *module)
- /*[clinic end generated code: output=2719c4bfcf27fb9f input=517e68db9ca32df6]*/
- {
- gid_t rgid, egid, sgid;
- if (getresgid(&rgid, &egid, &sgid) < 0)
- return posix_error();
- return Py_BuildValue("(NNN)", _PyLong_FromGid(rgid),
- _PyLong_FromGid(egid),
- _PyLong_FromGid(sgid));
- }
- #endif /* HAVE_GETRESGID */
- #ifdef USE_XATTRS
- /*[clinic input]
- os.getxattr
- path: path_t(allow_fd=True)
- attribute: path_t
- *
- follow_symlinks: bool = True
- Return the value of extended attribute attribute on path.
- path may be either a string, a path-like object, or an open file descriptor.
- If follow_symlinks is False, and the last element of the path is a symbolic
- link, getxattr will examine the symbolic link itself instead of the file
- the link points to.
- [clinic start generated code]*/
- static PyObject *
- os_getxattr_impl(PyObject *module, path_t *path, path_t *attribute,
- int follow_symlinks)
- /*[clinic end generated code: output=5f2f44200a43cff2 input=025789491708f7eb]*/
- {
- Py_ssize_t i;
- PyObject *buffer = NULL;
- if (fd_and_follow_symlinks_invalid("getxattr", path->fd, follow_symlinks))
- return NULL;
- if (PySys_Audit("os.getxattr", "OO", path->object, attribute->object) < 0) {
- return NULL;
- }
- for (i = 0; ; i++) {
- void *ptr;
- ssize_t result;
- static const Py_ssize_t buffer_sizes[] = {128, XATTR_SIZE_MAX, 0};
- Py_ssize_t buffer_size = buffer_sizes[i];
- if (!buffer_size) {
- path_error(path);
- return NULL;
- }
- buffer = PyBytes_FromStringAndSize(NULL, buffer_size);
- if (!buffer)
- return NULL;
- ptr = PyBytes_AS_STRING(buffer);
- Py_BEGIN_ALLOW_THREADS;
- if (path->fd >= 0)
- result = fgetxattr(path->fd, attribute->narrow, ptr, buffer_size);
- else if (follow_symlinks)
- result = getxattr(path->narrow, attribute->narrow, ptr, buffer_size);
- else
- result = lgetxattr(path->narrow, attribute->narrow, ptr, buffer_size);
- Py_END_ALLOW_THREADS;
- if (result < 0) {
- if (errno == ERANGE) {
- Py_DECREF(buffer);
- continue;
- }
- path_error(path);
- Py_DECREF(buffer);
- return NULL;
- }
- if (result != buffer_size) {
- /* Can only shrink. */
- _PyBytes_Resize(&buffer, result);
- }
- break;
- }
- return buffer;
- }
- /*[clinic input]
- os.setxattr
- path: path_t(allow_fd=True)
- attribute: path_t
- value: Py_buffer
- flags: int = 0
- *
- follow_symlinks: bool = True
- Set extended attribute attribute on path to value.
- path may be either a string, a path-like object, or an open file descriptor.
- If follow_symlinks is False, and the last element of the path is a symbolic
- link, setxattr will modify the symbolic link itself instead of the file
- the link points to.
- [clinic start generated code]*/
- static PyObject *
- os_setxattr_impl(PyObject *module, path_t *path, path_t *attribute,
- Py_buffer *value, int flags, int follow_symlinks)
- /*[clinic end generated code: output=98b83f63fdde26bb input=c17c0103009042f0]*/
- {
- ssize_t result;
- if (fd_and_follow_symlinks_invalid("setxattr", path->fd, follow_symlinks))
- return NULL;
- if (PySys_Audit("os.setxattr", "OOy#i", path->object, attribute->object,
- value->buf, value->len, flags) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS;
- if (path->fd > -1)
- result = fsetxattr(path->fd, attribute->narrow,
- value->buf, value->len, flags);
- else if (follow_symlinks)
- result = setxattr(path->narrow, attribute->narrow,
- value->buf, value->len, flags);
- else
- result = lsetxattr(path->narrow, attribute->narrow,
- value->buf, value->len, flags);
- Py_END_ALLOW_THREADS;
- if (result) {
- path_error(path);
- return NULL;
- }
- Py_RETURN_NONE;
- }
- /*[clinic input]
- os.removexattr
- path: path_t(allow_fd=True)
- attribute: path_t
- *
- follow_symlinks: bool = True
- Remove extended attribute attribute on path.
- path may be either a string, a path-like object, or an open file descriptor.
- If follow_symlinks is False, and the last element of the path is a symbolic
- link, removexattr will modify the symbolic link itself instead of the file
- the link points to.
- [clinic start generated code]*/
- static PyObject *
- os_removexattr_impl(PyObject *module, path_t *path, path_t *attribute,
- int follow_symlinks)
- /*[clinic end generated code: output=521a51817980cda6 input=3d9a7d36fe2f7c4e]*/
- {
- ssize_t result;
- if (fd_and_follow_symlinks_invalid("removexattr", path->fd, follow_symlinks))
- return NULL;
- if (PySys_Audit("os.removexattr", "OO", path->object, attribute->object) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS;
- if (path->fd > -1)
- result = fremovexattr(path->fd, attribute->narrow);
- else if (follow_symlinks)
- result = removexattr(path->narrow, attribute->narrow);
- else
- result = lremovexattr(path->narrow, attribute->narrow);
- Py_END_ALLOW_THREADS;
- if (result) {
- return path_error(path);
- }
- Py_RETURN_NONE;
- }
- /*[clinic input]
- os.listxattr
- path: path_t(allow_fd=True, nullable=True) = None
- *
- follow_symlinks: bool = True
- Return a list of extended attributes on path.
- path may be either None, a string, a path-like object, or an open file descriptor.
- if path is None, listxattr will examine the current directory.
- If follow_symlinks is False, and the last element of the path is a symbolic
- link, listxattr will examine the symbolic link itself instead of the file
- the link points to.
- [clinic start generated code]*/
- static PyObject *
- os_listxattr_impl(PyObject *module, path_t *path, int follow_symlinks)
- /*[clinic end generated code: output=bebdb4e2ad0ce435 input=9826edf9fdb90869]*/
- {
- Py_ssize_t i;
- PyObject *result = NULL;
- const char *name;
- char *buffer = NULL;
- if (fd_and_follow_symlinks_invalid("listxattr", path->fd, follow_symlinks))
- goto exit;
- if (PySys_Audit("os.listxattr", "(O)",
- path->object ? path->object : Py_None) < 0) {
- return NULL;
- }
- name = path->narrow ? path->narrow : ".";
- for (i = 0; ; i++) {
- const char *start, *trace, *end;
- ssize_t length;
- static const Py_ssize_t buffer_sizes[] = { 256, XATTR_LIST_MAX, 0 };
- Py_ssize_t buffer_size = buffer_sizes[i];
- if (!buffer_size) {
- /* ERANGE */
- path_error(path);
- break;
- }
- buffer = PyMem_Malloc(buffer_size);
- if (!buffer) {
- PyErr_NoMemory();
- break;
- }
- Py_BEGIN_ALLOW_THREADS;
- if (path->fd > -1)
- length = flistxattr(path->fd, buffer, buffer_size);
- else if (follow_symlinks)
- length = listxattr(name, buffer, buffer_size);
- else
- length = llistxattr(name, buffer, buffer_size);
- Py_END_ALLOW_THREADS;
- if (length < 0) {
- if (errno == ERANGE) {
- PyMem_Free(buffer);
- buffer = NULL;
- continue;
- }
- path_error(path);
- break;
- }
- result = PyList_New(0);
- if (!result) {
- goto exit;
- }
- end = buffer + length;
- for (trace = start = buffer; trace != end; trace++) {
- if (!*trace) {
- int error;
- PyObject *attribute = PyUnicode_DecodeFSDefaultAndSize(start,
- trace - start);
- if (!attribute) {
- Py_SETREF(result, NULL);
- goto exit;
- }
- error = PyList_Append(result, attribute);
- Py_DECREF(attribute);
- if (error) {
- Py_SETREF(result, NULL);
- goto exit;
- }
- start = trace + 1;
- }
- }
- break;
- }
- exit:
- if (buffer)
- PyMem_Free(buffer);
- return result;
- }
- #endif /* USE_XATTRS */
- /*[clinic input]
- os.urandom
- size: Py_ssize_t
- /
- Return a bytes object containing random bytes suitable for cryptographic use.
- [clinic start generated code]*/
- static PyObject *
- os_urandom_impl(PyObject *module, Py_ssize_t size)
- /*[clinic end generated code: output=42c5cca9d18068e9 input=4067cdb1b6776c29]*/
- {
- PyObject *bytes;
- int result;
- if (size < 0)
- return PyErr_Format(PyExc_ValueError,
- "negative argument not allowed");
- bytes = PyBytes_FromStringAndSize(NULL, size);
- if (bytes == NULL)
- return NULL;
- result = _PyOS_URandom(PyBytes_AS_STRING(bytes), PyBytes_GET_SIZE(bytes));
- if (result == -1) {
- Py_DECREF(bytes);
- return NULL;
- }
- return bytes;
- }
- #ifdef HAVE_MEMFD_CREATE
- /*[clinic input]
- os.memfd_create
- name: FSConverter
- flags: unsigned_int(bitwise=True, c_default="MFD_CLOEXEC") = MFD_CLOEXEC
- [clinic start generated code]*/
- static PyObject *
- os_memfd_create_impl(PyObject *module, PyObject *name, unsigned int flags)
- /*[clinic end generated code: output=6681ede983bdb9a6 input=a42cfc199bcd56e9]*/
- {
- int fd;
- const char *bytes = PyBytes_AS_STRING(name);
- Py_BEGIN_ALLOW_THREADS
- fd = memfd_create(bytes, flags);
- Py_END_ALLOW_THREADS
- if (fd == -1) {
- return PyErr_SetFromErrno(PyExc_OSError);
- }
- return PyLong_FromLong(fd);
- }
- #endif
- #if defined(HAVE_EVENTFD) && defined(EFD_CLOEXEC)
- /*[clinic input]
- os.eventfd
- initval: unsigned_int
- flags: int(c_default="EFD_CLOEXEC") = EFD_CLOEXEC
- Creates and returns an event notification file descriptor.
- [clinic start generated code]*/
- static PyObject *
- os_eventfd_impl(PyObject *module, unsigned int initval, int flags)
- /*[clinic end generated code: output=ce9c9bbd1446f2de input=66203e3c50c4028b]*/
- {
- /* initval is limited to uint32_t, internal counter is uint64_t */
- int fd;
- Py_BEGIN_ALLOW_THREADS
- fd = eventfd(initval, flags);
- Py_END_ALLOW_THREADS
- if (fd == -1) {
- return PyErr_SetFromErrno(PyExc_OSError);
- }
- return PyLong_FromLong(fd);
- }
- /*[clinic input]
- os.eventfd_read
- fd: fildes
- Read eventfd value
- [clinic start generated code]*/
- static PyObject *
- os_eventfd_read_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=8f2c7b59a3521fd1 input=110f8b57fa596afe]*/
- {
- eventfd_t value;
- int result;
- Py_BEGIN_ALLOW_THREADS
- result = eventfd_read(fd, &value);
- Py_END_ALLOW_THREADS
- if (result == -1) {
- return PyErr_SetFromErrno(PyExc_OSError);
- }
- return PyLong_FromUnsignedLongLong(value);
- }
- /*[clinic input]
- os.eventfd_write
- fd: fildes
- value: unsigned_long_long
- Write eventfd value.
- [clinic start generated code]*/
- static PyObject *
- os_eventfd_write_impl(PyObject *module, int fd, unsigned long long value)
- /*[clinic end generated code: output=bebd9040bbf987f5 input=156de8555be5a949]*/
- {
- int result;
- Py_BEGIN_ALLOW_THREADS
- result = eventfd_write(fd, value);
- Py_END_ALLOW_THREADS
- if (result == -1) {
- return PyErr_SetFromErrno(PyExc_OSError);
- }
- Py_RETURN_NONE;
- }
- #endif /* HAVE_EVENTFD && EFD_CLOEXEC */
- /* Terminal size querying */
- PyDoc_STRVAR(TerminalSize_docstring,
- "A tuple of (columns, lines) for holding terminal window size");
- static PyStructSequence_Field TerminalSize_fields[] = {
- {"columns", "width of the terminal window in characters"},
- {"lines", "height of the terminal window in characters"},
- {NULL, NULL}
- };
- static PyStructSequence_Desc TerminalSize_desc = {
- "os.terminal_size",
- TerminalSize_docstring,
- TerminalSize_fields,
- 2,
- };
- #if defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL)
- /*[clinic input]
- os.get_terminal_size
- fd: int(c_default="fileno(stdout)", py_default="<unrepresentable>") = -1
- /
- Return the size of the terminal window as (columns, lines).
- The optional argument fd (default standard output) specifies
- which file descriptor should be queried.
- If the file descriptor is not connected to a terminal, an OSError
- is thrown.
- This function will only be defined if an implementation is
- available for this system.
- shutil.get_terminal_size is the high-level function which should
- normally be used, os.get_terminal_size is the low-level implementation.
- [clinic start generated code]*/
- static PyObject *
- os_get_terminal_size_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=fbab93acef980508 input=ead5679b82ddb920]*/
- {
- int columns, lines;
- PyObject *termsize;
- /* Under some conditions stdout may not be connected and
- * fileno(stdout) may point to an invalid file descriptor. For example
- * GUI apps don't have valid standard streams by default.
- *
- * If this happens, and the optional fd argument is not present,
- * the ioctl below will fail returning EBADF. This is what we want.
- */
- #ifdef TERMSIZE_USE_IOCTL
- {
- struct winsize w;
- if (ioctl(fd, TIOCGWINSZ, &w))
- return PyErr_SetFromErrno(PyExc_OSError);
- columns = w.ws_col;
- lines = w.ws_row;
- }
- #endif /* TERMSIZE_USE_IOCTL */
- #ifdef TERMSIZE_USE_CONIO
- {
- HANDLE handle;
- CONSOLE_SCREEN_BUFFER_INFO csbi;
- handle = _Py_get_osfhandle(fd);
- if (handle == INVALID_HANDLE_VALUE)
- return NULL;
- if (!GetConsoleScreenBufferInfo(handle, &csbi))
- return PyErr_SetFromWindowsErr(0);
- columns = csbi.srWindow.Right - csbi.srWindow.Left + 1;
- lines = csbi.srWindow.Bottom - csbi.srWindow.Top + 1;
- }
- #endif /* TERMSIZE_USE_CONIO */
- PyObject *TerminalSizeType = get_posix_state(module)->TerminalSizeType;
- termsize = PyStructSequence_New((PyTypeObject *)TerminalSizeType);
- if (termsize == NULL)
- return NULL;
- int pos = 0;
- #define SET_TERMSIZE(CALL) \
- do { \
- PyObject *item = (CALL); \
- if (item == NULL) { \
- Py_DECREF(termsize); \
- return NULL; \
- } \
- PyStructSequence_SET_ITEM(termsize, pos++, item); \
- } while(0)
- SET_TERMSIZE(PyLong_FromLong(columns));
- SET_TERMSIZE(PyLong_FromLong(lines));
- #undef SET_TERMSIZE
- return termsize;
- }
- #endif /* defined(TERMSIZE_USE_CONIO) || defined(TERMSIZE_USE_IOCTL) */
- /*[clinic input]
- os.cpu_count
- Return the number of CPUs in the system; return None if indeterminable.
- This number is not equivalent to the number of CPUs the current process can
- use. The number of usable CPUs can be obtained with
- ``len(os.sched_getaffinity(0))``
- [clinic start generated code]*/
- static PyObject *
- os_cpu_count_impl(PyObject *module)
- /*[clinic end generated code: output=5fc29463c3936a9c input=e7c8f4ba6dbbadd3]*/
- {
- int ncpu = 0;
- #ifdef MS_WINDOWS
- #ifdef MS_WINDOWS_DESKTOP
- ncpu = GetActiveProcessorCount(ALL_PROCESSOR_GROUPS);
- #endif
- #elif defined(__hpux)
- ncpu = mpctl(MPC_GETNUMSPUS, NULL, NULL);
- #elif defined(HAVE_SYSCONF) && defined(_SC_NPROCESSORS_ONLN)
- ncpu = sysconf(_SC_NPROCESSORS_ONLN);
- #elif defined(__VXWORKS__)
- ncpu = _Py_popcount32(vxCpuEnabledGet());
- #elif defined(__DragonFly__) || \
- defined(__OpenBSD__) || \
- defined(__FreeBSD__) || \
- defined(__NetBSD__) || \
- defined(__APPLE__)
- int mib[2];
- size_t len = sizeof(ncpu);
- mib[0] = CTL_HW;
- mib[1] = HW_NCPU;
- if (sysctl(mib, 2, &ncpu, &len, NULL, 0) != 0)
- ncpu = 0;
- #endif
- if (ncpu >= 1)
- return PyLong_FromLong(ncpu);
- else
- Py_RETURN_NONE;
- }
- /*[clinic input]
- os.get_inheritable -> bool
- fd: int
- /
- Get the close-on-exe flag of the specified file descriptor.
- [clinic start generated code]*/
- static int
- os_get_inheritable_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=0445e20e149aa5b8 input=89ac008dc9ab6b95]*/
- {
- int return_value;
- _Py_BEGIN_SUPPRESS_IPH
- return_value = _Py_get_inheritable(fd);
- _Py_END_SUPPRESS_IPH
- return return_value;
- }
- /*[clinic input]
- os.set_inheritable
- fd: int
- inheritable: int
- /
- Set the inheritable flag of the specified file descriptor.
- [clinic start generated code]*/
- static PyObject *
- os_set_inheritable_impl(PyObject *module, int fd, int inheritable)
- /*[clinic end generated code: output=f1b1918a2f3c38c2 input=9ceaead87a1e2402]*/
- {
- int result;
- _Py_BEGIN_SUPPRESS_IPH
- result = _Py_set_inheritable(fd, inheritable, NULL);
- _Py_END_SUPPRESS_IPH
- if (result < 0)
- return NULL;
- Py_RETURN_NONE;
- }
- #ifdef MS_WINDOWS
- #ifndef HANDLE_FLAG_INHERIT
- #define HANDLE_FLAG_INHERIT 0x00000001
- #endif
- /*[clinic input]
- os.get_handle_inheritable -> bool
- handle: intptr_t
- /
- Get the close-on-exe flag of the specified file descriptor.
- [clinic start generated code]*/
- static int
- os_get_handle_inheritable_impl(PyObject *module, intptr_t handle)
- /*[clinic end generated code: output=36be5afca6ea84d8 input=cfe99f9c05c70ad1]*/
- {
- DWORD flags;
- if (!GetHandleInformation((HANDLE)handle, &flags)) {
- PyErr_SetFromWindowsErr(0);
- return -1;
- }
- return flags & HANDLE_FLAG_INHERIT;
- }
- /*[clinic input]
- os.set_handle_inheritable
- handle: intptr_t
- inheritable: bool
- /
- Set the inheritable flag of the specified handle.
- [clinic start generated code]*/
- static PyObject *
- os_set_handle_inheritable_impl(PyObject *module, intptr_t handle,
- int inheritable)
- /*[clinic end generated code: output=021d74fe6c96baa3 input=7a7641390d8364fc]*/
- {
- DWORD flags = inheritable ? HANDLE_FLAG_INHERIT : 0;
- if (!SetHandleInformation((HANDLE)handle, HANDLE_FLAG_INHERIT, flags)) {
- PyErr_SetFromWindowsErr(0);
- return NULL;
- }
- Py_RETURN_NONE;
- }
- #endif /* MS_WINDOWS */
- /*[clinic input]
- os.get_blocking -> bool
- fd: int
- /
- Get the blocking mode of the file descriptor.
- Return False if the O_NONBLOCK flag is set, True if the flag is cleared.
- [clinic start generated code]*/
- static int
- os_get_blocking_impl(PyObject *module, int fd)
- /*[clinic end generated code: output=336a12ad76a61482 input=f4afb59d51560179]*/
- {
- int blocking;
- _Py_BEGIN_SUPPRESS_IPH
- blocking = _Py_get_blocking(fd);
- _Py_END_SUPPRESS_IPH
- return blocking;
- }
- /*[clinic input]
- os.set_blocking
- fd: int
- blocking: bool
- /
- Set the blocking mode of the specified file descriptor.
- Set the O_NONBLOCK flag if blocking is False,
- clear the O_NONBLOCK flag otherwise.
- [clinic start generated code]*/
- static PyObject *
- os_set_blocking_impl(PyObject *module, int fd, int blocking)
- /*[clinic end generated code: output=384eb43aa0762a9d input=7e9dfc9b14804dd4]*/
- {
- int result;
- _Py_BEGIN_SUPPRESS_IPH
- result = _Py_set_blocking(fd, blocking);
- _Py_END_SUPPRESS_IPH
- if (result < 0)
- return NULL;
- Py_RETURN_NONE;
- }
- /*[clinic input]
- class os.DirEntry "DirEntry *" "DirEntryType"
- [clinic start generated code]*/
- /*[clinic end generated code: output=da39a3ee5e6b4b0d input=3c18c7a448247980]*/
- typedef struct {
- PyObject_HEAD
- PyObject *name;
- PyObject *path;
- PyObject *stat;
- PyObject *lstat;
- #ifdef MS_WINDOWS
- struct _Py_stat_struct win32_lstat;
- uint64_t win32_file_index;
- uint64_t win32_file_index_high;
- int got_file_index;
- #else /* POSIX */
- #ifdef HAVE_DIRENT_D_TYPE
- unsigned char d_type;
- #endif
- ino_t d_ino;
- int dir_fd;
- #endif
- } DirEntry;
- static void
- DirEntry_dealloc(DirEntry *entry)
- {
- PyTypeObject *tp = Py_TYPE(entry);
- Py_XDECREF(entry->name);
- Py_XDECREF(entry->path);
- Py_XDECREF(entry->stat);
- Py_XDECREF(entry->lstat);
- freefunc free_func = PyType_GetSlot(tp, Py_tp_free);
- free_func(entry);
- Py_DECREF(tp);
- }
- /* Forward reference */
- static int
- DirEntry_test_mode(PyTypeObject *defining_class, DirEntry *self,
- int follow_symlinks, unsigned short mode_bits);
- /*[clinic input]
- os.DirEntry.is_symlink -> bool
- defining_class: defining_class
- /
- Return True if the entry is a symbolic link; cached per entry.
- [clinic start generated code]*/
- static int
- os_DirEntry_is_symlink_impl(DirEntry *self, PyTypeObject *defining_class)
- /*[clinic end generated code: output=293096d589b6d47c input=e9acc5ee4d511113]*/
- {
- #ifdef MS_WINDOWS
- return (self->win32_lstat.st_mode & S_IFMT) == S_IFLNK;
- #elif defined(HAVE_DIRENT_D_TYPE)
- /* POSIX */
- if (self->d_type != DT_UNKNOWN)
- return self->d_type == DT_LNK;
- else
- return DirEntry_test_mode(defining_class, self, 0, S_IFLNK);
- #else
- /* POSIX without d_type */
- return DirEntry_test_mode(defining_class, self, 0, S_IFLNK);
- #endif
- }
- /*[clinic input]
- os.DirEntry.is_junction -> bool
- defining_class: defining_class
- /
- Return True if the entry is a junction; cached per entry.
- [clinic start generated code]*/
- static int
- os_DirEntry_is_junction_impl(DirEntry *self, PyTypeObject *defining_class)
- /*[clinic end generated code: output=7061a07b0ef2cd1f input=475cd36fb7d4723f]*/
- {
- #ifdef MS_WINDOWS
- return self->win32_lstat.st_reparse_tag == IO_REPARSE_TAG_MOUNT_POINT;
- #else
- return 0;
- #endif
- }
- static PyObject *
- DirEntry_fetch_stat(PyObject *module, DirEntry *self, int follow_symlinks)
- {
- int result;
- STRUCT_STAT st;
- PyObject *ub;
- #ifdef MS_WINDOWS
- if (!PyUnicode_FSDecoder(self->path, &ub))
- return NULL;
- wchar_t *path = PyUnicode_AsWideCharString(ub, NULL);
- Py_DECREF(ub);
- #else /* POSIX */
- if (!PyUnicode_FSConverter(self->path, &ub))
- return NULL;
- const char *path = PyBytes_AS_STRING(ub);
- if (self->dir_fd != DEFAULT_DIR_FD) {
- #ifdef HAVE_FSTATAT
- if (HAVE_FSTATAT_RUNTIME) {
- Py_BEGIN_ALLOW_THREADS
- result = fstatat(self->dir_fd, path, &st,
- follow_symlinks ? 0 : AT_SYMLINK_NOFOLLOW);
- Py_END_ALLOW_THREADS
- } else
- #endif /* HAVE_FSTATAT */
- {
- Py_DECREF(ub);
- PyErr_SetString(PyExc_NotImplementedError, "can't fetch stat");
- return NULL;
- }
- }
- else
- #endif
- {
- Py_BEGIN_ALLOW_THREADS
- if (follow_symlinks) {
- result = STAT(path, &st);
- }
- else {
- result = LSTAT(path, &st);
- }
- Py_END_ALLOW_THREADS
- }
- int saved_errno = errno;
- #if defined(MS_WINDOWS)
- PyMem_Free(path);
- #else
- Py_DECREF(ub);
- #endif
- if (result != 0) {
- errno = saved_errno;
- path_object_error(self->path);
- return NULL;
- }
- return _pystat_fromstructstat(module, &st);
- }
- static PyObject *
- DirEntry_get_lstat(PyTypeObject *defining_class, DirEntry *self)
- {
- if (!self->lstat) {
- PyObject *module = PyType_GetModule(defining_class);
- #ifdef MS_WINDOWS
- self->lstat = _pystat_fromstructstat(module, &self->win32_lstat);
- #else /* POSIX */
- self->lstat = DirEntry_fetch_stat(module, self, 0);
- #endif
- }
- return Py_XNewRef(self->lstat);
- }
- /*[clinic input]
- os.DirEntry.stat
- defining_class: defining_class
- /
- *
- follow_symlinks: bool = True
- Return stat_result object for the entry; cached per entry.
- [clinic start generated code]*/
- static PyObject *
- os_DirEntry_stat_impl(DirEntry *self, PyTypeObject *defining_class,
- int follow_symlinks)
- /*[clinic end generated code: output=23f803e19c3e780e input=e816273c4e67ee98]*/
- {
- if (!follow_symlinks) {
- return DirEntry_get_lstat(defining_class, self);
- }
- if (!self->stat) {
- int result = os_DirEntry_is_symlink_impl(self, defining_class);
- if (result == -1) {
- return NULL;
- }
- if (result) {
- PyObject *module = PyType_GetModule(defining_class);
- self->stat = DirEntry_fetch_stat(module, self, 1);
- }
- else {
- self->stat = DirEntry_get_lstat(defining_class, self);
- }
- }
- return Py_XNewRef(self->stat);
- }
- /* Set exception and return -1 on error, 0 for False, 1 for True */
- static int
- DirEntry_test_mode(PyTypeObject *defining_class, DirEntry *self,
- int follow_symlinks, unsigned short mode_bits)
- {
- PyObject *stat = NULL;
- PyObject *st_mode = NULL;
- long mode;
- int result;
- #if defined(MS_WINDOWS) || defined(HAVE_DIRENT_D_TYPE)
- int is_symlink;
- int need_stat;
- #endif
- #ifdef MS_WINDOWS
- unsigned long dir_bits;
- #endif
- #ifdef MS_WINDOWS
- is_symlink = (self->win32_lstat.st_mode & S_IFMT) == S_IFLNK;
- need_stat = follow_symlinks && is_symlink;
- #elif defined(HAVE_DIRENT_D_TYPE)
- is_symlink = self->d_type == DT_LNK;
- need_stat = self->d_type == DT_UNKNOWN || (follow_symlinks && is_symlink);
- #endif
- #if defined(MS_WINDOWS) || defined(HAVE_DIRENT_D_TYPE)
- if (need_stat) {
- #endif
- stat = os_DirEntry_stat_impl(self, defining_class, follow_symlinks);
- if (!stat) {
- if (PyErr_ExceptionMatches(PyExc_FileNotFoundError)) {
- /* If file doesn't exist (anymore), then return False
- (i.e., say it's not a file/directory) */
- PyErr_Clear();
- return 0;
- }
- goto error;
- }
- _posixstate* state = get_posix_state(PyType_GetModule(defining_class));
- st_mode = PyObject_GetAttr(stat, state->st_mode);
- if (!st_mode)
- goto error;
- mode = PyLong_AsLong(st_mode);
- if (mode == -1 && PyErr_Occurred())
- goto error;
- Py_CLEAR(st_mode);
- Py_CLEAR(stat);
- result = (mode & S_IFMT) == mode_bits;
- #if defined(MS_WINDOWS) || defined(HAVE_DIRENT_D_TYPE)
- }
- else if (is_symlink) {
- assert(mode_bits != S_IFLNK);
- result = 0;
- }
- else {
- assert(mode_bits == S_IFDIR || mode_bits == S_IFREG);
- #ifdef MS_WINDOWS
- dir_bits = self->win32_lstat.st_file_attributes & FILE_ATTRIBUTE_DIRECTORY;
- if (mode_bits == S_IFDIR)
- result = dir_bits != 0;
- else
- result = dir_bits == 0;
- #else /* POSIX */
- if (mode_bits == S_IFDIR)
- result = self->d_type == DT_DIR;
- else
- result = self->d_type == DT_REG;
- #endif
- }
- #endif
- return result;
- error:
- Py_XDECREF(st_mode);
- Py_XDECREF(stat);
- return -1;
- }
- /*[clinic input]
- os.DirEntry.is_dir -> bool
- defining_class: defining_class
- /
- *
- follow_symlinks: bool = True
- Return True if the entry is a directory; cached per entry.
- [clinic start generated code]*/
- static int
- os_DirEntry_is_dir_impl(DirEntry *self, PyTypeObject *defining_class,
- int follow_symlinks)
- /*[clinic end generated code: output=0cd453b9c0987fdf input=1a4ffd6dec9920cb]*/
- {
- return DirEntry_test_mode(defining_class, self, follow_symlinks, S_IFDIR);
- }
- /*[clinic input]
- os.DirEntry.is_file -> bool
- defining_class: defining_class
- /
- *
- follow_symlinks: bool = True
- Return True if the entry is a file; cached per entry.
- [clinic start generated code]*/
- static int
- os_DirEntry_is_file_impl(DirEntry *self, PyTypeObject *defining_class,
- int follow_symlinks)
- /*[clinic end generated code: output=f7c277ab5ba80908 input=0a64c5a12e802e3b]*/
- {
- return DirEntry_test_mode(defining_class, self, follow_symlinks, S_IFREG);
- }
- /*[clinic input]
- os.DirEntry.inode
- Return inode of the entry; cached per entry.
- [clinic start generated code]*/
- static PyObject *
- os_DirEntry_inode_impl(DirEntry *self)
- /*[clinic end generated code: output=156bb3a72162440e input=3ee7b872ae8649f0]*/
- {
- #ifdef MS_WINDOWS
- if (!self->got_file_index) {
- PyObject *unicode;
- STRUCT_STAT stat;
- int result;
- if (!PyUnicode_FSDecoder(self->path, &unicode))
- return NULL;
- wchar_t *path = PyUnicode_AsWideCharString(unicode, NULL);
- Py_DECREF(unicode);
- result = LSTAT(path, &stat);
- int saved_errno = errno;
- PyMem_Free(path);
- if (result != 0) {
- errno = saved_errno;
- return path_object_error(self->path);
- }
- self->win32_file_index = stat.st_ino;
- self->win32_file_index_high = stat.st_ino_high;
- self->got_file_index = 1;
- }
- return _pystat_l128_from_l64_l64(self->win32_file_index, self->win32_file_index_high);
- #else /* POSIX */
- static_assert(sizeof(unsigned long long) >= sizeof(self->d_ino),
- "DirEntry.d_ino is larger than unsigned long long");
- return PyLong_FromUnsignedLongLong(self->d_ino);
- #endif
- }
- static PyObject *
- DirEntry_repr(DirEntry *self)
- {
- return PyUnicode_FromFormat("<DirEntry %R>", self->name);
- }
- /*[clinic input]
- os.DirEntry.__fspath__
- Returns the path for the entry.
- [clinic start generated code]*/
- static PyObject *
- os_DirEntry___fspath___impl(DirEntry *self)
- /*[clinic end generated code: output=6dd7f7ef752e6f4f input=3c49d0cf38df4fac]*/
- {
- return Py_NewRef(self->path);
- }
- static PyMemberDef DirEntry_members[] = {
- {"name", T_OBJECT_EX, offsetof(DirEntry, name), READONLY,
- "the entry's base filename, relative to scandir() \"path\" argument"},
- {"path", T_OBJECT_EX, offsetof(DirEntry, path), READONLY,
- "the entry's full path name; equivalent to os.path.join(scandir_path, entry.name)"},
- {NULL}
- };
- #include "clinic/posixmodule.c.h"
- static PyMethodDef DirEntry_methods[] = {
- OS_DIRENTRY_IS_DIR_METHODDEF
- OS_DIRENTRY_IS_FILE_METHODDEF
- OS_DIRENTRY_IS_SYMLINK_METHODDEF
- OS_DIRENTRY_IS_JUNCTION_METHODDEF
- OS_DIRENTRY_STAT_METHODDEF
- OS_DIRENTRY_INODE_METHODDEF
- OS_DIRENTRY___FSPATH___METHODDEF
- {"__class_getitem__", Py_GenericAlias,
- METH_O|METH_CLASS, PyDoc_STR("See PEP 585")},
- {NULL}
- };
- static PyType_Slot DirEntryType_slots[] = {
- {Py_tp_dealloc, DirEntry_dealloc},
- {Py_tp_repr, DirEntry_repr},
- {Py_tp_methods, DirEntry_methods},
- {Py_tp_members, DirEntry_members},
- {0, 0},
- };
- static PyType_Spec DirEntryType_spec = {
- MODNAME ".DirEntry",
- sizeof(DirEntry),
- 0,
- Py_TPFLAGS_DEFAULT | Py_TPFLAGS_DISALLOW_INSTANTIATION,
- DirEntryType_slots
- };
- #ifdef MS_WINDOWS
- static wchar_t *
- join_path_filenameW(const wchar_t *path_wide, const wchar_t *filename)
- {
- Py_ssize_t path_len;
- Py_ssize_t size;
- wchar_t *result;
- wchar_t ch;
- if (!path_wide) { /* Default arg: "." */
- path_wide = L".";
- path_len = 1;
- }
- else {
- path_len = wcslen(path_wide);
- }
- /* The +1's are for the path separator and the NUL */
- size = path_len + 1 + wcslen(filename) + 1;
- result = PyMem_New(wchar_t, size);
- if (!result) {
- PyErr_NoMemory();
- return NULL;
- }
- wcscpy(result, path_wide);
- if (path_len > 0) {
- ch = result[path_len - 1];
- if (ch != SEP && ch != ALTSEP && ch != L':')
- result[path_len++] = SEP;
- wcscpy(result + path_len, filename);
- }
- return result;
- }
- static PyObject *
- DirEntry_from_find_data(PyObject *module, path_t *path, WIN32_FIND_DATAW *dataW)
- {
- DirEntry *entry;
- BY_HANDLE_FILE_INFORMATION file_info;
- ULONG reparse_tag;
- wchar_t *joined_path;
- PyObject *DirEntryType = get_posix_state(module)->DirEntryType;
- entry = PyObject_New(DirEntry, (PyTypeObject *)DirEntryType);
- if (!entry)
- return NULL;
- entry->name = NULL;
- entry->path = NULL;
- entry->stat = NULL;
- entry->lstat = NULL;
- entry->got_file_index = 0;
- entry->name = PyUnicode_FromWideChar(dataW->cFileName, -1);
- if (!entry->name)
- goto error;
- int return_bytes = path->wide && PyBytes_Check(path->object);
- if (return_bytes) {
- Py_SETREF(entry->name, PyUnicode_EncodeFSDefault(entry->name));
- if (!entry->name)
- goto error;
- }
- joined_path = join_path_filenameW(path->wide, dataW->cFileName);
- if (!joined_path)
- goto error;
- entry->path = PyUnicode_FromWideChar(joined_path, -1);
- PyMem_Free(joined_path);
- if (!entry->path)
- goto error;
- if (return_bytes) {
- Py_SETREF(entry->path, PyUnicode_EncodeFSDefault(entry->path));
- if (!entry->path)
- goto error;
- }
- find_data_to_file_info(dataW, &file_info, &reparse_tag);
- _Py_attribute_data_to_stat(&file_info, reparse_tag, NULL, NULL, &entry->win32_lstat);
- /* ctime is only deprecated from 3.12, so we copy birthtime across */
- entry->win32_lstat.st_ctime = entry->win32_lstat.st_birthtime;
- entry->win32_lstat.st_ctime_nsec = entry->win32_lstat.st_birthtime_nsec;
- return (PyObject *)entry;
- error:
- Py_DECREF(entry);
- return NULL;
- }
- #else /* POSIX */
- static char *
- join_path_filename(const char *path_narrow, const char* filename, Py_ssize_t filename_len)
- {
- Py_ssize_t path_len;
- Py_ssize_t size;
- char *result;
- if (!path_narrow) { /* Default arg: "." */
- path_narrow = ".";
- path_len = 1;
- }
- else {
- path_len = strlen(path_narrow);
- }
- if (filename_len == -1)
- filename_len = strlen(filename);
- /* The +1's are for the path separator and the NUL */
- size = path_len + 1 + filename_len + 1;
- result = PyMem_New(char, size);
- if (!result) {
- PyErr_NoMemory();
- return NULL;
- }
- strcpy(result, path_narrow);
- if (path_len > 0 && result[path_len - 1] != '/')
- result[path_len++] = '/';
- strcpy(result + path_len, filename);
- return result;
- }
- static PyObject *
- DirEntry_from_posix_info(PyObject *module, path_t *path, const char *name,
- Py_ssize_t name_len, ino_t d_ino
- #ifdef HAVE_DIRENT_D_TYPE
- , unsigned char d_type
- #endif
- )
- {
- DirEntry *entry;
- char *joined_path;
- PyObject *DirEntryType = get_posix_state(module)->DirEntryType;
- entry = PyObject_New(DirEntry, (PyTypeObject *)DirEntryType);
- if (!entry)
- return NULL;
- entry->name = NULL;
- entry->path = NULL;
- entry->stat = NULL;
- entry->lstat = NULL;
- if (path->fd != -1) {
- entry->dir_fd = path->fd;
- joined_path = NULL;
- }
- else {
- entry->dir_fd = DEFAULT_DIR_FD;
- joined_path = join_path_filename(path->narrow, name, name_len);
- if (!joined_path)
- goto error;
- }
- if (!path->narrow || !PyBytes_Check(path->object)) {
- entry->name = PyUnicode_DecodeFSDefaultAndSize(name, name_len);
- if (joined_path)
- entry->path = PyUnicode_DecodeFSDefault(joined_path);
- }
- else {
- entry->name = PyBytes_FromStringAndSize(name, name_len);
- if (joined_path)
- entry->path = PyBytes_FromString(joined_path);
- }
- PyMem_Free(joined_path);
- if (!entry->name)
- goto error;
- if (path->fd != -1) {
- entry->path = Py_NewRef(entry->name);
- }
- else if (!entry->path)
- goto error;
- #ifdef HAVE_DIRENT_D_TYPE
- entry->d_type = d_type;
- #endif
- entry->d_ino = d_ino;
- return (PyObject *)entry;
- error:
- Py_XDECREF(entry);
- return NULL;
- }
- #endif
- typedef struct {
- PyObject_HEAD
- path_t path;
- #ifdef MS_WINDOWS
- HANDLE handle;
- WIN32_FIND_DATAW file_data;
- int first_time;
- #else /* POSIX */
- DIR *dirp;
- #endif
- #ifdef HAVE_FDOPENDIR
- int fd;
- #endif
- } ScandirIterator;
- #ifdef MS_WINDOWS
- static int
- ScandirIterator_is_closed(ScandirIterator *iterator)
- {
- return iterator->handle == INVALID_HANDLE_VALUE;
- }
- static void
- ScandirIterator_closedir(ScandirIterator *iterator)
- {
- HANDLE handle = iterator->handle;
- if (handle == INVALID_HANDLE_VALUE)
- return;
- iterator->handle = INVALID_HANDLE_VALUE;
- Py_BEGIN_ALLOW_THREADS
- FindClose(handle);
- Py_END_ALLOW_THREADS
- }
- static PyObject *
- ScandirIterator_iternext(ScandirIterator *iterator)
- {
- WIN32_FIND_DATAW *file_data = &iterator->file_data;
- BOOL success;
- PyObject *entry;
- /* Happens if the iterator is iterated twice, or closed explicitly */
- if (iterator->handle == INVALID_HANDLE_VALUE)
- return NULL;
- while (1) {
- if (!iterator->first_time) {
- Py_BEGIN_ALLOW_THREADS
- success = FindNextFileW(iterator->handle, file_data);
- Py_END_ALLOW_THREADS
- if (!success) {
- /* Error or no more files */
- if (GetLastError() != ERROR_NO_MORE_FILES)
- path_error(&iterator->path);
- break;
- }
- }
- iterator->first_time = 0;
- /* Skip over . and .. */
- if (wcscmp(file_data->cFileName, L".") != 0 &&
- wcscmp(file_data->cFileName, L"..") != 0)
- {
- PyObject *module = PyType_GetModule(Py_TYPE(iterator));
- entry = DirEntry_from_find_data(module, &iterator->path, file_data);
- if (!entry)
- break;
- return entry;
- }
- /* Loop till we get a non-dot directory or finish iterating */
- }
- /* Error or no more files */
- ScandirIterator_closedir(iterator);
- return NULL;
- }
- #else /* POSIX */
- static int
- ScandirIterator_is_closed(ScandirIterator *iterator)
- {
- return !iterator->dirp;
- }
- static void
- ScandirIterator_closedir(ScandirIterator *iterator)
- {
- DIR *dirp = iterator->dirp;
- if (!dirp)
- return;
- iterator->dirp = NULL;
- Py_BEGIN_ALLOW_THREADS
- #ifdef HAVE_FDOPENDIR
- if (iterator->path.fd != -1)
- rewinddir(dirp);
- #endif
- closedir(dirp);
- Py_END_ALLOW_THREADS
- return;
- }
- static PyObject *
- ScandirIterator_iternext(ScandirIterator *iterator)
- {
- struct dirent *direntp;
- Py_ssize_t name_len;
- int is_dot;
- PyObject *entry;
- /* Happens if the iterator is iterated twice, or closed explicitly */
- if (!iterator->dirp)
- return NULL;
- while (1) {
- errno = 0;
- Py_BEGIN_ALLOW_THREADS
- direntp = readdir(iterator->dirp);
- Py_END_ALLOW_THREADS
- if (!direntp) {
- /* Error or no more files */
- if (errno != 0)
- path_error(&iterator->path);
- break;
- }
- /* Skip over . and .. */
- name_len = NAMLEN(direntp);
- is_dot = direntp->d_name[0] == '.' &&
- (name_len == 1 || (direntp->d_name[1] == '.' && name_len == 2));
- if (!is_dot) {
- PyObject *module = PyType_GetModule(Py_TYPE(iterator));
- entry = DirEntry_from_posix_info(module,
- &iterator->path, direntp->d_name,
- name_len, direntp->d_ino
- #ifdef HAVE_DIRENT_D_TYPE
- , direntp->d_type
- #endif
- );
- if (!entry)
- break;
- return entry;
- }
- /* Loop till we get a non-dot directory or finish iterating */
- }
- /* Error or no more files */
- ScandirIterator_closedir(iterator);
- return NULL;
- }
- #endif
- static PyObject *
- ScandirIterator_close(ScandirIterator *self, PyObject *args)
- {
- ScandirIterator_closedir(self);
- Py_RETURN_NONE;
- }
- static PyObject *
- ScandirIterator_enter(PyObject *self, PyObject *args)
- {
- return Py_NewRef(self);
- }
- static PyObject *
- ScandirIterator_exit(ScandirIterator *self, PyObject *args)
- {
- ScandirIterator_closedir(self);
- Py_RETURN_NONE;
- }
- static void
- ScandirIterator_finalize(ScandirIterator *iterator)
- {
- /* Save the current exception, if any. */
- PyObject *exc = PyErr_GetRaisedException();
- if (!ScandirIterator_is_closed(iterator)) {
- ScandirIterator_closedir(iterator);
- if (PyErr_ResourceWarning((PyObject *)iterator, 1,
- "unclosed scandir iterator %R", iterator)) {
- /* Spurious errors can appear at shutdown */
- if (PyErr_ExceptionMatches(PyExc_Warning)) {
- PyErr_WriteUnraisable((PyObject *) iterator);
- }
- }
- }
- path_cleanup(&iterator->path);
- /* Restore the saved exception. */
- PyErr_SetRaisedException(exc);
- }
- static void
- ScandirIterator_dealloc(ScandirIterator *iterator)
- {
- PyTypeObject *tp = Py_TYPE(iterator);
- if (PyObject_CallFinalizerFromDealloc((PyObject *)iterator) < 0)
- return;
- freefunc free_func = PyType_GetSlot(tp, Py_tp_free);
- free_func(iterator);
- Py_DECREF(tp);
- }
- static PyMethodDef ScandirIterator_methods[] = {
- {"__enter__", (PyCFunction)ScandirIterator_enter, METH_NOARGS},
- {"__exit__", (PyCFunction)ScandirIterator_exit, METH_VARARGS},
- {"close", (PyCFunction)ScandirIterator_close, METH_NOARGS},
- {NULL}
- };
- static PyType_Slot ScandirIteratorType_slots[] = {
- {Py_tp_dealloc, ScandirIterator_dealloc},
- {Py_tp_finalize, ScandirIterator_finalize},
- {Py_tp_iter, PyObject_SelfIter},
- {Py_tp_iternext, ScandirIterator_iternext},
- {Py_tp_methods, ScandirIterator_methods},
- {0, 0},
- };
- static PyType_Spec ScandirIteratorType_spec = {
- MODNAME ".ScandirIterator",
- sizeof(ScandirIterator),
- 0,
- // bpo-40549: Py_TPFLAGS_BASETYPE should not be used, since
- // PyType_GetModule(Py_TYPE(self)) doesn't work on a subclass instance.
- (Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_FINALIZE
- | Py_TPFLAGS_DISALLOW_INSTANTIATION),
- ScandirIteratorType_slots
- };
- /*[clinic input]
- os.scandir
- path : path_t(nullable=True, allow_fd='PATH_HAVE_FDOPENDIR') = None
- Return an iterator of DirEntry objects for given path.
- path can be specified as either str, bytes, or a path-like object. If path
- is bytes, the names of yielded DirEntry objects will also be bytes; in
- all other circumstances they will be str.
- If path is None, uses the path='.'.
- [clinic start generated code]*/
- static PyObject *
- os_scandir_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=6eb2668b675ca89e input=6bdd312708fc3bb0]*/
- {
- ScandirIterator *iterator;
- #ifdef MS_WINDOWS
- wchar_t *path_strW;
- #else
- const char *path_str;
- #ifdef HAVE_FDOPENDIR
- int fd = -1;
- #endif
- #endif
- if (PySys_Audit("os.scandir", "O",
- path->object ? path->object : Py_None) < 0) {
- return NULL;
- }
- PyObject *ScandirIteratorType = get_posix_state(module)->ScandirIteratorType;
- iterator = PyObject_New(ScandirIterator, (PyTypeObject *)ScandirIteratorType);
- if (!iterator)
- return NULL;
- #ifdef MS_WINDOWS
- iterator->handle = INVALID_HANDLE_VALUE;
- #else
- iterator->dirp = NULL;
- #endif
- /* Move the ownership to iterator->path */
- memcpy(&iterator->path, path, sizeof(path_t));
- memset(path, 0, sizeof(path_t));
- #ifdef MS_WINDOWS
- iterator->first_time = 1;
- path_strW = join_path_filenameW(iterator->path.wide, L"*.*");
- if (!path_strW)
- goto error;
- Py_BEGIN_ALLOW_THREADS
- iterator->handle = FindFirstFileW(path_strW, &iterator->file_data);
- Py_END_ALLOW_THREADS
- if (iterator->handle == INVALID_HANDLE_VALUE) {
- path_error(&iterator->path);
- PyMem_Free(path_strW);
- goto error;
- }
- PyMem_Free(path_strW);
- #else /* POSIX */
- errno = 0;
- #ifdef HAVE_FDOPENDIR
- if (iterator->path.fd != -1) {
- if (HAVE_FDOPENDIR_RUNTIME) {
- /* closedir() closes the FD, so we duplicate it */
- fd = _Py_dup(iterator->path.fd);
- if (fd == -1)
- goto error;
- Py_BEGIN_ALLOW_THREADS
- iterator->dirp = fdopendir(fd);
- Py_END_ALLOW_THREADS
- } else {
- PyErr_SetString(PyExc_TypeError,
- "scandir: path should be string, bytes, os.PathLike or None, not int");
- return NULL;
- }
- }
- else
- #endif
- {
- if (iterator->path.narrow)
- path_str = iterator->path.narrow;
- else
- path_str = ".";
- Py_BEGIN_ALLOW_THREADS
- iterator->dirp = opendir(path_str);
- Py_END_ALLOW_THREADS
- }
- if (!iterator->dirp) {
- path_error(&iterator->path);
- #ifdef HAVE_FDOPENDIR
- if (fd != -1) {
- Py_BEGIN_ALLOW_THREADS
- close(fd);
- Py_END_ALLOW_THREADS
- }
- #endif
- goto error;
- }
- #endif
- return (PyObject *)iterator;
- error:
- Py_DECREF(iterator);
- return NULL;
- }
- /*
- Return the file system path representation of the object.
- If the object is str or bytes, then allow it to pass through with
- an incremented refcount. If the object defines __fspath__(), then
- return the result of that method. All other types raise a TypeError.
- */
- PyObject *
- PyOS_FSPath(PyObject *path)
- {
- /* For error message reasons, this function is manually inlined in
- path_converter(). */
- PyObject *func = NULL;
- PyObject *path_repr = NULL;
- if (PyUnicode_Check(path) || PyBytes_Check(path)) {
- return Py_NewRef(path);
- }
- func = _PyObject_LookupSpecial(path, &_Py_ID(__fspath__));
- if (NULL == func) {
- return PyErr_Format(PyExc_TypeError,
- "expected str, bytes or os.PathLike object, "
- "not %.200s",
- _PyType_Name(Py_TYPE(path)));
- }
- path_repr = _PyObject_CallNoArgs(func);
- Py_DECREF(func);
- if (NULL == path_repr) {
- return NULL;
- }
- if (!(PyUnicode_Check(path_repr) || PyBytes_Check(path_repr))) {
- PyErr_Format(PyExc_TypeError,
- "expected %.200s.__fspath__() to return str or bytes, "
- "not %.200s", _PyType_Name(Py_TYPE(path)),
- _PyType_Name(Py_TYPE(path_repr)));
- Py_DECREF(path_repr);
- return NULL;
- }
- return path_repr;
- }
- /*[clinic input]
- os.fspath
- path: object
- Return the file system path representation of the object.
- If the object is str or bytes, then allow it to pass through as-is. If the
- object defines __fspath__(), then return the result of that method. All other
- types raise a TypeError.
- [clinic start generated code]*/
- static PyObject *
- os_fspath_impl(PyObject *module, PyObject *path)
- /*[clinic end generated code: output=c3c3b78ecff2914f input=e357165f7b22490f]*/
- {
- return PyOS_FSPath(path);
- }
- #ifdef HAVE_GETRANDOM_SYSCALL
- /*[clinic input]
- os.getrandom
- size: Py_ssize_t
- flags: int=0
- Obtain a series of random bytes.
- [clinic start generated code]*/
- static PyObject *
- os_getrandom_impl(PyObject *module, Py_ssize_t size, int flags)
- /*[clinic end generated code: output=b3a618196a61409c input=59bafac39c594947]*/
- {
- PyObject *bytes;
- Py_ssize_t n;
- if (size < 0) {
- errno = EINVAL;
- return posix_error();
- }
- bytes = PyBytes_FromStringAndSize(NULL, size);
- if (bytes == NULL) {
- PyErr_NoMemory();
- return NULL;
- }
- while (1) {
- n = syscall(SYS_getrandom,
- PyBytes_AS_STRING(bytes),
- PyBytes_GET_SIZE(bytes),
- flags);
- if (n < 0 && errno == EINTR) {
- if (PyErr_CheckSignals() < 0) {
- goto error;
- }
- /* getrandom() was interrupted by a signal: retry */
- continue;
- }
- break;
- }
- if (n < 0) {
- PyErr_SetFromErrno(PyExc_OSError);
- goto error;
- }
- if (n != size) {
- _PyBytes_Resize(&bytes, n);
- }
- return bytes;
- error:
- Py_DECREF(bytes);
- return NULL;
- }
- #endif /* HAVE_GETRANDOM_SYSCALL */
- #if defined(MS_WINDOWS_DESKTOP) || defined(MS_WINDOWS_APP) || defined(MS_WINDOWS_SYSTEM)
- /* bpo-36085: Helper functions for managing DLL search directories
- * on win32
- */
- /*[clinic input]
- os._add_dll_directory
- path: path_t
- Add a path to the DLL search path.
- This search path is used when resolving dependencies for imported
- extension modules (the module itself is resolved through sys.path),
- and also by ctypes.
- Returns an opaque value that may be passed to os.remove_dll_directory
- to remove this directory from the search path.
- [clinic start generated code]*/
- static PyObject *
- os__add_dll_directory_impl(PyObject *module, path_t *path)
- /*[clinic end generated code: output=80b025daebb5d683 input=1de3e6c13a5808c8]*/
- {
- DLL_DIRECTORY_COOKIE cookie = 0;
- DWORD err = 0;
- if (PySys_Audit("os.add_dll_directory", "(O)", path->object) < 0) {
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- if (!(cookie = AddDllDirectory(path->wide))) {
- err = GetLastError();
- }
- Py_END_ALLOW_THREADS
- if (err) {
- return win32_error_object_err("add_dll_directory",
- path->object, err);
- }
- return PyCapsule_New(cookie, "DLL directory cookie", NULL);
- }
- /*[clinic input]
- os._remove_dll_directory
- cookie: object
- Removes a path from the DLL search path.
- The parameter is an opaque value that was returned from
- os.add_dll_directory. You can only remove directories that you added
- yourself.
- [clinic start generated code]*/
- static PyObject *
- os__remove_dll_directory_impl(PyObject *module, PyObject *cookie)
- /*[clinic end generated code: output=594350433ae535bc input=c1d16a7e7d9dc5dc]*/
- {
- DLL_DIRECTORY_COOKIE cookieValue;
- DWORD err = 0;
- if (!PyCapsule_IsValid(cookie, "DLL directory cookie")) {
- PyErr_SetString(PyExc_TypeError,
- "Provided cookie was not returned from os.add_dll_directory");
- return NULL;
- }
- cookieValue = (DLL_DIRECTORY_COOKIE)PyCapsule_GetPointer(
- cookie, "DLL directory cookie");
- Py_BEGIN_ALLOW_THREADS
- if (!RemoveDllDirectory(cookieValue)) {
- err = GetLastError();
- }
- Py_END_ALLOW_THREADS
- if (err) {
- return win32_error_object_err("remove_dll_directory",
- NULL, err);
- }
- if (PyCapsule_SetName(cookie, NULL)) {
- return NULL;
- }
- Py_RETURN_NONE;
- }
- #endif /* MS_WINDOWS_APP || MS_WINDOWS_SYSTEM */
- /* Only check if WIFEXITED is available: expect that it comes
- with WEXITSTATUS, WIFSIGNALED, etc.
- os.waitstatus_to_exitcode() is implemented in C and not in Python, so
- subprocess can safely call it during late Python finalization without
- risking that used os attributes were set to None by finalize_modules(). */
- #if defined(WIFEXITED) || defined(MS_WINDOWS)
- /*[clinic input]
- os.waitstatus_to_exitcode
- status as status_obj: object
- Convert a wait status to an exit code.
- On Unix:
- * If WIFEXITED(status) is true, return WEXITSTATUS(status).
- * If WIFSIGNALED(status) is true, return -WTERMSIG(status).
- * Otherwise, raise a ValueError.
- On Windows, return status shifted right by 8 bits.
- On Unix, if the process is being traced or if waitpid() was called with
- WUNTRACED option, the caller must first check if WIFSTOPPED(status) is true.
- This function must not be called if WIFSTOPPED(status) is true.
- [clinic start generated code]*/
- static PyObject *
- os_waitstatus_to_exitcode_impl(PyObject *module, PyObject *status_obj)
- /*[clinic end generated code: output=db50b1b0ba3c7153 input=7fe2d7fdaea3db42]*/
- {
- #ifndef MS_WINDOWS
- int status = _PyLong_AsInt(status_obj);
- if (status == -1 && PyErr_Occurred()) {
- return NULL;
- }
- WAIT_TYPE wait_status;
- WAIT_STATUS_INT(wait_status) = status;
- int exitcode;
- if (WIFEXITED(wait_status)) {
- exitcode = WEXITSTATUS(wait_status);
- /* Sanity check to provide warranty on the function behavior.
- It should not occur in practice */
- if (exitcode < 0) {
- PyErr_Format(PyExc_ValueError, "invalid WEXITSTATUS: %i", exitcode);
- return NULL;
- }
- }
- else if (WIFSIGNALED(wait_status)) {
- int signum = WTERMSIG(wait_status);
- /* Sanity check to provide warranty on the function behavior.
- It should not occurs in practice */
- if (signum <= 0) {
- PyErr_Format(PyExc_ValueError, "invalid WTERMSIG: %i", signum);
- return NULL;
- }
- exitcode = -signum;
- } else if (WIFSTOPPED(wait_status)) {
- /* Status only received if the process is being traced
- or if waitpid() was called with WUNTRACED option. */
- int signum = WSTOPSIG(wait_status);
- PyErr_Format(PyExc_ValueError,
- "process stopped by delivery of signal %i",
- signum);
- return NULL;
- }
- else {
- PyErr_Format(PyExc_ValueError, "invalid wait status: %i", status);
- return NULL;
- }
- return PyLong_FromLong(exitcode);
- #else
- /* Windows implementation: see os.waitpid() implementation
- which uses _cwait(). */
- unsigned long long status = PyLong_AsUnsignedLongLong(status_obj);
- if (status == (unsigned long long)-1 && PyErr_Occurred()) {
- return NULL;
- }
- unsigned long long exitcode = (status >> 8);
- /* ExitProcess() accepts an UINT type:
- reject exit code which doesn't fit in an UINT */
- if (exitcode > UINT_MAX) {
- PyErr_Format(PyExc_ValueError, "invalid exit code: %llu", exitcode);
- return NULL;
- }
- return PyLong_FromUnsignedLong((unsigned long)exitcode);
- #endif
- }
- #endif
- static PyMethodDef posix_methods[] = {
- OS_STAT_METHODDEF
- OS_ACCESS_METHODDEF
- OS_TTYNAME_METHODDEF
- OS_CHDIR_METHODDEF
- OS_CHFLAGS_METHODDEF
- OS_CHMOD_METHODDEF
- OS_FCHMOD_METHODDEF
- OS_LCHMOD_METHODDEF
- OS_CHOWN_METHODDEF
- OS_FCHOWN_METHODDEF
- OS_LCHOWN_METHODDEF
- OS_LCHFLAGS_METHODDEF
- OS_CHROOT_METHODDEF
- OS_CTERMID_METHODDEF
- OS_GETCWD_METHODDEF
- OS_GETCWDB_METHODDEF
- OS_LINK_METHODDEF
- OS_LISTDIR_METHODDEF
- OS_LISTDRIVES_METHODDEF
- OS_LISTMOUNTS_METHODDEF
- OS_LISTVOLUMES_METHODDEF
- OS_LSTAT_METHODDEF
- OS_MKDIR_METHODDEF
- OS_NICE_METHODDEF
- OS_GETPRIORITY_METHODDEF
- OS_SETPRIORITY_METHODDEF
- OS_POSIX_SPAWN_METHODDEF
- OS_POSIX_SPAWNP_METHODDEF
- OS_READLINK_METHODDEF
- OS_COPY_FILE_RANGE_METHODDEF
- OS_SPLICE_METHODDEF
- OS_RENAME_METHODDEF
- OS_REPLACE_METHODDEF
- OS_RMDIR_METHODDEF
- OS_SYMLINK_METHODDEF
- OS_SYSTEM_METHODDEF
- OS_UMASK_METHODDEF
- OS_UNAME_METHODDEF
- OS_UNLINK_METHODDEF
- OS_REMOVE_METHODDEF
- OS_UTIME_METHODDEF
- OS_TIMES_METHODDEF
- OS__EXIT_METHODDEF
- OS__FCOPYFILE_METHODDEF
- OS_EXECV_METHODDEF
- OS_EXECVE_METHODDEF
- OS_SPAWNV_METHODDEF
- OS_SPAWNVE_METHODDEF
- OS_FORK1_METHODDEF
- OS_FORK_METHODDEF
- OS_REGISTER_AT_FORK_METHODDEF
- OS_SCHED_GET_PRIORITY_MAX_METHODDEF
- OS_SCHED_GET_PRIORITY_MIN_METHODDEF
- OS_SCHED_GETPARAM_METHODDEF
- OS_SCHED_GETSCHEDULER_METHODDEF
- OS_SCHED_RR_GET_INTERVAL_METHODDEF
- OS_SCHED_SETPARAM_METHODDEF
- OS_SCHED_SETSCHEDULER_METHODDEF
- OS_SCHED_YIELD_METHODDEF
- OS_SCHED_SETAFFINITY_METHODDEF
- OS_SCHED_GETAFFINITY_METHODDEF
- OS_OPENPTY_METHODDEF
- OS_LOGIN_TTY_METHODDEF
- OS_FORKPTY_METHODDEF
- OS_GETEGID_METHODDEF
- OS_GETEUID_METHODDEF
- OS_GETGID_METHODDEF
- OS_GETGROUPLIST_METHODDEF
- OS_GETGROUPS_METHODDEF
- OS_GETPID_METHODDEF
- OS_GETPGRP_METHODDEF
- OS_GETPPID_METHODDEF
- OS_GETUID_METHODDEF
- OS_GETLOGIN_METHODDEF
- OS_KILL_METHODDEF
- OS_KILLPG_METHODDEF
- OS_PLOCK_METHODDEF
- OS_STARTFILE_METHODDEF
- OS_SETUID_METHODDEF
- OS_SETEUID_METHODDEF
- OS_SETREUID_METHODDEF
- OS_SETGID_METHODDEF
- OS_SETEGID_METHODDEF
- OS_SETREGID_METHODDEF
- OS_SETGROUPS_METHODDEF
- OS_INITGROUPS_METHODDEF
- OS_GETPGID_METHODDEF
- OS_SETPGRP_METHODDEF
- OS_WAIT_METHODDEF
- OS_WAIT3_METHODDEF
- OS_WAIT4_METHODDEF
- OS_WAITID_METHODDEF
- OS_WAITPID_METHODDEF
- OS_PIDFD_OPEN_METHODDEF
- OS_GETSID_METHODDEF
- OS_SETSID_METHODDEF
- OS_SETPGID_METHODDEF
- OS_TCGETPGRP_METHODDEF
- OS_TCSETPGRP_METHODDEF
- OS_OPEN_METHODDEF
- OS_CLOSE_METHODDEF
- OS_CLOSERANGE_METHODDEF
- OS_DEVICE_ENCODING_METHODDEF
- OS_DUP_METHODDEF
- OS_DUP2_METHODDEF
- OS_LOCKF_METHODDEF
- OS_LSEEK_METHODDEF
- OS_READ_METHODDEF
- OS_READV_METHODDEF
- OS_PREAD_METHODDEF
- OS_PREADV_METHODDEF
- OS_WRITE_METHODDEF
- OS_WRITEV_METHODDEF
- OS_PWRITE_METHODDEF
- OS_PWRITEV_METHODDEF
- OS_SENDFILE_METHODDEF
- OS_FSTAT_METHODDEF
- OS_ISATTY_METHODDEF
- OS_PIPE_METHODDEF
- OS_PIPE2_METHODDEF
- OS_MKFIFO_METHODDEF
- OS_MKNOD_METHODDEF
- OS_MAJOR_METHODDEF
- OS_MINOR_METHODDEF
- OS_MAKEDEV_METHODDEF
- OS_FTRUNCATE_METHODDEF
- OS_TRUNCATE_METHODDEF
- OS_POSIX_FALLOCATE_METHODDEF
- OS_POSIX_FADVISE_METHODDEF
- OS_PUTENV_METHODDEF
- OS_UNSETENV_METHODDEF
- OS_STRERROR_METHODDEF
- OS_FCHDIR_METHODDEF
- OS_FSYNC_METHODDEF
- OS_SYNC_METHODDEF
- OS_FDATASYNC_METHODDEF
- OS_WCOREDUMP_METHODDEF
- OS_WIFCONTINUED_METHODDEF
- OS_WIFSTOPPED_METHODDEF
- OS_WIFSIGNALED_METHODDEF
- OS_WIFEXITED_METHODDEF
- OS_WEXITSTATUS_METHODDEF
- OS_WTERMSIG_METHODDEF
- OS_WSTOPSIG_METHODDEF
- OS_FSTATVFS_METHODDEF
- OS_STATVFS_METHODDEF
- OS_CONFSTR_METHODDEF
- OS_SYSCONF_METHODDEF
- OS_FPATHCONF_METHODDEF
- OS_PATHCONF_METHODDEF
- OS_ABORT_METHODDEF
- OS__GETFULLPATHNAME_METHODDEF
- OS__GETDISKUSAGE_METHODDEF
- OS__GETFINALPATHNAME_METHODDEF
- OS__GETVOLUMEPATHNAME_METHODDEF
- OS__PATH_SPLITROOT_METHODDEF
- OS__PATH_NORMPATH_METHODDEF
- OS_GETLOADAVG_METHODDEF
- OS_URANDOM_METHODDEF
- OS_SETRESUID_METHODDEF
- OS_SETRESGID_METHODDEF
- OS_GETRESUID_METHODDEF
- OS_GETRESGID_METHODDEF
- OS_GETXATTR_METHODDEF
- OS_SETXATTR_METHODDEF
- OS_REMOVEXATTR_METHODDEF
- OS_LISTXATTR_METHODDEF
- OS_GET_TERMINAL_SIZE_METHODDEF
- OS_CPU_COUNT_METHODDEF
- OS_GET_INHERITABLE_METHODDEF
- OS_SET_INHERITABLE_METHODDEF
- OS_GET_HANDLE_INHERITABLE_METHODDEF
- OS_SET_HANDLE_INHERITABLE_METHODDEF
- OS_GET_BLOCKING_METHODDEF
- OS_SET_BLOCKING_METHODDEF
- OS_SCANDIR_METHODDEF
- OS_FSPATH_METHODDEF
- OS_GETRANDOM_METHODDEF
- OS_MEMFD_CREATE_METHODDEF
- OS_EVENTFD_METHODDEF
- OS_EVENTFD_READ_METHODDEF
- OS_EVENTFD_WRITE_METHODDEF
- OS__ADD_DLL_DIRECTORY_METHODDEF
- OS__REMOVE_DLL_DIRECTORY_METHODDEF
- OS_WAITSTATUS_TO_EXITCODE_METHODDEF
- OS_SETNS_METHODDEF
- OS_UNSHARE_METHODDEF
- OS__PATH_ISDEVDRIVE_METHODDEF
- OS__PATH_ISDIR_METHODDEF
- OS__PATH_ISFILE_METHODDEF
- OS__PATH_ISLINK_METHODDEF
- OS__PATH_EXISTS_METHODDEF
- {NULL, NULL} /* Sentinel */
- };
- static int
- all_ins(PyObject *m)
- {
- #ifdef F_OK
- if (PyModule_AddIntMacro(m, F_OK)) return -1;
- #endif
- #ifdef R_OK
- if (PyModule_AddIntMacro(m, R_OK)) return -1;
- #endif
- #ifdef W_OK
- if (PyModule_AddIntMacro(m, W_OK)) return -1;
- #endif
- #ifdef X_OK
- if (PyModule_AddIntMacro(m, X_OK)) return -1;
- #endif
- #ifdef NGROUPS_MAX
- if (PyModule_AddIntMacro(m, NGROUPS_MAX)) return -1;
- #endif
- #ifdef TMP_MAX
- if (PyModule_AddIntMacro(m, TMP_MAX)) return -1;
- #endif
- #ifdef WCONTINUED
- if (PyModule_AddIntMacro(m, WCONTINUED)) return -1;
- #endif
- #ifdef WNOHANG
- if (PyModule_AddIntMacro(m, WNOHANG)) return -1;
- #endif
- #ifdef WUNTRACED
- if (PyModule_AddIntMacro(m, WUNTRACED)) return -1;
- #endif
- #ifdef O_RDONLY
- if (PyModule_AddIntMacro(m, O_RDONLY)) return -1;
- #endif
- #ifdef O_WRONLY
- if (PyModule_AddIntMacro(m, O_WRONLY)) return -1;
- #endif
- #ifdef O_RDWR
- if (PyModule_AddIntMacro(m, O_RDWR)) return -1;
- #endif
- #ifdef O_NDELAY
- if (PyModule_AddIntMacro(m, O_NDELAY)) return -1;
- #endif
- #ifdef O_NONBLOCK
- if (PyModule_AddIntMacro(m, O_NONBLOCK)) return -1;
- #endif
- #ifdef O_APPEND
- if (PyModule_AddIntMacro(m, O_APPEND)) return -1;
- #endif
- #ifdef O_DSYNC
- if (PyModule_AddIntMacro(m, O_DSYNC)) return -1;
- #endif
- #ifdef O_RSYNC
- if (PyModule_AddIntMacro(m, O_RSYNC)) return -1;
- #endif
- #ifdef O_SYNC
- if (PyModule_AddIntMacro(m, O_SYNC)) return -1;
- #endif
- #ifdef O_NOCTTY
- if (PyModule_AddIntMacro(m, O_NOCTTY)) return -1;
- #endif
- #ifdef O_CREAT
- if (PyModule_AddIntMacro(m, O_CREAT)) return -1;
- #endif
- #ifdef O_EXCL
- if (PyModule_AddIntMacro(m, O_EXCL)) return -1;
- #endif
- #ifdef O_TRUNC
- if (PyModule_AddIntMacro(m, O_TRUNC)) return -1;
- #endif
- #ifdef O_BINARY
- if (PyModule_AddIntMacro(m, O_BINARY)) return -1;
- #endif
- #ifdef O_TEXT
- if (PyModule_AddIntMacro(m, O_TEXT)) return -1;
- #endif
- #ifdef O_XATTR
- if (PyModule_AddIntMacro(m, O_XATTR)) return -1;
- #endif
- #ifdef O_LARGEFILE
- if (PyModule_AddIntMacro(m, O_LARGEFILE)) return -1;
- #endif
- #ifndef __GNU__
- #ifdef O_SHLOCK
- if (PyModule_AddIntMacro(m, O_SHLOCK)) return -1;
- #endif
- #ifdef O_EXLOCK
- if (PyModule_AddIntMacro(m, O_EXLOCK)) return -1;
- #endif
- #endif
- #ifdef O_EXEC
- if (PyModule_AddIntMacro(m, O_EXEC)) return -1;
- #endif
- #ifdef O_SEARCH
- if (PyModule_AddIntMacro(m, O_SEARCH)) return -1;
- #endif
- #ifdef O_PATH
- if (PyModule_AddIntMacro(m, O_PATH)) return -1;
- #endif
- #ifdef O_TTY_INIT
- if (PyModule_AddIntMacro(m, O_TTY_INIT)) return -1;
- #endif
- #ifdef O_TMPFILE
- if (PyModule_AddIntMacro(m, O_TMPFILE)) return -1;
- #endif
- #ifdef PRIO_PROCESS
- if (PyModule_AddIntMacro(m, PRIO_PROCESS)) return -1;
- #endif
- #ifdef PRIO_PGRP
- if (PyModule_AddIntMacro(m, PRIO_PGRP)) return -1;
- #endif
- #ifdef PRIO_USER
- if (PyModule_AddIntMacro(m, PRIO_USER)) return -1;
- #endif
- #ifdef PRIO_DARWIN_THREAD
- if (PyModule_AddIntMacro(m, PRIO_DARWIN_THREAD)) return -1;
- #endif
- #ifdef PRIO_DARWIN_PROCESS
- if (PyModule_AddIntMacro(m, PRIO_DARWIN_PROCESS)) return -1;
- #endif
- #ifdef PRIO_DARWIN_BG
- if (PyModule_AddIntMacro(m, PRIO_DARWIN_BG)) return -1;
- #endif
- #ifdef PRIO_DARWIN_NONUI
- if (PyModule_AddIntMacro(m, PRIO_DARWIN_NONUI)) return -1;
- #endif
- #ifdef O_CLOEXEC
- if (PyModule_AddIntMacro(m, O_CLOEXEC)) return -1;
- #endif
- #ifdef O_ACCMODE
- if (PyModule_AddIntMacro(m, O_ACCMODE)) return -1;
- #endif
- #ifdef O_EVTONLY
- if (PyModule_AddIntMacro(m, O_EVTONLY)) return -1;
- #endif
- #ifdef O_FSYNC
- if (PyModule_AddIntMacro(m, O_FSYNC)) return -1;
- #endif
- #ifdef O_SYMLINK
- if (PyModule_AddIntMacro(m, O_SYMLINK)) return -1;
- #endif
- #ifdef SEEK_HOLE
- if (PyModule_AddIntMacro(m, SEEK_HOLE)) return -1;
- #endif
- #ifdef SEEK_DATA
- if (PyModule_AddIntMacro(m, SEEK_DATA)) return -1;
- #endif
- /* MS Windows */
- #ifdef O_NOINHERIT
- /* Don't inherit in child processes. */
- if (PyModule_AddIntMacro(m, O_NOINHERIT)) return -1;
- #endif
- #ifdef _O_SHORT_LIVED
- /* Optimize for short life (keep in memory). */
- /* MS forgot to define this one with a non-underscore form too. */
- if (PyModule_AddIntConstant(m, "O_SHORT_LIVED", _O_SHORT_LIVED)) return -1;
- #endif
- #ifdef O_TEMPORARY
- /* Automatically delete when last handle is closed. */
- if (PyModule_AddIntMacro(m, O_TEMPORARY)) return -1;
- #endif
- #ifdef O_RANDOM
- /* Optimize for random access. */
- if (PyModule_AddIntMacro(m, O_RANDOM)) return -1;
- #endif
- #ifdef O_SEQUENTIAL
- /* Optimize for sequential access. */
- if (PyModule_AddIntMacro(m, O_SEQUENTIAL)) return -1;
- #endif
- /* GNU extensions. */
- #ifdef O_ASYNC
- /* Send a SIGIO signal whenever input or output
- becomes available on file descriptor */
- if (PyModule_AddIntMacro(m, O_ASYNC)) return -1;
- #endif
- #ifdef O_DIRECT
- /* Direct disk access. */
- if (PyModule_AddIntMacro(m, O_DIRECT)) return -1;
- #endif
- #ifdef O_DIRECTORY
- /* Must be a directory. */
- if (PyModule_AddIntMacro(m, O_DIRECTORY)) return -1;
- #endif
- #ifdef O_NOFOLLOW
- /* Do not follow links. */
- if (PyModule_AddIntMacro(m, O_NOFOLLOW)) return -1;
- #endif
- #ifdef O_NOFOLLOW_ANY
- if (PyModule_AddIntMacro(m, O_NOFOLLOW_ANY)) return -1;
- #endif
- #ifdef O_NOLINKS
- /* Fails if link count of the named file is greater than 1 */
- if (PyModule_AddIntMacro(m, O_NOLINKS)) return -1;
- #endif
- #ifdef O_NOATIME
- /* Do not update the access time. */
- if (PyModule_AddIntMacro(m, O_NOATIME)) return -1;
- #endif
- /* These come from sysexits.h */
- #ifdef EX_OK
- if (PyModule_AddIntMacro(m, EX_OK)) return -1;
- #endif /* EX_OK */
- #ifdef EX_USAGE
- if (PyModule_AddIntMacro(m, EX_USAGE)) return -1;
- #endif /* EX_USAGE */
- #ifdef EX_DATAERR
- if (PyModule_AddIntMacro(m, EX_DATAERR)) return -1;
- #endif /* EX_DATAERR */
- #ifdef EX_NOINPUT
- if (PyModule_AddIntMacro(m, EX_NOINPUT)) return -1;
- #endif /* EX_NOINPUT */
- #ifdef EX_NOUSER
- if (PyModule_AddIntMacro(m, EX_NOUSER)) return -1;
- #endif /* EX_NOUSER */
- #ifdef EX_NOHOST
- if (PyModule_AddIntMacro(m, EX_NOHOST)) return -1;
- #endif /* EX_NOHOST */
- #ifdef EX_UNAVAILABLE
- if (PyModule_AddIntMacro(m, EX_UNAVAILABLE)) return -1;
- #endif /* EX_UNAVAILABLE */
- #ifdef EX_SOFTWARE
- if (PyModule_AddIntMacro(m, EX_SOFTWARE)) return -1;
- #endif /* EX_SOFTWARE */
- #ifdef EX_OSERR
- if (PyModule_AddIntMacro(m, EX_OSERR)) return -1;
- #endif /* EX_OSERR */
- #ifdef EX_OSFILE
- if (PyModule_AddIntMacro(m, EX_OSFILE)) return -1;
- #endif /* EX_OSFILE */
- #ifdef EX_CANTCREAT
- if (PyModule_AddIntMacro(m, EX_CANTCREAT)) return -1;
- #endif /* EX_CANTCREAT */
- #ifdef EX_IOERR
- if (PyModule_AddIntMacro(m, EX_IOERR)) return -1;
- #endif /* EX_IOERR */
- #ifdef EX_TEMPFAIL
- if (PyModule_AddIntMacro(m, EX_TEMPFAIL)) return -1;
- #endif /* EX_TEMPFAIL */
- #ifdef EX_PROTOCOL
- if (PyModule_AddIntMacro(m, EX_PROTOCOL)) return -1;
- #endif /* EX_PROTOCOL */
- #ifdef EX_NOPERM
- if (PyModule_AddIntMacro(m, EX_NOPERM)) return -1;
- #endif /* EX_NOPERM */
- #ifdef EX_CONFIG
- if (PyModule_AddIntMacro(m, EX_CONFIG)) return -1;
- #endif /* EX_CONFIG */
- #ifdef EX_NOTFOUND
- if (PyModule_AddIntMacro(m, EX_NOTFOUND)) return -1;
- #endif /* EX_NOTFOUND */
- /* statvfs */
- #ifdef ST_RDONLY
- if (PyModule_AddIntMacro(m, ST_RDONLY)) return -1;
- #endif /* ST_RDONLY */
- #ifdef ST_NOSUID
- if (PyModule_AddIntMacro(m, ST_NOSUID)) return -1;
- #endif /* ST_NOSUID */
- /* GNU extensions */
- #ifdef ST_NODEV
- if (PyModule_AddIntMacro(m, ST_NODEV)) return -1;
- #endif /* ST_NODEV */
- #ifdef ST_NOEXEC
- if (PyModule_AddIntMacro(m, ST_NOEXEC)) return -1;
- #endif /* ST_NOEXEC */
- #ifdef ST_SYNCHRONOUS
- if (PyModule_AddIntMacro(m, ST_SYNCHRONOUS)) return -1;
- #endif /* ST_SYNCHRONOUS */
- #ifdef ST_MANDLOCK
- if (PyModule_AddIntMacro(m, ST_MANDLOCK)) return -1;
- #endif /* ST_MANDLOCK */
- #ifdef ST_WRITE
- if (PyModule_AddIntMacro(m, ST_WRITE)) return -1;
- #endif /* ST_WRITE */
- #ifdef ST_APPEND
- if (PyModule_AddIntMacro(m, ST_APPEND)) return -1;
- #endif /* ST_APPEND */
- #ifdef ST_NOATIME
- if (PyModule_AddIntMacro(m, ST_NOATIME)) return -1;
- #endif /* ST_NOATIME */
- #ifdef ST_NODIRATIME
- if (PyModule_AddIntMacro(m, ST_NODIRATIME)) return -1;
- #endif /* ST_NODIRATIME */
- #ifdef ST_RELATIME
- if (PyModule_AddIntMacro(m, ST_RELATIME)) return -1;
- #endif /* ST_RELATIME */
- /* FreeBSD sendfile() constants */
- #ifdef SF_NODISKIO
- if (PyModule_AddIntMacro(m, SF_NODISKIO)) return -1;
- #endif
- /* is obsolete since the 11.x release */
- #ifdef SF_MNOWAIT
- if (PyModule_AddIntMacro(m, SF_MNOWAIT)) return -1;
- #endif
- #ifdef SF_SYNC
- if (PyModule_AddIntMacro(m, SF_SYNC)) return -1;
- #endif
- #ifdef SF_NOCACHE
- if (PyModule_AddIntMacro(m, SF_NOCACHE)) return -1;
- #endif
- /* constants for posix_fadvise */
- #ifdef POSIX_FADV_NORMAL
- if (PyModule_AddIntMacro(m, POSIX_FADV_NORMAL)) return -1;
- #endif
- #ifdef POSIX_FADV_SEQUENTIAL
- if (PyModule_AddIntMacro(m, POSIX_FADV_SEQUENTIAL)) return -1;
- #endif
- #ifdef POSIX_FADV_RANDOM
- if (PyModule_AddIntMacro(m, POSIX_FADV_RANDOM)) return -1;
- #endif
- #ifdef POSIX_FADV_NOREUSE
- if (PyModule_AddIntMacro(m, POSIX_FADV_NOREUSE)) return -1;
- #endif
- #ifdef POSIX_FADV_WILLNEED
- if (PyModule_AddIntMacro(m, POSIX_FADV_WILLNEED)) return -1;
- #endif
- #ifdef POSIX_FADV_DONTNEED
- if (PyModule_AddIntMacro(m, POSIX_FADV_DONTNEED)) return -1;
- #endif
- /* constants for waitid */
- #if defined(HAVE_SYS_WAIT_H) && defined(HAVE_WAITID)
- if (PyModule_AddIntMacro(m, P_PID)) return -1;
- if (PyModule_AddIntMacro(m, P_PGID)) return -1;
- if (PyModule_AddIntMacro(m, P_ALL)) return -1;
- #ifdef P_PIDFD
- if (PyModule_AddIntMacro(m, P_PIDFD)) return -1;
- #endif
- #ifdef PIDFD_NONBLOCK
- if (PyModule_AddIntMacro(m, PIDFD_NONBLOCK)) return -1;
- #endif
- #endif
- #ifdef WEXITED
- if (PyModule_AddIntMacro(m, WEXITED)) return -1;
- #endif
- #ifdef WNOWAIT
- if (PyModule_AddIntMacro(m, WNOWAIT)) return -1;
- #endif
- #ifdef WSTOPPED
- if (PyModule_AddIntMacro(m, WSTOPPED)) return -1;
- #endif
- #ifdef CLD_EXITED
- if (PyModule_AddIntMacro(m, CLD_EXITED)) return -1;
- #endif
- #ifdef CLD_KILLED
- if (PyModule_AddIntMacro(m, CLD_KILLED)) return -1;
- #endif
- #ifdef CLD_DUMPED
- if (PyModule_AddIntMacro(m, CLD_DUMPED)) return -1;
- #endif
- #ifdef CLD_TRAPPED
- if (PyModule_AddIntMacro(m, CLD_TRAPPED)) return -1;
- #endif
- #ifdef CLD_STOPPED
- if (PyModule_AddIntMacro(m, CLD_STOPPED)) return -1;
- #endif
- #ifdef CLD_CONTINUED
- if (PyModule_AddIntMacro(m, CLD_CONTINUED)) return -1;
- #endif
- /* constants for lockf */
- #ifdef F_LOCK
- if (PyModule_AddIntMacro(m, F_LOCK)) return -1;
- #endif
- #ifdef F_TLOCK
- if (PyModule_AddIntMacro(m, F_TLOCK)) return -1;
- #endif
- #ifdef F_ULOCK
- if (PyModule_AddIntMacro(m, F_ULOCK)) return -1;
- #endif
- #ifdef F_TEST
- if (PyModule_AddIntMacro(m, F_TEST)) return -1;
- #endif
- #ifdef RWF_DSYNC
- if (PyModule_AddIntConstant(m, "RWF_DSYNC", RWF_DSYNC)) return -1;
- #endif
- #ifdef RWF_HIPRI
- if (PyModule_AddIntConstant(m, "RWF_HIPRI", RWF_HIPRI)) return -1;
- #endif
- #ifdef RWF_SYNC
- if (PyModule_AddIntConstant(m, "RWF_SYNC", RWF_SYNC)) return -1;
- #endif
- #ifdef RWF_NOWAIT
- if (PyModule_AddIntConstant(m, "RWF_NOWAIT", RWF_NOWAIT)) return -1;
- #endif
- #ifdef RWF_APPEND
- if (PyModule_AddIntConstant(m, "RWF_APPEND", RWF_APPEND)) return -1;
- #endif
- /* constants for splice */
- #if defined(HAVE_SPLICE) && defined(__linux__)
- if (PyModule_AddIntConstant(m, "SPLICE_F_MOVE", SPLICE_F_MOVE)) return -1;
- if (PyModule_AddIntConstant(m, "SPLICE_F_NONBLOCK", SPLICE_F_NONBLOCK)) return -1;
- if (PyModule_AddIntConstant(m, "SPLICE_F_MORE", SPLICE_F_MORE)) return -1;
- #endif
- /* constants for posix_spawn */
- #ifdef HAVE_POSIX_SPAWN
- if (PyModule_AddIntConstant(m, "POSIX_SPAWN_OPEN", POSIX_SPAWN_OPEN)) return -1;
- if (PyModule_AddIntConstant(m, "POSIX_SPAWN_CLOSE", POSIX_SPAWN_CLOSE)) return -1;
- if (PyModule_AddIntConstant(m, "POSIX_SPAWN_DUP2", POSIX_SPAWN_DUP2)) return -1;
- #endif
- #if defined(HAVE_SPAWNV) || defined (HAVE_RTPSPAWN)
- if (PyModule_AddIntConstant(m, "P_WAIT", _P_WAIT)) return -1;
- if (PyModule_AddIntConstant(m, "P_NOWAIT", _P_NOWAIT)) return -1;
- if (PyModule_AddIntConstant(m, "P_NOWAITO", _P_NOWAITO)) return -1;
- #endif
- #ifdef HAVE_SPAWNV
- if (PyModule_AddIntConstant(m, "P_OVERLAY", _OLD_P_OVERLAY)) return -1;
- if (PyModule_AddIntConstant(m, "P_DETACH", _P_DETACH)) return -1;
- #endif
- #ifdef HAVE_SCHED_H
- #ifdef SCHED_OTHER
- if (PyModule_AddIntMacro(m, SCHED_OTHER)) return -1;
- #endif
- #ifdef SCHED_FIFO
- if (PyModule_AddIntMacro(m, SCHED_FIFO)) return -1;
- #endif
- #ifdef SCHED_RR
- if (PyModule_AddIntMacro(m, SCHED_RR)) return -1;
- #endif
- #ifdef SCHED_SPORADIC
- if (PyModule_AddIntMacro(m, SCHED_SPORADIC)) return -1;
- #endif
- #ifdef SCHED_BATCH
- if (PyModule_AddIntMacro(m, SCHED_BATCH)) return -1;
- #endif
- #ifdef SCHED_IDLE
- if (PyModule_AddIntMacro(m, SCHED_IDLE)) return -1;
- #endif
- #ifdef SCHED_RESET_ON_FORK
- if (PyModule_AddIntMacro(m, SCHED_RESET_ON_FORK)) return -1;
- #endif
- #ifdef SCHED_SYS
- if (PyModule_AddIntMacro(m, SCHED_SYS)) return -1;
- #endif
- #ifdef SCHED_IA
- if (PyModule_AddIntMacro(m, SCHED_IA)) return -1;
- #endif
- #ifdef SCHED_FSS
- if (PyModule_AddIntMacro(m, SCHED_FSS)) return -1;
- #endif
- #ifdef SCHED_FX
- if (PyModule_AddIntConstant(m, "SCHED_FX", SCHED_FSS)) return -1;
- #endif
- /* constants for namespaces */
- #if defined(HAVE_SETNS) || defined(HAVE_UNSHARE)
- #ifdef CLONE_FS
- if (PyModule_AddIntMacro(m, CLONE_FS)) return -1;
- #endif
- #ifdef CLONE_FILES
- if (PyModule_AddIntMacro(m, CLONE_FILES)) return -1;
- #endif
- #ifdef CLONE_NEWNS
- if (PyModule_AddIntMacro(m, CLONE_NEWNS)) return -1;
- #endif
- #ifdef CLONE_NEWCGROUP
- if (PyModule_AddIntMacro(m, CLONE_NEWCGROUP)) return -1;
- #endif
- #ifdef CLONE_NEWUTS
- if (PyModule_AddIntMacro(m, CLONE_NEWUTS)) return -1;
- #endif
- #ifdef CLONE_NEWIPC
- if (PyModule_AddIntMacro(m, CLONE_NEWIPC)) return -1;
- #endif
- #ifdef CLONE_NEWUSER
- if (PyModule_AddIntMacro(m, CLONE_NEWUSER)) return -1;
- #endif
- #ifdef CLONE_NEWPID
- if (PyModule_AddIntMacro(m, CLONE_NEWPID)) return -1;
- #endif
- #ifdef CLONE_NEWNET
- if (PyModule_AddIntMacro(m, CLONE_NEWNET)) return -1;
- #endif
- #ifdef CLONE_NEWTIME
- if (PyModule_AddIntMacro(m, CLONE_NEWTIME)) return -1;
- #endif
- #ifdef CLONE_SYSVSEM
- if (PyModule_AddIntMacro(m, CLONE_SYSVSEM)) return -1;
- #endif
- #ifdef CLONE_THREAD
- if (PyModule_AddIntMacro(m, CLONE_THREAD)) return -1;
- #endif
- #ifdef CLONE_SIGHAND
- if (PyModule_AddIntMacro(m, CLONE_SIGHAND)) return -1;
- #endif
- #ifdef CLONE_VM
- if (PyModule_AddIntMacro(m, CLONE_VM)) return -1;
- #endif
- #endif
- #endif
- #ifdef USE_XATTRS
- if (PyModule_AddIntMacro(m, XATTR_CREATE)) return -1;
- if (PyModule_AddIntMacro(m, XATTR_REPLACE)) return -1;
- if (PyModule_AddIntMacro(m, XATTR_SIZE_MAX)) return -1;
- #endif
- #if HAVE_DECL_RTLD_LAZY
- if (PyModule_AddIntMacro(m, RTLD_LAZY)) return -1;
- #endif
- #if HAVE_DECL_RTLD_NOW
- if (PyModule_AddIntMacro(m, RTLD_NOW)) return -1;
- #endif
- #if HAVE_DECL_RTLD_GLOBAL
- if (PyModule_AddIntMacro(m, RTLD_GLOBAL)) return -1;
- #endif
- #if HAVE_DECL_RTLD_LOCAL
- if (PyModule_AddIntMacro(m, RTLD_LOCAL)) return -1;
- #endif
- #if HAVE_DECL_RTLD_NODELETE
- if (PyModule_AddIntMacro(m, RTLD_NODELETE)) return -1;
- #endif
- #if HAVE_DECL_RTLD_NOLOAD
- if (PyModule_AddIntMacro(m, RTLD_NOLOAD)) return -1;
- #endif
- #if HAVE_DECL_RTLD_DEEPBIND
- if (PyModule_AddIntMacro(m, RTLD_DEEPBIND)) return -1;
- #endif
- #if HAVE_DECL_RTLD_MEMBER
- if (PyModule_AddIntMacro(m, RTLD_MEMBER)) return -1;
- #endif
- #ifdef HAVE_GETRANDOM_SYSCALL
- if (PyModule_AddIntMacro(m, GRND_RANDOM)) return -1;
- if (PyModule_AddIntMacro(m, GRND_NONBLOCK)) return -1;
- #endif
- #ifdef HAVE_MEMFD_CREATE
- if (PyModule_AddIntMacro(m, MFD_CLOEXEC)) return -1;
- if (PyModule_AddIntMacro(m, MFD_ALLOW_SEALING)) return -1;
- #ifdef MFD_HUGETLB
- if (PyModule_AddIntMacro(m, MFD_HUGETLB)) return -1;
- #endif
- #ifdef MFD_HUGE_SHIFT
- if (PyModule_AddIntMacro(m, MFD_HUGE_SHIFT)) return -1;
- #endif
- #ifdef MFD_HUGE_MASK
- if (PyModule_AddIntMacro(m, MFD_HUGE_MASK)) return -1;
- #endif
- #ifdef MFD_HUGE_64KB
- if (PyModule_AddIntMacro(m, MFD_HUGE_64KB)) return -1;
- #endif
- #ifdef MFD_HUGE_512KB
- if (PyModule_AddIntMacro(m, MFD_HUGE_512KB)) return -1;
- #endif
- #ifdef MFD_HUGE_1MB
- if (PyModule_AddIntMacro(m, MFD_HUGE_1MB)) return -1;
- #endif
- #ifdef MFD_HUGE_2MB
- if (PyModule_AddIntMacro(m, MFD_HUGE_2MB)) return -1;
- #endif
- #ifdef MFD_HUGE_8MB
- if (PyModule_AddIntMacro(m, MFD_HUGE_8MB)) return -1;
- #endif
- #ifdef MFD_HUGE_16MB
- if (PyModule_AddIntMacro(m, MFD_HUGE_16MB)) return -1;
- #endif
- #ifdef MFD_HUGE_32MB
- if (PyModule_AddIntMacro(m, MFD_HUGE_32MB)) return -1;
- #endif
- #ifdef MFD_HUGE_256MB
- if (PyModule_AddIntMacro(m, MFD_HUGE_256MB)) return -1;
- #endif
- #ifdef MFD_HUGE_512MB
- if (PyModule_AddIntMacro(m, MFD_HUGE_512MB)) return -1;
- #endif
- #ifdef MFD_HUGE_1GB
- if (PyModule_AddIntMacro(m, MFD_HUGE_1GB)) return -1;
- #endif
- #ifdef MFD_HUGE_2GB
- if (PyModule_AddIntMacro(m, MFD_HUGE_2GB)) return -1;
- #endif
- #ifdef MFD_HUGE_16GB
- if (PyModule_AddIntMacro(m, MFD_HUGE_16GB)) return -1;
- #endif
- #endif /* HAVE_MEMFD_CREATE */
- #if defined(HAVE_EVENTFD) && defined(EFD_CLOEXEC)
- if (PyModule_AddIntMacro(m, EFD_CLOEXEC)) return -1;
- #ifdef EFD_NONBLOCK
- if (PyModule_AddIntMacro(m, EFD_NONBLOCK)) return -1;
- #endif
- #ifdef EFD_SEMAPHORE
- if (PyModule_AddIntMacro(m, EFD_SEMAPHORE)) return -1;
- #endif
- #endif /* HAVE_EVENTFD && EFD_CLOEXEC */
- #if defined(__APPLE__)
- if (PyModule_AddIntConstant(m, "_COPYFILE_DATA", COPYFILE_DATA)) return -1;
- if (PyModule_AddIntConstant(m, "_COPYFILE_STAT", COPYFILE_STAT)) return -1;
- if (PyModule_AddIntConstant(m, "_COPYFILE_ACL", COPYFILE_ACL)) return -1;
- if (PyModule_AddIntConstant(m, "_COPYFILE_XATTR", COPYFILE_XATTR)) return -1;
- #endif
- #ifdef MS_WINDOWS
- if (PyModule_AddIntConstant(m, "_LOAD_LIBRARY_SEARCH_DEFAULT_DIRS", LOAD_LIBRARY_SEARCH_DEFAULT_DIRS)) return -1;
- if (PyModule_AddIntConstant(m, "_LOAD_LIBRARY_SEARCH_APPLICATION_DIR", LOAD_LIBRARY_SEARCH_APPLICATION_DIR)) return -1;
- if (PyModule_AddIntConstant(m, "_LOAD_LIBRARY_SEARCH_SYSTEM32", LOAD_LIBRARY_SEARCH_SYSTEM32)) return -1;
- if (PyModule_AddIntConstant(m, "_LOAD_LIBRARY_SEARCH_USER_DIRS", LOAD_LIBRARY_SEARCH_USER_DIRS)) return -1;
- if (PyModule_AddIntConstant(m, "_LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR", LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR)) return -1;
- #endif
- return 0;
- }
- #define PROBE(name, test) \
- static int name(void) \
- { \
- if (test) { \
- return 1; \
- } else { \
- return 0; \
- } \
- }
- #ifdef HAVE_FSTATAT
- PROBE(probe_fstatat, HAVE_FSTATAT_RUNTIME)
- #endif
- #ifdef HAVE_FACCESSAT
- PROBE(probe_faccessat, HAVE_FACCESSAT_RUNTIME)
- #endif
- #ifdef HAVE_FCHMODAT
- PROBE(probe_fchmodat, HAVE_FCHMODAT_RUNTIME)
- #endif
- #ifdef HAVE_FCHOWNAT
- PROBE(probe_fchownat, HAVE_FCHOWNAT_RUNTIME)
- #endif
- #ifdef HAVE_LINKAT
- PROBE(probe_linkat, HAVE_LINKAT_RUNTIME)
- #endif
- #ifdef HAVE_FDOPENDIR
- PROBE(probe_fdopendir, HAVE_FDOPENDIR_RUNTIME)
- #endif
- #ifdef HAVE_MKDIRAT
- PROBE(probe_mkdirat, HAVE_MKDIRAT_RUNTIME)
- #endif
- #ifdef HAVE_MKFIFOAT
- PROBE(probe_mkfifoat, HAVE_MKFIFOAT_RUNTIME)
- #endif
- #ifdef HAVE_MKNODAT
- PROBE(probe_mknodat, HAVE_MKNODAT_RUNTIME)
- #endif
- #ifdef HAVE_RENAMEAT
- PROBE(probe_renameat, HAVE_RENAMEAT_RUNTIME)
- #endif
- #ifdef HAVE_UNLINKAT
- PROBE(probe_unlinkat, HAVE_UNLINKAT_RUNTIME)
- #endif
- #ifdef HAVE_OPENAT
- PROBE(probe_openat, HAVE_OPENAT_RUNTIME)
- #endif
- #ifdef HAVE_READLINKAT
- PROBE(probe_readlinkat, HAVE_READLINKAT_RUNTIME)
- #endif
- #ifdef HAVE_SYMLINKAT
- PROBE(probe_symlinkat, HAVE_SYMLINKAT_RUNTIME)
- #endif
- #ifdef HAVE_FUTIMENS
- PROBE(probe_futimens, HAVE_FUTIMENS_RUNTIME)
- #endif
- #ifdef HAVE_UTIMENSAT
- PROBE(probe_utimensat, HAVE_UTIMENSAT_RUNTIME)
- #endif
- static const struct have_function {
- const char * const label;
- int (*probe)(void);
- } have_functions[] = {
- #ifdef HAVE_EVENTFD
- {"HAVE_EVENTFD", NULL},
- #endif
- #ifdef HAVE_FACCESSAT
- { "HAVE_FACCESSAT", probe_faccessat },
- #endif
- #ifdef HAVE_FCHDIR
- { "HAVE_FCHDIR", NULL },
- #endif
- #ifdef HAVE_FCHMOD
- { "HAVE_FCHMOD", NULL },
- #endif
- #ifdef HAVE_FCHMODAT
- { "HAVE_FCHMODAT", probe_fchmodat },
- #endif
- #ifdef HAVE_FCHOWN
- { "HAVE_FCHOWN", NULL },
- #endif
- #ifdef HAVE_FCHOWNAT
- { "HAVE_FCHOWNAT", probe_fchownat },
- #endif
- #ifdef HAVE_FEXECVE
- { "HAVE_FEXECVE", NULL },
- #endif
- #ifdef HAVE_FDOPENDIR
- { "HAVE_FDOPENDIR", probe_fdopendir },
- #endif
- #ifdef HAVE_FPATHCONF
- { "HAVE_FPATHCONF", NULL },
- #endif
- #ifdef HAVE_FSTATAT
- { "HAVE_FSTATAT", probe_fstatat },
- #endif
- #ifdef HAVE_FSTATVFS
- { "HAVE_FSTATVFS", NULL },
- #endif
- #if defined HAVE_FTRUNCATE || defined MS_WINDOWS
- { "HAVE_FTRUNCATE", NULL },
- #endif
- #ifdef HAVE_FUTIMENS
- { "HAVE_FUTIMENS", probe_futimens },
- #endif
- #ifdef HAVE_FUTIMES
- { "HAVE_FUTIMES", NULL },
- #endif
- #ifdef HAVE_FUTIMESAT
- { "HAVE_FUTIMESAT", NULL },
- #endif
- #ifdef HAVE_LINKAT
- { "HAVE_LINKAT", probe_linkat },
- #endif
- #ifdef HAVE_LCHFLAGS
- { "HAVE_LCHFLAGS", NULL },
- #endif
- #ifdef HAVE_LCHMOD
- { "HAVE_LCHMOD", NULL },
- #endif
- #ifdef HAVE_LCHOWN
- { "HAVE_LCHOWN", NULL },
- #endif
- #ifdef HAVE_LSTAT
- { "HAVE_LSTAT", NULL },
- #endif
- #ifdef HAVE_LUTIMES
- { "HAVE_LUTIMES", NULL },
- #endif
- #ifdef HAVE_MEMFD_CREATE
- { "HAVE_MEMFD_CREATE", NULL },
- #endif
- #ifdef HAVE_MKDIRAT
- { "HAVE_MKDIRAT", probe_mkdirat },
- #endif
- #ifdef HAVE_MKFIFOAT
- { "HAVE_MKFIFOAT", probe_mkfifoat },
- #endif
- #ifdef HAVE_MKNODAT
- { "HAVE_MKNODAT", probe_mknodat },
- #endif
- #ifdef HAVE_OPENAT
- { "HAVE_OPENAT", probe_openat },
- #endif
- #ifdef HAVE_READLINKAT
- { "HAVE_READLINKAT", probe_readlinkat },
- #endif
- #ifdef HAVE_RENAMEAT
- { "HAVE_RENAMEAT", probe_renameat },
- #endif
- #ifdef HAVE_SYMLINKAT
- { "HAVE_SYMLINKAT", probe_symlinkat },
- #endif
- #ifdef HAVE_UNLINKAT
- { "HAVE_UNLINKAT", probe_unlinkat },
- #endif
- #ifdef HAVE_UTIMENSAT
- { "HAVE_UTIMENSAT", probe_utimensat },
- #endif
- #ifdef MS_WINDOWS
- { "MS_WINDOWS", NULL },
- #endif
- { NULL, NULL }
- };
- static int
- posixmodule_exec(PyObject *m)
- {
- _posixstate *state = get_posix_state(m);
- #if defined(HAVE_PWRITEV)
- if (HAVE_PWRITEV_RUNTIME) {} else {
- PyObject* dct = PyModule_GetDict(m);
- if (dct == NULL) {
- return -1;
- }
- if (PyDict_DelItemString(dct, "pwritev") == -1) {
- PyErr_Clear();
- }
- if (PyDict_DelItemString(dct, "preadv") == -1) {
- PyErr_Clear();
- }
- }
- #endif
- /* Initialize environ dictionary */
- if (_PyModule_Add(m, "environ", convertenviron()) != 0) {
- return -1;
- }
- if (all_ins(m))
- return -1;
- if (setup_confname_tables(m))
- return -1;
- if (PyModule_AddObjectRef(m, "error", PyExc_OSError) < 0) {
- return -1;
- }
- #if defined(HAVE_WAITID) && !defined(__APPLE__)
- waitid_result_desc.name = MODNAME ".waitid_result";
- state->WaitidResultType = (PyObject *)PyStructSequence_NewType(&waitid_result_desc);
- if (PyModule_AddObjectRef(m, "waitid_result", state->WaitidResultType) < 0) {
- return -1;
- }
- #endif
- stat_result_desc.name = "os.stat_result"; /* see issue #19209 */
- stat_result_desc.fields[7].name = PyStructSequence_UnnamedField;
- stat_result_desc.fields[8].name = PyStructSequence_UnnamedField;
- stat_result_desc.fields[9].name = PyStructSequence_UnnamedField;
- state->StatResultType = (PyObject *)PyStructSequence_NewType(&stat_result_desc);
- if (PyModule_AddObjectRef(m, "stat_result", state->StatResultType) < 0) {
- return -1;
- }
- state->statresult_new_orig = ((PyTypeObject *)state->StatResultType)->tp_new;
- ((PyTypeObject *)state->StatResultType)->tp_new = statresult_new;
- statvfs_result_desc.name = "os.statvfs_result"; /* see issue #19209 */
- state->StatVFSResultType = (PyObject *)PyStructSequence_NewType(&statvfs_result_desc);
- if (PyModule_AddObjectRef(m, "statvfs_result", state->StatVFSResultType) < 0) {
- return -1;
- }
- #if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM)
- sched_param_desc.name = MODNAME ".sched_param";
- state->SchedParamType = (PyObject *)PyStructSequence_NewType(&sched_param_desc);
- if (PyModule_AddObjectRef(m, "sched_param", state->SchedParamType) < 0) {
- return -1;
- }
- ((PyTypeObject *)state->SchedParamType)->tp_new = os_sched_param;
- if (_PyType_AddMethod((PyTypeObject *)state->SchedParamType,
- &os_sched_param_reduce_method) < 0)
- {
- return -1;
- }
- PyType_Modified((PyTypeObject *)state->SchedParamType);
- #endif
- /* initialize TerminalSize_info */
- state->TerminalSizeType = (PyObject *)PyStructSequence_NewType(&TerminalSize_desc);
- if (PyModule_AddObjectRef(m, "terminal_size", state->TerminalSizeType) < 0) {
- return -1;
- }
- /* initialize scandir types */
- PyObject *ScandirIteratorType = PyType_FromModuleAndSpec(m, &ScandirIteratorType_spec, NULL);
- if (ScandirIteratorType == NULL) {
- return -1;
- }
- state->ScandirIteratorType = ScandirIteratorType;
- state->DirEntryType = PyType_FromModuleAndSpec(m, &DirEntryType_spec, NULL);
- if (PyModule_AddObjectRef(m, "DirEntry", state->DirEntryType) < 0) {
- return -1;
- }
- times_result_desc.name = MODNAME ".times_result";
- state->TimesResultType = (PyObject *)PyStructSequence_NewType(×_result_desc);
- if (PyModule_AddObjectRef(m, "times_result", state->TimesResultType) < 0) {
- return -1;
- }
- state->UnameResultType = (PyObject *)PyStructSequence_NewType(&uname_result_desc);
- if (PyModule_AddObjectRef(m, "uname_result", state->UnameResultType) < 0) {
- return -1;
- }
- if ((state->billion = PyLong_FromLong(1000000000)) == NULL)
- return -1;
- #if defined(HAVE_WAIT3) || defined(HAVE_WAIT4)
- state->struct_rusage = PyUnicode_InternFromString("struct_rusage");
- if (state->struct_rusage == NULL)
- return -1;
- #endif
- state->st_mode = PyUnicode_InternFromString("st_mode");
- if (state->st_mode == NULL)
- return -1;
- /* suppress "function not used" warnings */
- {
- int ignored;
- fd_specified("", -1);
- follow_symlinks_specified("", 1);
- dir_fd_and_follow_symlinks_invalid("chmod", DEFAULT_DIR_FD, 1);
- dir_fd_converter(Py_None, &ignored);
- dir_fd_unavailable(Py_None, &ignored);
- }
- /*
- * provide list of locally available functions
- * so os.py can populate support_* lists
- */
- PyObject *list = PyList_New(0);
- if (!list) {
- return -1;
- }
- for (const struct have_function *trace = have_functions; trace->label; trace++) {
- PyObject *unicode;
- if (trace->probe && !trace->probe()) continue;
- unicode = PyUnicode_DecodeASCII(trace->label, strlen(trace->label), NULL);
- if (!unicode)
- return -1;
- if (PyList_Append(list, unicode))
- return -1;
- Py_DECREF(unicode);
- }
- return _PyModule_Add(m, "_have_functions", list);
- }
- static PyModuleDef_Slot posixmodile_slots[] = {
- {Py_mod_exec, posixmodule_exec},
- {Py_mod_multiple_interpreters, Py_MOD_PER_INTERPRETER_GIL_SUPPORTED},
- {0, NULL}
- };
- static struct PyModuleDef posixmodule = {
- PyModuleDef_HEAD_INIT,
- .m_name = MODNAME,
- .m_doc = posix__doc__,
- .m_size = sizeof(_posixstate),
- .m_methods = posix_methods,
- .m_slots = posixmodile_slots,
- .m_traverse = _posix_traverse,
- .m_clear = _posix_clear,
- .m_free = _posix_free,
- };
- PyMODINIT_FUNC
- INITFUNC(void)
- {
- return PyModuleDef_Init(&posixmodule);
- }
- #ifdef __cplusplus
- }
- #endif
|