PikaObj.c 150 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051
  1. /*
  2. * This file is part of the PikaPython project.
  3. * http://github.com/pikastech/pikapython
  4. *
  5. * MIT License
  6. *
  7. * Copyright (c) 2021 lyon liang6516@outlook.com
  8. * Copyright (c) 2023 Gorgon Meducer embedded_zhuroan@hotmail.com
  9. *
  10. * Permission is hereby granted, free of charge, to any person obtaining a
  11. * copy of this software and associated documentation files (the "Software"),
  12. * to deal in the Software without restriction, including without limitation
  13. * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  14. * and/or sell copies of the Software, and to permit persons to whom the
  15. * Software is furnished to do so, subject to the following conditions:
  16. *
  17. * The above copyright notice and this permission notice shall be included in
  18. * all copies or substantial portions of the Software.
  19. *
  20. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  21. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  22. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  23. * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  24. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  25. * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  26. * DEALINGS IN THE SOFTWARE.
  27. */
  28. #include "PikaObj.h"
  29. #include <stdint.h>
  30. #include "BaseObj.h"
  31. #include "PikaCompiler.h"
  32. #include "PikaParser.h"
  33. #include "PikaPlatform.h"
  34. #include "dataArg.h"
  35. #include "dataArgs.h"
  36. #include "dataMemory.h"
  37. #include "dataQueue.h"
  38. #include "dataString.h"
  39. #include "dataStrs.h"
  40. #if __linux
  41. #include "signal.h"
  42. #include "termios.h"
  43. #include "unistd.h"
  44. #endif
  45. extern volatile VMState g_PikaVMState;
  46. volatile PikaObjState g_PikaObjState = {
  47. .helpModulesCmodule = NULL,
  48. .inRootObj = pika_false,
  49. #if PIKA_GC_MARK_SWEEP_ENABLE
  50. .objCnt = 0,
  51. .objCntMax = 0,
  52. .objCntLastGC = 0,
  53. .gcChain = NULL,
  54. .markSweepBusy = 0,
  55. #endif
  56. };
  57. extern volatile PikaObj* __pikaMain;
  58. static volatile ShellConfig g_REPL;
  59. PikaObj* New_PikaStdData_Dict(Args* args);
  60. PikaObj* New_PikaStdData_dict_keys(Args* args);
  61. PikaObj* New_PikaStdData_List(Args* args);
  62. PikaObj* New_PikaStdData_Tuple(Args* args);
  63. void PikaStdData_Tuple___init__(PikaObj* self);
  64. void PikaStdData_List___init__(PikaObj* self);
  65. void PikaStdData_List_append(PikaObj* self, Arg* arg);
  66. void PikaStdData_Dict_set(PikaObj* self, char* key, Arg* value);
  67. void PikaStdData_Dict___init__(PikaObj* self);
  68. void _mem_cache_deinit(void);
  69. void _VMEvent_deinit(void);
  70. void pikaGC_markObj(PikaGC* gc, PikaObj* self);
  71. void _pikaGC_mark(PikaGC* gc);
  72. void obj_dump(PikaObj* self);
  73. void Locals_deinit(PikaObj* self);
  74. #if __linux
  75. static void disable_raw_mode(void);
  76. #endif
  77. static enum shellCTRL __obj_shellLineHandler_REPL(PikaObj* self,
  78. char* input_line,
  79. ShellConfig* shell);
  80. static const uint64_t __talbe_fast_atoi[][10] = {
  81. {0, 1e0, 2e0, 3e0, 4e0, 5e0, 6e0, 7e0, 8e0, 9e0},
  82. {0, 1e1, 2e1, 3e1, 4e1, 5e1, 6e1, 7e1, 8e1, 9e1},
  83. {0, 1e2, 2e2, 3e2, 4e2, 5e2, 6e2, 7e2, 8e2, 9e2},
  84. {0, 1e3, 2e3, 3e3, 4e3, 5e3, 6e3, 7e3, 8e3, 9e3},
  85. {0, 1e4, 2e4, 3e4, 4e4, 5e4, 6e4, 7e4, 8e4, 9e4},
  86. {0, 1e5, 2e5, 3e5, 4e5, 5e5, 6e5, 7e5, 8e5, 9e5},
  87. {0, 1e6, 2e6, 3e6, 4e6, 5e6, 6e6, 7e6, 8e6, 9e6},
  88. {0, 1e7, 2e7, 3e7, 4e7, 5e7, 6e7, 7e7, 8e7, 9e7},
  89. {0, 1e8, 2e8, 3e8, 4e8, 5e8, 6e8, 7e8, 8e8, 9e8},
  90. {0, 1e9, 2e9, 3e9, 4e9, 5e9, 6e9, 7e9, 8e9, 9e9},
  91. };
  92. int64_t fast_atoi(char* src) {
  93. const char* p = src;
  94. uint16_t size = strGetSize(src);
  95. p = p + size - 1;
  96. if (*p) {
  97. int64_t s = 0;
  98. const uint64_t* n = __talbe_fast_atoi[0];
  99. while (p != src) {
  100. s += n[(*p - '0')];
  101. n += 10;
  102. p--;
  103. }
  104. if (*p == '-') {
  105. return -s;
  106. }
  107. return s + n[(*p - '0')];
  108. }
  109. return 0;
  110. }
  111. PIKA_RES fast_atoi_safe(char* src, int64_t* out) {
  112. // Check is digit
  113. char* p = src;
  114. while (*p) {
  115. if (*p < '0' || *p > '9') {
  116. return PIKA_RES_ERR_INVALID_PARAM;
  117. }
  118. p++;
  119. }
  120. *out = fast_atoi(src);
  121. return PIKA_RES_OK;
  122. }
  123. static uint16_t const str100p[100] = {
  124. 0x3030, 0x3130, 0x3230, 0x3330, 0x3430, 0x3530, 0x3630, 0x3730, 0x3830,
  125. 0x3930, 0x3031, 0x3131, 0x3231, 0x3331, 0x3431, 0x3531, 0x3631, 0x3731,
  126. 0x3831, 0x3931, 0x3032, 0x3132, 0x3232, 0x3332, 0x3432, 0x3532, 0x3632,
  127. 0x3732, 0x3832, 0x3932, 0x3033, 0x3133, 0x3233, 0x3333, 0x3433, 0x3533,
  128. 0x3633, 0x3733, 0x3833, 0x3933, 0x3034, 0x3134, 0x3234, 0x3334, 0x3434,
  129. 0x3534, 0x3634, 0x3734, 0x3834, 0x3934, 0x3035, 0x3135, 0x3235, 0x3335,
  130. 0x3435, 0x3535, 0x3635, 0x3735, 0x3835, 0x3935, 0x3036, 0x3136, 0x3236,
  131. 0x3336, 0x3436, 0x3536, 0x3636, 0x3736, 0x3836, 0x3936, 0x3037, 0x3137,
  132. 0x3237, 0x3337, 0x3437, 0x3537, 0x3637, 0x3737, 0x3837, 0x3937, 0x3038,
  133. 0x3138, 0x3238, 0x3338, 0x3438, 0x3538, 0x3638, 0x3738, 0x3838, 0x3938,
  134. 0x3039, 0x3139, 0x3239, 0x3339, 0x3439, 0x3539, 0x3639, 0x3739, 0x3839,
  135. 0x3939,
  136. };
  137. char* fast_itoa(char* buf, uint32_t val) {
  138. char* p = &buf[10];
  139. *p = '\0';
  140. while (val >= 100) {
  141. uint32_t const old = val;
  142. p -= 2;
  143. val /= 100;
  144. pika_platform_memcpy(p, &str100p[old - (val * 100)], sizeof(uint16_t));
  145. }
  146. p -= 2;
  147. pika_platform_memcpy(p, &str100p[val], sizeof(uint16_t));
  148. return &p[val < 10];
  149. }
  150. static int32_t obj_deinit_no_del(PikaObj* self) {
  151. /* free the list */
  152. Locals_deinit(self);
  153. args_deinit_ex(self->list, pika_true);
  154. #if PIKA_KERNAL_DEBUG_ENABLE
  155. if (NULL != self->aName) {
  156. arg_deinit(self->aName);
  157. }
  158. #endif
  159. /* remove self from gc chain */
  160. obj_removeGcChain(self);
  161. /* free the pointer */
  162. pikaFree(self, sizeof(PikaObj));
  163. if (self == (PikaObj*)__pikaMain) {
  164. __pikaMain = NULL;
  165. }
  166. return 0;
  167. }
  168. int obj_GC(PikaObj* self) {
  169. if (!obj_checkAlive(self)) {
  170. return 0;
  171. }
  172. if (obj_getFlag(self, OBJ_FLAG_IN_DEL)) {
  173. /* in __del__, skip gc */
  174. return 0;
  175. }
  176. obj_refcntDec(self);
  177. int ref_cnt = obj_refcntNow(self);
  178. if (ref_cnt <= 0) {
  179. obj_deinit(self);
  180. }
  181. return 0;
  182. }
  183. int32_t obj_deinit(PikaObj* self) {
  184. pikaGC_lock();
  185. pika_bool bisRoot = pika_false;
  186. #if PIKA_KERNAL_DEBUG_ENABLE
  187. self->isAlive = pika_false;
  188. #endif
  189. Arg* del = obj_getMethodArgWithFullPath(self, "__del__");
  190. if (NULL != del) {
  191. obj_setFlag(self, OBJ_FLAG_IN_DEL);
  192. Arg* aRes = obj_runMethodArg0(self, del);
  193. if (NULL != aRes) {
  194. arg_deinit(aRes);
  195. }
  196. }
  197. extern volatile PikaObj* __pikaMain;
  198. if (self == (PikaObj*)__pikaMain) {
  199. bisRoot = pika_true;
  200. _mem_cache_deinit();
  201. #if PIKA_EVENT_ENABLE
  202. _VMEvent_deinit();
  203. #endif
  204. if (NULL != g_PikaObjState.helpModulesCmodule) {
  205. arg_deinit(g_PikaObjState.helpModulesCmodule);
  206. g_PikaObjState.helpModulesCmodule = NULL;
  207. }
  208. }
  209. int32_t ret = obj_deinit_no_del(self);
  210. pikaGC_unlock();
  211. if (bisRoot) {
  212. pikaGC_markSweep();
  213. shConfig_deinit((ShellConfig*)&g_REPL);
  214. #if __linux
  215. disable_raw_mode();
  216. pika_GIL_deinit();
  217. #endif
  218. }
  219. return ret;
  220. }
  221. PIKA_RES obj_setInt(PikaObj* self, char* argPath, int64_t val) {
  222. PikaObj* obj = obj_getHostObj(self, argPath);
  223. if (NULL == obj) {
  224. /* [error] object no found */
  225. return PIKA_RES_ERR_ARG_NO_FOUND;
  226. }
  227. char* name = strPointToLastToken(argPath, '.');
  228. args_setInt(obj->list, name, val);
  229. return PIKA_RES_OK;
  230. }
  231. PIKA_RES obj_setPtr(PikaObj* self, char* argPath, void* pointer) {
  232. PikaObj* obj = obj_getHostObj(self, argPath);
  233. if (NULL == obj) {
  234. /* [error] object no found */
  235. return PIKA_RES_ERR_ARG_NO_FOUND;
  236. }
  237. char* name = strPointToLastToken(argPath, '.');
  238. args_setPtr(obj->list, name, pointer);
  239. return PIKA_RES_OK;
  240. }
  241. PIKA_RES obj_setRef(PikaObj* self, char* argPath, PikaObj* pointer) {
  242. PikaObj* obj = obj_getHostObj(self, argPath);
  243. if (NULL == obj) {
  244. /* [error] object no found */
  245. return PIKA_RES_ERR_ARG_NO_FOUND;
  246. }
  247. char* name = strPointToLastToken(argPath, '.');
  248. args_setRef(obj->list, name, pointer);
  249. return PIKA_RES_OK;
  250. }
  251. PIKA_RES obj_setFloat(PikaObj* self, char* argPath, pika_float value) {
  252. PikaObj* obj = obj_getHostObj(self, argPath);
  253. if (NULL == obj) {
  254. /* [error] object no found */
  255. return PIKA_RES_ERR_ARG_NO_FOUND;
  256. }
  257. char* name = strPointToLastToken(argPath, '.');
  258. args_setFloat(obj->list, name, value);
  259. return PIKA_RES_OK;
  260. }
  261. PIKA_RES obj_setStr(PikaObj* self, char* argPath, char* str) {
  262. PikaObj* obj = obj_getHostObj(self, argPath);
  263. if (NULL == obj) {
  264. return PIKA_RES_ERR_ARG_NO_FOUND;
  265. }
  266. char* name = strPointToLastToken(argPath, '.');
  267. return args_setStr(obj->list, name, str);
  268. }
  269. PIKA_RES obj_setNone(PikaObj* self, char* argPath) {
  270. PikaObj* obj = obj_getHostObj(self, argPath);
  271. if (NULL == obj) {
  272. return PIKA_RES_ERR_ARG_NO_FOUND;
  273. }
  274. char* name = strPointToLastToken(argPath, '.');
  275. args_setNone(obj->list, name);
  276. return PIKA_RES_OK;
  277. }
  278. PIKA_RES obj_setBytes(PikaObj* self, char* argPath, uint8_t* src, size_t size) {
  279. PikaObj* obj = obj_getHostObj(self, argPath);
  280. if (NULL == obj) {
  281. return PIKA_RES_ERR_ARG_NO_FOUND;
  282. }
  283. char* name = strPointToLastToken(argPath, '.');
  284. args_setBytes(obj->list, name, src, size);
  285. return PIKA_RES_OK;
  286. }
  287. int64_t obj_getInt(PikaObj* self, char* argPath) {
  288. PikaObj* obj = obj_getHostObj(self, argPath);
  289. if (NULL == obj) {
  290. return _PIKA_INT_ERR;
  291. }
  292. char* argName = strPointToLastToken(argPath, '.');
  293. int64_t res = args_getInt(obj->list, argName);
  294. return res;
  295. }
  296. pika_bool obj_getBool(PikaObj* self, char* argPath) {
  297. PikaObj* obj = obj_getHostObj(self, argPath);
  298. if (NULL == obj) {
  299. return pika_false;
  300. }
  301. char* argName = strPointToLastToken(argPath, '.');
  302. pika_bool res = args_getBool(obj->list, argName);
  303. return res;
  304. }
  305. Arg* obj_getArg(PikaObj* self, char* argPath) {
  306. pika_assert(obj_checkAlive(self));
  307. pika_bool is_temp = pika_false;
  308. PikaObj* obj = obj_getHostObjWithIsTemp(self, argPath, &is_temp);
  309. if (NULL == obj) {
  310. return NULL;
  311. }
  312. Arg* res = NULL;
  313. char* argName = strPointToLastToken(argPath, '.');
  314. res = args_getArg(obj->list, argName);
  315. if (is_temp) {
  316. obj_setArg(self, "_buf", res);
  317. res = obj_getArg(self, "_buf");
  318. obj_deinit(obj);
  319. }
  320. return res;
  321. }
  322. uint8_t* obj_getBytes(PikaObj* self, char* argPath) {
  323. PikaObj* obj = obj_getHostObj(self, argPath);
  324. if (NULL == obj) {
  325. return NULL;
  326. }
  327. char* argName = strPointToLastToken(argPath, '.');
  328. return args_getBytes(obj->list, argName);
  329. }
  330. size_t obj_getBytesSize(PikaObj* self, char* argPath) {
  331. PikaObj* obj = obj_getHostObj(self, argPath);
  332. if (NULL == obj) {
  333. return 0;
  334. }
  335. char* argName = strPointToLastToken(argPath, '.');
  336. return args_getBytesSize(obj->list, argName);
  337. }
  338. size_t obj_loadBytes(PikaObj* self, char* argPath, uint8_t* out_buff) {
  339. size_t size_mem = obj_getBytesSize(self, argPath);
  340. void* src = obj_getBytes(self, argPath);
  341. if (0 == size_mem) {
  342. return 0;
  343. }
  344. if (NULL == src) {
  345. return 0;
  346. }
  347. pika_platform_memcpy(out_buff, src, size_mem);
  348. return size_mem;
  349. }
  350. void obj_setName(PikaObj* self, char* name) {
  351. #if !PIKA_KERNAL_DEBUG_ENABLE
  352. return;
  353. #else
  354. if (strEqu(name, "self")) {
  355. return;
  356. }
  357. if (NULL != self->aName) {
  358. if (!strstr(self->name, name)) {
  359. self->aName = arg_strAppend(self->aName, "|");
  360. self->aName = arg_strAppend(self->aName, name);
  361. }
  362. } else {
  363. self->aName = arg_newStr(name);
  364. }
  365. self->name = arg_getStr(self->aName);
  366. #endif
  367. }
  368. static PIKA_RES _obj_setArg(PikaObj* self,
  369. char* argPath,
  370. Arg* arg,
  371. uint8_t is_copy) {
  372. pika_assert(obj_checkAlive(self));
  373. /* setArg would copy arg */
  374. PikaObj* host = obj_getHostObj(self, argPath);
  375. PikaObj* oNew = NULL;
  376. pika_bool bNew = pika_false;
  377. if (NULL == host) {
  378. /* object no found */
  379. return PIKA_RES_ERR_ARG_NO_FOUND;
  380. }
  381. char* sArgName = strPointToLastToken(argPath, '.');
  382. Arg* aNew;
  383. if (is_copy) {
  384. aNew = arg_copy(arg);
  385. } else {
  386. aNew = arg;
  387. }
  388. aNew = arg_setName(aNew, sArgName);
  389. if (arg_isObject(aNew)) {
  390. oNew = arg_getPtr(aNew);
  391. bNew = pika_true;
  392. pika_assert(obj_checkAlive(oNew));
  393. #if PIKA_KERNAL_DEBUG_ENABLE
  394. if (host != oNew) {
  395. /* skip self ref */
  396. oNew->parent = host;
  397. }
  398. #endif
  399. obj_setName(oNew, sArgName);
  400. }
  401. #if 0
  402. if (argType_isObjectMethodActive(arg_getType(arg))) {
  403. PikaObj* host_self = methodArg_getHostObj(arg);
  404. obj_refcntInc(host_self);
  405. }
  406. #endif
  407. args_setArg(host->list, aNew);
  408. /* enable mark sweep to collect this object */
  409. if (bNew) {
  410. /* only enable mark sweep after setArg */
  411. obj_enableGC(oNew);
  412. }
  413. return PIKA_RES_OK;
  414. }
  415. PIKA_RES obj_setArg(PikaObj* self, char* argPath, Arg* arg) {
  416. return _obj_setArg(self, argPath, arg, 1);
  417. };
  418. PIKA_RES obj_setArg_noCopy(PikaObj* self, char* argPath, Arg* arg) {
  419. return _obj_setArg(self, argPath, arg, 0);
  420. }
  421. PIKA_RES obj_setObj(PikaObj* self, char* argPath, PikaObj* obj) {
  422. Arg* arg = arg_newObj(obj);
  423. return obj_setArg_noCopy(self, argPath, arg);
  424. }
  425. void* obj_getPtr(PikaObj* self, char* argPath) {
  426. PikaObj* obj = obj_getHostObj(self, argPath);
  427. if (NULL == obj) {
  428. return NULL;
  429. }
  430. char* argName = strPointToLastToken(argPath, '.');
  431. void* res = args_getPtr(obj->list, argName);
  432. return res;
  433. }
  434. pika_float obj_getFloat(PikaObj* self, char* argPath) {
  435. PikaObj* obj = obj_getHostObj(self, argPath);
  436. if (NULL == obj) {
  437. return -999.999;
  438. }
  439. char* argName = strPointToLastToken(argPath, '.');
  440. pika_float res = args_getFloat(obj->list, argName);
  441. return res;
  442. }
  443. char* obj_getStr(PikaObj* self, char* argPath) {
  444. PikaObj* obj = obj_getHostObj(self, argPath);
  445. if (NULL == obj) {
  446. return NULL;
  447. }
  448. char* argName = strPointToLastToken(argPath, '.');
  449. char* res = args_getStr(obj->list, argName);
  450. return res;
  451. }
  452. PikaObj* obj_newObjFromConstructor(PikaObj* context,
  453. char* name,
  454. NewFun constructor) {
  455. PikaObj* self = constructor(NULL);
  456. self->constructor = constructor;
  457. self->refcnt = 1;
  458. return self;
  459. }
  460. static PikaObj* _pika_new_obj_with_args(PikaObj* (*constructor)(),
  461. void (*init_func)(PikaObj*),
  462. int num_args,
  463. va_list args) {
  464. PikaObj* obj = newNormalObj(constructor);
  465. init_func(obj);
  466. if (constructor == New_PikaStdData_Tuple ||
  467. constructor == New_PikaStdData_List) {
  468. for (int i = 0; i < num_args; i++) {
  469. Arg* arg = va_arg(args, Arg*);
  470. if (num_args == 1 && NULL == arg) {
  471. /* empty tuple */
  472. return obj;
  473. }
  474. pikaList_append(obj, arg);
  475. }
  476. } else if (constructor == New_PikaStdData_Dict) {
  477. if (num_args == 1) {
  478. /* empty dict */
  479. return obj;
  480. }
  481. for (int i = 0; i + 1 < num_args; i += 2) {
  482. Arg* aKey = va_arg(args, Arg*);
  483. char* sKey = arg_getStr(aKey);
  484. Arg* value = va_arg(args, Arg*);
  485. pikaDict_set(obj, sKey, value);
  486. arg_deinit(aKey);
  487. }
  488. }
  489. return obj;
  490. }
  491. PikaObj* _pika_tuple_new(int num_args, ...) {
  492. va_list args;
  493. va_start(args, num_args);
  494. PikaObj* tuple = _pika_new_obj_with_args(
  495. New_PikaStdData_Tuple, PikaStdData_Tuple___init__, num_args, args);
  496. va_end(args);
  497. return tuple;
  498. }
  499. PikaObj* _pika_list_new(int num_args, ...) {
  500. va_list args;
  501. va_start(args, num_args);
  502. PikaObj* list = _pika_new_obj_with_args(
  503. New_PikaStdData_List, PikaStdData_List___init__, num_args, args);
  504. va_end(args);
  505. return list;
  506. }
  507. PikaObj* _pika_dict_new(int num_args, ...) {
  508. va_list args;
  509. va_start(args, num_args);
  510. PikaObj* dict = _pika_new_obj_with_args(
  511. New_PikaStdData_Dict, PikaStdData_Dict___init__, num_args, args);
  512. va_end(args);
  513. return dict;
  514. }
  515. NativeProperty* obj_getProp(PikaObj* self) {
  516. Arg* aProp = obj_getArg(self, "@p_");
  517. PikaObj* class_obj = NULL;
  518. if (NULL == aProp) {
  519. if (NULL != self->constructor) {
  520. class_obj = obj_getClassObj(self);
  521. aProp = obj_getArg(class_obj, "@p_");
  522. }
  523. }
  524. NativeProperty* prop = NULL;
  525. if (aProp == NULL) {
  526. goto __exit;
  527. }
  528. if (arg_getType(aProp) != ARG_TYPE_POINTER) {
  529. goto __exit;
  530. }
  531. prop = (NativeProperty*)arg_getPtr(aProp);
  532. __exit:
  533. if (NULL != class_obj) {
  534. obj_deinit_no_del(class_obj);
  535. }
  536. return prop;
  537. }
  538. Arg* _obj_getPropArg(PikaObj* obj, char* name) {
  539. NativeProperty* prop = obj_getProp(obj);
  540. Hash method_hash = hash_time33(name);
  541. Arg* aMethod = NULL;
  542. while (1) {
  543. if (prop == NULL) {
  544. break;
  545. }
  546. int size = prop->methodGroupCount;
  547. pika_assert(size >= 0);
  548. /* binary search */
  549. if (size == 0) {
  550. goto __next;
  551. }
  552. #if !PIKA_NANO_ENABLE
  553. if (size > 16) {
  554. int left = 0;
  555. int right = size - 1;
  556. int mid = 0;
  557. while (1) {
  558. if (left > right) {
  559. break;
  560. }
  561. mid = (right + left) >> 1;
  562. Arg* prop_this = (Arg*)(prop->methodGroup + mid);
  563. if (prop_this->name_hash == method_hash) {
  564. aMethod = prop_this;
  565. goto __exit;
  566. } else if (prop_this->name_hash < method_hash) {
  567. left = mid + 1;
  568. } else {
  569. right = mid - 1;
  570. }
  571. }
  572. goto __next;
  573. }
  574. #endif
  575. for (int i = 0; i < (int)prop->methodGroupCount; i++) {
  576. Arg* prop_this = (Arg*)(prop->methodGroup + i);
  577. if (prop_this->name_hash == method_hash) {
  578. aMethod = prop_this;
  579. goto __exit;
  580. }
  581. }
  582. goto __next;
  583. __next:
  584. prop = (NativeProperty*)prop->super;
  585. }
  586. __exit:
  587. return aMethod;
  588. }
  589. Arg* _obj_getMethodArg(PikaObj* obj, char* methodName, Arg* arg_reg) {
  590. Arg* aMethod = NULL;
  591. aMethod = obj_getArg(obj, methodName);
  592. if (NULL != aMethod) {
  593. aMethod = arg_copy_noalloc(aMethod, arg_reg);
  594. goto __exit;
  595. }
  596. aMethod = _obj_getPropArg(obj, methodName);
  597. __exit:
  598. return aMethod;
  599. }
  600. Arg* _obj_getMethodArgWithFullPath(PikaObj* obj,
  601. char* methodPath,
  602. Arg* arg_reg) {
  603. char* methodName = strPointToLastToken(methodPath, '.');
  604. return _obj_getMethodArg(obj, methodName, arg_reg);
  605. }
  606. Arg* obj_getMethodArgWithFullPath(PikaObj* obj, char* methodPath) {
  607. return _obj_getMethodArgWithFullPath(obj, methodPath, NULL);
  608. }
  609. Arg* obj_getMethodArgWithFullPath_noalloc(PikaObj* obj,
  610. char* methodPath,
  611. Arg* arg_reg) {
  612. return _obj_getMethodArgWithFullPath(obj, methodPath, arg_reg);
  613. }
  614. Arg* obj_getMethodArg(PikaObj* obj, char* methodName) {
  615. return _obj_getMethodArg(obj, methodName, NULL);
  616. }
  617. Arg* obj_getMethodArg_noalloc(PikaObj* obj, char* methodName, Arg* arg_reg) {
  618. return _obj_getMethodArg(obj, methodName, arg_reg);
  619. }
  620. NewFun obj_getClass(PikaObj* obj) {
  621. return (NewFun)obj->constructor;
  622. }
  623. PikaObj* obj_getClassObj(PikaObj* obj) {
  624. NewFun classPtr = obj_getClass(obj);
  625. if (NULL == classPtr) {
  626. return NULL;
  627. }
  628. PikaObj* classObj = obj_newObjFromConstructor(obj, "", classPtr);
  629. return classObj;
  630. }
  631. void* getNewClassObjFunByName(PikaObj* obj, char* name) {
  632. char* classPath = name;
  633. /* init the subprocess */
  634. void* newClassObjFun = args_getPtr(obj->list, classPath);
  635. return newClassObjFun;
  636. }
  637. PikaObj* removeMethodInfo(PikaObj* thisClass) {
  638. #if PIKA_METHOD_CACHE_ENABLE
  639. #else
  640. args_removeArg(thisClass->list, args_getArg(thisClass->list, "@p_"));
  641. #endif
  642. return thisClass;
  643. }
  644. PikaObj* newNormalObj(NewFun newObjFun) {
  645. PikaObj* thisClass = obj_newObjFromConstructor(NULL, "", newObjFun);
  646. obj_setFlag(thisClass, OBJ_FLAG_ALREADY_INIT);
  647. return removeMethodInfo(thisClass);
  648. }
  649. #ifdef __linux
  650. #include <errno.h>
  651. #include <stdio.h>
  652. #include <termios.h>
  653. #include <unistd.h>
  654. #define ECHOFLAGS (ECHO | ECHOE | ECHOK | ECHONL)
  655. static int set_disp_mode(int fd, int option) {
  656. int err;
  657. struct termios term;
  658. if (tcgetattr(fd, &term) == -1) {
  659. perror("Cannot get the attribution of the terminal");
  660. return 1;
  661. }
  662. if (option)
  663. term.c_lflag |= ECHOFLAGS;
  664. else
  665. term.c_lflag &= ~ECHOFLAGS;
  666. err = tcsetattr(fd, TCSAFLUSH, &term);
  667. if (err == -1) {
  668. perror("Cannot set the attribution of the terminal");
  669. return 1;
  670. }
  671. return 0;
  672. }
  673. #endif
  674. static volatile uint8_t logo_printed = 0;
  675. #if __linux
  676. struct termios original_termios;
  677. static void enable_raw_mode(void) {
  678. struct termios raw;
  679. tcgetattr(STDIN_FILENO, &original_termios);
  680. raw = original_termios;
  681. raw.c_lflag &= ~(ECHO | ICANON);
  682. tcsetattr(STDIN_FILENO, TCSAFLUSH, &raw);
  683. }
  684. static void disable_raw_mode(void) {
  685. tcsetattr(STDIN_FILENO, TCSAFLUSH, &original_termios);
  686. printf("\n");
  687. }
  688. static void signal_handler(int sig) {
  689. if (sig == SIGSEGV) {
  690. printf("Segmentation fault");
  691. } else if (sig == SIGINT) {
  692. printf("Ctrl+C");
  693. } else if (sig == SIGTERM) {
  694. printf("SIGTERM");
  695. } else if (sig == SIGHUP) {
  696. printf("SIGHUP");
  697. } else if (sig == SIGABRT) {
  698. printf("Aborted");
  699. }
  700. disable_raw_mode();
  701. exit(1);
  702. }
  703. #endif
  704. extern volatile PikaObj* __pikaMain;
  705. PikaObj* newRootObj(char* name, NewFun newObjFun) {
  706. g_PikaObjState.inRootObj = pika_true;
  707. #if PIKA_POOL_ENABLE
  708. mem_pool_init();
  709. #endif
  710. #ifdef __linux
  711. signal(SIGTERM, signal_handler);
  712. signal(SIGHUP, signal_handler);
  713. signal(SIGSEGV, signal_handler);
  714. signal(SIGABRT, signal_handler);
  715. signal(SIGQUIT, signal_handler);
  716. signal(SIGTRAP, signal_handler);
  717. signal(SIGHUP, signal_handler);
  718. enable_raw_mode();
  719. #endif
  720. PikaObj* newObj = newNormalObj(newObjFun);
  721. if (!logo_printed) {
  722. logo_printed = 1;
  723. pika_platform_printf("\r\n");
  724. pika_platform_printf("~~~/ POWERED BY \\~~~\r\n");
  725. pika_platform_printf("~ pikapython.com ~\r\n");
  726. pika_platform_printf("~~~~~~~~~~~~~~~~~~~~\r\n");
  727. }
  728. if (NULL != __pikaMain) {
  729. pika_platform_printf("Error: root object already exists\r\n");
  730. pika_platform_panic_handle();
  731. return NULL;
  732. }
  733. __pikaMain = newObj;
  734. obj_setName(newObj, name);
  735. g_PikaObjState.inRootObj = pika_false;
  736. return newObj;
  737. }
  738. Arg* arg_newMetaObj(NewFun new_obj_fun) {
  739. Arg* arg_new = New_arg(NULL);
  740. /* m means meta-object */
  741. arg_new = arg_setPtr(arg_new, "", ARG_TYPE_OBJECT_META, (void*)new_obj_fun);
  742. return arg_new;
  743. }
  744. Arg* arg_newDirectObj(NewFun new_obj_fun) {
  745. PikaObj* newObj = newNormalObj(new_obj_fun);
  746. Arg* arg_new = arg_newPtr(ARG_TYPE_OBJECT_NEW, newObj);
  747. return arg_new;
  748. }
  749. Arg* obj_newObjInPackage(NewFun new_obj_fun) {
  750. return arg_newDirectObj(new_obj_fun);
  751. }
  752. static PikaObj* _obj_initMetaObj(PikaObj* obj, char* name) {
  753. PikaObj* res = NULL;
  754. NewFun constructor = (NewFun)getNewClassObjFunByName(obj, name);
  755. Args buffs = {0};
  756. PikaObj* thisClass;
  757. PikaObj* oNew;
  758. if (NULL == constructor) {
  759. /* no such object */
  760. res = NULL;
  761. goto __exit;
  762. }
  763. thisClass = obj_newObjFromConstructor(obj, name, constructor);
  764. oNew = removeMethodInfo(thisClass);
  765. obj_setName(oNew, name);
  766. obj_runNativeMethod(oNew, "__init__", NULL);
  767. args_setPtrWithType(obj->list, name, ARG_TYPE_OBJECT, oNew);
  768. res = obj_getPtr(obj, name);
  769. // pikaGC_enable(res);
  770. goto __exit;
  771. __exit:
  772. strsDeinit(&buffs);
  773. return res;
  774. }
  775. PikaObj* _arg_to_obj(Arg* self, pika_bool* pIsTemp) {
  776. if (NULL == self) {
  777. return NULL;
  778. }
  779. if (arg_isObject(self)) {
  780. return arg_getPtr(self);
  781. }
  782. #if !PIKA_NANO_ENABLE
  783. if (arg_getType(self) == ARG_TYPE_STRING) {
  784. PikaObj* New_PikaStdData_String(Args * args);
  785. PikaObj* obj = newNormalObj(New_PikaStdData_String);
  786. obj_setStr(obj, "str", arg_getStr(self));
  787. if (NULL != pIsTemp) {
  788. *pIsTemp = pika_true;
  789. }
  790. return obj;
  791. }
  792. if (arg_getType(self) == ARG_TYPE_BYTES) {
  793. PikaObj* New_PikaStdData_ByteArray(Args * args);
  794. PikaObj* obj = newNormalObj(New_PikaStdData_ByteArray);
  795. obj_setArg(obj, "raw", self);
  796. if (NULL != pIsTemp) {
  797. *pIsTemp = pika_true;
  798. }
  799. return obj;
  800. }
  801. #endif
  802. return NULL;
  803. }
  804. static PikaObj* _obj_getObjDirect(PikaObj* self,
  805. char* name,
  806. pika_bool* pIsTemp) {
  807. *pIsTemp = pika_false;
  808. if (NULL == self) {
  809. return NULL;
  810. }
  811. /* finded object, check type*/
  812. Arg* aObj = args_getArg(self->list, name);
  813. ArgType type = ARG_TYPE_NONE;
  814. if (NULL == aObj) {
  815. aObj = _obj_getPropArg(self, name);
  816. }
  817. if (NULL == aObj) {
  818. return NULL;
  819. }
  820. type = arg_getType(aObj);
  821. /* found meta Object */
  822. if (type == ARG_TYPE_OBJECT_META) {
  823. return _obj_initMetaObj(self, name);
  824. }
  825. /* found Objcet */
  826. if (argType_isObject(type)) {
  827. return arg_getPtr(aObj);
  828. }
  829. #if !PIKA_NANO_ENABLE
  830. /* found class */
  831. if (type == ARG_TYPE_METHOD_NATIVE_CONSTRUCTOR ||
  832. type == ARG_TYPE_METHOD_CONSTRUCTOR) {
  833. *pIsTemp = pika_true;
  834. PikaObj* oMethodArgs = New_TinyObj(NULL);
  835. Arg* aClsObj = obj_runMethodArg(self, oMethodArgs, aObj);
  836. obj_deinit(oMethodArgs);
  837. Args* args = New_args(NULL);
  838. if (type == ARG_TYPE_METHOD_NATIVE_CONSTRUCTOR) {
  839. obj_runNativeMethod(arg_getPtr(aClsObj), "__init__", args);
  840. }
  841. PikaObj* res = arg_getPtr(aClsObj);
  842. arg_deinit(aClsObj);
  843. args_deinit(args);
  844. return res;
  845. }
  846. #endif
  847. return _arg_to_obj(aObj, pIsTemp);
  848. }
  849. static PikaObj* _obj_getObjWithKeepDeepth(PikaObj* self,
  850. char* objPath,
  851. pika_bool* pIsTemp,
  852. int32_t keepDeepth) {
  853. char objPath_buff[PIKA_PATH_BUFF_SIZE];
  854. char* objPath_ptr = objPath_buff;
  855. pika_assert(NULL != objPath);
  856. if ('.' == objPath[0] && '\0' == objPath[1]) {
  857. return self;
  858. }
  859. pika_assert(strGetSize(objPath) < PIKA_PATH_BUFF_SIZE);
  860. strcpy(objPath_buff, objPath);
  861. int32_t token_num = strGetTokenNum(objPath, '.');
  862. PikaObj* objThis = self;
  863. PikaObj* objNext = NULL;
  864. pika_bool bThisIsTemp = pika_false;
  865. for (int32_t i = 0; i < token_num - keepDeepth; i++) {
  866. char* token = strPopFirstToken(&objPath_ptr, '.');
  867. objNext = _obj_getObjDirect(objThis, token, pIsTemp);
  868. if (objNext == NULL) {
  869. objThis = NULL;
  870. goto __exit;
  871. }
  872. if (bThisIsTemp) {
  873. if (*pIsTemp == pika_false) {
  874. obj_refcntInc(objNext);
  875. *pIsTemp = pika_true;
  876. }
  877. obj_deinit(objThis);
  878. }
  879. pika_assert_obj_alive(objNext);
  880. objThis = objNext;
  881. bThisIsTemp = *pIsTemp;
  882. }
  883. goto __exit;
  884. __exit:
  885. if (NULL != objThis) {
  886. pika_assert(obj_checkAlive(objThis));
  887. }
  888. return objThis;
  889. }
  890. PikaObj* obj_getObj(PikaObj* self, char* objPath) {
  891. pika_assert(NULL != objPath);
  892. pika_bool is_temp = pika_false;
  893. return _obj_getObjWithKeepDeepth(self, objPath, &is_temp, 0);
  894. }
  895. PikaObj* obj_getHostObj(PikaObj* self, char* objPath) {
  896. pika_assert(NULL != objPath);
  897. pika_bool is_temp = pika_false;
  898. return _obj_getObjWithKeepDeepth(self, objPath, &is_temp, 1);
  899. }
  900. PikaObj* obj_getHostObjWithIsTemp(PikaObj* self,
  901. char* objPath,
  902. pika_bool* pIsTemp) {
  903. return _obj_getObjWithKeepDeepth(self, objPath, pIsTemp, 1);
  904. }
  905. Method methodArg_getPtr(Arg* method_arg) {
  906. MethodProp* method_store = (MethodProp*)arg_getContent(method_arg);
  907. return (Method)method_store->ptr;
  908. }
  909. Arg* _get_return_arg(PikaObj* locals);
  910. NativeProperty* methodArg_toProp(Arg* method_arg) {
  911. pika_assert(arg_getType(method_arg) == ARG_TYPE_METHOD_NATIVE_CONSTRUCTOR);
  912. PikaObj* locals = New_TinyObj(NULL);
  913. MethodProp* method_store = (MethodProp*)arg_getContent(method_arg);
  914. Method fMethod = (Method)method_store->ptr;
  915. fMethod(NULL, locals->list);
  916. Arg* aReturn = _get_return_arg(locals);
  917. if (arg_getType(aReturn) == ARG_TYPE_OBJECT_NEW) {
  918. arg_setType(aReturn, ARG_TYPE_OBJECT);
  919. }
  920. PikaObj* obj = arg_getPtr(aReturn);
  921. NativeProperty* prop = obj_getProp(obj);
  922. arg_deinit(aReturn);
  923. obj_deinit(locals);
  924. return prop;
  925. }
  926. char* methodArg_getTypeList(Arg* method_arg, char* buffs, size_t size) {
  927. pika_assert(arg_isCallable(method_arg));
  928. MethodProp* prop = (MethodProp*)arg_getContent(method_arg);
  929. if (NULL != prop->type_list) {
  930. pika_assert(strGetSize(prop->type_list) < size);
  931. return strcpy(buffs, prop->type_list);
  932. }
  933. char* method_dec = methodArg_getDec(method_arg);
  934. pika_assert(strGetSize(method_dec) <= size);
  935. if (strGetSize(method_dec) > size) {
  936. pika_platform_printf(
  937. "OverFlowError: please use bigger PIKA_LINE_BUFF_SIZE\r\n");
  938. while (1) {
  939. }
  940. }
  941. char* res = strCut(buffs, method_dec, '(', ')');
  942. return res;
  943. }
  944. MethodProp* methodArg_getProp(Arg* method_arg) {
  945. return (MethodProp*)arg_getContent(method_arg);
  946. }
  947. PikaObj* methodArg_getHostObj(Arg* method_arg) {
  948. pika_assert(argType_isObjectMethodActive(arg_getType(method_arg)));
  949. MethodProp* prop = (MethodProp*)arg_getContent(method_arg);
  950. return prop->host_obj;
  951. }
  952. Arg* methodArg_active(Arg* method_arg) {
  953. pika_assert(arg_getType(method_arg) == ARG_TYPE_METHOD_NATIVE);
  954. Arg* aActive = New_arg(NULL);
  955. MethodPropNative* propNative =
  956. (MethodPropNative*)arg_getContent(method_arg);
  957. MethodProp prop = {0};
  958. /* active the method */
  959. pika_platform_memcpy(&prop, propNative, sizeof(MethodPropNative));
  960. aActive = arg_setStruct(aActive, "", (uint8_t*)&prop, sizeof(MethodProp));
  961. arg_setType(aActive, ARG_TYPE_METHOD_NATIVE_ACTIVE);
  962. return aActive;
  963. }
  964. Arg* methodArg_setHostObj(Arg* method_arg, PikaObj* host_obj) {
  965. if (!argType_isObjectMethod(arg_getType(method_arg))) {
  966. return method_arg;
  967. }
  968. if (arg_getType(method_arg) == ARG_TYPE_METHOD_NATIVE) {
  969. method_arg = methodArg_active(method_arg);
  970. }
  971. MethodProp* prop = (MethodProp*)arg_getContent(method_arg);
  972. if (prop->host_obj == NULL) {
  973. prop->host_obj = host_obj;
  974. #if 0
  975. obj_refcntInc(host_obj);
  976. #endif
  977. return method_arg;
  978. }
  979. return method_arg;
  980. }
  981. char* methodArg_getName(Arg* method_arg, char* buffs, size_t size) {
  982. MethodProp* method_store = (MethodProp*)arg_getContent(method_arg);
  983. if (NULL != method_store->name) {
  984. return strcpy(buffs, method_store->name);
  985. }
  986. char* method_dec = methodArg_getDec(method_arg);
  987. pika_assert(strGetSize(method_dec) <= size);
  988. char* res = strGetFirstToken(buffs, method_dec, '(');
  989. return res;
  990. }
  991. char* _find_super_class_name(ByteCodeFrame* bcframe, int32_t pc_start);
  992. Arg* methodArg_super(Arg* aThis, NativeProperty** p_prop) {
  993. Arg* aSuper = NULL;
  994. PikaObj* builtins = NULL;
  995. ArgType type = arg_getType(aThis);
  996. *p_prop = NULL;
  997. if (!argType_isConstructor(type)) {
  998. aSuper = NULL;
  999. goto __exit;
  1000. }
  1001. if (type == ARG_TYPE_METHOD_CONSTRUCTOR) {
  1002. builtins = obj_getBuiltins();
  1003. MethodProp* method_store = (MethodProp*)arg_getContent(aThis);
  1004. ByteCodeFrame* bcframe = method_store->bytecode_frame;
  1005. int32_t pc = (uintptr_t)method_store->ptr -
  1006. (uintptr_t)bcframe->instruct_array.content_start;
  1007. char* sSuper = _find_super_class_name(bcframe, pc);
  1008. /* map TinyObj to object */
  1009. if (strEqu(sSuper, "TinyObj")) {
  1010. sSuper = "object";
  1011. }
  1012. PikaObj* context = method_store->def_context;
  1013. aSuper = obj_getMethodArgWithFullPath(context, sSuper);
  1014. if (NULL == aSuper) {
  1015. aSuper = obj_getMethodArgWithFullPath(builtins, sSuper);
  1016. }
  1017. goto __exit;
  1018. }
  1019. if (type == ARG_TYPE_METHOD_NATIVE_CONSTRUCTOR) {
  1020. NativeProperty* prop = methodArg_toProp(aThis);
  1021. *p_prop = prop;
  1022. arg_deinit(aThis);
  1023. aSuper = NULL;
  1024. goto __exit;
  1025. }
  1026. __exit:
  1027. if (NULL != builtins) {
  1028. obj_deinit(builtins);
  1029. }
  1030. if (NULL != aSuper) {
  1031. arg_deinit(aThis);
  1032. }
  1033. return aSuper;
  1034. }
  1035. Method obj_getNativeMethod(PikaObj* self, char* method_name) {
  1036. Arg* method_arg = obj_getMethodArgWithFullPath(self, method_name);
  1037. if (NULL == method_arg) {
  1038. return NULL;
  1039. }
  1040. Method res = methodArg_getPtr(method_arg);
  1041. arg_deinit(method_arg);
  1042. return res;
  1043. }
  1044. ByteCodeFrame* methodArg_getBytecodeFrame(Arg* method_arg) {
  1045. MethodProp* method_store = (MethodProp*)arg_getContent(method_arg);
  1046. return method_store->bytecode_frame;
  1047. }
  1048. char* methodArg_getDec(Arg* method_arg) {
  1049. MethodProp* method_store = (MethodProp*)arg_getContent(method_arg);
  1050. return method_store->declareation;
  1051. }
  1052. PikaObj* methodArg_getDefContext(Arg* method_arg) {
  1053. MethodProp* method_store = (MethodProp*)arg_getContent(method_arg);
  1054. return method_store->def_context;
  1055. }
  1056. void _update_proxy(PikaObj* self, char* name) {
  1057. #if PIKA_NANO_ENABLE
  1058. return;
  1059. #endif
  1060. if (name[0] != '_' || name[1] != '_') {
  1061. return;
  1062. }
  1063. if (!obj_getFlag(self, OBJ_FLAG_PROXY_GETATTRIBUTE)) {
  1064. if (strEqu(name, "__getattribute__")) {
  1065. obj_setFlag(self, OBJ_FLAG_PROXY_GETATTRIBUTE);
  1066. return;
  1067. }
  1068. }
  1069. if (!obj_getFlag(self, OBJ_FLAG_PROXY_GETATTR)) {
  1070. if (strEqu(name, "__getattr__")) {
  1071. obj_setFlag(self, OBJ_FLAG_PROXY_GETATTR);
  1072. return;
  1073. }
  1074. }
  1075. if (!obj_getFlag(self, OBJ_FLAG_PROXY_SETATTR)) {
  1076. if (strEqu(name, "__setattr__")) {
  1077. obj_setFlag(self, OBJ_FLAG_PROXY_SETATTR);
  1078. return;
  1079. }
  1080. }
  1081. if (!obj_getFlag(self, OBJ_FLAG_PROXY_METHOD)) {
  1082. if (strEqu(name, "__proxy__")) {
  1083. obj_setFlag(self, OBJ_FLAG_PROXY_METHOD);
  1084. return;
  1085. }
  1086. }
  1087. }
  1088. static void obj_saveMethodInfo(PikaObj* self, MethodInfo* tInfo) {
  1089. Arg* aMethod = New_arg(NULL);
  1090. MethodProp tProp = {
  1091. .ptr = tInfo->ptr,
  1092. .type_list = tInfo->typelist,
  1093. .name = tInfo->name,
  1094. .bytecode_frame = tInfo->bytecode_frame,
  1095. .def_context = tInfo->def_context,
  1096. .declareation = tInfo->dec, // const
  1097. .host_obj = NULL,
  1098. };
  1099. char* name = tInfo->name;
  1100. if (NULL == tInfo->name) {
  1101. char name_buff[PIKA_LINE_BUFF_SIZE / 2] = {0};
  1102. name = strGetFirstToken(name_buff, tInfo->dec, '(');
  1103. }
  1104. /* the first arg_value */
  1105. aMethod = arg_setStruct(aMethod, name, &tProp, sizeof(tProp));
  1106. pika_assert(NULL != aMethod);
  1107. arg_setType(aMethod, tInfo->type);
  1108. _update_proxy(self, name);
  1109. args_setArg(self->list, aMethod);
  1110. }
  1111. static int32_t __class_defineMethodWithType(PikaObj* self,
  1112. char* declareation,
  1113. char* name,
  1114. char* typelist,
  1115. Method method_ptr,
  1116. ArgType method_type,
  1117. PikaObj* def_context,
  1118. ByteCodeFrame* bytecode_frame) {
  1119. int32_t res = 0;
  1120. Args buffs = {0};
  1121. PikaObj* method_host = self;
  1122. MethodInfo method_info = {0};
  1123. if (NULL == method_host) {
  1124. /* no found method object */
  1125. res = 1;
  1126. goto __exit;
  1127. }
  1128. method_info.dec = declareation;
  1129. method_info.name = name;
  1130. method_info.ptr = (void*)method_ptr;
  1131. method_info.type = method_type;
  1132. method_info.def_context = def_context;
  1133. method_info.bytecode_frame = bytecode_frame;
  1134. method_info.typelist = typelist;
  1135. obj_saveMethodInfo(method_host, &method_info);
  1136. res = 0;
  1137. goto __exit;
  1138. __exit:
  1139. strsDeinit(&buffs);
  1140. return res;
  1141. }
  1142. /* define a constructor method */
  1143. int32_t class_defineConstructor(PikaObj* self,
  1144. char* name,
  1145. char* typelist,
  1146. Method methodPtr) {
  1147. return __class_defineMethodWithType(self, NULL, name, typelist, methodPtr,
  1148. ARG_TYPE_METHOD_NATIVE_CONSTRUCTOR,
  1149. NULL, NULL);
  1150. }
  1151. /* define a native method as default */
  1152. int32_t class_defineMethod(PikaObj* self,
  1153. char* name,
  1154. char* typelist,
  1155. Method methodPtr) {
  1156. return __class_defineMethodWithType(self, NULL, name, typelist, methodPtr,
  1157. ARG_TYPE_METHOD_NATIVE, NULL, NULL);
  1158. }
  1159. /* define object method, object method is which startwith (self) */
  1160. int32_t class_defineRunTimeConstructor(PikaObj* self,
  1161. char* declareation,
  1162. Method methodPtr,
  1163. PikaObj* def_context,
  1164. ByteCodeFrame* bytecode_frame) {
  1165. return __class_defineMethodWithType(self, declareation, NULL, NULL,
  1166. methodPtr, ARG_TYPE_METHOD_CONSTRUCTOR,
  1167. def_context, bytecode_frame);
  1168. }
  1169. /* define object method, object method is which startwith (self) */
  1170. int32_t class_defineObjectMethod(PikaObj* self,
  1171. char* declareation,
  1172. Method methodPtr,
  1173. PikaObj* def_context,
  1174. ByteCodeFrame* bytecode_frame) {
  1175. return __class_defineMethodWithType(self, declareation, NULL, NULL,
  1176. methodPtr, ARG_TYPE_METHOD_OBJECT,
  1177. def_context, bytecode_frame);
  1178. }
  1179. /* define a static method as default */
  1180. int32_t class_defineStaticMethod(PikaObj* self,
  1181. char* declareation,
  1182. Method methodPtr,
  1183. PikaObj* def_context,
  1184. ByteCodeFrame* bytecode_frame) {
  1185. return __class_defineMethodWithType(self, declareation, NULL, NULL,
  1186. methodPtr, ARG_TYPE_METHOD_STATIC,
  1187. def_context, bytecode_frame);
  1188. }
  1189. int32_t obj_removeArg(PikaObj* self, char* argPath) {
  1190. PikaObj* objHost = obj_getHostObj(self, argPath);
  1191. char* argName;
  1192. int32_t res;
  1193. int32_t err = 0;
  1194. if (NULL == objHost) {
  1195. /* [error] object no found */
  1196. err = 1;
  1197. goto __exit;
  1198. }
  1199. argName = strPointToLastToken(argPath, '.');
  1200. res = args_removeArg(objHost->list, args_getArg(objHost->list, argName));
  1201. if (1 == res) {
  1202. /*[error] not found arg*/
  1203. err = 2;
  1204. goto __exit;
  1205. }
  1206. goto __exit;
  1207. __exit:
  1208. return err;
  1209. }
  1210. pika_bool obj_isArgExist(PikaObj* self, char* argPath) {
  1211. if (NULL == argPath) {
  1212. return 0;
  1213. }
  1214. PikaObj* obj_host = obj_getHostObj(self, argPath);
  1215. if (obj_host == NULL) {
  1216. return pika_false;
  1217. }
  1218. int32_t res = 0;
  1219. char* argName;
  1220. Arg* arg;
  1221. pika_assert(NULL != obj_host);
  1222. argName = strPointToLastToken(argPath, '.');
  1223. arg = args_getArg(obj_host->list, argName);
  1224. if (NULL == arg) {
  1225. /* no found arg */
  1226. res = 0;
  1227. goto __exit;
  1228. }
  1229. /* found arg */
  1230. res = 1;
  1231. goto __exit;
  1232. __exit:
  1233. return res;
  1234. }
  1235. pika_bool obj_isMethodExist(PikaObj* self, char* method) {
  1236. Arg* arg = obj_getMethodArg(self, method);
  1237. if (NULL == arg) {
  1238. return pika_false;
  1239. }
  1240. arg_deinit(arg);
  1241. return pika_true;
  1242. }
  1243. VMParameters* obj_run(PikaObj* self, char* cmd) {
  1244. return pikaVM_run(self, cmd);
  1245. }
  1246. PIKA_RES obj_runNativeMethod(PikaObj* self, char* method_name, Args* args) {
  1247. Method native_method = obj_getNativeMethod(self, method_name);
  1248. if (NULL == native_method) {
  1249. return PIKA_RES_ERR_ARG_NO_FOUND;
  1250. }
  1251. native_method(self, args);
  1252. return PIKA_RES_OK;
  1253. }
  1254. static void __clearBuff(ShellConfig* shell) {
  1255. memset(shell->lineBuff, 0, PIKA_LINE_BUFF_SIZE);
  1256. shell->line_curpos = 0;
  1257. shell->line_position = 0;
  1258. }
  1259. enum PIKA_SHELL_STATE {
  1260. PIKA_SHELL_STATE_NORMAL,
  1261. PIKA_SHELL_STATE_WAIT_SPEC_KEY,
  1262. PIKA_SHELL_STATE_WAIT_FUNC_KEY,
  1263. };
  1264. static void _obj_runChar_beforeRun(PikaObj* self, ShellConfig* shell) {
  1265. /* create the line buff for the first time */
  1266. shell->inBlock = pika_false;
  1267. shell->stat = PIKA_SHELL_STATE_NORMAL;
  1268. shell->line_position = 0;
  1269. shell->line_curpos = 0;
  1270. __clearBuff(shell);
  1271. pika_platform_printf("%s", shell->prefix);
  1272. }
  1273. static void _putc_cmd(char KEY_POS, int pos) {
  1274. const char cmd[] = {0x1b, 0x5b, KEY_POS, 0x00};
  1275. for (int i = 0; i < pos; i++) {
  1276. pika_platform_printf((char*)cmd);
  1277. }
  1278. }
  1279. #if PIKA_SHELL_FILTER_ENABLE
  1280. typedef enum {
  1281. __FILTER_NO_RESULT,
  1282. __FILTER_FAIL_DROP_ONE,
  1283. __FILTER_SUCCESS_GET_ALL_PEEKED,
  1284. __FILTER_SUCCESS_DROP_ALL_PEEKED
  1285. } FilterReturn;
  1286. pika_bool _filter_msg_hi_pika_handler(FilterItem* msg,
  1287. PikaObj* self,
  1288. ShellConfig* shell) {
  1289. pika_platform_printf("Yes, I am here\r\n");
  1290. return pika_true;
  1291. }
  1292. pika_bool _filter_msg_bye_pika_handler(FilterItem* msg,
  1293. PikaObj* self,
  1294. ShellConfig* shell) {
  1295. pika_platform_printf("OK, see you\r\n");
  1296. return pika_true;
  1297. }
  1298. #define __MSG_DECLARE
  1299. #include "__default_filter_msg_table.h"
  1300. static const FilterItem g_default_filter_messages[] = {
  1301. #define __MSG_TABLE
  1302. #include "__default_filter_msg_table.h"
  1303. };
  1304. static FilterReturn _do_message_filter(PikaObj* self,
  1305. ShellConfig* shell,
  1306. FilterItem* msg,
  1307. uint_fast16_t count) {
  1308. pika_assert(NULL != msg);
  1309. pika_assert(count > 0);
  1310. ByteQueue* queue = &shell->filter_fifo.queue;
  1311. FilterReturn result = __FILTER_FAIL_DROP_ONE;
  1312. do {
  1313. do {
  1314. if (msg->ignore_mask & shell->filter_fifo.ignore_mask) {
  1315. /* this message should be ignored */
  1316. break;
  1317. }
  1318. if (NULL == msg->message) {
  1319. break;
  1320. }
  1321. uint_fast16_t message_size = msg->size;
  1322. if (!message_size) {
  1323. break;
  1324. }
  1325. byteQueue_resetPeek(queue);
  1326. /* do message comparison */
  1327. uint8_t* src = (uint8_t*)msg->message;
  1328. if (msg->is_case_insensitive) {
  1329. do {
  1330. uint8_t byte;
  1331. if (!byteQueue_peekOne(queue, &byte)) {
  1332. result = __FILTER_NO_RESULT;
  1333. break;
  1334. }
  1335. char letter = *src++;
  1336. if (letter >= 'A' && letter <= 'Z') {
  1337. letter += 'a' - 'A';
  1338. }
  1339. if (byte >= 'A' && byte <= 'Z') {
  1340. byte += 'a' - 'A';
  1341. }
  1342. if (letter != byte) {
  1343. break;
  1344. }
  1345. } while (--message_size);
  1346. } else {
  1347. do {
  1348. uint8_t byte;
  1349. if (!byteQueue_peekOne(queue, &byte)) {
  1350. result = __FILTER_NO_RESULT;
  1351. break;
  1352. }
  1353. if (*src++ != byte) {
  1354. break;
  1355. }
  1356. } while (--message_size);
  1357. }
  1358. if (0 == message_size) {
  1359. /* message match */
  1360. if (NULL != msg->handler) {
  1361. if (!msg->handler(msg, self, shell)) {
  1362. break;
  1363. }
  1364. }
  1365. /* message is handled */
  1366. if (msg->is_visible) {
  1367. return __FILTER_SUCCESS_GET_ALL_PEEKED;
  1368. }
  1369. return __FILTER_SUCCESS_DROP_ALL_PEEKED;
  1370. }
  1371. } while (0);
  1372. msg++;
  1373. } while (--count);
  1374. return result;
  1375. }
  1376. #ifndef dimof
  1377. #define dimof(__array) (sizeof(__array) / sizeof(__array[0]))
  1378. #endif
  1379. int16_t _do_stream_filter(PikaObj* self, ShellConfig* shell) {
  1380. ByteQueue* queue = &shell->filter_fifo.queue;
  1381. FilterReturn result =
  1382. _do_message_filter(self, shell, (FilterItem*)g_default_filter_messages,
  1383. dimof(g_default_filter_messages));
  1384. int16_t drop_count = 0;
  1385. switch (result) {
  1386. case __FILTER_NO_RESULT:
  1387. break;
  1388. case __FILTER_FAIL_DROP_ONE:
  1389. drop_count = 1;
  1390. break;
  1391. case __FILTER_SUCCESS_DROP_ALL_PEEKED:
  1392. byteQueue_dropAllPeeked(queue);
  1393. return 0;
  1394. case __FILTER_SUCCESS_GET_ALL_PEEKED:
  1395. drop_count = byteQueue_getPeekedNumber(queue);
  1396. return drop_count;
  1397. }
  1398. /* user registered message filter */
  1399. if (NULL != shell->messages && shell->message_count) {
  1400. result = _do_message_filter(self, shell, shell->messages,
  1401. shell->message_count);
  1402. switch (result) {
  1403. case __FILTER_NO_RESULT:
  1404. break;
  1405. case __FILTER_FAIL_DROP_ONE:
  1406. drop_count = 1;
  1407. break;
  1408. case __FILTER_SUCCESS_DROP_ALL_PEEKED:
  1409. byteQueue_dropAllPeeked(&shell->filter_fifo.queue);
  1410. return 0;
  1411. case __FILTER_SUCCESS_GET_ALL_PEEKED:
  1412. drop_count =
  1413. byteQueue_getPeekedNumber(&shell->filter_fifo.queue);
  1414. return drop_count;
  1415. }
  1416. }
  1417. return drop_count;
  1418. }
  1419. #endif
  1420. #if PIKA_SHELL_HISTORY_ENABLE
  1421. ShellHistory* shHistory_create(int max_size) {
  1422. ShellHistory* self = (ShellHistory*)pikaMalloc(sizeof(ShellHistory));
  1423. self->max_size = max_size;
  1424. self->current = -1;
  1425. self->count = 0;
  1426. self->last_offset = 0;
  1427. self->cached_current = 0;
  1428. self->history = (char**)pikaMalloc(max_size * sizeof(char*));
  1429. return self;
  1430. }
  1431. void shHistory_destroy(ShellHistory* self) {
  1432. for (int i = 0; i < self->count; i++) {
  1433. pikaFree(self->history[i], strGetSize(self->history[i]) + 1);
  1434. }
  1435. pikaFree(self->history, sizeof(char*) * self->max_size);
  1436. pikaFree(self, sizeof(ShellHistory));
  1437. }
  1438. void shHistory_add(ShellHistory* self, char* command) {
  1439. if (self->count == self->max_size) {
  1440. pikaFree(self->history[0], strGetSize(self->history[0]) + 1);
  1441. pika_platform_memmove(self->history, self->history + 1,
  1442. (self->max_size - 1) * sizeof(char*));
  1443. self->count--;
  1444. }
  1445. /* filter for empty command */
  1446. if (self->count > 0 && self->history[self->count - 1][0] == '\0') {
  1447. pikaFree(self->history[self->count - 1],
  1448. strGetSize(self->history[self->count - 1]) + 1);
  1449. self->count--;
  1450. }
  1451. /* filter for same command */
  1452. if (self->count > 0 && strEqu(self->history[self->count - 1], command)) {
  1453. return;
  1454. }
  1455. self->history[self->count] = pikaMalloc(strGetSize(command) + 1);
  1456. pika_platform_memcpy(self->history[self->count], command,
  1457. strGetSize(command) + 1);
  1458. self->count++;
  1459. self->current = self->count - 1;
  1460. self->last_offset = 0;
  1461. }
  1462. char* shHistory_get(ShellHistory* self, int offset) {
  1463. int actual_offset = offset + self->last_offset;
  1464. int index = self->current + actual_offset;
  1465. if (index < 0 || index >= self->count) {
  1466. return NULL;
  1467. }
  1468. self->last_offset = actual_offset;
  1469. return self->history[index];
  1470. }
  1471. char* shHistory_getPrev(ShellHistory* self) {
  1472. return shHistory_get(self, -1);
  1473. }
  1474. char* shHistory_getNext(ShellHistory* self) {
  1475. return shHistory_get(self, 1);
  1476. }
  1477. #endif
  1478. #define PIKA_BACKSPACE() pika_platform_printf(" \b")
  1479. #if __linux
  1480. #define PIKA_BACKSPACE_FORCE() printf("\b \b")
  1481. #else
  1482. #define PIKA_BACKSPACE_FORCE() pika_platform_printf("\b \b")
  1483. #endif
  1484. static void handle_history_navigation(char inputChar,
  1485. ShellConfig* shell,
  1486. pika_bool bIsUp) {
  1487. #if PIKA_SHELL_HISTORY_ENABLE
  1488. if (NULL == shell->history) {
  1489. shell->history = shHistory_create(PIKA_SHELL_HISTORY_NUM);
  1490. }
  1491. if (0 == shell->history->cached_current) {
  1492. /* save the current line */
  1493. shHistory_add(shell->history, shell->lineBuff);
  1494. shell->history->cached_current = 1;
  1495. }
  1496. char* history_line = bIsUp ? shHistory_getPrev(shell->history)
  1497. : shHistory_getNext(shell->history);
  1498. if (NULL == history_line) {
  1499. return;
  1500. }
  1501. /* move to the last position */
  1502. for (int i = 0; i < shell->line_position - shell->line_curpos; i++) {
  1503. _putc_cmd(PIKA_KEY_RIGHT, 1);
  1504. }
  1505. /* clear the current line */
  1506. for (int i = 0; i < shell->line_position; i++) {
  1507. PIKA_BACKSPACE_FORCE();
  1508. }
  1509. pika_platform_memcpy(shell->lineBuff, history_line,
  1510. strGetSize(history_line) + 1);
  1511. /* show the previous line */
  1512. pika_platform_printf("%s", shell->lineBuff);
  1513. shell->line_position = strGetSize(history_line);
  1514. shell->line_curpos = shell->line_position;
  1515. #endif
  1516. return;
  1517. }
  1518. enum shellCTRL _inner_do_obj_runChar(PikaObj* self,
  1519. char inputChar,
  1520. ShellConfig* shell) {
  1521. char* input_line = NULL;
  1522. enum shellCTRL ctrl = SHELL_CTRL_CONTINUE;
  1523. if (inputChar == 0x7F) {
  1524. inputChar = '\b';
  1525. }
  1526. if (g_REPL.no_echo == pika_false) {
  1527. #if __linux
  1528. printf("%c", inputChar);
  1529. #elif !(defined(_WIN32))
  1530. pika_platform_printf("%c", inputChar);
  1531. #endif
  1532. }
  1533. if (inputChar == '\n' && shell->lastChar == '\r') {
  1534. ctrl = SHELL_CTRL_CONTINUE;
  1535. goto __exit;
  1536. }
  1537. if (inputChar == 0x1b) {
  1538. shell->stat = PIKA_SHELL_STATE_WAIT_SPEC_KEY;
  1539. ctrl = SHELL_CTRL_CONTINUE;
  1540. goto __exit;
  1541. }
  1542. if (shell->stat == PIKA_SHELL_STATE_WAIT_SPEC_KEY) {
  1543. if (inputChar == 0x5b) {
  1544. shell->stat = PIKA_SHELL_STATE_WAIT_FUNC_KEY;
  1545. ctrl = SHELL_CTRL_CONTINUE;
  1546. goto __exit;
  1547. } else {
  1548. shell->stat = PIKA_SHELL_STATE_NORMAL;
  1549. }
  1550. }
  1551. if (shell->stat == PIKA_SHELL_STATE_WAIT_FUNC_KEY) {
  1552. shell->stat = PIKA_SHELL_STATE_NORMAL;
  1553. if (inputChar == PIKA_KEY_LEFT) {
  1554. if (shell->line_curpos) {
  1555. shell->line_curpos--;
  1556. } else {
  1557. pika_platform_printf(" ");
  1558. }
  1559. ctrl = SHELL_CTRL_CONTINUE;
  1560. goto __exit;
  1561. }
  1562. if (inputChar == PIKA_KEY_RIGHT) {
  1563. if (shell->line_curpos < shell->line_position) {
  1564. // pika_platform_printf("%c",
  1565. // shell->lineBuff[shell->line_curpos]);
  1566. shell->line_curpos++;
  1567. } else {
  1568. pika_platform_printf("\b");
  1569. }
  1570. ctrl = SHELL_CTRL_CONTINUE;
  1571. goto __exit;
  1572. }
  1573. if (inputChar == PIKA_KEY_UP) {
  1574. _putc_cmd(PIKA_KEY_DOWN, 1);
  1575. ctrl = SHELL_CTRL_CONTINUE;
  1576. handle_history_navigation(inputChar, shell, pika_true);
  1577. goto __exit;
  1578. }
  1579. if (inputChar == PIKA_KEY_DOWN) {
  1580. ctrl = SHELL_CTRL_CONTINUE;
  1581. handle_history_navigation(inputChar, shell, pika_false);
  1582. goto __exit;
  1583. }
  1584. }
  1585. if (inputChar == '\b') {
  1586. if (shell->line_curpos == 0) {
  1587. #if __linux
  1588. printf(" ");
  1589. #else
  1590. pika_platform_printf(" ");
  1591. #endif
  1592. ctrl = SHELL_CTRL_CONTINUE;
  1593. goto __exit;
  1594. }
  1595. PIKA_BACKSPACE();
  1596. shell->line_position--;
  1597. shell->line_curpos--;
  1598. pika_platform_memmove(shell->lineBuff + shell->line_curpos,
  1599. shell->lineBuff + shell->line_curpos + 1,
  1600. shell->line_position - shell->line_curpos);
  1601. shell->lineBuff[shell->line_position] = 0;
  1602. if (shell->line_curpos != shell->line_position) {
  1603. /* update screen */
  1604. pika_platform_printf("%s", shell->lineBuff + shell->line_curpos);
  1605. pika_platform_printf(" ");
  1606. _putc_cmd(PIKA_KEY_LEFT,
  1607. shell->line_position - shell->line_curpos + 1);
  1608. }
  1609. ctrl = SHELL_CTRL_CONTINUE;
  1610. goto __exit;
  1611. }
  1612. if ((inputChar != '\r') && (inputChar != '\n')) {
  1613. if (shell->line_position + 1 >= PIKA_LINE_BUFF_SIZE) {
  1614. pika_platform_printf(
  1615. "\r\nError: line buff overflow, please use bigger "
  1616. "'PIKA_LINE_BUFF_SIZE'\r\n");
  1617. ctrl = SHELL_CTRL_EXIT;
  1618. __clearBuff(shell);
  1619. goto __exit;
  1620. }
  1621. if ('\0' != inputChar) {
  1622. pika_platform_memmove(shell->lineBuff + shell->line_curpos + 1,
  1623. shell->lineBuff + shell->line_curpos,
  1624. shell->line_position - shell->line_curpos);
  1625. shell->lineBuff[shell->line_position + 1] = 0;
  1626. if (shell->line_curpos != shell->line_position) {
  1627. pika_platform_printf("%s",
  1628. shell->lineBuff + shell->line_curpos + 1);
  1629. _putc_cmd(PIKA_KEY_LEFT,
  1630. shell->line_position - shell->line_curpos);
  1631. }
  1632. shell->lineBuff[shell->line_curpos] = inputChar;
  1633. shell->line_position++;
  1634. shell->line_curpos++;
  1635. }
  1636. ctrl = SHELL_CTRL_CONTINUE;
  1637. goto __exit;
  1638. }
  1639. if ((inputChar == '\r') || (inputChar == '\n')) {
  1640. #if !(defined(__linux) || defined(_WIN32) || PIKA_SHELL_NO_NEWLINE)
  1641. pika_platform_printf("\r\n");
  1642. #endif
  1643. #if PIKA_SHELL_HISTORY_ENABLE
  1644. if (NULL == shell->history) {
  1645. shell->history = shHistory_create(PIKA_SHELL_HISTORY_NUM);
  1646. }
  1647. if (shell->lineBuff[0] != '\0') {
  1648. shHistory_add(shell->history, shell->lineBuff);
  1649. }
  1650. shell->history->last_offset = 0;
  1651. shell->history->cached_current = 0;
  1652. #endif
  1653. /* still in block */
  1654. if (shell->blockBuffName != NULL && shell->inBlock) {
  1655. /* load new line into buff */
  1656. Args buffs = {0};
  1657. char _n = '\n';
  1658. strAppendWithSize(shell->lineBuff, &_n, 1);
  1659. char* shell_buff_new =
  1660. strsAppend(&buffs, obj_getStr(self, shell->blockBuffName),
  1661. shell->lineBuff);
  1662. obj_setStr(self, shell->blockBuffName, shell_buff_new);
  1663. strsDeinit(&buffs);
  1664. /* go out from block */
  1665. if ((shell->lineBuff[0] != ' ') && (shell->lineBuff[0] != '\t')) {
  1666. shell->inBlock = pika_false;
  1667. input_line = obj_getStr(self, shell->blockBuffName);
  1668. ctrl = shell->handler(self, input_line, shell);
  1669. __clearBuff(shell);
  1670. pika_platform_printf(">>> ");
  1671. goto __exit;
  1672. } else {
  1673. pika_platform_printf("... ");
  1674. }
  1675. __clearBuff(shell);
  1676. ctrl = SHELL_CTRL_CONTINUE;
  1677. goto __exit;
  1678. }
  1679. /* go in block */
  1680. if (shell->blockBuffName != NULL && 0 != strGetSize(shell->lineBuff)) {
  1681. if (shell->lineBuff[strGetSize(shell->lineBuff) - 1] == ':') {
  1682. shell->inBlock = pika_true;
  1683. char _n = '\n';
  1684. strAppendWithSize(shell->lineBuff, &_n, 1);
  1685. obj_setStr(self, shell->blockBuffName, shell->lineBuff);
  1686. __clearBuff(shell);
  1687. pika_platform_printf("... ");
  1688. ctrl = SHELL_CTRL_CONTINUE;
  1689. goto __exit;
  1690. }
  1691. }
  1692. shell->lineBuff[shell->line_position] = '\0';
  1693. ctrl = shell->handler(self, shell->lineBuff, shell);
  1694. if (SHELL_CTRL_EXIT != ctrl) {
  1695. pika_platform_printf("%s", shell->prefix);
  1696. }
  1697. __clearBuff(shell);
  1698. goto __exit;
  1699. }
  1700. __exit:
  1701. shell->lastChar = inputChar;
  1702. return ctrl;
  1703. }
  1704. PIKA_WEAK
  1705. enum shellCTRL _do_obj_runChar(PikaObj* self,
  1706. char inputChar,
  1707. ShellConfig* shell) {
  1708. #if PIKA_SHELL_FILTER_ENABLE
  1709. ByteQueue* queue = &(shell->filter_fifo.queue);
  1710. /* validation */
  1711. if (NULL == queue->buffer) {
  1712. /* need initialize first */
  1713. byteQueue_init(queue, &shell->filter_fifo.buffer,
  1714. sizeof(shell->filter_fifo.buffer), pika_false);
  1715. }
  1716. pika_bool result = byteQueue_writeOne(queue, inputChar);
  1717. (void)result;
  1718. pika_assert(result != pika_false);
  1719. int16_t byte_count;
  1720. do {
  1721. if (0 == byteQueue_peekAvailableCount(queue)) {
  1722. break;
  1723. }
  1724. byte_count = _do_stream_filter(self, shell);
  1725. int16_t n = byte_count;
  1726. while (n--) {
  1727. result = byteQueue_readOne(queue, (uint8_t*)&inputChar);
  1728. pika_assert(pika_false != result);
  1729. if (SHELL_CTRL_EXIT ==
  1730. _inner_do_obj_runChar(self, inputChar, shell)) {
  1731. return SHELL_CTRL_EXIT;
  1732. }
  1733. }
  1734. } while (byte_count);
  1735. return SHELL_CTRL_CONTINUE;
  1736. #else
  1737. return _inner_do_obj_runChar(self, inputChar, shell);
  1738. #endif
  1739. }
  1740. enum shellCTRL obj_runChar(PikaObj* self, char inputChar) {
  1741. ShellConfig* shell = args_getHeapStruct(self->list, "@shcfg");
  1742. if (NULL == shell) {
  1743. /* init the shell */
  1744. ShellConfig newShell = {0};
  1745. newShell.prefix = ">>> ";
  1746. newShell.blockBuffName = "@sh1";
  1747. newShell.handler = __obj_shellLineHandler_REPL;
  1748. args_setHeapStruct(self->list, "@shcfg", newShell, shConfig_deinit);
  1749. shell = args_getHeapStruct(self->list, "@shcfg");
  1750. _obj_runChar_beforeRun(self, shell);
  1751. }
  1752. return _do_obj_runChar(self, inputChar, shell);
  1753. }
  1754. static void _save_file(char* file_name, uint8_t* buff, size_t size) {
  1755. pika_platform_printf("[ Info] Saving file to '%s'...\r\n", file_name);
  1756. FILE* fp = pika_platform_fopen(file_name, "wb+");
  1757. if (NULL == fp) {
  1758. pika_platform_printf("[ Error] Open file '%s' error!\r\n", file_name);
  1759. pika_platform_fclose(fp);
  1760. } else {
  1761. if (pika_platform_fwrite(buff, 1, size, fp) != size) {
  1762. pika_platform_printf("[ Error] Failed to write to '%s'...\r\n",
  1763. file_name);
  1764. pika_platform_fclose(fp);
  1765. pika_platform_printf("[ Info] Removing '%s'...\r\n", file_name);
  1766. pika_platform_remove(file_name);
  1767. return;
  1768. } else {
  1769. pika_platform_printf("[ Info] Writing %d bytes to '%s'...\r\n",
  1770. (int)(size), file_name);
  1771. }
  1772. pika_platform_fclose(fp);
  1773. pika_platform_printf("[ OK ] Writing to '%s' succeed!\r\n",
  1774. file_name);
  1775. }
  1776. }
  1777. char _await_getchar(sh_getchar fn_getchar) {
  1778. pika_GIL_EXIT();
  1779. char ret = fn_getchar();
  1780. pika_GIL_ENTER();
  1781. return ret;
  1782. }
  1783. #define PIKA_MAGIC_CODE_LEN 4
  1784. /* return file size */
  1785. uint32_t _pikaShell_recv_file_direct(ShellConfig* cfg,
  1786. uint8_t* magic_code,
  1787. uint8_t** pbuff) {
  1788. uint32_t size = 0;
  1789. for (int i = 0; i < 4; i++) {
  1790. uint8_t* size_byte = (uint8_t*)&size;
  1791. size_byte[i] = cfg->fn_getchar();
  1792. }
  1793. if (magic_code[3] == 'o') {
  1794. size += sizeof(uint32_t) * 2;
  1795. }
  1796. uint8_t* buff = pikaMalloc(size);
  1797. /* save magic code and size */
  1798. memcpy(buff, magic_code, PIKA_MAGIC_CODE_LEN);
  1799. memcpy(buff + PIKA_MAGIC_CODE_LEN, &size, sizeof(size));
  1800. for (uint32_t i = sizeof(uint32_t) * 2; i < size; i++) {
  1801. buff[i] = cfg->fn_getchar();
  1802. }
  1803. *pbuff = buff;
  1804. return size;
  1805. }
  1806. /* return file size */
  1807. PIKA_WEAK uint32_t _pikaShell_recv_file(ShellConfig* cfg,
  1808. uint8_t* magic_code,
  1809. uint8_t** pbuff) {
  1810. return _pikaShell_recv_file_direct(cfg, magic_code, pbuff);
  1811. }
  1812. void _do_pikaScriptShell(PikaObj* self, ShellConfig* cfg) {
  1813. /* init the shell */
  1814. _obj_runChar_beforeRun(self, cfg);
  1815. /* getchar and run */
  1816. char inputChar[2] = {0};
  1817. while (1) {
  1818. inputChar[1] = inputChar[0];
  1819. inputChar[0] = _await_getchar(cfg->fn_getchar);
  1820. #if !PIKA_NANO_ENABLE
  1821. /* run python script */
  1822. if (inputChar[0] == '!' && inputChar[1] == '#') {
  1823. /* #! xxx */
  1824. /* start */
  1825. char* buff = pikaMalloc(PIKA_READ_FILE_BUFF_SIZE);
  1826. char input[2] = {0};
  1827. int buff_i = 0;
  1828. pika_bool is_exit = pika_false;
  1829. pika_bool is_first_line = pika_true;
  1830. while (1) {
  1831. input[1] = input[0];
  1832. input[0] = cfg->fn_getchar();
  1833. if (input[0] == '!' && input[1] == '#') {
  1834. buff[buff_i - 1] = 0;
  1835. for (int i = 0; i < 4; i++) {
  1836. /* eat 'pika' */
  1837. cfg->fn_getchar();
  1838. }
  1839. break;
  1840. }
  1841. if ('\r' == input[0]) {
  1842. continue;
  1843. }
  1844. if (is_first_line) {
  1845. if ('\n' == input[0]) {
  1846. is_first_line = pika_false;
  1847. }
  1848. continue;
  1849. }
  1850. buff[buff_i++] = input[0];
  1851. }
  1852. /* end */
  1853. pika_platform_printf(
  1854. "\r\n=============== [Code] ===============\r\n");
  1855. size_t len = strGetSize(buff);
  1856. for (size_t i = 0; i < len; i++) {
  1857. if (buff[i] == '\r') {
  1858. continue;
  1859. }
  1860. if (buff[i] == '\n') {
  1861. pika_platform_printf("\r\n");
  1862. continue;
  1863. }
  1864. pika_platform_printf("%c", buff[i]);
  1865. }
  1866. pika_platform_printf("\r\n");
  1867. pika_platform_printf("=============== [File] ===============\r\n");
  1868. pika_platform_printf(
  1869. "[ Info] File buff used: %d/%d (%0.2f%%)\r\n", (int)len,
  1870. (int)PIKA_READ_FILE_BUFF_SIZE,
  1871. ((float)len / (float)PIKA_READ_FILE_BUFF_SIZE));
  1872. #if PIKA_SHELL_SAVE_FILE_ENABLE
  1873. _save_file(PIKA_SHELL_SAVE_FILE_PATH, (uint8_t*)buff, len);
  1874. #endif
  1875. pika_platform_printf("=============== [ Run] ===============\r\n");
  1876. obj_run(self, (char*)buff);
  1877. if (NULL != strstr(buff, "exit()")) {
  1878. is_exit = pika_true;
  1879. }
  1880. pikaFree(buff, PIKA_READ_FILE_BUFF_SIZE);
  1881. if (is_exit) {
  1882. return;
  1883. }
  1884. pika_platform_printf("%s", cfg->prefix);
  1885. continue;
  1886. }
  1887. /* run xx.py.o */
  1888. if (inputChar[0] == 'p' && inputChar[1] == 0x0f) {
  1889. uint8_t magic_code[PIKA_MAGIC_CODE_LEN] = {0x0f, 'p', 0x00, 0x00};
  1890. for (int i = 0; i < 2; i++) {
  1891. /* eat 'yo' */
  1892. magic_code[2 + i] = cfg->fn_getchar();
  1893. }
  1894. uint8_t* recv = NULL;
  1895. uint32_t size = 0;
  1896. if (magic_code[2] == 'y' && magic_code[3] == 'a') {
  1897. size = _pikaShell_recv_file(cfg, magic_code, &recv);
  1898. } else {
  1899. size = _pikaShell_recv_file_direct(cfg, magic_code, &recv);
  1900. }
  1901. pika_platform_printf(
  1902. "\r\n=============== [File] ===============\r\n");
  1903. pika_platform_printf("[ Info] Recived size: %d\r\n", size);
  1904. if (magic_code[3] == 'o') {
  1905. #if PIKA_SHELL_SAVE_BYTECODE_ENABLE
  1906. _save_file(PIKA_SHELL_SAVE_BYTECODE_PATH, (uint8_t*)recv, size);
  1907. #endif
  1908. pika_platform_printf(
  1909. "=============== [ RUN] ===============\r\n");
  1910. pikaVM_runByteCodeInconstant(self, recv);
  1911. pikaFree(recv, size);
  1912. // pika_platform_printf("%s", cfg->prefix);
  1913. // continue;
  1914. //! Return for example like lvgl handler to continue.
  1915. return;
  1916. }
  1917. if (magic_code[3] == 'a') {
  1918. _save_file(PIKA_SHELL_SAVE_APP_PATH, (uint8_t*)recv, size);
  1919. pika_platform_printf(
  1920. "=============== [REBOOT] ===============\r\n");
  1921. pika_platform_reboot();
  1922. pikaFree(recv, size);
  1923. return;
  1924. }
  1925. }
  1926. #endif
  1927. if (SHELL_CTRL_EXIT == _do_obj_runChar(self, inputChar[0], cfg)) {
  1928. break;
  1929. }
  1930. }
  1931. }
  1932. void _temp__do_pikaScriptShell(PikaObj* self, ShellConfig* shell) {
  1933. /* init the shell */
  1934. _obj_runChar_beforeRun(self, shell);
  1935. /* getchar and run */
  1936. while (1) {
  1937. char inputChar = shell->fn_getchar();
  1938. if (SHELL_CTRL_EXIT == _do_obj_runChar(self, inputChar, shell)) {
  1939. break;
  1940. }
  1941. }
  1942. }
  1943. static enum shellCTRL __obj_shellLineHandler_REPL(PikaObj* self,
  1944. char* input_line,
  1945. ShellConfig* shell) {
  1946. /* exit */
  1947. if (strEqu("exit()", input_line)) {
  1948. /* exit pika shell */
  1949. return SHELL_CTRL_EXIT;
  1950. }
  1951. /* run single line */
  1952. pikaVM_run_ex_cfg cfg = {0};
  1953. cfg.in_repl = pika_true;
  1954. pikaVM_run_ex(self, input_line, &cfg);
  1955. return SHELL_CTRL_CONTINUE;
  1956. }
  1957. static volatile ShellConfig g_REPL = {
  1958. .handler = __obj_shellLineHandler_REPL,
  1959. .prefix = ">>> ",
  1960. .blockBuffName = "@sh0",
  1961. #if PIKA_SHELL_HISTORY_ENABLE
  1962. .history = NULL,
  1963. #endif
  1964. .no_echo = PIKA_SHELL_NO_ECHO,
  1965. };
  1966. void pikaScriptShell_withGetchar(PikaObj* self, sh_getchar getchar_fn) {
  1967. g_REPL.fn_getchar = getchar_fn;
  1968. _do_pikaScriptShell(self, (ShellConfig*)&g_REPL);
  1969. }
  1970. int shConfig_deinit(ShellConfig* self) {
  1971. #if PIKA_SHELL_HISTORY_ENABLE
  1972. if (NULL != self->history) {
  1973. shHistory_destroy(self->history);
  1974. self->history = NULL;
  1975. }
  1976. #endif
  1977. return 0;
  1978. }
  1979. char pika_repl_getchar(void) {
  1980. char c = 0;
  1981. pika_platform_repl_recv((uint8_t*)&c, 1, PIKA_TIMEOUT_FOREVER);
  1982. return c;
  1983. }
  1984. void pikaPythonShell(PikaObj* self) {
  1985. pikaScriptShell_withGetchar(self, pika_repl_getchar);
  1986. }
  1987. void pikaShellSetEcho(pika_bool enable_echo) {
  1988. if (enable_echo) {
  1989. g_REPL.no_echo = pika_false;
  1990. } else {
  1991. g_REPL.no_echo = pika_true;
  1992. }
  1993. }
  1994. void obj_setErrorCode(PikaObj* self, int32_t errCode) {
  1995. pika_assert(NULL != self->vmFrame);
  1996. self->vmFrame->error.code = errCode;
  1997. }
  1998. void method_returnBytes(Args* args, uint8_t* val) {
  1999. args_pushArg_name(args, "@rt", arg_newBytes(val, PIKA_BYTES_DEFAULT_SIZE));
  2000. }
  2001. void method_returnStr(Args* args, char* val) {
  2002. args_pushArg_name(args, "@rt", arg_newStr(val));
  2003. }
  2004. void method_returnInt(Args* args, int64_t val) {
  2005. args_pushArg_name(args, "@rt", arg_newInt(val));
  2006. }
  2007. void method_returnBool(Args* args, pika_bool val) {
  2008. if (val == _PIKA_BOOL_ERR) {
  2009. return;
  2010. }
  2011. args_pushArg_name(args, "@rt", arg_newBool(val));
  2012. }
  2013. void method_returnFloat(Args* args, pika_float val) {
  2014. args_pushArg_name(args, "@rt", arg_newFloat(val));
  2015. }
  2016. void method_returnPtr(Args* args, void* val) {
  2017. args_pushArg_name(args, "@rt", arg_newPtr(ARG_TYPE_POINTER, val));
  2018. }
  2019. void method_returnObj(Args* args, void* val) {
  2020. if (NULL == val) {
  2021. args_pushArg_name(args, "@rt", arg_newNone());
  2022. return;
  2023. }
  2024. ArgType type;
  2025. PikaObj* obj = (PikaObj*)val;
  2026. if (obj_getFlag(obj, OBJ_FLAG_ALREADY_INIT)) {
  2027. type = ARG_TYPE_OBJECT;
  2028. } else {
  2029. type = ARG_TYPE_OBJECT_NEW;
  2030. }
  2031. args_pushArg_name(args, "@rt", arg_newPtr(type, val));
  2032. }
  2033. void method_returnArg(Args* args, Arg* arg) {
  2034. args_pushArg_name(args, "@rt", arg);
  2035. }
  2036. int64_t method_getInt(Args* args, char* argName) {
  2037. return args_getInt(args, argName);
  2038. }
  2039. pika_float method_getFloat(Args* args, char* argName) {
  2040. return args_getFloat(args, argName);
  2041. }
  2042. char* method_getStr(Args* args, char* argName) {
  2043. return args_getStr(args, argName);
  2044. }
  2045. #if PIKA_GC_MARK_SWEEP_ENABLE
  2046. PikaObj* pikaGC_getLast(PikaObj* self) {
  2047. PikaObj* obj = g_PikaObjState.gcChain;
  2048. PikaObj* last = NULL;
  2049. while (NULL != obj) {
  2050. if (obj == self) {
  2051. return last;
  2052. }
  2053. last = obj;
  2054. obj = obj->gcNext;
  2055. }
  2056. return NULL;
  2057. }
  2058. void pikaGC_clean(PikaGC* gc) {
  2059. PikaObj* obj = g_PikaObjState.gcChain;
  2060. while (NULL != obj) {
  2061. obj_clearFlag(obj, OBJ_FLAG_GC_MARKED);
  2062. // obj->gcRoot = NULL;
  2063. obj = obj->gcNext;
  2064. }
  2065. }
  2066. uint32_t pikaGC_count(void) {
  2067. uint32_t count = 0;
  2068. PikaObj* obj = g_PikaObjState.gcChain;
  2069. while (NULL != obj) {
  2070. count++;
  2071. obj = obj->gcNext;
  2072. }
  2073. return count;
  2074. }
  2075. uint32_t pikaGC_countMarked(void) {
  2076. uint32_t count = 0;
  2077. PikaObj* obj = g_PikaObjState.gcChain;
  2078. while (NULL != obj) {
  2079. if (obj_getFlag(obj, OBJ_FLAG_GC_MARKED)) {
  2080. count++;
  2081. }
  2082. obj = obj->gcNext;
  2083. }
  2084. return count;
  2085. }
  2086. uint32_t pikaGC_printFreeList(void) {
  2087. uint32_t count = 0;
  2088. PikaObj* obj = g_PikaObjState.gcChain;
  2089. pika_platform_printf("-----\r\n");
  2090. while (NULL != obj) {
  2091. if (!obj_getFlag(obj, OBJ_FLAG_GC_MARKED)) {
  2092. count++;
  2093. pika_platform_printf("gc free: ");
  2094. obj_dump(obj);
  2095. }
  2096. obj = obj->gcNext;
  2097. }
  2098. pika_platform_printf("-----\r\n");
  2099. return count;
  2100. }
  2101. void obj_dump(PikaObj* self) {
  2102. #if PIKA_KERNAL_DEBUG_ENABLE
  2103. pika_platform_printf("[\033[32m%s\033[0m]", self->name);
  2104. #endif
  2105. pika_platform_printf(" \033[36m@%p\033[0m", self);
  2106. pika_platform_printf("\r\n");
  2107. }
  2108. uint32_t pikaGC_markSweepOnce(PikaGC* gc) {
  2109. _pikaGC_mark(gc);
  2110. uint32_t count = 0;
  2111. PikaObj* freeList[16] = {0};
  2112. PikaObj* obj = g_PikaObjState.gcChain;
  2113. while (NULL != obj) {
  2114. if (!obj_getFlag(obj, OBJ_FLAG_GC_MARKED)) {
  2115. if (count > dimof(freeList) - 1) {
  2116. break;
  2117. }
  2118. freeList[count] = obj;
  2119. count++;
  2120. }
  2121. obj = obj->gcNext;
  2122. }
  2123. if (count > 0) {
  2124. // pikaGC_markDump();
  2125. // pikaGC_printFreeList();
  2126. for (uint32_t i = 0; i < count; i++) {
  2127. pika_platform_printf("GC Free:");
  2128. obj_dump(freeList[i]);
  2129. }
  2130. for (uint32_t i = 0; i < count; i++) {
  2131. obj_GC(freeList[i]);
  2132. }
  2133. }
  2134. return count;
  2135. }
  2136. int32_t _pikaGC_markHandler(Arg* argEach, void* context) {
  2137. PikaGC* gc = (PikaGC*)context;
  2138. if (arg_isObject(argEach)) {
  2139. PikaObj* obj = (PikaObj*)arg_getPtr(argEach);
  2140. #if PIKA_KERNAL_DEBUG_ENABLE
  2141. obj->gcRoot = (void*)gc->oThis;
  2142. #endif
  2143. pikaGC_markObj(gc, obj);
  2144. }
  2145. return 0;
  2146. }
  2147. void pikaGC_markObj(PikaGC* gc, PikaObj* self) {
  2148. gc->oThis = self;
  2149. gc->markDeepth++;
  2150. if (NULL == self) {
  2151. goto __exit;
  2152. }
  2153. if (obj_getFlag(self, OBJ_FLAG_GC_MARKED)) {
  2154. goto __exit;
  2155. }
  2156. obj_setFlag(self, OBJ_FLAG_GC_MARKED);
  2157. if (NULL != gc->onMarkObj) {
  2158. gc->onMarkObj(gc);
  2159. }
  2160. args_foreach(self->list, _pikaGC_markHandler, gc);
  2161. if (self->constructor == New_PikaStdData_Dict) {
  2162. Args* dict = _OBJ2DICT(self);
  2163. if (NULL == dict) {
  2164. goto __exit;
  2165. }
  2166. args_foreach(dict, _pikaGC_markHandler, (void*)gc);
  2167. goto __exit;
  2168. }
  2169. if (self->constructor == New_PikaStdData_List ||
  2170. self->constructor == New_PikaStdData_Tuple) {
  2171. Args* list = _OBJ2LIST(self);
  2172. if (NULL == list) {
  2173. goto __exit;
  2174. }
  2175. args_foreach(list, _pikaGC_markHandler, (void*)gc);
  2176. goto __exit;
  2177. }
  2178. __exit:
  2179. gc->markDeepth--;
  2180. return;
  2181. }
  2182. void _pikaGC_mark(PikaGC* gc) {
  2183. pikaGC_clean(gc);
  2184. PikaObj* root = g_PikaObjState.gcChain;
  2185. while (NULL != root) {
  2186. if (obj_getFlag(root, OBJ_FLAG_GC_ROOT)) {
  2187. pikaGC_markObj(gc, root);
  2188. }
  2189. root = root->gcNext;
  2190. }
  2191. }
  2192. void pikaGC_mark(void) {
  2193. PikaGC gc = {0};
  2194. _pikaGC_mark(&gc);
  2195. }
  2196. int _pikaGC_markDumpHandler(PikaGC* gc) {
  2197. for (uint32_t i = 0; i < gc->markDeepth - 1; i++) {
  2198. pika_platform_printf(" |");
  2199. }
  2200. if (gc->markDeepth != 1) {
  2201. pika_platform_printf("- ");
  2202. }
  2203. obj_dump(gc->oThis);
  2204. return 0;
  2205. }
  2206. #endif
  2207. pika_bool obj_checkAlive(PikaObj* self) {
  2208. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2209. return pika_true;
  2210. #else
  2211. pika_bool ret = pika_false;
  2212. if (NULL == g_PikaObjState.gcChain) {
  2213. ret = pika_false;
  2214. goto __exit;
  2215. }
  2216. PikaObj* obj = g_PikaObjState.gcChain;
  2217. while (NULL != obj) {
  2218. if (obj == self) {
  2219. ret = pika_true;
  2220. goto __exit;
  2221. }
  2222. obj = obj->gcNext;
  2223. }
  2224. __exit:
  2225. #if PIKA_KERNAL_DEBUG_ENABLE
  2226. if (ret == pika_true) {
  2227. self->isAlive = ret;
  2228. }
  2229. #endif
  2230. return ret;
  2231. #endif
  2232. }
  2233. uint32_t pikaGC_markSweep(void) {
  2234. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2235. return 0;
  2236. #else
  2237. PikaGC gc = {0};
  2238. uint32_t count = 0;
  2239. if (pikaGC_islock()) {
  2240. return 0;
  2241. }
  2242. pikaGC_lock();
  2243. while (pikaGC_markSweepOnce(&gc) != 0) {
  2244. count++;
  2245. };
  2246. if (count > 0) {
  2247. // pikaGC_markDump();
  2248. }
  2249. /* update gc state */
  2250. g_PikaObjState.objCntLastGC = g_PikaObjState.objCnt;
  2251. pikaGC_unlock();
  2252. return count;
  2253. #endif
  2254. }
  2255. void pikaGC_markDump(void) {
  2256. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2257. return;
  2258. #else
  2259. PikaGC gc = {0};
  2260. pika_platform_printf(
  2261. "\033[32m"
  2262. "========= PIKA GC DUMP =========\r\n"
  2263. "\033[0m");
  2264. gc.onMarkObj = _pikaGC_markDumpHandler;
  2265. _pikaGC_mark(&gc);
  2266. #endif
  2267. }
  2268. void pikaGC_checkThreshold(void) {
  2269. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2270. return;
  2271. #else
  2272. if (g_PikaObjState.objCnt >
  2273. g_PikaObjState.objCntLastGC + PIKA_GC_MARK_SWEEP_THRESHOLD) {
  2274. pikaGC_markSweep();
  2275. }
  2276. #endif
  2277. }
  2278. void pikaGC_append(PikaObj* self) {
  2279. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2280. return;
  2281. #else
  2282. g_PikaObjState.objCnt++;
  2283. if (g_PikaObjState.objCntMax < g_PikaObjState.objCnt) {
  2284. g_PikaObjState.objCntMax = g_PikaObjState.objCnt;
  2285. }
  2286. /* gc single chain */
  2287. if (NULL == g_PikaObjState.gcChain) {
  2288. g_PikaObjState.gcChain = self;
  2289. return;
  2290. }
  2291. /* append to head of gc chain */
  2292. self->gcNext = g_PikaObjState.gcChain;
  2293. g_PikaObjState.gcChain = self;
  2294. #endif
  2295. }
  2296. void obj_removeGcChain(PikaObj* self) {
  2297. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2298. return;
  2299. #else
  2300. g_PikaObjState.objCnt--;
  2301. PikaObj* last = pikaGC_getLast(self);
  2302. if (NULL == last) {
  2303. /* remove head */
  2304. g_PikaObjState.gcChain = self->gcNext;
  2305. return;
  2306. }
  2307. last->gcNext = self->gcNext;
  2308. #endif
  2309. }
  2310. void obj_enableGC(PikaObj* self) {
  2311. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2312. return;
  2313. #else
  2314. obj_clearFlag(self, OBJ_FLAG_GC_ROOT);
  2315. #endif
  2316. }
  2317. void pikaGC_lock(void) {
  2318. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2319. return;
  2320. #else
  2321. g_PikaObjState.markSweepBusy++;
  2322. #endif
  2323. }
  2324. void pikaGC_unlock(void) {
  2325. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2326. return;
  2327. #else
  2328. g_PikaObjState.markSweepBusy--;
  2329. #endif
  2330. }
  2331. pika_bool pikaGC_islock(void) {
  2332. #if !PIKA_GC_MARK_SWEEP_ENABLE
  2333. return pika_false;
  2334. #else
  2335. return g_PikaObjState.markSweepBusy > 0;
  2336. #endif
  2337. }
  2338. PikaObj* New_PikaObj(Args* args) {
  2339. PikaObj* self = pikaMalloc(sizeof(PikaObj));
  2340. /* List */
  2341. self->list = New_args(NULL);
  2342. self->refcnt = 0;
  2343. self->constructor = New_PikaObj;
  2344. self->flag = 0;
  2345. self->vmFrame = NULL;
  2346. #if PIKA_GC_MARK_SWEEP_ENABLE
  2347. self->gcNext = NULL;
  2348. obj_setFlag(self, OBJ_FLAG_GC_ROOT);
  2349. #endif
  2350. #if PIKA_KERNAL_DEBUG_ENABLE
  2351. self->aName = NULL;
  2352. self->name = "PikaObj";
  2353. self->parent = NULL;
  2354. self->isAlive = pika_true;
  2355. #endif
  2356. #if PIKA_GC_MARK_SWEEP_ENABLE && PIKA_KERNAL_DEBUG_ENABLE
  2357. self->gcRoot = NULL;
  2358. #endif
  2359. /* append to gc chain */
  2360. pikaGC_append(self);
  2361. pikaGC_checkThreshold();
  2362. return self;
  2363. }
  2364. Arg* arg_setObj(Arg* self, char* name, PikaObj* obj) {
  2365. return arg_setPtr(self, name, ARG_TYPE_OBJECT, obj);
  2366. }
  2367. Arg* arg_setRef(Arg* self, char* name, PikaObj* obj) {
  2368. pika_assert(NULL != obj);
  2369. obj_refcntInc(obj);
  2370. return arg_setObj(self, name, obj);
  2371. }
  2372. Arg* arg_setWeakRef(Arg* self, char* name, PikaObj* obj) {
  2373. pika_assert(NULL != obj);
  2374. return arg_setPtr(self, name, ARG_TYPE_OBJECT_WEAK, obj);
  2375. }
  2376. int32_t obj_newDirectObj(PikaObj* self, char* objName, NewFun newFunPtr) {
  2377. Arg* aNewObj = arg_newDirectObj(newFunPtr);
  2378. aNewObj = arg_setName(aNewObj, objName);
  2379. obj_setName(arg_getPtr(aNewObj), objName);
  2380. arg_setType(aNewObj, ARG_TYPE_OBJECT);
  2381. // pikaGC_enable(arg_getPtr(aNewObj));
  2382. obj_setArg_noCopy(self, objName, aNewObj);
  2383. return 0;
  2384. }
  2385. int32_t obj_newHostObj(PikaObj* self, char* objName) {
  2386. Args buffs = {0};
  2387. size_t tokenCnt = strCountSign(objName, '.');
  2388. if (0 == tokenCnt) {
  2389. return 0;
  2390. }
  2391. PikaObj* this = self;
  2392. objName = strsCopy(&buffs, objName);
  2393. for (int i = 0; i < tokenCnt; i++) {
  2394. char* name = strsPopToken(&buffs, &objName, '.');
  2395. if (!obj_isArgExist(this, name)) {
  2396. obj_newDirectObj(this, name, New_TinyObj);
  2397. this = obj_getObj(this, name);
  2398. }
  2399. }
  2400. strsDeinit(&buffs);
  2401. return 0;
  2402. }
  2403. int32_t obj_newMetaObj(PikaObj* self, char* objName, NewFun newFunPtr) {
  2404. /* add meta Obj, no inited */
  2405. Arg* aMetaObj = arg_newMetaObj(newFunPtr);
  2406. aMetaObj = arg_setName(aMetaObj, objName);
  2407. args_setArg(self->list, aMetaObj);
  2408. return 0;
  2409. }
  2410. static void _append_help(char* name) {
  2411. if (NULL == g_PikaObjState.helpModulesCmodule) {
  2412. g_PikaObjState.helpModulesCmodule = arg_newStr("");
  2413. }
  2414. Arg* _help = g_PikaObjState.helpModulesCmodule;
  2415. _help = arg_strAppend(_help, name);
  2416. _help = arg_strAppend(_help, "\r\n");
  2417. g_PikaObjState.helpModulesCmodule = _help;
  2418. }
  2419. int32_t obj_newObj(PikaObj* self,
  2420. char* objName,
  2421. char* className,
  2422. NewFun newFunPtr) {
  2423. /* before init root object */
  2424. if (g_PikaObjState.inRootObj) {
  2425. _append_help(objName);
  2426. }
  2427. return obj_newMetaObj(self, objName, newFunPtr);
  2428. }
  2429. PikaObj* obj_importModuleWithByteCode(PikaObj* self,
  2430. char* name,
  2431. uint8_t* byteCode) {
  2432. if (!obj_isArgExist((PikaObj*)__pikaMain, name)) {
  2433. /* import to main module context */
  2434. obj_newHostObj((PikaObj*)__pikaMain, name);
  2435. obj_newDirectObj((PikaObj*)__pikaMain, name, New_TinyObj);
  2436. PikaObj* module_obj = obj_getObj((PikaObj*)__pikaMain, name);
  2437. pikaVM_runBytecode_ex_cfg cfg = {0};
  2438. cfg.globals = module_obj;
  2439. cfg.locals = module_obj;
  2440. cfg.name = name;
  2441. cfg.vm_thread = self->vmFrame->vm_thread;
  2442. cfg.is_const_bytecode = pika_true;
  2443. pikaVM_runByteCode_ex(module_obj, byteCode, &cfg);
  2444. }
  2445. if (self != (PikaObj*)__pikaMain) {
  2446. /* import to other module context */
  2447. Arg* aModule = obj_getArg((PikaObj*)__pikaMain, name);
  2448. PikaObj* oModule = arg_getPtr(aModule);
  2449. obj_newHostObj(self, name);
  2450. obj_setArg(self, name, aModule);
  2451. arg_setIsWeakRef(obj_getArg(self, name), pika_true);
  2452. pika_assert(arg_isObject(aModule));
  2453. /* decrase refcnt to avoid circle reference */
  2454. obj_refcntDec(oModule);
  2455. }
  2456. return self;
  2457. }
  2458. PikaObj* obj_importModuleWithByteCodeFrame(PikaObj* self,
  2459. char* name,
  2460. ByteCodeFrame* byteCode_frame) {
  2461. PikaObj* New_PikaStdLib_SysObj(Args * args);
  2462. obj_newDirectObj(self, name, New_PikaStdLib_SysObj);
  2463. pikaVM_runByteCodeFrame(obj_getObj(self, name), byteCode_frame);
  2464. return self;
  2465. }
  2466. int obj_linkLibraryFile(PikaObj* self, char* input_file_name) {
  2467. obj_newMetaObj(self, "@lib", New_LibObj);
  2468. LibObj* lib = obj_getObj(self, "@lib");
  2469. return LibObj_loadLibraryFile(lib, input_file_name);
  2470. }
  2471. PikaObj* obj_linkLibrary(PikaObj* self, uint8_t* library_bytes) {
  2472. obj_newMetaObj(self, "@lib", New_LibObj);
  2473. LibObj* lib = obj_getObj(self, "@lib");
  2474. LibObj_loadLibrary(lib, library_bytes);
  2475. return self;
  2476. }
  2477. void obj_printModules(PikaObj* self) {
  2478. LibObj* lib = obj_getObj(self, "@lib");
  2479. if (lib == NULL) {
  2480. pika_platform_printf(
  2481. "Error: Not found LibObj, please execute obj_linkLibrary()\r\n");
  2482. return;
  2483. }
  2484. pika_platform_printf(arg_getStr((Arg*)g_PikaObjState.helpModulesCmodule));
  2485. LibObj_printModules(lib);
  2486. }
  2487. PikaObj* obj_linkLibObj(PikaObj* self, LibObj* library) {
  2488. obj_setRef(self, "@lib", library);
  2489. return self;
  2490. }
  2491. LibObj* pika_getLibObj(void) {
  2492. // Ensure __pikaMain exists
  2493. if (__pikaMain == NULL) {
  2494. return NULL;
  2495. }
  2496. // Cast __pikaMain to PikaObj and fetch library
  2497. PikaObj* self = (PikaObj*)__pikaMain;
  2498. return obj_getPtr(self, "@lib");
  2499. }
  2500. uint8_t* pika_getByteCodeFromModule(char* module_name) {
  2501. // Check if module_name is not NULL
  2502. pika_assert(NULL != module_name);
  2503. // Fetch library using pika_getLibObj
  2504. LibObj* lib = pika_getLibObj();
  2505. // Exit if there's no library
  2506. if (NULL == lib) {
  2507. return NULL;
  2508. }
  2509. // Fetch the module from the library
  2510. PikaObj* module = LibObj_getModule(lib, module_name);
  2511. // Check if module exists
  2512. if (NULL == module) {
  2513. return NULL;
  2514. }
  2515. // Return bytecode of the module
  2516. return obj_getPtr(module, "bytecode");
  2517. }
  2518. int obj_runModule(PikaObj* self, char* module_name) {
  2519. uint8_t* bytecode = pika_getByteCodeFromModule(module_name);
  2520. if (NULL == bytecode) {
  2521. return 1;
  2522. }
  2523. PikaVMThread* vm_thread = pikaVMThread_require();
  2524. pikaVM_runBytecode_ex_cfg cfg = {0};
  2525. cfg.globals = self;
  2526. cfg.locals = self;
  2527. cfg.name = module_name;
  2528. cfg.vm_thread = vm_thread;
  2529. cfg.is_const_bytecode = pika_true;
  2530. pikaVM_runByteCode_ex(self, bytecode, &cfg);
  2531. return 0;
  2532. }
  2533. PikaObj* obj_runFile(PikaObj* self, char* file_name) {
  2534. return pikaVM_runFile(self, file_name);
  2535. }
  2536. PikaObj* obj_runSingleFile(PikaObj* self, char* file_name) {
  2537. return pikaVM_runSingleFile(self, file_name);
  2538. }
  2539. int obj_importModule(PikaObj* self, char* module_name) {
  2540. if (NULL == module_name) {
  2541. return -1;
  2542. }
  2543. /* import bytecode of the module */
  2544. uint8_t* bytecode = pika_getByteCodeFromModule(module_name);
  2545. if (NULL == bytecode) {
  2546. return -1;
  2547. }
  2548. obj_importModuleWithByteCode(self, module_name, bytecode);
  2549. return 0;
  2550. }
  2551. PikaObj* arg_getObj(Arg* self) {
  2552. return (PikaObj*)arg_getPtr(self);
  2553. }
  2554. char* obj_toStr(PikaObj* self) {
  2555. /* check method arg */
  2556. Arg* aMethod = obj_getMethodArgWithFullPath(self, "__str__");
  2557. if (NULL != aMethod) {
  2558. Arg* aStr = obj_runMethodArg0(self, aMethod);
  2559. char* str_res = obj_cacheStr(self, arg_getStr(aStr));
  2560. arg_deinit(aStr);
  2561. return str_res;
  2562. }
  2563. /* normal object */
  2564. Args buffs = {0};
  2565. char* str_res =
  2566. strsFormat(&buffs, PIKA_SPRINTF_BUFF_SIZE, "<object at %p>", self);
  2567. obj_setStr(self, "@res_str", str_res);
  2568. strsDeinit(&buffs);
  2569. return obj_getStr(self, "@res_str");
  2570. }
  2571. void pika_eventListener_registEventHandler(PikaEventListener* self,
  2572. uintptr_t eventId,
  2573. PikaObj* eventHandleObj) {
  2574. Args buffs = {0};
  2575. char* event_name =
  2576. strsFormat(&buffs, PIKA_SPRINTF_BUFF_SIZE, "%ld", eventId);
  2577. obj_newDirectObj(self, event_name, New_TinyObj);
  2578. PikaObj* event_item = obj_getObj(self, event_name);
  2579. obj_setRef(event_item, "eventHandleObj", eventHandleObj);
  2580. strsDeinit(&buffs);
  2581. }
  2582. PIKA_RES obj_setEventCallback(PikaObj* self,
  2583. uintptr_t eventId,
  2584. Arg* eventCallback,
  2585. PikaEventListener* listener) {
  2586. obj_setArg(self, "eventCallBack", eventCallback);
  2587. if (argType_isObjectMethodActive(arg_getType(eventCallback))) {
  2588. PikaObj* method_self = methodArg_getHostObj(eventCallback);
  2589. obj_setRef(self, "eventCallBackHost", method_self);
  2590. }
  2591. pika_eventListener_registEventHandler(listener, eventId, self);
  2592. return PIKA_RES_OK;
  2593. }
  2594. void pika_eventListener_registEventCallback(PikaEventListener* listener,
  2595. uintptr_t eventId,
  2596. Arg* eventCallback) {
  2597. pika_assert(NULL != listener);
  2598. char hash_str[32] = {0};
  2599. pika_sprintf(hash_str, "C%d", eventId);
  2600. obj_newDirectObj(listener, hash_str, New_TinyObj);
  2601. PikaObj* oHandle = obj_getPtr(listener, hash_str);
  2602. obj_setEventCallback(oHandle, eventId, eventCallback, listener);
  2603. }
  2604. Args buffs = {0};
  2605. void pika_eventListener_removeEvent(PikaEventListener* self,
  2606. uintptr_t eventId) {
  2607. char* event_name =
  2608. strsFormat(&buffs, PIKA_SPRINTF_BUFF_SIZE, "%ld", eventId);
  2609. obj_removeArg(self, event_name);
  2610. strsDeinit(&buffs);
  2611. }
  2612. PikaObj* pika_eventListener_getEventHandleObj(PikaEventListener* self,
  2613. uintptr_t eventId) {
  2614. Args buffs = {0};
  2615. char* event_name =
  2616. strsFormat(&buffs, PIKA_SPRINTF_BUFF_SIZE, "%ld", eventId);
  2617. PikaObj* event_item = obj_getObj(self, event_name);
  2618. PikaObj* eventHandleObj = obj_getPtr(event_item, "eventHandleObj");
  2619. strsDeinit(&buffs);
  2620. return eventHandleObj;
  2621. }
  2622. void pika_eventListener_init(PikaEventListener** p_self) {
  2623. *p_self = newNormalObj(New_TinyObj);
  2624. }
  2625. void pika_eventListener_deinit(PikaEventListener** p_self) {
  2626. if (NULL != *p_self) {
  2627. obj_deinit(*p_self);
  2628. *p_self = NULL;
  2629. }
  2630. }
  2631. Arg* pika_runFunction0_ex(PikaObj* host, Arg* functionArg) {
  2632. obj_setArg(host, "@f", functionArg);
  2633. /* clang-format off */
  2634. PIKA_PYTHON(
  2635. @r = @f()
  2636. )
  2637. /* clang-format on */
  2638. const uint8_t bytes[] = {
  2639. 0x08, 0x00, 0x00, 0x00, /* instruct array size */
  2640. 0x00, 0x82, 0x01, 0x00, 0x00, 0x04, 0x04, 0x00, /* instruct array */
  2641. 0x07, 0x00, 0x00, 0x00, /* const pool size */
  2642. 0x00, 0x40, 0x66, 0x00, 0x40, 0x72, 0x00, /* const pool */
  2643. };
  2644. Arg* res = pikaVM_runByteCodeReturn(host, bytes, "@r");
  2645. arg_deinit(functionArg);
  2646. return res;
  2647. }
  2648. Arg* pika_runFunction1_ex(PikaObj* host, Arg* functionArg, Arg* arg1) {
  2649. obj_setArg(host, "@d", arg1);
  2650. obj_setArg(host, "@f", functionArg);
  2651. /* clang-format off */
  2652. PIKA_PYTHON(
  2653. @r = @f(@d)
  2654. )
  2655. /* clang-format on */
  2656. const uint8_t bytes[] = {
  2657. 0x0c, 0x00, 0x00, 0x00, /* instruct array size */
  2658. 0x10, 0x81, 0x01, 0x00, 0x00, 0x02, 0x04, 0x00, 0x00, 0x04, 0x07, 0x00,
  2659. /* instruct array */
  2660. 0x0a, 0x00, 0x00, 0x00, /* const pool size */
  2661. 0x00, 0x40, 0x64, 0x00, 0x40, 0x66, 0x00, 0x40, 0x72,
  2662. 0x00, /* const pool */
  2663. };
  2664. Arg* res = pikaVM_runByteCodeReturn(host, bytes, "@r");
  2665. arg_deinit(arg1);
  2666. arg_deinit(functionArg);
  2667. return res;
  2668. }
  2669. Arg* pika_runFunction2_ex(PikaObj* host,
  2670. Arg* functionArg,
  2671. Arg* arg1,
  2672. Arg* arg2) {
  2673. obj_setArg(host, "@d", arg1);
  2674. obj_setArg(host, "@e", arg2);
  2675. obj_setArg(host, "@f", functionArg);
  2676. /* clang-format off */
  2677. PIKA_PYTHON(
  2678. @r = @f(@d, @e)
  2679. )
  2680. /* clang-format on */
  2681. const uint8_t bytes[] = {
  2682. 0x10, 0x00, 0x00, 0x00, /* instruct array size */
  2683. 0x10, 0x81, 0x01, 0x00, 0x10, 0x01, 0x04, 0x00,
  2684. 0x00, 0x02, 0x07, 0x00, 0x00, 0x04, 0x0a, 0x00, /* instruct array */
  2685. 0x0d, 0x00, 0x00, 0x00, /* const pool size */
  2686. 0x00, 0x40, 0x64, 0x00, 0x40, 0x65, 0x00, 0x40,
  2687. 0x66, 0x00, 0x40, 0x72, 0x00,
  2688. /* const pool */
  2689. };
  2690. Arg* res = pikaVM_runByteCodeReturn(host, bytes, "@r");
  2691. arg_deinit(arg1);
  2692. arg_deinit(arg2);
  2693. arg_deinit(functionArg);
  2694. return res;
  2695. }
  2696. Arg* pika_runFunction0(Arg* functionArg) {
  2697. PikaObj* locals = New_TinyObj(NULL);
  2698. Arg* res = pika_runFunction0_ex(locals, functionArg);
  2699. obj_deinit(locals);
  2700. return res;
  2701. }
  2702. Arg* pika_runFunction1(Arg* functionArg, Arg* arg1) {
  2703. PikaObj* locals = New_TinyObj(NULL);
  2704. Arg* res = pika_runFunction1_ex(locals, functionArg, arg1);
  2705. obj_deinit(locals);
  2706. return res;
  2707. }
  2708. Arg* pika_runFunction2(Arg* functionArg, Arg* arg1, Arg* arg2) {
  2709. PikaObj* locals = New_TinyObj(NULL);
  2710. Arg* res = pika_runFunction2_ex(locals, functionArg, arg1, arg2);
  2711. obj_deinit(locals);
  2712. return res;
  2713. }
  2714. Arg* obj_runMethod0(PikaObj* self, char* methodName) {
  2715. Arg* aMethod = obj_getMethodArg(self, methodName);
  2716. if (NULL == aMethod) {
  2717. return NULL;
  2718. }
  2719. Arg* aRes = pika_runFunction0_ex(self, aMethod);
  2720. return aRes;
  2721. }
  2722. Arg* obj_runMethod1(PikaObj* self, char* methodName, Arg* arg1) {
  2723. Arg* aMethod = obj_getMethodArg(self, methodName);
  2724. if (NULL == aMethod) {
  2725. return NULL;
  2726. }
  2727. Arg* aRes = pika_runFunction1_ex(self, aMethod, arg1);
  2728. return aRes;
  2729. }
  2730. Arg* obj_runMethod2(PikaObj* self, char* methodName, Arg* arg1, Arg* arg2) {
  2731. Arg* aMethod = obj_getMethodArg(self, methodName);
  2732. if (NULL == aMethod) {
  2733. return NULL;
  2734. }
  2735. Arg* aRes = pika_runFunction2_ex(self, aMethod, arg1, arg2);
  2736. return aRes;
  2737. }
  2738. Arg* obj_runMethodArg0(PikaObj* self, Arg* methodArg) {
  2739. return pika_runFunction0_ex(self, methodArg);
  2740. }
  2741. Arg* obj_runMethodArg1(PikaObj* self, Arg* methodArg, Arg* arg1) {
  2742. return pika_runFunction1_ex(self, methodArg, arg1);
  2743. }
  2744. Arg* obj_runMethodArg2(PikaObj* self, Arg* methodArg, Arg* arg1, Arg* arg2) {
  2745. return pika_runFunction2_ex(self, methodArg, arg1, arg2);
  2746. }
  2747. Arg* __eventListener_runEvent(PikaEventListener* listener,
  2748. uintptr_t eventId,
  2749. Arg* eventData) {
  2750. PikaObj* handler = pika_eventListener_getEventHandleObj(listener, eventId);
  2751. pika_debug("event handler: %p", handler);
  2752. if (NULL == handler) {
  2753. pika_platform_printf(
  2754. "Error: can not find event handler by id: [0x%02x]\r\n", eventId);
  2755. return NULL;
  2756. }
  2757. Arg* eventCallBack = obj_getArg(handler, "eventCallBack");
  2758. pika_debug("run event handler: %p", handler);
  2759. Arg* res = pika_runFunction1(arg_copy(eventCallBack), arg_copy(eventData));
  2760. return res;
  2761. }
  2762. void pika_debug_bytes(uint8_t* buff, size_t len) {
  2763. pika_debug_raw("[");
  2764. for (size_t i = 0; i < len; i++) {
  2765. pika_debug_raw("0x%02X", buff[i]);
  2766. if (i < len - 1) {
  2767. pika_debug_raw(",");
  2768. }
  2769. }
  2770. pika_debug_raw("]\n");
  2771. }
  2772. static void _thread_event(void* arg) {
  2773. pika_assert(_VM_is_first_lock());
  2774. while (1) {
  2775. pika_GIL_ENTER();
  2776. #if PIKA_EVENT_ENABLE
  2777. if (g_PikaVMState.event_thread_exit) {
  2778. g_PikaVMState.event_thread_exit_done = 1;
  2779. break;
  2780. }
  2781. #endif
  2782. _VMEvent_pickupEvent();
  2783. pika_GIL_EXIT();
  2784. pika_platform_thread_yield();
  2785. }
  2786. pika_GIL_EXIT();
  2787. }
  2788. PIKA_RES _do_pika_eventListener_send(PikaEventListener* self,
  2789. uintptr_t eventId,
  2790. Arg* eventData,
  2791. int eventSignal,
  2792. PIKA_BOOL pickupWhenNoVM) {
  2793. #if !PIKA_EVENT_ENABLE
  2794. pika_platform_printf("PIKA_EVENT_ENABLE is not enable");
  2795. while (1) {
  2796. };
  2797. #else
  2798. if (NULL != eventData && !_VM_is_first_lock()) {
  2799. #if PIKA_EVENT_THREAD_ENABLE
  2800. _VM_lock_init();
  2801. #else
  2802. pika_platform_printf(
  2803. "Error: can not send arg event data without thread support\r\n");
  2804. arg_deinit(eventData);
  2805. return PIKA_RES_ERR_RUNTIME_ERROR;
  2806. #endif
  2807. }
  2808. if (NULL == eventData) {
  2809. // for event signal
  2810. if (PIKA_RES_OK !=
  2811. __eventListener_pushSignal(self, eventId, eventSignal)) {
  2812. return PIKA_RES_ERR_RUNTIME_ERROR;
  2813. }
  2814. }
  2815. /* using multi thread */
  2816. if (pika_GIL_isInit()) {
  2817. /* python thread is running */
  2818. /* wait python thread get first lock */
  2819. while (1) {
  2820. if (_VM_is_first_lock()) {
  2821. break;
  2822. }
  2823. if (g_PikaVMState.vm_cnt == 0) {
  2824. break;
  2825. }
  2826. if (pika_GIL_getBareLock() == 0) {
  2827. break;
  2828. }
  2829. pika_platform_thread_yield();
  2830. }
  2831. pika_GIL_ENTER();
  2832. #if PIKA_EVENT_THREAD_ENABLE
  2833. if (!g_PikaVMState.event_thread) {
  2834. // avoid _VMEvent_pickupEvent() in _time.c as soon as
  2835. // possible
  2836. g_PikaVMState.event_thread = pika_platform_thread_init(
  2837. "pika_event", _thread_event, NULL, PIKA_EVENT_THREAD_STACK_SIZE,
  2838. PIKA_THREAD_PRIO, PIKA_THREAD_TICK);
  2839. pika_debug("event thread init");
  2840. }
  2841. #endif
  2842. if (NULL != eventData) {
  2843. if (PIKA_RES_OK !=
  2844. __eventListener_pushEvent(self, eventId, eventData)) {
  2845. goto __gil_exit;
  2846. }
  2847. }
  2848. if (pickupWhenNoVM) {
  2849. int vmCnt = _VMEvent_getVMCnt();
  2850. if (0 == vmCnt) {
  2851. /* no vm running, pick up event imediately */
  2852. pika_debug("vmCnt: %d, pick up imediately", vmCnt);
  2853. _VMEvent_pickupEvent();
  2854. }
  2855. }
  2856. __gil_exit:
  2857. pika_GIL_EXIT();
  2858. }
  2859. return (PIKA_RES)0;
  2860. #endif
  2861. }
  2862. PIKA_RES pika_eventListener_send(PikaEventListener* self,
  2863. uintptr_t eventId,
  2864. Arg* eventData) {
  2865. return _do_pika_eventListener_send(self, eventId, eventData, 0, pika_false);
  2866. }
  2867. PIKA_RES pika_eventListener_sendSignal(PikaEventListener* self,
  2868. uintptr_t eventId,
  2869. int eventSignal) {
  2870. return _do_pika_eventListener_send(self, eventId, NULL, eventSignal,
  2871. pika_false);
  2872. }
  2873. Arg* pika_eventListener_sendSignalAwaitResult(PikaEventListener* self,
  2874. uintptr_t eventId,
  2875. int eventSignal) {
  2876. /*
  2877. * Await result from event.
  2878. * need implement `pika_platform_thread_delay()` to support thread switch */
  2879. #if !PIKA_EVENT_ENABLE
  2880. pika_platform_printf("PIKA_EVENT_ENABLE is not enable");
  2881. while (1) {
  2882. };
  2883. #else
  2884. extern volatile VMState g_PikaVMState;
  2885. int tail = g_PikaVMState.cq.tail;
  2886. pika_eventListener_sendSignal(self, eventId, eventSignal);
  2887. while (1) {
  2888. Arg* res = g_PikaVMState.cq.res[tail];
  2889. pika_platform_thread_yield();
  2890. if (NULL != res) {
  2891. return res;
  2892. }
  2893. }
  2894. #endif
  2895. }
  2896. Arg* pika_eventListener_syncSendAwaitResult(PikaEventListener* self,
  2897. uintptr_t eventId,
  2898. Arg* eventData) {
  2899. return __eventListener_runEvent(self, eventId, eventData);
  2900. }
  2901. PIKA_RES pika_eventListener_syncSend(PikaEventListener* self,
  2902. uintptr_t eventId,
  2903. Arg* eventData) {
  2904. Arg* res = __eventListener_runEvent(self, eventId, eventData);
  2905. PIKA_RES ret = PIKA_RES_OK;
  2906. if (NULL == res) {
  2907. ret = PIKA_RES_ERR_RUNTIME_ERROR;
  2908. } else {
  2909. arg_deinit(res);
  2910. }
  2911. arg_deinit(eventData);
  2912. return ret;
  2913. }
  2914. PIKA_RES pika_eventListener_syncSendSignal(PikaEventListener* self,
  2915. uintptr_t eventId,
  2916. int eventSignal) {
  2917. Arg* eventData = arg_newInt(eventSignal);
  2918. PIKA_RES res = pika_eventListener_syncSend(self, eventId, eventData);
  2919. return res;
  2920. }
  2921. Arg* pika_eventListener_syncSendSignalAwaitResult(PikaEventListener* self,
  2922. uintptr_t eventId,
  2923. int eventSignal) {
  2924. Arg* eventData = arg_newInt(eventSignal);
  2925. Arg* ret = pika_eventListener_syncSendAwaitResult(self, eventId, eventData);
  2926. arg_deinit(eventData);
  2927. return ret;
  2928. }
  2929. /* print major version info */
  2930. void pika_printVersion(void) {
  2931. pika_platform_printf("pikascript-core==v%d.%d.%d (%s)\r\n",
  2932. PIKA_VERSION_MAJOR, PIKA_VERSION_MINOR,
  2933. PIKA_VERSION_MICRO, PIKA_EDIT_TIME);
  2934. }
  2935. void pika_getVersion(char* buff) {
  2936. pika_sprintf(buff, "%d.%d.%d", PIKA_VERSION_MAJOR, PIKA_VERSION_MINOR,
  2937. PIKA_VERSION_MICRO);
  2938. }
  2939. void* obj_getStruct(PikaObj* self, char* name) {
  2940. pika_assert(self != NULL);
  2941. return args_getStruct(self->list, name);
  2942. }
  2943. char* obj_cacheStr(PikaObj* self, char* str) {
  2944. pika_assert(self != NULL);
  2945. return args_cacheStr(self->list, str);
  2946. }
  2947. void _obj_updateProxyFlag(PikaObj* self) {
  2948. if (!obj_getFlag(self, OBJ_FLAG_PROXY_GETATTRIBUTE)) {
  2949. if (NULL != _obj_getPropArg(self, "__getattribute__")) {
  2950. obj_setFlag(self, OBJ_FLAG_PROXY_GETATTRIBUTE);
  2951. }
  2952. }
  2953. if (!obj_getFlag(self, OBJ_FLAG_PROXY_GETATTR)) {
  2954. if (NULL != _obj_getPropArg(self, "__getattr__")) {
  2955. obj_setFlag(self, OBJ_FLAG_PROXY_GETATTR);
  2956. }
  2957. }
  2958. if (!obj_getFlag(self, OBJ_FLAG_PROXY_SETATTR)) {
  2959. if (NULL != _obj_getPropArg(self, "__setattr__")) {
  2960. obj_setFlag(self, OBJ_FLAG_PROXY_SETATTR);
  2961. }
  2962. }
  2963. };
  2964. void builtins_remove(PikaObj* self, char* argPath) {
  2965. obj_setErrorCode(self, 0);
  2966. int32_t res = obj_removeArg(self, argPath);
  2967. if (1 == res) {
  2968. obj_setSysOut(self, "[error] del: object no found.");
  2969. obj_setErrorCode(self, 1);
  2970. return;
  2971. }
  2972. if (2 == res) {
  2973. obj_setSysOut(self, "[error] del: arg not match.");
  2974. obj_setErrorCode(self, 2);
  2975. return;
  2976. }
  2977. }
  2978. Arg* _type(Arg* arg);
  2979. Arg* builtins_type(PikaObj* self, Arg* arg) {
  2980. #if PIKA_NANO_ENABLE
  2981. pika_platform_printf("PIKA_NANO_ENABLE is not enable");
  2982. return NULL;
  2983. #else
  2984. if (NULL == arg) {
  2985. obj_setSysOut(self, "[error] type: arg no found.");
  2986. obj_setErrorCode(self, 1);
  2987. return NULL;
  2988. }
  2989. return _type(arg);
  2990. #endif
  2991. }
  2992. pika_float builtins_float(PikaObj* self, Arg* arg) {
  2993. ArgType type = arg_getType(arg);
  2994. if (ARG_TYPE_INT == type) {
  2995. return (pika_float)arg_getInt(arg);
  2996. }
  2997. if (ARG_TYPE_FLOAT == type) {
  2998. return (pika_float)arg_getFloat(arg);
  2999. }
  3000. if (ARG_TYPE_STRING == type) {
  3001. return strtod(arg_getStr(arg), NULL);
  3002. }
  3003. if (ARG_TYPE_BOOL == type) {
  3004. return (pika_float)arg_getBool(arg);
  3005. }
  3006. obj_setSysOut(self, "[error] convert to pika_float type failed.");
  3007. obj_setErrorCode(self, 1);
  3008. return _PIKA_FLOAT_ERR;
  3009. }
  3010. PIKA_RES _transeInt(Arg* arg, int base, int64_t* res) {
  3011. ArgType type = arg_getType(arg);
  3012. if (ARG_TYPE_INT == type) {
  3013. *res = arg_getInt(arg);
  3014. return PIKA_RES_OK;
  3015. }
  3016. if (ARG_TYPE_BOOL == type) {
  3017. *res = (int64_t)arg_getBool(arg);
  3018. return PIKA_RES_OK;
  3019. }
  3020. if (ARG_TYPE_FLOAT == type) {
  3021. *res = (int64_t)arg_getFloat(arg);
  3022. return PIKA_RES_OK;
  3023. }
  3024. if (ARG_TYPE_STRING == type) {
  3025. char* end = NULL;
  3026. char* str = arg_getStr(arg);
  3027. *res = strtoll(str, &end, base);
  3028. if (NULL != end && strGetSize(end) != 0) {
  3029. return PIKA_RES_ERR_INVALID_PARAM;
  3030. }
  3031. return PIKA_RES_OK;
  3032. }
  3033. if (ARG_TYPE_BYTES == type) {
  3034. size_t size = arg_getBytesSize(arg);
  3035. if (size != 1) {
  3036. return PIKA_RES_ERR_INVALID_PARAM;
  3037. }
  3038. uint8_t val = *arg_getBytes(arg);
  3039. *res = val;
  3040. return PIKA_RES_OK;
  3041. }
  3042. return PIKA_RES_ERR_INVALID_PARAM;
  3043. }
  3044. PIKA_RES _transeBool(Arg* arg, pika_bool* res) {
  3045. int64_t iRes = 0;
  3046. if (arg_getType(arg) == ARG_TYPE_BOOL) {
  3047. *res = arg_getBool(arg);
  3048. return PIKA_RES_OK;
  3049. }
  3050. if (arg_getType(arg) == ARG_TYPE_NONE) {
  3051. *res = pika_false;
  3052. return PIKA_RES_OK;
  3053. }
  3054. if (arg_getType(arg) == ARG_TYPE_STRING ||
  3055. arg_getType(arg) == ARG_TYPE_BYTES) {
  3056. if (arg_getStr(arg)[0] == '\0') {
  3057. *res = pika_false;
  3058. return PIKA_RES_OK;
  3059. }
  3060. *res = pika_true;
  3061. return PIKA_RES_OK;
  3062. }
  3063. if (arg_isObject(arg)) {
  3064. int64_t len = obj_getSize(arg_getObj(arg));
  3065. if (len < 0) {
  3066. *res = pika_true;
  3067. return PIKA_RES_OK;
  3068. }
  3069. if (len == 0) {
  3070. *res = pika_false;
  3071. return PIKA_RES_OK;
  3072. }
  3073. *res = pika_true;
  3074. return PIKA_RES_OK;
  3075. }
  3076. if (_transeInt(arg, 10, &iRes) == PIKA_RES_OK) {
  3077. *res = iRes ? pika_true : pika_false;
  3078. return PIKA_RES_OK;
  3079. }
  3080. return PIKA_RES_ERR_INVALID_PARAM;
  3081. }
  3082. Arg* builtins_int(PikaObj* self, Arg* arg, PikaTuple* base) {
  3083. int64_t res = 0;
  3084. int64_t iBase = 10;
  3085. if (pikaTuple_getSize(base) > 0) {
  3086. if (arg_getType(arg) != ARG_TYPE_STRING &&
  3087. arg_getType(arg) != ARG_TYPE_BYTES) {
  3088. obj_setSysOut(self,
  3089. "TypeError: int() can't convert non-string with "
  3090. "explicit base");
  3091. obj_setErrorCode(self, 1);
  3092. return NULL;
  3093. }
  3094. iBase = (int64_t)pikaTuple_getInt(base, 0);
  3095. }
  3096. if (_transeInt(arg, iBase, &res) == PIKA_RES_OK) {
  3097. return arg_newInt(res);
  3098. }
  3099. obj_setSysOut(self, "ValueError: invalid literal for int(): '%s'",
  3100. arg_getStr(arg));
  3101. obj_setErrorCode(self, 1);
  3102. return NULL;
  3103. }
  3104. pika_bool builtins_bool(PikaObj* self, Arg* arg) {
  3105. pika_bool res = 0;
  3106. if (_transeBool(arg, &res) == PIKA_RES_OK) {
  3107. return res;
  3108. }
  3109. obj_setSysOut(self, "ValueError: invalid literal for bool()");
  3110. obj_setErrorCode(self, 1);
  3111. return _PIKA_BOOL_ERR;
  3112. }
  3113. char* builtins_str(PikaObj* self, Arg* arg) {
  3114. pika_assert(NULL != arg);
  3115. pika_assert(NULL != self);
  3116. // if (arg_getType(arg) == ARG_TYPE_BYTES) {
  3117. // return obj_cacheStr(self, (char*)arg_getBytes(arg));
  3118. // }
  3119. Arg* arg_str = arg_toStrArg(arg);
  3120. if (NULL == arg_str) {
  3121. obj_setSysOut(self, "Error: convert to str type failed.");
  3122. obj_setErrorCode(self, 1);
  3123. return NULL;
  3124. }
  3125. char* str = obj_cacheStr(self, arg_getStr(arg_str));
  3126. arg_deinit(arg_str);
  3127. return str;
  3128. }
  3129. PikaObj* New_builtins_RangeObj(Args* args);
  3130. Arg* builtins_iter(PikaObj* self, Arg* arg) {
  3131. pika_assert(NULL != arg);
  3132. /* object */
  3133. pika_bool bIsTemp = pika_false;
  3134. PikaObj* oArg = _arg_to_obj(arg, &bIsTemp);
  3135. pika_assert(NULL != oArg);
  3136. NewFun _clsptr = (NewFun)oArg->constructor;
  3137. if (_clsptr == New_builtins_RangeObj) {
  3138. /* found RangeObj, return directly */
  3139. return arg_copy(arg);
  3140. }
  3141. /* clang-format off */
  3142. PIKA_PYTHON(
  3143. @res_iter = __iter__()
  3144. )
  3145. /* clang-format on */
  3146. const uint8_t bytes[] = {
  3147. 0x08, 0x00, 0x00, 0x00, /* instruct array size */
  3148. 0x00, 0x82, 0x01, 0x00, 0x00, 0x04, 0x0a, 0x00, /* instruct array */
  3149. 0x14, 0x00, 0x00, 0x00, /* const pool size */
  3150. 0x00, 0x5f, 0x5f, 0x69, 0x74, 0x65, 0x72, 0x5f, 0x5f, 0x00, 0x40,
  3151. 0x72, 0x65, 0x73, 0x5f, 0x69, 0x74, 0x65, 0x72, 0x00, /* const pool */
  3152. };
  3153. Arg* res = pikaVM_runByteCodeReturn(oArg, (uint8_t*)bytes, "@res_iter");
  3154. if (bIsTemp) {
  3155. obj_refcntDec(oArg);
  3156. }
  3157. return res;
  3158. }
  3159. Arg* builtins_range(PikaObj* self, PikaTuple* ax) {
  3160. /* set template arg to create rangeObj */
  3161. Arg* aRangeObj = arg_newDirectObj(New_builtins_RangeObj);
  3162. PikaObj* oRangeObj = arg_getPtr(aRangeObj);
  3163. RangeData tRangeData = {0};
  3164. if (pikaTuple_getSize(ax) == 1) {
  3165. int start = 0;
  3166. int end = arg_getInt(pikaTuple_getArg(ax, 0));
  3167. tRangeData.start = start;
  3168. tRangeData.end = end;
  3169. tRangeData.step = 1;
  3170. } else if (pikaTuple_getSize(ax) == 2) {
  3171. int start = arg_getInt(pikaTuple_getArg(ax, 0));
  3172. int end = arg_getInt(pikaTuple_getArg(ax, 1));
  3173. tRangeData.start = start;
  3174. tRangeData.end = end;
  3175. tRangeData.step = 1;
  3176. } else if (pikaTuple_getSize(ax) == 3) {
  3177. int start = arg_getInt(pikaTuple_getArg(ax, 0));
  3178. int end = arg_getInt(pikaTuple_getArg(ax, 1));
  3179. int step = arg_getInt(pikaTuple_getArg(ax, 2));
  3180. tRangeData.start = start;
  3181. tRangeData.end = end;
  3182. tRangeData.step = step;
  3183. }
  3184. tRangeData.i = tRangeData.start;
  3185. obj_setStruct(oRangeObj, "_", tRangeData);
  3186. return aRangeObj;
  3187. }
  3188. Arg* builtins___getitem__(PikaObj* self, Arg* obj, Arg* key) {
  3189. return _vm_get(NULL, self, key, obj);
  3190. }
  3191. Arg* builtins___setitem__(PikaObj* self, Arg* obj, Arg* key, Arg* val) {
  3192. ArgType obj_type = arg_getType(obj);
  3193. if (ARG_TYPE_STRING == obj_type) {
  3194. int index = arg_getInt(key);
  3195. char* str_val = arg_getStr(val);
  3196. char* str_pyload = arg_getStr(obj);
  3197. str_pyload[index] = str_val[0];
  3198. return arg_newStr(str_pyload);
  3199. }
  3200. if (ARG_TYPE_BYTES == obj_type) {
  3201. int index = arg_getInt(key);
  3202. uint8_t byte_val = 0;
  3203. if (ARG_TYPE_BYTES == arg_getType(val)) {
  3204. uint8_t* bytes_val = arg_getBytes(val);
  3205. byte_val = bytes_val[0];
  3206. }
  3207. if (ARG_TYPE_INT == arg_getType(val)) {
  3208. byte_val = arg_getInt(val);
  3209. }
  3210. uint8_t* bytes_pyload = arg_getBytes(obj);
  3211. size_t bytes_len = arg_getBytesSize(obj);
  3212. bytes_pyload[index] = byte_val;
  3213. return arg_newBytes(bytes_pyload, bytes_len);
  3214. }
  3215. if (argType_isObject(obj_type)) {
  3216. PikaObj* arg_obj = arg_getPtr(obj);
  3217. obj_setArg(arg_obj, "__key", key);
  3218. obj_setArg(arg_obj, "__val", val);
  3219. /* clang-format off */
  3220. PIKA_PYTHON(
  3221. __setitem__(__key, __val)
  3222. )
  3223. /* clang-format on */
  3224. const uint8_t bytes[] = {
  3225. 0x0c, 0x00, 0x00, 0x00, /* instruct array size */
  3226. 0x10, 0x81, 0x01, 0x00, 0x10, 0x01, 0x07, 0x00, 0x00, 0x02, 0x0d,
  3227. 0x00,
  3228. /* instruct array */
  3229. 0x19, 0x00, 0x00, 0x00, /* const pool size */
  3230. 0x00, 0x5f, 0x5f, 0x6b, 0x65, 0x79, 0x00, 0x5f, 0x5f, 0x76, 0x61,
  3231. 0x6c, 0x00, 0x5f, 0x5f, 0x73, 0x65, 0x74, 0x69, 0x74, 0x65, 0x6d,
  3232. 0x5f, 0x5f, 0x00,
  3233. /* const pool */
  3234. };
  3235. pikaVM_runByteCode(arg_obj, (uint8_t*)bytes);
  3236. return arg_newRef(arg_obj);
  3237. }
  3238. return NULL;
  3239. }
  3240. pika_bool arg_isList(Arg* arg) {
  3241. if (!arg_isObject(arg)) {
  3242. return pika_false;
  3243. }
  3244. return arg_getObj(arg)->constructor == New_PikaStdData_List;
  3245. }
  3246. pika_bool arg_isDict(Arg* arg) {
  3247. if (!arg_isObject(arg)) {
  3248. return pika_false;
  3249. }
  3250. return arg_getObj(arg)->constructor == New_PikaStdData_Dict;
  3251. }
  3252. pika_bool arg_isTuple(Arg* arg) {
  3253. if (!arg_isObject(arg)) {
  3254. return pika_false;
  3255. }
  3256. return arg_getObj(arg)->constructor == New_PikaStdData_Tuple;
  3257. }
  3258. int64_t obj_getSize(PikaObj* arg_obj) {
  3259. Arg* aRes = obj_runMethod0(arg_obj, "__len__");
  3260. if (NULL == aRes) {
  3261. return -1;
  3262. }
  3263. int64_t res = arg_getInt(aRes);
  3264. arg_deinit(aRes);
  3265. return res;
  3266. }
  3267. int builtins_len(PikaObj* self, Arg* arg) {
  3268. if (ARG_TYPE_STRING == arg_getType(arg)) {
  3269. return strGetSize(arg_getStr(arg));
  3270. }
  3271. if (ARG_TYPE_BYTES == arg_getType(arg)) {
  3272. return arg_getBytesSize(arg);
  3273. }
  3274. if (arg_isObject(arg)) {
  3275. PikaObj* arg_obj = arg_getPtr(arg);
  3276. return obj_getSize(arg_obj);
  3277. }
  3278. obj_setErrorCode(self, 1);
  3279. __platform_printf("[Error] len: arg type not support\r\n");
  3280. return -1;
  3281. }
  3282. Arg* builtins_list(PikaObj* self, PikaTuple* val) {
  3283. #if PIKA_BUILTIN_STRUCT_ENABLE
  3284. if (1 == pikaTuple_getSize(val)) {
  3285. Arg* aInput = pikaTuple_getArg(val, 0);
  3286. if (!arg_isIterable(aInput)) {
  3287. obj_setErrorCode(self, 1);
  3288. obj_setSysOut(self, "Error: input arg must be iterable");
  3289. return NULL;
  3290. }
  3291. obj_setArg(self, "__list", aInput);
  3292. /* clang-format off */
  3293. PIKA_PYTHON(
  3294. @res_list = []
  3295. for __item in __list:
  3296. @res_list.append(__item)
  3297. del __item
  3298. del __list
  3299. )
  3300. /* clang-format on */
  3301. const uint8_t bytes[] = {
  3302. 0x40, 0x00, 0x00, 0x00, /* instruct array size */
  3303. 0x00, 0x95, 0x00, 0x00, 0x00, 0x22, 0x00, 0x00, 0x00, 0x04, 0x01,
  3304. 0x00, 0x10, 0x81, 0x0b, 0x00, 0x00, 0x02, 0x12, 0x00, 0x00, 0x04,
  3305. 0x17, 0x00, 0x00, 0x82, 0x1b, 0x00, 0x00, 0x04, 0x28, 0x00, 0x00,
  3306. 0x0d, 0x28, 0x00, 0x00, 0x07, 0x2f, 0x00, 0x11, 0x81, 0x28, 0x00,
  3307. 0x01, 0x02, 0x31, 0x00, 0x00, 0x86, 0x42, 0x00, 0x00, 0x8c, 0x17,
  3308. 0x00, 0x00, 0x8c, 0x28, 0x00, 0x00, 0x8c, 0x0b, 0x00, /* instruct
  3309. array */
  3310. 0x45, 0x00, 0x00, 0x00, /* const pool size */
  3311. 0x00, 0x40, 0x72, 0x65, 0x73, 0x5f, 0x6c, 0x69, 0x73, 0x74, 0x00,
  3312. 0x5f, 0x5f, 0x6c, 0x69, 0x73, 0x74, 0x00, 0x69, 0x74, 0x65, 0x72,
  3313. 0x00, 0x24, 0x6c, 0x30, 0x00, 0x24, 0x6c, 0x30, 0x2e, 0x5f, 0x5f,
  3314. 0x6e, 0x65, 0x78, 0x74, 0x5f, 0x5f, 0x00, 0x5f, 0x5f, 0x69, 0x74,
  3315. 0x65, 0x6d, 0x00, 0x32, 0x00, 0x40, 0x72, 0x65, 0x73, 0x5f, 0x6c,
  3316. 0x69, 0x73, 0x74, 0x2e, 0x61, 0x70, 0x70, 0x65, 0x6e, 0x64, 0x00,
  3317. 0x2d, 0x31, 0x00, /* const pool */
  3318. };
  3319. return pikaVM_runByteCodeReturn(self, (uint8_t*)bytes, "@res_list");
  3320. }
  3321. return arg_newObj(New_pikaListFrom(NULL));
  3322. #else
  3323. obj_setErrorCode(self, 1);
  3324. obj_setSysOut(self, "[Error] built-in list is not enabled.");
  3325. return arg_newNull();
  3326. #endif
  3327. }
  3328. Arg* builtins_dict(PikaObj* self, PikaTuple* val) {
  3329. #if PIKA_BUILTIN_STRUCT_ENABLE
  3330. return arg_newObj(New_PikaDict());
  3331. #else
  3332. obj_setErrorCode(self, 1);
  3333. obj_setSysOut(self, "[Error] built-in dist is not enabled.");
  3334. return arg_newNull();
  3335. #endif
  3336. }
  3337. Arg* builtins_tuple(PikaObj* self, PikaTuple* val) {
  3338. #if PIKA_BUILTIN_STRUCT_ENABLE
  3339. Arg* tuple = builtins_list(self, val);
  3340. if (NULL == tuple) {
  3341. return NULL;
  3342. }
  3343. PikaObj* oTuple = arg_getObj(tuple);
  3344. oTuple->constructor = New_PikaStdData_Tuple;
  3345. return tuple;
  3346. #else
  3347. obj_setErrorCode(self, 1);
  3348. obj_setSysOut(self, "[Error] built-in tuple is not enabled.");
  3349. return arg_newNull();
  3350. #endif
  3351. }
  3352. char* builtins_hex(PikaObj* self, int val) {
  3353. char buff[PIKA_SPRINTF_BUFF_SIZE] = {0};
  3354. if (val >= 0) {
  3355. __platform_sprintf(buff, "0x%02x", val);
  3356. } else {
  3357. __platform_sprintf(buff, "-0x%02x", -val);
  3358. }
  3359. /* load the string from stack to heap */
  3360. obj_setStr(self, "__buf", buff);
  3361. return obj_getStr(self, "__buf");
  3362. }
  3363. int builtins_ord(PikaObj* self, char* val) {
  3364. return (int)val[0];
  3365. }
  3366. char* builtins_chr(PikaObj* self, int val) {
  3367. char buff[PIKA_SPRINTF_BUFF_SIZE] = {0};
  3368. char to_str[] = "0";
  3369. to_str[0] = val;
  3370. __platform_sprintf(buff, "%s", to_str);
  3371. /* load the string from stack to heap */
  3372. obj_setStr(self, "__buf", buff);
  3373. return obj_getStr(self, "__buf");
  3374. }
  3375. Arg* builtins_bytes(PikaObj* self, Arg* val) {
  3376. ArgType type = arg_getType(val);
  3377. if (ARG_TYPE_INT == type) {
  3378. int size = arg_getInt(val);
  3379. /* src is NULL so the bytes are all '\0' */
  3380. Arg* bytes = arg_newBytes(NULL, size);
  3381. return bytes;
  3382. }
  3383. if (ARG_TYPE_BYTES == type) {
  3384. return arg_copy(val);
  3385. }
  3386. if (ARG_TYPE_STRING == type) {
  3387. int size = strGetSize(arg_getStr(val));
  3388. Arg* bytes = arg_newBytes((uint8_t*)arg_getStr(val), size);
  3389. return bytes;
  3390. }
  3391. #if !PIKA_NANO_ENABLE
  3392. if (argType_isObject(type)) {
  3393. PikaObj* obj = arg_getPtr(val);
  3394. PikaObj* New_PikaStdData_List(Args * args);
  3395. PikaObj* New_PikaStdData_Tuple(Args * args);
  3396. if (obj->constructor == New_PikaStdData_List ||
  3397. obj->constructor == New_PikaStdData_Tuple) {
  3398. Arg* bytes = arg_newBytes(NULL, pikaList_getSize(obj));
  3399. uint8_t* bytes_raw = arg_getBytes(bytes);
  3400. for (size_t i = 0; i < pikaList_getSize(obj); i++) {
  3401. bytes_raw[i] = (uint8_t)pikaList_getInt(obj, i);
  3402. }
  3403. return bytes;
  3404. }
  3405. }
  3406. #endif
  3407. obj_setErrorCode(self, 1);
  3408. obj_setSysOut(self, "Error: input arg type not supported.");
  3409. return arg_newNone();
  3410. }
  3411. void builtins_print(PikaObj* self, PikaTuple* val, PikaDict* ops) {
  3412. int arg_size = pikaTuple_getSize(val);
  3413. pika_assert(arg_size >= 0);
  3414. char* end = pikaDict_getStr(ops, "end");
  3415. if (NULL == end) {
  3416. /* default */
  3417. end = "\r\n";
  3418. }
  3419. if (arg_size == 1) {
  3420. arg_print(pikaTuple_getArg(val, 0), pika_false, end);
  3421. return;
  3422. }
  3423. Arg* print_out_arg = NULL;
  3424. pika_bool is_get_print = pika_false;
  3425. for (int i = 0; i < arg_size; i++) {
  3426. Arg* arg = pikaTuple_getArg(val, i);
  3427. Arg* item_arg_str = arg_toStrArg(arg);
  3428. if (NULL != item_arg_str) {
  3429. is_get_print = PIKA_TRUE;
  3430. if (NULL == print_out_arg) {
  3431. print_out_arg = arg_newStr("");
  3432. }
  3433. print_out_arg =
  3434. arg_strAppend(print_out_arg, arg_getStr(item_arg_str));
  3435. if (i < arg_size - 1) {
  3436. print_out_arg = arg_strAppend(print_out_arg, " ");
  3437. }
  3438. arg_deinit(item_arg_str);
  3439. }
  3440. }
  3441. if (PIKA_TRUE == is_get_print) {
  3442. __platform_printf("%s%s", arg_getStr(print_out_arg), end);
  3443. }
  3444. if (NULL != print_out_arg) {
  3445. arg_deinit(print_out_arg);
  3446. }
  3447. }
  3448. char* builtins_cformat(PikaObj* self, char* fmt, PikaTuple* var) {
  3449. #if PIKA_SYNTAX_FORMAT_ENABLE
  3450. Args buffs = {0};
  3451. pikaMemMaxReset();
  3452. char* res = strsFormatList(&buffs, fmt, var);
  3453. obj_setStr(self, "_buf", res);
  3454. res = obj_getStr(self, "_buf");
  3455. strsDeinit(&buffs);
  3456. return res;
  3457. #else
  3458. obj_setErrorCode(self, 1);
  3459. obj_setSysOut(self, "[Error] PIKA_SYNTAX_FORMAT_ENABLE is not enabled.");
  3460. return NULL;
  3461. #endif
  3462. }
  3463. int builtins_id(PikaObj* self, Arg* obj) {
  3464. uintptr_t ptr = 0;
  3465. if (arg_isObject(obj)) {
  3466. ptr = (uintptr_t)arg_getPtr(obj);
  3467. } else {
  3468. ptr = (uintptr_t)obj;
  3469. }
  3470. return ptr & (0x7FFFFFFF);
  3471. }
  3472. int PikaStdData_FILEIO_init(PikaObj* self, char* path, char* mode);
  3473. PikaObj* New_PikaStdData_FILEIO(Args* args);
  3474. PikaObj* builtins_open(PikaObj* self, char* path, char* mode) {
  3475. #if PIKA_FILEIO_ENABLE
  3476. PikaObj* file = newNormalObj(New_PikaStdData_FILEIO);
  3477. if (0 != PikaStdData_FILEIO_init(file, path, mode)) {
  3478. obj_setErrorCode(self, 1);
  3479. obj_setSysOut(self, "[Error] open: can not open file.");
  3480. obj_deinit(file);
  3481. return NULL;
  3482. }
  3483. return file;
  3484. #else
  3485. obj_setErrorCode(self, 1);
  3486. obj_setSysOut(self, "[Error] PIKA_FILEIO_ENABLE is not enabled.");
  3487. return NULL;
  3488. #endif
  3489. }
  3490. /* __dir_each */
  3491. int32_t __dir_each(Arg* argEach, void* context) {
  3492. if (argType_isCallable(arg_getType(argEach))) {
  3493. char name_buff[PIKA_LINE_BUFF_SIZE] = {0};
  3494. char* method_name =
  3495. methodArg_getName(argEach, name_buff, sizeof(name_buff));
  3496. Arg* arg_str = arg_newStr(method_name);
  3497. PikaList* list = args_getPtr(context, "_list");
  3498. pikaList_append(list, arg_str);
  3499. }
  3500. return 0;
  3501. }
  3502. PikaObj* builtins_dir(PikaObj* self, Arg* arg) {
  3503. if (!arg_isObject(arg)) {
  3504. obj_setErrorCode(self, 1);
  3505. obj_setSysOut(self, "[Error] dir: not support type.");
  3506. return NULL;
  3507. }
  3508. PikaObj* obj = arg_getPtr(arg);
  3509. PikaObj* New_PikaStdData_List(Args * args);
  3510. PikaObj* list = New_PikaList();
  3511. Args* context = New_args(NULL);
  3512. args_setPtr(context, "_list", list);
  3513. args_foreach(obj->list, __dir_each, context);
  3514. args_deinit(context);
  3515. return list;
  3516. }
  3517. void builtins_exec(PikaObj* self, char* code) {
  3518. #if PIKA_EXEC_ENABLE
  3519. obj_run(self, code);
  3520. #else
  3521. obj_setErrorCode(self, 1);
  3522. obj_setSysOut(self, "[Error] PIKA_EXEC_ENABLE is not enabled.");
  3523. #endif
  3524. }
  3525. Arg* builtins_getattr(PikaObj* self, PikaObj* obj, char* name) {
  3526. if (NULL == obj) {
  3527. obj_setErrorCode(self, 1);
  3528. obj_setSysOut(self, "[Error] getattr: can not get attr of NULL.");
  3529. return NULL;
  3530. }
  3531. Arg* aRes = obj_getArg(obj, name);
  3532. if (NULL == aRes) {
  3533. aRes = obj_getMethodArgWithFullPath(obj, name);
  3534. }
  3535. if (NULL != aRes) {
  3536. aRes = methodArg_setHostObj(aRes, obj);
  3537. if (arg_getType(aRes) != ARG_TYPE_METHOD_NATIVE_ACTIVE) {
  3538. aRes = arg_copy(aRes);
  3539. }
  3540. }
  3541. return aRes;
  3542. }
  3543. void builtins_setattr(PikaObj* self, PikaObj* obj, char* name, Arg* val) {
  3544. if (NULL == obj) {
  3545. obj_setErrorCode(self, 1);
  3546. obj_setSysOut(self, "[Error] setattr: obj is null.");
  3547. goto __exit;
  3548. }
  3549. obj_setArg(obj, name, val);
  3550. __exit:
  3551. return;
  3552. }
  3553. void builtins_exit(PikaObj* self) {
  3554. pika_vm_exit();
  3555. }
  3556. int builtins_hasattr(PikaObj* self, PikaObj* obj, char* name) {
  3557. if (NULL == obj) {
  3558. obj_setErrorCode(self, 1);
  3559. obj_setSysOut(self, "[Error] hasattr: obj is null.");
  3560. return 0;
  3561. }
  3562. if (obj_isArgExist(obj, name)) {
  3563. return 1;
  3564. }
  3565. Arg* method = obj_getMethodArgWithFullPath(obj, name);
  3566. if (NULL != method) {
  3567. arg_deinit(method);
  3568. return 1;
  3569. }
  3570. return 0;
  3571. }
  3572. Arg* builtins_eval(PikaObj* self, char* code) {
  3573. Args buffs = {0};
  3574. char* cmd = strsAppend(&buffs, "@res = ", code);
  3575. obj_run(self, cmd);
  3576. Arg* res = arg_copy(obj_getArg(self, "@res"));
  3577. strsDeinit(&buffs);
  3578. obj_removeArg(self, "@res");
  3579. return res;
  3580. }
  3581. static enum shellCTRL __obj_shellLineHandler_input(PikaObj* self,
  3582. char* input_line,
  3583. struct ShellConfig* cfg) {
  3584. cfg->context = arg_newStr(input_line);
  3585. return SHELL_CTRL_EXIT;
  3586. }
  3587. Arg* _type(Arg* arg) {
  3588. Arg* result;
  3589. PikaObj* oBuiltins = NULL;
  3590. ArgType type = arg_getType(arg);
  3591. oBuiltins = obj_getBuiltins();
  3592. if (ARG_TYPE_INT == type) {
  3593. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "int"));
  3594. goto __exit;
  3595. }
  3596. if (ARG_TYPE_FLOAT == type) {
  3597. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "float"));
  3598. goto __exit;
  3599. }
  3600. if (ARG_TYPE_STRING == type) {
  3601. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "str"));
  3602. goto __exit;
  3603. }
  3604. if (ARG_TYPE_BOOL == type) {
  3605. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "bool"));
  3606. goto __exit;
  3607. }
  3608. if (ARG_TYPE_BYTES == type) {
  3609. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "bytes"));
  3610. goto __exit;
  3611. }
  3612. if (argType_isObject(type)) {
  3613. PikaObj* obj = arg_getPtr(arg);
  3614. NewFun clsptr = obj_getClass(obj);
  3615. PikaObj* New_PikaStdData_List(Args * args);
  3616. if (clsptr == New_PikaStdData_List) {
  3617. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "list"));
  3618. goto __exit;
  3619. }
  3620. PikaObj* New_PikaStdData_Dict(Args * args);
  3621. if (clsptr == New_PikaStdData_Dict) {
  3622. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "dict"));
  3623. goto __exit;
  3624. }
  3625. PikaObj* New_PikaStdData_Tuple(Args * args);
  3626. if (clsptr == New_PikaStdData_Tuple) {
  3627. result = arg_copy(obj_getMethodArgWithFullPath(oBuiltins, "tuple"));
  3628. goto __exit;
  3629. }
  3630. #if PIKA_TYPE_FULL_FEATURE_ENABLE
  3631. Arg* aMethod = obj_getArg(obj, "__class__");
  3632. if (NULL != aMethod) {
  3633. result = arg_copy(aMethod);
  3634. goto __exit;
  3635. }
  3636. #endif
  3637. result = arg_newStr("<class 'object'>");
  3638. goto __exit;
  3639. }
  3640. if (ARG_TYPE_OBJECT_META == type) {
  3641. result = arg_newStr("<class 'meta object'>");
  3642. goto __exit;
  3643. }
  3644. if (ARG_TYPE_METHOD_OBJECT == type) {
  3645. result = arg_newStr("<class 'method'>");
  3646. goto __exit;
  3647. }
  3648. if (ARG_TYPE_METHOD_STATIC == type) {
  3649. result = arg_newStr("<class 'function'>");
  3650. goto __exit;
  3651. }
  3652. if (ARG_TYPE_NONE == type) {
  3653. result = arg_newStr("<class 'NoneType'>");
  3654. goto __exit;
  3655. }
  3656. result = arg_newStr("<class 'buitin_function_or_method'>");
  3657. __exit:
  3658. if (NULL != oBuiltins) {
  3659. obj_deinit(oBuiltins);
  3660. }
  3661. return result;
  3662. }
  3663. pika_bool _isinstance(Arg* aObj, Arg* classinfo) {
  3664. pika_bool res = pika_false;
  3665. Arg* aObjType = NULL;
  3666. aObjType = _type(aObj);
  3667. NativeProperty* objProp = NULL;
  3668. while (1) {
  3669. if (arg_getPtr(aObjType) == arg_getPtr(classinfo)) {
  3670. res = pika_true;
  3671. goto __exit;
  3672. }
  3673. Arg* aObjSuper = methodArg_super(aObjType, &objProp);
  3674. if ((NULL == aObjSuper) && (NULL == objProp)) {
  3675. res = pika_false;
  3676. goto __exit;
  3677. }
  3678. aObjType = aObjSuper;
  3679. if (NULL != objProp) {
  3680. if (!(arg_getType(classinfo) ==
  3681. ARG_TYPE_METHOD_NATIVE_CONSTRUCTOR)) {
  3682. res = pika_false;
  3683. goto __exit;
  3684. }
  3685. NativeProperty* classProp = methodArg_toProp(classinfo);
  3686. while (1) {
  3687. if (objProp == classProp) {
  3688. res = pika_true;
  3689. goto __exit;
  3690. }
  3691. if (objProp->super == NULL) {
  3692. res = pika_false;
  3693. goto __exit;
  3694. }
  3695. objProp = (NativeProperty*)objProp->super;
  3696. }
  3697. }
  3698. if (NULL == aObjType) {
  3699. res = pika_false;
  3700. goto __exit;
  3701. }
  3702. }
  3703. __exit:
  3704. if (NULL != aObjType) {
  3705. arg_deinit(aObjType);
  3706. }
  3707. return res;
  3708. }
  3709. char* builtins_input(PikaObj* self, PikaTuple* info) {
  3710. struct ShellConfig cfg = {
  3711. .prefix = "",
  3712. .context = NULL,
  3713. .handler = __obj_shellLineHandler_input,
  3714. .fn_getchar = __platform_getchar,
  3715. };
  3716. if (pikaTuple_getSize(info) > 0) {
  3717. __platform_printf("%s", pikaTuple_getStr(info, 0));
  3718. }
  3719. _temp__do_pikaScriptShell(self, &cfg);
  3720. char* res = obj_cacheStr(self, arg_getStr(cfg.context));
  3721. arg_deinit(cfg.context);
  3722. return res;
  3723. }
  3724. extern volatile PikaObj* __pikaMain;
  3725. void builtins_help(PikaObj* self, char* name) {
  3726. if (strEqu(name, "modules")) {
  3727. obj_printModules((PikaObj*)__pikaMain);
  3728. return;
  3729. }
  3730. pika_platform_printf(
  3731. "Warning: help() only support help('modules') now.\r\n");
  3732. }
  3733. void builtins_reboot(PikaObj* self) {
  3734. pika_platform_reboot();
  3735. }
  3736. void builtins_clear(PikaObj* self) {
  3737. pika_platform_clear();
  3738. }
  3739. void builtins_gcdump(PikaObj* self) {
  3740. pikaGC_markDump();
  3741. }
  3742. Arg* builtins_abs(PikaObj* self, Arg* val) {
  3743. ArgType type = arg_getType(val);
  3744. if (type == ARG_TYPE_INT) {
  3745. int64_t v = arg_getInt(val);
  3746. if (v < 0) {
  3747. v = -v;
  3748. }
  3749. return arg_newInt(v);
  3750. }
  3751. if (type == ARG_TYPE_FLOAT) {
  3752. pika_float v = arg_getFloat(val);
  3753. if (v < 0) {
  3754. v = -v;
  3755. }
  3756. return arg_newFloat(v);
  3757. }
  3758. obj_setSysOut(self, "TypeError: bad operand type for abs()");
  3759. obj_setErrorCode(self, PIKA_RES_ERR_INVALID_PARAM);
  3760. return NULL;
  3761. }
  3762. PikaObj* New_PikaStdData_Tuple(Args* args);
  3763. /* clang-format off */
  3764. PIKA_PYTHON(
  3765. @res_max = @list[0]
  3766. for @item in @list:
  3767. if @item > @res_max:
  3768. @res_max = @item
  3769. )
  3770. /* clang-format on */
  3771. const uint8_t bc_max[] = {
  3772. 0x4c, 0x00, 0x00, 0x00, /* instruct array size */
  3773. 0x10, 0x81, 0x01, 0x00, 0x10, 0x05, 0x07, 0x00, 0x00, 0x1d, 0x00, 0x00,
  3774. 0x00, 0x04, 0x09, 0x00, 0x10, 0x81, 0x01, 0x00, 0x00, 0x02, 0x12, 0x00,
  3775. 0x00, 0x04, 0x17, 0x00, 0x00, 0x82, 0x1b, 0x00, 0x00, 0x04, 0x28, 0x00,
  3776. 0x00, 0x0d, 0x28, 0x00, 0x00, 0x07, 0x2e, 0x00, 0x11, 0x81, 0x28, 0x00,
  3777. 0x11, 0x01, 0x09, 0x00, 0x01, 0x08, 0x30, 0x00, 0x01, 0x07, 0x32, 0x00,
  3778. 0x02, 0x81, 0x28, 0x00, 0x02, 0x04, 0x09, 0x00, 0x00, 0x86, 0x34, 0x00,
  3779. 0x00, 0x8c, 0x17, 0x00, /* instruct array */
  3780. 0x37, 0x00, 0x00, 0x00, /* const pool size */
  3781. 0x00, 0x40, 0x6c, 0x69, 0x73, 0x74, 0x00, 0x30, 0x00, 0x40, 0x72, 0x65,
  3782. 0x73, 0x5f, 0x6d, 0x61, 0x78, 0x00, 0x69, 0x74, 0x65, 0x72, 0x00, 0x24,
  3783. 0x6c, 0x30, 0x00, 0x24, 0x6c, 0x30, 0x2e, 0x5f, 0x5f, 0x6e, 0x65, 0x78,
  3784. 0x74, 0x5f, 0x5f, 0x00, 0x40, 0x69, 0x74, 0x65, 0x6d, 0x00, 0x32, 0x00,
  3785. 0x3e, 0x00, 0x31, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  3786. };
  3787. /* clang-format off */
  3788. PIKA_PYTHON(
  3789. @res_max = @list[0]
  3790. for @item in @list:
  3791. if @item < @res_max:
  3792. @res_max = @item
  3793. )
  3794. /* clang-format on */
  3795. const uint8_t bc_min[] = {
  3796. 0x4c, 0x00, 0x00, 0x00, /* instruct array size */
  3797. 0x10, 0x81, 0x01, 0x00, 0x10, 0x05, 0x07, 0x00, 0x00, 0x1d, 0x00, 0x00,
  3798. 0x00, 0x04, 0x09, 0x00, 0x10, 0x81, 0x01, 0x00, 0x00, 0x02, 0x12, 0x00,
  3799. 0x00, 0x04, 0x17, 0x00, 0x00, 0x82, 0x1b, 0x00, 0x00, 0x04, 0x28, 0x00,
  3800. 0x00, 0x0d, 0x28, 0x00, 0x00, 0x07, 0x2e, 0x00, 0x11, 0x81, 0x28, 0x00,
  3801. 0x11, 0x01, 0x09, 0x00, 0x01, 0x08, 0x30, 0x00, 0x01, 0x07, 0x32, 0x00,
  3802. 0x02, 0x81, 0x28, 0x00, 0x02, 0x04, 0x09, 0x00, 0x00, 0x86, 0x34, 0x00,
  3803. 0x00, 0x8c, 0x17, 0x00, /* instruct array */
  3804. 0x37, 0x00, 0x00, 0x00, /* const pool size */
  3805. 0x00, 0x40, 0x6c, 0x69, 0x73, 0x74, 0x00, 0x30, 0x00, 0x40, 0x72, 0x65,
  3806. 0x73, 0x5f, 0x6d, 0x61, 0x78, 0x00, 0x69, 0x74, 0x65, 0x72, 0x00, 0x24,
  3807. 0x6c, 0x30, 0x00, 0x24, 0x6c, 0x30, 0x2e, 0x5f, 0x5f, 0x6e, 0x65, 0x78,
  3808. 0x74, 0x5f, 0x5f, 0x00, 0x40, 0x69, 0x74, 0x65, 0x6d, 0x00, 0x32, 0x00,
  3809. 0x3c, 0x00, 0x31, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  3810. };
  3811. Arg* _max_min(PikaObj* self, PikaTuple* val, uint8_t* bc) {
  3812. int size = pikaTuple_getSize(val);
  3813. if (size == 0) {
  3814. obj_setSysOut(self, "TypeError: max expected 1 arguments, got 0");
  3815. obj_setErrorCode(self, PIKA_RES_ERR_INVALID_PARAM);
  3816. return NULL;
  3817. }
  3818. if (size == 1) {
  3819. ArgType type = arg_getType(pikaTuple_getArg(val, 0));
  3820. if (!argType_isIterable(type)) {
  3821. obj_setSysOut(self, "TypeError: object is not iterable");
  3822. obj_setErrorCode(self, PIKA_RES_ERR_INVALID_PARAM);
  3823. return NULL;
  3824. }
  3825. obj_setArg(self, "@list", pikaTuple_getArg(val, 0));
  3826. return pikaVM_runByteCodeReturn(self, (uint8_t*)bc, "@res_max");
  3827. }
  3828. PikaTuple* oTuple = newNormalObj(New_PikaStdData_Tuple);
  3829. obj_setPtr(oTuple, "list", obj_getPtr(val, "list"));
  3830. obj_setInt(oTuple, "needfree", 0);
  3831. Arg* aTuple = arg_newObj(oTuple);
  3832. obj_setArg(self, "@list", aTuple);
  3833. Arg* aRet = pikaVM_runByteCodeReturn(self, (uint8_t*)bc, "@res_max");
  3834. arg_deinit(aTuple);
  3835. return aRet;
  3836. }
  3837. Arg* builtins_max(PikaObj* self, PikaTuple* val) {
  3838. return _max_min(self, val, (uint8_t*)bc_max);
  3839. }
  3840. Arg* builtins_min(PikaObj* self, PikaTuple* val) {
  3841. return _max_min(self, val, (uint8_t*)bc_min);
  3842. }
  3843. static PIKA_BOOL _check_no_buff_format(char* format) {
  3844. while (*format) {
  3845. if (*format == '%') {
  3846. ++format;
  3847. if (*format != 's' && *format != '%') {
  3848. return PIKA_FALSE;
  3849. }
  3850. }
  3851. ++format;
  3852. }
  3853. return PIKA_TRUE;
  3854. }
  3855. int pika_pvsprintf(char** buff, const char* fmt, va_list args) {
  3856. int required_size;
  3857. int current_size = PIKA_SPRINTF_BUFF_SIZE;
  3858. *buff = (char*)pika_platform_malloc(current_size * sizeof(char));
  3859. if (*buff == NULL) {
  3860. return -1; // Memory allocation failed
  3861. }
  3862. va_list args_copy;
  3863. va_copy(args_copy, args);
  3864. required_size =
  3865. pika_platform_vsnprintf(*buff, current_size, fmt, args_copy);
  3866. va_end(args_copy);
  3867. while (required_size >= current_size) {
  3868. current_size *= 2;
  3869. char* new_buff = (char*)pika_reallocn(*buff, current_size / 2,
  3870. current_size * sizeof(char));
  3871. if (new_buff == NULL) {
  3872. pika_platform_free(*buff);
  3873. *buff = NULL;
  3874. return -1; // Memory allocation failed
  3875. } else {
  3876. *buff = new_buff;
  3877. }
  3878. va_copy(args_copy, args);
  3879. required_size =
  3880. pika_platform_vsnprintf(*buff, current_size, fmt, args_copy);
  3881. va_end(args_copy);
  3882. }
  3883. return required_size;
  3884. }
  3885. static int _no_buff_vprintf(char* fmt, va_list args) {
  3886. int written = 0;
  3887. while (*fmt) {
  3888. if (*fmt == '%') {
  3889. ++fmt;
  3890. if (*fmt == 's') {
  3891. char* str = va_arg(args, char*);
  3892. if (str == NULL) {
  3893. str = "(null)";
  3894. }
  3895. int len = strlen(str);
  3896. written += len;
  3897. for (int i = 0; i < len; i++) {
  3898. pika_putchar(str[i]);
  3899. }
  3900. } else if (*fmt == '%') {
  3901. pika_putchar('%');
  3902. ++written;
  3903. }
  3904. } else {
  3905. pika_putchar(*fmt);
  3906. ++written;
  3907. }
  3908. ++fmt;
  3909. }
  3910. return written;
  3911. }
  3912. int pika_vprintf(char* fmt, va_list args) {
  3913. int ret = 0;
  3914. if (_check_no_buff_format(fmt)) {
  3915. _no_buff_vprintf(fmt, args);
  3916. return 0;
  3917. }
  3918. char* buff = NULL;
  3919. int required_size = pika_pvsprintf(&buff, fmt, args);
  3920. if (required_size < 0) {
  3921. ret = -1; // Memory allocation or other error occurred
  3922. goto __exit;
  3923. }
  3924. // putchar
  3925. for (int i = 0; i < strlen(buff); i++) {
  3926. pika_putchar(buff[i]);
  3927. }
  3928. __exit:
  3929. if (NULL != buff) {
  3930. pika_platform_free(buff);
  3931. }
  3932. return ret;
  3933. }
  3934. int pika_sprintf(char* buff, char* fmt, ...) {
  3935. va_list args;
  3936. va_start(args, fmt);
  3937. int res = pika_platform_vsnprintf(buff, PIKA_SPRINTF_BUFF_SIZE, fmt, args);
  3938. va_end(args);
  3939. if (res >= PIKA_SPRINTF_BUFF_SIZE) {
  3940. pika_platform_printf(
  3941. "OverflowError: sprintf buff size overflow, please use bigger "
  3942. "PIKA_SPRINTF_BUFF_SIZE\r\n");
  3943. pika_platform_printf("Info: buff size request: %d\r\n", res);
  3944. pika_platform_printf("Info: buff size now: %d\r\n",
  3945. PIKA_SPRINTF_BUFF_SIZE);
  3946. while (1)
  3947. ;
  3948. }
  3949. return res;
  3950. }
  3951. int pika_vsprintf(char* buff, char* fmt, va_list args) {
  3952. /* vsnprintf */
  3953. return pika_platform_vsnprintf(buff, PIKA_SPRINTF_BUFF_SIZE, fmt, args);
  3954. }
  3955. int pika_snprintf(char* buff, size_t size, const char* fmt, ...) {
  3956. va_list args;
  3957. va_start(args, fmt);
  3958. int ret = pika_platform_vsnprintf(buff, size, fmt, args);
  3959. va_end(args);
  3960. return ret;
  3961. }
  3962. void _do_vsysOut(char* fmt, va_list args) {
  3963. char* fmt_buff = pikaMalloc(strGetSize(fmt) + 2);
  3964. pika_platform_memcpy(fmt_buff, fmt, strGetSize(fmt));
  3965. fmt_buff[strGetSize(fmt)] = '\n';
  3966. fmt_buff[strGetSize(fmt) + 1] = '\0';
  3967. char* out_buff = pikaMalloc(PIKA_SPRINTF_BUFF_SIZE);
  3968. pika_platform_vsnprintf(out_buff, PIKA_SPRINTF_BUFF_SIZE, fmt_buff, args);
  3969. pika_platform_printf("%s", out_buff);
  3970. pikaFree(out_buff, PIKA_SPRINTF_BUFF_SIZE);
  3971. pikaFree(fmt_buff, strGetSize(fmt) + 2);
  3972. }
  3973. void obj_setSysOut(PikaObj* self, char* fmt, ...) {
  3974. if (NULL != self->vmFrame) {
  3975. if (self->vmFrame->error.code == PIKA_RES_OK) {
  3976. self->vmFrame->error.code = PIKA_RES_ERR_RUNTIME_ERROR;
  3977. }
  3978. if (self->vmFrame->vm_thread->try_state == TRY_STATE_INNER) {
  3979. return;
  3980. }
  3981. }
  3982. va_list args;
  3983. va_start(args, fmt);
  3984. _do_vsysOut(fmt, args);
  3985. va_end(args);
  3986. }
  3987. pika_bool builtins_isinstance(PikaObj* self, Arg* object, Arg* classinfo) {
  3988. if (!argType_isConstructor(arg_getType(classinfo)) &&
  3989. !argType_isCallable(arg_getType(classinfo))) {
  3990. obj_setErrorCode(self, 1);
  3991. obj_setSysOut(self, "TypeError: isinstance() arg 2 must be a type");
  3992. return pika_false;
  3993. }
  3994. return _isinstance(object, classinfo);
  3995. }
  3996. Arg* builtins_StringObj___next__(PikaObj* self) {
  3997. return arg_newNone();
  3998. }
  3999. Arg* builtins_RangeObj___next__(PikaObj* self) {
  4000. RangeData* _ = (RangeData*)args_getStruct(self->list, "_");
  4001. int end = _->end;
  4002. int step = _->step;
  4003. /* exit */
  4004. if (_->i >= end) {
  4005. return arg_newNone();
  4006. }
  4007. Arg* res = arg_newInt(_->i);
  4008. _->i += step;
  4009. return res;
  4010. }
  4011. PikaObj* New_builtins(Args* args);
  4012. PikaObj* obj_getBuiltins(void) {
  4013. return newNormalObj(New_builtins);
  4014. }
  4015. void builtins_bytearray___init__(PikaObj* self, Arg* bytes) {
  4016. obj_setArg_noCopy(self, "raw", builtins_bytes(self, bytes));
  4017. }
  4018. Arg* builtins_bytearray___iter__(PikaObj* self) {
  4019. obj_setInt(self, "__iter_i", 0);
  4020. return arg_newRef(self);
  4021. }
  4022. int builtins_bytearray___len__(PikaObj* self) {
  4023. return obj_getBytesSize(self, "raw");
  4024. }
  4025. Arg* builtins_bytearray___next__(PikaObj* self) {
  4026. int __iter_i = args_getInt(self->list, "__iter_i");
  4027. uint8_t* data = obj_getBytes(self, "raw");
  4028. uint16_t len = obj_getBytesSize(self, "raw");
  4029. Arg* res = NULL;
  4030. char char_buff[] = " ";
  4031. if (__iter_i < len) {
  4032. char_buff[0] = data[__iter_i];
  4033. res = arg_newInt(char_buff[0]);
  4034. } else {
  4035. return arg_newNone();
  4036. }
  4037. args_setInt(self->list, "__iter_i", __iter_i + 1);
  4038. return res;
  4039. }
  4040. int builtins_bytearray___getitem__(PikaObj* self, int __key) {
  4041. uint8_t* data = obj_getBytes(self, "raw");
  4042. uint16_t len = obj_getBytesSize(self, "raw");
  4043. if (__key < len) {
  4044. return data[__key];
  4045. } else {
  4046. return 0;
  4047. }
  4048. }
  4049. pika_bool _bytes_contains(Arg* self, Arg* others) {
  4050. ArgType type = arg_getType(others);
  4051. if (type == ARG_TYPE_BYTES) {
  4052. uint8_t* bytes1 = arg_getBytes(self);
  4053. uint8_t* bytes2 = arg_getBytes(others);
  4054. size_t size1 = arg_getBytesSize(self);
  4055. size_t size2 = arg_getBytesSize(others);
  4056. if (size1 > size2) {
  4057. return pika_false;
  4058. }
  4059. size_t i = 0, j = 0, start_j = 0;
  4060. while (j < size2) {
  4061. if (bytes1[i] == bytes2[j]) {
  4062. i++;
  4063. j++;
  4064. if (i == size1) {
  4065. return pika_true;
  4066. }
  4067. } else {
  4068. j = ++start_j; // Move `j` to the next start position
  4069. i = 0; // Reset `i`
  4070. }
  4071. }
  4072. }
  4073. return pika_false;
  4074. }
  4075. int builtins_bytearray___contains__(PikaObj* self, Arg* others) {
  4076. if (arg_isObject(others)) {
  4077. others = obj_getArg(arg_getObj(others), "raw");
  4078. }
  4079. return _bytes_contains(others, obj_getArg(self, "raw"));
  4080. }
  4081. void builtins_bytearray___setitem__(PikaObj* self, int __key, int __val) {
  4082. uint8_t* data = obj_getBytes(self, "raw");
  4083. uint16_t len = obj_getBytesSize(self, "raw");
  4084. if (__key < len) {
  4085. data[__key] = __val;
  4086. }
  4087. }
  4088. char* builtins_bytearray___str__(PikaObj* self) {
  4089. uint8_t* data = obj_getBytes(self, "raw");
  4090. uint16_t len = obj_getBytesSize(self, "raw");
  4091. Arg* str_arg = arg_newStr("");
  4092. str_arg = arg_strAppend(str_arg, "bytearray(b'");
  4093. for (int i = 0; i < len; i++) {
  4094. char u8_str[] = "\\x00";
  4095. uint8_t u8 = data[i];
  4096. __platform_sprintf(u8_str, "\\x%02x", u8);
  4097. str_arg = arg_strAppend(str_arg, u8_str);
  4098. }
  4099. str_arg = arg_strAppend(str_arg, "')");
  4100. obj_removeArg(self, "_buf");
  4101. obj_setStr(self, "_buf", arg_getStr(str_arg));
  4102. arg_deinit(str_arg);
  4103. return obj_getStr(self, "_buf");
  4104. }
  4105. char* builtins_bytearray_decode(PikaObj* self) {
  4106. uint8_t* data = obj_getBytes(self, "raw");
  4107. return obj_cacheStr(self, (char*)data);
  4108. }
  4109. PikaObj* builtins_bytearray_split(PikaObj* self, PikaTuple* vars) {
  4110. int max_splite = -1;
  4111. Arg* sep = NULL;
  4112. pika_bool sep_need_free = pika_false;
  4113. if (pikaTuple_getSize(vars) >= 1) {
  4114. sep = pikaTuple_getArg(vars, 0);
  4115. } else {
  4116. sep = arg_newBytes((uint8_t*)" ", 1);
  4117. sep_need_free = pika_true;
  4118. }
  4119. if (pikaTuple_getSize(vars) == 2) {
  4120. max_splite = pikaTuple_getInt(vars, 1);
  4121. }
  4122. if (arg_getType(sep) != ARG_TYPE_BYTES) {
  4123. obj_setErrorCode(self, 1);
  4124. obj_setSysOut(self, "TypeError: bytes is required");
  4125. return NULL;
  4126. }
  4127. PikaList* list = NULL;
  4128. uint8_t* sep_data = arg_getBytes(sep);
  4129. size_t sep_len = arg_getBytesSize(sep);
  4130. if (sep_len == 0) {
  4131. obj_setErrorCode(self, 1);
  4132. obj_setSysOut(self, "ValueError: empty separator");
  4133. goto __exit;
  4134. }
  4135. list = New_PikaList();
  4136. uint8_t* data = obj_getBytes(self, "raw");
  4137. size_t len = obj_getBytesSize(self, "raw");
  4138. if (len == 0) {
  4139. pikaList_append(list, arg_newBytes((uint8_t*)"", 0));
  4140. goto __exit;
  4141. }
  4142. size_t start = 0;
  4143. int splits = 0;
  4144. for (size_t i = 0; i < len; i++) {
  4145. // Check if the current position can accommodate the separator
  4146. if (i <= len - sep_len && memcmp(data + i, sep_data, sep_len) == 0) {
  4147. // Found a separator, slice the bytearray from start to current
  4148. // position
  4149. uint8_t* target_data = data + start;
  4150. size_t target_len = i - start;
  4151. pikaList_append(list, arg_newBytes(target_data, target_len));
  4152. // Move the start index past the separator
  4153. start = i + sep_len;
  4154. i = start - 1; // Adjust `i` as it will be incremented by the loop
  4155. // Increment the split count and check if we've reached the maximum
  4156. // number of splits
  4157. splits++;
  4158. if (max_splite != -1 && splits >= max_splite) {
  4159. break;
  4160. }
  4161. }
  4162. }
  4163. // Add the last segment if any remains
  4164. if (start < len) {
  4165. uint8_t* target_data = data + start;
  4166. size_t target_len = len - start;
  4167. pikaList_append(list, arg_newBytes(target_data, target_len));
  4168. }
  4169. // After loop
  4170. if (sep_len > 0 && start == len) {
  4171. pikaList_append(list, arg_newBytes((uint8_t*)"", 0));
  4172. }
  4173. __exit:
  4174. if (sep_need_free) {
  4175. arg_deinit(sep);
  4176. }
  4177. return list;
  4178. }
  4179. int32_t pikaDict_forEach(PikaObj* self,
  4180. int32_t (*eachHandle)(PikaObj* self,
  4181. Arg* keyEach,
  4182. Arg* valEach,
  4183. void* context),
  4184. void* context) {
  4185. Args* keys = _OBJ2KEYS(self);
  4186. Args* dict = _OBJ2DICT(self);
  4187. pika_assert(NULL != dict);
  4188. pika_assert(NULL != keys);
  4189. size_t size = args_getSize(keys);
  4190. for (int i = size - 1; i >= 0; i--) {
  4191. Arg* item_key = args_getArgByIndex(keys, i);
  4192. Arg* item_val = args_getArgByIndex(dict, i);
  4193. pika_assert(NULL != item_key);
  4194. // Call the handle function on each key-value pair
  4195. if (eachHandle(self, item_key, item_val, context) != 0) {
  4196. return -1;
  4197. }
  4198. }
  4199. return 0;
  4200. }
  4201. int32_t pikaList_forEach(PikaObj* self,
  4202. int32_t (*eachHandle)(PikaObj* self,
  4203. int itemIndex,
  4204. Arg* itemEach,
  4205. void* context),
  4206. void* context) {
  4207. int i = 0;
  4208. while (PIKA_TRUE) {
  4209. Arg* item = pikaList_get(self, i);
  4210. if (NULL == item) {
  4211. break;
  4212. }
  4213. // Call the handler function with the item.
  4214. int32_t result = eachHandle(self, i, item, context);
  4215. if (result != 0) {
  4216. // If the handler function returns a non-zero value,
  4217. // stop the iteration.
  4218. return result;
  4219. }
  4220. i++;
  4221. }
  4222. return 0;
  4223. }
  4224. void pika_sleep_ms(uint32_t ms) {
  4225. int64_t tick = pika_platform_get_tick();
  4226. while (1) {
  4227. pika_platform_thread_yield();
  4228. #if PIKA_EVENT_ENABLE
  4229. if (!pika_GIL_isInit()) {
  4230. _VMEvent_pickupEvent();
  4231. }
  4232. #endif
  4233. if (pika_platform_get_tick() - tick >= ms) {
  4234. break;
  4235. }
  4236. }
  4237. }
  4238. PikaObj* _New_pikaListOrTuple(int isTuple) {
  4239. Args* list = New_args(NULL);
  4240. /* set top index for append */
  4241. args_pushArg_name(list, "top", arg_newInt(0));
  4242. PikaObj* self = NULL;
  4243. if (isTuple) {
  4244. self = newNormalObj(New_PikaStdData_Tuple);
  4245. } else {
  4246. self = newNormalObj(New_PikaStdData_List);
  4247. }
  4248. obj_setPtr(self, "list", list);
  4249. return self;
  4250. }
  4251. PikaList* New_PikaList(void) {
  4252. return _New_pikaListOrTuple(0);
  4253. }
  4254. PikaTuple* New_pikaTuple(void) {
  4255. return _New_pikaListOrTuple(1);
  4256. }
  4257. void pikaDict_init(PikaObj* self) {
  4258. Args* dict = New_args(NULL);
  4259. Args* keys = New_args(NULL);
  4260. obj_setPtr(self, "dict", dict);
  4261. obj_setPtr(self, "_keys", keys);
  4262. }
  4263. PikaDict* New_PikaDict(void) {
  4264. PikaDict* self = newNormalObj(New_PikaStdData_Dict);
  4265. pikaDict_init(self);
  4266. return self;
  4267. }
  4268. PIKA_RES pikaList_set(PikaList* self, int index, Arg* arg) {
  4269. char buff[11];
  4270. char* i_str = fast_itoa(buff, index);
  4271. int top = pikaList_getSize(self);
  4272. if (index > top) {
  4273. return PIKA_RES_ERR_OUT_OF_RANGE;
  4274. }
  4275. arg = arg_setName(arg, i_str);
  4276. args_setArg(_OBJ2LIST(self), arg);
  4277. return PIKA_RES_OK;
  4278. }
  4279. Arg* pikaList_get(PikaList* self, int index) {
  4280. pika_assert(NULL != self);
  4281. char buff[11];
  4282. char* i_str = fast_itoa(buff, index);
  4283. return args_getArg(_OBJ2LIST(self), i_str);
  4284. }
  4285. int pikaList_getInt(PikaList* self, int index) {
  4286. Arg* arg = pikaList_get(self, index);
  4287. return arg_getInt(arg);
  4288. }
  4289. pika_bool pikaList_getBool(PikaList* self, int index) {
  4290. Arg* arg = pikaList_get(self, index);
  4291. return arg_getBool(arg);
  4292. }
  4293. pika_float pikaList_getFloat(PikaList* self, int index) {
  4294. Arg* arg = pikaList_get(self, index);
  4295. return arg_getFloat(arg);
  4296. }
  4297. char* pikaList_getStr(PikaList* self, int index) {
  4298. Arg* arg = pikaList_get(self, index);
  4299. return arg_getStr(arg);
  4300. }
  4301. void* pikaList_getPtr(PikaList* self, int index) {
  4302. Arg* arg = pikaList_get(self, index);
  4303. return arg_getPtr(arg);
  4304. }
  4305. PikaObj* pikaList_getObj(PikaList* self, int index) {
  4306. Arg* arg = pikaList_get(self, index);
  4307. return arg_getObj(arg);
  4308. }
  4309. PIKA_RES pikaList_append(PikaList* self, Arg* arg) {
  4310. if (NULL == arg) {
  4311. return PIKA_RES_ERR_ARG_NO_FOUND;
  4312. }
  4313. int top = pikaList_getSize(self);
  4314. char buff[11];
  4315. char* topStr = fast_itoa(buff, top);
  4316. Arg* arg_to_push = arg;
  4317. arg_setName(arg_to_push, topStr);
  4318. args_setArg(_OBJ2LIST(self), arg_to_push);
  4319. /* top++ */
  4320. return args_setInt(_OBJ2LIST(self), "top", top + 1);
  4321. }
  4322. void pikaList_deinit(PikaList* self) {
  4323. args_deinit(_OBJ2LIST(self));
  4324. }
  4325. void pikaTuple_deinit(PikaTuple* self) {
  4326. pikaList_deinit(self);
  4327. }
  4328. ArgType pikaList_getType(PikaList* self, int index) {
  4329. Arg* arg = pikaList_get(self, index);
  4330. return arg_getType(arg);
  4331. }
  4332. Arg* pikaList_pop_withIndex(PikaList* self, int index) {
  4333. int top = pikaList_getSize(self);
  4334. if (top <= 0) {
  4335. return NULL;
  4336. }
  4337. if (index < 0) {
  4338. index = top + index;
  4339. }
  4340. Arg* arg = pikaList_get(self, index);
  4341. Arg* res = arg_copy(arg);
  4342. pikaList_remove(self, arg);
  4343. return res;
  4344. }
  4345. Arg* pikaList_pop(PikaList* self) {
  4346. int top = pikaList_getSize(self);
  4347. if (top <= 0) {
  4348. return NULL;
  4349. }
  4350. return pikaList_pop_withIndex(self, top - 1);
  4351. }
  4352. PIKA_RES pikaList_remove(PikaList* self, Arg* arg) {
  4353. int top = pikaList_getSize(self);
  4354. int i_remove = 0;
  4355. if (top <= 0) {
  4356. return PIKA_RES_ERR_OUT_OF_RANGE;
  4357. }
  4358. for (int i = 0; i < top; i++) {
  4359. Arg* arg_now = pikaList_get(self, i);
  4360. if (arg_isEqual(arg_now, arg)) {
  4361. i_remove = i;
  4362. args_removeArg(_OBJ2LIST(self), arg_now);
  4363. break;
  4364. }
  4365. }
  4366. /* move args */
  4367. for (int i = i_remove + 1; i < top; i++) {
  4368. char buff[11];
  4369. char* i_str = fast_itoa(buff, i - 1);
  4370. Arg* arg_now = pikaList_get(self, i);
  4371. arg_setName(arg_now, i_str);
  4372. }
  4373. args_setInt(_OBJ2LIST(self), "top", top - 1);
  4374. return PIKA_RES_OK;
  4375. }
  4376. PIKA_RES pikaList_insert(PikaList* self, int index, Arg* arg) {
  4377. int top = pikaList_getSize(self);
  4378. if (index > top) {
  4379. return PIKA_RES_ERR_OUT_OF_RANGE;
  4380. }
  4381. /* move args */
  4382. for (int i = top - 1; i >= index; i--) {
  4383. char buff[11];
  4384. char* i_str = fast_itoa(buff, i + 1);
  4385. Arg* arg_now = pikaList_get(self, i);
  4386. arg_setName(arg_now, i_str);
  4387. }
  4388. char buff[11];
  4389. char* i_str = fast_itoa(buff, index);
  4390. Arg* arg_to_push = arg_copy(arg);
  4391. arg_setName(arg_to_push, i_str);
  4392. args_setArg(_OBJ2LIST(self), arg_to_push);
  4393. args_setInt(_OBJ2LIST(self), "top", top + 1);
  4394. return PIKA_RES_OK;
  4395. }
  4396. size_t pikaList_getSize(PikaList* self) {
  4397. if (NULL == self) {
  4398. return 0;
  4399. }
  4400. int64_t ret = args_getInt(_OBJ2LIST(self), "top");
  4401. pika_assert(ret >= 0);
  4402. return ret;
  4403. }
  4404. void pikaList_init(PikaObj* self) {
  4405. Args* list = New_args(NULL);
  4406. args_pushArg_name(list, "top", arg_newInt(0));
  4407. obj_setPtr(self, "list", list);
  4408. }
  4409. void pikaList_reverse(PikaList* self) {
  4410. pika_assert(NULL != self);
  4411. int top = pikaList_getSize(self);
  4412. for (int i = 0; i < top / 2; i++) {
  4413. Arg* arg_i = arg_copy(pikaList_get(self, i));
  4414. Arg* arg_top = arg_copy(pikaList_get(self, top - i - 1));
  4415. pikaList_set(self, i, arg_top);
  4416. pikaList_set(self, top - i - 1, arg_i);
  4417. }
  4418. }
  4419. Arg* pikaTuple_getArg(PikaTuple* self, int index) {
  4420. return pikaList_get(self, (index));
  4421. }
  4422. size_t pikaTuple_getSize(PikaTuple* self) {
  4423. if (self == NULL) {
  4424. return 0;
  4425. }
  4426. return pikaList_getSize(self);
  4427. }
  4428. int64_t pikaTuple_getInt(PikaTuple* self, int index) {
  4429. return pikaList_getInt(self, (index));
  4430. }
  4431. pika_bool pikaTuple_getBool(PikaTuple* self, int index) {
  4432. return pikaList_getBool(self, (index));
  4433. }
  4434. pika_float pikaTuple_getFloat(PikaTuple* self, int index) {
  4435. return pikaList_getFloat(self, (index));
  4436. }
  4437. char* pikaTuple_getStr(PikaTuple* self, int index) {
  4438. return pikaList_getStr(self, (index));
  4439. }
  4440. void* pikaTuple_getPtr(PikaTuple* self, int index) {
  4441. return pikaList_getPtr(self, (index));
  4442. }
  4443. ArgType pikaTuple_getType(PikaTuple* self, int index) {
  4444. return pikaList_getType(self, (index));
  4445. }
  4446. PIKA_RES pikaDict_setInt(PikaDict* self, char* name, int64_t val) {
  4447. return pikaDict_set(self, name, arg_newInt(val));
  4448. }
  4449. pika_bool pikaDict_setBool(PikaDict* self, char* name, pika_bool val) {
  4450. return pikaDict_set(self, name, arg_newBool(val));
  4451. }
  4452. PIKA_RES pikaDict_setFloat(PikaDict* self, char* name, pika_float val) {
  4453. return pikaDict_set(self, name, arg_newFloat(val));
  4454. }
  4455. PIKA_RES pikaDict_setStr(PikaDict* self, char* name, char* val) {
  4456. return pikaDict_set(self, name, arg_newStr(val));
  4457. }
  4458. PIKA_RES pikaDict_setPtr(PikaDict* self, char* name, void* val) {
  4459. return pikaDict_set(self, name, arg_newPtr(ARG_TYPE_POINTER, val));
  4460. }
  4461. PIKA_RES _pikaDict_setVal(PikaDict* self, Arg* val) {
  4462. return args_setArg(_OBJ2DICT(self), (val));
  4463. }
  4464. PIKA_RES pikaDict_set(PikaDict* self, char* name, Arg* val) {
  4465. val = arg_setName(val, name);
  4466. _pikaDict_setVal(self, val);
  4467. return args_setStr(_OBJ2KEYS(self), (name), (name));
  4468. }
  4469. PIKA_RES pikaDict_removeArg(PikaDict* self, Arg* val) {
  4470. // args_removeArg(_OBJ2KEYS(self), (val));
  4471. return args_removeArg(_OBJ2DICT(self), (val));
  4472. }
  4473. PIKA_RES pikaDict_reverse(PikaDict* self) {
  4474. args_reverse(_OBJ2KEYS(self));
  4475. args_reverse(_OBJ2DICT(self));
  4476. return PIKA_RES_OK;
  4477. }
  4478. PIKA_RES pikaDict_setBytes(PikaDict* self,
  4479. char* name,
  4480. uint8_t* val,
  4481. size_t size) {
  4482. return pikaDict_set(self, name, arg_newBytes(val, size));
  4483. }
  4484. int64_t pikaDict_getInt(PikaDict* self, char* name) {
  4485. return args_getInt(_OBJ2DICT(self), (name));
  4486. }
  4487. pika_bool pikaDict_getBool(PikaDict* self, char* name) {
  4488. return args_getBool(_OBJ2DICT(self), (name));
  4489. }
  4490. pika_float pikaDict_getFloat(PikaDict* self, char* name) {
  4491. return args_getFloat(_OBJ2DICT(self), (name));
  4492. }
  4493. char* pikaDict_getStr(PikaDict* self, char* name) {
  4494. return args_getStr(_OBJ2DICT(self), (name));
  4495. }
  4496. void* pikaDict_getPtr(PikaDict* self, char* name) {
  4497. return args_getPtr(_OBJ2DICT(self), (name));
  4498. }
  4499. PikaObj* pikaDict_getObj(PikaDict* self, char* name) {
  4500. return args_getPtr(_OBJ2DICT(self), (name));
  4501. }
  4502. int pikaDict_getSize(PikaDict* self) {
  4503. return args_getSize(_OBJ2DICT(self));
  4504. }
  4505. Arg* pikaDict_getArgByindex(PikaDict* self, int index) {
  4506. return args_getArgByIndex(_OBJ2DICT(self), (index));
  4507. }
  4508. Arg* pikaDict_get(PikaDict* self, char* name) {
  4509. return args_getArg(_OBJ2DICT(self), (name));
  4510. }
  4511. int32_t pikaDict_isArgExist(PikaDict* self, char* name) {
  4512. return args_isArgExist(_OBJ2DICT(self), (name));
  4513. }
  4514. uint8_t* pikaDict_getBytes(PikaDict* self, char* name) {
  4515. return args_getBytes(_OBJ2DICT(self), (name));
  4516. }
  4517. ArgType pikaDict_getType(PikaDict* self, char* name) {
  4518. return args_getType(_OBJ2DICT(self), (name));
  4519. }
  4520. size_t pikaDict_getBytesSize(PikaDict* self, char* name) {
  4521. return args_getBytesSize(_OBJ2DICT(self), (name));
  4522. }
  4523. void pikaDict_deinit(PikaDict* self) {
  4524. obj_deinit(self);
  4525. }
  4526. PIKA_RES insert_label_at_line(const char* filename,
  4527. int N,
  4528. char* buff,
  4529. int buff_size) {
  4530. FILE* file = pika_platform_fopen(filename, "r");
  4531. if (!file) {
  4532. return PIKA_RES_ERR_IO_ERROR;
  4533. }
  4534. int line_count = 1;
  4535. int buff_index = 0;
  4536. char ch;
  4537. while ((pika_platform_fread(&ch, 1, 1, file)) != 0) {
  4538. if (line_count == N) {
  4539. char* label = "#!label\n";
  4540. int label_length = strGetSize(label);
  4541. if (buff_index + label_length >= buff_size) {
  4542. pika_platform_fclose(file);
  4543. return PIKA_RES_ERR_IO_ERROR; // Insufficient buffer size
  4544. }
  4545. strCopy(buff + buff_index, label);
  4546. buff_index += label_length;
  4547. line_count++; // Skip this line since we're adding a label
  4548. }
  4549. if (ch == '\n') {
  4550. line_count++;
  4551. }
  4552. if (buff_index >=
  4553. buff_size - 1) { // -1 to leave space for null terminator
  4554. pika_platform_fclose(file);
  4555. return PIKA_RES_ERR_IO_ERROR; // Insufficient buffer size
  4556. }
  4557. buff[buff_index] = ch;
  4558. buff_index++;
  4559. }
  4560. buff[buff_index] = '\0'; // Null terminate the buffer
  4561. pika_platform_fclose(file);
  4562. return PIKA_RES_OK;
  4563. }
  4564. int32_t pika_debug_find_break_point_pc(char* pyFile, uint32_t pyLine) {
  4565. ByteCodeFrame bytecode_frame;
  4566. byteCodeFrame_init(&bytecode_frame);
  4567. char* file_buff = pikaMalloc(PIKA_READ_FILE_BUFF_SIZE);
  4568. FILE* file = pika_platform_fopen(pyFile, "r");
  4569. if (!file) {
  4570. goto __exit;
  4571. }
  4572. if (PIKA_RES_OK != insert_label_at_line(pyFile, pyLine, file_buff,
  4573. PIKA_READ_FILE_BUFF_SIZE)) {
  4574. goto __exit;
  4575. }
  4576. pika_lines2Bytes(&bytecode_frame, file_buff);
  4577. __exit:
  4578. if (NULL != file) {
  4579. pika_platform_fclose(file);
  4580. }
  4581. byteCodeFrame_deinit(&bytecode_frame);
  4582. pikaFree(file_buff, PIKA_READ_FILE_BUFF_SIZE);
  4583. return bytecode_frame.label_pc;
  4584. }