123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873687468756876687768786879688068816882688368846885688668876888688968906891689268936894689568966897689868996900690169026903690469056906690769086909691069116912691369146915691669176918691969206921692269236924692569266927692869296930693169326933693469356936693769386939694069416942694369446945694669476948694969506951695269536954695569566957695869596960696169626963696469656966696769686969697069716972697369746975697669776978697969806981698269836984698569866987698869896990699169926993699469956996699769986999700070017002700370047005700670077008700970107011701270137014701570167017701870197020702170227023702470257026702770287029703070317032703370347035703670377038703970407041704270437044704570467047704870497050705170527053705470557056705770587059706070617062706370647065706670677068706970707071707270737074707570767077707870797080708170827083708470857086708770887089709070917092709370947095709670977098709971007101710271037104710571067107710871097110711171127113711471157116711771187119712071217122712371247125712671277128712971307131713271337134713571367137713871397140714171427143714471457146714771487149715071517152715371547155715671577158715971607161716271637164716571667167716871697170717171727173717471757176717771787179718071817182718371847185718671877188718971907191719271937194719571967197719871997200720172027203720472057206720772087209721072117212721372147215721672177218721972207221722272237224722572267227722872297230723172327233723472357236723772387239724072417242724372447245724672477248724972507251725272537254725572567257725872597260726172627263726472657266726772687269727072717272727372747275727672777278727972807281728272837284728572867287728872897290729172927293729472957296729772987299730073017302730373047305730673077308730973107311731273137314731573167317731873197320732173227323732473257326732773287329733073317332733373347335733673377338733973407341734273437344734573467347734873497350735173527353735473557356735773587359736073617362736373647365736673677368736973707371737273737374737573767377737873797380738173827383738473857386738773887389739073917392739373947395739673977398739974007401740274037404740574067407740874097410741174127413741474157416741774187419742074217422742374247425742674277428742974307431743274337434743574367437743874397440744174427443744474457446744774487449745074517452745374547455745674577458745974607461746274637464746574667467746874697470747174727473747474757476747774787479748074817482748374847485748674877488748974907491749274937494749574967497749874997500750175027503750475057506750775087509751075117512751375147515751675177518751975207521752275237524752575267527752875297530753175327533753475357536753775387539754075417542754375447545754675477548754975507551755275537554755575567557755875597560756175627563756475657566756775687569757075717572757375747575757675777578757975807581758275837584758575867587758875897590759175927593759475957596759775987599760076017602760376047605760676077608760976107611761276137614761576167617761876197620762176227623762476257626762776287629763076317632763376347635763676377638763976407641764276437644764576467647764876497650765176527653765476557656765776587659766076617662766376647665766676677668766976707671767276737674767576767677767876797680768176827683768476857686768776887689769076917692769376947695769676977698769977007701770277037704770577067707770877097710771177127713771477157716771777187719772077217722772377247725772677277728772977307731773277337734773577367737773877397740774177427743774477457746774777487749775077517752775377547755775677577758775977607761776277637764776577667767776877697770777177727773777477757776777777787779778077817782778377847785778677877788778977907791779277937794779577967797779877997800780178027803780478057806780778087809781078117812781378147815781678177818781978207821782278237824782578267827782878297830783178327833783478357836783778387839784078417842784378447845784678477848784978507851785278537854785578567857785878597860786178627863786478657866786778687869787078717872787378747875787678777878787978807881788278837884788578867887788878897890789178927893789478957896789778987899790079017902790379047905790679077908790979107911791279137914791579167917791879197920792179227923792479257926792779287929793079317932793379347935793679377938793979407941794279437944794579467947794879497950795179527953795479557956795779587959796079617962796379647965796679677968796979707971797279737974797579767977797879797980798179827983798479857986798779887989799079917992799379947995799679977998799980008001800280038004800580068007800880098010801180128013801480158016801780188019802080218022802380248025802680278028802980308031803280338034803580368037803880398040804180428043804480458046804780488049805080518052805380548055805680578058805980608061806280638064806580668067806880698070807180728073807480758076807780788079808080818082808380848085808680878088808980908091809280938094809580968097809880998100810181028103810481058106810781088109811081118112811381148115811681178118811981208121812281238124812581268127812881298130813181328133813481358136813781388139814081418142814381448145814681478148814981508151815281538154815581568157815881598160816181628163816481658166816781688169817081718172817381748175817681778178817981808181818281838184818581868187818881898190819181928193819481958196819781988199820082018202820382048205820682078208820982108211821282138214821582168217821882198220822182228223822482258226822782288229823082318232823382348235823682378238823982408241824282438244824582468247824882498250825182528253825482558256825782588259826082618262826382648265826682678268826982708271827282738274827582768277827882798280828182828283828482858286828782888289829082918292829382948295829682978298829983008301830283038304830583068307830883098310831183128313831483158316831783188319832083218322832383248325832683278328832983308331833283338334833583368337833883398340834183428343834483458346834783488349835083518352835383548355835683578358835983608361836283638364836583668367836883698370837183728373837483758376837783788379838083818382838383848385838683878388838983908391839283938394839583968397839883998400840184028403840484058406840784088409841084118412841384148415841684178418841984208421842284238424842584268427842884298430843184328433843484358436843784388439844084418442844384448445844684478448844984508451845284538454845584568457845884598460846184628463846484658466846784688469847084718472847384748475847684778478847984808481848284838484848584868487848884898490849184928493849484958496849784988499850085018502850385048505850685078508850985108511851285138514851585168517851885198520852185228523852485258526852785288529853085318532853385348535853685378538853985408541854285438544854585468547854885498550855185528553855485558556855785588559856085618562856385648565856685678568856985708571857285738574857585768577857885798580858185828583858485858586858785888589859085918592859385948595859685978598859986008601860286038604860586068607860886098610861186128613861486158616861786188619862086218622862386248625862686278628862986308631863286338634863586368637863886398640864186428643864486458646864786488649865086518652865386548655865686578658865986608661866286638664866586668667866886698670867186728673867486758676867786788679868086818682868386848685868686878688868986908691869286938694869586968697869886998700870187028703870487058706870787088709871087118712871387148715871687178718871987208721872287238724872587268727872887298730873187328733873487358736873787388739874087418742874387448745874687478748874987508751875287538754875587568757875887598760876187628763876487658766876787688769877087718772877387748775877687778778877987808781878287838784878587868787878887898790879187928793879487958796879787988799880088018802880388048805880688078808880988108811881288138814881588168817881888198820882188228823882488258826882788288829883088318832883388348835883688378838883988408841884288438844884588468847884888498850885188528853885488558856885788588859886088618862886388648865886688678868886988708871887288738874887588768877887888798880888188828883888488858886888788888889889088918892889388948895889688978898889989008901890289038904890589068907890889098910891189128913891489158916891789188919892089218922892389248925892689278928892989308931893289338934893589368937893889398940894189428943894489458946894789488949895089518952895389548955895689578958895989608961896289638964896589668967896889698970897189728973897489758976897789788979898089818982898389848985898689878988898989908991899289938994899589968997899889999000900190029003900490059006900790089009901090119012901390149015901690179018901990209021902290239024902590269027902890299030903190329033903490359036903790389039904090419042904390449045904690479048904990509051905290539054905590569057905890599060906190629063906490659066906790689069907090719072907390749075907690779078907990809081908290839084908590869087908890899090909190929093909490959096909790989099910091019102910391049105910691079108910991109111911291139114911591169117911891199120912191229123912491259126912791289129913091319132913391349135913691379138913991409141914291439144914591469147914891499150915191529153915491559156915791589159916091619162916391649165916691679168916991709171917291739174917591769177917891799180918191829183918491859186918791889189919091919192919391949195919691979198919992009201920292039204920592069207920892099210921192129213921492159216921792189219922092219222922392249225922692279228922992309231923292339234923592369237923892399240924192429243924492459246924792489249925092519252925392549255925692579258925992609261926292639264926592669267926892699270927192729273927492759276927792789279928092819282928392849285928692879288928992909291929292939294929592969297929892999300930193029303930493059306930793089309931093119312931393149315931693179318931993209321932293239324932593269327932893299330933193329333933493359336933793389339934093419342934393449345934693479348934993509351935293539354935593569357935893599360936193629363936493659366936793689369937093719372937393749375937693779378937993809381938293839384938593869387938893899390939193929393939493959396939793989399940094019402940394049405940694079408940994109411941294139414941594169417941894199420942194229423942494259426942794289429943094319432943394349435943694379438943994409441944294439444944594469447944894499450945194529453945494559456945794589459946094619462946394649465946694679468946994709471947294739474947594769477947894799480948194829483948494859486948794889489949094919492949394949495949694979498949995009501950295039504950595069507950895099510951195129513951495159516951795189519952095219522952395249525952695279528952995309531953295339534953595369537953895399540954195429543954495459546954795489549955095519552955395549555955695579558955995609561956295639564956595669567956895699570957195729573957495759576957795789579958095819582958395849585958695879588958995909591959295939594959595969597959895999600960196029603960496059606960796089609961096119612961396149615961696179618961996209621962296239624962596269627962896299630963196329633963496359636963796389639964096419642964396449645964696479648964996509651965296539654965596569657965896599660966196629663966496659666966796689669967096719672967396749675967696779678967996809681968296839684968596869687968896899690969196929693969496959696969796989699970097019702970397049705970697079708970997109711971297139714971597169717971897199720972197229723972497259726972797289729973097319732973397349735973697379738973997409741974297439744974597469747974897499750975197529753975497559756975797589759976097619762976397649765976697679768976997709771977297739774977597769777977897799780978197829783978497859786978797889789979097919792979397949795979697979798979998009801980298039804980598069807980898099810981198129813981498159816981798189819982098219822982398249825982698279828982998309831983298339834983598369837983898399840984198429843984498459846984798489849985098519852985398549855985698579858985998609861986298639864986598669867986898699870987198729873987498759876987798789879988098819882988398849885988698879888988998909891989298939894989598969897989898999900990199029903990499059906990799089909991099119912991399149915991699179918991999209921992299239924992599269927992899299930993199329933993499359936993799389939994099419942994399449945994699479948994999509951995299539954995599569957995899599960996199629963996499659966996799689969997099719972997399749975997699779978997999809981998299839984998599869987998899899990999199929993999499959996999799989999100001000110002100031000410005100061000710008100091001010011100121001310014100151001610017100181001910020100211002210023100241002510026100271002810029100301003110032100331003410035100361003710038100391004010041100421004310044100451004610047100481004910050100511005210053100541005510056100571005810059100601006110062100631006410065100661006710068100691007010071100721007310074100751007610077100781007910080100811008210083100841008510086100871008810089100901009110092100931009410095100961009710098100991010010101101021010310104101051010610107101081010910110101111011210113101141011510116101171011810119101201012110122101231012410125101261012710128101291013010131101321013310134101351013610137101381013910140101411014210143101441014510146101471014810149101501015110152101531015410155101561015710158101591016010161101621016310164101651016610167101681016910170101711017210173101741017510176101771017810179101801018110182101831018410185101861018710188101891019010191101921019310194101951019610197101981019910200102011020210203102041020510206102071020810209102101021110212102131021410215102161021710218102191022010221102221022310224102251022610227102281022910230102311023210233102341023510236102371023810239102401024110242102431024410245102461024710248102491025010251102521025310254102551025610257102581025910260102611026210263102641026510266102671026810269102701027110272102731027410275102761027710278102791028010281102821028310284102851028610287102881028910290102911029210293102941029510296102971029810299103001030110302103031030410305103061030710308103091031010311103121031310314103151031610317103181031910320103211032210323103241032510326103271032810329103301033110332103331033410335103361033710338103391034010341103421034310344103451034610347103481034910350103511035210353103541035510356103571035810359103601036110362103631036410365103661036710368103691037010371103721037310374103751037610377103781037910380103811038210383103841038510386103871038810389103901039110392103931039410395103961039710398103991040010401104021040310404104051040610407104081040910410104111041210413104141041510416104171041810419104201042110422104231042410425104261042710428104291043010431104321043310434104351043610437104381043910440104411044210443104441044510446104471044810449104501045110452104531045410455104561045710458104591046010461104621046310464104651046610467104681046910470104711047210473104741047510476104771047810479104801048110482104831048410485104861048710488104891049010491104921049310494104951049610497104981049910500105011050210503105041050510506105071050810509105101051110512105131051410515105161051710518105191052010521105221052310524105251052610527105281052910530105311053210533105341053510536105371053810539105401054110542105431054410545105461054710548105491055010551105521055310554105551055610557105581055910560105611056210563105641056510566105671056810569105701057110572105731057410575105761057710578105791058010581105821058310584105851058610587105881058910590105911059210593105941059510596105971059810599106001060110602106031060410605106061060710608106091061010611106121061310614106151061610617106181061910620106211062210623106241062510626106271062810629106301063110632106331063410635106361063710638106391064010641106421064310644106451064610647106481064910650106511065210653106541065510656106571065810659106601066110662106631066410665106661066710668106691067010671106721067310674106751067610677106781067910680106811068210683106841068510686106871068810689106901069110692106931069410695106961069710698106991070010701107021070310704107051070610707107081070910710107111071210713107141071510716107171071810719107201072110722107231072410725107261072710728107291073010731107321073310734107351073610737107381073910740107411074210743107441074510746107471074810749107501075110752107531075410755107561075710758107591076010761107621076310764107651076610767107681076910770107711077210773107741077510776107771077810779107801078110782107831078410785107861078710788107891079010791107921079310794107951079610797107981079910800108011080210803108041080510806108071080810809108101081110812108131081410815108161081710818108191082010821108221082310824108251082610827108281082910830108311083210833108341083510836108371083810839108401084110842108431084410845108461084710848108491085010851108521085310854108551085610857108581085910860108611086210863108641086510866108671086810869108701087110872108731087410875108761087710878108791088010881108821088310884108851088610887108881088910890108911089210893108941089510896108971089810899109001090110902109031090410905109061090710908109091091010911109121091310914109151091610917109181091910920109211092210923109241092510926109271092810929109301093110932109331093410935109361093710938109391094010941109421094310944109451094610947109481094910950109511095210953109541095510956109571095810959109601096110962109631096410965109661096710968109691097010971109721097310974109751097610977109781097910980109811098210983109841098510986109871098810989109901099110992109931099410995109961099710998109991100011001110021100311004110051100611007110081100911010110111101211013110141101511016110171101811019110201102111022110231102411025110261102711028110291103011031110321103311034110351103611037110381103911040110411104211043110441104511046110471104811049110501105111052110531105411055110561105711058110591106011061110621106311064110651106611067110681106911070110711107211073110741107511076110771107811079110801108111082110831108411085110861108711088110891109011091110921109311094110951109611097110981109911100111011110211103111041110511106111071110811109111101111111112111131111411115111161111711118111191112011121111221112311124111251112611127111281112911130111311113211133111341113511136111371113811139111401114111142111431114411145111461114711148111491115011151111521115311154111551115611157111581115911160111611116211163111641116511166111671116811169111701117111172111731117411175111761117711178111791118011181111821118311184111851118611187111881118911190111911119211193111941119511196111971119811199112001120111202112031120411205112061120711208112091121011211112121121311214112151121611217112181121911220112211122211223112241122511226112271122811229112301123111232112331123411235112361123711238112391124011241112421124311244112451124611247112481124911250112511125211253112541125511256112571125811259112601126111262112631126411265112661126711268112691127011271112721127311274112751127611277112781127911280112811128211283112841128511286112871128811289112901129111292112931129411295112961129711298112991130011301113021130311304113051130611307113081130911310113111131211313113141131511316113171131811319113201132111322113231132411325113261132711328113291133011331113321133311334113351133611337113381133911340113411134211343113441134511346113471134811349113501135111352113531135411355113561135711358113591136011361113621136311364113651136611367113681136911370113711137211373113741137511376113771137811379113801138111382113831138411385113861138711388113891139011391113921139311394113951139611397113981139911400114011140211403114041140511406114071140811409114101141111412114131141411415114161141711418114191142011421114221142311424114251142611427114281142911430114311143211433114341143511436114371143811439114401144111442114431144411445114461144711448114491145011451114521145311454114551145611457114581145911460114611146211463114641146511466114671146811469114701147111472114731147411475114761147711478114791148011481114821148311484114851148611487114881148911490114911149211493114941149511496114971149811499115001150111502115031150411505115061150711508115091151011511115121151311514115151151611517115181151911520115211152211523115241152511526115271152811529115301153111532115331153411535115361153711538115391154011541115421154311544115451154611547115481154911550115511155211553115541155511556115571155811559115601156111562115631156411565115661156711568115691157011571115721157311574115751157611577115781157911580115811158211583115841158511586115871158811589115901159111592115931159411595115961159711598115991160011601116021160311604116051160611607116081160911610116111161211613116141161511616116171161811619116201162111622116231162411625116261162711628116291163011631116321163311634116351163611637116381163911640116411164211643116441164511646116471164811649116501165111652116531165411655116561165711658116591166011661116621166311664116651166611667116681166911670116711167211673116741167511676116771167811679116801168111682116831168411685116861168711688116891169011691116921169311694116951169611697116981169911700117011170211703117041170511706117071170811709117101171111712117131171411715117161171711718117191172011721117221172311724117251172611727117281172911730117311173211733117341173511736117371173811739117401174111742117431174411745117461174711748117491175011751117521175311754117551175611757117581175911760117611176211763117641176511766117671176811769117701177111772117731177411775117761177711778117791178011781117821178311784117851178611787117881178911790117911179211793117941179511796117971179811799118001180111802118031180411805118061180711808118091181011811118121181311814118151181611817118181181911820118211182211823118241182511826118271182811829118301183111832118331183411835118361183711838118391184011841118421184311844118451184611847118481184911850118511185211853118541185511856118571185811859118601186111862118631186411865118661186711868118691187011871118721187311874118751187611877118781187911880118811188211883118841188511886118871188811889118901189111892118931189411895118961189711898118991190011901119021190311904119051190611907119081190911910119111191211913119141191511916119171191811919119201192111922119231192411925119261192711928119291193011931119321193311934119351193611937119381193911940119411194211943119441194511946119471194811949119501195111952119531195411955119561195711958119591196011961119621196311964119651196611967119681196911970119711197211973119741197511976119771197811979119801198111982119831198411985119861198711988119891199011991119921199311994119951199611997119981199912000120011200212003120041200512006120071200812009120101201112012120131201412015120161201712018120191202012021120221202312024120251202612027120281202912030120311203212033120341203512036120371203812039120401204112042120431204412045120461204712048120491205012051120521205312054120551205612057120581205912060120611206212063120641206512066120671206812069120701207112072120731207412075120761207712078120791208012081120821208312084120851208612087120881208912090120911209212093120941209512096120971209812099121001210112102121031210412105121061210712108121091211012111121121211312114121151211612117121181211912120121211212212123121241212512126121271212812129121301213112132121331213412135121361213712138121391214012141121421214312144121451214612147121481214912150121511215212153121541215512156121571215812159121601216112162121631216412165121661216712168121691217012171121721217312174121751217612177121781217912180121811218212183121841218512186121871218812189121901219112192121931219412195121961219712198121991220012201122021220312204122051220612207122081220912210122111221212213122141221512216122171221812219122201222112222122231222412225122261222712228122291223012231122321223312234122351223612237122381223912240122411224212243122441224512246122471224812249122501225112252122531225412255122561225712258122591226012261122621226312264122651226612267122681226912270122711227212273122741227512276122771227812279122801228112282122831228412285122861228712288122891229012291122921229312294122951229612297122981229912300123011230212303123041230512306123071230812309123101231112312123131231412315123161231712318123191232012321123221232312324123251232612327123281232912330123311233212333123341233512336123371233812339123401234112342123431234412345123461234712348123491235012351123521235312354123551235612357123581235912360123611236212363123641236512366123671236812369123701237112372123731237412375123761237712378123791238012381123821238312384123851238612387123881238912390123911239212393123941239512396123971239812399124001240112402124031240412405124061240712408124091241012411124121241312414124151241612417124181241912420124211242212423124241242512426124271242812429124301243112432124331243412435124361243712438124391244012441124421244312444124451244612447124481244912450124511245212453124541245512456124571245812459124601246112462124631246412465124661246712468124691247012471124721247312474124751247612477124781247912480124811248212483124841248512486124871248812489124901249112492124931249412495124961249712498124991250012501125021250312504125051250612507125081250912510125111251212513125141251512516125171251812519125201252112522125231252412525125261252712528125291253012531125321253312534125351253612537125381253912540125411254212543125441254512546125471254812549125501255112552125531255412555125561255712558125591256012561125621256312564125651256612567125681256912570125711257212573125741257512576125771257812579125801258112582125831258412585125861258712588125891259012591125921259312594125951259612597125981259912600126011260212603126041260512606126071260812609126101261112612126131261412615126161261712618126191262012621126221262312624126251262612627126281262912630126311263212633126341263512636126371263812639126401264112642126431264412645126461264712648126491265012651126521265312654126551265612657126581265912660126611266212663126641266512666126671266812669126701267112672126731267412675126761267712678126791268012681126821268312684126851268612687126881268912690126911269212693126941269512696126971269812699127001270112702127031270412705127061270712708127091271012711127121271312714127151271612717127181271912720127211272212723127241272512726127271272812729127301273112732127331273412735127361273712738127391274012741127421274312744127451274612747127481274912750127511275212753127541275512756127571275812759127601276112762127631276412765127661276712768127691277012771127721277312774127751277612777127781277912780127811278212783127841278512786127871278812789127901279112792127931279412795127961279712798127991280012801128021280312804128051280612807128081280912810128111281212813128141281512816128171281812819128201282112822128231282412825128261282712828128291283012831128321283312834128351283612837128381283912840128411284212843128441284512846128471284812849128501285112852128531285412855128561285712858128591286012861128621286312864128651286612867128681286912870128711287212873128741287512876128771287812879128801288112882128831288412885128861288712888128891289012891128921289312894128951289612897128981289912900129011290212903129041290512906129071290812909129101291112912129131291412915129161291712918129191292012921129221292312924129251292612927129281292912930129311293212933129341293512936129371293812939129401294112942129431294412945129461294712948129491295012951129521295312954129551295612957129581295912960129611296212963129641296512966129671296812969129701297112972129731297412975129761297712978129791298012981129821298312984129851298612987129881298912990129911299212993129941299512996129971299812999130001300113002130031300413005130061300713008130091301013011130121301313014130151301613017130181301913020130211302213023130241302513026130271302813029130301303113032130331303413035130361303713038130391304013041130421304313044130451304613047130481304913050130511305213053130541305513056130571305813059130601306113062130631306413065130661306713068130691307013071130721307313074130751307613077130781307913080130811308213083130841308513086130871308813089130901309113092130931309413095130961309713098130991310013101131021310313104131051310613107131081310913110131111311213113131141311513116131171311813119131201312113122131231312413125131261312713128131291313013131131321313313134131351313613137131381313913140131411314213143131441314513146131471314813149131501315113152131531315413155131561315713158131591316013161131621316313164131651316613167131681316913170131711317213173131741317513176131771317813179131801318113182131831318413185131861318713188131891319013191131921319313194131951319613197131981319913200132011320213203132041320513206132071320813209132101321113212132131321413215132161321713218132191322013221132221322313224132251322613227132281322913230132311323213233132341323513236132371323813239132401324113242132431324413245132461324713248132491325013251132521325313254132551325613257132581325913260132611326213263132641326513266132671326813269132701327113272132731327413275132761327713278132791328013281132821328313284132851328613287132881328913290132911329213293132941329513296132971329813299133001330113302133031330413305133061330713308133091331013311133121331313314133151331613317133181331913320133211332213323133241332513326133271332813329133301333113332133331333413335133361333713338133391334013341133421334313344133451334613347133481334913350133511335213353133541335513356133571335813359133601336113362133631336413365133661336713368133691337013371133721337313374133751337613377133781337913380133811338213383133841338513386133871338813389133901339113392133931339413395133961339713398133991340013401134021340313404134051340613407134081340913410134111341213413134141341513416134171341813419134201342113422134231342413425134261342713428134291343013431134321343313434134351343613437134381343913440134411344213443134441344513446134471344813449134501345113452134531345413455134561345713458134591346013461134621346313464134651346613467134681346913470134711347213473134741347513476134771347813479134801348113482134831348413485134861348713488134891349013491134921349313494134951349613497134981349913500135011350213503135041350513506135071350813509135101351113512135131351413515135161351713518135191352013521135221352313524135251352613527135281352913530135311353213533135341353513536135371353813539135401354113542135431354413545135461354713548135491355013551135521355313554135551355613557135581355913560135611356213563135641356513566135671356813569135701357113572135731357413575135761357713578135791358013581135821358313584135851358613587135881358913590135911359213593135941359513596135971359813599136001360113602136031360413605136061360713608136091361013611136121361313614136151361613617136181361913620136211362213623136241362513626136271362813629136301363113632136331363413635136361363713638136391364013641136421364313644136451364613647136481364913650136511365213653136541365513656136571365813659136601366113662136631366413665136661366713668136691367013671136721367313674136751367613677136781367913680136811368213683136841368513686136871368813689136901369113692136931369413695136961369713698136991370013701137021370313704137051370613707137081370913710137111371213713137141371513716137171371813719137201372113722137231372413725137261372713728137291373013731137321373313734137351373613737137381373913740137411374213743137441374513746137471374813749137501375113752137531375413755137561375713758137591376013761137621376313764137651376613767137681376913770137711377213773137741377513776137771377813779137801378113782137831378413785137861378713788137891379013791137921379313794137951379613797137981379913800138011380213803138041380513806138071380813809138101381113812138131381413815138161381713818138191382013821138221382313824138251382613827138281382913830138311383213833138341383513836138371383813839138401384113842138431384413845138461384713848138491385013851138521385313854138551385613857138581385913860138611386213863138641386513866138671386813869138701387113872138731387413875138761387713878138791388013881138821388313884138851388613887138881388913890138911389213893138941389513896138971389813899139001390113902139031390413905139061390713908139091391013911139121391313914139151391613917139181391913920139211392213923139241392513926139271392813929139301393113932139331393413935139361393713938139391394013941139421394313944139451394613947139481394913950139511395213953139541395513956139571395813959139601396113962139631396413965139661396713968139691397013971139721397313974139751397613977139781397913980139811398213983139841398513986139871398813989139901399113992139931399413995139961399713998139991400014001140021400314004140051400614007140081400914010140111401214013140141401514016140171401814019140201402114022140231402414025140261402714028140291403014031140321403314034140351403614037140381403914040140411404214043140441404514046140471404814049140501405114052140531405414055140561405714058140591406014061140621406314064140651406614067140681406914070140711407214073140741407514076140771407814079140801408114082140831408414085140861408714088140891409014091140921409314094140951409614097140981409914100141011410214103141041410514106141071410814109141101411114112141131411414115141161411714118141191412014121141221412314124141251412614127141281412914130141311413214133141341413514136141371413814139141401414114142141431414414145141461414714148141491415014151141521415314154141551415614157141581415914160141611416214163141641416514166141671416814169141701417114172141731417414175141761417714178141791418014181141821418314184141851418614187141881418914190141911419214193141941419514196141971419814199142001420114202142031420414205142061420714208142091421014211142121421314214142151421614217142181421914220142211422214223142241422514226142271422814229142301423114232142331423414235142361423714238142391424014241142421424314244142451424614247142481424914250142511425214253142541425514256142571425814259142601426114262142631426414265142661426714268142691427014271142721427314274142751427614277142781427914280142811428214283142841428514286142871428814289142901429114292142931429414295142961429714298142991430014301143021430314304143051430614307143081430914310143111431214313143141431514316143171431814319143201432114322143231432414325143261432714328143291433014331143321433314334143351433614337143381433914340143411434214343143441434514346143471434814349143501435114352143531435414355143561435714358143591436014361143621436314364143651436614367143681436914370143711437214373143741437514376143771437814379143801438114382143831438414385143861438714388143891439014391143921439314394143951439614397143981439914400144011440214403144041440514406144071440814409144101441114412144131441414415144161441714418144191442014421144221442314424144251442614427144281442914430144311443214433144341443514436144371443814439144401444114442144431444414445144461444714448144491445014451144521445314454144551445614457144581445914460144611446214463144641446514466144671446814469144701447114472144731447414475144761447714478144791448014481144821448314484144851448614487144881448914490144911449214493144941449514496144971449814499145001450114502145031450414505145061450714508145091451014511145121451314514145151451614517145181451914520145211452214523145241452514526145271452814529145301453114532145331453414535145361453714538145391454014541145421454314544145451454614547145481454914550145511455214553145541455514556145571455814559145601456114562145631456414565145661456714568145691457014571145721457314574145751457614577145781457914580145811458214583145841458514586145871458814589145901459114592145931459414595145961459714598145991460014601146021460314604146051460614607146081460914610146111461214613146141461514616146171461814619146201462114622146231462414625146261462714628146291463014631146321463314634146351463614637146381463914640146411464214643146441464514646146471464814649146501465114652146531465414655146561465714658146591466014661146621466314664146651466614667146681466914670146711467214673146741467514676146771467814679146801468114682146831468414685146861468714688146891469014691146921469314694146951469614697146981469914700147011470214703147041470514706147071470814709147101471114712147131471414715147161471714718147191472014721147221472314724147251472614727147281472914730147311473214733147341473514736147371473814739147401474114742147431474414745147461474714748147491475014751147521475314754147551475614757147581475914760147611476214763147641476514766147671476814769147701477114772147731477414775147761477714778147791478014781147821478314784147851478614787147881478914790147911479214793147941479514796147971479814799148001480114802148031480414805148061480714808148091481014811148121481314814148151481614817148181481914820148211482214823148241482514826148271482814829148301483114832148331483414835148361483714838148391484014841148421484314844148451484614847148481484914850148511485214853148541485514856148571485814859148601486114862148631486414865148661486714868148691487014871148721487314874148751487614877148781487914880148811488214883148841488514886148871488814889148901489114892148931489414895148961489714898148991490014901149021490314904149051490614907149081490914910149111491214913149141491514916149171491814919149201492114922149231492414925149261492714928149291493014931149321493314934149351493614937149381493914940149411494214943149441494514946149471494814949149501495114952149531495414955149561495714958149591496014961149621496314964149651496614967149681496914970149711497214973149741497514976149771497814979149801498114982149831498414985149861498714988149891499014991149921499314994149951499614997149981499915000150011500215003150041500515006150071500815009150101501115012150131501415015150161501715018150191502015021150221502315024150251502615027150281502915030150311503215033150341503515036150371503815039150401504115042150431504415045150461504715048150491505015051150521505315054150551505615057150581505915060150611506215063150641506515066150671506815069150701507115072150731507415075150761507715078150791508015081150821508315084150851508615087150881508915090150911509215093150941509515096150971509815099151001510115102151031510415105151061510715108151091511015111151121511315114151151511615117151181511915120151211512215123151241512515126151271512815129151301513115132151331513415135151361513715138151391514015141151421514315144151451514615147151481514915150151511515215153151541515515156151571515815159151601516115162151631516415165151661516715168151691517015171151721517315174151751517615177151781517915180151811518215183151841518515186151871518815189151901519115192151931519415195151961519715198151991520015201152021520315204152051520615207152081520915210152111521215213152141521515216152171521815219152201522115222152231522415225152261522715228152291523015231152321523315234152351523615237152381523915240152411524215243152441524515246152471524815249152501525115252152531525415255152561525715258152591526015261152621526315264152651526615267152681526915270152711527215273152741527515276152771527815279152801528115282152831528415285152861528715288152891529015291152921529315294152951529615297152981529915300153011530215303153041530515306153071530815309153101531115312153131531415315153161531715318153191532015321153221532315324153251532615327153281532915330153311533215333153341533515336153371533815339153401534115342153431534415345153461534715348153491535015351153521535315354153551535615357153581535915360153611536215363153641536515366153671536815369153701537115372153731537415375153761537715378153791538015381153821538315384153851538615387153881538915390153911539215393153941539515396153971539815399154001540115402154031540415405154061540715408154091541015411154121541315414154151541615417154181541915420154211542215423154241542515426154271542815429154301543115432154331543415435154361543715438154391544015441154421544315444154451544615447154481544915450154511545215453154541545515456154571545815459154601546115462154631546415465154661546715468154691547015471154721547315474154751547615477154781547915480154811548215483154841548515486154871548815489154901549115492154931549415495154961549715498154991550015501155021550315504155051550615507155081550915510155111551215513155141551515516155171551815519155201552115522155231552415525155261552715528155291553015531155321553315534155351553615537155381553915540155411554215543155441554515546155471554815549155501555115552155531555415555155561555715558155591556015561155621556315564155651556615567155681556915570155711557215573155741557515576155771557815579155801558115582155831558415585155861558715588155891559015591155921559315594155951559615597155981559915600156011560215603156041560515606156071560815609156101561115612156131561415615156161561715618156191562015621156221562315624156251562615627156281562915630156311563215633156341563515636156371563815639156401564115642156431564415645156461564715648156491565015651156521565315654156551565615657156581565915660156611566215663156641566515666156671566815669156701567115672156731567415675156761567715678156791568015681156821568315684156851568615687156881568915690156911569215693156941569515696156971569815699157001570115702157031570415705157061570715708157091571015711157121571315714157151571615717157181571915720157211572215723157241572515726157271572815729157301573115732157331573415735157361573715738157391574015741157421574315744157451574615747157481574915750157511575215753157541575515756157571575815759157601576115762157631576415765157661576715768157691577015771157721577315774157751577615777157781577915780157811578215783157841578515786157871578815789157901579115792157931579415795157961579715798157991580015801158021580315804158051580615807158081580915810158111581215813158141581515816158171581815819158201582115822158231582415825158261582715828158291583015831158321583315834158351583615837158381583915840158411584215843158441584515846158471584815849158501585115852158531585415855158561585715858158591586015861158621586315864158651586615867158681586915870158711587215873158741587515876158771587815879158801588115882158831588415885158861588715888158891589015891158921589315894158951589615897158981589915900159011590215903159041590515906159071590815909159101591115912159131591415915159161591715918159191592015921159221592315924159251592615927159281592915930159311593215933159341593515936159371593815939159401594115942159431594415945159461594715948159491595015951159521595315954159551595615957159581595915960159611596215963159641596515966159671596815969159701597115972159731597415975159761597715978159791598015981159821598315984159851598615987159881598915990159911599215993159941599515996159971599815999160001600116002160031600416005160061600716008160091601016011160121601316014160151601616017160181601916020160211602216023160241602516026160271602816029160301603116032160331603416035160361603716038160391604016041160421604316044160451604616047160481604916050160511605216053160541605516056160571605816059160601606116062160631606416065160661606716068160691607016071160721607316074160751607616077160781607916080160811608216083160841608516086160871608816089160901609116092160931609416095160961609716098160991610016101161021610316104161051610616107161081610916110161111611216113161141611516116161171611816119161201612116122161231612416125161261612716128161291613016131161321613316134161351613616137161381613916140161411614216143161441614516146161471614816149161501615116152161531615416155161561615716158161591616016161161621616316164161651616616167161681616916170161711617216173161741617516176161771617816179161801618116182161831618416185161861618716188161891619016191161921619316194161951619616197161981619916200162011620216203162041620516206162071620816209162101621116212162131621416215162161621716218162191622016221162221622316224162251622616227162281622916230162311623216233162341623516236162371623816239162401624116242162431624416245162461624716248162491625016251162521625316254162551625616257162581625916260162611626216263162641626516266162671626816269162701627116272162731627416275162761627716278162791628016281162821628316284162851628616287162881628916290162911629216293162941629516296162971629816299163001630116302163031630416305163061630716308163091631016311163121631316314163151631616317163181631916320163211632216323163241632516326163271632816329163301633116332163331633416335163361633716338163391634016341163421634316344163451634616347163481634916350163511635216353163541635516356163571635816359163601636116362163631636416365163661636716368163691637016371163721637316374163751637616377163781637916380163811638216383163841638516386163871638816389163901639116392163931639416395163961639716398163991640016401164021640316404164051640616407164081640916410164111641216413164141641516416164171641816419164201642116422164231642416425164261642716428164291643016431164321643316434164351643616437164381643916440164411644216443164441644516446164471644816449164501645116452164531645416455164561645716458164591646016461164621646316464164651646616467164681646916470164711647216473164741647516476164771647816479164801648116482164831648416485164861648716488164891649016491164921649316494164951649616497164981649916500165011650216503165041650516506165071650816509165101651116512165131651416515165161651716518165191652016521165221652316524165251652616527165281652916530165311653216533165341653516536165371653816539165401654116542165431654416545165461654716548165491655016551165521655316554165551655616557165581655916560165611656216563165641656516566165671656816569165701657116572165731657416575165761657716578165791658016581165821658316584165851658616587165881658916590165911659216593165941659516596165971659816599166001660116602166031660416605166061660716608166091661016611166121661316614166151661616617166181661916620166211662216623166241662516626166271662816629166301663116632166331663416635166361663716638166391664016641166421664316644166451664616647166481664916650166511665216653166541665516656166571665816659166601666116662166631666416665166661666716668166691667016671166721667316674166751667616677166781667916680166811668216683166841668516686166871668816689166901669116692166931669416695166961669716698166991670016701167021670316704167051670616707167081670916710167111671216713167141671516716167171671816719167201672116722167231672416725167261672716728167291673016731167321673316734167351673616737167381673916740167411674216743167441674516746167471674816749167501675116752167531675416755167561675716758167591676016761167621676316764167651676616767167681676916770167711677216773167741677516776167771677816779167801678116782167831678416785167861678716788167891679016791167921679316794167951679616797167981679916800168011680216803168041680516806168071680816809168101681116812168131681416815168161681716818168191682016821168221682316824168251682616827168281682916830168311683216833168341683516836168371683816839168401684116842168431684416845168461684716848168491685016851168521685316854168551685616857168581685916860168611686216863168641686516866168671686816869168701687116872168731687416875168761687716878168791688016881168821688316884168851688616887168881688916890168911689216893168941689516896168971689816899169001690116902169031690416905169061690716908169091691016911169121691316914169151691616917169181691916920169211692216923169241692516926169271692816929169301693116932169331693416935169361693716938169391694016941169421694316944169451694616947169481694916950169511695216953169541695516956169571695816959169601696116962169631696416965169661696716968169691697016971169721697316974169751697616977169781697916980169811698216983169841698516986169871698816989169901699116992169931699416995169961699716998169991700017001170021700317004170051700617007170081700917010170111701217013170141701517016170171701817019170201702117022170231702417025170261702717028170291703017031170321703317034170351703617037170381703917040170411704217043170441704517046170471704817049 |
- /* 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_moduleobject.h" // _PyModule_GetState()
- #include "pycore_object.h" // _PyObject_LookupSpecial()
- #include "pycore_pystate.h" // _PyInterpreterState_GET()
- #include "pycore_signal.h" // Py_NSIG
- #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
- # 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
- # 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 */
- #define _PyLong_FromDev PyLong_FromLongLong
- #if (defined(HAVE_MKNOD) && defined(HAVE_MAKEDEV)) || defined(HAVE_DEVICE_MACROS)
- static int
- _Py_Dev_Converter(PyObject *obj, void *p)
- {
- *((dev_t *)p) = PyLong_AsUnsignedLongLong(obj);
- if (PyErr_Occurred())
- return 0;
- 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 the platform:
- *
- * * On Windows, 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.
- *
- * * On all other platforms, strings are encoded
- * to bytes using PyUnicode_FSConverter, then we
- * extract the char * from the bytes object 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.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
- * and was not encoded. (Only used on Windows.)
- * path.narrow
- * Points to the path if it was expressed as bytes,
- * or it was Unicode and was encoded to bytes. (On Windows,
- * is a non-zero integer if the path was expressed as bytes.
- * The type is deliberately incompatible to prevent misuse.)
- * 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.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 {
- const char *function_name;
- const char *argument_name;
- int nullable;
- int allow_fd;
- const wchar_t *wide;
- #ifdef MS_WINDOWS
- BOOL narrow;
- #else
- const char *narrow;
- #endif
- int fd;
- Py_ssize_t length;
- PyObject *object;
- PyObject *cleanup;
- } path_t;
- #ifdef MS_WINDOWS
- #define PATH_T_INITIALIZE(function_name, argument_name, nullable, allow_fd) \
- {function_name, argument_name, nullable, allow_fd, NULL, FALSE, -1, 0, NULL, NULL}
- #else
- #define PATH_T_INITIALIZE(function_name, argument_name, nullable, allow_fd) \
- {function_name, argument_name, nullable, allow_fd, NULL, NULL, -1, 0, NULL, NULL}
- #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;
- #ifdef MS_WINDOWS
- PyObject *wo = NULL;
- wchar_t *wide = NULL;
- #endif
- #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;
- #ifdef MS_WINDOWS
- path->narrow = FALSE;
- #else
- path->narrow = NULL;
- #endif
- 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) {
- #ifdef MS_WINDOWS
- wide = PyUnicode_AsWideCharString(o, &length);
- if (!wide) {
- goto error_exit;
- }
- if (length > 32767) {
- FORMAT_EXCEPTION(PyExc_ValueError, "%s too long for Windows");
- goto error_exit;
- }
- if (wcslen(wide) != length) {
- FORMAT_EXCEPTION(PyExc_ValueError, "embedded null character in %s");
- goto error_exit;
- }
- path->wide = wide;
- path->narrow = FALSE;
- path->fd = -1;
- wide = NULL;
- goto success_exit;
- #else
- if (!PyUnicode_FSConverter(o, &bytes)) {
- goto error_exit;
- }
- #endif
- }
- else if (is_bytes) {
- bytes = Py_NewRef(o);
- }
- else if (is_index) {
- if (!_fd_converter(o, &path->fd)) {
- goto error_exit;
- }
- path->wide = NULL;
- #ifdef MS_WINDOWS
- path->narrow = FALSE;
- #else
- path->narrow = NULL;
- #endif
- 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 ((size_t)length != strlen(narrow)) {
- FORMAT_EXCEPTION(PyExc_ValueError, "embedded null character in %s");
- goto error_exit;
- }
- #ifdef MS_WINDOWS
- wo = PyUnicode_DecodeFSDefaultAndSize(
- narrow,
- length
- );
- if (!wo) {
- goto error_exit;
- }
- wide = PyUnicode_AsWideCharString(wo, &length);
- Py_DECREF(wo);
- if (!wide) {
- goto error_exit;
- }
- if (length > 32767) {
- FORMAT_EXCEPTION(PyExc_ValueError, "%s too long for Windows");
- goto error_exit;
- }
- if (wcslen(wide) != length) {
- FORMAT_EXCEPTION(PyExc_ValueError, "embedded null character in %s");
- goto error_exit;
- }
- path->wide = wide;
- path->narrow = TRUE;
- 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;
- }
- #endif
- path->fd = -1;
- success_exit:
- path->length = length;
- path->object = o;
- return Py_CLEANUP_SUPPORTED;
- error_exit:
- Py_XDECREF(o);
- Py_XDECREF(bytes);
- #ifdef MS_WINDOWS
- PyMem_Free(wide);
- #endif
- return 0;
- }
- 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)
- #ifndef MS_WINDOWS
- && !path->narrow
- #endif
- ) {
- 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, nullable=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.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
- self.c_default = 'PATH_T_INITIALIZE("{}", "{}", {}, {})'.format(
- self.function.name,
- self.name,
- strify(self.nullable),
- 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=3338733161aa7879]*/
- /*[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;
- 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;
- } else {
- po_wchars = path->wide;
- len = wcslen(path->wide);
- }
- /* 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 (path->narrow && 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 (path->narrow) {
- 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 && path->narrow) {
- 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 (path->narrow)
- 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;
- }
- /*[clinic input]
- os._path_isdir
- s: 'O'
- Return true if the pathname refers to an existing directory.
- [clinic start generated code]*/
- static PyObject *
- os__path_isdir_impl(PyObject *module, PyObject *s)
- /*[clinic end generated code: output=9d87ab3c8b8a4e61 input=c17f7ef21d22d64e]*/
- {
- HANDLE hfile;
- BOOL close_file = TRUE;
- FILE_BASIC_INFO info;
- path_t _path = PATH_T_INITIALIZE("isdir", "s", 0, 1);
- int result;
- BOOL slow_path = TRUE;
- FILE_STAT_BASIC_INFORMATION statInfo;
- if (!path_converter(s, &_path)) {
- path_cleanup(&_path);
- if (PyErr_ExceptionMatches(PyExc_ValueError)) {
- PyErr_Clear();
- Py_RETURN_FALSE;
- }
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- if (_path.wide) {
- if (_Py_GetFileInformationByName(_path.wide, FileStatBasicByNameInfo,
- &statInfo, sizeof(statInfo))) {
- if (!(statInfo.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)) {
- slow_path = FALSE;
- result = statInfo.FileAttributes & FILE_ATTRIBUTE_DIRECTORY;
- } else if (!(statInfo.FileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
- slow_path = FALSE;
- result = 0;
- }
- } else if (_Py_GetFileInformationByName_ErrorIsTrustworthy(GetLastError())) {
- slow_path = FALSE;
- result = 0;
- }
- }
- if (slow_path) {
- if (_path.fd != -1) {
- hfile = _Py_get_osfhandle_noraise(_path.fd);
- close_file = FALSE;
- }
- else {
- hfile = CreateFileW(_path.wide, FILE_READ_ATTRIBUTES, 0, NULL,
- OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
- }
- if (hfile != INVALID_HANDLE_VALUE) {
- if (GetFileInformationByHandleEx(hfile, FileBasicInfo, &info,
- sizeof(info)))
- {
- result = info.FileAttributes & FILE_ATTRIBUTE_DIRECTORY;
- }
- else {
- result = 0;
- }
- if (close_file) {
- CloseHandle(hfile);
- }
- }
- else {
- STRUCT_STAT st;
- switch (GetLastError()) {
- case ERROR_ACCESS_DENIED:
- case ERROR_SHARING_VIOLATION:
- case ERROR_CANT_ACCESS_FILE:
- case ERROR_INVALID_PARAMETER:
- if (STAT(_path.wide, &st)) {
- result = 0;
- }
- else {
- result = S_ISDIR(st.st_mode);
- }
- break;
- default:
- result = 0;
- }
- }
- }
- Py_END_ALLOW_THREADS
- path_cleanup(&_path);
- if (result) {
- Py_RETURN_TRUE;
- }
- Py_RETURN_FALSE;
- }
- /*[clinic input]
- os._path_isfile
- path: 'O'
- Test whether a path is a regular file
- [clinic start generated code]*/
- static PyObject *
- os__path_isfile_impl(PyObject *module, PyObject *path)
- /*[clinic end generated code: output=2394ed7c4b5cfd85 input=de22d74960ade365]*/
- {
- HANDLE hfile;
- BOOL close_file = TRUE;
- FILE_BASIC_INFO info;
- path_t _path = PATH_T_INITIALIZE("isfile", "path", 0, 1);
- int result;
- BOOL slow_path = TRUE;
- FILE_STAT_BASIC_INFORMATION statInfo;
- if (!path_converter(path, &_path)) {
- path_cleanup(&_path);
- if (PyErr_ExceptionMatches(PyExc_ValueError)) {
- PyErr_Clear();
- Py_RETURN_FALSE;
- }
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- if (_path.wide) {
- if (_Py_GetFileInformationByName(_path.wide, FileStatBasicByNameInfo,
- &statInfo, sizeof(statInfo))) {
- if (!(statInfo.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)) {
- slow_path = FALSE;
- result = !(statInfo.FileAttributes & FILE_ATTRIBUTE_DIRECTORY);
- } else if (statInfo.FileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
- slow_path = FALSE;
- result = 0;
- }
- } else if (_Py_GetFileInformationByName_ErrorIsTrustworthy(GetLastError())) {
- slow_path = FALSE;
- result = 0;
- }
- }
- if (slow_path) {
- if (_path.fd != -1) {
- hfile = _Py_get_osfhandle_noraise(_path.fd);
- close_file = FALSE;
- }
- else {
- hfile = CreateFileW(_path.wide, FILE_READ_ATTRIBUTES, 0, NULL,
- OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
- }
- if (hfile != INVALID_HANDLE_VALUE) {
- if (GetFileInformationByHandleEx(hfile, FileBasicInfo, &info,
- sizeof(info)))
- {
- result = !(info.FileAttributes & FILE_ATTRIBUTE_DIRECTORY);
- }
- else {
- result = 0;
- }
- if (close_file) {
- CloseHandle(hfile);
- }
- }
- else {
- STRUCT_STAT st;
- switch (GetLastError()) {
- case ERROR_ACCESS_DENIED:
- case ERROR_SHARING_VIOLATION:
- case ERROR_CANT_ACCESS_FILE:
- case ERROR_INVALID_PARAMETER:
- if (STAT(_path.wide, &st)) {
- result = 0;
- }
- else {
- result = S_ISREG(st.st_mode);
- }
- break;
- default:
- result = 0;
- }
- }
- }
- Py_END_ALLOW_THREADS
- path_cleanup(&_path);
- if (result) {
- Py_RETURN_TRUE;
- }
- Py_RETURN_FALSE;
- }
- /*[clinic input]
- os._path_exists
- path: 'O'
- Test whether a path exists. Returns False for broken symbolic links
- [clinic start generated code]*/
- static PyObject *
- os__path_exists_impl(PyObject *module, PyObject *path)
- /*[clinic end generated code: output=f508c3b35e13a249 input=380f77cdfa0f7ae8]*/
- {
- HANDLE hfile;
- BOOL close_file = TRUE;
- path_t _path = PATH_T_INITIALIZE("exists", "path", 0, 1);
- int result;
- BOOL slow_path = TRUE;
- FILE_STAT_BASIC_INFORMATION statInfo;
- if (!path_converter(path, &_path)) {
- path_cleanup(&_path);
- if (PyErr_ExceptionMatches(PyExc_ValueError)) {
- PyErr_Clear();
- Py_RETURN_FALSE;
- }
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- if (_path.wide) {
- if (_Py_GetFileInformationByName(_path.wide, FileStatBasicByNameInfo,
- &statInfo, sizeof(statInfo))) {
- if (!(statInfo.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)) {
- slow_path = FALSE;
- result = 1;
- }
- } else if (_Py_GetFileInformationByName_ErrorIsTrustworthy(GetLastError())) {
- slow_path = FALSE;
- result = 0;
- }
- }
- if (slow_path) {
- if (_path.fd != -1) {
- hfile = _Py_get_osfhandle_noraise(_path.fd);
- close_file = FALSE;
- }
- else {
- hfile = CreateFileW(_path.wide, FILE_READ_ATTRIBUTES, 0, NULL,
- OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
- }
- if (hfile != INVALID_HANDLE_VALUE) {
- result = 1;
- if (close_file) {
- CloseHandle(hfile);
- }
- }
- else {
- STRUCT_STAT st;
- switch (GetLastError()) {
- case ERROR_ACCESS_DENIED:
- case ERROR_SHARING_VIOLATION:
- case ERROR_CANT_ACCESS_FILE:
- case ERROR_INVALID_PARAMETER:
- if (STAT(_path.wide, &st)) {
- result = 0;
- }
- else {
- result = 1;
- }
- break;
- default:
- result = 0;
- }
- }
- }
- Py_END_ALLOW_THREADS
- path_cleanup(&_path);
- if (result) {
- Py_RETURN_TRUE;
- }
- Py_RETURN_FALSE;
- }
- /*[clinic input]
- os._path_islink
- path: 'O'
- Test whether a path is a symbolic link
- [clinic start generated code]*/
- static PyObject *
- os__path_islink_impl(PyObject *module, PyObject *path)
- /*[clinic end generated code: output=6d8640b1a390c054 input=38a3cb937ccf59bf]*/
- {
- HANDLE hfile;
- BOOL close_file = TRUE;
- FILE_ATTRIBUTE_TAG_INFO info;
- path_t _path = PATH_T_INITIALIZE("islink", "path", 0, 1);
- int result;
- BOOL slow_path = TRUE;
- FILE_STAT_BASIC_INFORMATION statInfo;
- if (!path_converter(path, &_path)) {
- path_cleanup(&_path);
- if (PyErr_ExceptionMatches(PyExc_ValueError)) {
- PyErr_Clear();
- Py_RETURN_FALSE;
- }
- return NULL;
- }
- Py_BEGIN_ALLOW_THREADS
- if (_path.wide) {
- if (_Py_GetFileInformationByName(_path.wide, FileStatBasicByNameInfo,
- &statInfo, sizeof(statInfo))) {
- slow_path = FALSE;
- if (statInfo.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
- result = (statInfo.ReparseTag == IO_REPARSE_TAG_SYMLINK);
- }
- else {
- result = 0;
- }
- } else if (_Py_GetFileInformationByName_ErrorIsTrustworthy(GetLastError())) {
- slow_path = FALSE;
- result = 0;
- }
- }
- if (slow_path) {
- if (_path.fd != -1) {
- hfile = _Py_get_osfhandle_noraise(_path.fd);
- close_file = FALSE;
- }
- else {
- hfile = CreateFileW(_path.wide, FILE_READ_ATTRIBUTES, 0, NULL,
- OPEN_EXISTING,
- FILE_FLAG_OPEN_REPARSE_POINT | FILE_FLAG_BACKUP_SEMANTICS,
- NULL);
- }
- if (hfile != INVALID_HANDLE_VALUE) {
- if (GetFileInformationByHandleEx(hfile, FileAttributeTagInfo, &info,
- sizeof(info)))
- {
- result = (info.ReparseTag == IO_REPARSE_TAG_SYMLINK);
- }
- else {
- result = 0;
- }
- if (close_file) {
- CloseHandle(hfile);
- }
- }
- else {
- STRUCT_STAT st;
- switch (GetLastError()) {
- case ERROR_ACCESS_DENIED:
- case ERROR_SHARING_VIOLATION:
- case ERROR_CANT_ACCESS_FILE:
- case ERROR_INVALID_PARAMETER:
- if (LSTAT(_path.wide, &st)) {
- result = 0;
- }
- else {
- result = S_ISLNK(st.st_mode);
- }
- break;
- default:
- result = 0;
- }
- }
- }
- Py_END_ALLOW_THREADS
- path_cleanup(&_path);
- if (result) {
- Py_RETURN_TRUE;
- }
- Py_RETURN_FALSE;
- }
- #endif /* MS_WINDOWS */
- /*[clinic input]
- os._path_normpath
- path: object
- Basic path normalization.
- [clinic start generated code]*/
- static PyObject *
- os__path_normpath_impl(PyObject *module, PyObject *path)
- /*[clinic end generated code: output=b94d696d828019da input=5e90c39e12549dc0]*/
- {
- if (!PyUnicode_Check(path)) {
- PyErr_Format(PyExc_TypeError, "expected 'str', not '%.200s'",
- Py_TYPE(path)->tp_name);
- return NULL;
- }
- Py_ssize_t len;
- wchar_t *buffer = PyUnicode_AsWideCharString(path, &len);
- if (!buffer) {
- return NULL;
- }
- Py_ssize_t norm_len;
- wchar_t *norm_path = _Py_normpath_and_size(buffer, len, &norm_len);
- PyObject *result = PyUnicode_FromWideChar(norm_path, norm_len);
- PyMem_Free(buffer);
- 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 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
- result = CreateDirectoryW(path->wide, NULL);
- Py_END_ALLOW_THREADS
- 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 */
- // 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();
- }
- }
- #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;
- }
- 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 && path->narrow) {
- 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=2bf9df3d6ab2e48b]*/
- #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
- /*[clinic input]
- os.major -> unsigned_int
- device: dev_t
- /
- Extracts a device major number from a raw device number.
- [clinic start generated code]*/
- static unsigned int
- os_major_impl(PyObject *module, dev_t device)
- /*[clinic end generated code: output=5b3b2589bafb498e input=1e16a4d30c4d4462]*/
- {
- return major(device);
- }
- /*[clinic input]
- os.minor -> unsigned_int
- device: dev_t
- /
- Extracts a device minor number from a raw device number.
- [clinic start generated code]*/
- static unsigned int
- os_minor_impl(PyObject *module, dev_t device)
- /*[clinic end generated code: output=5e1a25e630b0157d input=0842c6d23f24c65e]*/
- {
- return minor(device);
- }
- /*[clinic input]
- os.makedev -> dev_t
- major: int
- minor: int
- /
- Composes a raw device number from the major and minor device numbers.
- [clinic start generated code]*/
- static dev_t
- os_makedev_impl(PyObject *module, int major, int minor)
- /*[clinic end generated code: output=881aaa4aba6f6a52 input=4b9fd8fc73cbe48f]*/
- {
- 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;
- if (path->narrow) {
- 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 (path->narrow) {
- 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;
- #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
|