12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429543054315432543354345435543654375438543954405441544254435444544554465447544854495450545154525453545454555456545754585459546054615462546354645465546654675468546954705471547254735474547554765477547854795480548154825483548454855486548754885489549054915492549354945495549654975498549955005501550255035504550555065507550855095510551155125513551455155516551755185519552055215522552355245525552655275528552955305531553255335534553555365537553855395540554155425543554455455546554755485549555055515552555355545555555655575558555955605561556255635564556555665567556855695570557155725573557455755576557755785579558055815582558355845585558655875588558955905591559255935594559555965597559855995600560156025603560456055606560756085609561056115612561356145615561656175618561956205621562256235624562556265627562856295630563156325633563456355636563756385639564056415642564356445645564656475648564956505651565256535654565556565657565856595660566156625663566456655666566756685669 |
- #!/bin/sh
- # Prime Mover
- #
- # (C) 2006 David Given.
- # Prime Mover is licensed under the MIT open source license. To get the full
- # license text, run this file with the '--license' option.
- #
- # $Id: shell 52 2006-10-03 22:12:33Z dtrg $
- if [ -x "$(which arch 2>/dev/null)" ]; then
- ARCH="$(arch)"
- elif [ -x "$(which machine 2>/dev/null)" ]; then
- ARCH="$(machine)"
- elif [ -x "$(which uname 2>/dev/null)" ]; then
- ARCH="$(uname -m)"
- else
- echo "pm: unable to determine target type, proceeding anyway"
- ARCH=unknown
- fi
-
- THISFILE="$0"
- PMEXEC="./.pm-exec-$ARCH"
- set -e
- GZFILE=/tmp/pm-$$.gz
- CFILE=/tmp/pm-$$.c
- trap "rm -f $GZFILE $CFILE" EXIT
- extract_section() {
- sed -e "1,/^XXXXSTART$1/d" "$THISFILE" | (
- read size
- dd bs=1 count=$size 2> /dev/null
- ) > $GZFILE
- cat $GZFILE | cat
- }
- # If the bootstrap's built, run it.
- if [ "$PMEXEC" -nt $0 ]; then
- extract_section script | "$PMEXEC" /dev/stdin "$@"
- exit $?
- fi
- # Otherwise, compile it and restart.
- echo "pm: bootstrapping..."
- if [ -x "$(which gcc 2>/dev/null)" ]; then
- CC="gcc -O -s"
- else
- CC="cc"
- fi
- extract_section interpreter > /tmp/pm-$$.c
- $CC $CFILE -o "$PMEXEC" && exec "$THISFILE" "$@"
- echo "pm: bootstrap failed."
- exit 1
- XXXXSTARTscript
- 38405
- #!/usr/bin/lua
- -- Prime Mover
- --
- -- © 2006 David Given.
- -- Prime Mover is licensed under the MIT open source license. Search
- -- for 'MIT' in this file to find the full license text.
- --
- -- $Id: pm 93 2007-02-24 21:20:53Z dtrg $
- -- ======================================================================= --
- -- GLOBALS --
- -- ======================================================================= --
- local VERSION = "0.1.1"
- -- Fast versions of useful system variables.
- local stdin = io.stdin
- local stdout = io.stdout
- local stderr = io.stderr
- local string_find = string.find
- local string_gsub = string.gsub
- local string_sub = string.sub
- local string_byte = string.byte
- local table_insert = table.insert
- local table_getn = table.getn
- local table_concat = table.concat
- local posix_stat = posix.stat
- local posix_readlink = posix.readlink
- local posix_unlink = posix.unlink
- local posix_rmdir = posix.rmdir
- local os_time = os.time
- local _G = _G
- local _
- -- Option settings.
- local delete_output_files_on_error = true
- local purge_intermediate_cache = false
- local no_execute = false
- local input_files = {}
- local targets = {}
- intermediate_cache_dir = ".pm-cache/"
- verbose = false
- quiet = false
- -- Application globals.
- local sandbox = {}
- local scope = {object=sandbox, next=nil}
- local intermediate_cache = {}
- local intermediate_cache_count = 0
- local buildstages = 0
- -- Atoms.
- local PARENT = {}
- local EMPTY = {}
- local REDIRECT = {}
- -- Exported symbols (set to dummy values).
- message = 0
- filetime = 0
- filetouch = 0
- install = 0
- rendertable = 0
- stringmodifier = {}
- setmetatable(_G, {__newindex = function(t, key, value)
- error("Attempt to write to new global "..key)
- end})
- -- ======================================================================= --
- -- UTILITIES --
- -- ======================================================================= --
- local function message(...)
- stderr:write("pm: ")
- stderr:write(unpack(arg))
- stderr:write("\n")
- end
- _G.message = message
- local function usererror(...)
- stderr:write("pm: ")
- stderr:write(unpack(arg))
- stderr:write("\n")
- os.exit(1)
- end
- local function traceoutput(...)
- stdout:write(unpack(arg))
- stdout:write("\n")
- end
- local function assert(message, result, e)
- if result then
- return result
- end
-
- if (type(message) == "string") then
- message = {message}
- end
-
- table.insert(message, ": ")
- table.insert(message, e)
- usererror(unpack(message))
- end
- -- Concatenates the contents of its arguments to the specified table.
- -- (Numeric indices only.)
- local function table_append(t, ...)
- for _, i in ipairs(arg) do
- if (type(i) == "table") then
- for _, j in ipairs(i) do
- table_insert(t, j)
- end
- else
- table_insert(t, i)
- end
- end
- end
- -- Merge the contents of its arguments to the specified table.
- -- (Name indices. Will break on numeric indices.)
- local function table_merge(t, ...)
- for _, i in ipairs(arg) do
- for j, k in pairs(i) do
- t[j] = k
- end
- end
- end
- -- Turn a list of strings into a single quoted string.
- function rendertable(i)
- if (type(i) == "string") or (type(i) == "number") then
- return i
- end
-
- if (i == nil) or (i == EMPTY) then
- return ""
- end
-
- local t = {}
- for _, j in ipairs(i) do
- local r = string_gsub(j, "\\", "\\\\")
- r = string_gsub(r, '"', '\\"')
- table_insert(t, r)
- end
- return '"'..table_concat(t, '" "')..'"'
- end
- local rendertable = rendertable
- -- Returns just the directory part of a path.
- local function dirname(f)
- local f, n = string_gsub(f, "/[^/]*$", "")
- if (n == 0) then
- return "."
- end
- return f
- end
- posix.dirname = dirname
- -- Makes an absolute path.
- local function absname(f)
- if string.find(f, "^/") then
- return f
- end
-
- return posix.getcwd().."/"..f
- end
- posix.absname = absname
- -- Copies a file.
- local function copy(src, dest)
- local s = string_gsub(src, "'", "'\"'\"'")
- local d = string_gsub(dest, "'", "'\"'\"'")
- local r = os.execute("cp '"..s.."' '"..d.."'")
- if (r ~= 0) then
- return nil, "unable to copy file"
- end
- return 0, nil
- end
- posix.copy = copy
-
- -- Makes all directories that contain f
- local function mkcontainerdir(f)
- f = dirname(f)
- if not posix_stat(f, "type") then
- mkcontainerdir(f)
- local r = posix.mkdir(f)
- if not r then
- usererror("unable to create directory '"..f.."'")
- end
- end
- end
- -- Install a file (suitable as a command list entry).
- local function do_install(self, src, dest)
- src = absname(self:__expand(src))
- dest = absname(self:__expand(dest))
- if verbose then
- message("installing '", src, "' --> '", dest, "'")
- end
-
- mkcontainerdir(dest)
- local f, e = posix.symlink(src, dest)
- if f then
- return
- end
- if (e ~= nil) then
- f, e = posix.unlink(dest)
- if f then
- f, e = posix.symlink(src, dest)
- if f then
- return
- end
- end
- end
-
- self:__error("couldn't install '", src, "' to '", dest,
- "': ", e)
- end
- function install(src, dest)
- return function(self, inputs, outputs)
- local src = src
- local dest = dest
-
- if (dest == nil) then
- dest = src
- src = outputs[1]
- end
- if (type(src) ~= "string") then
- self:__error("pm.install needs a string or an object for an input")
- end
- if (type(dest) ~= "string") then
- self:__error("pm.install needs a string for a destination")
- end
- return do_install(self, src, dest)
- end
- end
- -- Perform an error traceback.
- local function traceback(e)
- local i = 1
- while true do
- local t = debug.getinfo(i)
- if not t then
- break
- end
- if (t.short_src ~= "stdin") and (t.short_src ~= "[C]") then
- if (t.currentline == -1) then
- t.currentline = ""
- end
- message(" ", t.short_src, ":", t.currentline)
- end
- i = i + 1
- end
-
- e = string_gsub(e, "^stdin:[0-9]*: ", "")
- usererror("error: ", e)
- end
- -- ======================================================================= --
- -- CACHE MANAGEMENT --
- -- ======================================================================= --
- local statted_files = {}
- local function clear_stat_cache()
- statted_files = {}
- end
- -- Returns the timestamp of a file, or 0 if it doesn't exist.
- local statted_files = {}
- local function filetime(f)
- local t = statted_files[f]
- if t then
- return t
- end
-
- -- Stupid BeOS doesn't dereference symlinks on stat().
-
- local realf = f
- while true do
- local newf, e = posix_readlink(realf)
- if e then
- break
- end
- realf = newf
- end
-
- t = posix_stat(realf, "mtime") or 0
- statted_files[f] = t
- return t
- end
- _G.filetime = filetime
- -- Pretends to touch a file by manipulating the stat cache.
- local function filetouch(f)
- if (type(f) == "string") then
- f = {f}
- end
-
- local t = os_time()
- for _, i in ipairs(f) do
- statted_files[i] = t
- end
- end
- _G.filetouch = filetouch
- local function create_intermediate_cache()
- local d = dirname(intermediate_cache_dir)
- if not quiet then
- message("creating new intermediate file cache in '"..d.."'")
- end
-
- -- Attempt to wipe the old cache directory.
-
- local f = posix.files(d)
- if not f then
- -- The directory doesn't exist, so create it.
-
- mkcontainerdir(d)
- f = posix.mkdir(d)
- if not f then
- usererror("unable to create intermediate file cache directory")
- end
- else
- -- The directory exists. Delete all files in it recursively.
-
- local function rmdir(root)
- local f = posix.files(root)
- if not f then
- return
- end
-
- for i in f do
- if ((i ~= ".") and (i ~= "..")) then
- local fn = root.."/"..i
- local t = posix_stat(fn, "type")
- if (t == "regular") then
- if not posix_unlink(fn) then
- usererror("unable to purge intermediate file cache directory")
- end
- elseif (t == "directory") then
- rmdir(fn)
- posix_rmdir(fn)
- end
- end
- end
- end
-
- rmdir(d)
- end
- end
- local function save_intermediate_cache()
- local fn = intermediate_cache_dir.."index"
- local f = io.open(fn, "w")
- if not f then
- usererror("unable to save intermediate cache index file '", fn, "'")
- end
-
- f:write(intermediate_cache_count, "\n")
- for i, j in pairs(intermediate_cache) do
- f:write(i, "\n")
- f:write(j, "\n")
- end
-
- f:close()
- end
- local function load_intermediate_cache()
- local fn = intermediate_cache_dir.."index"
- local f = io.open(fn, "r")
- if not f then
- create_intermediate_cache()
- return
- end
-
- intermediate_cache_count = f:read("*l")
- while true do
- local l1 = f:read("*l")
- local l2 = f:read("*l")
-
- if (l1 == nil) or (l2 == nil) then
- break
- end
-
- intermediate_cache[l1] = l2
- end
-
- f:close()
- end
- local function create_intermediate_cache_key(key)
- local u = intermediate_cache[key]
- if not u then
- intermediate_cache_count = intermediate_cache_count + 1
- u = intermediate_cache_count
- intermediate_cache[key] = u
- save_intermediate_cache()
- end
-
- return u
- end
- -- ======================================================================= --
- -- STRING MODIFIERS --
- -- ======================================================================= --
- function stringmodifier.dirname(self, s)
- if (type(s) == "table") then
- if (table_getn(s) == 1) then
- s = s[1]
- else
- self:__error("tried to use string modifier 'dirname' on a table with more than one entry")
- end
- end
-
- return dirname(s)
- end
- -- ======================================================================= --
- -- CLASS SYSTEM --
- -- ======================================================================= --
- --- Base class --------------------------------------------------------------
- local metaclass = {
- class = "metaclass",
-
- -- Creates a new instance of a class by creating a new object and cloning
- -- all properties of the called class onto it.
-
- __call = function (self, ...)
- local o = {}
- for i, j in pairs(self) do
- o[i] = j
- end
- setmetatable(o, o)
-
- -- Determine where this object was defined.
-
- local i = 1
- while true do
- local s = debug.getinfo(i, "Sl")
- if s then
- if (string_byte(s.source) == 64) then
- o.definedat = string_sub(s.source, 2)..":"..s.currentline
- end
- else
- break
- end
- i = i + 1
- end
- -- Call the object's constructor and return it.
-
- o:__init(unpack(arg))
- return o
- end,
-
- -- Dummy constructor.
-
- __init = function (self, ...)
- end,
- }
- setmetatable(metaclass, metaclass)
- --- Top-level build node ----------------------------------------------------
- local node = metaclass()
- node.class = "node"
- -- When constructed, nodes initialise themselves from a supplied table of
- -- properties. All node children take exactly one argument, allowing the
- -- "constructor {properties}" construction pattern.
- function node:__init(t)
- metaclass.__init(self)
-
- if (type(t) == "string") then
- t = {t}
- end
- if (type(t) ~= "table") then
- self:__error("can't be constructed with a ", type(t), "; try a table or a string")
- end
-
- -- Copy over non-numeric parameters.
-
- for i, j in pairs(t) do
- if (tonumber(i) == nil) then
- self[i] = j
- end
- end
- -- Copy any numeric parameters.
-
- for _, i in ipairs(t) do
- table_insert(self, i)
- end
-
- -- If we're a class, don't verify.
-
- if t.class then
- return
- end
- -- ensure_n_children
- -- When true, ensures that the node has exactly the number of children
- -- specified.
-
- if self.ensure_n_children then
- local n = self.ensure_n_children
- if (table.getn(self) ~= n) then
- local one
- if (n == 1) then
- one = "one child"
- else
- one = n.." children"
- end
- self:_error("must have exactly ", one)
- end
- end
-
- -- ensure_at_least_one_child
- -- When true, ensures the the node has at least one child.
-
- if self.ensure_at_least_one_child then
- if (table_getn(self) < 1) then
- self:__error("must have at least one child")
- end
- end
-
- -- construct_string_children_with
- -- If set, any string children are automatically converted using the
- -- specified constructor.
-
- if self.construct_string_children_with then
- local constructor = self.construct_string_children_with
- for i, j in ipairs(self) do
- if (type(j) == "string") then
- self[i] = constructor {j}
- end
- end
- end
-
- -- all_children_are_objects
- -- When true, verifies that all children are objects and not something
- -- else (such as strings).
- if self.all_children_are_objects then
- for i, j in ipairs(self) do
- if (type(j) ~= "table") then
- self:__error("doesn't know what to do with child ", i,
- ", which is a ", type(j))
- end
- end
- end
-
- -- Ensure that self.install is valid.
-
- if self.install then
- local t = type(self.install)
- if (t == "string") or
- (t == "function") then
- self.install = {self.install}
- end
-
- if (type(self.install) ~= "table") then
- self:__error("doesn't know what to do with its installation command, ",
- "which is a ", type(self.install), " but should be a table, function ",
- "or string")
- end
- end
- end
- -- If an attempt is made to access a variable on a node that doesn't exist,
- -- and the variable starts with a capital letter, it's looked up in the
- -- property scope.
- function node:__index(key)
- local i = string_byte(key, 1)
- if (i >= 65) and (i <= 90) then
- -- Scan up the class hierarchy.
-
- local recurse
- recurse = function(s, key)
- if not s then
- return nil
- end
- local o = rawget(s.object, key)
- if o then
- if (type(o) == "table") then
- -- Handle lists of the form {PARENT, "foo", "bar"...}
- if (o[1] == PARENT) then
- local parent = recurse(s.next, key)
- local newo = {}
-
- if parent then
- if (type(parent) ~= "table") then
- parent = {parent}
- end
- for _, j in ipairs(parent) do
- table_insert(newo, j)
- end
- end
- for _, j in ipairs(o) do
- if (j ~= PARENT) then
- table_insert(newo, j)
- end
- end
- return newo
- -- Handle lists of the form {REDIRECT, "newkey"}
- elseif (o[1] == REDIRECT) then
- return self:__index(o[2])
- end
- end
- return o
- end
- -- Tail recursion.
- return recurse(s.next, key)
- end
-
- -- We want this node looked at first, so fake up a scope entry for it.
- local fakescope = {
- next = scope,
- object = self
- }
-
- -- Tail recursion.
- return recurse(fakescope, key)
- end
-
- -- For local properties, just return what's here.
- return rawget(self, key)
- end
- -- Little utility that emits an error message.
- function node:__error(...)
- usererror("object '", self.class, "', defined at ",
- self.definedat, ", ", unpack(arg))
- end
- -- Causes a node to return its outputs; that is, the files the node will
- -- produce when built. The parameter contains a list of input filenames; the
- -- outputs of the node's children.
- function node:__outputs(inputs)
- self:__error("didn't implement __outputs when it should have")
- end
- -- Causes a node to return its dependencies; that is, a list of *filenames*
- -- whose timestamps need to be considered when checking whether a node needs
- -- to be rebuilt. This is usually, but not always, the same as the inputs.
- function node:__dependencies(inputs, outputs)
- return inputs
- end
- -- Returns the node's timestamp. It will only get built if this is older than its
- -- children's timestamps.
- function node:__timestamp(inputs, outputs)
- local t = 0
- for _, i in ipairs(outputs) do
- local tt = filetime(i)
- if (tt > t) then
- t = tt
- end
- end
- return t
- end
- -- Unconditionally builds the nodes' children, collating their outputs. We
- -- push a new scope while we do so, to make this object's definitions visible
- -- to the children. (Almost never overridden. Only file() will want to do
- -- this, most likely.)
- function node:__buildchildren()
- local inputs = {}
- scope = {object=self, next=scope}
-
- for _, i in ipairs(self) do
- table_append(inputs, i:__build())
- end
- self:__buildadditionalchildren()
- scope = scope.next
- return inputs
- end
- -- Provides a hook for building any additional children that aren't actually
- -- in the child list.
- function node:__buildadditionalchildren()
- end
- -- Cause the node's children to be built, collating their outputs, and if
- -- any output is newer than the node itself, causes the node to be built.
- function node:__build()
- -- Build children and collate their outputs. These will become this node's
- -- inputs.
-
- local inputs = self:__buildchildren()
-
- -- Determine the node's outputs. This will usually be automatically
- -- generated, in which case the name will depend on the overall environment ---
- -- including the inputs.
-
- local outputs = self:__outputs(inputs)
-
- -- Get the current node's timestamp. If anything this node depends on is
- -- newer than that, the node needs rebuilding.
-
- local t = self:__timestamp(inputs, outputs)
- local depends = self:__dependencies(inputs, outputs)
- local rebuild = false
- if (t == 0) then
- rebuild = true
- end
-
- if (not rebuild and depends) then
- for _, i in ipairs(depends) do
- local tt = filetime(i)
- -- message("comparing ", t, " with ", tt, " (", rendertable({i}), ")")
- if (tt > t) then
- if verbose then
- message("rebuilding ", self.class, " because ", i, " newer than ", rendertable(outputs))
- end
- rebuild = true
- break
- end
- end
- end
- if rebuild then
- self:__dobuild(inputs, outputs)
- filetouch(outputs)
- end
-
- -- If an installation command was specified, execute it now.
-
- if self.install then
- self:__invoke(self.install, inputs, outputs)
- end
-
- -- And return this nodes' outputs.
-
- return outputs
- end
- -- Builds this node from the specified input files (the node's childrens'
- -- outputs).
- function node:__dobuild(inputs, outputs)
- self:__error("didn't implement __dobuild when it should have")
- end
- -- Recursively expands any variables in a string.
- function node:__expand(s)
- local searching = true
- while searching do
- searching = false
-
- -- Expand %{expressions}%
-
- s = string_gsub(s, "%%{(.-)}%%", function (expr)
- searching = true
- local f, e = loadstring(expr, "expression")
- if not f then
- self:__error("couldn't compile the expression '", expr, "': ", e)
- end
-
- local env = {self=self}
- setmetatable(env, {
- __index = function(_, key)
- return sandbox[key]
- end
- })
- setfenv(f, env)
-
- f, e = pcall(f, self)
- if not f then
- self:__error("couldn't evaluate the expression '", expr, "': ", e)
- end
- return rendertable(e)
- end)
- -- Expand %varnames%
-
- s = string_gsub(s, "%%(.-)%%", function (varname)
- searching = true
- -- Parse the string reference.
-
- local _, _, leftcolon, rightcolon = string_find(varname, "([^:]*):?(.*)$")
- local _, _, varname, selectfrom, hyphen, selectto = string_find(leftcolon, "^([^[]*)%[?([^-%]]*)(%-?)([^%]]*)]?$")
-
- -- Get the basic value that the rest of the reference is going to
- -- depend on.
-
- local result = self:__index(varname)
- if not result then
- self:__error("doesn't understand variable '", varname, "'")
- end
- -- Process any selector, if specified.
-
- if (selectfrom ~= "") or (hyphen ~= "") or (selectto ~= "") then
- if (type(result) ~= "table") then
- self:__error("tried to use a [] selector on variable '", varname,
- "', which doesn't contain a table")
- end
- local n = table_getn(result)
- selectfrom = tonumber(selectfrom)
- selectto = tonumber(selectto)
-
- if (hyphen ~= "") then
- if not selectfrom then
- selectfrom = 1
- end
- if not selectto then
- selectto = n
- end
- else
- if not selectto then
- selectto = selectfrom
- end
- if not selectfrom then
- self:__error("tried to use an empty selector on variable '", varname, "'")
- end
- end
-
- if (selectfrom < 1) or (selectto < 1) or
- (selectfrom > n) or (selectto > n) or
- (selectto < selectfrom) then
- self:__error("tried to use an invalid selector [",
- selectfrom, "-", selectto, "] on variable '", varname,
- "'; only [1-", n, "] is valid")
- end
-
- local newresult = {}
- for i = selectfrom, selectto do
- table_insert(newresult, result[i])
- end
- result = newresult
- end
-
- -- Process any string modifier, if supplied.
-
- if (rightcolon ~= "") then
- local f = stringmodifier[rightcolon]
- if not f then
- self:__error("tried to use an unknown string modifier '",
- rightcolon, "' on variable '", varname, "'")
- end
-
- result = f(self, result)
- end
-
- return rendertable(result)
- end)
- end
-
- -- Any remaining %% sequences must be empty, and so convert them into
- -- single % sequences.
-
- s = string_gsub(s, "%%%%", "%")
- return s
- end
- -- Expands any variables in a command table, and executes it.
- function node:__invoke(command, inputs, outputs)
- if (type(command) ~= "table") then
- command = {command}
- end
-
- for _, s in ipairs(command) do
- if (type(s) == "string") then
- s = self:__expand(s)
- if not quiet then
- traceoutput(s)
- end
- if not no_execute then
- local r = os.execute(s)
- if (r ~= 0) then
- return r
- end
- end
- elseif (type(s) == "function") then
- local r = s(self, inputs, outputs)
- if r then
- return r
- end
- end
- end
- return false
- end
- -- ======================================================================= --
- -- PROLOGUE --
- -- ======================================================================= --
- -- The prologue contains the standard library that all pmfiles can refer to.
- -- For simplicity, it's implemented by code running inside the sandbox,
- -- which means that it's basically identical to user code (and could, in
- -- fact, be kept in a seperate file).
- -- Here we set up the sandbox.
- table_merge(sandbox, {
- VERSION = VERSION,
-
- assert = assert,
- collectgarbage = collectgarbage,
- dofile = dofile,
- error = error,
- getfenv = getfenv,
- getmetatable = getmetatable,
- gcinfo = gcinfo,
- ipairs = ipairs,
- loadfile = loadfile,
- loadlib = loadlib,
- loadstring = loadstring,
- next = next,
- pairs = pairs,
- pcall = pcall,
- print = print,
- rawequal = rawequal,
- rawget = rawget,
- rawset = rawset,
- require = require,
- setfenv = setfenv,
- setmetatable = setmetatable,
- tonumber = tonumber,
- tostring = tostring,
- type = type,
- unpack = unpack,
- _VERSION = _VERSION,
- xpcall = xpcall,
-
- table = table,
- io = io,
- os = os,
- posix = posix,
- string = string,
- debug = debug,
- loadlib = loadlib,
-
- pm = _G,
- node = node,
-
- PARENT = PARENT,
- EMPTY = EMPTY,
- REDIRECT = REDIRECT,
- })
- -- Cause any reads from undefined keys in the sandbox to fail with an error.
- -- This helps debugging pmfiles somewhat.
- setmetatable(sandbox, {
- __index = function(self, key)
- local value = rawget(self, key)
- if (value == nil) then
- error(key.." could not be found in any applicable scope")
- end
- return value
- end
- })
- -- Switch into sandbox mode.
- setfenv(1, sandbox)
- --- Assorted utilities ------------------------------------------------------
- -- Includes a file.
- function include(f, ...)
- local c, e = loadfile(f)
- if not c then
- usererror("script compilation error: ", e)
- end
-
- setfenv(c, sandbox)
- local arguments = arg
- xpcall(
- function()
- c(unpack(arguments))
- end,
- function(e)
- message("script execution error --- traceback follows:")
- traceback(e)
- end
- )
- end
- --- file --------------------------------------------------------------------
- -- file() is pretty much the simplest clause. It takes a list of filenames,
- -- and outputs them.
- --
- -- * Building does nothing.
- -- * Its outputs are its inputs.
- --
- -- Note: this clause only takes *strings* as its children. If a reference is
- -- made to a file that doesn't exist, an error occurs.
- file = node {
- class = "file",
- ensure_at_least_one_child = true,
-
- __init = function(self, p)
- node.__init(self, p)
-
- -- If we're a class, don't verify.
-
- if ((type(p) == "table") and p.class) then
- return
- end
-
- -- Ensure that the file's children are strings.
-
- for i, j in ipairs(self) do
- if (type(j) ~= "string") then
- self:__error("doesn't know what to do with child ", i,
- ", which is a ", type(j))
- end
- end
- end,
- -- File's timestamp is special and will bail if it meets a nonexistant file.
-
- __timestamp = function(self, inputs, outputs)
- local t = 0
- for _, i in ipairs(outputs) do
- i = self:__expand(i)
- local tt = filetime(i)
- if (tt == 0) then
- self:__error("is referring to the file '", i, "' which does not exist")
- end
- if (tt > t) then
- t = tt
- end
- end
- return t
- end,
- -- Outputs are inputs.
-
- __outputs = function(self, inputs)
- local o = {}
- local n
- if self.only_n_children_are_outputs then
- n = self.only_n_children_are_outputs
- else
- n = table_getn(inputs)
- end
-
- for i = 1, n do
- o[i] = inputs[i]
- end
-
- return o
- end,
-
- -- Building children does nothing; outputs are inputs.
-
- __buildchildren = function(self)
- local outputs = {}
- table_append(outputs, self)
- return outputs
- end,
-
- -- Building does nothing.
-
- __dobuild = function(self, inputs, outputs)
- end,
- }
-
- --- group -------------------------------------------------------------------
- -- group() is also the simplest clause. It does nothing, existing only to
- -- group together its children.
- group = node {
- class = "group",
-
- -- Outputs are inputs.
-
- __outputs = function(self, inputs)
- return inputs
- end,
-
- -- Building does nothing.
-
- __dobuild = function(self, inputs, outputs)
- end,
- }
- --- deponly -----------------------------------------------------------------
- -- deponly() is the one-and-a-halfth most simplest clause. It acts like
- -- group {}, but returns no outputs. It's useful for ensuring that building
- -- one node causes another node to be built without actually using the
- -- second node's outputs.
- deponly = node {
- class = "deponly",
- ensure_at_least_one_child = true,
-
- -- Emits no outputs
-
- __outputs = function(self, inputs)
- return {}
- end,
-
- -- Building does nothing.
-
- __dobuild = function(self, inputs, outputs)
- end,
- }
- --- ith ---------------------------------------------------------------------
- -- ith() is the second simplest clause. It acts like group {}, but returns
- -- only some of the specified output. It is suitable for extracting, say,
- -- one output from a clause to pass to cfile {}.
- ith = node {
- class = "ith",
- ensure_at_least_one_child = true,
-
- __init = function(self, p)
- node.__init(self, p)
-
- -- If we're a class, don't verify.
-
- if ((type(p) == "table") and p.class) then
- return
- end
- -- If we have an i property, ensure we don't have a from or
- -- to property.
-
- if self.i then
- if self.from or self.to then
- self:__error("can't have both an i property and a from or to property")
- end
-
- if (type(self.i) ~= "number") then
- self:__error("doesn't know what to do with its i property, ",
- "which is a ", type(self.i), " where a number was expected")
- end
- self.from = self.i
- self.to = self.i
- end
-
- -- Ensure the from and to properties are numbers, if they exist.
-
- if self.from then
- if (type(self.from) ~= "number") then
- self:__error("doesn't know what to do with its from property, ",
- "which is a ", type(self.from), " where a number was expected")
- end
- end
-
- if self.to then
- if (type(self.to) ~= "number") then
- self:__error("doesn't know what to do with its to property, ",
- "which is a ", type(self.to), " where a number was expected")
- end
- end
- end,
- -- Emits one output, which is one of the inputs.
-
- __outputs = function(self, inputs)
- local n = table_getn(inputs)
- local from = self.from or 1
- local to = self.to or n
-
- if (from < 1) or (to > n) then
- self:__error("tried to select range ", from, " to ", to,
- " from only ", n, " inputs")
- end
-
- local range = {}
- for i = from, to do
- table_append(range, inputs[i])
- end
- return range
- end,
-
- -- Building does nothing.
-
- __dobuild = function(self, inputs, outputs)
- end,
- }
- --- foreach -----------------------------------------------------------------
- -- foreach {} is the counterpart to ith {}. It applies a particular rule to
- -- all of its children.
- foreach = node {
- class = "foreach",
-
- __init = function(self, p)
- node.__init(self, p)
-
- -- If we're a class, don't verify.
-
- if ((type(p) == "table") and p.class) then
- return
- end
- -- Ensure we have a rule property which is a table.
-
- if not self.rule then
- self:__error("must have a rule property")
- end
- if (type(self.rule) ~= "table") then
- self:__error("doesn't know what to do with its rule property, ",
- "which is a ", type(self.rule), " where a table was expected")
- end
- end,
- -- Build all our children via the rule.
- --
- -- This is pretty much a copy of node.__buildchildren().
-
- __buildchildren = function(self)
- scope = {object=self, next=scope}
- local intermediate = {}
- for _, i in ipairs(self) do
- table_append(intermediate, i:__build())
- end
-
- local inputs = {}
- for _, i in ipairs(intermediate) do
- local r = self.rule { i }
- table_append(inputs, r:__build())
- end
-
- self:__buildadditionalchildren()
- scope = scope.next
- return inputs
- end,
- -- Inputs are outputs --- because __buildchildren has already done the
- -- necessary work.
-
- __outputs = function(self, inputs)
- return inputs
- end,
- -- Building does nothing.
-
- __dobuild = function(self, inputs, outputs)
- end,
- }
- --- Simple ---------------------------------------------------------------
- -- simple is the most common clause, and implements make-like behaviour:
- -- the named command is executed in order to rebuild the node.
- --
- -- * The timestamp is the newest timestamp of its outputs.
- -- * Building executes the command.
- -- * Its outputs are automatically generated by expanding the templates
- -- in the 'outputs' variable.
- simple = node {
- class = "file",
- construct_string_children_with = file,
- all_children_are_objects = true,
-
- __init = function(self, p)
- node.__init(self, p)
-
- -- If we're a class, don't verify.
-
- if ((type(p) == "table") and p.class) then
- return
- end
- -- outputs must exist, and must be a table.
-
- if not self.outputs then
- self:__error("must have an outputs template set")
- end
-
- if (type(self.outputs) ~= "table") then
- self:__error("doesn't know what to do with its outputs, which is a ",
- type(self.outputs), " but should be a table")
- end
-
- -- There must be a command which must be a string or table.
-
- if not self.command then
- self:__error("must have a command specified")
- end
- if (type(self.command) == "string") then
- self.command = {self.command}
- end
- if (type(self.command) ~= "table") then
- self:__error("doesn't know what to do with its command, which is a ",
- type(self.command), " but should be a string or a table")
- end
- end,
-
- -- Outputs are specified manually.
-
- __outputs = function(self, inputs)
- self["in"] = inputs
-
- local input
- if inputs then
- input = inputs[1]
- end
- if not input then
- input = ""
- end
-
- self.I = string_gsub(input, "^.*/", "")
- self.I = string_gsub(self.I, "%..-$", "")
- -- Construct an outputs array for use in the cache key. This mirrors
- -- what the final array will be, but the unique ID is going to be 0.
-
- self.out = {}
- self.U = 0
- for _, i in ipairs(self.outputs) do
- i = self:__expand(i)
- table_append(self.out, i)
- end
-
- -- Determine the cache key we're going to use.
-
- local cachekey = table_concat(self.command, " && ")
- cachekey = self:__expand(cachekey)
- cachekey = create_intermediate_cache_key(cachekey)
- -- Work out the unique ID.
- --
- -- Note: we're running in the sandbox, so we need to fully qualify
- -- pm.intermediate_cache_dir.
-
- self.U = pm.intermediate_cache_dir..cachekey
-
- -- Construct the real outputs array.
-
- self.out = {}
- for _, i in ipairs(self.outputs) do
- i = self:__expand(i)
- mkcontainerdir(i)
- table_append(self.out, i)
- end
- return self.out
- end,
-
- -- Building causes the command to be expanded and invoked. The 'children'
- -- variable is set to the input files.
-
- __dobuild = function(self, inputs, outputs)
- self["in"] = inputs
- self.out = outputs
- local r = self:__invoke(self.command, inputs, outputs)
- if r then
- if delete_output_files_on_error then
- self:__invoke({"%RM% %out%"})
- end
- self:__error("failed to build with return code ", r)
- end
- end,
- }
- --- End of prologue ---------------------------------------------------------
- -- Set a few useful global variables.
- RM = "rm -f"
- INSTALL = "ln -f"
- -- Now we're done, switch out of sandbox mode again. This only works
- -- because we made _G local at the top of the file, which makes it
- -- lexically scoped rather than looked up via the environment.
- setfenv(1, _G)
- -- ======================================================================= --
- -- APPLICATION DRIVER --
- -- ======================================================================= --
- -- Parse and process the command line options.
- do
- local function do_help(opt)
- message("Prime Mover version ", VERSION, " © 2006 David Given")
- stdout:write([[
- Syntax: pm [<options...>] [<targets>]
- Options:
- -h --help Displays this message.
- --license List Prime Mover's redistribution license.
- -cX --cachedir X Sets the object file cache to directory X.
- -p --purge Purges the cache before execution.
- WARNING: will remove *everything* in the cache dir!
- -fX --file X Reads in the pmfile X. May be specified multiple times.
- -DX=Y --define X=Y Defines variable X to value Y (or true if Y omitted)
- -n --no-execute Don't actually execute anything
- -v --verbose Be more verbose
- -q --quiet Be more quiet
-
- If no pmfiles are explicitly specified, 'pmfile' is read.
- If no targets are explicitly specified, 'default' is built.
- Options and targets may be specified in any order.
- ]])
- os.exit(0)
- end
-
- local function do_license(opt)
- message("Prime Mover version ", VERSION, " © 2006 David Given")
- stdout:write([[
-
- Prime Mover is licensed under the MIT open source license.
- Copyright © 2006 David Given
- Permission is hereby granted, free of charge, to any person obtaining a copy
- of this software and associated documentation files (the "Software"), to deal
- in the Software without restriction, including without limitation the rights
- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- copies of the Software, and to permit persons to whom the Software is
- furnished to do so, subject to the following conditions:
- The above copyright notice and this permission notice shall be included in
- all copies or substantial portions of the Software.
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- SOFTWARE.
- ]])
- os.exit(0)
- end
- local function needarg(opt)
- if not opt then
- usererror("missing option parameter")
- end
- end
-
- local function do_cachedir(opt)
- needarg(opt)
- intermediate_cache_dir = opt
- return 1
- end
-
- local function do_inputfile(opt)
- needarg(opt)
- table_append(input_files, opt)
- return 1
- end
-
- local function do_purgecache(opt)
- purge_intermediate_cache = true
- return 0
- end
-
- local function do_define(opt)
- needarg(opt)
-
- local s, e, key, value = string_find(opt, "^([^=]*)=(.*)$")
- if not key then
- key = opt
- value = true
- end
-
- sandbox[key] = value
- return 1
- end
-
- local function do_no_execute(opt)
- no_execute = true
- return 0
- end
-
- local function do_verbose(opt)
- verbose = true
- return 0
- end
-
- local function do_quiet(opt)
- quiet = true
- return 0
- end
-
- local argmap = {
- ["h"] = do_help,
- ["help"] = do_help,
- ["c"] = do_cachedir,
- ["cachedir"] = do_cachedir,
- ["p"] = do_purgecache,
- ["purge"] = do_purgecache,
- ["f"] = do_inputfile,
- ["file"] = do_inputfile,
- ["D"] = do_define,
- ["define"] = do_define,
- ["n"] = do_no_execute,
- ["no-execute"] = do_no_execute,
- ["v"] = do_verbose,
- ["verbose"] = do_verbose,
- ["q"] = do_quiet,
- ["quiet"] = do_quiet,
- ["license"] = do_license,
- }
-
- -- Called on an unrecognised option.
-
- local function unrecognisedarg(arg)
- usererror("unrecognised option '", arg, "' --- try --help for help")
- end
-
- -- Do the actual argument parsing.
-
- for i = 1, table_getn(arg) do
- local o = arg[i]
- local op
-
- if (string_byte(o, 1) == 45) then
- -- This is an option.
- if (string_byte(o, 2) == 45) then
- -- ...with a -- prefix.
- o = string_sub(o, 3)
- local fn = argmap[o]
- if not fn then
- unrecognisedarg("--"..o)
- end
- local op = arg[i+1]
- i = i + fn(op)
- else
- -- ...without a -- prefix.
- local od = string_sub(o, 2, 2)
- local fn = argmap[od]
- if not fn then
- unrecognisedarg("-"..od)
- end
- op = string_sub(o, 3)
- if (op == "") then
- op = arg[i+1]
- i = i + fn(op)
- else
- fn(op)
- end
- end
- else
- -- This is a target name.
- table_append(targets, o)
- end
- end
-
- -- Option fallbacks.
-
- if (table_getn(input_files) == 0) then
- input_files = {"pmfile"}
- end
-
- if (table_getn(targets) == 0) then
- targets = {"default"}
- end
- end
- -- Load any input files.
- for _, i in ipairs(input_files) do
- sandbox.include(i, unpack(arg))
- end
- -- Set up the intermediate cache.
- if purge_intermediate_cache then
- create_intermediate_cache()
- else
- load_intermediate_cache()
- end
- -- Build any targets.
- for _, i in ipairs(targets) do
- local o = sandbox[i]
- if not o then
- usererror("don't know how to build '", i, "'")
- end
- if ((type(o) ~= "table") and not o.class) then
- usererror("'", i, "' doesn't seem to be a valid target")
- end
- xpcall(
- function()
- o:__build()
- end,
- function(e)
- message("rule engine execution error --- traceback follows:")
- traceback(e)
- end
- )
- end
- XXXXSTARTinterpreter
- 253583
- #include <signal.h>
- #include <sys/wait.h>
- #include <stdarg.h>
- #include <errno.h>
- #include <stdio.h>
- #include <locale.h>
- #include <sys/types.h>
- #include <sys/utsname.h>
- #include <stddef.h>
- #include <grp.h>
- #include <time.h>
- #include <assert.h>
- #include <sys/stat.h>
- #include <ctype.h>
- #include <utime.h>
- #include <setjmp.h>
- #include <dirent.h>
- #include <sys/times.h>
- #include <pwd.h>
- #include <unistd.h>
- #include <string.h>
- #include <limits.h>
- #include <fcntl.h>
- #include <stdlib.h>
- #line 1 "lapi.c"
- #define lapi_c
- #line 1 "lua.h"
- #ifndef lua_h
- #define lua_h
- #define LUA_NUMBER int
- #define LUA_NUMBER_SCAN "%d"
- #define LUA_NUMBER_FMT "%d"
- #define LUA_VERSION "Lua 5.0.2 (patched for Prime Mover)"
- #define LUA_COPYRIGHT "Copyright (C) 1994-2004 Tecgraf, PUC-Rio"
- #define LUA_AUTHORS "R. Ierusalimschy, L. H. de Figueiredo & W. Celes"
- #define LUA_MULTRET (-1)
- #define LUA_REGISTRYINDEX (-10000)
- #define LUA_GLOBALSINDEX (-10001)
- #define lua_upvalueindex(i) (LUA_GLOBALSINDEX-(i))
- #define LUA_ERRRUN 1
- #define LUA_ERRFILE 2
- #define LUA_ERRSYNTAX 3
- #define LUA_ERRMEM 4
- #define LUA_ERRERR 5
- typedef struct lua_State lua_State;typedef int(*lua_CFunction)(lua_State*L);
- typedef const char*(*lua_Chunkreader)(lua_State*L,void*ud,size_t*sz);typedef
- int(*lua_Chunkwriter)(lua_State*L,const void*p,size_t sz,void*ud);
- #define LUA_TNONE (-1)
- #define LUA_TNIL 0
- #define LUA_TBOOLEAN 1
- #define LUA_TLIGHTUSERDATA 2
- #define LUA_TNUMBER 3
- #define LUA_TSTRING 4
- #define LUA_TTABLE 5
- #define LUA_TFUNCTION 6
- #define LUA_TUSERDATA 7
- #define LUA_TTHREAD 8
- #define LUA_MINSTACK 20
- #ifdef LUA_USER_H
- #include LUA_USER_H
- #endif
- #ifndef LUA_NUMBER
- typedef double lua_Number;
- #else
- typedef LUA_NUMBER lua_Number;
- #endif
- #ifndef LUA_API
- #define LUA_API extern
- #endif
- LUA_API lua_State*lua_open(void);LUA_API void lua_close(lua_State*L);LUA_API
- lua_State*lua_newthread(lua_State*L);LUA_API lua_CFunction lua_atpanic(
- lua_State*L,lua_CFunction panicf);LUA_API int lua_gettop(lua_State*L);LUA_API
- void lua_settop(lua_State*L,int idx);LUA_API void lua_pushvalue(lua_State*L,
- int idx);LUA_API void lua_remove(lua_State*L,int idx);LUA_API void lua_insert(
- lua_State*L,int idx);LUA_API void lua_replace(lua_State*L,int idx);LUA_API int
- lua_checkstack(lua_State*L,int sz);LUA_API void lua_xmove(lua_State*from,
- lua_State*to,int n);LUA_API int lua_isnumber(lua_State*L,int idx);LUA_API int
- lua_isstring(lua_State*L,int idx);LUA_API int lua_iscfunction(lua_State*L,int
- idx);LUA_API int lua_isuserdata(lua_State*L,int idx);LUA_API int lua_type(
- lua_State*L,int idx);LUA_API const char*lua_typename(lua_State*L,int tp);
- LUA_API int lua_equal(lua_State*L,int idx1,int idx2);LUA_API int lua_rawequal(
- lua_State*L,int idx1,int idx2);LUA_API int lua_lessthan(lua_State*L,int idx1,
- int idx2);LUA_API lua_Number lua_tonumber(lua_State*L,int idx);LUA_API int
- lua_toboolean(lua_State*L,int idx);LUA_API const char*lua_tostring(lua_State*L
- ,int idx);LUA_API size_t lua_strlen(lua_State*L,int idx);LUA_API lua_CFunction
- lua_tocfunction(lua_State*L,int idx);LUA_API void*lua_touserdata(lua_State*L,
- int idx);LUA_API lua_State*lua_tothread(lua_State*L,int idx);LUA_API const
- void*lua_topointer(lua_State*L,int idx);LUA_API void lua_pushnil(lua_State*L);
- LUA_API void lua_pushnumber(lua_State*L,lua_Number n);LUA_API void
- lua_pushlstring(lua_State*L,const char*s,size_t l);LUA_API void lua_pushstring
- (lua_State*L,const char*s);LUA_API const char*lua_pushvfstring(lua_State*L,
- const char*fmt,va_list argp);LUA_API const char*lua_pushfstring(lua_State*L,
- const char*fmt,...);LUA_API void lua_pushcclosure(lua_State*L,lua_CFunction fn
- ,int n);LUA_API void lua_pushboolean(lua_State*L,int b);LUA_API void
- lua_pushlightuserdata(lua_State*L,void*p);LUA_API void lua_gettable(lua_State*
- L,int idx);LUA_API void lua_rawget(lua_State*L,int idx);LUA_API void
- lua_rawgeti(lua_State*L,int idx,int n);LUA_API void lua_newtable(lua_State*L);
- LUA_API void*lua_newuserdata(lua_State*L,size_t sz);LUA_API int
- lua_getmetatable(lua_State*L,int objindex);LUA_API void lua_getfenv(lua_State*
- L,int idx);LUA_API void lua_settable(lua_State*L,int idx);LUA_API void
- lua_rawset(lua_State*L,int idx);LUA_API void lua_rawseti(lua_State*L,int idx,
- int n);LUA_API int lua_setmetatable(lua_State*L,int objindex);LUA_API int
- lua_setfenv(lua_State*L,int idx);LUA_API void lua_call(lua_State*L,int nargs,
- int nresults);LUA_API int lua_pcall(lua_State*L,int nargs,int nresults,int
- errfunc);LUA_API int lua_cpcall(lua_State*L,lua_CFunction func,void*ud);
- LUA_API int lua_load(lua_State*L,lua_Chunkreader reader,void*dt,const char*
- chunkname);LUA_API int lua_dump(lua_State*L,lua_Chunkwriter writer,void*data);
- LUA_API int lua_yield(lua_State*L,int nresults);LUA_API int lua_resume(
- lua_State*L,int narg);LUA_API int lua_getgcthreshold(lua_State*L);LUA_API int
- lua_getgccount(lua_State*L);LUA_API void lua_setgcthreshold(lua_State*L,int
- newthreshold);LUA_API const char*lua_version(void);LUA_API int lua_error(
- lua_State*L);LUA_API int lua_next(lua_State*L,int idx);LUA_API void lua_concat
- (lua_State*L,int n);
- #define lua_boxpointer(L,u) (*(void**)(lua_newuserdata(L,sizeof(void*)))=(u))
- #define lua_unboxpointer(L,i) (*(void**)(lua_touserdata(L,i)))
- #define lua_pop(L,n) lua_settop(L,-(n)-1)
- #define lua_register(L,n,f) (lua_pushstring(L,n),lua_pushcfunction(L,f),\
- lua_settable(L,LUA_GLOBALSINDEX))
- #define lua_pushcfunction(L,f) lua_pushcclosure(L,f,0)
- #define lua_isfunction(L,n) (lua_type(L,n)==LUA_TFUNCTION)
- #define lua_istable(L,n) (lua_type(L,n)==LUA_TTABLE)
- #define lua_islightuserdata(L,n) (lua_type(L,n)==LUA_TLIGHTUSERDATA)
- #define lua_isnil(L,n) (lua_type(L,n)==LUA_TNIL)
- #define lua_isboolean(L,n) (lua_type(L,n)==LUA_TBOOLEAN)
- #define lua_isnone(L,n) (lua_type(L,n)==LUA_TNONE)
- #define lua_isnoneornil(L, n)(lua_type(L,n)<=0)
- #define lua_pushliteral(L, s)lua_pushlstring(L,""s,(sizeof(s)/sizeof(char))-1)
- LUA_API int lua_pushupvalues(lua_State*L);
- #define lua_getregistry(L) lua_pushvalue(L,LUA_REGISTRYINDEX)
- #define lua_setglobal(L,s) (lua_pushstring(L,s),lua_insert(L,-2),lua_settable(\
- L,LUA_GLOBALSINDEX))
- #define lua_getglobal(L,s) (lua_pushstring(L,s),lua_gettable(L,\
- LUA_GLOBALSINDEX))
- #define LUA_NOREF (-2)
- #define LUA_REFNIL (-1)
- #define lua_ref(L,lock) ((lock)?luaL_ref(L,LUA_REGISTRYINDEX):(lua_pushstring(\
- L,"unlocked references are obsolete"),lua_error(L),0))
- #define lua_unref(L,ref) luaL_unref(L,LUA_REGISTRYINDEX,(ref))
- #define lua_getref(L,ref) lua_rawgeti(L,LUA_REGISTRYINDEX,ref)
- #ifndef LUA_NUMBER_SCAN
- #define LUA_NUMBER_SCAN "%lf"
- #endif
- #ifndef LUA_NUMBER_FMT
- #define LUA_NUMBER_FMT "%.14g"
- #endif
- #define LUA_HOOKCALL 0
- #define LUA_HOOKRET 1
- #define LUA_HOOKLINE 2
- #define LUA_HOOKCOUNT 3
- #define LUA_HOOKTAILRET 4
- #define LUA_MASKCALL (1<<LUA_HOOKCALL)
- #define LUA_MASKRET (1<<LUA_HOOKRET)
- #define LUA_MASKLINE (1<<LUA_HOOKLINE)
- #define LUA_MASKCOUNT (1<<LUA_HOOKCOUNT)
- typedef struct lua_Debug lua_Debug;typedef void(*lua_Hook)(lua_State*L,
- lua_Debug*ar);LUA_API int lua_getstack(lua_State*L,int level,lua_Debug*ar);
- LUA_API int lua_getinfo(lua_State*L,const char*what,lua_Debug*ar);LUA_API
- const char*lua_getlocal(lua_State*L,const lua_Debug*ar,int n);LUA_API const
- char*lua_setlocal(lua_State*L,const lua_Debug*ar,int n);LUA_API const char*
- lua_getupvalue(lua_State*L,int funcindex,int n);LUA_API const char*
- lua_setupvalue(lua_State*L,int funcindex,int n);LUA_API int lua_sethook(
- lua_State*L,lua_Hook func,int mask,int count);LUA_API lua_Hook lua_gethook(
- lua_State*L);LUA_API int lua_gethookmask(lua_State*L);LUA_API int
- lua_gethookcount(lua_State*L);
- #define LUA_IDSIZE 60
- struct lua_Debug{int event;const char*name;const char*namewhat;const char*what
- ;const char*source;int currentline;int nups;int linedefined;char short_src[
- LUA_IDSIZE];int i_ci;};
- #endif
- #line 14 "lapi.c"
- #line 1 "lapi.h"
- #ifndef lapi_h
- #define lapi_h
- #line 1 "lobject.h"
- #ifndef lobject_h
- #define lobject_h
- #line 1 "llimits.h"
- #ifndef llimits_h
- #define llimits_h
- #ifndef BITS_INT
- #if INT_MAX-20 <32760
- #define BITS_INT 16
- #else
- #if INT_MAX >2147483640L
- #define BITS_INT 32
- #else
- #error "you must define BITS_INT with number of bits in an integer"
- #endif
- #endif
- #endif
- typedef unsigned int lu_hash;typedef int ls_hash;typedef unsigned long lu_mem;
- #define MAX_LUMEM ULONG_MAX
- typedef long ls_nstr;typedef unsigned char lu_byte;
- #define MAX_SIZET ((size_t)(~(size_t)0)-2)
- #define MAX_INT (INT_MAX-2)
- #define IntPoint(p) ((lu_hash)(p))
- #ifndef LUSER_ALIGNMENT_T
- typedef union{double u;void*s;long l;}L_Umaxalign;
- #else
- typedef LUSER_ALIGNMENT_T L_Umaxalign;
- #endif
- #ifndef LUA_UACNUMBER
- typedef double l_uacNumber;
- #else
- typedef LUA_UACNUMBER l_uacNumber;
- #endif
- #ifndef lua_assert
- #define lua_assert(c)
- #endif
- #ifndef check_exp
- #define check_exp(c,e) (e)
- #endif
- #ifndef UNUSED
- #define UNUSED(x) ((void)(x))
- #endif
- #ifndef cast
- #define cast(t, exp)((t)(exp))
- #endif
- typedef unsigned long Instruction;
- #ifndef LUA_MAXCALLS
- #define LUA_MAXCALLS 4096
- #endif
- #ifndef LUA_MAXCCALLS
- #define LUA_MAXCCALLS 200
- #endif
- #ifndef LUA_MAXCSTACK
- #define LUA_MAXCSTACK 2048
- #endif
- #define MAXSTACK 250
- #ifndef MAXVARS
- #define MAXVARS 200
- #endif
- #ifndef MAXUPVALUES
- #define MAXUPVALUES 32
- #endif
- #ifndef MAXPARAMS
- #define MAXPARAMS 100
- #endif
- #ifndef MINSTRTABSIZE
- #define MINSTRTABSIZE 32
- #endif
- #ifndef LUA_MINBUFFER
- #define LUA_MINBUFFER 32
- #endif
- #ifndef LUA_MAXPARSERLEVEL
- #define LUA_MAXPARSERLEVEL 200
- #endif
- #endif
- #line 12 "lobject.h"
- #define NUM_TAGS LUA_TTHREAD
- #define LUA_TPROTO (NUM_TAGS+1)
- #define LUA_TUPVAL (NUM_TAGS+2)
- typedef union GCObject GCObject;
- #define CommonHeader GCObject*next;lu_byte tt;lu_byte marked
- typedef struct GCheader{CommonHeader;}GCheader;typedef union{GCObject*gc;void*
- p;lua_Number n;int b;}Value;typedef struct lua_TObject{int tt;Value value;}
- TObject;
- #define ttisnil(o) (ttype(o)==LUA_TNIL)
- #define ttisnumber(o) (ttype(o)==LUA_TNUMBER)
- #define ttisstring(o) (ttype(o)==LUA_TSTRING)
- #define ttistable(o) (ttype(o)==LUA_TTABLE)
- #define ttisfunction(o) (ttype(o)==LUA_TFUNCTION)
- #define ttisboolean(o) (ttype(o)==LUA_TBOOLEAN)
- #define ttisuserdata(o) (ttype(o)==LUA_TUSERDATA)
- #define ttisthread(o) (ttype(o)==LUA_TTHREAD)
- #define ttislightuserdata(o) (ttype(o)==LUA_TLIGHTUSERDATA)
- #define ttype(o) ((o)->tt)
- #define gcvalue(o) check_exp(iscollectable(o),(o)->value.gc)
- #define pvalue(o) check_exp(ttislightuserdata(o),(o)->value.p)
- #define nvalue(o) check_exp(ttisnumber(o),(o)->value.n)
- #define tsvalue(o) check_exp(ttisstring(o),&(o)->value.gc->ts)
- #define uvalue(o) check_exp(ttisuserdata(o),&(o)->value.gc->u)
- #define clvalue(o) check_exp(ttisfunction(o),&(o)->value.gc->cl)
- #define hvalue(o) check_exp(ttistable(o),&(o)->value.gc->h)
- #define bvalue(o) check_exp(ttisboolean(o),(o)->value.b)
- #define thvalue(o) check_exp(ttisthread(o),&(o)->value.gc->th)
- #define l_isfalse(o) (ttisnil(o)||(ttisboolean(o)&&bvalue(o)==0))
- #define setnvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TNUMBER;i_o->value.n=(\
- x);}
- #define chgnvalue(obj,x) check_exp(ttype(obj)==LUA_TNUMBER,(obj)->value.n=(x))
- #define setpvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TLIGHTUSERDATA;i_o->\
- value.p=(x);}
- #define setbvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TBOOLEAN;i_o->value.b=\
- (x);}
- #define setsvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TSTRING;i_o->value.gc=\
- cast(GCObject*,(x));lua_assert(i_o->value.gc->gch.tt==LUA_TSTRING);}
- #define setuvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TUSERDATA;i_o->value.\
- gc=cast(GCObject*,(x));lua_assert(i_o->value.gc->gch.tt==LUA_TUSERDATA);}
- #define setthvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TTHREAD;i_o->value.gc\
- =cast(GCObject*,(x));lua_assert(i_o->value.gc->gch.tt==LUA_TTHREAD);}
- #define setclvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TFUNCTION;i_o->value.\
- gc=cast(GCObject*,(x));lua_assert(i_o->value.gc->gch.tt==LUA_TFUNCTION);}
- #define sethvalue(obj,x) {TObject*i_o=(obj);i_o->tt=LUA_TTABLE;i_o->value.gc=\
- cast(GCObject*,(x));lua_assert(i_o->value.gc->gch.tt==LUA_TTABLE);}
- #define setnilvalue(obj) ((obj)->tt=LUA_TNIL)
- #define checkconsistency(obj) lua_assert(!iscollectable(obj)||(ttype(obj)==(\
- obj)->value.gc->gch.tt))
- #define setobj(obj1,obj2) {const TObject*o2=(obj2);TObject*o1=(obj1);\
- checkconsistency(o2);o1->tt=o2->tt;o1->value=o2->value;}
- #define setobjs2s setobj
- #define setobj2s setobj
- #define setsvalue2s setsvalue
- #define setobjt2t setobj
- #define setobj2t setobj
- #define setobj2n setobj
- #define setsvalue2n setsvalue
- #define setttype(obj, tt)(ttype(obj)=(tt))
- #define iscollectable(o) (ttype(o)>=LUA_TSTRING)
- typedef TObject*StkId;typedef union TString{L_Umaxalign dummy;struct{
- CommonHeader;lu_byte reserved;lu_hash hash;size_t len;}tsv;}TString;
- #define getstr(ts) cast(const char*,(ts)+1)
- #define svalue(o) getstr(tsvalue(o))
- typedef union Udata{L_Umaxalign dummy;struct{CommonHeader;struct Table*
- metatable;size_t len;}uv;}Udata;typedef struct Proto{CommonHeader;TObject*k;
- Instruction*code;struct Proto**p;int*lineinfo;struct LocVar*locvars;TString**
- upvalues;TString*source;int sizeupvalues;int sizek;int sizecode;int
- sizelineinfo;int sizep;int sizelocvars;int lineDefined;GCObject*gclist;lu_byte
- nups;lu_byte numparams;lu_byte is_vararg;lu_byte maxstacksize;}Proto;typedef
- struct LocVar{TString*varname;int startpc;int endpc;}LocVar;typedef struct
- UpVal{CommonHeader;TObject*v;TObject value;}UpVal;
- #define ClosureHeader CommonHeader;lu_byte isC;lu_byte nupvalues;GCObject*\
- gclist
- typedef struct CClosure{ClosureHeader;lua_CFunction f;TObject upvalue[1];}
- CClosure;typedef struct LClosure{ClosureHeader;struct Proto*p;TObject g;UpVal*
- upvals[1];}LClosure;typedef union Closure{CClosure c;LClosure l;}Closure;
- #define iscfunction(o) (ttype(o)==LUA_TFUNCTION&&clvalue(o)->c.isC)
- #define isLfunction(o) (ttype(o)==LUA_TFUNCTION&&!clvalue(o)->c.isC)
- typedef struct Node{TObject i_key;TObject i_val;struct Node*next;}Node;typedef
- struct Table{CommonHeader;lu_byte flags;lu_byte lsizenode;struct Table*
- metatable;TObject*array;Node*node;Node*firstfree;GCObject*gclist;int sizearray
- ;}Table;
- #define lmod(s,size) check_exp((size&(size-1))==0,(cast(int,(s)&((size)-1))))
- #define twoto(x) (1<<(x))
- #define sizenode(t) (twoto((t)->lsizenode))
- extern const TObject luaO_nilobject;int luaO_log2(unsigned int x);int
- luaO_int2fb(unsigned int x);
- #define fb2int(x) (((x)&7)<<((x)>>3))
- int luaO_rawequalObj(const TObject*t1,const TObject*t2);int luaO_str2d(const
- char*s,lua_Number*result);const char*luaO_pushvfstring(lua_State*L,const char*
- fmt,va_list argp);const char*luaO_pushfstring(lua_State*L,const char*fmt,...);
- void luaO_chunkid(char*out,const char*source,int len);
- #endif
- #line 12 "lapi.h"
- void luaA_pushobject(lua_State*L,const TObject*o);
- #endif
- #line 16 "lapi.c"
- #line 1 "ldebug.h"
- #ifndef ldebug_h
- #define ldebug_h
- #line 1 "lstate.h"
- #ifndef lstate_h
- #define lstate_h
- #line 1 "ltm.h"
- #ifndef ltm_h
- #define ltm_h
- typedef enum{TM_INDEX,TM_NEWINDEX,TM_GC,TM_MODE,TM_EQ,TM_ADD,TM_SUB,TM_MUL,
- TM_DIV,TM_POW,TM_UNM,TM_LT,TM_LE,TM_CONCAT,TM_CALL,TM_N}TMS;
- #define gfasttm(g,et,e) (((et)->flags&(1u<<(e)))?NULL:luaT_gettm(et,e,(g)->\
- tmname[e]))
- #define fasttm(l,et,e) gfasttm(G(l),et,e)
- const TObject*luaT_gettm(Table*events,TMS event,TString*ename);const TObject*
- luaT_gettmbyobj(lua_State*L,const TObject*o,TMS event);void luaT_init(
- lua_State*L);extern const char*const luaT_typenames[];
- #endif
- #line 14 "lstate.h"
- #line 1 "lzio.h"
- #ifndef lzio_h
- #define lzio_h
- #define EOZ (-1)
- typedef struct Zio ZIO;
- #define char2int(c) cast(int,cast(unsigned char,(c)))
- #define zgetc(z) (((z)->n--)>0?char2int(*(z)->p++):luaZ_fill(z))
- #define zname(z) ((z)->name)
- void luaZ_init(ZIO*z,lua_Chunkreader reader,void*data,const char*name);size_t
- luaZ_read(ZIO*z,void*b,size_t n);int luaZ_lookahead(ZIO*z);typedef struct
- Mbuffer{char*buffer;size_t buffsize;}Mbuffer;char*luaZ_openspace(lua_State*L,
- Mbuffer*buff,size_t n);
- #define luaZ_initbuffer(L, buff)((buff)->buffer=NULL,(buff)->buffsize=0)
- #define luaZ_sizebuffer(buff) ((buff)->buffsize)
- #define luaZ_buffer(buff) ((buff)->buffer)
- #define luaZ_resizebuffer(L, buff,size)(luaM_reallocvector(L,(buff)->buffer,(\
- buff)->buffsize,size,char),(buff)->buffsize=size)
- #define luaZ_freebuffer(L, buff)luaZ_resizebuffer(L,buff,0)
- struct Zio{size_t n;const char*p;lua_Chunkreader reader;void*data;const char*
- name;};int luaZ_fill(ZIO*z);
- #endif
- #line 15 "lstate.h"
- #ifndef lua_lock
- #define lua_lock(L) ((void)0)
- #endif
- #ifndef lua_unlock
- #define lua_unlock(L) ((void)0)
- #endif
- #ifndef lua_userstateopen
- #define lua_userstateopen(l)
- #endif
- struct lua_longjmp;
- #define defaultmeta(L) (&G(L)->_defaultmeta)
- #define gt(L) (&L->_gt)
- #define registry(L) (&G(L)->_registry)
- #define EXTRA_STACK 5
- #define BASIC_CI_SIZE 8
- #define BASIC_STACK_SIZE (2*LUA_MINSTACK)
- typedef struct stringtable{GCObject**hash;ls_nstr nuse;int size;}stringtable;
- typedef struct CallInfo{StkId base;StkId top;int state;union{struct{const
- Instruction*savedpc;const Instruction**pc;int tailcalls;}l;struct{int dummy;}c
- ;}u;}CallInfo;
- #define CI_C (1<<0)
- #define CI_HASFRAME (1<<1)
- #define CI_CALLING (1<<2)
- #define CI_SAVEDPC (1<<3)
- #define CI_YIELD (1<<4)
- #define ci_func(ci) (clvalue((ci)->base-1))
- typedef struct global_State{stringtable strt;GCObject*rootgc;GCObject*
- rootudata;GCObject*tmudata;Mbuffer buff;lu_mem GCthreshold;lu_mem nblocks;
- lua_CFunction panic;TObject _registry;TObject _defaultmeta;struct lua_State*
- mainthread;Node dummynode[1];TString*tmname[TM_N];}global_State;struct
- lua_State{CommonHeader;StkId top;StkId base;global_State*l_G;CallInfo*ci;StkId
- stack_last;StkId stack;int stacksize;CallInfo*end_ci;CallInfo*base_ci;
- unsigned short size_ci;unsigned short nCcalls;lu_byte hookmask;lu_byte
- allowhook;lu_byte hookinit;int basehookcount;int hookcount;lua_Hook hook;
- TObject _gt;GCObject*openupval;GCObject*gclist;struct lua_longjmp*errorJmp;
- ptrdiff_t errfunc;};
- #define G(L) (L->l_G)
- union GCObject{GCheader gch;union TString ts;union Udata u;union Closure cl;
- struct Table h;struct Proto p;struct UpVal uv;struct lua_State th;};
- #define gcotots(o) check_exp((o)->gch.tt==LUA_TSTRING,&((o)->ts))
- #define gcotou(o) check_exp((o)->gch.tt==LUA_TUSERDATA,&((o)->u))
- #define gcotocl(o) check_exp((o)->gch.tt==LUA_TFUNCTION,&((o)->cl))
- #define gcotoh(o) check_exp((o)->gch.tt==LUA_TTABLE,&((o)->h))
- #define gcotop(o) check_exp((o)->gch.tt==LUA_TPROTO,&((o)->p))
- #define gcotouv(o) check_exp((o)->gch.tt==LUA_TUPVAL,&((o)->uv))
- #define ngcotouv(o) check_exp((o)==NULL||(o)->gch.tt==LUA_TUPVAL,&((o)->uv))
- #define gcototh(o) check_exp((o)->gch.tt==LUA_TTHREAD,&((o)->th))
- #define valtogco(v) (cast(GCObject*,(v)))
- lua_State*luaE_newthread(lua_State*L);void luaE_freethread(lua_State*L,
- lua_State*L1);
- #endif
- #line 12 "ldebug.h"
- #define pcRel(pc, p)(cast(int,(pc)-(p)->code)-1)
- #define getline(f,pc) (((f)->lineinfo)?(f)->lineinfo[pc]:0)
- #define resethookcount(L) (L->hookcount=L->basehookcount)
- void luaG_inithooks(lua_State*L);void luaG_typeerror(lua_State*L,const TObject
- *o,const char*opname);void luaG_concaterror(lua_State*L,StkId p1,StkId p2);
- void luaG_aritherror(lua_State*L,const TObject*p1,const TObject*p2);int
- luaG_ordererror(lua_State*L,const TObject*p1,const TObject*p2);void
- luaG_runerror(lua_State*L,const char*fmt,...);void luaG_errormsg(lua_State*L);
- int luaG_checkcode(const Proto*pt);
- #endif
- #line 17 "lapi.c"
- #line 1 "ldo.h"
- #ifndef ldo_h
- #define ldo_h
- #ifndef HARDSTACKTESTS
- #define condhardstacktests(x) {}
- #else
- #define condhardstacktests(x) x
- #endif
- #define luaD_checkstack(L,n) if((char*)L->stack_last-(char*)L->top<=(n)*(int)\
- sizeof(TObject))luaD_growstack(L,n);else condhardstacktests(luaD_reallocstack(\
- L,L->stacksize));
- #define incr_top(L) {luaD_checkstack(L,1);L->top++;}
- #define savestack(L,p) ((char*)(p)-(char*)L->stack)
- #define restorestack(L,n) ((TObject*)((char*)L->stack+(n)))
- #define saveci(L,p) ((char*)(p)-(char*)L->base_ci)
- #define restoreci(L,n) ((CallInfo*)((char*)L->base_ci+(n)))
- typedef void(*Pfunc)(lua_State*L,void*ud);void luaD_resetprotection(lua_State*
- L);int luaD_protectedparser(lua_State*L,ZIO*z,int bin);void luaD_callhook(
- lua_State*L,int event,int line);StkId luaD_precall(lua_State*L,StkId func);
- void luaD_call(lua_State*L,StkId func,int nResults);int luaD_pcall(lua_State*L
- ,Pfunc func,void*u,ptrdiff_t oldtop,ptrdiff_t ef);void luaD_poscall(lua_State*
- L,int wanted,StkId firstResult);void luaD_reallocCI(lua_State*L,int newsize);
- void luaD_reallocstack(lua_State*L,int newsize);void luaD_growstack(lua_State*
- L,int n);void luaD_throw(lua_State*L,int errcode);int luaD_rawrunprotected(
- lua_State*L,Pfunc f,void*ud);
- #endif
- #line 18 "lapi.c"
- #line 1 "lfunc.h"
- #ifndef lfunc_h
- #define lfunc_h
- Proto*luaF_newproto(lua_State*L);Closure*luaF_newCclosure(lua_State*L,int
- nelems);Closure*luaF_newLclosure(lua_State*L,int nelems,TObject*e);UpVal*
- luaF_findupval(lua_State*L,StkId level);void luaF_close(lua_State*L,StkId
- level);void luaF_freeproto(lua_State*L,Proto*f);void luaF_freeclosure(
- lua_State*L,Closure*c);const char*luaF_getlocalname(const Proto*func,int
- local_number,int pc);
- #endif
- #line 19 "lapi.c"
- #line 1 "lgc.h"
- #ifndef lgc_h
- #define lgc_h
- #define luaC_checkGC(L) {lua_assert(!(L->ci->state&CI_CALLING));if(G(L)->\
- nblocks>=G(L)->GCthreshold)luaC_collectgarbage(L);}
- size_t luaC_separateudata(lua_State*L);void luaC_callGCTM(lua_State*L);void
- luaC_sweep(lua_State*L,int all);void luaC_collectgarbage(lua_State*L);void
- luaC_link(lua_State*L,GCObject*o,lu_byte tt);
- #endif
- #line 20 "lapi.c"
- #line 1 "lmem.h"
- #ifndef lmem_h
- #define lmem_h
- #define MEMERRMSG "not enough memory"
- void*luaM_realloc(lua_State*L,void*oldblock,lu_mem oldsize,lu_mem size);void*
- luaM_growaux(lua_State*L,void*block,int*size,int size_elem,int limit,const
- char*errormsg);
- #define luaM_free(L, b,s)luaM_realloc(L,(b),(s),0)
- #define luaM_freelem(L, b)luaM_realloc(L,(b),sizeof(*(b)),0)
- #define luaM_freearray(L, b,n,t)luaM_realloc(L,(b),cast(lu_mem,n)*cast(lu_mem,\
- sizeof(t)),0)
- #define luaM_malloc(L, t)luaM_realloc(L,NULL,0,(t))
- #define luaM_new(L, t)cast(t*,luaM_malloc(L,sizeof(t)))
- #define luaM_newvector(L, n,t)cast(t*,luaM_malloc(L,cast(lu_mem,n)*cast(lu_mem\
- ,sizeof(t))))
- #define luaM_growvector(L,v,nelems,size,t,limit,e) if(((nelems)+1)>(size))((v)\
- =cast(t*,luaM_growaux(L,v,&(size),sizeof(t),limit,e)))
- #define luaM_reallocvector(L, v,oldn,n,t)((v)=cast(t*,luaM_realloc(L,v,cast(\
- lu_mem,oldn)*cast(lu_mem,sizeof(t)),cast(lu_mem,n)*cast(lu_mem,sizeof(t)))))
- #endif
- #line 21 "lapi.c"
- #line 1 "lstring.h"
- #ifndef lstring_h
- #define lstring_h
- #define sizestring(l) (cast(lu_mem,sizeof(union TString))+(cast(lu_mem,l)+1)*\
- sizeof(char))
- #define sizeudata(l) (cast(lu_mem,sizeof(union Udata))+(l))
- #define luaS_new(L, s)(luaS_newlstr(L,s,strlen(s)))
- #define luaS_newliteral(L, s)(luaS_newlstr(L,""s,(sizeof(s)/sizeof(char))-1))
- #define luaS_fix(s) ((s)->tsv.marked|=(1<<4))
- void luaS_resize(lua_State*L,int newsize);Udata*luaS_newudata(lua_State*L,
- size_t s);void luaS_freeall(lua_State*L);TString*luaS_newlstr(lua_State*L,
- const char*str,size_t l);
- #endif
- #line 24 "lapi.c"
- #line 1 "ltable.h"
- #ifndef ltable_h
- #define ltable_h
- #define gnode(t,i) (&(t)->node[i])
- #define gkey(n) (&(n)->i_key)
- #define gval(n) (&(n)->i_val)
- const TObject*luaH_getnum(Table*t,int key);TObject*luaH_setnum(lua_State*L,
- Table*t,int key);const TObject*luaH_getstr(Table*t,TString*key);const TObject*
- luaH_get(Table*t,const TObject*key);TObject*luaH_set(lua_State*L,Table*t,const
- TObject*key);Table*luaH_new(lua_State*L,int narray,int lnhash);void luaH_free
- (lua_State*L,Table*t);int luaH_next(lua_State*L,Table*t,StkId key);Node*
- luaH_mainposition(const Table*t,const TObject*key);
- #endif
- #line 25 "lapi.c"
- #line 1 "lundump.h"
- #ifndef lundump_h
- #define lundump_h
- Proto*luaU_undump(lua_State*L,ZIO*Z,Mbuffer*buff);int luaU_endianness(void);
- void luaU_dump(lua_State*L,const Proto*Main,lua_Chunkwriter w,void*data);void
- luaU_print(const Proto*Main);
- #define LUA_SIGNATURE "\033Lua"
- #define VERSION 0x50
- #define VERSION0 0x50
- #define TEST_NUMBER ((lua_Number)3.14159265358979323846E7)
- #endif
- #line 27 "lapi.c"
- #line 1 "lvm.h"
- #ifndef lvm_h
- #define lvm_h
- #define tostring(L,o) ((ttype(o)==LUA_TSTRING)||(luaV_tostring(L,o)))
- #define tonumber(o,n) (ttype(o)==LUA_TNUMBER||(((o)=luaV_tonumber(o,n))!=NULL)\
- )
- #define equalobj(L,o1,o2) (ttype(o1)==ttype(o2)&&luaV_equalval(L,o1,o2))
- int luaV_lessthan(lua_State*L,const TObject*l,const TObject*r);int
- luaV_equalval(lua_State*L,const TObject*t1,const TObject*t2);const TObject*
- luaV_tonumber(const TObject*obj,TObject*n);int luaV_tostring(lua_State*L,StkId
- obj);const TObject*luaV_gettable(lua_State*L,const TObject*t,TObject*key,int
- loop);void luaV_settable(lua_State*L,const TObject*t,TObject*key,StkId val);
- StkId luaV_execute(lua_State*L);void luaV_concat(lua_State*L,int total,int
- last);
- #endif
- #line 28 "lapi.c"
- const char lua_ident[]="$Lua: "LUA_VERSION" "LUA_COPYRIGHT" $\n""$Authors: "
- LUA_AUTHORS" $\n""$URL: www.lua.org $\n";
- #ifndef api_check
- #define api_check(L, o)
- #endif
- #define api_checknelems(L, n)api_check(L,(n)<=(L->top-L->base))
- #define api_incr_top(L) {api_check(L,L->top<L->ci->top);L->top++;}
- static TObject*negindex(lua_State*L,int idx){if(idx>LUA_REGISTRYINDEX){
- api_check(L,idx!=0&&-idx<=L->top-L->base);return L->top+idx;}else switch(idx){
- case LUA_REGISTRYINDEX:return registry(L);case LUA_GLOBALSINDEX:return gt(L);
- default:{TObject*func=(L->base-1);idx=LUA_GLOBALSINDEX-idx;lua_assert(
- iscfunction(func));return(idx<=clvalue(func)->c.nupvalues)?&clvalue(func)->c.
- upvalue[idx-1]:NULL;}}}static TObject*luaA_index(lua_State*L,int idx){if(idx>0
- ){api_check(L,idx<=L->top-L->base);return L->base+idx-1;}else{TObject*o=
- negindex(L,idx);api_check(L,o!=NULL);return o;}}static TObject*
- luaA_indexAcceptable(lua_State*L,int idx){if(idx>0){TObject*o=L->base+(idx-1);
- api_check(L,idx<=L->stack_last-L->base);if(o>=L->top)return NULL;else return o
- ;}else return negindex(L,idx);}void luaA_pushobject(lua_State*L,const TObject*
- o){setobj2s(L->top,o);incr_top(L);}LUA_API int lua_checkstack(lua_State*L,int
- size){int res;lua_lock(L);if((L->top-L->base+size)>LUA_MAXCSTACK)res=0;else{
- luaD_checkstack(L,size);if(L->ci->top<L->top+size)L->ci->top=L->top+size;res=1
- ;}lua_unlock(L);return res;}LUA_API void lua_xmove(lua_State*from,lua_State*to
- ,int n){int i;lua_lock(to);api_checknelems(from,n);from->top-=n;for(i=0;i<n;i
- ++){setobj2s(to->top,from->top+i);api_incr_top(to);}lua_unlock(to);}LUA_API
- lua_CFunction lua_atpanic(lua_State*L,lua_CFunction panicf){lua_CFunction old;
- lua_lock(L);old=G(L)->panic;G(L)->panic=panicf;lua_unlock(L);return old;}
- LUA_API lua_State*lua_newthread(lua_State*L){lua_State*L1;lua_lock(L);
- luaC_checkGC(L);L1=luaE_newthread(L);setthvalue(L->top,L1);api_incr_top(L);
- lua_unlock(L);lua_userstateopen(L1);return L1;}LUA_API int lua_gettop(
- lua_State*L){return(L->top-L->base);}LUA_API void lua_settop(lua_State*L,int
- idx){lua_lock(L);if(idx>=0){api_check(L,idx<=L->stack_last-L->base);while(L->
- top<L->base+idx)setnilvalue(L->top++);L->top=L->base+idx;}else{api_check(L,-(
- idx+1)<=(L->top-L->base));L->top+=idx+1;}lua_unlock(L);}LUA_API void
- lua_remove(lua_State*L,int idx){StkId p;lua_lock(L);p=luaA_index(L,idx);while(
- ++p<L->top)setobjs2s(p-1,p);L->top--;lua_unlock(L);}LUA_API void lua_insert(
- lua_State*L,int idx){StkId p;StkId q;lua_lock(L);p=luaA_index(L,idx);for(q=L->
- top;q>p;q--)setobjs2s(q,q-1);setobjs2s(p,L->top);lua_unlock(L);}LUA_API void
- lua_replace(lua_State*L,int idx){lua_lock(L);api_checknelems(L,1);setobj(
- luaA_index(L,idx),L->top-1);L->top--;lua_unlock(L);}LUA_API void lua_pushvalue
- (lua_State*L,int idx){lua_lock(L);setobj2s(L->top,luaA_index(L,idx));
- api_incr_top(L);lua_unlock(L);}LUA_API int lua_type(lua_State*L,int idx){StkId
- o=luaA_indexAcceptable(L,idx);return(o==NULL)?LUA_TNONE:ttype(o);}LUA_API
- const char*lua_typename(lua_State*L,int t){UNUSED(L);return(t==LUA_TNONE)?
- "no value":luaT_typenames[t];}LUA_API int lua_iscfunction(lua_State*L,int idx)
- {StkId o=luaA_indexAcceptable(L,idx);return(o==NULL)?0:iscfunction(o);}LUA_API
- int lua_isnumber(lua_State*L,int idx){TObject n;const TObject*o=
- luaA_indexAcceptable(L,idx);return(o!=NULL&&tonumber(o,&n));}LUA_API int
- lua_isstring(lua_State*L,int idx){int t=lua_type(L,idx);return(t==LUA_TSTRING
- ||t==LUA_TNUMBER);}LUA_API int lua_isuserdata(lua_State*L,int idx){const
- TObject*o=luaA_indexAcceptable(L,idx);return(o!=NULL&&(ttisuserdata(o)||
- ttislightuserdata(o)));}LUA_API int lua_rawequal(lua_State*L,int index1,int
- index2){StkId o1=luaA_indexAcceptable(L,index1);StkId o2=luaA_indexAcceptable(
- L,index2);return(o1==NULL||o2==NULL)?0:luaO_rawequalObj(o1,o2);}LUA_API int
- lua_equal(lua_State*L,int index1,int index2){StkId o1,o2;int i;lua_lock(L);o1=
- luaA_indexAcceptable(L,index1);o2=luaA_indexAcceptable(L,index2);i=(o1==NULL||
- o2==NULL)?0:equalobj(L,o1,o2);lua_unlock(L);return i;}LUA_API int lua_lessthan
- (lua_State*L,int index1,int index2){StkId o1,o2;int i;lua_lock(L);o1=
- luaA_indexAcceptable(L,index1);o2=luaA_indexAcceptable(L,index2);i=(o1==NULL||
- o2==NULL)?0:luaV_lessthan(L,o1,o2);lua_unlock(L);return i;}LUA_API lua_Number
- lua_tonumber(lua_State*L,int idx){TObject n;const TObject*o=
- luaA_indexAcceptable(L,idx);if(o!=NULL&&tonumber(o,&n))return nvalue(o);else
- return 0;}LUA_API int lua_toboolean(lua_State*L,int idx){const TObject*o=
- luaA_indexAcceptable(L,idx);return(o!=NULL)&&!l_isfalse(o);}LUA_API const char
- *lua_tostring(lua_State*L,int idx){StkId o=luaA_indexAcceptable(L,idx);if(o==
- NULL)return NULL;else if(ttisstring(o))return svalue(o);else{const char*s;
- lua_lock(L);s=(luaV_tostring(L,o)?svalue(o):NULL);luaC_checkGC(L);lua_unlock(L
- );return s;}}LUA_API size_t lua_strlen(lua_State*L,int idx){StkId o=
- luaA_indexAcceptable(L,idx);if(o==NULL)return 0;else if(ttisstring(o))return
- tsvalue(o)->tsv.len;else{size_t l;lua_lock(L);l=(luaV_tostring(L,o)?tsvalue(o)
- ->tsv.len:0);lua_unlock(L);return l;}}LUA_API lua_CFunction lua_tocfunction(
- lua_State*L,int idx){StkId o=luaA_indexAcceptable(L,idx);return(o==NULL||!
- iscfunction(o))?NULL:clvalue(o)->c.f;}LUA_API void*lua_touserdata(lua_State*L,
- int idx){StkId o=luaA_indexAcceptable(L,idx);if(o==NULL)return NULL;switch(
- ttype(o)){case LUA_TUSERDATA:return(uvalue(o)+1);case LUA_TLIGHTUSERDATA:
- return pvalue(o);default:return NULL;}}LUA_API lua_State*lua_tothread(
- lua_State*L,int idx){StkId o=luaA_indexAcceptable(L,idx);return(o==NULL||!
- ttisthread(o))?NULL:thvalue(o);}LUA_API const void*lua_topointer(lua_State*L,
- int idx){StkId o=luaA_indexAcceptable(L,idx);if(o==NULL)return NULL;else{
- switch(ttype(o)){case LUA_TTABLE:return hvalue(o);case LUA_TFUNCTION:return
- clvalue(o);case LUA_TTHREAD:return thvalue(o);case LUA_TUSERDATA:case
- LUA_TLIGHTUSERDATA:return lua_touserdata(L,idx);default:return NULL;}}}LUA_API
- void lua_pushnil(lua_State*L){lua_lock(L);setnilvalue(L->top);api_incr_top(L)
- ;lua_unlock(L);}LUA_API void lua_pushnumber(lua_State*L,lua_Number n){lua_lock
- (L);setnvalue(L->top,n);api_incr_top(L);lua_unlock(L);}LUA_API void
- lua_pushlstring(lua_State*L,const char*s,size_t len){lua_lock(L);luaC_checkGC(
- L);setsvalue2s(L->top,luaS_newlstr(L,s,len));api_incr_top(L);lua_unlock(L);}
- LUA_API void lua_pushstring(lua_State*L,const char*s){if(s==NULL)lua_pushnil(L
- );else lua_pushlstring(L,s,strlen(s));}LUA_API const char*lua_pushvfstring(
- lua_State*L,const char*fmt,va_list argp){const char*ret;lua_lock(L);
- luaC_checkGC(L);ret=luaO_pushvfstring(L,fmt,argp);lua_unlock(L);return ret;}
- LUA_API const char*lua_pushfstring(lua_State*L,const char*fmt,...){const char*
- ret;va_list argp;lua_lock(L);luaC_checkGC(L);va_start(argp,fmt);ret=
- luaO_pushvfstring(L,fmt,argp);va_end(argp);lua_unlock(L);return ret;}LUA_API
- void lua_pushcclosure(lua_State*L,lua_CFunction fn,int n){Closure*cl;lua_lock(
- L);luaC_checkGC(L);api_checknelems(L,n);cl=luaF_newCclosure(L,n);cl->c.f=fn;L
- ->top-=n;while(n--)setobj2n(&cl->c.upvalue[n],L->top+n);setclvalue(L->top,cl);
- api_incr_top(L);lua_unlock(L);}LUA_API void lua_pushboolean(lua_State*L,int b)
- {lua_lock(L);setbvalue(L->top,(b!=0));api_incr_top(L);lua_unlock(L);}LUA_API
- void lua_pushlightuserdata(lua_State*L,void*p){lua_lock(L);setpvalue(L->top,p)
- ;api_incr_top(L);lua_unlock(L);}LUA_API void lua_gettable(lua_State*L,int idx)
- {StkId t;lua_lock(L);t=luaA_index(L,idx);setobj2s(L->top-1,luaV_gettable(L,t,L
- ->top-1,0));lua_unlock(L);}LUA_API void lua_rawget(lua_State*L,int idx){StkId
- t;lua_lock(L);t=luaA_index(L,idx);api_check(L,ttistable(t));setobj2s(L->top-1,
- luaH_get(hvalue(t),L->top-1));lua_unlock(L);}LUA_API void lua_rawgeti(
- lua_State*L,int idx,int n){StkId o;lua_lock(L);o=luaA_index(L,idx);api_check(L
- ,ttistable(o));setobj2s(L->top,luaH_getnum(hvalue(o),n));api_incr_top(L);
- lua_unlock(L);}LUA_API void lua_newtable(lua_State*L){lua_lock(L);luaC_checkGC
- (L);sethvalue(L->top,luaH_new(L,0,0));api_incr_top(L);lua_unlock(L);}LUA_API
- int lua_getmetatable(lua_State*L,int objindex){const TObject*obj;Table*mt=NULL
- ;int res;lua_lock(L);obj=luaA_indexAcceptable(L,objindex);if(obj!=NULL){switch
- (ttype(obj)){case LUA_TTABLE:mt=hvalue(obj)->metatable;break;case
- LUA_TUSERDATA:mt=uvalue(obj)->uv.metatable;break;}}if(mt==NULL||mt==hvalue(
- defaultmeta(L)))res=0;else{sethvalue(L->top,mt);api_incr_top(L);res=1;}
- lua_unlock(L);return res;}LUA_API void lua_getfenv(lua_State*L,int idx){StkId
- o;lua_lock(L);o=luaA_index(L,idx);setobj2s(L->top,isLfunction(o)?&clvalue(o)->
- l.g:gt(L));api_incr_top(L);lua_unlock(L);}LUA_API void lua_settable(lua_State*
- L,int idx){StkId t;lua_lock(L);api_checknelems(L,2);t=luaA_index(L,idx);
- luaV_settable(L,t,L->top-2,L->top-1);L->top-=2;lua_unlock(L);}LUA_API void
- lua_rawset(lua_State*L,int idx){StkId t;lua_lock(L);api_checknelems(L,2);t=
- luaA_index(L,idx);api_check(L,ttistable(t));setobj2t(luaH_set(L,hvalue(t),L->
- top-2),L->top-1);L->top-=2;lua_unlock(L);}LUA_API void lua_rawseti(lua_State*L
- ,int idx,int n){StkId o;lua_lock(L);api_checknelems(L,1);o=luaA_index(L,idx);
- api_check(L,ttistable(o));setobj2t(luaH_setnum(L,hvalue(o),n),L->top-1);L->top
- --;lua_unlock(L);}LUA_API int lua_setmetatable(lua_State*L,int objindex){
- TObject*obj,*mt;int res=1;lua_lock(L);api_checknelems(L,1);obj=luaA_index(L,
- objindex);mt=(!ttisnil(L->top-1))?L->top-1:defaultmeta(L);api_check(L,
- ttistable(mt));switch(ttype(obj)){case LUA_TTABLE:{hvalue(obj)->metatable=
- hvalue(mt);break;}case LUA_TUSERDATA:{uvalue(obj)->uv.metatable=hvalue(mt);
- break;}default:{res=0;break;}}L->top--;lua_unlock(L);return res;}LUA_API int
- lua_setfenv(lua_State*L,int idx){StkId o;int res=0;lua_lock(L);api_checknelems
- (L,1);o=luaA_index(L,idx);L->top--;api_check(L,ttistable(L->top));if(
- isLfunction(o)){res=1;clvalue(o)->l.g=*(L->top);}lua_unlock(L);return res;}
- LUA_API void lua_call(lua_State*L,int nargs,int nresults){StkId func;lua_lock(
- L);api_checknelems(L,nargs+1);func=L->top-(nargs+1);luaD_call(L,func,nresults)
- ;lua_unlock(L);}struct CallS{StkId func;int nresults;};static void f_call(
- lua_State*L,void*ud){struct CallS*c=cast(struct CallS*,ud);luaD_call(L,c->func
- ,c->nresults);}LUA_API int lua_pcall(lua_State*L,int nargs,int nresults,int
- errfunc){struct CallS c;int status;ptrdiff_t func;lua_lock(L);func=(errfunc==0
- )?0:savestack(L,luaA_index(L,errfunc));c.func=L->top-(nargs+1);c.nresults=
- nresults;status=luaD_pcall(L,f_call,&c,savestack(L,c.func),func);lua_unlock(L)
- ;return status;}struct CCallS{lua_CFunction func;void*ud;};static void f_Ccall
- (lua_State*L,void*ud){struct CCallS*c=cast(struct CCallS*,ud);Closure*cl;cl=
- luaF_newCclosure(L,0);cl->c.f=c->func;setclvalue(L->top,cl);incr_top(L);
- setpvalue(L->top,c->ud);incr_top(L);luaD_call(L,L->top-2,0);}LUA_API int
- lua_cpcall(lua_State*L,lua_CFunction func,void*ud){struct CCallS c;int status;
- lua_lock(L);c.func=func;c.ud=ud;status=luaD_pcall(L,f_Ccall,&c,savestack(L,L->
- top),0);lua_unlock(L);return status;}LUA_API int lua_load(lua_State*L,
- lua_Chunkreader reader,void*data,const char*chunkname){ZIO z;int status;int c;
- lua_lock(L);if(!chunkname)chunkname="?";luaZ_init(&z,reader,data,chunkname);c=
- luaZ_lookahead(&z);status=luaD_protectedparser(L,&z,(c==LUA_SIGNATURE[0]));
- lua_unlock(L);return status;}LUA_API int lua_dump(lua_State*L,lua_Chunkwriter
- writer,void*data){int status;TObject*o;lua_lock(L);api_checknelems(L,1);o=L->
- top-1;if(isLfunction(o)&&clvalue(o)->l.nupvalues==0){luaU_dump(L,clvalue(o)->l
- .p,writer,data);status=1;}else status=0;lua_unlock(L);return status;}
- #define GCscalel(x) ((x)>>10)
- #define GCscale(x) (cast(int,GCscalel(x)))
- #define GCunscale(x) (cast(lu_mem,x)<<10)
- LUA_API int lua_getgcthreshold(lua_State*L){int threshold;lua_lock(L);
- threshold=GCscale(G(L)->GCthreshold);lua_unlock(L);return threshold;}LUA_API
- int lua_getgccount(lua_State*L){int count;lua_lock(L);count=GCscale(G(L)->
- nblocks);lua_unlock(L);return count;}LUA_API void lua_setgcthreshold(lua_State
- *L,int newthreshold){lua_lock(L);if(cast(lu_mem,newthreshold)>GCscalel(
- MAX_LUMEM))G(L)->GCthreshold=MAX_LUMEM;else G(L)->GCthreshold=GCunscale(
- newthreshold);luaC_checkGC(L);lua_unlock(L);}LUA_API const char*lua_version(
- void){return LUA_VERSION;}LUA_API int lua_error(lua_State*L){lua_lock(L);
- api_checknelems(L,1);luaG_errormsg(L);lua_unlock(L);return 0;}LUA_API int
- lua_next(lua_State*L,int idx){StkId t;int more;lua_lock(L);t=luaA_index(L,idx)
- ;api_check(L,ttistable(t));more=luaH_next(L,hvalue(t),L->top-1);if(more){
- api_incr_top(L);}else L->top-=1;lua_unlock(L);return more;}LUA_API void
- lua_concat(lua_State*L,int n){lua_lock(L);luaC_checkGC(L);api_checknelems(L,n)
- ;if(n>=2){luaV_concat(L,n,L->top-L->base-1);L->top-=(n-1);}else if(n==0){
- setsvalue2s(L->top,luaS_newlstr(L,NULL,0));api_incr_top(L);}lua_unlock(L);}
- LUA_API void*lua_newuserdata(lua_State*L,size_t size){Udata*u;lua_lock(L);
- luaC_checkGC(L);u=luaS_newudata(L,size);setuvalue(L->top,u);api_incr_top(L);
- lua_unlock(L);return u+1;}LUA_API int lua_pushupvalues(lua_State*L){Closure*
- func;int n,i;lua_lock(L);api_check(L,iscfunction(L->base-1));func=clvalue(L->
- base-1);n=func->c.nupvalues;luaD_checkstack(L,n+LUA_MINSTACK);for(i=0;i<n;i++)
- {setobj2s(L->top,&func->c.upvalue[i]);L->top++;}lua_unlock(L);return n;}static
- const char*aux_upvalue(lua_State*L,int funcindex,int n,TObject**val){Closure*
- f;StkId fi=luaA_index(L,funcindex);if(!ttisfunction(fi))return NULL;f=clvalue(
- fi);if(f->c.isC){if(n>f->c.nupvalues)return NULL;*val=&f->c.upvalue[n-1];
- return"";}else{Proto*p=f->l.p;if(n>p->sizeupvalues)return NULL;*val=f->l.
- upvals[n-1]->v;return getstr(p->upvalues[n-1]);}}LUA_API const char*
- lua_getupvalue(lua_State*L,int funcindex,int n){const char*name;TObject*val;
- lua_lock(L);name=aux_upvalue(L,funcindex,n,&val);if(name){setobj2s(L->top,val)
- ;api_incr_top(L);}lua_unlock(L);return name;}LUA_API const char*lua_setupvalue
- (lua_State*L,int funcindex,int n){const char*name;TObject*val;lua_lock(L);
- api_checknelems(L,1);name=aux_upvalue(L,funcindex,n,&val);if(name){L->top--;
- setobj(val,L->top);}lua_unlock(L);return name;}
- #line 1 "lauxlib.c"
- #define lauxlib_c
- #line 1 "lauxlib.h"
- #ifndef lauxlib_h
- #define lauxlib_h
- #ifndef LUALIB_API
- #define LUALIB_API LUA_API
- #endif
- typedef struct luaL_reg{const char*name;lua_CFunction func;}luaL_reg;
- LUALIB_API void luaL_openlib(lua_State*L,const char*libname,const luaL_reg*l,
- int nup);LUALIB_API int luaL_getmetafield(lua_State*L,int obj,const char*e);
- LUALIB_API int luaL_callmeta(lua_State*L,int obj,const char*e);LUALIB_API int
- luaL_typerror(lua_State*L,int narg,const char*tname);LUALIB_API int
- luaL_argerror(lua_State*L,int numarg,const char*extramsg);LUALIB_API const
- char*luaL_checklstring(lua_State*L,int numArg,size_t*l);LUALIB_API const char*
- luaL_optlstring(lua_State*L,int numArg,const char*def,size_t*l);LUALIB_API
- lua_Number luaL_checknumber(lua_State*L,int numArg);LUALIB_API lua_Number
- luaL_optnumber(lua_State*L,int nArg,lua_Number def);LUALIB_API void
- luaL_checkstack(lua_State*L,int sz,const char*msg);LUALIB_API void
- luaL_checktype(lua_State*L,int narg,int t);LUALIB_API void luaL_checkany(
- lua_State*L,int narg);LUALIB_API int luaL_newmetatable(lua_State*L,const char*
- tname);LUALIB_API void luaL_getmetatable(lua_State*L,const char*tname);
- LUALIB_API void*luaL_checkudata(lua_State*L,int ud,const char*tname);
- LUALIB_API void luaL_where(lua_State*L,int lvl);LUALIB_API int luaL_error(
- lua_State*L,const char*fmt,...);LUALIB_API int luaL_findstring(const char*st,
- const char*const lst[]);LUALIB_API int luaL_ref(lua_State*L,int t);LUALIB_API
- void luaL_unref(lua_State*L,int t,int ref);LUALIB_API int luaL_getn(lua_State*
- L,int t);LUALIB_API void luaL_setn(lua_State*L,int t,int n);LUALIB_API int
- luaL_loadfile(lua_State*L,const char*filename);LUALIB_API int luaL_loadbuffer(
- lua_State*L,const char*buff,size_t sz,const char*name);
- #define luaL_argcheck(L, cond,numarg,extramsg)if(!(cond))luaL_argerror(L,\
- numarg,extramsg)
- #define luaL_checkstring(L,n) (luaL_checklstring(L,(n),NULL))
- #define luaL_optstring(L,n,d) (luaL_optlstring(L,(n),(d),NULL))
- #define luaL_checkint(L,n) ((int)luaL_checknumber(L,n))
- #define luaL_checklong(L,n) ((long)luaL_checknumber(L,n))
- #define luaL_optint(L,n,d) ((int)luaL_optnumber(L,n,(lua_Number)(d)))
- #define luaL_optlong(L,n,d) ((long)luaL_optnumber(L,n,(lua_Number)(d)))
- #ifndef LUAL_BUFFERSIZE
- #define LUAL_BUFFERSIZE BUFSIZ
- #endif
- typedef struct luaL_Buffer{char*p;int lvl;lua_State*L;char buffer[
- LUAL_BUFFERSIZE];}luaL_Buffer;
- #define luaL_putchar(B,c) ((void)((B)->p<((B)->buffer+LUAL_BUFFERSIZE)||\
- luaL_prepbuffer(B)),(*(B)->p++=(char)(c)))
- #define luaL_addsize(B,n) ((B)->p+=(n))
- LUALIB_API void luaL_buffinit(lua_State*L,luaL_Buffer*B);LUALIB_API char*
- luaL_prepbuffer(luaL_Buffer*B);LUALIB_API void luaL_addlstring(luaL_Buffer*B,
- const char*s,size_t l);LUALIB_API void luaL_addstring(luaL_Buffer*B,const char
- *s);LUALIB_API void luaL_addvalue(luaL_Buffer*B);LUALIB_API void
- luaL_pushresult(luaL_Buffer*B);LUALIB_API int lua_dofile(lua_State*L,const
- char*filename);LUALIB_API int lua_dostring(lua_State*L,const char*str);
- LUALIB_API int lua_dobuffer(lua_State*L,const char*buff,size_t sz,const char*n
- );
- #define luaL_check_lstr luaL_checklstring
- #define luaL_opt_lstr luaL_optlstring
- #define luaL_check_number luaL_checknumber
- #define luaL_opt_number luaL_optnumber
- #define luaL_arg_check luaL_argcheck
- #define luaL_check_string luaL_checkstring
- #define luaL_opt_string luaL_optstring
- #define luaL_check_int luaL_checkint
- #define luaL_check_long luaL_checklong
- #define luaL_opt_int luaL_optint
- #define luaL_opt_long luaL_optlong
- #endif
- #line 24 "lauxlib.c"
- #define RESERVED_REFS 2
- #define FREELIST_REF 1
- #define ARRAYSIZE_REF 2
- #define abs_index(L, i)((i)>0||(i)<=LUA_REGISTRYINDEX?(i):lua_gettop(L)+(i)+1)
- LUALIB_API int luaL_argerror(lua_State*L,int narg,const char*extramsg){
- lua_Debug ar;lua_getstack(L,0,&ar);lua_getinfo(L,"n",&ar);if(strcmp(ar.
- namewhat,"method")==0){narg--;if(narg==0)return luaL_error(L,
- "calling `%s' on bad self (%s)",ar.name,extramsg);}if(ar.name==NULL)ar.name=
- "?";return luaL_error(L,"bad argument #%d to `%s' (%s)",narg,ar.name,extramsg)
- ;}LUALIB_API int luaL_typerror(lua_State*L,int narg,const char*tname){const
- char*msg=lua_pushfstring(L,"%s expected, got %s",tname,lua_typename(L,lua_type
- (L,narg)));return luaL_argerror(L,narg,msg);}static void tag_error(lua_State*L
- ,int narg,int tag){luaL_typerror(L,narg,lua_typename(L,tag));}LUALIB_API void
- luaL_where(lua_State*L,int level){lua_Debug ar;if(lua_getstack(L,level,&ar)){
- lua_getinfo(L,"Snl",&ar);if(ar.currentline>0){lua_pushfstring(L,"%s:%d: ",ar.
- short_src,ar.currentline);return;}}lua_pushliteral(L,"");}LUALIB_API int
- luaL_error(lua_State*L,const char*fmt,...){va_list argp;va_start(argp,fmt);
- luaL_where(L,1);lua_pushvfstring(L,fmt,argp);va_end(argp);lua_concat(L,2);
- return lua_error(L);}LUALIB_API int luaL_findstring(const char*name,const char
- *const list[]){int i;for(i=0;list[i];i++)if(strcmp(list[i],name)==0)return i;
- return-1;}LUALIB_API int luaL_newmetatable(lua_State*L,const char*tname){
- lua_pushstring(L,tname);lua_rawget(L,LUA_REGISTRYINDEX);if(!lua_isnil(L,-1))
- return 0;lua_pop(L,1);lua_newtable(L);lua_pushstring(L,tname);lua_pushvalue(L,
- -2);lua_rawset(L,LUA_REGISTRYINDEX);lua_pushvalue(L,-1);lua_pushstring(L,tname
- );lua_rawset(L,LUA_REGISTRYINDEX);return 1;}LUALIB_API void luaL_getmetatable(
- lua_State*L,const char*tname){lua_pushstring(L,tname);lua_rawget(L,
- LUA_REGISTRYINDEX);}LUALIB_API void*luaL_checkudata(lua_State*L,int ud,const
- char*tname){const char*tn;if(!lua_getmetatable(L,ud))return NULL;lua_rawget(L,
- LUA_REGISTRYINDEX);tn=lua_tostring(L,-1);if(tn&&(strcmp(tn,tname)==0)){lua_pop
- (L,1);return lua_touserdata(L,ud);}else{lua_pop(L,1);return NULL;}}LUALIB_API
- void luaL_checkstack(lua_State*L,int space,const char*mes){if(!lua_checkstack(
- L,space))luaL_error(L,"stack overflow (%s)",mes);}LUALIB_API void
- luaL_checktype(lua_State*L,int narg,int t){if(lua_type(L,narg)!=t)tag_error(L,
- narg,t);}LUALIB_API void luaL_checkany(lua_State*L,int narg){if(lua_type(L,
- narg)==LUA_TNONE)luaL_argerror(L,narg,"value expected");}LUALIB_API const char
- *luaL_checklstring(lua_State*L,int narg,size_t*len){const char*s=lua_tostring(
- L,narg);if(!s)tag_error(L,narg,LUA_TSTRING);if(len)*len=lua_strlen(L,narg);
- return s;}LUALIB_API const char*luaL_optlstring(lua_State*L,int narg,const
- char*def,size_t*len){if(lua_isnoneornil(L,narg)){if(len)*len=(def?strlen(def):
- 0);return def;}else return luaL_checklstring(L,narg,len);}LUALIB_API
- lua_Number luaL_checknumber(lua_State*L,int narg){lua_Number d=lua_tonumber(L,
- narg);if(d==0&&!lua_isnumber(L,narg))tag_error(L,narg,LUA_TNUMBER);return d;}
- LUALIB_API lua_Number luaL_optnumber(lua_State*L,int narg,lua_Number def){if(
- lua_isnoneornil(L,narg))return def;else return luaL_checknumber(L,narg);}
- LUALIB_API int luaL_getmetafield(lua_State*L,int obj,const char*event){if(!
- lua_getmetatable(L,obj))return 0;lua_pushstring(L,event);lua_rawget(L,-2);if(
- lua_isnil(L,-1)){lua_pop(L,2);return 0;}else{lua_remove(L,-2);return 1;}}
- LUALIB_API int luaL_callmeta(lua_State*L,int obj,const char*event){obj=
- abs_index(L,obj);if(!luaL_getmetafield(L,obj,event))return 0;lua_pushvalue(L,
- obj);lua_call(L,1,1);return 1;}LUALIB_API void luaL_openlib(lua_State*L,const
- char*libname,const luaL_reg*l,int nup){if(libname){lua_pushstring(L,libname);
- lua_gettable(L,LUA_GLOBALSINDEX);if(lua_isnil(L,-1)){lua_pop(L,1);lua_newtable
- (L);lua_pushstring(L,libname);lua_pushvalue(L,-2);lua_settable(L,
- LUA_GLOBALSINDEX);}lua_insert(L,-(nup+1));}for(;l->name;l++){int i;
- lua_pushstring(L,l->name);for(i=0;i<nup;i++)lua_pushvalue(L,-(nup+1));
- lua_pushcclosure(L,l->func,nup);lua_settable(L,-(nup+3));}lua_pop(L,nup);}
- static int checkint(lua_State*L,int topop){int n=(int)lua_tonumber(L,-1);if(n
- ==0&&!lua_isnumber(L,-1))n=-1;lua_pop(L,topop);return n;}static void getsizes(
- lua_State*L){lua_rawgeti(L,LUA_REGISTRYINDEX,ARRAYSIZE_REF);if(lua_isnil(L,-1)
- ){lua_pop(L,1);lua_newtable(L);lua_pushvalue(L,-1);lua_setmetatable(L,-2);
- lua_pushliteral(L,"__mode");lua_pushliteral(L,"k");lua_rawset(L,-3);
- lua_pushvalue(L,-1);lua_rawseti(L,LUA_REGISTRYINDEX,ARRAYSIZE_REF);}}void
- luaL_setn(lua_State*L,int t,int n){t=abs_index(L,t);lua_pushliteral(L,"n");
- lua_rawget(L,t);if(checkint(L,1)>=0){lua_pushliteral(L,"n");lua_pushnumber(L,(
- lua_Number)n);lua_rawset(L,t);}else{getsizes(L);lua_pushvalue(L,t);
- lua_pushnumber(L,(lua_Number)n);lua_rawset(L,-3);lua_pop(L,1);}}int luaL_getn(
- lua_State*L,int t){int n;t=abs_index(L,t);lua_pushliteral(L,"n");lua_rawget(L,
- t);if((n=checkint(L,1))>=0)return n;getsizes(L);lua_pushvalue(L,t);lua_rawget(
- L,-2);if((n=checkint(L,2))>=0)return n;for(n=1;;n++){lua_rawgeti(L,t,n);if(
- lua_isnil(L,-1))break;lua_pop(L,1);}lua_pop(L,1);return n-1;}
- #define bufflen(B) ((B)->p-(B)->buffer)
- #define bufffree(B) ((size_t)(LUAL_BUFFERSIZE-bufflen(B)))
- #define LIMIT (LUA_MINSTACK/2)
- static int emptybuffer(luaL_Buffer*B){size_t l=bufflen(B);if(l==0)return 0;
- else{lua_pushlstring(B->L,B->buffer,l);B->p=B->buffer;B->lvl++;return 1;}}
- static void adjuststack(luaL_Buffer*B){if(B->lvl>1){lua_State*L=B->L;int toget
- =1;size_t toplen=lua_strlen(L,-1);do{size_t l=lua_strlen(L,-(toget+1));if(B->
- lvl-toget+1>=LIMIT||toplen>l){toplen+=l;toget++;}else break;}while(toget<B->
- lvl);lua_concat(L,toget);B->lvl=B->lvl-toget+1;}}LUALIB_API char*
- luaL_prepbuffer(luaL_Buffer*B){if(emptybuffer(B))adjuststack(B);return B->
- buffer;}LUALIB_API void luaL_addlstring(luaL_Buffer*B,const char*s,size_t l){
- while(l--)luaL_putchar(B,*s++);}LUALIB_API void luaL_addstring(luaL_Buffer*B,
- const char*s){luaL_addlstring(B,s,strlen(s));}LUALIB_API void luaL_pushresult(
- luaL_Buffer*B){emptybuffer(B);lua_concat(B->L,B->lvl);B->lvl=1;}LUALIB_API
- void luaL_addvalue(luaL_Buffer*B){lua_State*L=B->L;size_t vl=lua_strlen(L,-1);
- if(vl<=bufffree(B)){memcpy(B->p,lua_tostring(L,-1),vl);B->p+=vl;lua_pop(L,1);}
- else{if(emptybuffer(B))lua_insert(L,-2);B->lvl++;adjuststack(B);}}LUALIB_API
- void luaL_buffinit(lua_State*L,luaL_Buffer*B){B->L=L;B->p=B->buffer;B->lvl=0;}
- LUALIB_API int luaL_ref(lua_State*L,int t){int ref;t=abs_index(L,t);if(
- lua_isnil(L,-1)){lua_pop(L,1);return LUA_REFNIL;}lua_rawgeti(L,t,FREELIST_REF)
- ;ref=(int)lua_tonumber(L,-1);lua_pop(L,1);if(ref!=0){lua_rawgeti(L,t,ref);
- lua_rawseti(L,t,FREELIST_REF);}else{ref=luaL_getn(L,t);if(ref<RESERVED_REFS)
- ref=RESERVED_REFS;ref++;luaL_setn(L,t,ref);}lua_rawseti(L,t,ref);return ref;}
- LUALIB_API void luaL_unref(lua_State*L,int t,int ref){if(ref>=0){t=abs_index(L
- ,t);lua_rawgeti(L,t,FREELIST_REF);lua_rawseti(L,t,ref);lua_pushnumber(L,(
- lua_Number)ref);lua_rawseti(L,t,FREELIST_REF);}}typedef struct LoadF{FILE*f;
- char buff[LUAL_BUFFERSIZE];}LoadF;static const char*getF(lua_State*L,void*ud,
- size_t*size){LoadF*lf=(LoadF*)ud;(void)L;if(feof(lf->f))return NULL;*size=
- fread(lf->buff,1,LUAL_BUFFERSIZE,lf->f);return(*size>0)?lf->buff:NULL;}static
- int errfile(lua_State*L,int fnameindex){const char*filename=lua_tostring(L,
- fnameindex)+1;lua_pushfstring(L,"cannot read %s: %s",filename,strerror(errno))
- ;lua_remove(L,fnameindex);return LUA_ERRFILE;}LUALIB_API int luaL_loadfile(
- lua_State*L,const char*filename){LoadF lf;int status,readstatus;int c;int
- fnameindex=lua_gettop(L)+1;if(filename==NULL){lua_pushliteral(L,"=stdin");lf.f
- =stdin;}else{lua_pushfstring(L,"@%s",filename);lf.f=fopen(filename,"r");}if(lf
- .f==NULL)return errfile(L,fnameindex);c=ungetc(getc(lf.f),lf.f);if(!(isspace(c
- )||isprint(c))&&lf.f!=stdin){fclose(lf.f);lf.f=fopen(filename,"rb");if(lf.f==
- NULL)return errfile(L,fnameindex);}status=lua_load(L,getF,&lf,lua_tostring(L,-
- 1));readstatus=ferror(lf.f);if(lf.f!=stdin)fclose(lf.f);if(readstatus){
- lua_settop(L,fnameindex);return errfile(L,fnameindex);}lua_remove(L,fnameindex
- );return status;}typedef struct LoadS{const char*s;size_t size;}LoadS;static
- const char*getS(lua_State*L,void*ud,size_t*size){LoadS*ls=(LoadS*)ud;(void)L;
- if(ls->size==0)return NULL;*size=ls->size;ls->size=0;return ls->s;}LUALIB_API
- int luaL_loadbuffer(lua_State*L,const char*buff,size_t size,const char*name){
- LoadS ls;ls.s=buff;ls.size=size;return lua_load(L,getS,&ls,name);}static void
- callalert(lua_State*L,int status){if(status!=0){lua_getglobal(L,"_ALERT");if(
- lua_isfunction(L,-1)){lua_insert(L,-2);lua_call(L,1,0);}else{fprintf(stderr,
- "%s\n",lua_tostring(L,-2));lua_pop(L,2);}}}static int aux_do(lua_State*L,int
- status){if(status==0){status=lua_pcall(L,0,LUA_MULTRET,0);}callalert(L,status)
- ;return status;}LUALIB_API int lua_dofile(lua_State*L,const char*filename){
- return aux_do(L,luaL_loadfile(L,filename));}LUALIB_API int lua_dobuffer(
- lua_State*L,const char*buff,size_t size,const char*name){return aux_do(L,
- luaL_loadbuffer(L,buff,size,name));}LUALIB_API int lua_dostring(lua_State*L,
- const char*str){return lua_dobuffer(L,str,strlen(str),str);}
- #line 1 "lbaselib.c"
- #define lbaselib_c
- #line 1 "lualib.h"
- #ifndef lualib_h
- #define lualib_h
- #ifndef LUALIB_API
- #define LUALIB_API LUA_API
- #endif
- #define LUA_COLIBNAME "coroutine"
- LUALIB_API int luaopen_base(lua_State*L);
- #define LUA_TABLIBNAME "table"
- LUALIB_API int luaopen_table(lua_State*L);
- #define LUA_IOLIBNAME "io"
- #define LUA_OSLIBNAME "os"
- LUALIB_API int luaopen_io(lua_State*L);
- #define LUA_STRLIBNAME "string"
- LUALIB_API int luaopen_string(lua_State*L);
- #define LUA_MATHLIBNAME "math"
- LUALIB_API int luaopen_math(lua_State*L);
- #define LUA_DBLIBNAME "debug"
- LUALIB_API int luaopen_debug(lua_State*L);LUALIB_API int luaopen_loadlib(
- lua_State*L);
- #ifndef lua_assert
- #define lua_assert(c)
- #endif
- #define lua_baselibopen luaopen_base
- #define lua_tablibopen luaopen_table
- #define lua_iolibopen luaopen_io
- #define lua_strlibopen luaopen_string
- #define lua_mathlibopen luaopen_math
- #define lua_dblibopen luaopen_debug
- #endif
- #line 20 "lbaselib.c"
- static int luaB_print(lua_State*L){int n=lua_gettop(L);int i;lua_getglobal(L,
- "tostring");for(i=1;i<=n;i++){const char*s;lua_pushvalue(L,-1);lua_pushvalue(L
- ,i);lua_call(L,1,1);s=lua_tostring(L,-1);if(s==NULL)return luaL_error(L,
- "`tostring' must return a string to `print'");if(i>1)fputs("\t",stdout);fputs(
- s,stdout);lua_pop(L,1);}fputs("\n",stdout);return 0;}static int luaB_tonumber(
- lua_State*L){int base=luaL_optint(L,2,10);if(base==10){luaL_checkany(L,1);if(
- lua_isnumber(L,1)){lua_pushnumber(L,lua_tonumber(L,1));return 1;}}else{const
- char*s1=luaL_checkstring(L,1);char*s2;unsigned long n;luaL_argcheck(L,2<=base
- &&base<=36,2,"base out of range");n=strtoul(s1,&s2,base);if(s1!=s2){while(
- isspace((unsigned char)(*s2)))s2++;if(*s2=='\0'){lua_pushnumber(L,(lua_Number)
- n);return 1;}}}lua_pushnil(L);return 1;}static int luaB_error(lua_State*L){int
- level=luaL_optint(L,2,1);luaL_checkany(L,1);if(!lua_isstring(L,1)||level==0)
- lua_pushvalue(L,1);else{luaL_where(L,level);lua_pushvalue(L,1);lua_concat(L,2)
- ;}return lua_error(L);}static int luaB_getmetatable(lua_State*L){luaL_checkany
- (L,1);if(!lua_getmetatable(L,1)){lua_pushnil(L);return 1;}luaL_getmetafield(L,
- 1,"__metatable");return 1;}static int luaB_setmetatable(lua_State*L){int t=
- lua_type(L,2);luaL_checktype(L,1,LUA_TTABLE);luaL_argcheck(L,t==LUA_TNIL||t==
- LUA_TTABLE,2,"nil or table expected");if(luaL_getmetafield(L,1,"__metatable"))
- luaL_error(L,"cannot change a protected metatable");lua_settop(L,2);
- lua_setmetatable(L,1);return 1;}static void getfunc(lua_State*L){if(
- lua_isfunction(L,1))lua_pushvalue(L,1);else{lua_Debug ar;int level=luaL_optint
- (L,1,1);luaL_argcheck(L,level>=0,1,"level must be non-negative");if(
- lua_getstack(L,level,&ar)==0)luaL_argerror(L,1,"invalid level");lua_getinfo(L,
- "f",&ar);if(lua_isnil(L,-1))luaL_error(L,
- "no function environment for tail call at level %d",level);}}static int
- aux_getfenv(lua_State*L){lua_getfenv(L,-1);lua_pushliteral(L,"__fenv");
- lua_rawget(L,-2);return!lua_isnil(L,-1);}static int luaB_getfenv(lua_State*L){
- getfunc(L);if(!aux_getfenv(L))lua_pop(L,1);return 1;}static int luaB_setfenv(
- lua_State*L){luaL_checktype(L,2,LUA_TTABLE);getfunc(L);if(aux_getfenv(L))
- luaL_error(L,"`setfenv' cannot change a protected environment");else lua_pop(L
- ,2);lua_pushvalue(L,2);if(lua_isnumber(L,1)&&lua_tonumber(L,1)==0)lua_replace(
- L,LUA_GLOBALSINDEX);else if(lua_setfenv(L,-2)==0)luaL_error(L,
- "`setfenv' cannot change environment of given function");return 0;}static int
- luaB_rawequal(lua_State*L){luaL_checkany(L,1);luaL_checkany(L,2);
- lua_pushboolean(L,lua_rawequal(L,1,2));return 1;}static int luaB_rawget(
- lua_State*L){luaL_checktype(L,1,LUA_TTABLE);luaL_checkany(L,2);lua_rawget(L,1)
- ;return 1;}static int luaB_rawset(lua_State*L){luaL_checktype(L,1,LUA_TTABLE);
- luaL_checkany(L,2);luaL_checkany(L,3);lua_rawset(L,1);return 1;}static int
- luaB_gcinfo(lua_State*L){lua_pushnumber(L,(lua_Number)lua_getgccount(L));
- lua_pushnumber(L,(lua_Number)lua_getgcthreshold(L));return 2;}static int
- luaB_collectgarbage(lua_State*L){lua_setgcthreshold(L,luaL_optint(L,1,0));
- return 0;}static int luaB_type(lua_State*L){luaL_checkany(L,1);lua_pushstring(
- L,lua_typename(L,lua_type(L,1)));return 1;}static int luaB_next(lua_State*L){
- luaL_checktype(L,1,LUA_TTABLE);lua_settop(L,2);if(lua_next(L,1))return 2;else{
- lua_pushnil(L);return 1;}}static int luaB_pairs(lua_State*L){luaL_checktype(L,
- 1,LUA_TTABLE);lua_pushliteral(L,"next");lua_rawget(L,LUA_GLOBALSINDEX);
- lua_pushvalue(L,1);lua_pushnil(L);return 3;}static int luaB_ipairs(lua_State*L
- ){lua_Number i=lua_tonumber(L,2);luaL_checktype(L,1,LUA_TTABLE);if(i==0&&
- lua_isnone(L,2)){lua_pushliteral(L,"ipairs");lua_rawget(L,LUA_GLOBALSINDEX);
- lua_pushvalue(L,1);lua_pushnumber(L,0);return 3;}else{i++;lua_pushnumber(L,i);
- lua_rawgeti(L,1,(int)i);return(lua_isnil(L,-1))?0:2;}}static int load_aux(
- lua_State*L,int status){if(status==0)return 1;else{lua_pushnil(L);lua_insert(L
- ,-2);return 2;}}static int luaB_loadstring(lua_State*L){size_t l;const char*s=
- luaL_checklstring(L,1,&l);const char*chunkname=luaL_optstring(L,2,s);return
- load_aux(L,luaL_loadbuffer(L,s,l,chunkname));}static int luaB_loadfile(
- lua_State*L){const char*fname=luaL_optstring(L,1,NULL);return load_aux(L,
- luaL_loadfile(L,fname));}static int luaB_dofile(lua_State*L){const char*fname=
- luaL_optstring(L,1,NULL);int n=lua_gettop(L);int status=luaL_loadfile(L,fname)
- ;if(status!=0)lua_error(L);lua_call(L,0,LUA_MULTRET);return lua_gettop(L)-n;}
- static int luaB_assert(lua_State*L){luaL_checkany(L,1);if(!lua_toboolean(L,1))
- return luaL_error(L,"%s",luaL_optstring(L,2,"assertion failed!"));lua_settop(L
- ,1);return 1;}static int luaB_unpack(lua_State*L){int n,i;luaL_checktype(L,1,
- LUA_TTABLE);n=luaL_getn(L,1);luaL_checkstack(L,n,"table too big to unpack");
- for(i=1;i<=n;i++)lua_rawgeti(L,1,i);return n;}static int luaB_pcall(lua_State*
- L){int status;luaL_checkany(L,1);status=lua_pcall(L,lua_gettop(L)-1,
- LUA_MULTRET,0);lua_pushboolean(L,(status==0));lua_insert(L,1);return
- lua_gettop(L);}static int luaB_xpcall(lua_State*L){int status;luaL_checkany(L,
- 2);lua_settop(L,2);lua_insert(L,1);status=lua_pcall(L,0,LUA_MULTRET,1);
- lua_pushboolean(L,(status==0));lua_replace(L,1);return lua_gettop(L);}static
- int luaB_tostring(lua_State*L){char buff[128];luaL_checkany(L,1);if(
- luaL_callmeta(L,1,"__tostring"))return 1;switch(lua_type(L,1)){case
- LUA_TNUMBER:lua_pushstring(L,lua_tostring(L,1));return 1;case LUA_TSTRING:
- lua_pushvalue(L,1);return 1;case LUA_TBOOLEAN:lua_pushstring(L,(lua_toboolean(
- L,1)?"true":"false"));return 1;case LUA_TTABLE:sprintf(buff,"table: %p",
- lua_topointer(L,1));break;case LUA_TFUNCTION:sprintf(buff,"function: %p",
- lua_topointer(L,1));break;case LUA_TUSERDATA:case LUA_TLIGHTUSERDATA:sprintf(
- buff,"userdata: %p",lua_touserdata(L,1));break;case LUA_TTHREAD:sprintf(buff,
- "thread: %p",(void*)lua_tothread(L,1));break;case LUA_TNIL:lua_pushliteral(L,
- "nil");return 1;}lua_pushstring(L,buff);return 1;}static int luaB_newproxy(
- lua_State*L){lua_settop(L,1);lua_newuserdata(L,0);if(lua_toboolean(L,1)==0)
- return 1;else if(lua_isboolean(L,1)){lua_newtable(L);lua_pushvalue(L,-1);
- lua_pushboolean(L,1);lua_rawset(L,lua_upvalueindex(1));}else{int validproxy=0;
- if(lua_getmetatable(L,1)){lua_rawget(L,lua_upvalueindex(1));validproxy=
- lua_toboolean(L,-1);lua_pop(L,1);}luaL_argcheck(L,validproxy,1,
- "boolean or proxy expected");lua_getmetatable(L,1);}lua_setmetatable(L,2);
- return 1;}
- #define REQTAB "_LOADED"
- #define LUA_PATH "LUA_PATH"
- #ifndef LUA_PATH_SEP
- #define LUA_PATH_SEP ';'
- #endif
- #ifndef LUA_PATH_MARK
- #define LUA_PATH_MARK '?'
- #endif
- #ifndef LUA_PATH_DEFAULT
- #define LUA_PATH_DEFAULT "?;?.lua"
- #endif
- static const char*getpath(lua_State*L){const char*path;lua_getglobal(L,
- LUA_PATH);path=lua_tostring(L,-1);lua_pop(L,1);if(path)return path;path=getenv
- (LUA_PATH);if(path)return path;return LUA_PATH_DEFAULT;}static const char*
- pushnextpath(lua_State*L,const char*path){const char*l;if(*path=='\0')return
- NULL;if(*path==LUA_PATH_SEP)path++;l=strchr(path,LUA_PATH_SEP);if(l==NULL)l=
- path+strlen(path);lua_pushlstring(L,path,l-path);return l;}static void
- pushcomposename(lua_State*L){const char*path=lua_tostring(L,-1);const char*
- wild;int n=1;while((wild=strchr(path,LUA_PATH_MARK))!=NULL){luaL_checkstack(L,
- 3,"too many marks in a path component");lua_pushlstring(L,path,wild-path);
- lua_pushvalue(L,1);path=wild+1;n+=2;}lua_pushstring(L,path);lua_concat(L,n);}
- static int luaB_require(lua_State*L){const char*path;int status=LUA_ERRFILE;
- luaL_checkstring(L,1);lua_settop(L,1);lua_getglobal(L,REQTAB);if(!lua_istable(
- L,2))return luaL_error(L,"`"REQTAB"' is not a table");path=getpath(L);
- lua_pushvalue(L,1);lua_rawget(L,2);if(lua_toboolean(L,-1))return 1;else{while(
- status==LUA_ERRFILE){lua_settop(L,3);if((path=pushnextpath(L,path))==NULL)
- break;pushcomposename(L);status=luaL_loadfile(L,lua_tostring(L,-1));}}switch(
- status){case 0:{lua_getglobal(L,"_REQUIREDNAME");lua_insert(L,-2);
- lua_pushvalue(L,1);lua_setglobal(L,"_REQUIREDNAME");lua_call(L,0,1);lua_insert
- (L,-2);lua_setglobal(L,"_REQUIREDNAME");if(lua_isnil(L,-1)){lua_pushboolean(L,
- 1);lua_replace(L,-2);}lua_pushvalue(L,1);lua_pushvalue(L,-2);lua_rawset(L,2);
- return 1;}case LUA_ERRFILE:{return luaL_error(L,
- "could not load package `%s' from path `%s'",lua_tostring(L,1),getpath(L));}
- default:{return luaL_error(L,"error loading package `%s' (%s)",lua_tostring(L,
- 1),lua_tostring(L,-1));}}}static const luaL_reg base_funcs[]={{"error",
- luaB_error},{"getmetatable",luaB_getmetatable},{"setmetatable",
- luaB_setmetatable},{"getfenv",luaB_getfenv},{"setfenv",luaB_setfenv},{"next",
- luaB_next},{"ipairs",luaB_ipairs},{"pairs",luaB_pairs},{"print",luaB_print},{
- "tonumber",luaB_tonumber},{"tostring",luaB_tostring},{"type",luaB_type},{
- "assert",luaB_assert},{"unpack",luaB_unpack},{"rawequal",luaB_rawequal},{
- "rawget",luaB_rawget},{"rawset",luaB_rawset},{"pcall",luaB_pcall},{"xpcall",
- luaB_xpcall},{"collectgarbage",luaB_collectgarbage},{"gcinfo",luaB_gcinfo},{
- "loadfile",luaB_loadfile},{"dofile",luaB_dofile},{"loadstring",luaB_loadstring
- },{"require",luaB_require},{NULL,NULL}};static int auxresume(lua_State*L,
- lua_State*co,int narg){int status;if(!lua_checkstack(co,narg))luaL_error(L,
- "too many arguments to resume");lua_xmove(L,co,narg);status=lua_resume(co,narg
- );if(status==0){int nres=lua_gettop(co);if(!lua_checkstack(L,nres))luaL_error(
- L,"too many results to resume");lua_xmove(co,L,nres);return nres;}else{
- lua_xmove(co,L,1);return-1;}}static int luaB_coresume(lua_State*L){lua_State*
- co=lua_tothread(L,1);int r;luaL_argcheck(L,co,1,"coroutine expected");r=
- auxresume(L,co,lua_gettop(L)-1);if(r<0){lua_pushboolean(L,0);lua_insert(L,-2);
- return 2;}else{lua_pushboolean(L,1);lua_insert(L,-(r+1));return r+1;}}static
- int luaB_auxwrap(lua_State*L){lua_State*co=lua_tothread(L,lua_upvalueindex(1))
- ;int r=auxresume(L,co,lua_gettop(L));if(r<0){if(lua_isstring(L,-1)){luaL_where
- (L,1);lua_insert(L,-2);lua_concat(L,2);}lua_error(L);}return r;}static int
- luaB_cocreate(lua_State*L){lua_State*NL=lua_newthread(L);luaL_argcheck(L,
- lua_isfunction(L,1)&&!lua_iscfunction(L,1),1,"Lua function expected");
- lua_pushvalue(L,1);lua_xmove(L,NL,1);return 1;}static int luaB_cowrap(
- lua_State*L){luaB_cocreate(L);lua_pushcclosure(L,luaB_auxwrap,1);return 1;}
- static int luaB_yield(lua_State*L){return lua_yield(L,lua_gettop(L));}static
- int luaB_costatus(lua_State*L){lua_State*co=lua_tothread(L,1);luaL_argcheck(L,
- co,1,"coroutine expected");if(L==co)lua_pushliteral(L,"running");else{
- lua_Debug ar;if(lua_getstack(co,0,&ar)==0&&lua_gettop(co)==0)lua_pushliteral(L
- ,"dead");else lua_pushliteral(L,"suspended");}return 1;}static const luaL_reg
- co_funcs[]={{"create",luaB_cocreate},{"wrap",luaB_cowrap},{"resume",
- luaB_coresume},{"yield",luaB_yield},{"status",luaB_costatus},{NULL,NULL}};
- static void base_open(lua_State*L){lua_pushliteral(L,"_G");lua_pushvalue(L,
- LUA_GLOBALSINDEX);luaL_openlib(L,NULL,base_funcs,0);lua_pushliteral(L,
- "_VERSION");lua_pushliteral(L,LUA_VERSION);lua_rawset(L,-3);lua_pushliteral(L,
- "newproxy");lua_newtable(L);lua_pushvalue(L,-1);lua_setmetatable(L,-2);
- lua_pushliteral(L,"__mode");lua_pushliteral(L,"k");lua_rawset(L,-3);
- lua_pushcclosure(L,luaB_newproxy,1);lua_rawset(L,-3);lua_rawset(L,-1);}
- LUALIB_API int luaopen_base(lua_State*L){base_open(L);luaL_openlib(L,
- LUA_COLIBNAME,co_funcs,0);lua_newtable(L);lua_setglobal(L,REQTAB);return 0;}
- #line 1 "lcode.c"
- #define lcode_c
- #line 1 "lcode.h"
- #ifndef lcode_h
- #define lcode_h
- #line 1 "llex.h"
- #ifndef llex_h
- #define llex_h
- #define FIRST_RESERVED 257
- #define TOKEN_LEN (sizeof("function")/sizeof(char))
- enum RESERVED{TK_AND=FIRST_RESERVED,TK_BREAK,TK_DO,TK_ELSE,TK_ELSEIF,TK_END,
- TK_FALSE,TK_FOR,TK_FUNCTION,TK_IF,TK_IN,TK_LOCAL,TK_NIL,TK_NOT,TK_OR,TK_REPEAT
- ,TK_RETURN,TK_THEN,TK_TRUE,TK_UNTIL,TK_WHILE,TK_NAME,TK_CONCAT,TK_DOTS,TK_EQ,
- TK_GE,TK_LE,TK_NE,TK_NUMBER,TK_STRING,TK_EOS};
- #define NUM_RESERVED (cast(int,TK_WHILE-FIRST_RESERVED+1))
- typedef union{lua_Number r;TString*ts;}SemInfo;typedef struct Token{int token;
- SemInfo seminfo;}Token;typedef struct LexState{int current;int linenumber;int
- lastline;Token t;Token lookahead;struct FuncState*fs;struct lua_State*L;ZIO*z;
- Mbuffer*buff;TString*source;int nestlevel;}LexState;void luaX_init(lua_State*L
- );void luaX_setinput(lua_State*L,LexState*LS,ZIO*z,TString*source);int
- luaX_lex(LexState*LS,SemInfo*seminfo);void luaX_checklimit(LexState*ls,int val
- ,int limit,const char*msg);void luaX_syntaxerror(LexState*ls,const char*s);
- void luaX_errorline(LexState*ls,const char*s,const char*token,int line);const
- char*luaX_token2str(LexState*ls,int token);
- #endif
- #line 11 "lcode.h"
- #line 1 "lopcodes.h"
- #ifndef lopcodes_h
- #define lopcodes_h
- enum OpMode{iABC,iABx,iAsBx};
- #define SIZE_C 9
- #define SIZE_B 9
- #define SIZE_Bx (SIZE_C+SIZE_B)
- #define SIZE_A 8
- #define SIZE_OP 6
- #define POS_C SIZE_OP
- #define POS_B (POS_C+SIZE_C)
- #define POS_Bx POS_C
- #define POS_A (POS_B+SIZE_B)
- #if SIZE_Bx <BITS_INT-1
- #define MAXARG_Bx ((1<<SIZE_Bx)-1)
- #define MAXARG_sBx (MAXARG_Bx>>1)
- #else
- #define MAXARG_Bx MAX_INT
- #define MAXARG_sBx MAX_INT
- #endif
- #define MAXARG_A ((1<<SIZE_A)-1)
- #define MAXARG_B ((1<<SIZE_B)-1)
- #define MAXARG_C ((1<<SIZE_C)-1)
- #define MASK1(n,p) ((~((~(Instruction)0)<<n))<<p)
- #define MASK0(n,p) (~MASK1(n,p))
- #define GET_OPCODE(i) (cast(OpCode,(i)&MASK1(SIZE_OP,0)))
- #define SET_OPCODE(i,o) ((i)=(((i)&MASK0(SIZE_OP,0))|cast(Instruction,o)))
- #define GETARG_A(i) (cast(int,(i)>>POS_A))
- #define SETARG_A(i,u) ((i)=(((i)&MASK0(SIZE_A,POS_A))|((cast(Instruction,u)<<\
- POS_A)&MASK1(SIZE_A,POS_A))))
- #define GETARG_B(i) (cast(int,((i)>>POS_B)&MASK1(SIZE_B,0)))
- #define SETARG_B(i,b) ((i)=(((i)&MASK0(SIZE_B,POS_B))|((cast(Instruction,b)<<\
- POS_B)&MASK1(SIZE_B,POS_B))))
- #define GETARG_C(i) (cast(int,((i)>>POS_C)&MASK1(SIZE_C,0)))
- #define SETARG_C(i,b) ((i)=(((i)&MASK0(SIZE_C,POS_C))|((cast(Instruction,b)<<\
- POS_C)&MASK1(SIZE_C,POS_C))))
- #define GETARG_Bx(i) (cast(int,((i)>>POS_Bx)&MASK1(SIZE_Bx,0)))
- #define SETARG_Bx(i,b) ((i)=(((i)&MASK0(SIZE_Bx,POS_Bx))|((cast(Instruction,b)\
- <<POS_Bx)&MASK1(SIZE_Bx,POS_Bx))))
- #define GETARG_sBx(i) (GETARG_Bx(i)-MAXARG_sBx)
- #define SETARG_sBx(i,b) SETARG_Bx((i),cast(unsigned int,(b)+MAXARG_sBx))
- #define CREATE_ABC(o,a,b,c) (cast(Instruction,o)|(cast(Instruction,a)<<POS_A)|\
- (cast(Instruction,b)<<POS_B)|(cast(Instruction,c)<<POS_C))
- #define CREATE_ABx(o,a,bc) (cast(Instruction,o)|(cast(Instruction,a)<<POS_A)|(\
- cast(Instruction,bc)<<POS_Bx))
- #define NO_REG MAXARG_A
- typedef enum{OP_MOVE,OP_LOADK,OP_LOADBOOL,OP_LOADNIL,OP_GETUPVAL,OP_GETGLOBAL,
- OP_GETTABLE,OP_SETGLOBAL,OP_SETUPVAL,OP_SETTABLE,OP_NEWTABLE,OP_SELF,OP_ADD,
- OP_SUB,OP_MUL,OP_DIV,OP_POW,OP_UNM,OP_NOT,OP_CONCAT,OP_JMP,OP_EQ,OP_LT,OP_LE,
- OP_TEST,OP_CALL,OP_TAILCALL,OP_RETURN,OP_FORLOOP,OP_TFORLOOP,OP_TFORPREP,
- OP_SETLIST,OP_SETLISTO,OP_CLOSE,OP_CLOSURE}OpCode;
- #define NUM_OPCODES (cast(int,OP_CLOSURE+1))
- enum OpModeMask{OpModeBreg=2,OpModeBrk,OpModeCrk,OpModesetA,OpModeK,OpModeT};
- extern const lu_byte luaP_opmodes[NUM_OPCODES];
- #define getOpMode(m) (cast(enum OpMode,luaP_opmodes[m]&3))
- #define testOpMode(m, b)(luaP_opmodes[m]&(1<<(b)))
- #ifdef LUA_OPNAMES
- extern const char*const luaP_opnames[];
- #endif
- #define LFIELDS_PER_FLUSH 32
- #endif
- #line 13 "lcode.h"
- #line 1 "lparser.h"
- #ifndef lparser_h
- #define lparser_h
- typedef enum{VVOID,VNIL,VTRUE,VFALSE,VK,VLOCAL,VUPVAL,VGLOBAL,VINDEXED,VJMP,
- VRELOCABLE,VNONRELOC,VCALL}expkind;typedef struct expdesc{expkind k;int info,
- aux;int t;int f;}expdesc;struct BlockCnt;typedef struct FuncState{Proto*f;
- Table*h;struct FuncState*prev;struct LexState*ls;struct lua_State*L;struct
- BlockCnt*bl;int pc;int lasttarget;int jpc;int freereg;int nk;int np;int
- nlocvars;int nactvar;expdesc upvalues[MAXUPVALUES];int actvar[MAXVARS];}
- FuncState;Proto*luaY_parser(lua_State*L,ZIO*z,Mbuffer*buff);
- #endif
- #line 14 "lcode.h"
- #define NO_JUMP (-1)
- typedef enum BinOpr{OPR_ADD,OPR_SUB,OPR_MULT,OPR_DIV,OPR_POW,OPR_CONCAT,OPR_NE
- ,OPR_EQ,OPR_LT,OPR_LE,OPR_GT,OPR_GE,OPR_AND,OPR_OR,OPR_NOBINOPR}BinOpr;
- #define binopistest(op) ((op)>=OPR_NE)
- typedef enum UnOpr{OPR_MINUS,OPR_NOT,OPR_NOUNOPR}UnOpr;
- #define getcode(fs,e) ((fs)->f->code[(e)->info])
- #define luaK_codeAsBx(fs,o,A,sBx) luaK_codeABx(fs,o,A,(sBx)+MAXARG_sBx)
- int luaK_code(FuncState*fs,Instruction i,int line);int luaK_codeABx(FuncState*
- fs,OpCode o,int A,unsigned int Bx);int luaK_codeABC(FuncState*fs,OpCode o,int
- A,int B,int C);void luaK_fixline(FuncState*fs,int line);void luaK_nil(
- FuncState*fs,int from,int n);void luaK_reserveregs(FuncState*fs,int n);void
- luaK_checkstack(FuncState*fs,int n);int luaK_stringK(FuncState*fs,TString*s);
- int luaK_numberK(FuncState*fs,lua_Number r);void luaK_dischargevars(FuncState*
- fs,expdesc*e);int luaK_exp2anyreg(FuncState*fs,expdesc*e);void
- luaK_exp2nextreg(FuncState*fs,expdesc*e);void luaK_exp2val(FuncState*fs,
- expdesc*e);int luaK_exp2RK(FuncState*fs,expdesc*e);void luaK_self(FuncState*fs
- ,expdesc*e,expdesc*key);void luaK_indexed(FuncState*fs,expdesc*t,expdesc*k);
- void luaK_goiftrue(FuncState*fs,expdesc*e);void luaK_goiffalse(FuncState*fs,
- expdesc*e);void luaK_storevar(FuncState*fs,expdesc*var,expdesc*e);void
- luaK_setcallreturns(FuncState*fs,expdesc*var,int nresults);int luaK_jump(
- FuncState*fs);void luaK_patchlist(FuncState*fs,int list,int target);void
- luaK_patchtohere(FuncState*fs,int list);void luaK_concat(FuncState*fs,int*l1,
- int l2);int luaK_getlabel(FuncState*fs);void luaK_prefix(FuncState*fs,UnOpr op
- ,expdesc*v);void luaK_infix(FuncState*fs,BinOpr op,expdesc*v);void luaK_posfix
- (FuncState*fs,BinOpr op,expdesc*v1,expdesc*v2);
- #endif
- #line 15 "lcode.c"
- #define hasjumps(e) ((e)->t!=(e)->f)
- void luaK_nil(FuncState*fs,int from,int n){Instruction*previous;if(fs->pc>fs->
- lasttarget&&GET_OPCODE(*(previous=&fs->f->code[fs->pc-1]))==OP_LOADNIL){int
- pfrom=GETARG_A(*previous);int pto=GETARG_B(*previous);if(pfrom<=from&&from<=
- pto+1){if(from+n-1>pto)SETARG_B(*previous,from+n-1);return;}}luaK_codeABC(fs,
- OP_LOADNIL,from,from+n-1,0);}int luaK_jump(FuncState*fs){int jpc=fs->jpc;int j
- ;fs->jpc=NO_JUMP;j=luaK_codeAsBx(fs,OP_JMP,0,NO_JUMP);luaK_concat(fs,&j,jpc);
- return j;}static int luaK_condjump(FuncState*fs,OpCode op,int A,int B,int C){
- luaK_codeABC(fs,op,A,B,C);return luaK_jump(fs);}static void luaK_fixjump(
- FuncState*fs,int pc,int dest){Instruction*jmp=&fs->f->code[pc];int offset=dest
- -(pc+1);lua_assert(dest!=NO_JUMP);if(abs(offset)>MAXARG_sBx)luaX_syntaxerror(
- fs->ls,"control structure too long");SETARG_sBx(*jmp,offset);}int
- luaK_getlabel(FuncState*fs){fs->lasttarget=fs->pc;return fs->pc;}static int
- luaK_getjump(FuncState*fs,int pc){int offset=GETARG_sBx(fs->f->code[pc]);if(
- offset==NO_JUMP)return NO_JUMP;else return(pc+1)+offset;}static Instruction*
- getjumpcontrol(FuncState*fs,int pc){Instruction*pi=&fs->f->code[pc];if(pc>=1&&
- testOpMode(GET_OPCODE(*(pi-1)),OpModeT))return pi-1;else return pi;}static int
- need_value(FuncState*fs,int list,int cond){for(;list!=NO_JUMP;list=
- luaK_getjump(fs,list)){Instruction i=*getjumpcontrol(fs,list);if(GET_OPCODE(i)
- !=OP_TEST||GETARG_C(i)!=cond)return 1;}return 0;}static void patchtestreg(
- Instruction*i,int reg){if(reg==NO_REG)reg=GETARG_B(*i);SETARG_A(*i,reg);}
- static void luaK_patchlistaux(FuncState*fs,int list,int ttarget,int treg,int
- ftarget,int freg,int dtarget){while(list!=NO_JUMP){int next=luaK_getjump(fs,
- list);Instruction*i=getjumpcontrol(fs,list);if(GET_OPCODE(*i)!=OP_TEST){
- lua_assert(dtarget!=NO_JUMP);luaK_fixjump(fs,list,dtarget);}else{if(GETARG_C(*
- i)){lua_assert(ttarget!=NO_JUMP);patchtestreg(i,treg);luaK_fixjump(fs,list,
- ttarget);}else{lua_assert(ftarget!=NO_JUMP);patchtestreg(i,freg);luaK_fixjump(
- fs,list,ftarget);}}list=next;}}static void luaK_dischargejpc(FuncState*fs){
- luaK_patchlistaux(fs,fs->jpc,fs->pc,NO_REG,fs->pc,NO_REG,fs->pc);fs->jpc=
- NO_JUMP;}void luaK_patchlist(FuncState*fs,int list,int target){if(target==fs->
- pc)luaK_patchtohere(fs,list);else{lua_assert(target<fs->pc);luaK_patchlistaux(
- fs,list,target,NO_REG,target,NO_REG,target);}}void luaK_patchtohere(FuncState*
- fs,int list){luaK_getlabel(fs);luaK_concat(fs,&fs->jpc,list);}void luaK_concat
- (FuncState*fs,int*l1,int l2){if(l2==NO_JUMP)return;else if(*l1==NO_JUMP)*l1=l2
- ;else{int list=*l1;int next;while((next=luaK_getjump(fs,list))!=NO_JUMP)list=
- next;luaK_fixjump(fs,list,l2);}}void luaK_checkstack(FuncState*fs,int n){int
- newstack=fs->freereg+n;if(newstack>fs->f->maxstacksize){if(newstack>=MAXSTACK)
- luaX_syntaxerror(fs->ls,"function or expression too complex");fs->f->
- maxstacksize=cast(lu_byte,newstack);}}void luaK_reserveregs(FuncState*fs,int n
- ){luaK_checkstack(fs,n);fs->freereg+=n;}static void freereg(FuncState*fs,int
- reg){if(reg>=fs->nactvar&®<MAXSTACK){fs->freereg--;lua_assert(reg==fs->
- freereg);}}static void freeexp(FuncState*fs,expdesc*e){if(e->k==VNONRELOC)
- freereg(fs,e->info);}static int addk(FuncState*fs,TObject*k,TObject*v){const
- TObject*idx=luaH_get(fs->h,k);if(ttisnumber(idx)){lua_assert(luaO_rawequalObj(
- &fs->f->k[cast(int,nvalue(idx))],v));return cast(int,nvalue(idx));}else{Proto*
- f=fs->f;luaM_growvector(fs->L,f->k,fs->nk,f->sizek,TObject,MAXARG_Bx,
- "constant table overflow");setobj2n(&f->k[fs->nk],v);setnvalue(luaH_set(fs->L,
- fs->h,k),cast(lua_Number,fs->nk));return fs->nk++;}}int luaK_stringK(FuncState
- *fs,TString*s){TObject o;setsvalue(&o,s);return addk(fs,&o,&o);}int
- luaK_numberK(FuncState*fs,lua_Number r){TObject o;setnvalue(&o,r);return addk(
- fs,&o,&o);}static int nil_constant(FuncState*fs){TObject k,v;setnilvalue(&v);
- sethvalue(&k,fs->h);return addk(fs,&k,&v);}void luaK_setcallreturns(FuncState*
- fs,expdesc*e,int nresults){if(e->k==VCALL){SETARG_C(getcode(fs,e),nresults+1);
- if(nresults==1){e->k=VNONRELOC;e->info=GETARG_A(getcode(fs,e));}}}void
- luaK_dischargevars(FuncState*fs,expdesc*e){switch(e->k){case VLOCAL:{e->k=
- VNONRELOC;break;}case VUPVAL:{e->info=luaK_codeABC(fs,OP_GETUPVAL,0,e->info,0)
- ;e->k=VRELOCABLE;break;}case VGLOBAL:{e->info=luaK_codeABx(fs,OP_GETGLOBAL,0,e
- ->info);e->k=VRELOCABLE;break;}case VINDEXED:{freereg(fs,e->aux);freereg(fs,e
- ->info);e->info=luaK_codeABC(fs,OP_GETTABLE,0,e->info,e->aux);e->k=VRELOCABLE;
- break;}case VCALL:{luaK_setcallreturns(fs,e,1);break;}default:break;}}static
- int code_label(FuncState*fs,int A,int b,int jump){luaK_getlabel(fs);return
- luaK_codeABC(fs,OP_LOADBOOL,A,b,jump);}static void discharge2reg(FuncState*fs,
- expdesc*e,int reg){luaK_dischargevars(fs,e);switch(e->k){case VNIL:{luaK_nil(
- fs,reg,1);break;}case VFALSE:case VTRUE:{luaK_codeABC(fs,OP_LOADBOOL,reg,e->k
- ==VTRUE,0);break;}case VK:{luaK_codeABx(fs,OP_LOADK,reg,e->info);break;}case
- VRELOCABLE:{Instruction*pc=&getcode(fs,e);SETARG_A(*pc,reg);break;}case
- VNONRELOC:{if(reg!=e->info)luaK_codeABC(fs,OP_MOVE,reg,e->info,0);break;}
- default:{lua_assert(e->k==VVOID||e->k==VJMP);return;}}e->info=reg;e->k=
- VNONRELOC;}static void discharge2anyreg(FuncState*fs,expdesc*e){if(e->k!=
- VNONRELOC){luaK_reserveregs(fs,1);discharge2reg(fs,e,fs->freereg-1);}}static
- void luaK_exp2reg(FuncState*fs,expdesc*e,int reg){discharge2reg(fs,e,reg);if(e
- ->k==VJMP)luaK_concat(fs,&e->t,e->info);if(hasjumps(e)){int final;int p_f=
- NO_JUMP;int p_t=NO_JUMP;if(need_value(fs,e->t,1)||need_value(fs,e->f,0)){int
- fj=NO_JUMP;if(e->k!=VJMP)fj=luaK_jump(fs);p_f=code_label(fs,reg,0,1);p_t=
- code_label(fs,reg,1,0);luaK_patchtohere(fs,fj);}final=luaK_getlabel(fs);
- luaK_patchlistaux(fs,e->f,p_f,NO_REG,final,reg,p_f);luaK_patchlistaux(fs,e->t,
- final,reg,p_t,NO_REG,p_t);}e->f=e->t=NO_JUMP;e->info=reg;e->k=VNONRELOC;}void
- luaK_exp2nextreg(FuncState*fs,expdesc*e){luaK_dischargevars(fs,e);freeexp(fs,e
- );luaK_reserveregs(fs,1);luaK_exp2reg(fs,e,fs->freereg-1);}int luaK_exp2anyreg
- (FuncState*fs,expdesc*e){luaK_dischargevars(fs,e);if(e->k==VNONRELOC){if(!
- hasjumps(e))return e->info;if(e->info>=fs->nactvar){luaK_exp2reg(fs,e,e->info)
- ;return e->info;}}luaK_exp2nextreg(fs,e);return e->info;}void luaK_exp2val(
- FuncState*fs,expdesc*e){if(hasjumps(e))luaK_exp2anyreg(fs,e);else
- luaK_dischargevars(fs,e);}int luaK_exp2RK(FuncState*fs,expdesc*e){luaK_exp2val
- (fs,e);switch(e->k){case VNIL:{if(fs->nk+MAXSTACK<=MAXARG_C){e->info=
- nil_constant(fs);e->k=VK;return e->info+MAXSTACK;}else break;}case VK:{if(e->
- info+MAXSTACK<=MAXARG_C)return e->info+MAXSTACK;else break;}default:break;}
- return luaK_exp2anyreg(fs,e);}void luaK_storevar(FuncState*fs,expdesc*var,
- expdesc*exp){switch(var->k){case VLOCAL:{freeexp(fs,exp);luaK_exp2reg(fs,exp,
- var->info);return;}case VUPVAL:{int e=luaK_exp2anyreg(fs,exp);luaK_codeABC(fs,
- OP_SETUPVAL,e,var->info,0);break;}case VGLOBAL:{int e=luaK_exp2anyreg(fs,exp);
- luaK_codeABx(fs,OP_SETGLOBAL,e,var->info);break;}case VINDEXED:{int e=
- luaK_exp2RK(fs,exp);luaK_codeABC(fs,OP_SETTABLE,var->info,var->aux,e);break;}
- default:{lua_assert(0);break;}}freeexp(fs,exp);}void luaK_self(FuncState*fs,
- expdesc*e,expdesc*key){int func;luaK_exp2anyreg(fs,e);freeexp(fs,e);func=fs->
- freereg;luaK_reserveregs(fs,2);luaK_codeABC(fs,OP_SELF,func,e->info,
- luaK_exp2RK(fs,key));freeexp(fs,key);e->info=func;e->k=VNONRELOC;}static void
- invertjump(FuncState*fs,expdesc*e){Instruction*pc=getjumpcontrol(fs,e->info);
- lua_assert(testOpMode(GET_OPCODE(*pc),OpModeT)&&GET_OPCODE(*pc)!=OP_TEST);
- SETARG_A(*pc,!(GETARG_A(*pc)));}static int jumponcond(FuncState*fs,expdesc*e,
- int cond){if(e->k==VRELOCABLE){Instruction ie=getcode(fs,e);if(GET_OPCODE(ie)
- ==OP_NOT){fs->pc--;return luaK_condjump(fs,OP_TEST,NO_REG,GETARG_B(ie),!cond);
- }}discharge2anyreg(fs,e);freeexp(fs,e);return luaK_condjump(fs,OP_TEST,NO_REG,
- e->info,cond);}void luaK_goiftrue(FuncState*fs,expdesc*e){int pc;
- luaK_dischargevars(fs,e);switch(e->k){case VK:case VTRUE:{pc=NO_JUMP;break;}
- case VFALSE:{pc=luaK_jump(fs);break;}case VJMP:{invertjump(fs,e);pc=e->info;
- break;}default:{pc=jumponcond(fs,e,0);break;}}luaK_concat(fs,&e->f,pc);}void
- luaK_goiffalse(FuncState*fs,expdesc*e){int pc;luaK_dischargevars(fs,e);switch(
- e->k){case VNIL:case VFALSE:{pc=NO_JUMP;break;}case VTRUE:{pc=luaK_jump(fs);
- break;}case VJMP:{pc=e->info;break;}default:{pc=jumponcond(fs,e,1);break;}}
- luaK_concat(fs,&e->t,pc);}static void codenot(FuncState*fs,expdesc*e){
- luaK_dischargevars(fs,e);switch(e->k){case VNIL:case VFALSE:{e->k=VTRUE;break;
- }case VK:case VTRUE:{e->k=VFALSE;break;}case VJMP:{invertjump(fs,e);break;}
- case VRELOCABLE:case VNONRELOC:{discharge2anyreg(fs,e);freeexp(fs,e);e->info=
- luaK_codeABC(fs,OP_NOT,0,e->info,0);e->k=VRELOCABLE;break;}default:{lua_assert
- (0);break;}}{int temp=e->f;e->f=e->t;e->t=temp;}}void luaK_indexed(FuncState*
- fs,expdesc*t,expdesc*k){t->aux=luaK_exp2RK(fs,k);t->k=VINDEXED;}void
- luaK_prefix(FuncState*fs,UnOpr op,expdesc*e){if(op==OPR_MINUS){luaK_exp2val(fs
- ,e);if(e->k==VK&&ttisnumber(&fs->f->k[e->info]))e->info=luaK_numberK(fs,-
- nvalue(&fs->f->k[e->info]));else{luaK_exp2anyreg(fs,e);freeexp(fs,e);e->info=
- luaK_codeABC(fs,OP_UNM,0,e->info,0);e->k=VRELOCABLE;}}else codenot(fs,e);}void
- luaK_infix(FuncState*fs,BinOpr op,expdesc*v){switch(op){case OPR_AND:{
- luaK_goiftrue(fs,v);luaK_patchtohere(fs,v->t);v->t=NO_JUMP;break;}case OPR_OR:
- {luaK_goiffalse(fs,v);luaK_patchtohere(fs,v->f);v->f=NO_JUMP;break;}case
- OPR_CONCAT:{luaK_exp2nextreg(fs,v);break;}default:{luaK_exp2RK(fs,v);break;}}}
- static void codebinop(FuncState*fs,expdesc*res,BinOpr op,int o1,int o2){if(op
- <=OPR_POW){OpCode opc=cast(OpCode,(op-OPR_ADD)+OP_ADD);res->info=luaK_codeABC(
- fs,opc,0,o1,o2);res->k=VRELOCABLE;}else{static const OpCode ops[]={OP_EQ,OP_EQ
- ,OP_LT,OP_LE,OP_LT,OP_LE};int cond=1;if(op>=OPR_GT){int temp;temp=o1;o1=o2;o2=
- temp;}else if(op==OPR_NE)cond=0;res->info=luaK_condjump(fs,ops[op-OPR_NE],cond
- ,o1,o2);res->k=VJMP;}}void luaK_posfix(FuncState*fs,BinOpr op,expdesc*e1,
- expdesc*e2){switch(op){case OPR_AND:{lua_assert(e1->t==NO_JUMP);
- luaK_dischargevars(fs,e2);luaK_concat(fs,&e1->f,e2->f);e1->k=e2->k;e1->info=e2
- ->info;e1->aux=e2->aux;e1->t=e2->t;break;}case OPR_OR:{lua_assert(e1->f==
- NO_JUMP);luaK_dischargevars(fs,e2);luaK_concat(fs,&e1->t,e2->t);e1->k=e2->k;e1
- ->info=e2->info;e1->aux=e2->aux;e1->f=e2->f;break;}case OPR_CONCAT:{
- luaK_exp2val(fs,e2);if(e2->k==VRELOCABLE&&GET_OPCODE(getcode(fs,e2))==
- OP_CONCAT){lua_assert(e1->info==GETARG_B(getcode(fs,e2))-1);freeexp(fs,e1);
- SETARG_B(getcode(fs,e2),e1->info);e1->k=e2->k;e1->info=e2->info;}else{
- luaK_exp2nextreg(fs,e2);freeexp(fs,e2);freeexp(fs,e1);e1->info=luaK_codeABC(fs
- ,OP_CONCAT,0,e1->info,e2->info);e1->k=VRELOCABLE;}break;}default:{int o1=
- luaK_exp2RK(fs,e1);int o2=luaK_exp2RK(fs,e2);freeexp(fs,e2);freeexp(fs,e1);
- codebinop(fs,e1,op,o1,o2);}}}void luaK_fixline(FuncState*fs,int line){fs->f->
- lineinfo[fs->pc-1]=line;}int luaK_code(FuncState*fs,Instruction i,int line){
- Proto*f=fs->f;luaK_dischargejpc(fs);luaM_growvector(fs->L,f->code,fs->pc,f->
- sizecode,Instruction,MAX_INT,"code size overflow");f->code[fs->pc]=i;
- luaM_growvector(fs->L,f->lineinfo,fs->pc,f->sizelineinfo,int,MAX_INT,
- "code size overflow");f->lineinfo[fs->pc]=line;return fs->pc++;}int
- luaK_codeABC(FuncState*fs,OpCode o,int a,int b,int c){lua_assert(getOpMode(o)
- ==iABC);return luaK_code(fs,CREATE_ABC(o,a,b,c),fs->ls->lastline);}int
- luaK_codeABx(FuncState*fs,OpCode o,int a,unsigned int bc){lua_assert(getOpMode
- (o)==iABx||getOpMode(o)==iAsBx);return luaK_code(fs,CREATE_ABx(o,a,bc),fs->ls
- ->lastline);}
- #line 1 "ldblib.c"
- #define ldblib_c
- static void settabss(lua_State*L,const char*i,const char*v){lua_pushstring(L,i
- );lua_pushstring(L,v);lua_rawset(L,-3);}static void settabsi(lua_State*L,const
- char*i,int v){lua_pushstring(L,i);lua_pushnumber(L,(lua_Number)v);lua_rawset(
- L,-3);}static int getinfo(lua_State*L){lua_Debug ar;const char*options=
- luaL_optstring(L,2,"flnSu");if(lua_isnumber(L,1)){if(!lua_getstack(L,(int)(
- lua_tonumber(L,1)),&ar)){lua_pushnil(L);return 1;}}else if(lua_isfunction(L,1)
- ){lua_pushfstring(L,">%s",options);options=lua_tostring(L,-1);lua_pushvalue(L,
- 1);}else return luaL_argerror(L,1,"function or level expected");if(!
- lua_getinfo(L,options,&ar))return luaL_argerror(L,2,"invalid option");
- lua_newtable(L);for(;*options;options++){switch(*options){case'S':settabss(L,
- "source",ar.source);settabss(L,"short_src",ar.short_src);settabsi(L,
- "linedefined",ar.linedefined);settabss(L,"what",ar.what);break;case'l':
- settabsi(L,"currentline",ar.currentline);break;case'u':settabsi(L,"nups",ar.
- nups);break;case'n':settabss(L,"name",ar.name);settabss(L,"namewhat",ar.
- namewhat);break;case'f':lua_pushliteral(L,"func");lua_pushvalue(L,-3);
- lua_rawset(L,-3);break;}}return 1;}static int getlocal(lua_State*L){lua_Debug
- ar;const char*name;if(!lua_getstack(L,luaL_checkint(L,1),&ar))return
- luaL_argerror(L,1,"level out of range");name=lua_getlocal(L,&ar,luaL_checkint(
- L,2));if(name){lua_pushstring(L,name);lua_pushvalue(L,-2);return 2;}else{
- lua_pushnil(L);return 1;}}static int setlocal(lua_State*L){lua_Debug ar;if(!
- lua_getstack(L,luaL_checkint(L,1),&ar))return luaL_argerror(L,1,
- "level out of range");luaL_checkany(L,3);lua_pushstring(L,lua_setlocal(L,&ar,
- luaL_checkint(L,2)));return 1;}static int auxupvalue(lua_State*L,int get){
- const char*name;int n=luaL_checkint(L,2);luaL_checktype(L,1,LUA_TFUNCTION);if(
- lua_iscfunction(L,1))return 0;name=get?lua_getupvalue(L,1,n):lua_setupvalue(L,
- 1,n);if(name==NULL)return 0;lua_pushstring(L,name);lua_insert(L,-(get+1));
- return get+1;}static int getupvalue(lua_State*L){return auxupvalue(L,1);}
- static int setupvalue(lua_State*L){luaL_checkany(L,3);return auxupvalue(L,0);}
- static const char KEY_HOOK='h';static void hookf(lua_State*L,lua_Debug*ar){
- static const char*const hooknames[]={"call","return","line","count",
- "tail return"};lua_pushlightuserdata(L,(void*)&KEY_HOOK);lua_rawget(L,
- LUA_REGISTRYINDEX);if(lua_isfunction(L,-1)){lua_pushstring(L,hooknames[(int)ar
- ->event]);if(ar->currentline>=0)lua_pushnumber(L,(lua_Number)ar->currentline);
- else lua_pushnil(L);lua_assert(lua_getinfo(L,"lS",ar));lua_call(L,2,0);}else
- lua_pop(L,1);}static int makemask(const char*smask,int count){int mask=0;if(
- strchr(smask,'c'))mask|=LUA_MASKCALL;if(strchr(smask,'r'))mask|=LUA_MASKRET;if
- (strchr(smask,'l'))mask|=LUA_MASKLINE;if(count>0)mask|=LUA_MASKCOUNT;return
- mask;}static char*unmakemask(int mask,char*smask){int i=0;if(mask&LUA_MASKCALL
- )smask[i++]='c';if(mask&LUA_MASKRET)smask[i++]='r';if(mask&LUA_MASKLINE)smask[
- i++]='l';smask[i]='\0';return smask;}static int sethook(lua_State*L){if(
- lua_isnoneornil(L,1)){lua_settop(L,1);lua_sethook(L,NULL,0,0);}else{const char
- *smask=luaL_checkstring(L,2);int count=luaL_optint(L,3,0);luaL_checktype(L,1,
- LUA_TFUNCTION);lua_sethook(L,hookf,makemask(smask,count),count);}
- lua_pushlightuserdata(L,(void*)&KEY_HOOK);lua_pushvalue(L,1);lua_rawset(L,
- LUA_REGISTRYINDEX);return 0;}static int gethook(lua_State*L){char buff[5];int
- mask=lua_gethookmask(L);lua_Hook hook=lua_gethook(L);if(hook!=NULL&&hook!=
- hookf)lua_pushliteral(L,"external hook");else{lua_pushlightuserdata(L,(void*)&
- KEY_HOOK);lua_rawget(L,LUA_REGISTRYINDEX);}lua_pushstring(L,unmakemask(mask,
- buff));lua_pushnumber(L,(lua_Number)lua_gethookcount(L));return 3;}static int
- debug(lua_State*L){for(;;){char buffer[250];fputs("lua_debug> ",stderr);if(
- fgets(buffer,sizeof(buffer),stdin)==0||strcmp(buffer,"cont\n")==0)return 0;
- lua_dostring(L,buffer);lua_settop(L,0);}}
- #define LEVELS1 12
- #define LEVELS2 10
- static int errorfb(lua_State*L){int level=1;int firstpart=1;lua_Debug ar;if(
- lua_gettop(L)==0)lua_pushliteral(L,"");else if(!lua_isstring(L,1))return 1;
- else lua_pushliteral(L,"\n");lua_pushliteral(L,"stack traceback:");while(
- lua_getstack(L,level++,&ar)){if(level>LEVELS1&&firstpart){if(!lua_getstack(L,
- level+LEVELS2,&ar))level--;else{lua_pushliteral(L,"\n\t...");while(
- lua_getstack(L,level+LEVELS2,&ar))level++;}firstpart=0;continue;}
- lua_pushliteral(L,"\n\t");lua_getinfo(L,"Snl",&ar);lua_pushfstring(L,"%s:",ar.
- short_src);if(ar.currentline>0)lua_pushfstring(L,"%d:",ar.currentline);switch(
- *ar.namewhat){case'g':case'l':case'f':case'm':lua_pushfstring(L,
- " in function `%s'",ar.name);break;default:{if(*ar.what=='m')lua_pushfstring(L
- ," in main chunk");else if(*ar.what=='C'||*ar.what=='t')lua_pushliteral(L," ?"
- );else lua_pushfstring(L," in function <%s:%d>",ar.short_src,ar.linedefined);}
- }lua_concat(L,lua_gettop(L));}lua_concat(L,lua_gettop(L));return 1;}static
- const luaL_reg dblib[]={{"getlocal",getlocal},{"getinfo",getinfo},{"gethook",
- gethook},{"getupvalue",getupvalue},{"sethook",sethook},{"setlocal",setlocal},{
- "setupvalue",setupvalue},{"debug",debug},{"traceback",errorfb},{NULL,NULL}};
- LUALIB_API int luaopen_debug(lua_State*L){luaL_openlib(L,LUA_DBLIBNAME,dblib,0
- );lua_pushliteral(L,"_TRACEBACK");lua_pushcfunction(L,errorfb);lua_settable(L,
- LUA_GLOBALSINDEX);return 1;}
- #line 1 "ldebug.c"
- #define ldebug_c
- static const char*getfuncname(CallInfo*ci,const char**name);
- #define isLua(ci) (!((ci)->state&CI_C))
- static int currentpc(CallInfo*ci){if(!isLua(ci))return-1;if(ci->state&
- CI_HASFRAME)ci->u.l.savedpc=*ci->u.l.pc;return pcRel(ci->u.l.savedpc,ci_func(
- ci)->l.p);}static int currentline(CallInfo*ci){int pc=currentpc(ci);if(pc<0)
- return-1;else return getline(ci_func(ci)->l.p,pc);}void luaG_inithooks(
- lua_State*L){CallInfo*ci;for(ci=L->ci;ci!=L->base_ci;ci--)currentpc(ci);L->
- hookinit=1;}LUA_API int lua_sethook(lua_State*L,lua_Hook func,int mask,int
- count){if(func==NULL||mask==0){mask=0;func=NULL;}L->hook=func;L->basehookcount
- =count;resethookcount(L);L->hookmask=cast(lu_byte,mask);L->hookinit=0;return 1
- ;}LUA_API lua_Hook lua_gethook(lua_State*L){return L->hook;}LUA_API int
- lua_gethookmask(lua_State*L){return L->hookmask;}LUA_API int lua_gethookcount(
- lua_State*L){return L->basehookcount;}LUA_API int lua_getstack(lua_State*L,int
- level,lua_Debug*ar){int status;CallInfo*ci;lua_lock(L);for(ci=L->ci;level>0&&
- ci>L->base_ci;ci--){level--;if(!(ci->state&CI_C))level-=ci->u.l.tailcalls;}if(
- level>0||ci==L->base_ci)status=0;else if(level<0){status=1;ar->i_ci=0;}else{
- status=1;ar->i_ci=ci-L->base_ci;}lua_unlock(L);return status;}static Proto*
- getluaproto(CallInfo*ci){return(isLua(ci)?ci_func(ci)->l.p:NULL);}LUA_API
- const char*lua_getlocal(lua_State*L,const lua_Debug*ar,int n){const char*name;
- CallInfo*ci;Proto*fp;lua_lock(L);name=NULL;ci=L->base_ci+ar->i_ci;fp=
- getluaproto(ci);if(fp){name=luaF_getlocalname(fp,n,currentpc(ci));if(name)
- luaA_pushobject(L,ci->base+(n-1));}lua_unlock(L);return name;}LUA_API const
- char*lua_setlocal(lua_State*L,const lua_Debug*ar,int n){const char*name;
- CallInfo*ci;Proto*fp;lua_lock(L);name=NULL;ci=L->base_ci+ar->i_ci;fp=
- getluaproto(ci);L->top--;if(fp){name=luaF_getlocalname(fp,n,currentpc(ci));if(
- !name||name[0]=='(')name=NULL;else setobjs2s(ci->base+(n-1),L->top);}
- lua_unlock(L);return name;}static void funcinfo(lua_Debug*ar,StkId func){
- Closure*cl=clvalue(func);if(cl->c.isC){ar->source="=[C]";ar->linedefined=-1;ar
- ->what="C";}else{ar->source=getstr(cl->l.p->source);ar->linedefined=cl->l.p->
- lineDefined;ar->what=(ar->linedefined==0)?"main":"Lua";}luaO_chunkid(ar->
- short_src,ar->source,LUA_IDSIZE);}static const char*travglobals(lua_State*L,
- const TObject*o){Table*g=hvalue(gt(L));int i=sizenode(g);while(i--){Node*n=
- gnode(g,i);if(luaO_rawequalObj(o,gval(n))&&ttisstring(gkey(n)))return getstr(
- tsvalue(gkey(n)));}return NULL;}static void info_tailcall(lua_State*L,
- lua_Debug*ar){ar->name=ar->namewhat="";ar->what="tail";ar->linedefined=ar->
- currentline=-1;ar->source="=(tail call)";luaO_chunkid(ar->short_src,ar->source
- ,LUA_IDSIZE);ar->nups=0;setnilvalue(L->top);}static int auxgetinfo(lua_State*L
- ,const char*what,lua_Debug*ar,StkId f,CallInfo*ci){int status=1;for(;*what;
- what++){switch(*what){case'S':{funcinfo(ar,f);break;}case'l':{ar->currentline=
- (ci)?currentline(ci):-1;break;}case'u':{ar->nups=clvalue(f)->c.nupvalues;break
- ;}case'n':{ar->namewhat=(ci)?getfuncname(ci,&ar->name):NULL;if(ar->namewhat==
- NULL){if((ar->name=travglobals(L,f))!=NULL)ar->namewhat="global";else ar->
- namewhat="";}break;}case'f':{setobj2s(L->top,f);break;}default:status=0;}}
- return status;}LUA_API int lua_getinfo(lua_State*L,const char*what,lua_Debug*
- ar){int status=1;lua_lock(L);if(*what=='>'){StkId f=L->top-1;if(!ttisfunction(
- f))luaG_runerror(L,"value for `lua_getinfo' is not a function");status=
- auxgetinfo(L,what+1,ar,f,NULL);L->top--;}else if(ar->i_ci!=0){CallInfo*ci=L->
- base_ci+ar->i_ci;lua_assert(ttisfunction(ci->base-1));status=auxgetinfo(L,what
- ,ar,ci->base-1,ci);}else info_tailcall(L,ar);if(strchr(what,'f'))incr_top(L);
- lua_unlock(L);return status;}
- #define check(x) if(!(x))return 0;
- #define checkjump(pt,pc) check(0<=pc&&pc<pt->sizecode)
- #define checkreg(pt,reg) check((reg)<(pt)->maxstacksize)
- static int precheck(const Proto*pt){check(pt->maxstacksize<=MAXSTACK);check(pt
- ->sizelineinfo==pt->sizecode||pt->sizelineinfo==0);lua_assert(pt->numparams+pt
- ->is_vararg<=pt->maxstacksize);check(GET_OPCODE(pt->code[pt->sizecode-1])==
- OP_RETURN);return 1;}static int checkopenop(const Proto*pt,int pc){Instruction
- i=pt->code[pc+1];switch(GET_OPCODE(i)){case OP_CALL:case OP_TAILCALL:case
- OP_RETURN:{check(GETARG_B(i)==0);return 1;}case OP_SETLISTO:return 1;default:
- return 0;}}static int checkRK(const Proto*pt,int r){return(r<pt->maxstacksize
- ||(r>=MAXSTACK&&r-MAXSTACK<pt->sizek));}static Instruction luaG_symbexec(const
- Proto*pt,int lastpc,int reg){int pc;int last;last=pt->sizecode-1;check(
- precheck(pt));for(pc=0;pc<lastpc;pc++){const Instruction i=pt->code[pc];OpCode
- op=GET_OPCODE(i);int a=GETARG_A(i);int b=0;int c=0;checkreg(pt,a);switch(
- getOpMode(op)){case iABC:{b=GETARG_B(i);c=GETARG_C(i);if(testOpMode(op,
- OpModeBreg)){checkreg(pt,b);}else if(testOpMode(op,OpModeBrk))check(checkRK(pt
- ,b));if(testOpMode(op,OpModeCrk))check(checkRK(pt,c));break;}case iABx:{b=
- GETARG_Bx(i);if(testOpMode(op,OpModeK))check(b<pt->sizek);break;}case iAsBx:{b
- =GETARG_sBx(i);break;}}if(testOpMode(op,OpModesetA)){if(a==reg)last=pc;}if(
- testOpMode(op,OpModeT)){check(pc+2<pt->sizecode);check(GET_OPCODE(pt->code[pc+
- 1])==OP_JMP);}switch(op){case OP_LOADBOOL:{check(c==0||pc+2<pt->sizecode);
- break;}case OP_LOADNIL:{if(a<=reg&®<=b)last=pc;break;}case OP_GETUPVAL:case
- OP_SETUPVAL:{check(b<pt->nups);break;}case OP_GETGLOBAL:case OP_SETGLOBAL:{
- check(ttisstring(&pt->k[b]));break;}case OP_SELF:{checkreg(pt,a+1);if(reg==a+1
- )last=pc;break;}case OP_CONCAT:{check(c<MAXSTACK&&b<c);break;}case OP_TFORLOOP
- :checkreg(pt,a+c+5);if(reg>=a)last=pc;case OP_FORLOOP:checkreg(pt,a+2);case
- OP_JMP:{int dest=pc+1+b;check(0<=dest&&dest<pt->sizecode);if(reg!=NO_REG&&pc<
- dest&&dest<=lastpc)pc+=b;break;}case OP_CALL:case OP_TAILCALL:{if(b!=0){
- checkreg(pt,a+b-1);}c--;if(c==LUA_MULTRET){check(checkopenop(pt,pc));}else if(
- c!=0)checkreg(pt,a+c-1);if(reg>=a)last=pc;break;}case OP_RETURN:{b--;if(b>0)
- checkreg(pt,a+b-1);break;}case OP_SETLIST:{checkreg(pt,a+(b&(LFIELDS_PER_FLUSH
- -1))+1);break;}case OP_CLOSURE:{int nup;check(b<pt->sizep);nup=pt->p[b]->nups;
- check(pc+nup<pt->sizecode);for(;nup>0;nup--){OpCode op1=GET_OPCODE(pt->code[pc
- +nup]);check(op1==OP_GETUPVAL||op1==OP_MOVE);}break;}default:break;}}return pt
- ->code[last];}
- #undef check
- #undef checkjump
- #undef checkreg
- int luaG_checkcode(const Proto*pt){return luaG_symbexec(pt,pt->sizecode,NO_REG
- );}static const char*kname(Proto*p,int c){c=c-MAXSTACK;if(c>=0&&ttisstring(&p
- ->k[c]))return svalue(&p->k[c]);else return"?";}static const char*getobjname(
- CallInfo*ci,int stackpos,const char**name){if(isLua(ci)){Proto*p=ci_func(ci)->
- l.p;int pc=currentpc(ci);Instruction i;*name=luaF_getlocalname(p,stackpos+1,pc
- );if(*name)return"local";i=luaG_symbexec(p,pc,stackpos);lua_assert(pc!=-1);
- switch(GET_OPCODE(i)){case OP_GETGLOBAL:{int g=GETARG_Bx(i);lua_assert(
- ttisstring(&p->k[g]));*name=svalue(&p->k[g]);return"global";}case OP_MOVE:{int
- a=GETARG_A(i);int b=GETARG_B(i);if(b<a)return getobjname(ci,b,name);break;}
- case OP_GETTABLE:{int k=GETARG_C(i);*name=kname(p,k);return"field";}case
- OP_SELF:{int k=GETARG_C(i);*name=kname(p,k);return"method";}default:break;}}
- return NULL;}static const char*getfuncname(CallInfo*ci,const char**name){
- Instruction i;if((isLua(ci)&&ci->u.l.tailcalls>0)||!isLua(ci-1))return NULL;ci
- --;i=ci_func(ci)->l.p->code[currentpc(ci)];if(GET_OPCODE(i)==OP_CALL||
- GET_OPCODE(i)==OP_TAILCALL)return getobjname(ci,GETARG_A(i),name);else return
- NULL;}static int isinstack(CallInfo*ci,const TObject*o){StkId p;for(p=ci->base
- ;p<ci->top;p++)if(o==p)return 1;return 0;}void luaG_typeerror(lua_State*L,
- const TObject*o,const char*op){const char*name=NULL;const char*t=
- luaT_typenames[ttype(o)];const char*kind=(isinstack(L->ci,o))?getobjname(L->ci
- ,o-L->base,&name):NULL;if(kind)luaG_runerror(L,
- "attempt to %s %s `%s' (a %s value)",op,kind,name,t);else luaG_runerror(L,
- "attempt to %s a %s value",op,t);}void luaG_concaterror(lua_State*L,StkId p1,
- StkId p2){if(ttisstring(p1))p1=p2;lua_assert(!ttisstring(p1));luaG_typeerror(L
- ,p1,"concatenate");}void luaG_aritherror(lua_State*L,const TObject*p1,const
- TObject*p2){TObject temp;if(luaV_tonumber(p1,&temp)==NULL)p2=p1;luaG_typeerror
- (L,p2,"perform arithmetic on");}int luaG_ordererror(lua_State*L,const TObject*
- p1,const TObject*p2){const char*t1=luaT_typenames[ttype(p1)];const char*t2=
- luaT_typenames[ttype(p2)];if(t1[2]==t2[2])luaG_runerror(L,
- "attempt to compare two %s values",t1);else luaG_runerror(L,
- "attempt to compare %s with %s",t1,t2);return 0;}static void addinfo(lua_State
- *L,const char*msg){CallInfo*ci=L->ci;if(isLua(ci)){char buff[LUA_IDSIZE];int
- line=currentline(ci);luaO_chunkid(buff,getstr(getluaproto(ci)->source),
- LUA_IDSIZE);luaO_pushfstring(L,"%s:%d: %s",buff,line,msg);}}void luaG_errormsg
- (lua_State*L){if(L->errfunc!=0){StkId errfunc=restorestack(L,L->errfunc);if(!
- ttisfunction(errfunc))luaD_throw(L,LUA_ERRERR);setobjs2s(L->top,L->top-1);
- setobjs2s(L->top-1,errfunc);incr_top(L);luaD_call(L,L->top-2,1);}luaD_throw(L,
- LUA_ERRRUN);}void luaG_runerror(lua_State*L,const char*fmt,...){va_list argp;
- va_start(argp,fmt);addinfo(L,luaO_pushvfstring(L,fmt,argp));va_end(argp);
- luaG_errormsg(L);}
- #line 1 "ldo.c"
- #define ldo_c
- struct lua_longjmp{struct lua_longjmp*previous;jmp_buf b;volatile int status;}
- ;static void seterrorobj(lua_State*L,int errcode,StkId oldtop){switch(errcode)
- {case LUA_ERRMEM:{setsvalue2s(oldtop,luaS_new(L,MEMERRMSG));break;}case
- LUA_ERRERR:{setsvalue2s(oldtop,luaS_new(L,"error in error handling"));break;}
- case LUA_ERRSYNTAX:case LUA_ERRRUN:{setobjs2s(oldtop,L->top-1);break;}}L->top=
- oldtop+1;}void luaD_throw(lua_State*L,int errcode){if(L->errorJmp){L->errorJmp
- ->status=errcode;longjmp(L->errorJmp->b,1);}else{G(L)->panic(L);exit(
- EXIT_FAILURE);}}int luaD_rawrunprotected(lua_State*L,Pfunc f,void*ud){struct
- lua_longjmp lj;lj.status=0;lj.previous=L->errorJmp;L->errorJmp=&lj;if(setjmp(
- lj.b)==0)(*f)(L,ud);L->errorJmp=lj.previous;return lj.status;}static void
- restore_stack_limit(lua_State*L){L->stack_last=L->stack+L->stacksize-1;if(L->
- size_ci>LUA_MAXCALLS){int inuse=(L->ci-L->base_ci);if(inuse+1<LUA_MAXCALLS)
- luaD_reallocCI(L,LUA_MAXCALLS);}}static void correctstack(lua_State*L,TObject*
- oldstack){CallInfo*ci;GCObject*up;L->top=(L->top-oldstack)+L->stack;for(up=L->
- openupval;up!=NULL;up=up->gch.next)gcotouv(up)->v=(gcotouv(up)->v-oldstack)+L
- ->stack;for(ci=L->base_ci;ci<=L->ci;ci++){ci->top=(ci->top-oldstack)+L->stack;
- ci->base=(ci->base-oldstack)+L->stack;}L->base=L->ci->base;}void
- luaD_reallocstack(lua_State*L,int newsize){TObject*oldstack=L->stack;
- luaM_reallocvector(L,L->stack,L->stacksize,newsize,TObject);L->stacksize=
- newsize;L->stack_last=L->stack+newsize-1-EXTRA_STACK;correctstack(L,oldstack);
- }void luaD_reallocCI(lua_State*L,int newsize){CallInfo*oldci=L->base_ci;
- luaM_reallocvector(L,L->base_ci,L->size_ci,newsize,CallInfo);L->size_ci=cast(
- unsigned short,newsize);L->ci=(L->ci-oldci)+L->base_ci;L->end_ci=L->base_ci+L
- ->size_ci;}void luaD_growstack(lua_State*L,int n){if(n<=L->stacksize)
- luaD_reallocstack(L,2*L->stacksize);else luaD_reallocstack(L,L->stacksize+n+
- EXTRA_STACK);}static void luaD_growCI(lua_State*L){if(L->size_ci>LUA_MAXCALLS)
- luaD_throw(L,LUA_ERRERR);else{luaD_reallocCI(L,2*L->size_ci);if(L->size_ci>
- LUA_MAXCALLS)luaG_runerror(L,"stack overflow");}}void luaD_callhook(lua_State*
- L,int event,int line){lua_Hook hook=L->hook;if(hook&&L->allowhook){ptrdiff_t
- top=savestack(L,L->top);ptrdiff_t ci_top=savestack(L,L->ci->top);lua_Debug ar;
- ar.event=event;ar.currentline=line;if(event==LUA_HOOKTAILRET)ar.i_ci=0;else ar
- .i_ci=L->ci-L->base_ci;luaD_checkstack(L,LUA_MINSTACK);L->ci->top=L->top+
- LUA_MINSTACK;L->allowhook=0;lua_unlock(L);(*hook)(L,&ar);lua_lock(L);
- lua_assert(!L->allowhook);L->allowhook=1;L->ci->top=restorestack(L,ci_top);L->
- top=restorestack(L,top);}}static void adjust_varargs(lua_State*L,int nfixargs,
- StkId base){int i;Table*htab;TObject nname;int actual=L->top-base;if(actual<
- nfixargs){luaD_checkstack(L,nfixargs-actual);for(;actual<nfixargs;++actual)
- setnilvalue(L->top++);}actual-=nfixargs;htab=luaH_new(L,actual,1);for(i=0;i<
- actual;i++)setobj2n(luaH_setnum(L,htab,i+1),L->top-actual+i);setsvalue(&nname,
- luaS_newliteral(L,"n"));setnvalue(luaH_set(L,htab,&nname),cast(lua_Number,
- actual));L->top-=actual;sethvalue(L->top,htab);incr_top(L);}static StkId
- tryfuncTM(lua_State*L,StkId func){const TObject*tm=luaT_gettmbyobj(L,func,
- TM_CALL);StkId p;ptrdiff_t funcr=savestack(L,func);if(!ttisfunction(tm))
- luaG_typeerror(L,func,"call");for(p=L->top;p>func;p--)setobjs2s(p,p-1);
- incr_top(L);func=restorestack(L,funcr);setobj2s(func,tm);return func;}StkId
- luaD_precall(lua_State*L,StkId func){LClosure*cl;ptrdiff_t funcr=savestack(L,
- func);if(!ttisfunction(func))func=tryfuncTM(L,func);if(L->ci+1==L->end_ci)
- luaD_growCI(L);else condhardstacktests(luaD_reallocCI(L,L->size_ci));cl=&
- clvalue(func)->l;if(!cl->isC){CallInfo*ci;Proto*p=cl->p;if(p->is_vararg)
- adjust_varargs(L,p->numparams,func+1);luaD_checkstack(L,p->maxstacksize);ci=++
- L->ci;L->base=L->ci->base=restorestack(L,funcr)+1;ci->top=L->base+p->
- maxstacksize;ci->u.l.savedpc=p->code;ci->u.l.tailcalls=0;ci->state=CI_SAVEDPC;
- while(L->top<ci->top)setnilvalue(L->top++);L->top=ci->top;return NULL;}else{
- CallInfo*ci;int n;luaD_checkstack(L,LUA_MINSTACK);ci=++L->ci;L->base=L->ci->
- base=restorestack(L,funcr)+1;ci->top=L->top+LUA_MINSTACK;ci->state=CI_C;if(L->
- hookmask&LUA_MASKCALL)luaD_callhook(L,LUA_HOOKCALL,-1);lua_unlock(L);
- #ifdef LUA_COMPATUPVALUES
- lua_pushupvalues(L);
- #endif
- n=(*clvalue(L->base-1)->c.f)(L);lua_lock(L);return L->top-n;}}static StkId
- callrethooks(lua_State*L,StkId firstResult){ptrdiff_t fr=savestack(L,
- firstResult);luaD_callhook(L,LUA_HOOKRET,-1);if(!(L->ci->state&CI_C)){while(L
- ->ci->u.l.tailcalls--)luaD_callhook(L,LUA_HOOKTAILRET,-1);}return restorestack
- (L,fr);}void luaD_poscall(lua_State*L,int wanted,StkId firstResult){StkId res;
- if(L->hookmask&LUA_MASKRET)firstResult=callrethooks(L,firstResult);res=L->base
- -1;L->ci--;L->base=L->ci->base;while(wanted!=0&&firstResult<L->top){setobjs2s(
- res++,firstResult++);wanted--;}while(wanted-->0)setnilvalue(res++);L->top=res;
- }void luaD_call(lua_State*L,StkId func,int nResults){StkId firstResult;
- lua_assert(!(L->ci->state&CI_CALLING));if(++L->nCcalls>=LUA_MAXCCALLS){if(L->
- nCcalls==LUA_MAXCCALLS)luaG_runerror(L,"C stack overflow");else if(L->nCcalls
- >=(LUA_MAXCCALLS+(LUA_MAXCCALLS>>3)))luaD_throw(L,LUA_ERRERR);}firstResult=
- luaD_precall(L,func);if(firstResult==NULL)firstResult=luaV_execute(L);
- luaD_poscall(L,nResults,firstResult);L->nCcalls--;luaC_checkGC(L);}static void
- resume(lua_State*L,void*ud){StkId firstResult;int nargs=*cast(int*,ud);
- CallInfo*ci=L->ci;if(ci==L->base_ci){lua_assert(nargs<L->top-L->base);
- luaD_precall(L,L->top-(nargs+1));}else{lua_assert(ci->state&CI_YIELD);if(ci->
- state&CI_C){int nresults;lua_assert((ci-1)->state&CI_SAVEDPC);lua_assert(
- GET_OPCODE(*((ci-1)->u.l.savedpc-1))==OP_CALL||GET_OPCODE(*((ci-1)->u.l.
- savedpc-1))==OP_TAILCALL);nresults=GETARG_C(*((ci-1)->u.l.savedpc-1))-1;
- luaD_poscall(L,nresults,L->top-nargs);if(nresults>=0)L->top=L->ci->top;}else{
- ci->state&=~CI_YIELD;}}firstResult=luaV_execute(L);if(firstResult!=NULL)
- luaD_poscall(L,LUA_MULTRET,firstResult);}static int resume_error(lua_State*L,
- const char*msg){L->top=L->ci->base;setsvalue2s(L->top,luaS_new(L,msg));
- incr_top(L);lua_unlock(L);return LUA_ERRRUN;}LUA_API int lua_resume(lua_State*
- L,int nargs){int status;lu_byte old_allowhooks;lua_lock(L);if(L->ci==L->
- base_ci){if(nargs>=L->top-L->base)return resume_error(L,
- "cannot resume dead coroutine");}else if(!(L->ci->state&CI_YIELD))return
- resume_error(L,"cannot resume non-suspended coroutine");old_allowhooks=L->
- allowhook;lua_assert(L->errfunc==0&&L->nCcalls==0);status=luaD_rawrunprotected
- (L,resume,&nargs);if(status!=0){L->ci=L->base_ci;L->base=L->ci->base;L->
- nCcalls=0;luaF_close(L,L->base);seterrorobj(L,status,L->base);L->allowhook=
- old_allowhooks;restore_stack_limit(L);}lua_unlock(L);return status;}LUA_API
- int lua_yield(lua_State*L,int nresults){CallInfo*ci;lua_lock(L);ci=L->ci;if(L
- ->nCcalls>0)luaG_runerror(L,
- "attempt to yield across metamethod/C-call boundary");if(ci->state&CI_C){if((
- ci-1)->state&CI_C)luaG_runerror(L,"cannot yield a C function");if(L->top-
- nresults>L->base){int i;for(i=0;i<nresults;i++)setobjs2s(L->base+i,L->top-
- nresults+i);L->top=L->base+nresults;}}ci->state|=CI_YIELD;lua_unlock(L);return
- -1;}int luaD_pcall(lua_State*L,Pfunc func,void*u,ptrdiff_t old_top,ptrdiff_t
- ef){int status;unsigned short oldnCcalls=L->nCcalls;ptrdiff_t old_ci=saveci(L,
- L->ci);lu_byte old_allowhooks=L->allowhook;ptrdiff_t old_errfunc=L->errfunc;L
- ->errfunc=ef;status=luaD_rawrunprotected(L,func,u);if(status!=0){StkId oldtop=
- restorestack(L,old_top);luaF_close(L,oldtop);seterrorobj(L,status,oldtop);L->
- nCcalls=oldnCcalls;L->ci=restoreci(L,old_ci);L->base=L->ci->base;L->allowhook=
- old_allowhooks;restore_stack_limit(L);}L->errfunc=old_errfunc;return status;}
- struct SParser{ZIO*z;Mbuffer buff;int bin;};static void f_parser(lua_State*L,
- void*ud){struct SParser*p;Proto*tf;Closure*cl;luaC_checkGC(L);p=cast(struct
- SParser*,ud);tf=p->bin?luaU_undump(L,p->z,&p->buff):luaY_parser(L,p->z,&p->
- buff);cl=luaF_newLclosure(L,0,gt(L));cl->l.p=tf;setclvalue(L->top,cl);incr_top
- (L);}int luaD_protectedparser(lua_State*L,ZIO*z,int bin){struct SParser p;int
- status;ptrdiff_t oldtopr=savestack(L,L->top);p.z=z;p.bin=bin;luaZ_initbuffer(L
- ,&p.buff);status=luaD_rawrunprotected(L,f_parser,&p);luaZ_freebuffer(L,&p.buff
- );if(status!=0){StkId oldtop=restorestack(L,oldtopr);seterrorobj(L,status,
- oldtop);}return status;}
- #line 1 "ldump.c"
- #define ldump_c
- #define DumpVector(b,n,size,D) DumpBlock(b,(n)*(size),D)
- #define DumpLiteral(s,D) DumpBlock(""s,(sizeof(s))-1,D)
- typedef struct{lua_State*L;lua_Chunkwriter write;void*data;}DumpState;static
- void DumpBlock(const void*b,size_t size,DumpState*D){lua_unlock(D->L);(*D->
- write)(D->L,b,size,D->data);lua_lock(D->L);}static void DumpByte(int y,
- DumpState*D){char x=(char)y;DumpBlock(&x,sizeof(x),D);}static void DumpInt(int
- x,DumpState*D){DumpBlock(&x,sizeof(x),D);}static void DumpSize(size_t x,
- DumpState*D){DumpBlock(&x,sizeof(x),D);}static void DumpNumber(lua_Number x,
- DumpState*D){DumpBlock(&x,sizeof(x),D);}static void DumpString(TString*s,
- DumpState*D){if(s==NULL||getstr(s)==NULL)DumpSize(0,D);else{size_t size=s->tsv
- .len+1;DumpSize(size,D);DumpBlock(getstr(s),size,D);}}static void DumpCode(
- const Proto*f,DumpState*D){DumpInt(f->sizecode,D);DumpVector(f->code,f->
- sizecode,sizeof(*f->code),D);}static void DumpLocals(const Proto*f,DumpState*D
- ){int i,n=f->sizelocvars;DumpInt(n,D);for(i=0;i<n;i++){DumpString(f->locvars[i
- ].varname,D);DumpInt(f->locvars[i].startpc,D);DumpInt(f->locvars[i].endpc,D);}
- }static void DumpLines(const Proto*f,DumpState*D){DumpInt(f->sizelineinfo,D);
- DumpVector(f->lineinfo,f->sizelineinfo,sizeof(*f->lineinfo),D);}static void
- DumpUpvalues(const Proto*f,DumpState*D){int i,n=f->sizeupvalues;DumpInt(n,D);
- for(i=0;i<n;i++)DumpString(f->upvalues[i],D);}static void DumpFunction(const
- Proto*f,const TString*p,DumpState*D);static void DumpConstants(const Proto*f,
- DumpState*D){int i,n;DumpInt(n=f->sizek,D);for(i=0;i<n;i++){const TObject*o=&f
- ->k[i];DumpByte(ttype(o),D);switch(ttype(o)){case LUA_TNUMBER:DumpNumber(
- nvalue(o),D);break;case LUA_TSTRING:DumpString(tsvalue(o),D);break;case
- LUA_TNIL:break;default:lua_assert(0);break;}}DumpInt(n=f->sizep,D);for(i=0;i<n
- ;i++)DumpFunction(f->p[i],f->source,D);}static void DumpFunction(const Proto*f
- ,const TString*p,DumpState*D){DumpString((f->source==p)?NULL:f->source,D);
- DumpInt(f->lineDefined,D);DumpByte(f->nups,D);DumpByte(f->numparams,D);
- DumpByte(f->is_vararg,D);DumpByte(f->maxstacksize,D);DumpLines(f,D);DumpLocals
- (f,D);DumpUpvalues(f,D);DumpConstants(f,D);DumpCode(f,D);}static void
- DumpHeader(DumpState*D){DumpLiteral(LUA_SIGNATURE,D);DumpByte(VERSION,D);
- DumpByte(luaU_endianness(),D);DumpByte(sizeof(int),D);DumpByte(sizeof(size_t),
- D);DumpByte(sizeof(Instruction),D);DumpByte(SIZE_OP,D);DumpByte(SIZE_A,D);
- DumpByte(SIZE_B,D);DumpByte(SIZE_C,D);DumpByte(sizeof(lua_Number),D);
- DumpNumber(TEST_NUMBER,D);}void luaU_dump(lua_State*L,const Proto*Main,
- lua_Chunkwriter w,void*data){DumpState D;D.L=L;D.write=w;D.data=data;
- DumpHeader(&D);DumpFunction(Main,NULL,&D);}
- #line 1 "lfunc.c"
- #define lfunc_c
- #define sizeCclosure(n) (cast(int,sizeof(CClosure))+cast(int,sizeof(TObject)*(\
- (n)-1)))
- #define sizeLclosure(n) (cast(int,sizeof(LClosure))+cast(int,sizeof(TObject*)*\
- ((n)-1)))
- Closure*luaF_newCclosure(lua_State*L,int nelems){Closure*c=cast(Closure*,
- luaM_malloc(L,sizeCclosure(nelems)));luaC_link(L,valtogco(c),LUA_TFUNCTION);c
- ->c.isC=1;c->c.nupvalues=cast(lu_byte,nelems);return c;}Closure*
- luaF_newLclosure(lua_State*L,int nelems,TObject*e){Closure*c=cast(Closure*,
- luaM_malloc(L,sizeLclosure(nelems)));luaC_link(L,valtogco(c),LUA_TFUNCTION);c
- ->l.isC=0;c->l.g=*e;c->l.nupvalues=cast(lu_byte,nelems);return c;}UpVal*
- luaF_findupval(lua_State*L,StkId level){GCObject**pp=&L->openupval;UpVal*p;
- UpVal*v;while((p=ngcotouv(*pp))!=NULL&&p->v>=level){if(p->v==level)return p;pp
- =&p->next;}v=luaM_new(L,UpVal);v->tt=LUA_TUPVAL;v->marked=1;v->v=level;v->next
- =*pp;*pp=valtogco(v);return v;}void luaF_close(lua_State*L,StkId level){UpVal*
- p;while((p=ngcotouv(L->openupval))!=NULL&&p->v>=level){setobj(&p->value,p->v);
- p->v=&p->value;L->openupval=p->next;luaC_link(L,valtogco(p),LUA_TUPVAL);}}
- Proto*luaF_newproto(lua_State*L){Proto*f=luaM_new(L,Proto);luaC_link(L,
- valtogco(f),LUA_TPROTO);f->k=NULL;f->sizek=0;f->p=NULL;f->sizep=0;f->code=NULL
- ;f->sizecode=0;f->sizelineinfo=0;f->sizeupvalues=0;f->nups=0;f->upvalues=NULL;
- f->numparams=0;f->is_vararg=0;f->maxstacksize=0;f->lineinfo=NULL;f->
- sizelocvars=0;f->locvars=NULL;f->lineDefined=0;f->source=NULL;return f;}void
- luaF_freeproto(lua_State*L,Proto*f){luaM_freearray(L,f->code,f->sizecode,
- Instruction);luaM_freearray(L,f->p,f->sizep,Proto*);luaM_freearray(L,f->k,f->
- sizek,TObject);luaM_freearray(L,f->lineinfo,f->sizelineinfo,int);
- luaM_freearray(L,f->locvars,f->sizelocvars,struct LocVar);luaM_freearray(L,f->
- upvalues,f->sizeupvalues,TString*);luaM_freelem(L,f);}void luaF_freeclosure(
- lua_State*L,Closure*c){int size=(c->c.isC)?sizeCclosure(c->c.nupvalues):
- sizeLclosure(c->l.nupvalues);luaM_free(L,c,size);}const char*luaF_getlocalname
- (const Proto*f,int local_number,int pc){int i;for(i=0;i<f->sizelocvars&&f->
- locvars[i].startpc<=pc;i++){if(pc<f->locvars[i].endpc){local_number--;if(
- local_number==0)return getstr(f->locvars[i].varname);}}return NULL;}
- #line 1 "lgc.c"
- #define lgc_c
- typedef struct GCState{GCObject*tmark;GCObject*wk;GCObject*wv;GCObject*wkv;
- global_State*g;}GCState;
- #define setbit(x,b) ((x)|=(1<<(b)))
- #define resetbit(x,b) ((x)&=cast(lu_byte,~(1<<(b))))
- #define testbit(x,b) ((x)&(1<<(b)))
- #define unmark(x) resetbit((x)->gch.marked,0)
- #define ismarked(x) ((x)->gch.marked&((1<<4)|1))
- #define stringmark(s) setbit((s)->tsv.marked,0)
- #define isfinalized(u) (!testbit((u)->uv.marked,1))
- #define markfinalized(u) resetbit((u)->uv.marked,1)
- #define KEYWEAKBIT 1
- #define VALUEWEAKBIT 2
- #define KEYWEAK (1<<KEYWEAKBIT)
- #define VALUEWEAK (1<<VALUEWEAKBIT)
- #define markobject(st,o) {checkconsistency(o);if(iscollectable(o)&&!ismarked(\
- gcvalue(o)))reallymarkobject(st,gcvalue(o));}
- #define condmarkobject(st,o,c) {checkconsistency(o);if(iscollectable(o)&&!\
- ismarked(gcvalue(o))&&(c))reallymarkobject(st,gcvalue(o));}
- #define markvalue(st,t) {if(!ismarked(valtogco(t)))reallymarkobject(st,\
- valtogco(t));}
- static void reallymarkobject(GCState*st,GCObject*o){lua_assert(!ismarked(o));
- setbit(o->gch.marked,0);switch(o->gch.tt){case LUA_TUSERDATA:{markvalue(st,
- gcotou(o)->uv.metatable);break;}case LUA_TFUNCTION:{gcotocl(o)->c.gclist=st->
- tmark;st->tmark=o;break;}case LUA_TTABLE:{gcotoh(o)->gclist=st->tmark;st->
- tmark=o;break;}case LUA_TTHREAD:{gcototh(o)->gclist=st->tmark;st->tmark=o;
- break;}case LUA_TPROTO:{gcotop(o)->gclist=st->tmark;st->tmark=o;break;}default
- :lua_assert(o->gch.tt==LUA_TSTRING);}}static void marktmu(GCState*st){GCObject
- *u;for(u=st->g->tmudata;u;u=u->gch.next){unmark(u);reallymarkobject(st,u);}}
- size_t luaC_separateudata(lua_State*L){size_t deadmem=0;GCObject**p=&G(L)->
- rootudata;GCObject*curr;GCObject*collected=NULL;GCObject**lastcollected=&
- collected;while((curr=*p)!=NULL){lua_assert(curr->gch.tt==LUA_TUSERDATA);if(
- ismarked(curr)||isfinalized(gcotou(curr)))p=&curr->gch.next;else if(fasttm(L,
- gcotou(curr)->uv.metatable,TM_GC)==NULL){markfinalized(gcotou(curr));p=&curr->
- gch.next;}else{deadmem+=sizeudata(gcotou(curr)->uv.len);*p=curr->gch.next;curr
- ->gch.next=NULL;*lastcollected=curr;lastcollected=&curr->gch.next;}}*
- lastcollected=G(L)->tmudata;G(L)->tmudata=collected;return deadmem;}static
- void removekey(Node*n){setnilvalue(gval(n));if(iscollectable(gkey(n)))setttype
- (gkey(n),LUA_TNONE);}static void traversetable(GCState*st,Table*h){int i;int
- weakkey=0;int weakvalue=0;const TObject*mode;markvalue(st,h->metatable);
- lua_assert(h->lsizenode||h->node==st->g->dummynode);mode=gfasttm(st->g,h->
- metatable,TM_MODE);if(mode&&ttisstring(mode)){weakkey=(strchr(svalue(mode),'k'
- )!=NULL);weakvalue=(strchr(svalue(mode),'v')!=NULL);if(weakkey||weakvalue){
- GCObject**weaklist;h->marked&=~(KEYWEAK|VALUEWEAK);h->marked|=cast(lu_byte,(
- weakkey<<KEYWEAKBIT)|(weakvalue<<VALUEWEAKBIT));weaklist=(weakkey&&weakvalue)?
- &st->wkv:(weakkey)?&st->wk:&st->wv;h->gclist=*weaklist;*weaklist=valtogco(h);}
- }if(!weakvalue){i=h->sizearray;while(i--)markobject(st,&h->array[i]);}i=
- sizenode(h);while(i--){Node*n=gnode(h,i);if(!ttisnil(gval(n))){lua_assert(!
- ttisnil(gkey(n)));condmarkobject(st,gkey(n),!weakkey);condmarkobject(st,gval(n
- ),!weakvalue);}}}static void traverseproto(GCState*st,Proto*f){int i;
- stringmark(f->source);for(i=0;i<f->sizek;i++){if(ttisstring(f->k+i))stringmark
- (tsvalue(f->k+i));}for(i=0;i<f->sizeupvalues;i++)stringmark(f->upvalues[i]);
- for(i=0;i<f->sizep;i++)markvalue(st,f->p[i]);for(i=0;i<f->sizelocvars;i++)
- stringmark(f->locvars[i].varname);lua_assert(luaG_checkcode(f));}static void
- traverseclosure(GCState*st,Closure*cl){if(cl->c.isC){int i;for(i=0;i<cl->c.
- nupvalues;i++)markobject(st,&cl->c.upvalue[i]);}else{int i;lua_assert(cl->l.
- nupvalues==cl->l.p->nups);markvalue(st,hvalue(&cl->l.g));markvalue(st,cl->l.p)
- ;for(i=0;i<cl->l.nupvalues;i++){UpVal*u=cl->l.upvals[i];if(!u->marked){
- markobject(st,&u->value);u->marked=1;}}}}static void checkstacksizes(lua_State
- *L,StkId max){int used=L->ci-L->base_ci;if(4*used<L->size_ci&&2*BASIC_CI_SIZE<
- L->size_ci)luaD_reallocCI(L,L->size_ci/2);else condhardstacktests(
- luaD_reallocCI(L,L->size_ci));used=max-L->stack;if(4*used<L->stacksize&&2*(
- BASIC_STACK_SIZE+EXTRA_STACK)<L->stacksize)luaD_reallocstack(L,L->stacksize/2)
- ;else condhardstacktests(luaD_reallocstack(L,L->stacksize));}static void
- traversestack(GCState*st,lua_State*L1){StkId o,lim;CallInfo*ci;markobject(st,
- gt(L1));lim=L1->top;for(ci=L1->base_ci;ci<=L1->ci;ci++){lua_assert(ci->top<=L1
- ->stack_last);lua_assert(ci->state&(CI_C|CI_HASFRAME|CI_SAVEDPC));if(lim<ci->
- top)lim=ci->top;}for(o=L1->stack;o<L1->top;o++)markobject(st,o);for(;o<=lim;o
- ++)setnilvalue(o);checkstacksizes(L1,lim);}static void propagatemarks(GCState*
- st){while(st->tmark){switch(st->tmark->gch.tt){case LUA_TTABLE:{Table*h=gcotoh
- (st->tmark);st->tmark=h->gclist;traversetable(st,h);break;}case LUA_TFUNCTION:
- {Closure*cl=gcotocl(st->tmark);st->tmark=cl->c.gclist;traverseclosure(st,cl);
- break;}case LUA_TTHREAD:{lua_State*th=gcototh(st->tmark);st->tmark=th->gclist;
- traversestack(st,th);break;}case LUA_TPROTO:{Proto*p=gcotop(st->tmark);st->
- tmark=p->gclist;traverseproto(st,p);break;}default:lua_assert(0);}}}static int
- valismarked(const TObject*o){if(ttisstring(o))stringmark(tsvalue(o));return!
- iscollectable(o)||testbit(o->value.gc->gch.marked,0);}static void
- cleartablekeys(GCObject*l){while(l){Table*h=gcotoh(l);int i=sizenode(h);
- lua_assert(h->marked&KEYWEAK);while(i--){Node*n=gnode(h,i);if(!valismarked(
- gkey(n)))removekey(n);}l=h->gclist;}}static void cleartablevalues(GCObject*l){
- while(l){Table*h=gcotoh(l);int i=h->sizearray;lua_assert(h->marked&VALUEWEAK);
- while(i--){TObject*o=&h->array[i];if(!valismarked(o))setnilvalue(o);}i=
- sizenode(h);while(i--){Node*n=gnode(h,i);if(!valismarked(gval(n)))removekey(n)
- ;}l=h->gclist;}}static void freeobj(lua_State*L,GCObject*o){switch(o->gch.tt){
- case LUA_TPROTO:luaF_freeproto(L,gcotop(o));break;case LUA_TFUNCTION:
- luaF_freeclosure(L,gcotocl(o));break;case LUA_TUPVAL:luaM_freelem(L,gcotouv(o)
- );break;case LUA_TTABLE:luaH_free(L,gcotoh(o));break;case LUA_TTHREAD:{
- lua_assert(gcototh(o)!=L&&gcototh(o)!=G(L)->mainthread);luaE_freethread(L,
- gcototh(o));break;}case LUA_TSTRING:{luaM_free(L,o,sizestring(gcotots(o)->tsv.
- len));break;}case LUA_TUSERDATA:{luaM_free(L,o,sizeudata(gcotou(o)->uv.len));
- break;}default:lua_assert(0);}}static int sweeplist(lua_State*L,GCObject**p,
- int limit){GCObject*curr;int count=0;while((curr=*p)!=NULL){if(curr->gch.
- marked>limit){unmark(curr);p=&curr->gch.next;}else{count++;*p=curr->gch.next;
- freeobj(L,curr);}}return count;}static void sweepstrings(lua_State*L,int all){
- int i;for(i=0;i<G(L)->strt.size;i++){G(L)->strt.nuse-=sweeplist(L,&G(L)->strt.
- hash[i],all);}}static void checkSizes(lua_State*L,size_t deadmem){if(G(L)->
- strt.nuse<cast(ls_nstr,G(L)->strt.size/4)&&G(L)->strt.size>MINSTRTABSIZE*2)
- luaS_resize(L,G(L)->strt.size/2);if(luaZ_sizebuffer(&G(L)->buff)>LUA_MINBUFFER
- *2){size_t newsize=luaZ_sizebuffer(&G(L)->buff)/2;luaZ_resizebuffer(L,&G(L)->
- buff,newsize);}G(L)->GCthreshold=2*G(L)->nblocks-deadmem;}static void do1gcTM(
- lua_State*L,Udata*udata){const TObject*tm=fasttm(L,udata->uv.metatable,TM_GC);
- if(tm!=NULL){setobj2s(L->top,tm);setuvalue(L->top+1,udata);L->top+=2;luaD_call
- (L,L->top-2,0);}}void luaC_callGCTM(lua_State*L){lu_byte oldah=L->allowhook;L
- ->allowhook=0;L->top++;while(G(L)->tmudata!=NULL){GCObject*o=G(L)->tmudata;
- Udata*udata=gcotou(o);G(L)->tmudata=udata->uv.next;udata->uv.next=G(L)->
- rootudata;G(L)->rootudata=o;setuvalue(L->top-1,udata);unmark(o);markfinalized(
- udata);do1gcTM(L,udata);}L->top--;L->allowhook=oldah;}void luaC_sweep(
- lua_State*L,int all){if(all)all=256;sweeplist(L,&G(L)->rootudata,all);
- sweepstrings(L,all);sweeplist(L,&G(L)->rootgc,all);}static void markroot(
- GCState*st,lua_State*L){global_State*g=st->g;markobject(st,defaultmeta(L));
- markobject(st,registry(L));traversestack(st,g->mainthread);if(L!=g->mainthread
- )markvalue(st,L);}static size_t mark(lua_State*L){size_t deadmem;GCState st;
- GCObject*wkv;st.g=G(L);st.tmark=NULL;st.wkv=st.wk=st.wv=NULL;markroot(&st,L);
- propagatemarks(&st);cleartablevalues(st.wkv);cleartablevalues(st.wv);wkv=st.
- wkv;st.wkv=NULL;st.wv=NULL;deadmem=luaC_separateudata(L);marktmu(&st);
- propagatemarks(&st);cleartablekeys(wkv);cleartablekeys(st.wk);cleartablevalues
- (st.wv);cleartablekeys(st.wkv);cleartablevalues(st.wkv);return deadmem;}void
- luaC_collectgarbage(lua_State*L){size_t deadmem=mark(L);luaC_sweep(L,0);
- checkSizes(L,deadmem);luaC_callGCTM(L);}void luaC_link(lua_State*L,GCObject*o,
- lu_byte tt){o->gch.next=G(L)->rootgc;G(L)->rootgc=o;o->gch.marked=0;o->gch.tt=
- tt;}
- #line 1 "liolib.c"
- #define liolib_c
- #ifndef USE_TMPNAME
- #ifdef __GNUC__
- #define USE_TMPNAME 0
- #else
- #define USE_TMPNAME 1
- #endif
- #endif
- #ifndef USE_POPEN
- #ifdef _POSIX_C_SOURCE
- #if _POSIX_C_SOURCE >=2
- #define USE_POPEN 1
- #endif
- #endif
- #endif
- #ifndef USE_POPEN
- #define USE_POPEN 0
- #endif
- #if !USE_POPEN
- #define pclose(f) (-1)
- #endif
- #define FILEHANDLE "FILE*"
- #define IO_INPUT "_input"
- #define IO_OUTPUT "_output"
- static int pushresult(lua_State*L,int i,const char*filename){if(i){
- lua_pushboolean(L,1);return 1;}else{lua_pushnil(L);if(filename)lua_pushfstring
- (L,"%s: %s",filename,strerror(errno));else lua_pushfstring(L,"%s",strerror(
- errno));lua_pushnumber(L,errno);return 3;}}static FILE**topfile(lua_State*L,
- int findex){FILE**f=(FILE**)luaL_checkudata(L,findex,FILEHANDLE);if(f==NULL)
- luaL_argerror(L,findex,"bad file");return f;}static int io_type(lua_State*L){
- FILE**f=(FILE**)luaL_checkudata(L,1,FILEHANDLE);if(f==NULL)lua_pushnil(L);else
- if(*f==NULL)lua_pushliteral(L,"closed file");else lua_pushliteral(L,"file");
- return 1;}static FILE*tofile(lua_State*L,int findex){FILE**f=topfile(L,findex)
- ;if(*f==NULL)luaL_error(L,"attempt to use a closed file");return*f;}static
- FILE**newfile(lua_State*L){FILE**pf=(FILE**)lua_newuserdata(L,sizeof(FILE*));*
- pf=NULL;luaL_getmetatable(L,FILEHANDLE);lua_setmetatable(L,-2);return pf;}
- static void registerfile(lua_State*L,FILE*f,const char*name,const char*impname
- ){lua_pushstring(L,name);*newfile(L)=f;if(impname){lua_pushstring(L,impname);
- lua_pushvalue(L,-2);lua_settable(L,-6);}lua_settable(L,-3);}static int
- aux_close(lua_State*L){FILE*f=tofile(L,1);if(f==stdin||f==stdout||f==stderr)
- return 0;else{int ok=(pclose(f)!=-1)||(fclose(f)==0);if(ok)*(FILE**)
- lua_touserdata(L,1)=NULL;return ok;}}static int io_close(lua_State*L){if(
- lua_isnone(L,1)&&lua_type(L,lua_upvalueindex(1))==LUA_TTABLE){lua_pushstring(L
- ,IO_OUTPUT);lua_rawget(L,lua_upvalueindex(1));}return pushresult(L,aux_close(L
- ),NULL);}static int io_gc(lua_State*L){FILE**f=topfile(L,1);if(*f!=NULL)
- aux_close(L);return 0;}static int io_tostring(lua_State*L){char buff[128];FILE
- **f=topfile(L,1);if(*f==NULL)strcpy(buff,"closed");else sprintf(buff,"%p",
- lua_touserdata(L,1));lua_pushfstring(L,"file (%s)",buff);return 1;}static int
- io_open(lua_State*L){const char*filename=luaL_checkstring(L,1);const char*mode
- =luaL_optstring(L,2,"r");FILE**pf=newfile(L);*pf=fopen(filename,mode);return(*
- pf==NULL)?pushresult(L,0,filename):1;}static int io_popen(lua_State*L){
- #if !USE_POPEN
- luaL_error(L,"`popen' not supported");return 0;
- #else
- const char*filename=luaL_checkstring(L,1);const char*mode=luaL_optstring(L,2,
- "r");FILE**pf=newfile(L);*pf=popen(filename,mode);return(*pf==NULL)?pushresult
- (L,0,filename):1;
- #endif
- }static int io_tmpfile(lua_State*L){FILE**pf=newfile(L);*pf=tmpfile();return(*
- pf==NULL)?pushresult(L,0,NULL):1;}static FILE*getiofile(lua_State*L,const char
- *name){lua_pushstring(L,name);lua_rawget(L,lua_upvalueindex(1));return tofile(
- L,-1);}static int g_iofile(lua_State*L,const char*name,const char*mode){if(!
- lua_isnoneornil(L,1)){const char*filename=lua_tostring(L,1);lua_pushstring(L,
- name);if(filename){FILE**pf=newfile(L);*pf=fopen(filename,mode);if(*pf==NULL){
- lua_pushfstring(L,"%s: %s",filename,strerror(errno));luaL_argerror(L,1,
- lua_tostring(L,-1));}}else{tofile(L,1);lua_pushvalue(L,1);}lua_rawset(L,
- lua_upvalueindex(1));}lua_pushstring(L,name);lua_rawget(L,lua_upvalueindex(1))
- ;return 1;}static int io_input(lua_State*L){return g_iofile(L,IO_INPUT,"r");}
- static int io_output(lua_State*L){return g_iofile(L,IO_OUTPUT,"w");}static int
- io_readline(lua_State*L);static void aux_lines(lua_State*L,int idx,int close)
- {lua_pushliteral(L,FILEHANDLE);lua_rawget(L,LUA_REGISTRYINDEX);lua_pushvalue(L
- ,idx);lua_pushboolean(L,close);lua_pushcclosure(L,io_readline,3);}static int
- f_lines(lua_State*L){tofile(L,1);aux_lines(L,1,0);return 1;}static int
- io_lines(lua_State*L){if(lua_isnoneornil(L,1)){lua_pushstring(L,IO_INPUT);
- lua_rawget(L,lua_upvalueindex(1));return f_lines(L);}else{const char*filename=
- luaL_checkstring(L,1);FILE**pf=newfile(L);*pf=fopen(filename,"r");
- luaL_argcheck(L,*pf,1,strerror(errno));aux_lines(L,lua_gettop(L),1);return 1;}
- }static int read_number(lua_State*L,FILE*f){lua_Number d;if(fscanf(f,
- LUA_NUMBER_SCAN,&d)==1){lua_pushnumber(L,d);return 1;}else return 0;}static
- int test_eof(lua_State*L,FILE*f){int c=getc(f);ungetc(c,f);lua_pushlstring(L,
- NULL,0);return(c!=EOF);}static int read_line(lua_State*L,FILE*f){luaL_Buffer b
- ;luaL_buffinit(L,&b);for(;;){size_t l;char*p=luaL_prepbuffer(&b);if(fgets(p,
- LUAL_BUFFERSIZE,f)==NULL){luaL_pushresult(&b);return(lua_strlen(L,-1)>0);}l=
- strlen(p);if(p[l-1]!='\n')luaL_addsize(&b,l);else{luaL_addsize(&b,l-1);
- luaL_pushresult(&b);return 1;}}}static int read_chars(lua_State*L,FILE*f,
- size_t n){size_t rlen;size_t nr;luaL_Buffer b;luaL_buffinit(L,&b);rlen=
- LUAL_BUFFERSIZE;do{char*p=luaL_prepbuffer(&b);if(rlen>n)rlen=n;nr=fread(p,
- sizeof(char),rlen,f);luaL_addsize(&b,nr);n-=nr;}while(n>0&&nr==rlen);
- luaL_pushresult(&b);return(n==0||lua_strlen(L,-1)>0);}static int g_read(
- lua_State*L,FILE*f,int first){int nargs=lua_gettop(L)-1;int success;int n;if(
- nargs==0){success=read_line(L,f);n=first+1;}else{luaL_checkstack(L,nargs+
- LUA_MINSTACK,"too many arguments");success=1;for(n=first;nargs--&&success;n++)
- {if(lua_type(L,n)==LUA_TNUMBER){size_t l=(size_t)lua_tonumber(L,n);success=(l
- ==0)?test_eof(L,f):read_chars(L,f,l);}else{const char*p=lua_tostring(L,n);
- luaL_argcheck(L,p&&p[0]=='*',n,"invalid option");switch(p[1]){case'n':success=
- read_number(L,f);break;case'l':success=read_line(L,f);break;case'a':read_chars
- (L,f,~((size_t)0));success=1;break;case'w':return luaL_error(L,
- "obsolete option `*w' to `read'");default:return luaL_argerror(L,n,
- "invalid format");}}}}if(!success){lua_pop(L,1);lua_pushnil(L);}return n-first
- ;}static int io_read(lua_State*L){return g_read(L,getiofile(L,IO_INPUT),1);}
- static int f_read(lua_State*L){return g_read(L,tofile(L,1),2);}static int
- io_readline(lua_State*L){FILE*f=*(FILE**)lua_touserdata(L,lua_upvalueindex(2))
- ;if(f==NULL)luaL_error(L,"file is already closed");if(read_line(L,f))return 1;
- else{if(lua_toboolean(L,lua_upvalueindex(3))){lua_settop(L,0);lua_pushvalue(L,
- lua_upvalueindex(2));aux_close(L);}return 0;}}static int g_write(lua_State*L,
- FILE*f,int arg){int nargs=lua_gettop(L)-1;int status=1;for(;nargs--;arg++){if(
- lua_type(L,arg)==LUA_TNUMBER){status=status&&fprintf(f,LUA_NUMBER_FMT,
- lua_tonumber(L,arg))>0;}else{size_t l;const char*s=luaL_checklstring(L,arg,&l)
- ;status=status&&(fwrite(s,sizeof(char),l,f)==l);}}return pushresult(L,status,
- NULL);}static int io_write(lua_State*L){return g_write(L,getiofile(L,IO_OUTPUT
- ),1);}static int f_write(lua_State*L){return g_write(L,tofile(L,1),2);}static
- int f_seek(lua_State*L){static const int mode[]={SEEK_SET,SEEK_CUR,SEEK_END};
- static const char*const modenames[]={"set","cur","end",NULL};FILE*f=tofile(L,1
- );int op=luaL_findstring(luaL_optstring(L,2,"cur"),modenames);long offset=
- luaL_optlong(L,3,0);luaL_argcheck(L,op!=-1,2,"invalid mode");op=fseek(f,offset
- ,mode[op]);if(op)return pushresult(L,0,NULL);else{lua_pushnumber(L,ftell(f));
- return 1;}}static int io_flush(lua_State*L){return pushresult(L,fflush(
- getiofile(L,IO_OUTPUT))==0,NULL);}static int f_flush(lua_State*L){return
- pushresult(L,fflush(tofile(L,1))==0,NULL);}static const luaL_reg iolib[]={{
- "input",io_input},{"output",io_output},{"lines",io_lines},{"close",io_close},{
- "flush",io_flush},{"open",io_open},{"popen",io_popen},{"read",io_read},{
- "tmpfile",io_tmpfile},{"type",io_type},{"write",io_write},{NULL,NULL}};static
- const luaL_reg flib[]={{"flush",f_flush},{"read",f_read},{"lines",f_lines},{
- "seek",f_seek},{"write",f_write},{"close",io_close},{"__gc",io_gc},{
- "__tostring",io_tostring},{NULL,NULL}};static void createmeta(lua_State*L){
- luaL_newmetatable(L,FILEHANDLE);lua_pushliteral(L,"__index");lua_pushvalue(L,-
- 2);lua_rawset(L,-3);luaL_openlib(L,NULL,flib,0);}static int io_execute(
- lua_State*L){lua_pushnumber(L,system(luaL_checkstring(L,1)));return 1;}static
- int io_remove(lua_State*L){const char*filename=luaL_checkstring(L,1);return
- pushresult(L,remove(filename)==0,filename);}static int io_rename(lua_State*L){
- const char*fromname=luaL_checkstring(L,1);const char*toname=luaL_checkstring(L
- ,2);return pushresult(L,rename(fromname,toname)==0,fromname);}static int
- io_tmpname(lua_State*L){
- #if !USE_TMPNAME
- luaL_error(L,"`tmpname' not supported");return 0;
- #else
- char buff[L_tmpnam];if(tmpnam(buff)!=buff)return luaL_error(L,
- "unable to generate a unique filename in `tmpname'");lua_pushstring(L,buff);
- return 1;
- #endif
- }static int io_getenv(lua_State*L){lua_pushstring(L,getenv(luaL_checkstring(L,
- 1)));return 1;}static int io_clock(lua_State*L){lua_pushnumber(L,((lua_Number)
- clock())/(lua_Number)CLOCKS_PER_SEC);return 1;}static void setfield(lua_State*
- L,const char*key,int value){lua_pushstring(L,key);lua_pushnumber(L,value);
- lua_rawset(L,-3);}static void setboolfield(lua_State*L,const char*key,int
- value){lua_pushstring(L,key);lua_pushboolean(L,value);lua_rawset(L,-3);}static
- int getboolfield(lua_State*L,const char*key){int res;lua_pushstring(L,key);
- lua_gettable(L,-2);res=lua_toboolean(L,-1);lua_pop(L,1);return res;}static int
- getfield(lua_State*L,const char*key,int d){int res;lua_pushstring(L,key);
- lua_gettable(L,-2);if(lua_isnumber(L,-1))res=(int)(lua_tonumber(L,-1));else{if
- (d==-2)return luaL_error(L,"field `%s' missing in date table",key);res=d;}
- lua_pop(L,1);return res;}static int io_date(lua_State*L){const char*s=
- luaL_optstring(L,1,"%c");time_t t=(time_t)(luaL_optnumber(L,2,-1));struct tm*
- stm;if(t==(time_t)(-1))t=time(NULL);if(*s=='!'){stm=gmtime(&t);s++;}else stm=
- localtime(&t);if(stm==NULL)lua_pushnil(L);else if(strcmp(s,"*t")==0){
- lua_newtable(L);setfield(L,"sec",stm->tm_sec);setfield(L,"min",stm->tm_min);
- setfield(L,"hour",stm->tm_hour);setfield(L,"day",stm->tm_mday);setfield(L,
- "month",stm->tm_mon+1);setfield(L,"year",stm->tm_year+1900);setfield(L,"wday",
- stm->tm_wday+1);setfield(L,"yday",stm->tm_yday+1);setboolfield(L,"isdst",stm->
- tm_isdst);}else{char b[256];if(strftime(b,sizeof(b),s,stm))lua_pushstring(L,b)
- ;else return luaL_error(L,"`date' format too long");}return 1;}static int
- io_time(lua_State*L){if(lua_isnoneornil(L,1))lua_pushnumber(L,time(NULL));else
- {time_t t;struct tm ts;luaL_checktype(L,1,LUA_TTABLE);lua_settop(L,1);ts.
- tm_sec=getfield(L,"sec",0);ts.tm_min=getfield(L,"min",0);ts.tm_hour=getfield(L
- ,"hour",12);ts.tm_mday=getfield(L,"day",-2);ts.tm_mon=getfield(L,"month",-2)-1
- ;ts.tm_year=getfield(L,"year",-2)-1900;ts.tm_isdst=getboolfield(L,"isdst");t=
- mktime(&ts);if(t==(time_t)(-1))lua_pushnil(L);else lua_pushnumber(L,t);}return
- 1;}static int io_difftime(lua_State*L){lua_pushnumber(L,difftime((time_t)(
- luaL_checknumber(L,1)),(time_t)(luaL_optnumber(L,2,0))));return 1;}static int
- io_setloc(lua_State*L){static const int cat[]={LC_ALL,LC_COLLATE,LC_CTYPE,
- LC_MONETARY,LC_NUMERIC,LC_TIME};static const char*const catnames[]={"all",
- "collate","ctype","monetary","numeric","time",NULL};const char*l=lua_tostring(
- L,1);int op=luaL_findstring(luaL_optstring(L,2,"all"),catnames);luaL_argcheck(
- L,l||lua_isnoneornil(L,1),1,"string expected");luaL_argcheck(L,op!=-1,2,
- "invalid option");lua_pushstring(L,setlocale(cat[op],l));return 1;}static int
- io_exit(lua_State*L){exit(luaL_optint(L,1,EXIT_SUCCESS));return 0;}static
- const luaL_reg syslib[]={{"clock",io_clock},{"date",io_date},{"difftime",
- io_difftime},{"execute",io_execute},{"exit",io_exit},{"getenv",io_getenv},{
- "remove",io_remove},{"rename",io_rename},{"setlocale",io_setloc},{"time",
- io_time},{"tmpname",io_tmpname},{NULL,NULL}};LUALIB_API int luaopen_io(
- lua_State*L){luaL_openlib(L,LUA_OSLIBNAME,syslib,0);createmeta(L);
- lua_pushvalue(L,-1);luaL_openlib(L,LUA_IOLIBNAME,iolib,1);registerfile(L,stdin
- ,"stdin",IO_INPUT);registerfile(L,stdout,"stdout",IO_OUTPUT);registerfile(L,
- stderr,"stderr",NULL);return 1;}
- #line 1 "llex.c"
- #define llex_c
- #define next(LS) (LS->current=zgetc(LS->z))
- static const char*const token2string[]={"and","break","do","else","elseif",
- "end","false","for","function","if","in","local","nil","not","or","repeat",
- "return","then","true","until","while","*name","..","...","==",">=","<=","~=",
- "*number","*string","<eof>"};void luaX_init(lua_State*L){int i;for(i=0;i<
- NUM_RESERVED;i++){TString*ts=luaS_new(L,token2string[i]);luaS_fix(ts);
- lua_assert(strlen(token2string[i])+1<=TOKEN_LEN);ts->tsv.reserved=cast(lu_byte
- ,i+1);}}
- #define MAXSRC 80
- void luaX_checklimit(LexState*ls,int val,int limit,const char*msg){if(val>
- limit){msg=luaO_pushfstring(ls->L,"too many %s (limit=%d)",msg,limit);
- luaX_syntaxerror(ls,msg);}}void luaX_errorline(LexState*ls,const char*s,const
- char*token,int line){lua_State*L=ls->L;char buff[MAXSRC];luaO_chunkid(buff,
- getstr(ls->source),MAXSRC);luaO_pushfstring(L,"%s:%d: %s near `%s'",buff,line,
- s,token);luaD_throw(L,LUA_ERRSYNTAX);}static void luaX_error(LexState*ls,const
- char*s,const char*token){luaX_errorline(ls,s,token,ls->linenumber);}void
- luaX_syntaxerror(LexState*ls,const char*msg){const char*lasttoken;switch(ls->t
- .token){case TK_NAME:lasttoken=getstr(ls->t.seminfo.ts);break;case TK_STRING:
- case TK_NUMBER:lasttoken=luaZ_buffer(ls->buff);break;default:lasttoken=
- luaX_token2str(ls,ls->t.token);break;}luaX_error(ls,msg,lasttoken);}const char
- *luaX_token2str(LexState*ls,int token){if(token<FIRST_RESERVED){lua_assert(
- token==(unsigned char)token);return luaO_pushfstring(ls->L,"%c",token);}else
- return token2string[token-FIRST_RESERVED];}static void luaX_lexerror(LexState*
- ls,const char*s,int token){if(token==TK_EOS)luaX_error(ls,s,luaX_token2str(ls,
- token));else luaX_error(ls,s,luaZ_buffer(ls->buff));}static void inclinenumber
- (LexState*LS){next(LS);++LS->linenumber;luaX_checklimit(LS,LS->linenumber,
- MAX_INT,"lines in a chunk");}void luaX_setinput(lua_State*L,LexState*LS,ZIO*z,
- TString*source){LS->L=L;LS->lookahead.token=TK_EOS;LS->z=z;LS->fs=NULL;LS->
- linenumber=1;LS->lastline=1;LS->source=source;next(LS);if(LS->current=='#'){do
- {next(LS);}while(LS->current!='\n'&&LS->current!=EOZ);}}
- #define EXTRABUFF 32
- #define MAXNOCHECK 5
- #define checkbuffer(LS, len)if(((len)+MAXNOCHECK)*sizeof(char)>luaZ_sizebuffer\
- ((LS)->buff))luaZ_openspace((LS)->L,(LS)->buff,(len)+EXTRABUFF)
- #define save(LS, c,l)(luaZ_buffer((LS)->buff)[l++]=cast(char,c))
- #define save_and_next(LS, l)(save(LS,LS->current,l),next(LS))
- static size_t readname(LexState*LS){size_t l=0;checkbuffer(LS,l);do{
- checkbuffer(LS,l);save_and_next(LS,l);}while(isalnum(LS->current)||LS->current
- =='_');save(LS,'\0',l);return l-1;}static void read_numeral(LexState*LS,int
- comma,SemInfo*seminfo){size_t l=0;checkbuffer(LS,l);if(comma)save(LS,'.',l);
- while(isdigit(LS->current)){checkbuffer(LS,l);save_and_next(LS,l);}if(LS->
- current=='.'){save_and_next(LS,l);if(LS->current=='.'){save_and_next(LS,l);
- save(LS,'\0',l);luaX_lexerror(LS,
- "ambiguous syntax (decimal point x string concatenation)",TK_NUMBER);}}while(
- isdigit(LS->current)){checkbuffer(LS,l);save_and_next(LS,l);}if(LS->current==
- 'e'||LS->current=='E'){save_and_next(LS,l);if(LS->current=='+'||LS->current==
- '-')save_and_next(LS,l);while(isdigit(LS->current)){checkbuffer(LS,l);
- save_and_next(LS,l);}}save(LS,'\0',l);if(!luaO_str2d(luaZ_buffer(LS->buff),&
- seminfo->r))luaX_lexerror(LS,"malformed number",TK_NUMBER);}static void
- read_long_string(LexState*LS,SemInfo*seminfo){int cont=0;size_t l=0;
- checkbuffer(LS,l);save(LS,'[',l);save_and_next(LS,l);if(LS->current=='\n')
- inclinenumber(LS);for(;;){checkbuffer(LS,l);switch(LS->current){case EOZ:save(
- LS,'\0',l);luaX_lexerror(LS,(seminfo)?"unfinished long string":
- "unfinished long comment",TK_EOS);break;case'[':save_and_next(LS,l);if(LS->
- current=='['){cont++;save_and_next(LS,l);}continue;case']':save_and_next(LS,l)
- ;if(LS->current==']'){if(cont==0)goto endloop;cont--;save_and_next(LS,l);}
- continue;case'\n':save(LS,'\n',l);inclinenumber(LS);if(!seminfo)l=0;continue;
- default:save_and_next(LS,l);}}endloop:save_and_next(LS,l);save(LS,'\0',l);if(
- seminfo)seminfo->ts=luaS_newlstr(LS->L,luaZ_buffer(LS->buff)+2,l-5);}static
- void read_string(LexState*LS,int del,SemInfo*seminfo){size_t l=0;checkbuffer(
- LS,l);save_and_next(LS,l);while(LS->current!=del){checkbuffer(LS,l);switch(LS
- ->current){case EOZ:save(LS,'\0',l);luaX_lexerror(LS,"unfinished string",
- TK_EOS);break;case'\n':save(LS,'\0',l);luaX_lexerror(LS,"unfinished string",
- TK_STRING);break;case'\\':next(LS);switch(LS->current){case'a':save(LS,'\a',l)
- ;next(LS);break;case'b':save(LS,'\b',l);next(LS);break;case'f':save(LS,'\f',l)
- ;next(LS);break;case'n':save(LS,'\n',l);next(LS);break;case'r':save(LS,'\r',l)
- ;next(LS);break;case't':save(LS,'\t',l);next(LS);break;case'v':save(LS,'\v',l)
- ;next(LS);break;case'\n':save(LS,'\n',l);inclinenumber(LS);break;case EOZ:
- break;default:{if(!isdigit(LS->current))save_and_next(LS,l);else{int c=0;int i
- =0;do{c=10*c+(LS->current-'0');next(LS);}while(++i<3&&isdigit(LS->current));if
- (c>UCHAR_MAX){save(LS,'\0',l);luaX_lexerror(LS,"escape sequence too large",
- TK_STRING);}save(LS,c,l);}}}break;default:save_and_next(LS,l);}}save_and_next(
- LS,l);save(LS,'\0',l);seminfo->ts=luaS_newlstr(LS->L,luaZ_buffer(LS->buff)+1,l
- -3);}int luaX_lex(LexState*LS,SemInfo*seminfo){for(;;){switch(LS->current){
- case'\n':{inclinenumber(LS);continue;}case'-':{next(LS);if(LS->current!='-')
- return'-';next(LS);if(LS->current=='['&&(next(LS),LS->current=='['))
- read_long_string(LS,NULL);else while(LS->current!='\n'&&LS->current!=EOZ)next(
- LS);continue;}case'[':{next(LS);if(LS->current!='[')return'[';else{
- read_long_string(LS,seminfo);return TK_STRING;}}case'=':{next(LS);if(LS->
- current!='=')return'=';else{next(LS);return TK_EQ;}}case'<':{next(LS);if(LS->
- current!='=')return'<';else{next(LS);return TK_LE;}}case'>':{next(LS);if(LS->
- current!='=')return'>';else{next(LS);return TK_GE;}}case'~':{next(LS);if(LS->
- current!='=')return'~';else{next(LS);return TK_NE;}}case'"':case'\'':{
- read_string(LS,LS->current,seminfo);return TK_STRING;}case'.':{next(LS);if(LS
- ->current=='.'){next(LS);if(LS->current=='.'){next(LS);return TK_DOTS;}else
- return TK_CONCAT;}else if(!isdigit(LS->current))return'.';else{read_numeral(LS
- ,1,seminfo);return TK_NUMBER;}}case EOZ:{return TK_EOS;}default:{if(isspace(LS
- ->current)){next(LS);continue;}else if(isdigit(LS->current)){read_numeral(LS,0
- ,seminfo);return TK_NUMBER;}else if(isalpha(LS->current)||LS->current=='_'){
- size_t l=readname(LS);TString*ts=luaS_newlstr(LS->L,luaZ_buffer(LS->buff),l);
- if(ts->tsv.reserved>0)return ts->tsv.reserved-1+FIRST_RESERVED;seminfo->ts=ts;
- return TK_NAME;}else{int c=LS->current;if(iscntrl(c))luaX_error(LS,
- "invalid control char",luaO_pushfstring(LS->L,"char(%d)",c));next(LS);return c
- ;}}}}}
- #undef next
- #line 1 "lmem.c"
- #define lmem_c
- #ifndef l_realloc
- #define l_realloc(b,os,s) realloc(b,s)
- #endif
- #ifndef l_free
- #define l_free(b,os) free(b)
- #endif
- #define MINSIZEARRAY 4
- void*luaM_growaux(lua_State*L,void*block,int*size,int size_elems,int limit,
- const char*errormsg){void*newblock;int newsize=(*size)*2;if(newsize<
- MINSIZEARRAY)newsize=MINSIZEARRAY;else if(*size>=limit/2){if(*size<limit-
- MINSIZEARRAY)newsize=limit;else luaG_runerror(L,errormsg);}newblock=
- luaM_realloc(L,block,cast(lu_mem,*size)*cast(lu_mem,size_elems),cast(lu_mem,
- newsize)*cast(lu_mem,size_elems));*size=newsize;return newblock;}void*
- luaM_realloc(lua_State*L,void*block,lu_mem oldsize,lu_mem size){lua_assert((
- oldsize==0)==(block==NULL));if(size==0){if(block!=NULL){l_free(block,oldsize);
- block=NULL;}else return NULL;}else if(size>=MAX_SIZET)luaG_runerror(L,
- "memory allocation error: block too big");else{block=l_realloc(block,oldsize,
- size);if(block==NULL){if(L)luaD_throw(L,LUA_ERRMEM);else return NULL;}}if(L){
- lua_assert(G(L)!=NULL&&G(L)->nblocks>0);G(L)->nblocks-=oldsize;G(L)->nblocks+=
- size;}return block;}
- #line 1 "loadlib.c"
- #undef LOADLIB
- #ifdef USE_DLOPEN
- #define LOADLIB
- #include <dlfcn.h> /* dg: magic anchor comment */
- static int loadlib(lua_State*L){const char*path=luaL_checkstring(L,1);const
- char*init=luaL_checkstring(L,2);void*lib=dlopen(path,RTLD_NOW);if(lib!=NULL){
- lua_CFunction f=(lua_CFunction)dlsym(lib,init);if(f!=NULL){
- lua_pushlightuserdata(L,lib);lua_pushcclosure(L,f,1);return 1;}}lua_pushnil(L)
- ;lua_pushstring(L,dlerror());lua_pushstring(L,(lib!=NULL)?"init":"open");if(
- lib!=NULL)dlclose(lib);return 3;}
- #endif
- #ifndef USE_DLL
- #ifdef _WIN32
- #define USE_DLL 1
- #else
- #define USE_DLL 0
- #endif
- #endif
- #if USE_DLL
- #define LOADLIB
- #include <windows.h> /* dg: magic anchor comment */
- static void pusherror(lua_State*L){int error=GetLastError();char buffer[128];
- if(FormatMessage(FORMAT_MESSAGE_IGNORE_INSERTS|FORMAT_MESSAGE_FROM_SYSTEM,0,
- error,0,buffer,sizeof(buffer),0))lua_pushstring(L,buffer);else lua_pushfstring
- (L,"system error %d\n",error);}static int loadlib(lua_State*L){const char*path
- =luaL_checkstring(L,1);const char*init=luaL_checkstring(L,2);HINSTANCE lib=
- LoadLibrary(path);if(lib!=NULL){lua_CFunction f=(lua_CFunction)GetProcAddress(
- lib,init);if(f!=NULL){lua_pushlightuserdata(L,lib);lua_pushcclosure(L,f,1);
- return 1;}}lua_pushnil(L);pusherror(L);lua_pushstring(L,(lib!=NULL)?"init":
- "open");if(lib!=NULL)FreeLibrary(lib);return 3;}
- #endif
- #ifndef LOADLIB
- #ifdef linux
- #define LOADLIB
- #endif
- #ifdef sun
- #define LOADLIB
- #endif
- #ifdef sgi
- #define LOADLIB
- #endif
- #ifdef BSD
- #define LOADLIB
- #endif
- #ifdef _WIN32
- #define LOADLIB
- #endif
- #ifdef LOADLIB
- #undef LOADLIB
- #define LOADLIB "`loadlib' not installed (check your Lua configuration)"
- #else
- #define LOADLIB "`loadlib' not supported"
- #endif
- static int loadlib(lua_State*L){lua_pushnil(L);lua_pushliteral(L,LOADLIB);
- lua_pushliteral(L,"absent");return 3;}
- #endif
- LUALIB_API int luaopen_loadlib(lua_State*L){lua_register(L,"loadlib",loadlib);
- return 0;}
- #line 1 "lobject.c"
- #define lobject_c
- #ifndef lua_str2number
- #define lua_str2number(s,p) strtod((s),(p))
- #endif
- const TObject luaO_nilobject={LUA_TNIL,{NULL}};int luaO_int2fb(unsigned int x)
- {int m=0;while(x>=(1<<3)){x=(x+1)>>1;m++;}return(m<<3)|cast(int,x);}int
- luaO_log2(unsigned int x){static const lu_byte log_8[255]={0,1,1,2,2,2,2,3,3,3
- ,3,3,3,3,3,4,4,4,4,4,4,4,4,4,4,4,4,4,4,4,4,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5
- ,5,5,5,5,5,5,5,5,5,5,5,5,5,5,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6
- ,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6
- ,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7
- ,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7
- ,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7
- ,7,7,7,7,7,7,7,7,7,7,7};if(x>=0x00010000){if(x>=0x01000000)return log_8[((x>>
- 24)&0xff)-1]+24;else return log_8[((x>>16)&0xff)-1]+16;}else{if(x>=0x00000100)
- return log_8[((x>>8)&0xff)-1]+8;else if(x)return log_8[(x&0xff)-1];return-1;}}
- int luaO_rawequalObj(const TObject*t1,const TObject*t2){if(ttype(t1)!=ttype(t2
- ))return 0;else switch(ttype(t1)){case LUA_TNIL:return 1;case LUA_TNUMBER:
- return nvalue(t1)==nvalue(t2);case LUA_TBOOLEAN:return bvalue(t1)==bvalue(t2);
- case LUA_TLIGHTUSERDATA:return pvalue(t1)==pvalue(t2);default:lua_assert(
- iscollectable(t1));return gcvalue(t1)==gcvalue(t2);}}int luaO_str2d(const char
- *s,lua_Number*result){char*endptr;lua_Number res=lua_str2number(s,&endptr);if(
- endptr==s)return 0;while(isspace((unsigned char)(*endptr)))endptr++;if(*endptr
- !='\0')return 0;*result=res;return 1;}static void pushstr(lua_State*L,const
- char*str){setsvalue2s(L->top,luaS_new(L,str));incr_top(L);}const char*
- luaO_pushvfstring(lua_State*L,const char*fmt,va_list argp){int n=1;pushstr(L,
- "");for(;;){const char*e=strchr(fmt,'%');if(e==NULL)break;setsvalue2s(L->top,
- luaS_newlstr(L,fmt,e-fmt));incr_top(L);switch(*(e+1)){case's':pushstr(L,va_arg
- (argp,char*));break;case'c':{char buff[2];buff[0]=cast(char,va_arg(argp,int));
- buff[1]='\0';pushstr(L,buff);break;}case'd':setnvalue(L->top,cast(lua_Number,
- va_arg(argp,int)));incr_top(L);break;case'f':setnvalue(L->top,cast(lua_Number,
- va_arg(argp,l_uacNumber)));incr_top(L);break;case'%':pushstr(L,"%");break;
- default:lua_assert(0);}n+=2;fmt=e+2;}pushstr(L,fmt);luaV_concat(L,n+1,L->top-L
- ->base-1);L->top-=n;return svalue(L->top-1);}const char*luaO_pushfstring(
- lua_State*L,const char*fmt,...){const char*msg;va_list argp;va_start(argp,fmt)
- ;msg=luaO_pushvfstring(L,fmt,argp);va_end(argp);return msg;}void luaO_chunkid(
- char*out,const char*source,int bufflen){if(*source=='='){strncpy(out,source+1,
- bufflen);out[bufflen-1]='\0';}else{if(*source=='@'){int l;source++;bufflen-=
- sizeof(" `...' ");l=strlen(source);strcpy(out,"");if(l>bufflen){source+=(l-
- bufflen);strcat(out,"...");}strcat(out,source);}else{int len=strcspn(source,
- "\n");bufflen-=sizeof(" [string \"...\"] ");if(len>bufflen)len=bufflen;strcpy(
- out,"[string \"");if(source[len]!='\0'){strncat(out,source,len);strcat(out,
- "...");}else strcat(out,source);strcat(out,"\"]");}}}
- #line 1 "lopcodes.c"
- #define lopcodes_c
- #ifdef LUA_OPNAMES
- const char*const luaP_opnames[]={"MOVE","LOADK","LOADBOOL","LOADNIL",
- "GETUPVAL","GETGLOBAL","GETTABLE","SETGLOBAL","SETUPVAL","SETTABLE","NEWTABLE"
- ,"SELF","ADD","SUB","MUL","DIV","POW","UNM","NOT","CONCAT","JMP","EQ","LT",
- "LE","TEST","CALL","TAILCALL","RETURN","FORLOOP","TFORLOOP","TFORPREP",
- "SETLIST","SETLISTO","CLOSE","CLOSURE"};
- #endif
- #define opmode(t,b,bk,ck,sa,k,m) (((t)<<OpModeT)|((b)<<OpModeBreg)|((bk)<<\
- OpModeBrk)|((ck)<<OpModeCrk)|((sa)<<OpModesetA)|((k)<<OpModeK)|(m))
- const lu_byte luaP_opmodes[NUM_OPCODES]={opmode(0,1,0,0,1,0,iABC),opmode(0,0,0
- ,0,1,1,iABx),opmode(0,0,0,0,1,0,iABC),opmode(0,1,0,0,1,0,iABC),opmode(0,0,0,0,
- 1,0,iABC),opmode(0,0,0,0,1,1,iABx),opmode(0,1,0,1,1,0,iABC),opmode(0,0,0,0,0,1
- ,iABx),opmode(0,0,0,0,0,0,iABC),opmode(0,0,1,1,0,0,iABC),opmode(0,0,0,0,1,0,
- iABC),opmode(0,1,0,1,1,0,iABC),opmode(0,0,1,1,1,0,iABC),opmode(0,0,1,1,1,0,
- iABC),opmode(0,0,1,1,1,0,iABC),opmode(0,0,1,1,1,0,iABC),opmode(0,0,1,1,1,0,
- iABC),opmode(0,1,0,0,1,0,iABC),opmode(0,1,0,0,1,0,iABC),opmode(0,1,0,1,1,0,
- iABC),opmode(0,0,0,0,0,0,iAsBx),opmode(1,0,1,1,0,0,iABC),opmode(1,0,1,1,0,0,
- iABC),opmode(1,0,1,1,0,0,iABC),opmode(1,1,0,0,1,0,iABC),opmode(0,0,0,0,0,0,
- iABC),opmode(0,0,0,0,0,0,iABC),opmode(0,0,0,0,0,0,iABC),opmode(0,0,0,0,0,0,
- iAsBx),opmode(1,0,0,0,0,0,iABC),opmode(0,0,0,0,0,0,iAsBx),opmode(0,0,0,0,0,0,
- iABx),opmode(0,0,0,0,0,0,iABx),opmode(0,0,0,0,0,0,iABC),opmode(0,0,0,0,1,0,
- iABx)};
- #line 1 "lparser.c"
- #define lparser_c
- #define getlocvar(fs, i)((fs)->f->locvars[(fs)->actvar[i]])
- #define enterlevel(ls) if(++(ls)->nestlevel>LUA_MAXPARSERLEVEL)\
- luaX_syntaxerror(ls,"too many syntax levels");
- #define leavelevel(ls) ((ls)->nestlevel--)
- typedef struct BlockCnt{struct BlockCnt*previous;int breaklist;int nactvar;int
- upval;int isbreakable;}BlockCnt;static void chunk(LexState*ls);static void
- expr(LexState*ls,expdesc*v);static void next(LexState*ls){ls->lastline=ls->
- linenumber;if(ls->lookahead.token!=TK_EOS){ls->t=ls->lookahead;ls->lookahead.
- token=TK_EOS;}else ls->t.token=luaX_lex(ls,&ls->t.seminfo);}static void
- lookahead(LexState*ls){lua_assert(ls->lookahead.token==TK_EOS);ls->lookahead.
- token=luaX_lex(ls,&ls->lookahead.seminfo);}static void error_expected(LexState
- *ls,int token){luaX_syntaxerror(ls,luaO_pushfstring(ls->L,"`%s' expected",
- luaX_token2str(ls,token)));}static int testnext(LexState*ls,int c){if(ls->t.
- token==c){next(ls);return 1;}else return 0;}static void check(LexState*ls,int
- c){if(!testnext(ls,c))error_expected(ls,c);}
- #define check_condition(ls,c,msg) {if(!(c))luaX_syntaxerror(ls,msg);}
- static void check_match(LexState*ls,int what,int who,int where){if(!testnext(
- ls,what)){if(where==ls->linenumber)error_expected(ls,what);else{
- luaX_syntaxerror(ls,luaO_pushfstring(ls->L,
- "`%s' expected (to close `%s' at line %d)",luaX_token2str(ls,what),
- luaX_token2str(ls,who),where));}}}static TString*str_checkname(LexState*ls){
- TString*ts;check_condition(ls,(ls->t.token==TK_NAME),"<name> expected");ts=ls
- ->t.seminfo.ts;next(ls);return ts;}static void init_exp(expdesc*e,expkind k,
- int i){e->f=e->t=NO_JUMP;e->k=k;e->info=i;}static void codestring(LexState*ls,
- expdesc*e,TString*s){init_exp(e,VK,luaK_stringK(ls->fs,s));}static void
- checkname(LexState*ls,expdesc*e){codestring(ls,e,str_checkname(ls));}static
- int luaI_registerlocalvar(LexState*ls,TString*varname){FuncState*fs=ls->fs;
- Proto*f=fs->f;luaM_growvector(ls->L,f->locvars,fs->nlocvars,f->sizelocvars,
- LocVar,MAX_INT,"");f->locvars[fs->nlocvars].varname=varname;return fs->
- nlocvars++;}static void new_localvar(LexState*ls,TString*name,int n){FuncState
- *fs=ls->fs;luaX_checklimit(ls,fs->nactvar+n+1,MAXVARS,"local variables");fs->
- actvar[fs->nactvar+n]=luaI_registerlocalvar(ls,name);}static void
- adjustlocalvars(LexState*ls,int nvars){FuncState*fs=ls->fs;fs->nactvar+=nvars;
- for(;nvars;nvars--){getlocvar(fs,fs->nactvar-nvars).startpc=fs->pc;}}static
- void removevars(LexState*ls,int tolevel){FuncState*fs=ls->fs;while(fs->nactvar
- >tolevel)getlocvar(fs,--fs->nactvar).endpc=fs->pc;}static void new_localvarstr
- (LexState*ls,const char*name,int n){new_localvar(ls,luaS_new(ls->L,name),n);}
- static void create_local(LexState*ls,const char*name){new_localvarstr(ls,name,
- 0);adjustlocalvars(ls,1);}static int indexupvalue(FuncState*fs,TString*name,
- expdesc*v){int i;Proto*f=fs->f;for(i=0;i<f->nups;i++){if(fs->upvalues[i].k==v
- ->k&&fs->upvalues[i].info==v->info){lua_assert(fs->f->upvalues[i]==name);
- return i;}}luaX_checklimit(fs->ls,f->nups+1,MAXUPVALUES,"upvalues");
- luaM_growvector(fs->L,fs->f->upvalues,f->nups,fs->f->sizeupvalues,TString*,
- MAX_INT,"");fs->f->upvalues[f->nups]=name;fs->upvalues[f->nups]=*v;return f->
- nups++;}static int searchvar(FuncState*fs,TString*n){int i;for(i=fs->nactvar-1
- ;i>=0;i--){if(n==getlocvar(fs,i).varname)return i;}return-1;}static void
- markupval(FuncState*fs,int level){BlockCnt*bl=fs->bl;while(bl&&bl->nactvar>
- level)bl=bl->previous;if(bl)bl->upval=1;}static void singlevaraux(FuncState*fs
- ,TString*n,expdesc*var,int base){if(fs==NULL)init_exp(var,VGLOBAL,NO_REG);else
- {int v=searchvar(fs,n);if(v>=0){init_exp(var,VLOCAL,v);if(!base)markupval(fs,v
- );}else{singlevaraux(fs->prev,n,var,0);if(var->k==VGLOBAL){if(base)var->info=
- luaK_stringK(fs,n);}else{var->info=indexupvalue(fs,n,var);var->k=VUPVAL;}}}}
- static TString*singlevar(LexState*ls,expdesc*var,int base){TString*varname=
- str_checkname(ls);singlevaraux(ls->fs,varname,var,base);return varname;}static
- void adjust_assign(LexState*ls,int nvars,int nexps,expdesc*e){FuncState*fs=ls
- ->fs;int extra=nvars-nexps;if(e->k==VCALL){extra++;if(extra<=0)extra=0;else
- luaK_reserveregs(fs,extra-1);luaK_setcallreturns(fs,e,extra);}else{if(e->k!=
- VVOID)luaK_exp2nextreg(fs,e);if(extra>0){int reg=fs->freereg;luaK_reserveregs(
- fs,extra);luaK_nil(fs,reg,extra);}}}static void code_params(LexState*ls,int
- nparams,int dots){FuncState*fs=ls->fs;adjustlocalvars(ls,nparams);
- luaX_checklimit(ls,fs->nactvar,MAXPARAMS,"parameters");fs->f->numparams=cast(
- lu_byte,fs->nactvar);fs->f->is_vararg=cast(lu_byte,dots);if(dots)create_local(
- ls,"arg");luaK_reserveregs(fs,fs->nactvar);}static void enterblock(FuncState*
- fs,BlockCnt*bl,int isbreakable){bl->breaklist=NO_JUMP;bl->isbreakable=
- isbreakable;bl->nactvar=fs->nactvar;bl->upval=0;bl->previous=fs->bl;fs->bl=bl;
- lua_assert(fs->freereg==fs->nactvar);}static void leaveblock(FuncState*fs){
- BlockCnt*bl=fs->bl;fs->bl=bl->previous;removevars(fs->ls,bl->nactvar);if(bl->
- upval)luaK_codeABC(fs,OP_CLOSE,bl->nactvar,0,0);lua_assert(bl->nactvar==fs->
- nactvar);fs->freereg=fs->nactvar;luaK_patchtohere(fs,bl->breaklist);}static
- void pushclosure(LexState*ls,FuncState*func,expdesc*v){FuncState*fs=ls->fs;
- Proto*f=fs->f;int i;luaM_growvector(ls->L,f->p,fs->np,f->sizep,Proto*,
- MAXARG_Bx,"constant table overflow");f->p[fs->np++]=func->f;init_exp(v,
- VRELOCABLE,luaK_codeABx(fs,OP_CLOSURE,0,fs->np-1));for(i=0;i<func->f->nups;i++
- ){OpCode o=(func->upvalues[i].k==VLOCAL)?OP_MOVE:OP_GETUPVAL;luaK_codeABC(fs,o
- ,0,func->upvalues[i].info,0);}}static void open_func(LexState*ls,FuncState*fs)
- {Proto*f=luaF_newproto(ls->L);fs->f=f;fs->prev=ls->fs;fs->ls=ls;fs->L=ls->L;ls
- ->fs=fs;fs->pc=0;fs->lasttarget=0;fs->jpc=NO_JUMP;fs->freereg=0;fs->nk=0;fs->h
- =luaH_new(ls->L,0,0);fs->np=0;fs->nlocvars=0;fs->nactvar=0;fs->bl=NULL;f->
- source=ls->source;f->maxstacksize=2;}static void close_func(LexState*ls){
- lua_State*L=ls->L;FuncState*fs=ls->fs;Proto*f=fs->f;removevars(ls,0);
- luaK_codeABC(fs,OP_RETURN,0,1,0);luaM_reallocvector(L,f->code,f->sizecode,fs->
- pc,Instruction);f->sizecode=fs->pc;luaM_reallocvector(L,f->lineinfo,f->
- sizelineinfo,fs->pc,int);f->sizelineinfo=fs->pc;luaM_reallocvector(L,f->k,f->
- sizek,fs->nk,TObject);f->sizek=fs->nk;luaM_reallocvector(L,f->p,f->sizep,fs->
- np,Proto*);f->sizep=fs->np;luaM_reallocvector(L,f->locvars,f->sizelocvars,fs->
- nlocvars,LocVar);f->sizelocvars=fs->nlocvars;luaM_reallocvector(L,f->upvalues,
- f->sizeupvalues,f->nups,TString*);f->sizeupvalues=f->nups;lua_assert(
- luaG_checkcode(f));lua_assert(fs->bl==NULL);ls->fs=fs->prev;}Proto*luaY_parser
- (lua_State*L,ZIO*z,Mbuffer*buff){struct LexState lexstate;struct FuncState
- funcstate;lexstate.buff=buff;lexstate.nestlevel=0;luaX_setinput(L,&lexstate,z,
- luaS_new(L,zname(z)));open_func(&lexstate,&funcstate);next(&lexstate);chunk(&
- lexstate);check_condition(&lexstate,(lexstate.t.token==TK_EOS),
- "<eof> expected");close_func(&lexstate);lua_assert(funcstate.prev==NULL);
- lua_assert(funcstate.f->nups==0);lua_assert(lexstate.nestlevel==0);return
- funcstate.f;}static void luaY_field(LexState*ls,expdesc*v){FuncState*fs=ls->fs
- ;expdesc key;luaK_exp2anyreg(fs,v);next(ls);checkname(ls,&key);luaK_indexed(fs
- ,v,&key);}static void luaY_index(LexState*ls,expdesc*v){next(ls);expr(ls,v);
- luaK_exp2val(ls->fs,v);check(ls,']');}struct ConsControl{expdesc v;expdesc*t;
- int nh;int na;int tostore;};static void recfield(LexState*ls,struct
- ConsControl*cc){FuncState*fs=ls->fs;int reg=ls->fs->freereg;expdesc key,val;if
- (ls->t.token==TK_NAME){luaX_checklimit(ls,cc->nh,MAX_INT,
- "items in a constructor");cc->nh++;checkname(ls,&key);}else luaY_index(ls,&key
- );check(ls,'=');luaK_exp2RK(fs,&key);expr(ls,&val);luaK_codeABC(fs,OP_SETTABLE
- ,cc->t->info,luaK_exp2RK(fs,&key),luaK_exp2RK(fs,&val));fs->freereg=reg;}
- static void closelistfield(FuncState*fs,struct ConsControl*cc){if(cc->v.k==
- VVOID)return;luaK_exp2nextreg(fs,&cc->v);cc->v.k=VVOID;if(cc->tostore==
- LFIELDS_PER_FLUSH){luaK_codeABx(fs,OP_SETLIST,cc->t->info,cc->na-1);cc->
- tostore=0;fs->freereg=cc->t->info+1;}}static void lastlistfield(FuncState*fs,
- struct ConsControl*cc){if(cc->tostore==0)return;if(cc->v.k==VCALL){
- luaK_setcallreturns(fs,&cc->v,LUA_MULTRET);luaK_codeABx(fs,OP_SETLISTO,cc->t->
- info,cc->na-1);}else{if(cc->v.k!=VVOID)luaK_exp2nextreg(fs,&cc->v);
- luaK_codeABx(fs,OP_SETLIST,cc->t->info,cc->na-1);}fs->freereg=cc->t->info+1;}
- static void listfield(LexState*ls,struct ConsControl*cc){expr(ls,&cc->v);
- luaX_checklimit(ls,cc->na,MAXARG_Bx,"items in a constructor");cc->na++;cc->
- tostore++;}static void constructor(LexState*ls,expdesc*t){FuncState*fs=ls->fs;
- int line=ls->linenumber;int pc=luaK_codeABC(fs,OP_NEWTABLE,0,0,0);struct
- ConsControl cc;cc.na=cc.nh=cc.tostore=0;cc.t=t;init_exp(t,VRELOCABLE,pc);
- init_exp(&cc.v,VVOID,0);luaK_exp2nextreg(ls->fs,t);check(ls,'{');do{lua_assert
- (cc.v.k==VVOID||cc.tostore>0);testnext(ls,';');if(ls->t.token=='}')break;
- closelistfield(fs,&cc);switch(ls->t.token){case TK_NAME:{lookahead(ls);if(ls->
- lookahead.token!='=')listfield(ls,&cc);else recfield(ls,&cc);break;}case'[':{
- recfield(ls,&cc);break;}default:{listfield(ls,&cc);break;}}}while(testnext(ls,
- ',')||testnext(ls,';'));check_match(ls,'}','{',line);lastlistfield(fs,&cc);
- SETARG_B(fs->f->code[pc],luaO_int2fb(cc.na));SETARG_C(fs->f->code[pc],
- luaO_log2(cc.nh)+1);}static void parlist(LexState*ls){int nparams=0;int dots=0
- ;if(ls->t.token!=')'){do{switch(ls->t.token){case TK_DOTS:dots=1;next(ls);
- break;case TK_NAME:new_localvar(ls,str_checkname(ls),nparams++);break;default:
- luaX_syntaxerror(ls,"<name> or `...' expected");}}while(!dots&&testnext(ls,','
- ));}code_params(ls,nparams,dots);}static void body(LexState*ls,expdesc*e,int
- needself,int line){FuncState new_fs;open_func(ls,&new_fs);new_fs.f->
- lineDefined=line;check(ls,'(');if(needself)create_local(ls,"self");parlist(ls)
- ;check(ls,')');chunk(ls);check_match(ls,TK_END,TK_FUNCTION,line);close_func(ls
- );pushclosure(ls,&new_fs,e);}static int explist1(LexState*ls,expdesc*v){int n=
- 1;expr(ls,v);while(testnext(ls,',')){luaK_exp2nextreg(ls->fs,v);expr(ls,v);n++
- ;}return n;}static void funcargs(LexState*ls,expdesc*f){FuncState*fs=ls->fs;
- expdesc args;int base,nparams;int line=ls->linenumber;switch(ls->t.token){case
- '(':{if(line!=ls->lastline)luaX_syntaxerror(ls,
- "ambiguous syntax (function call x new statement)");next(ls);if(ls->t.token==
- ')')args.k=VVOID;else{explist1(ls,&args);luaK_setcallreturns(fs,&args,
- LUA_MULTRET);}check_match(ls,')','(',line);break;}case'{':{constructor(ls,&
- args);break;}case TK_STRING:{codestring(ls,&args,ls->t.seminfo.ts);next(ls);
- break;}default:{luaX_syntaxerror(ls,"function arguments expected");return;}}
- lua_assert(f->k==VNONRELOC);base=f->info;if(args.k==VCALL)nparams=LUA_MULTRET;
- else{if(args.k!=VVOID)luaK_exp2nextreg(fs,&args);nparams=fs->freereg-(base+1);
- }init_exp(f,VCALL,luaK_codeABC(fs,OP_CALL,base,nparams+1,2));luaK_fixline(fs,
- line);fs->freereg=base+1;}static void prefixexp(LexState*ls,expdesc*v){switch(
- ls->t.token){case'(':{int line=ls->linenumber;next(ls);expr(ls,v);check_match(
- ls,')','(',line);luaK_dischargevars(ls->fs,v);return;}case TK_NAME:{singlevar(
- ls,v,1);return;}
- #ifdef LUA_COMPATUPSYNTAX
- case'%':{TString*varname;int line=ls->linenumber;next(ls);varname=singlevar(ls
- ,v,1);if(v->k!=VUPVAL)luaX_errorline(ls,"global upvalues are obsolete",getstr(
- varname),line);return;}
- #endif
- default:{luaX_syntaxerror(ls,"unexpected symbol");return;}}}static void
- primaryexp(LexState*ls,expdesc*v){FuncState*fs=ls->fs;prefixexp(ls,v);for(;;){
- switch(ls->t.token){case'.':{luaY_field(ls,v);break;}case'[':{expdesc key;
- luaK_exp2anyreg(fs,v);luaY_index(ls,&key);luaK_indexed(fs,v,&key);break;}case
- ':':{expdesc key;next(ls);checkname(ls,&key);luaK_self(fs,v,&key);funcargs(ls,
- v);break;}case'(':case TK_STRING:case'{':{luaK_exp2nextreg(fs,v);funcargs(ls,v
- );break;}default:return;}}}static void simpleexp(LexState*ls,expdesc*v){switch
- (ls->t.token){case TK_NUMBER:{init_exp(v,VK,luaK_numberK(ls->fs,ls->t.seminfo.
- r));next(ls);break;}case TK_STRING:{codestring(ls,v,ls->t.seminfo.ts);next(ls)
- ;break;}case TK_NIL:{init_exp(v,VNIL,0);next(ls);break;}case TK_TRUE:{init_exp
- (v,VTRUE,0);next(ls);break;}case TK_FALSE:{init_exp(v,VFALSE,0);next(ls);break
- ;}case'{':{constructor(ls,v);break;}case TK_FUNCTION:{next(ls);body(ls,v,0,ls
- ->linenumber);break;}default:{primaryexp(ls,v);break;}}}static UnOpr getunopr(
- int op){switch(op){case TK_NOT:return OPR_NOT;case'-':return OPR_MINUS;default
- :return OPR_NOUNOPR;}}static BinOpr getbinopr(int op){switch(op){case'+':
- return OPR_ADD;case'-':return OPR_SUB;case'*':return OPR_MULT;case'/':return
- OPR_DIV;case'^':return OPR_POW;case TK_CONCAT:return OPR_CONCAT;case TK_NE:
- return OPR_NE;case TK_EQ:return OPR_EQ;case'<':return OPR_LT;case TK_LE:return
- OPR_LE;case'>':return OPR_GT;case TK_GE:return OPR_GE;case TK_AND:return
- OPR_AND;case TK_OR:return OPR_OR;default:return OPR_NOBINOPR;}}static const
- struct{lu_byte left;lu_byte right;}priority[]={{6,6},{6,6},{7,7},{7,7},{10,9},
- {5,4},{3,3},{3,3},{3,3},{3,3},{3,3},{3,3},{2,2},{1,1}};
- #define UNARY_PRIORITY 8
- static BinOpr subexpr(LexState*ls,expdesc*v,int limit){BinOpr op;UnOpr uop;
- enterlevel(ls);uop=getunopr(ls->t.token);if(uop!=OPR_NOUNOPR){next(ls);subexpr
- (ls,v,UNARY_PRIORITY);luaK_prefix(ls->fs,uop,v);}else simpleexp(ls,v);op=
- getbinopr(ls->t.token);while(op!=OPR_NOBINOPR&&cast(int,priority[op].left)>
- limit){expdesc v2;BinOpr nextop;next(ls);luaK_infix(ls->fs,op,v);nextop=
- subexpr(ls,&v2,cast(int,priority[op].right));luaK_posfix(ls->fs,op,v,&v2);op=
- nextop;}leavelevel(ls);return op;}static void expr(LexState*ls,expdesc*v){
- subexpr(ls,v,-1);}static int block_follow(int token){switch(token){case
- TK_ELSE:case TK_ELSEIF:case TK_END:case TK_UNTIL:case TK_EOS:return 1;default:
- return 0;}}static void block(LexState*ls){FuncState*fs=ls->fs;BlockCnt bl;
- enterblock(fs,&bl,0);chunk(ls);lua_assert(bl.breaklist==NO_JUMP);leaveblock(fs
- );}struct LHS_assign{struct LHS_assign*prev;expdesc v;};static void
- check_conflict(LexState*ls,struct LHS_assign*lh,expdesc*v){FuncState*fs=ls->fs
- ;int extra=fs->freereg;int conflict=0;for(;lh;lh=lh->prev){if(lh->v.k==
- VINDEXED){if(lh->v.info==v->info){conflict=1;lh->v.info=extra;}if(lh->v.aux==v
- ->info){conflict=1;lh->v.aux=extra;}}}if(conflict){luaK_codeABC(fs,OP_MOVE,fs
- ->freereg,v->info,0);luaK_reserveregs(fs,1);}}static void assignment(LexState*
- ls,struct LHS_assign*lh,int nvars){expdesc e;check_condition(ls,VLOCAL<=lh->v.
- k&&lh->v.k<=VINDEXED,"syntax error");if(testnext(ls,',')){struct LHS_assign nv
- ;nv.prev=lh;primaryexp(ls,&nv.v);if(nv.v.k==VLOCAL)check_conflict(ls,lh,&nv.v)
- ;assignment(ls,&nv,nvars+1);}else{int nexps;check(ls,'=');nexps=explist1(ls,&e
- );if(nexps!=nvars){adjust_assign(ls,nvars,nexps,&e);if(nexps>nvars)ls->fs->
- freereg-=nexps-nvars;}else{luaK_setcallreturns(ls->fs,&e,1);luaK_storevar(ls->
- fs,&lh->v,&e);return;}}init_exp(&e,VNONRELOC,ls->fs->freereg-1);luaK_storevar(
- ls->fs,&lh->v,&e);}static void cond(LexState*ls,expdesc*v){expr(ls,v);if(v->k
- ==VNIL)v->k=VFALSE;luaK_goiftrue(ls->fs,v);luaK_patchtohere(ls->fs,v->t);}
- #ifndef MAXEXPWHILE
- #define MAXEXPWHILE 100
- #endif
- #define EXTRAEXP 5
- static void whilestat(LexState*ls,int line){Instruction codeexp[MAXEXPWHILE+
- EXTRAEXP];int lineexp;int i;int sizeexp;FuncState*fs=ls->fs;int whileinit,
- blockinit,expinit;expdesc v;BlockCnt bl;next(ls);whileinit=luaK_jump(fs);
- expinit=luaK_getlabel(fs);expr(ls,&v);if(v.k==VK)v.k=VTRUE;lineexp=ls->
- linenumber;luaK_goiffalse(fs,&v);luaK_concat(fs,&v.f,fs->jpc);fs->jpc=NO_JUMP;
- sizeexp=fs->pc-expinit;if(sizeexp>MAXEXPWHILE)luaX_syntaxerror(ls,
- "`while' condition too complex");for(i=0;i<sizeexp;i++)codeexp[i]=fs->f->code[
- expinit+i];fs->pc=expinit;enterblock(fs,&bl,1);check(ls,TK_DO);blockinit=
- luaK_getlabel(fs);block(ls);luaK_patchtohere(fs,whileinit);if(v.t!=NO_JUMP)v.t
- +=fs->pc-expinit;if(v.f!=NO_JUMP)v.f+=fs->pc-expinit;for(i=0;i<sizeexp;i++)
- luaK_code(fs,codeexp[i],lineexp);check_match(ls,TK_END,TK_WHILE,line);
- leaveblock(fs);luaK_patchlist(fs,v.t,blockinit);luaK_patchtohere(fs,v.f);}
- static void repeatstat(LexState*ls,int line){FuncState*fs=ls->fs;int
- repeat_init=luaK_getlabel(fs);expdesc v;BlockCnt bl;enterblock(fs,&bl,1);next(
- ls);block(ls);check_match(ls,TK_UNTIL,TK_REPEAT,line);cond(ls,&v);
- luaK_patchlist(fs,v.f,repeat_init);leaveblock(fs);}static int exp1(LexState*ls
- ){expdesc e;int k;expr(ls,&e);k=e.k;luaK_exp2nextreg(ls->fs,&e);return k;}
- static void forbody(LexState*ls,int base,int line,int nvars,int isnum){
- BlockCnt bl;FuncState*fs=ls->fs;int prep,endfor;adjustlocalvars(ls,nvars);
- check(ls,TK_DO);enterblock(fs,&bl,1);prep=luaK_getlabel(fs);block(ls);
- luaK_patchtohere(fs,prep-1);endfor=(isnum)?luaK_codeAsBx(fs,OP_FORLOOP,base,
- NO_JUMP):luaK_codeABC(fs,OP_TFORLOOP,base,0,nvars-3);luaK_fixline(fs,line);
- luaK_patchlist(fs,(isnum)?endfor:luaK_jump(fs),prep);leaveblock(fs);}static
- void fornum(LexState*ls,TString*varname,int line){FuncState*fs=ls->fs;int base
- =fs->freereg;new_localvar(ls,varname,0);new_localvarstr(ls,"(for limit)",1);
- new_localvarstr(ls,"(for step)",2);check(ls,'=');exp1(ls);check(ls,',');exp1(
- ls);if(testnext(ls,','))exp1(ls);else{luaK_codeABx(fs,OP_LOADK,fs->freereg,
- luaK_numberK(fs,1));luaK_reserveregs(fs,1);}luaK_codeABC(fs,OP_SUB,fs->freereg
- -3,fs->freereg-3,fs->freereg-1);luaK_jump(fs);forbody(ls,base,line,3,1);}
- static void forlist(LexState*ls,TString*indexname){FuncState*fs=ls->fs;expdesc
- e;int nvars=0;int line;int base=fs->freereg;new_localvarstr(ls,
- "(for generator)",nvars++);new_localvarstr(ls,"(for state)",nvars++);
- new_localvar(ls,indexname,nvars++);while(testnext(ls,','))new_localvar(ls,
- str_checkname(ls),nvars++);check(ls,TK_IN);line=ls->linenumber;adjust_assign(
- ls,nvars,explist1(ls,&e),&e);luaK_checkstack(fs,3);luaK_codeAsBx(fs,
- OP_TFORPREP,base,NO_JUMP);forbody(ls,base,line,nvars,0);}static void forstat(
- LexState*ls,int line){FuncState*fs=ls->fs;TString*varname;BlockCnt bl;
- enterblock(fs,&bl,0);next(ls);varname=str_checkname(ls);switch(ls->t.token){
- case'=':fornum(ls,varname,line);break;case',':case TK_IN:forlist(ls,varname);
- break;default:luaX_syntaxerror(ls,"`=' or `in' expected");}check_match(ls,
- TK_END,TK_FOR,line);leaveblock(fs);}static void test_then_block(LexState*ls,
- expdesc*v){next(ls);cond(ls,v);check(ls,TK_THEN);block(ls);}static void ifstat
- (LexState*ls,int line){FuncState*fs=ls->fs;expdesc v;int escapelist=NO_JUMP;
- test_then_block(ls,&v);while(ls->t.token==TK_ELSEIF){luaK_concat(fs,&
- escapelist,luaK_jump(fs));luaK_patchtohere(fs,v.f);test_then_block(ls,&v);}if(
- ls->t.token==TK_ELSE){luaK_concat(fs,&escapelist,luaK_jump(fs));
- luaK_patchtohere(fs,v.f);next(ls);block(ls);}else luaK_concat(fs,&escapelist,v
- .f);luaK_patchtohere(fs,escapelist);check_match(ls,TK_END,TK_IF,line);}static
- void localfunc(LexState*ls){expdesc v,b;FuncState*fs=ls->fs;new_localvar(ls,
- str_checkname(ls),0);init_exp(&v,VLOCAL,fs->freereg);luaK_reserveregs(fs,1);
- adjustlocalvars(ls,1);body(ls,&b,0,ls->linenumber);luaK_storevar(fs,&v,&b);
- getlocvar(fs,fs->nactvar-1).startpc=fs->pc;}static void localstat(LexState*ls)
- {int nvars=0;int nexps;expdesc e;do{new_localvar(ls,str_checkname(ls),nvars++)
- ;}while(testnext(ls,','));if(testnext(ls,'='))nexps=explist1(ls,&e);else{e.k=
- VVOID;nexps=0;}adjust_assign(ls,nvars,nexps,&e);adjustlocalvars(ls,nvars);}
- static int funcname(LexState*ls,expdesc*v){int needself=0;singlevar(ls,v,1);
- while(ls->t.token=='.')luaY_field(ls,v);if(ls->t.token==':'){needself=1;
- luaY_field(ls,v);}return needself;}static void funcstat(LexState*ls,int line){
- int needself;expdesc v,b;next(ls);needself=funcname(ls,&v);body(ls,&b,needself
- ,line);luaK_storevar(ls->fs,&v,&b);luaK_fixline(ls->fs,line);}static void
- exprstat(LexState*ls){FuncState*fs=ls->fs;struct LHS_assign v;primaryexp(ls,&v
- .v);if(v.v.k==VCALL){luaK_setcallreturns(fs,&v.v,0);}else{v.prev=NULL;
- assignment(ls,&v,1);}}static void retstat(LexState*ls){FuncState*fs=ls->fs;
- expdesc e;int first,nret;next(ls);if(block_follow(ls->t.token)||ls->t.token==
- ';')first=nret=0;else{nret=explist1(ls,&e);if(e.k==VCALL){luaK_setcallreturns(
- fs,&e,LUA_MULTRET);if(nret==1){SET_OPCODE(getcode(fs,&e),OP_TAILCALL);
- lua_assert(GETARG_A(getcode(fs,&e))==fs->nactvar);}first=fs->nactvar;nret=
- LUA_MULTRET;}else{if(nret==1)first=luaK_exp2anyreg(fs,&e);else{
- luaK_exp2nextreg(fs,&e);first=fs->nactvar;lua_assert(nret==fs->freereg-first);
- }}}luaK_codeABC(fs,OP_RETURN,first,nret+1,0);}static void breakstat(LexState*
- ls){FuncState*fs=ls->fs;BlockCnt*bl=fs->bl;int upval=0;next(ls);while(bl&&!bl
- ->isbreakable){upval|=bl->upval;bl=bl->previous;}if(!bl)luaX_syntaxerror(ls,
- "no loop to break");if(upval)luaK_codeABC(fs,OP_CLOSE,bl->nactvar,0,0);
- luaK_concat(fs,&bl->breaklist,luaK_jump(fs));}static int statement(LexState*ls
- ){int line=ls->linenumber;switch(ls->t.token){case TK_IF:{ifstat(ls,line);
- return 0;}case TK_WHILE:{whilestat(ls,line);return 0;}case TK_DO:{next(ls);
- block(ls);check_match(ls,TK_END,TK_DO,line);return 0;}case TK_FOR:{forstat(ls,
- line);return 0;}case TK_REPEAT:{repeatstat(ls,line);return 0;}case TK_FUNCTION
- :{funcstat(ls,line);return 0;}case TK_LOCAL:{next(ls);if(testnext(ls,
- TK_FUNCTION))localfunc(ls);else localstat(ls);return 0;}case TK_RETURN:{
- retstat(ls);return 1;}case TK_BREAK:{breakstat(ls);return 1;}default:{exprstat
- (ls);return 0;}}}static void chunk(LexState*ls){int islast=0;enterlevel(ls);
- while(!islast&&!block_follow(ls->t.token)){islast=statement(ls);testnext(ls,
- ';');lua_assert(ls->fs->freereg>=ls->fs->nactvar);ls->fs->freereg=ls->fs->
- nactvar;}leavelevel(ls);}
- #line 1 "lposix.c"
- #define MYNAME "posix"
- #define MYVERSION MYNAME" library for "LUA_VERSION" / Nov 2003"
- #ifndef MYBUFSIZ
- #define MYBUFSIZ 512
- #endif
- #line 1 "modemuncher.h"
- struct modeLookup{char rwx;mode_t bits;};typedef struct modeLookup modeLookup;
- static modeLookup modesel[]={{'r',S_IRUSR},{'w',S_IWUSR},{'x',S_IXUSR},{'r',
- S_IRGRP},{'w',S_IWGRP},{'x',S_IXGRP},{'r',S_IROTH},{'w',S_IWOTH},{'x',S_IXOTH}
- ,{(char)NULL,(mode_t)-1}};static int rwxrwxrwx(mode_t*mode,const char*p){int
- count;mode_t tmp_mode=*mode;tmp_mode&=~(S_ISUID|S_ISGID);for(count=0;count<9;
- count++){if(*p==modesel[count].rwx)tmp_mode|=modesel[count].bits;else if(*p==
- '-')tmp_mode&=~modesel[count].bits;else if(*p=='s')switch(count){case 2:
- tmp_mode|=S_ISUID|S_IXUSR;break;case 5:tmp_mode|=S_ISGID|S_IXGRP;break;default
- :return-4;break;}p++;}*mode=tmp_mode;return 0;}static void modechopper(mode_t
- mode,char*p){int count;char*pp;pp=p;for(count=0;count<9;count++){if(mode&
- modesel[count].bits)*p=modesel[count].rwx;else*p='-';p++;}*p=0;if(mode&S_ISUID
- )pp[2]=(mode&S_IXUSR)?'s':'S';if(mode&S_ISGID)pp[5]=(mode&S_IXGRP)?'s':'S';}
- static int mode_munch(mode_t*mode,const char*p){char op=0;mode_t affected_bits
- ,ch_mode;int doneFlag=0;
- #ifdef DEBUG
- char tmp[10];
- #endif
- #ifdef DEBUG
- modechopper(*mode,tmp);printf("modemuncher: got base mode = %s\n",tmp);
- #endif
- while(!doneFlag){affected_bits=0;ch_mode=0;
- #ifdef DEBUG
- printf("modemuncher step 1\n");
- #endif
- if(*p=='r'||*p=='-')return rwxrwxrwx(mode,p);for(;;p++)switch(*p){case'u':
- affected_bits|=04700;break;case'g':affected_bits|=02070;break;case'o':
- affected_bits|=01007;break;case'a':affected_bits|=07777;break;case' ':break;
- default:goto no_more_affected;}no_more_affected:if(affected_bits==0)
- affected_bits=07777;
- #ifdef DEBUG
- printf("modemuncher step 2 (*p='%c')\n",*p);
- #endif
- switch(*p){case'+':case'-':case'=':op=*p;break;case' ':break;default:return-1;
- }
- #ifdef DEBUG
- printf("modemuncher step 3\n");
- #endif
- for(p++;*p!=0;p++)switch(*p){case'r':ch_mode|=00444;break;case'w':ch_mode|=
- 00222;break;case'x':ch_mode|=00111;break;case's':ch_mode|=06000;break;case' ':
- break;default:goto specs_done;}specs_done:
- #ifdef DEBUG
- printf("modemuncher step 4\n");
- #endif
- if(*p!=',')doneFlag=1;if(*p!=0&&*p!=' '&&*p!=','){
- #ifdef DEBUG
- printf("modemuncher: comma error!\n");printf("modemuncher: doneflag = %u\n",
- doneFlag);
- #endif
- return-2;}p++;if(ch_mode)switch(op){case'+':*mode=*mode|=ch_mode&affected_bits
- ;break;case'-':*mode=*mode&=~(ch_mode&affected_bits);break;case'=':*mode=
- ch_mode&affected_bits;break;default:return-3;}}
- #ifdef DEBUG
- modechopper(*mode,tmp);printf("modemuncher: returning mode = %s\n",tmp);
- #endif
- return 0;}
- #line 38 "lposix.c"
- static const char*filetype(mode_t m){if(S_ISREG(m))return"regular";else if(
- S_ISLNK(m))return"link";else if(S_ISDIR(m))return"directory";else if(S_ISCHR(m
- ))return"character device";else if(S_ISBLK(m))return"block device";else if(
- S_ISFIFO(m))return"fifo";
- #ifdef S_ISSOCK
- else if(S_ISSOCK(m))return"socket";
- #endif
- else return"?";}typedef int(*Selector)(lua_State*L,int i,const void*data);
- static int doselection(lua_State*L,int i,const char*const S[],Selector F,const
- void*data){if(lua_isnone(L,i)){lua_newtable(L);for(i=0;S[i]!=NULL;i++){
- lua_pushstring(L,S[i]);F(L,i,data);lua_settable(L,-3);}return 1;}else{int j=
- luaL_findstring(luaL_checkstring(L,i),S);if(j==-1)luaL_argerror(L,i,
- "unknown selector");return F(L,j,data);}}static void storeindex(lua_State*L,
- int i,const char*value){lua_pushstring(L,value);lua_rawseti(L,-2,i);}static
- void storestring(lua_State*L,const char*name,const char*value){lua_pushstring(
- L,name);lua_pushstring(L,value);lua_settable(L,-3);}static void storenumber(
- lua_State*L,const char*name,lua_Number value){lua_pushstring(L,name);
- lua_pushnumber(L,value);lua_settable(L,-3);}static int pusherror(lua_State*L,
- const char*info){lua_pushnil(L);if(info==NULL)lua_pushstring(L,strerror(errno)
- );else lua_pushfstring(L,"%s: %s",info,strerror(errno));lua_pushnumber(L,errno
- );return 3;}static int lposix_pushresult(lua_State*L,int i,const char*info){if
- (i!=-1){lua_pushnumber(L,i);return 1;}else return pusherror(L,info);}static
- void badoption(lua_State*L,int i,const char*what,int option){luaL_argerror(L,2
- ,lua_pushfstring(L,"unknown %s option `%c'",what,option));}static uid_t
- mygetuid(lua_State*L,int i){if(lua_isnone(L,i))return-1;else if(lua_isnumber(L
- ,i))return(uid_t)lua_tonumber(L,i);else if(lua_isstring(L,i)){struct passwd*p=
- getpwnam(lua_tostring(L,i));return(p==NULL)?-1:p->pw_uid;}else return
- luaL_typerror(L,i,"string or number");}static gid_t mygetgid(lua_State*L,int i
- ){if(lua_isnone(L,i))return-1;else if(lua_isnumber(L,i))return(gid_t)
- lua_tonumber(L,i);else if(lua_isstring(L,i)){struct group*g=getgrnam(
- lua_tostring(L,i));return(g==NULL)?-1:g->gr_gid;}else return luaL_typerror(L,i
- ,"string or number");}static int Perrno(lua_State*L){lua_pushstring(L,strerror
- (errno));lua_pushnumber(L,errno);return 2;}static int Pdir(lua_State*L){const
- char*path=luaL_optstring(L,1,".");DIR*d=opendir(path);if(d==NULL)return
- pusherror(L,path);else{int i;struct dirent*entry;lua_newtable(L);for(i=1;(
- entry=readdir(d))!=NULL;i++)storeindex(L,i,entry->d_name);closedir(d);return 1
- ;}}static int aux_files(lua_State*L){DIR*d=lua_touserdata(L,lua_upvalueindex(1
- ));struct dirent*entry;if(d==NULL)luaL_error(L,"attempt to use closed dir");
- entry=readdir(d);if(entry==NULL){closedir(d);lua_pushnil(L);lua_replace(L,
- lua_upvalueindex(1));lua_pushnil(L);}else{lua_pushstring(L,entry->d_name);
- #if 0
- #ifdef _DIRENT_HAVE_D_TYPE
- lua_pushstring(L,filetype(DTTOIF(entry->d_type)));return 2;
- #endif
- #endif
- }return 1;}static int Pfiles(lua_State*L){const char*path=luaL_optstring(L,1,
- ".");DIR*d=opendir(path);if(d==NULL)return pusherror(L,path);else{
- lua_pushlightuserdata(L,d);lua_pushcclosure(L,aux_files,1);return 1;}}static
- int Pgetcwd(lua_State*L){char buf[MYBUFSIZ];if(getcwd(buf,sizeof(buf))==NULL)
- return pusherror(L,".");else{lua_pushstring(L,buf);return 1;}}static int
- Pmkdir(lua_State*L){const char*path=luaL_checkstring(L,1);return
- lposix_pushresult(L,mkdir(path,0777),path);}static int Pchdir(lua_State*L){
- const char*path=luaL_checkstring(L,1);return lposix_pushresult(L,chdir(path),
- path);}static int Prmdir(lua_State*L){const char*path=luaL_checkstring(L,1);
- return lposix_pushresult(L,rmdir(path),path);}static int Punlink(lua_State*L){
- const char*path=luaL_checkstring(L,1);return lposix_pushresult(L,unlink(path),
- path);}static int Plink(lua_State*L){const char*oldpath=luaL_checkstring(L,1);
- const char*newpath=luaL_checkstring(L,2);return lposix_pushresult(L,link(
- oldpath,newpath),NULL);}static int Psymlink(lua_State*L){const char*oldpath=
- luaL_checkstring(L,1);const char*newpath=luaL_checkstring(L,2);return
- lposix_pushresult(L,symlink(oldpath,newpath),NULL);}static int Preadlink(
- lua_State*L){char buf[MYBUFSIZ];const char*path=luaL_checkstring(L,1);int n=
- readlink(path,buf,sizeof(buf));if(n==-1)return pusherror(L,path);
- lua_pushlstring(L,buf,n);return 1;}static int Paccess(lua_State*L){int mode=
- F_OK;const char*path=luaL_checkstring(L,1);const char*s;for(s=luaL_optstring(L
- ,2,"f");*s!=0;s++)switch(*s){case' ':break;case'r':mode|=R_OK;break;case'w':
- mode|=W_OK;break;case'x':mode|=X_OK;break;case'f':mode|=F_OK;break;default:
- badoption(L,2,"mode",*s);break;}return lposix_pushresult(L,access(path,mode),
- path);}static int Pmkfifo(lua_State*L){const char*path=luaL_checkstring(L,1);
- return lposix_pushresult(L,mkfifo(path,0777),path);}static int Pexec(lua_State
- *L){const char*path=luaL_checkstring(L,1);int i,n=lua_gettop(L);char**argv=
- malloc((n+1)*sizeof(char*));if(argv==NULL)luaL_error(L,"not enough memory");
- argv[0]=(char*)path;for(i=1;i<n;i++)argv[i]=(char*)luaL_checkstring(L,i+1);
- argv[i]=NULL;execvp(path,argv);return pusherror(L,path);}static int Pfork(
- lua_State*L){return lposix_pushresult(L,fork(),NULL);}static int Pwait(
- lua_State*L){pid_t pid=luaL_optint(L,1,-1);return lposix_pushresult(L,waitpid(
- pid,NULL,0),NULL);}static int Pkill(lua_State*L){pid_t pid=luaL_checkint(L,1);
- int sig=luaL_optint(L,2,SIGTERM);return lposix_pushresult(L,kill(pid,sig),NULL
- );}static int Psleep(lua_State*L){unsigned int seconds=luaL_checkint(L,1);
- lua_pushnumber(L,sleep(seconds));return 1;}static int Pputenv(lua_State*L){
- size_t l;const char*s=luaL_checklstring(L,1,&l);char*e=malloc(++l);return
- lposix_pushresult(L,(e==NULL)?-1:putenv(memcpy(e,s,l)),s);}
- #ifdef linux
- static int Psetenv(lua_State*L){const char*name=luaL_checkstring(L,1);const
- char*value=luaL_checkstring(L,2);int overwrite=lua_isnoneornil(L,3)||
- lua_toboolean(L,3);return lposix_pushresult(L,setenv(name,value,overwrite),
- name);}static int Punsetenv(lua_State*L){const char*name=luaL_checkstring(L,1)
- ;unsetenv(name);return 0;}
- #endif
- static int Pgetenv(lua_State*L){if(lua_isnone(L,1)){extern char**environ;char*
- *e;if(*environ==NULL)lua_pushnil(L);else lua_newtable(L);for(e=environ;*e!=
- NULL;e++){char*s=*e;char*eq=strchr(s,'=');if(eq==NULL){lua_pushstring(L,s);
- lua_pushboolean(L,0);}else{lua_pushlstring(L,s,eq-s);lua_pushstring(L,eq+1);}
- lua_settable(L,-3);}}else lua_pushstring(L,getenv(luaL_checkstring(L,1)));
- return 1;}static int Pumask(lua_State*L){char m[10];mode_t mode;umask(mode=
- umask(0));mode=(~mode)&0777;if(!lua_isnone(L,1)){if(mode_munch(&mode,
- luaL_checkstring(L,1))){lua_pushnil(L);return 1;}mode&=0777;umask(~mode);}
- modechopper(mode,m);lua_pushstring(L,m);return 1;}static int Pchmod(lua_State*
- L){mode_t mode;struct stat s;const char*path=luaL_checkstring(L,1);const char*
- modestr=luaL_checkstring(L,2);if(stat(path,&s))return pusherror(L,path);mode=s
- .st_mode;if(mode_munch(&mode,modestr))luaL_argerror(L,2,"bad mode");return
- lposix_pushresult(L,chmod(path,mode),path);}static int Pchown(lua_State*L){
- const char*path=luaL_checkstring(L,1);uid_t uid=mygetuid(L,2);gid_t gid=
- mygetgid(L,3);return lposix_pushresult(L,chown(path,uid,gid),path);}static int
- Putime(lua_State*L){struct utimbuf times;time_t currtime=time(NULL);const
- char*path=luaL_checkstring(L,1);times.modtime=luaL_optnumber(L,2,currtime);
- times.actime=luaL_optnumber(L,3,currtime);return lposix_pushresult(L,utime(
- path,×),path);}static int FgetID(lua_State*L,int i,const void*data){
- switch(i){case 0:lua_pushnumber(L,getegid());break;case 1:lua_pushnumber(L,
- geteuid());break;case 2:lua_pushnumber(L,getgid());break;case 3:lua_pushnumber
- (L,getuid());break;case 4:lua_pushnumber(L,getpgrp());break;case 5:
- lua_pushnumber(L,getpid());break;case 6:lua_pushnumber(L,getppid());break;}
- return 1;}static const char*const SgetID[]={"egid","euid","gid","uid","pgrp",
- "pid","ppid",NULL};static int Pgetprocessid(lua_State*L){return doselection(L,
- 1,SgetID,FgetID,NULL);}static int Pttyname(lua_State*L){int fd=luaL_optint(L,1
- ,0);lua_pushstring(L,ttyname(fd));return 1;}static int Pctermid(lua_State*L){
- char b[L_ctermid];lua_pushstring(L,ctermid(b));return 1;}static int Pgetlogin(
- lua_State*L){lua_pushstring(L,getlogin());return 1;}static int Fgetpasswd(
- lua_State*L,int i,const void*data){const struct passwd*p=data;switch(i){case 0
- :lua_pushstring(L,p->pw_name);break;case 1:lua_pushnumber(L,p->pw_uid);break;
- case 2:lua_pushnumber(L,p->pw_gid);break;case 3:lua_pushstring(L,p->pw_dir);
- break;case 4:lua_pushstring(L,p->pw_shell);break;case 5:lua_pushstring(L,p->
- pw_gecos);break;case 6:lua_pushstring(L,p->pw_passwd);break;}return 1;}static
- const char*const Sgetpasswd[]={"name","uid","gid","dir","shell","gecos",
- "passwd",NULL};static int Pgetpasswd(lua_State*L){struct passwd*p=NULL;if(
- lua_isnoneornil(L,1))p=getpwuid(geteuid());else if(lua_isnumber(L,1))p=
- getpwuid((uid_t)lua_tonumber(L,1));else if(lua_isstring(L,1))p=getpwnam(
- lua_tostring(L,1));else luaL_typerror(L,1,"string or number");if(p==NULL)
- lua_pushnil(L);else doselection(L,2,Sgetpasswd,Fgetpasswd,p);return 1;}static
- int Pgetgroup(lua_State*L){struct group*g=NULL;if(lua_isnumber(L,1))g=getgrgid
- ((gid_t)lua_tonumber(L,1));else if(lua_isstring(L,1))g=getgrnam(lua_tostring(L
- ,1));else luaL_typerror(L,1,"string or number");if(g==NULL)lua_pushnil(L);else
- {int i;lua_newtable(L);storestring(L,"name",g->gr_name);storenumber(L,"gid",g
- ->gr_gid);for(i=0;g->gr_mem[i]!=NULL;i++)storeindex(L,i+1,g->gr_mem[i]);}
- return 1;}static int Psetuid(lua_State*L){return lposix_pushresult(L,setuid(
- mygetuid(L,1)),NULL);}static int Psetgid(lua_State*L){return lposix_pushresult
- (L,setgid(mygetgid(L,1)),NULL);}struct mytimes{struct tms t;clock_t elapsed;};
- #define pushtime(L,x) lua_pushnumber(L,((lua_Number)x)/CLOCKS_PER_SEC)
- static int Ftimes(lua_State*L,int i,const void*data){const struct mytimes*t=
- data;switch(i){case 0:pushtime(L,t->t.tms_utime);break;case 1:pushtime(L,t->t.
- tms_stime);break;case 2:pushtime(L,t->t.tms_cutime);break;case 3:pushtime(L,t
- ->t.tms_cstime);break;case 4:pushtime(L,t->elapsed);break;}return 1;}static
- const char*const Stimes[]={"utime","stime","cutime","cstime","elapsed",NULL};
- #define storetime(L,name,x) storenumber(L,name,(lua_Number)x/CLK_TCK)
- static int Ptimes(lua_State*L){struct mytimes t;t.elapsed=times(&t.t);return
- doselection(L,1,Stimes,Ftimes,&t);}struct mystat{struct stat s;char mode[10];
- const char*type;};static int Fstat(lua_State*L,int i,const void*data){const
- struct mystat*s=data;switch(i){case 0:lua_pushstring(L,s->mode);break;case 1:
- lua_pushnumber(L,s->s.st_ino);break;case 2:lua_pushnumber(L,s->s.st_dev);break
- ;case 3:lua_pushnumber(L,s->s.st_nlink);break;case 4:lua_pushnumber(L,s->s.
- st_uid);break;case 5:lua_pushnumber(L,s->s.st_gid);break;case 6:lua_pushnumber
- (L,s->s.st_size);break;case 7:lua_pushnumber(L,s->s.st_atime);break;case 8:
- lua_pushnumber(L,s->s.st_mtime);break;case 9:lua_pushnumber(L,s->s.st_ctime);
- break;case 10:lua_pushstring(L,s->type);break;case 11:lua_pushnumber(L,s->s.
- st_mode);break;}return 1;}static const char*const Sstat[]={"mode","ino","dev",
- "nlink","uid","gid","size","atime","mtime","ctime","type","_mode",NULL};static
- int Pstat(lua_State*L){struct mystat s;const char*path=luaL_checkstring(L,1);
- if(stat(path,&s.s)==-1)return pusherror(L,path);s.type=filetype(s.s.st_mode);
- modechopper(s.s.st_mode,s.mode);return doselection(L,2,Sstat,Fstat,&s);}static
- int Puname(lua_State*L){struct utsname u;luaL_Buffer b;const char*s;if(uname(
- &u)==-1)return pusherror(L,NULL);luaL_buffinit(L,&b);for(s=luaL_optstring(L,1,
- "%s %n %r %v %m");*s;s++)if(*s!='%')luaL_putchar(&b,*s);else switch(*++s){case
- '%':luaL_putchar(&b,*s);break;case'm':luaL_addstring(&b,u.machine);break;case
- 'n':luaL_addstring(&b,u.nodename);break;case'r':luaL_addstring(&b,u.release);
- break;case's':luaL_addstring(&b,u.sysname);break;case'v':luaL_addstring(&b,u.
- version);break;default:badoption(L,2,"format",*s);break;}luaL_pushresult(&b);
- return 1;}static const int Kpathconf[]={_PC_LINK_MAX,_PC_MAX_CANON,
- _PC_MAX_INPUT,_PC_NAME_MAX,_PC_PATH_MAX,_PC_PIPE_BUF,_PC_CHOWN_RESTRICTED,
- _PC_NO_TRUNC,_PC_VDISABLE,-1};static int Fpathconf(lua_State*L,int i,const
- void*data){const char*path=data;lua_pushnumber(L,pathconf(path,Kpathconf[i]));
- return 1;}static const char*const Spathconf[]={"link_max","max_canon",
- "max_input","name_max","path_max","pipe_buf","chown_restricted","no_trunc",
- "vdisable",NULL};static int Ppathconf(lua_State*L){const char*path=
- luaL_checkstring(L,1);return doselection(L,2,Spathconf,Fpathconf,path);}static
- const int Ksysconf[]={_SC_ARG_MAX,_SC_CHILD_MAX,_SC_CLK_TCK,_SC_NGROUPS_MAX,
- _SC_STREAM_MAX,_SC_TZNAME_MAX,_SC_OPEN_MAX,_SC_JOB_CONTROL,_SC_SAVED_IDS,
- _SC_VERSION,-1};static int Fsysconf(lua_State*L,int i,const void*data){
- lua_pushnumber(L,sysconf(Ksysconf[i]));return 1;}static const char*const
- Ssysconf[]={"arg_max","child_max","clk_tck","ngroups_max","stream_max",
- "tzname_max","open_max","job_control","saved_ids","version",NULL};static int
- Psysconf(lua_State*L){return doselection(L,1,Ssysconf,Fsysconf,NULL);}static
- const luaL_reg R[]={{"access",Paccess},{"chdir",Pchdir},{"chmod",Pchmod},{
- "chown",Pchown},{"ctermid",Pctermid},{"dir",Pdir},{"errno",Perrno},{"exec",
- Pexec},{"files",Pfiles},{"fork",Pfork},{"getcwd",Pgetcwd},{"getenv",Pgetenv},{
- "getgroup",Pgetgroup},{"getlogin",Pgetlogin},{"getpasswd",Pgetpasswd},{
- "getprocessid",Pgetprocessid},{"kill",Pkill},{"link",Plink},{"mkdir",Pmkdir},{
- "mkfifo",Pmkfifo},{"pathconf",Ppathconf},{"putenv",Pputenv},{"readlink",
- Preadlink},{"rmdir",Prmdir},{"setgid",Psetgid},{"setuid",Psetuid},{"sleep",
- Psleep},{"stat",Pstat},{"symlink",Psymlink},{"sysconf",Psysconf},{"times",
- Ptimes},{"ttyname",Pttyname},{"umask",Pumask},{"uname",Puname},{"unlink",
- Punlink},{"utime",Putime},{"wait",Pwait},
- #ifdef linux
- {"setenv",Psetenv},{"unsetenv",Punsetenv},
- #endif
- {NULL,NULL}};LUALIB_API int luaopen_posix(lua_State*L){luaL_openlib(L,MYNAME,R
- ,0);lua_pushliteral(L,"version");lua_pushliteral(L,MYVERSION);lua_settable(L,-
- 3);return 1;}
- #line 1 "lstate.c"
- #define lstate_c
- #ifndef LUA_USERSTATE
- #define EXTRASPACE 0
- #else
- union UEXTRASPACE{L_Umaxalign a;LUA_USERSTATE b;};
- #define EXTRASPACE (sizeof(union UEXTRASPACE))
- #endif
- static int default_panic(lua_State*L){UNUSED(L);return 0;}static lua_State*
- mallocstate(lua_State*L){lu_byte*block=(lu_byte*)luaM_malloc(L,sizeof(
- lua_State)+EXTRASPACE);if(block==NULL)return NULL;else{block+=EXTRASPACE;
- return cast(lua_State*,block);}}static void freestate(lua_State*L,lua_State*L1
- ){luaM_free(L,cast(lu_byte*,L1)-EXTRASPACE,sizeof(lua_State)+EXTRASPACE);}
- static void stack_init(lua_State*L1,lua_State*L){L1->stack=luaM_newvector(L,
- BASIC_STACK_SIZE+EXTRA_STACK,TObject);L1->stacksize=BASIC_STACK_SIZE+
- EXTRA_STACK;L1->top=L1->stack;L1->stack_last=L1->stack+(L1->stacksize-
- EXTRA_STACK)-1;L1->base_ci=luaM_newvector(L,BASIC_CI_SIZE,CallInfo);L1->ci=L1
- ->base_ci;L1->ci->state=CI_C;setnilvalue(L1->top++);L1->base=L1->ci->base=L1->
- top;L1->ci->top=L1->top+LUA_MINSTACK;L1->size_ci=BASIC_CI_SIZE;L1->end_ci=L1->
- base_ci+L1->size_ci;}static void freestack(lua_State*L,lua_State*L1){
- luaM_freearray(L,L1->base_ci,L1->size_ci,CallInfo);luaM_freearray(L,L1->stack,
- L1->stacksize,TObject);}static void f_luaopen(lua_State*L,void*ud){
- global_State*g=luaM_new(NULL,global_State);UNUSED(ud);if(g==NULL)luaD_throw(L,
- LUA_ERRMEM);L->l_G=g;g->mainthread=L;g->GCthreshold=0;g->strt.size=0;g->strt.
- nuse=0;g->strt.hash=NULL;setnilvalue(defaultmeta(L));setnilvalue(registry(L));
- luaZ_initbuffer(L,&g->buff);g->panic=default_panic;g->rootgc=NULL;g->rootudata
- =NULL;g->tmudata=NULL;setnilvalue(gkey(g->dummynode));setnilvalue(gval(g->
- dummynode));g->dummynode->next=NULL;g->nblocks=sizeof(lua_State)+sizeof(
- global_State);stack_init(L,L);defaultmeta(L)->tt=LUA_TTABLE;sethvalue(
- defaultmeta(L),luaH_new(L,0,0));hvalue(defaultmeta(L))->metatable=hvalue(
- defaultmeta(L));sethvalue(gt(L),luaH_new(L,0,4));sethvalue(registry(L),
- luaH_new(L,4,4));luaS_resize(L,MINSTRTABSIZE);luaT_init(L);luaX_init(L);
- luaS_fix(luaS_newliteral(L,MEMERRMSG));g->GCthreshold=4*G(L)->nblocks;}static
- void preinit_state(lua_State*L){L->stack=NULL;L->stacksize=0;L->errorJmp=NULL;
- L->hook=NULL;L->hookmask=L->hookinit=0;L->basehookcount=0;L->allowhook=1;
- resethookcount(L);L->openupval=NULL;L->size_ci=0;L->nCcalls=0;L->base_ci=L->ci
- =NULL;L->errfunc=0;setnilvalue(gt(L));}static void close_state(lua_State*L){
- luaF_close(L,L->stack);if(G(L)){luaC_sweep(L,1);lua_assert(G(L)->rootgc==NULL)
- ;lua_assert(G(L)->rootudata==NULL);luaS_freeall(L);luaZ_freebuffer(L,&G(L)->
- buff);}freestack(L,L);if(G(L)){lua_assert(G(L)->nblocks==sizeof(lua_State)+
- sizeof(global_State));luaM_freelem(NULL,G(L));}freestate(NULL,L);}lua_State*
- luaE_newthread(lua_State*L){lua_State*L1=mallocstate(L);luaC_link(L,valtogco(
- L1),LUA_TTHREAD);preinit_state(L1);L1->l_G=L->l_G;stack_init(L1,L);setobj2n(gt
- (L1),gt(L));return L1;}void luaE_freethread(lua_State*L,lua_State*L1){
- luaF_close(L1,L1->stack);lua_assert(L1->openupval==NULL);freestack(L,L1);
- freestate(L,L1);}LUA_API lua_State*lua_open(void){lua_State*L=mallocstate(NULL
- );if(L){L->tt=LUA_TTHREAD;L->marked=0;L->next=L->gclist=NULL;preinit_state(L);
- L->l_G=NULL;if(luaD_rawrunprotected(L,f_luaopen,NULL)!=0){close_state(L);L=
- NULL;}}lua_userstateopen(L);return L;}static void callallgcTM(lua_State*L,void
- *ud){UNUSED(ud);luaC_callGCTM(L);}LUA_API void lua_close(lua_State*L){lua_lock
- (L);L=G(L)->mainthread;luaF_close(L,L->stack);luaC_separateudata(L);L->errfunc
- =0;do{L->ci=L->base_ci;L->base=L->top=L->ci->base;L->nCcalls=0;}while(
- luaD_rawrunprotected(L,callallgcTM,NULL)!=0);lua_assert(G(L)->tmudata==NULL);
- close_state(L);}
- #line 1 "lstring.c"
- #define lstring_c
- void luaS_freeall(lua_State*L){lua_assert(G(L)->strt.nuse==0);luaM_freearray(L
- ,G(L)->strt.hash,G(L)->strt.size,TString*);}void luaS_resize(lua_State*L,int
- newsize){GCObject**newhash=luaM_newvector(L,newsize,GCObject*);stringtable*tb=
- &G(L)->strt;int i;for(i=0;i<newsize;i++)newhash[i]=NULL;for(i=0;i<tb->size;i++
- ){GCObject*p=tb->hash[i];while(p){GCObject*next=p->gch.next;lu_hash h=gcotots(
- p)->tsv.hash;int h1=lmod(h,newsize);lua_assert(cast(int,h%newsize)==lmod(h,
- newsize));p->gch.next=newhash[h1];newhash[h1]=p;p=next;}}luaM_freearray(L,tb->
- hash,tb->size,TString*);tb->size=newsize;tb->hash=newhash;}static TString*
- newlstr(lua_State*L,const char*str,size_t l,lu_hash h){TString*ts=cast(TString
- *,luaM_malloc(L,sizestring(l)));stringtable*tb;ts->tsv.len=l;ts->tsv.hash=h;ts
- ->tsv.marked=0;ts->tsv.tt=LUA_TSTRING;ts->tsv.reserved=0;memcpy(ts+1,str,l*
- sizeof(char));((char*)(ts+1))[l]='\0';tb=&G(L)->strt;h=lmod(h,tb->size);ts->
- tsv.next=tb->hash[h];tb->hash[h]=valtogco(ts);tb->nuse++;if(tb->nuse>cast(
- ls_nstr,tb->size)&&tb->size<=MAX_INT/2)luaS_resize(L,tb->size*2);return ts;}
- TString*luaS_newlstr(lua_State*L,const char*str,size_t l){GCObject*o;lu_hash h
- =(lu_hash)l;size_t step=(l>>5)+1;size_t l1;for(l1=l;l1>=step;l1-=step)h=h^((h
- <<5)+(h>>2)+(unsigned char)(str[l1-1]));for(o=G(L)->strt.hash[lmod(h,G(L)->
- strt.size)];o!=NULL;o=o->gch.next){TString*ts=gcotots(o);if(ts->tsv.len==l&&(
- memcmp(str,getstr(ts),l)==0))return ts;}return newlstr(L,str,l,h);}Udata*
- luaS_newudata(lua_State*L,size_t s){Udata*u;u=cast(Udata*,luaM_malloc(L,
- sizeudata(s)));u->uv.marked=(1<<1);u->uv.tt=LUA_TUSERDATA;u->uv.len=s;u->uv.
- metatable=hvalue(defaultmeta(L));u->uv.next=G(L)->rootudata;G(L)->rootudata=
- valtogco(u);return u;}
- #line 1 "lstrlib.c"
- #define lstrlib_c
- #ifndef uchar
- #define uchar(c) ((unsigned char)(c))
- #endif
- typedef long sint32;static int str_len(lua_State*L){size_t l;luaL_checklstring
- (L,1,&l);lua_pushnumber(L,(lua_Number)l);return 1;}static sint32 posrelat(
- sint32 pos,size_t len){return(pos>=0)?pos:(sint32)len+pos+1;}static int
- str_sub(lua_State*L){size_t l;const char*s=luaL_checklstring(L,1,&l);sint32
- start=posrelat(luaL_checklong(L,2),l);sint32 end=posrelat(luaL_optlong(L,3,-1)
- ,l);if(start<1)start=1;if(end>(sint32)l)end=(sint32)l;if(start<=end)
- lua_pushlstring(L,s+start-1,end-start+1);else lua_pushliteral(L,"");return 1;}
- static int str_lower(lua_State*L){size_t l;size_t i;luaL_Buffer b;const char*s
- =luaL_checklstring(L,1,&l);luaL_buffinit(L,&b);for(i=0;i<l;i++)luaL_putchar(&b
- ,tolower(uchar(s[i])));luaL_pushresult(&b);return 1;}static int str_upper(
- lua_State*L){size_t l;size_t i;luaL_Buffer b;const char*s=luaL_checklstring(L,
- 1,&l);luaL_buffinit(L,&b);for(i=0;i<l;i++)luaL_putchar(&b,toupper(uchar(s[i]))
- );luaL_pushresult(&b);return 1;}static int str_rep(lua_State*L){size_t l;
- luaL_Buffer b;const char*s=luaL_checklstring(L,1,&l);int n=luaL_checkint(L,2);
- luaL_buffinit(L,&b);while(n-->0)luaL_addlstring(&b,s,l);luaL_pushresult(&b);
- return 1;}static int str_byte(lua_State*L){size_t l;const char*s=
- luaL_checklstring(L,1,&l);sint32 pos=posrelat(luaL_optlong(L,2,1),l);if(pos<=0
- ||(size_t)(pos)>l)return 0;lua_pushnumber(L,uchar(s[pos-1]));return 1;}static
- int str_char(lua_State*L){int n=lua_gettop(L);int i;luaL_Buffer b;
- luaL_buffinit(L,&b);for(i=1;i<=n;i++){int c=luaL_checkint(L,i);luaL_argcheck(L
- ,uchar(c)==c,i,"invalid value");luaL_putchar(&b,uchar(c));}luaL_pushresult(&b)
- ;return 1;}static int writer(lua_State*L,const void*b,size_t size,void*B){(
- void)L;luaL_addlstring((luaL_Buffer*)B,(const char*)b,size);return 1;}static
- int str_dump(lua_State*L){luaL_Buffer b;luaL_checktype(L,1,LUA_TFUNCTION);
- luaL_buffinit(L,&b);if(!lua_dump(L,writer,&b))luaL_error(L,
- "unable to dump given function");luaL_pushresult(&b);return 1;}
- #ifndef MAX_CAPTURES
- #define MAX_CAPTURES 32
- #endif
- #define CAP_UNFINISHED (-1)
- #define CAP_POSITION (-2)
- typedef struct MatchState{const char*src_init;const char*src_end;lua_State*L;
- int level;struct{const char*init;sint32 len;}capture[MAX_CAPTURES];}MatchState
- ;
- #define ESC '%'
- #define SPECIALS "^$*+?.([%-"
- static int check_capture(MatchState*ms,int l){l-='1';if(l<0||l>=ms->level||ms
- ->capture[l].len==CAP_UNFINISHED)return luaL_error(ms->L,
- "invalid capture index");return l;}static int capture_to_close(MatchState*ms){
- int level=ms->level;for(level--;level>=0;level--)if(ms->capture[level].len==
- CAP_UNFINISHED)return level;return luaL_error(ms->L,"invalid pattern capture")
- ;}static const char*luaI_classend(MatchState*ms,const char*p){switch(*p++){
- case ESC:{if(*p=='\0')luaL_error(ms->L,"malformed pattern (ends with `%')");
- return p+1;}case'[':{if(*p=='^')p++;do{if(*p=='\0')luaL_error(ms->L,
- "malformed pattern (missing `]')");if(*(p++)==ESC&&*p!='\0')p++;}while(*p!=']'
- );return p+1;}default:{return p;}}}static int match_class(int c,int cl){int
- res;switch(tolower(cl)){case'a':res=isalpha(c);break;case'c':res=iscntrl(c);
- break;case'd':res=isdigit(c);break;case'l':res=islower(c);break;case'p':res=
- ispunct(c);break;case's':res=isspace(c);break;case'u':res=isupper(c);break;
- case'w':res=isalnum(c);break;case'x':res=isxdigit(c);break;case'z':res=(c==0);
- break;default:return(cl==c);}return(islower(cl)?res:!res);}static int
- matchbracketclass(int c,const char*p,const char*ec){int sig=1;if(*(p+1)=='^'){
- sig=0;p++;}while(++p<ec){if(*p==ESC){p++;if(match_class(c,*p))return sig;}else
- if((*(p+1)=='-')&&(p+2<ec)){p+=2;if(uchar(*(p-2))<=c&&c<=uchar(*p))return sig
- ;}else if(uchar(*p)==c)return sig;}return!sig;}static int luaI_singlematch(int
- c,const char*p,const char*ep){switch(*p){case'.':return 1;case ESC:return
- match_class(c,*(p+1));case'[':return matchbracketclass(c,p,ep-1);default:
- return(uchar(*p)==c);}}static const char*match(MatchState*ms,const char*s,
- const char*p);static const char*matchbalance(MatchState*ms,const char*s,const
- char*p){if(*p==0||*(p+1)==0)luaL_error(ms->L,"unbalanced pattern");if(*s!=*p)
- return NULL;else{int b=*p;int e=*(p+1);int cont=1;while(++s<ms->src_end){if(*s
- ==e){if(--cont==0)return s+1;}else if(*s==b)cont++;}}return NULL;}static const
- char*max_expand(MatchState*ms,const char*s,const char*p,const char*ep){sint32
- i=0;while((s+i)<ms->src_end&&luaI_singlematch(uchar(*(s+i)),p,ep))i++;while(i
- >=0){const char*res=match(ms,(s+i),ep+1);if(res)return res;i--;}return NULL;}
- static const char*min_expand(MatchState*ms,const char*s,const char*p,const
- char*ep){for(;;){const char*res=match(ms,s,ep+1);if(res!=NULL)return res;else
- if(s<ms->src_end&&luaI_singlematch(uchar(*s),p,ep))s++;else return NULL;}}
- static const char*start_capture(MatchState*ms,const char*s,const char*p,int
- what){const char*res;int level=ms->level;if(level>=MAX_CAPTURES)luaL_error(ms
- ->L,"too many captures");ms->capture[level].init=s;ms->capture[level].len=what
- ;ms->level=level+1;if((res=match(ms,s,p))==NULL)ms->level--;return res;}static
- const char*end_capture(MatchState*ms,const char*s,const char*p){int l=
- capture_to_close(ms);const char*res;ms->capture[l].len=s-ms->capture[l].init;
- if((res=match(ms,s,p))==NULL)ms->capture[l].len=CAP_UNFINISHED;return res;}
- static const char*match_capture(MatchState*ms,const char*s,int l){size_t len;l
- =check_capture(ms,l);len=ms->capture[l].len;if((size_t)(ms->src_end-s)>=len&&
- memcmp(ms->capture[l].init,s,len)==0)return s+len;else return NULL;}static
- const char*match(MatchState*ms,const char*s,const char*p){init:switch(*p){case
- '(':{if(*(p+1)==')')return start_capture(ms,s,p+2,CAP_POSITION);else return
- start_capture(ms,s,p+1,CAP_UNFINISHED);}case')':{return end_capture(ms,s,p+1);
- }case ESC:{switch(*(p+1)){case'b':{s=matchbalance(ms,s,p+2);if(s==NULL)return
- NULL;p+=4;goto init;}case'f':{const char*ep;char previous;p+=2;if(*p!='[')
- luaL_error(ms->L,"missing `[' after `%%f' in pattern");ep=luaI_classend(ms,p);
- previous=(s==ms->src_init)?'\0':*(s-1);if(matchbracketclass(uchar(previous),p,
- ep-1)||!matchbracketclass(uchar(*s),p,ep-1))return NULL;p=ep;goto init;}
- default:{if(isdigit(uchar(*(p+1)))){s=match_capture(ms,s,*(p+1));if(s==NULL)
- return NULL;p+=2;goto init;}goto dflt;}}}case'\0':{return s;}case'$':{if(*(p+1
- )=='\0')return(s==ms->src_end)?s:NULL;else goto dflt;}default:dflt:{const char
- *ep=luaI_classend(ms,p);int m=s<ms->src_end&&luaI_singlematch(uchar(*s),p,ep);
- switch(*ep){case'?':{const char*res;if(m&&((res=match(ms,s+1,ep+1))!=NULL))
- return res;p=ep+1;goto init;}case'*':{return max_expand(ms,s,p,ep);}case'+':{
- return(m?max_expand(ms,s+1,p,ep):NULL);}case'-':{return min_expand(ms,s,p,ep);
- }default:{if(!m)return NULL;s++;p=ep;goto init;}}}}}static const char*lmemfind
- (const char*s1,size_t l1,const char*s2,size_t l2){if(l2==0)return s1;else if(
- l2>l1)return NULL;else{const char*init;l2--;l1=l1-l2;while(l1>0&&(init=(const
- char*)memchr(s1,*s2,l1))!=NULL){init++;if(memcmp(init,s2+1,l2)==0)return init-
- 1;else{l1-=init-s1;s1=init;}}return NULL;}}static void push_onecapture(
- MatchState*ms,int i){int l=ms->capture[i].len;if(l==CAP_UNFINISHED)luaL_error(
- ms->L,"unfinished capture");if(l==CAP_POSITION)lua_pushnumber(ms->L,(
- lua_Number)(ms->capture[i].init-ms->src_init+1));else lua_pushlstring(ms->L,ms
- ->capture[i].init,l);}static int push_captures(MatchState*ms,const char*s,
- const char*e){int i;luaL_checkstack(ms->L,ms->level,"too many captures");if(ms
- ->level==0&&s){lua_pushlstring(ms->L,s,e-s);return 1;}else{for(i=0;i<ms->level
- ;i++)push_onecapture(ms,i);return ms->level;}}static int str_find(lua_State*L)
- {size_t l1,l2;const char*s=luaL_checklstring(L,1,&l1);const char*p=
- luaL_checklstring(L,2,&l2);sint32 init=posrelat(luaL_optlong(L,3,1),l1)-1;if(
- init<0)init=0;else if((size_t)(init)>l1)init=(sint32)l1;if(lua_toboolean(L,4)
- ||strpbrk(p,SPECIALS)==NULL){const char*s2=lmemfind(s+init,l1-init,p,l2);if(s2
- ){lua_pushnumber(L,(lua_Number)(s2-s+1));lua_pushnumber(L,(lua_Number)(s2-s+l2
- ));return 2;}}else{MatchState ms;int anchor=(*p=='^')?(p++,1):0;const char*s1=
- s+init;ms.L=L;ms.src_init=s;ms.src_end=s+l1;do{const char*res;ms.level=0;if((
- res=match(&ms,s1,p))!=NULL){lua_pushnumber(L,(lua_Number)(s1-s+1));
- lua_pushnumber(L,(lua_Number)(res-s));return push_captures(&ms,NULL,0)+2;}}
- while(s1++<ms.src_end&&!anchor);}lua_pushnil(L);return 1;}static int gfind_aux
- (lua_State*L){MatchState ms;const char*s=lua_tostring(L,lua_upvalueindex(1));
- size_t ls=lua_strlen(L,lua_upvalueindex(1));const char*p=lua_tostring(L,
- lua_upvalueindex(2));const char*src;ms.L=L;ms.src_init=s;ms.src_end=s+ls;for(
- src=s+(size_t)lua_tonumber(L,lua_upvalueindex(3));src<=ms.src_end;src++){const
- char*e;ms.level=0;if((e=match(&ms,src,p))!=NULL){int newstart=e-s;if(e==src)
- newstart++;lua_pushnumber(L,(lua_Number)newstart);lua_replace(L,
- lua_upvalueindex(3));return push_captures(&ms,src,e);}}return 0;}static int
- gfind(lua_State*L){luaL_checkstring(L,1);luaL_checkstring(L,2);lua_settop(L,2)
- ;lua_pushnumber(L,0);lua_pushcclosure(L,gfind_aux,3);return 1;}static void
- add_s(MatchState*ms,luaL_Buffer*b,const char*s,const char*e){lua_State*L=ms->L
- ;if(lua_isstring(L,3)){const char*news=lua_tostring(L,3);size_t l=lua_strlen(L
- ,3);size_t i;for(i=0;i<l;i++){if(news[i]!=ESC)luaL_putchar(b,news[i]);else{i++
- ;if(!isdigit(uchar(news[i])))luaL_putchar(b,news[i]);else{int level=
- check_capture(ms,news[i]);push_onecapture(ms,level);luaL_addvalue(b);}}}}else{
- int n;lua_pushvalue(L,3);n=push_captures(ms,s,e);lua_call(L,n,1);if(
- lua_isstring(L,-1))luaL_addvalue(b);else lua_pop(L,1);}}static int str_gsub(
- lua_State*L){size_t srcl;const char*src=luaL_checklstring(L,1,&srcl);const
- char*p=luaL_checkstring(L,2);int max_s=luaL_optint(L,4,srcl+1);int anchor=(*p
- =='^')?(p++,1):0;int n=0;MatchState ms;luaL_Buffer b;luaL_argcheck(L,
- lua_gettop(L)>=3&&(lua_isstring(L,3)||lua_isfunction(L,3)),3,
- "string or function expected");luaL_buffinit(L,&b);ms.L=L;ms.src_init=src;ms.
- src_end=src+srcl;while(n<max_s){const char*e;ms.level=0;e=match(&ms,src,p);if(
- e){n++;add_s(&ms,&b,src,e);}if(e&&e>src)src=e;else if(src<ms.src_end)
- luaL_putchar(&b,*src++);else break;if(anchor)break;}luaL_addlstring(&b,src,ms.
- src_end-src);luaL_pushresult(&b);lua_pushnumber(L,(lua_Number)n);return 2;}
- #define MAX_ITEM 512
- #define MAX_FORMAT 20
- static void luaI_addquoted(lua_State*L,luaL_Buffer*b,int arg){size_t l;const
- char*s=luaL_checklstring(L,arg,&l);luaL_putchar(b,'"');while(l--){switch(*s){
- case'"':case'\\':case'\n':{luaL_putchar(b,'\\');luaL_putchar(b,*s);break;}case
- '\0':{luaL_addlstring(b,"\\000",4);break;}default:{luaL_putchar(b,*s);break;}}
- s++;}luaL_putchar(b,'"');}static const char*scanformat(lua_State*L,const char*
- strfrmt,char*form,int*hasprecision){const char*p=strfrmt;while(strchr("-+ #0",
- *p))p++;if(isdigit(uchar(*p)))p++;if(isdigit(uchar(*p)))p++;if(*p=='.'){p++;*
- hasprecision=1;if(isdigit(uchar(*p)))p++;if(isdigit(uchar(*p)))p++;}if(isdigit
- (uchar(*p)))luaL_error(L,"invalid format (width or precision too long)");if(p-
- strfrmt+2>MAX_FORMAT)luaL_error(L,"invalid format (too long)");form[0]='%';
- strncpy(form+1,strfrmt,p-strfrmt+1);form[p-strfrmt+2]=0;return p;}static int
- str_format(lua_State*L){int arg=1;size_t sfl;const char*strfrmt=
- luaL_checklstring(L,arg,&sfl);const char*strfrmt_end=strfrmt+sfl;luaL_Buffer b
- ;luaL_buffinit(L,&b);while(strfrmt<strfrmt_end){if(*strfrmt!='%')luaL_putchar(
- &b,*strfrmt++);else if(*++strfrmt=='%')luaL_putchar(&b,*strfrmt++);else{char
- form[MAX_FORMAT];char buff[MAX_ITEM];int hasprecision=0;if(isdigit(uchar(*
- strfrmt))&&*(strfrmt+1)=='$')return luaL_error(L,
- "obsolete option (d$) to `format'");arg++;strfrmt=scanformat(L,strfrmt,form,&
- hasprecision);switch(*strfrmt++){case'c':case'd':case'i':{sprintf(buff,form,
- luaL_checkint(L,arg));break;}case'o':case'u':case'x':case'X':{sprintf(buff,
- form,(unsigned int)(luaL_checknumber(L,arg)));break;}case'e':case'E':case'f':
- case'g':case'G':{sprintf(buff,form,luaL_checknumber(L,arg));break;}case'q':{
- luaI_addquoted(L,&b,arg);continue;}case's':{size_t l;const char*s=
- luaL_checklstring(L,arg,&l);if(!hasprecision&&l>=100){lua_pushvalue(L,arg);
- luaL_addvalue(&b);continue;}else{sprintf(buff,form,s);break;}}default:{return
- luaL_error(L,"invalid option to `format'");}}luaL_addlstring(&b,buff,strlen(
- buff));}}luaL_pushresult(&b);return 1;}static const luaL_reg strlib[]={{"len",
- str_len},{"sub",str_sub},{"lower",str_lower},{"upper",str_upper},{"char",
- str_char},{"rep",str_rep},{"byte",str_byte},{"format",str_format},{"dump",
- str_dump},{"find",str_find},{"gfind",gfind},{"gsub",str_gsub},{NULL,NULL}};
- LUALIB_API int luaopen_string(lua_State*L){luaL_openlib(L,LUA_STRLIBNAME,
- strlib,0);return 1;}
- #line 1 "ltable.c"
- #define ltable_c
- #if BITS_INT >26
- #define MAXBITS 24
- #else
- #define MAXBITS (BITS_INT-2)
- #endif
- #define toobig(x) ((((x)-1)>>MAXBITS)!=0)
- #ifndef lua_number2int
- #define lua_number2int(i,n) ((i)=(int)(n))
- #endif
- #define hashpow2(t,n) (gnode(t,lmod((n),sizenode(t))))
- #define hashstr(t,str) hashpow2(t,(str)->tsv.hash)
- #define hashboolean(t,p) hashpow2(t,p)
- #define hashmod(t,n) (gnode(t,((n)%((sizenode(t)-1)|1))))
- #define hashpointer(t,p) hashmod(t,IntPoint(p))
- #define numints cast(int,sizeof(lua_Number)/sizeof(int))
- static Node*hashnum(const Table*t,lua_Number n){unsigned int a[numints];int i;
- n+=1;lua_assert(sizeof(a)<=sizeof(n));memcpy(a,&n,sizeof(a));for(i=1;i<numints
- ;i++)a[0]+=a[i];return hashmod(t,cast(lu_hash,a[0]));}Node*luaH_mainposition(
- const Table*t,const TObject*key){switch(ttype(key)){case LUA_TNUMBER:return
- hashnum(t,nvalue(key));case LUA_TSTRING:return hashstr(t,tsvalue(key));case
- LUA_TBOOLEAN:return hashboolean(t,bvalue(key));case LUA_TLIGHTUSERDATA:return
- hashpointer(t,pvalue(key));default:return hashpointer(t,gcvalue(key));}}static
- int arrayindex(const TObject*key){if(ttisnumber(key)){int k;lua_number2int(k,
- (nvalue(key)));if(cast(lua_Number,k)==nvalue(key)&&k>=1&&!toobig(k))return k;}
- return-1;}static int luaH_index(lua_State*L,Table*t,StkId key){int i;if(
- ttisnil(key))return-1;i=arrayindex(key);if(0<=i&&i<=t->sizearray){return i-1;}
- else{const TObject*v=luaH_get(t,key);if(v==&luaO_nilobject)luaG_runerror(L,
- "invalid key for `next'");i=cast(int,(cast(const lu_byte*,v)-cast(const
- lu_byte*,gval(gnode(t,0))))/sizeof(Node));return i+t->sizearray;}}int
- luaH_next(lua_State*L,Table*t,StkId key){int i=luaH_index(L,t,key);for(i++;i<t
- ->sizearray;i++){if(!ttisnil(&t->array[i])){setnvalue(key,cast(lua_Number,i+1)
- );setobj2s(key+1,&t->array[i]);return 1;}}for(i-=t->sizearray;i<sizenode(t);i
- ++){if(!ttisnil(gval(gnode(t,i)))){setobj2s(key,gkey(gnode(t,i)));setobj2s(key
- +1,gval(gnode(t,i)));return 1;}}return 0;}static void computesizes(int nums[],
- int ntotal,int*narray,int*nhash){int i;int a=nums[0];int na=a;int n=(na==0)?-1
- :0;for(i=1;a<*narray&&*narray>=twoto(i-1);i++){if(nums[i]>0){a+=nums[i];if(a>=
- twoto(i-1)){n=i;na=a;}}}lua_assert(na<=*narray&&*narray<=ntotal);*nhash=ntotal
- -na;*narray=(n==-1)?0:twoto(n);lua_assert(na<=*narray&&na>=*narray/2);}static
- void numuse(const Table*t,int*narray,int*nhash){int nums[MAXBITS+1];int i,lg;
- int totaluse=0;for(i=0,lg=0;lg<=MAXBITS;lg++){int ttlg=twoto(lg);if(ttlg>t->
- sizearray){ttlg=t->sizearray;if(i>=ttlg)break;}nums[lg]=0;for(;i<ttlg;i++){if(
- !ttisnil(&t->array[i])){nums[lg]++;totaluse++;}}}for(;lg<=MAXBITS;lg++)nums[lg
- ]=0;*narray=totaluse;i=sizenode(t);while(i--){Node*n=&t->node[i];if(!ttisnil(
- gval(n))){int k=arrayindex(gkey(n));if(k>=0){nums[luaO_log2(k-1)+1]++;(*narray
- )++;}totaluse++;}}computesizes(nums,totaluse,narray,nhash);}static void
- setarrayvector(lua_State*L,Table*t,int size){int i;luaM_reallocvector(L,t->
- array,t->sizearray,size,TObject);for(i=t->sizearray;i<size;i++)setnilvalue(&t
- ->array[i]);t->sizearray=size;}static void setnodevector(lua_State*L,Table*t,
- int lsize){int i;int size=twoto(lsize);if(lsize>MAXBITS)luaG_runerror(L,
- "table overflow");if(lsize==0){t->node=G(L)->dummynode;lua_assert(ttisnil(gkey
- (t->node)));lua_assert(ttisnil(gval(t->node)));lua_assert(t->node->next==NULL)
- ;}else{t->node=luaM_newvector(L,size,Node);for(i=0;i<size;i++){t->node[i].next
- =NULL;setnilvalue(gkey(gnode(t,i)));setnilvalue(gval(gnode(t,i)));}}t->
- lsizenode=cast(lu_byte,lsize);t->firstfree=gnode(t,size-1);}static void resize
- (lua_State*L,Table*t,int nasize,int nhsize){int i;int oldasize=t->sizearray;
- int oldhsize=t->lsizenode;Node*nold;Node temp[1];if(oldhsize)nold=t->node;else
- {lua_assert(t->node==G(L)->dummynode);temp[0]=t->node[0];nold=temp;setnilvalue
- (gkey(G(L)->dummynode));setnilvalue(gval(G(L)->dummynode));lua_assert(G(L)->
- dummynode->next==NULL);}if(nasize>oldasize)setarrayvector(L,t,nasize);
- setnodevector(L,t,nhsize);if(nasize<oldasize){t->sizearray=nasize;for(i=nasize
- ;i<oldasize;i++){if(!ttisnil(&t->array[i]))setobjt2t(luaH_setnum(L,t,i+1),&t->
- array[i]);}luaM_reallocvector(L,t->array,oldasize,nasize,TObject);}for(i=twoto
- (oldhsize)-1;i>=0;i--){Node*old=nold+i;if(!ttisnil(gval(old)))setobjt2t(
- luaH_set(L,t,gkey(old)),gval(old));}if(oldhsize)luaM_freearray(L,nold,twoto(
- oldhsize),Node);}static void rehash(lua_State*L,Table*t){int nasize,nhsize;
- numuse(t,&nasize,&nhsize);resize(L,t,nasize,luaO_log2(nhsize)+1);}Table*
- luaH_new(lua_State*L,int narray,int lnhash){Table*t=luaM_new(L,Table);
- luaC_link(L,valtogco(t),LUA_TTABLE);t->metatable=hvalue(defaultmeta(L));t->
- flags=cast(lu_byte,~0);t->array=NULL;t->sizearray=0;t->lsizenode=0;t->node=
- NULL;setarrayvector(L,t,narray);setnodevector(L,t,lnhash);return t;}void
- luaH_free(lua_State*L,Table*t){if(t->lsizenode)luaM_freearray(L,t->node,
- sizenode(t),Node);luaM_freearray(L,t->array,t->sizearray,TObject);luaM_freelem
- (L,t);}
- #if 0
- void luaH_remove(Table*t,Node*e){Node*mp=luaH_mainposition(t,gkey(e));if(e!=mp
- ){while(mp->next!=e)mp=mp->next;mp->next=e->next;}else{if(e->next!=NULL)??}
- lua_assert(ttisnil(gval(node)));setnilvalue(gkey(e));e->next=NULL;}
- #endif
- static TObject*newkey(lua_State*L,Table*t,const TObject*key){TObject*val;Node*
- mp=luaH_mainposition(t,key);if(!ttisnil(gval(mp))){Node*othern=
- luaH_mainposition(t,gkey(mp));Node*n=t->firstfree;if(othern!=mp){while(othern
- ->next!=mp)othern=othern->next;othern->next=n;*n=*mp;mp->next=NULL;setnilvalue
- (gval(mp));}else{n->next=mp->next;mp->next=n;mp=n;}}setobj2t(gkey(mp),key);
- lua_assert(ttisnil(gval(mp)));for(;;){if(ttisnil(gkey(t->firstfree)))return
- gval(mp);else if(t->firstfree==t->node)break;else(t->firstfree)--;}setbvalue(
- gval(mp),0);rehash(L,t);val=cast(TObject*,luaH_get(t,key));lua_assert(
- ttisboolean(val));setnilvalue(val);return val;}static const TObject*
- luaH_getany(Table*t,const TObject*key){if(ttisnil(key))return&luaO_nilobject;
- else{Node*n=luaH_mainposition(t,key);do{if(luaO_rawequalObj(gkey(n),key))
- return gval(n);else n=n->next;}while(n);return&luaO_nilobject;}}const TObject*
- luaH_getnum(Table*t,int key){if(1<=key&&key<=t->sizearray)return&t->array[key-
- 1];else{lua_Number nk=cast(lua_Number,key);Node*n=hashnum(t,nk);do{if(
- ttisnumber(gkey(n))&&nvalue(gkey(n))==nk)return gval(n);else n=n->next;}while(
- n);return&luaO_nilobject;}}const TObject*luaH_getstr(Table*t,TString*key){Node
- *n=hashstr(t,key);do{if(ttisstring(gkey(n))&&tsvalue(gkey(n))==key)return gval
- (n);else n=n->next;}while(n);return&luaO_nilobject;}const TObject*luaH_get(
- Table*t,const TObject*key){switch(ttype(key)){case LUA_TSTRING:return
- luaH_getstr(t,tsvalue(key));case LUA_TNUMBER:{int k;lua_number2int(k,(nvalue(
- key)));if(cast(lua_Number,k)==nvalue(key))return luaH_getnum(t,k);}default:
- return luaH_getany(t,key);}}TObject*luaH_set(lua_State*L,Table*t,const TObject
- *key){const TObject*p=luaH_get(t,key);t->flags=0;if(p!=&luaO_nilobject)return
- cast(TObject*,p);else{if(ttisnil(key))luaG_runerror(L,"table index is nil");
- else if(ttisnumber(key)&&nvalue(key)!=nvalue(key))luaG_runerror(L,
- "table index is NaN");return newkey(L,t,key);}}TObject*luaH_setnum(lua_State*L
- ,Table*t,int key){const TObject*p=luaH_getnum(t,key);if(p!=&luaO_nilobject)
- return cast(TObject*,p);else{TObject k;setnvalue(&k,cast(lua_Number,key));
- return newkey(L,t,&k);}}
- #line 1 "ltablib.c"
- #define ltablib_c
- #define aux_getn(L,n) (luaL_checktype(L,n,LUA_TTABLE),luaL_getn(L,n))
- static int luaB_foreachi(lua_State*L){int i;int n=aux_getn(L,1);luaL_checktype
- (L,2,LUA_TFUNCTION);for(i=1;i<=n;i++){lua_pushvalue(L,2);lua_pushnumber(L,(
- lua_Number)i);lua_rawgeti(L,1,i);lua_call(L,2,1);if(!lua_isnil(L,-1))return 1;
- lua_pop(L,1);}return 0;}static int luaB_foreach(lua_State*L){luaL_checktype(L,
- 1,LUA_TTABLE);luaL_checktype(L,2,LUA_TFUNCTION);lua_pushnil(L);for(;;){if(
- lua_next(L,1)==0)return 0;lua_pushvalue(L,2);lua_pushvalue(L,-3);lua_pushvalue
- (L,-3);lua_call(L,2,1);if(!lua_isnil(L,-1))return 1;lua_pop(L,2);}}static int
- luaB_getn(lua_State*L){lua_pushnumber(L,(lua_Number)aux_getn(L,1));return 1;}
- static int luaB_setn(lua_State*L){luaL_checktype(L,1,LUA_TTABLE);luaL_setn(L,1
- ,luaL_checkint(L,2));return 0;}static int luaB_tinsert(lua_State*L){int v=
- lua_gettop(L);int n=aux_getn(L,1)+1;int pos;if(v==2)pos=n;else{pos=
- luaL_checkint(L,2);if(pos>n)n=pos;v=3;}luaL_setn(L,1,n);while(--n>=pos){
- lua_rawgeti(L,1,n);lua_rawseti(L,1,n+1);}lua_pushvalue(L,v);lua_rawseti(L,1,
- pos);return 0;}static int luaB_tremove(lua_State*L){int n=aux_getn(L,1);int
- pos=luaL_optint(L,2,n);if(n<=0)return 0;luaL_setn(L,1,n-1);lua_rawgeti(L,1,pos
- );for(;pos<n;pos++){lua_rawgeti(L,1,pos+1);lua_rawseti(L,1,pos);}lua_pushnil(L
- );lua_rawseti(L,1,n);return 1;}static int str_concat(lua_State*L){luaL_Buffer
- b;size_t lsep;const char*sep=luaL_optlstring(L,2,"",&lsep);int i=luaL_optint(L
- ,3,1);int n=luaL_optint(L,4,0);luaL_checktype(L,1,LUA_TTABLE);if(n==0)n=
- luaL_getn(L,1);luaL_buffinit(L,&b);for(;i<=n;i++){lua_rawgeti(L,1,i);
- luaL_argcheck(L,lua_isstring(L,-1),1,"table contains non-strings");
- luaL_addvalue(&b);if(i!=n)luaL_addlstring(&b,sep,lsep);}luaL_pushresult(&b);
- return 1;}static void set2(lua_State*L,int i,int j){lua_rawseti(L,1,i);
- lua_rawseti(L,1,j);}static int sort_comp(lua_State*L,int a,int b){if(!
- lua_isnil(L,2)){int res;lua_pushvalue(L,2);lua_pushvalue(L,a-1);lua_pushvalue(
- L,b-2);lua_call(L,2,1);res=lua_toboolean(L,-1);lua_pop(L,1);return res;}else
- return lua_lessthan(L,a,b);}static void auxsort(lua_State*L,int l,int u){while
- (l<u){int i,j;lua_rawgeti(L,1,l);lua_rawgeti(L,1,u);if(sort_comp(L,-1,-2))set2
- (L,l,u);else lua_pop(L,2);if(u-l==1)break;i=(l+u)/2;lua_rawgeti(L,1,i);
- lua_rawgeti(L,1,l);if(sort_comp(L,-2,-1))set2(L,i,l);else{lua_pop(L,1);
- lua_rawgeti(L,1,u);if(sort_comp(L,-1,-2))set2(L,i,u);else lua_pop(L,2);}if(u-l
- ==2)break;lua_rawgeti(L,1,i);lua_pushvalue(L,-1);lua_rawgeti(L,1,u-1);set2(L,i
- ,u-1);i=l;j=u-1;for(;;){while(lua_rawgeti(L,1,++i),sort_comp(L,-1,-2)){if(i>u)
- luaL_error(L,"invalid order function for sorting");lua_pop(L,1);}while(
- lua_rawgeti(L,1,--j),sort_comp(L,-3,-1)){if(j<l)luaL_error(L,
- "invalid order function for sorting");lua_pop(L,1);}if(j<i){lua_pop(L,3);break
- ;}set2(L,i,j);}lua_rawgeti(L,1,u-1);lua_rawgeti(L,1,i);set2(L,u-1,i);if(i-l<u-
- i){j=l;i=i-1;l=i+2;}else{j=i+1;i=u;u=j-2;}auxsort(L,j,i);}}static int
- luaB_sort(lua_State*L){int n=aux_getn(L,1);luaL_checkstack(L,40,"");if(!
- lua_isnoneornil(L,2))luaL_checktype(L,2,LUA_TFUNCTION);lua_settop(L,2);auxsort
- (L,1,n);return 0;}static const luaL_reg tab_funcs[]={{"concat",str_concat},{
- "foreach",luaB_foreach},{"foreachi",luaB_foreachi},{"getn",luaB_getn},{"setn",
- luaB_setn},{"sort",luaB_sort},{"insert",luaB_tinsert},{"remove",luaB_tremove},
- {NULL,NULL}};LUALIB_API int luaopen_table(lua_State*L){luaL_openlib(L,
- LUA_TABLIBNAME,tab_funcs,0);return 1;}
- #line 1 "ltests.c"
- #define ltests_c
- #ifdef LUA_DEBUG
- #define lua_pushintegral(L,i) lua_pushnumber(L,cast(lua_Number,(i)))
- static lua_State*lua_state=NULL;int islocked=0;
- #define func_at(L,k) (L->ci->base+(k)-1)
- static void setnameval(lua_State*L,const char*name,int val){lua_pushstring(L,
- name);lua_pushintegral(L,val);lua_settable(L,-3);}
- #define MARK 0x55
- #ifndef EXTERNMEMCHECK
- #define HEADER (sizeof(L_Umaxalign))
- #define MARKSIZE 16
- #define blockhead(b) (cast(char*,b)-HEADER)
- #define setsize(newblock, size)(*cast(size_t*,newblock)=size)
- #define checkblocksize(b, size)(size==(*cast(size_t*,blockhead(b))))
- #define fillmem(mem,size) memset(mem,-MARK,size)
- #else
- #define HEADER 0
- #define MARKSIZE 0
- #define blockhead(b) (b)
- #define setsize(newblock, size)
- #define checkblocksize(b,size) (1)
- #define fillmem(mem,size)
- #endif
- unsigned long memdebug_numblocks=0;unsigned long memdebug_total=0;unsigned
- long memdebug_maxmem=0;unsigned long memdebug_memlimit=ULONG_MAX;static void*
- checkblock(void*block,size_t size){void*b=blockhead(block);int i;for(i=0;i<
- MARKSIZE;i++)lua_assert(*(cast(char*,b)+HEADER+size+i)==MARK+i);return b;}
- static void freeblock(void*block,size_t size){if(block){lua_assert(
- checkblocksize(block,size));block=checkblock(block,size);fillmem(block,size+
- HEADER+MARKSIZE);free(block);memdebug_numblocks--;memdebug_total-=size;}}void*
- debug_realloc(void*block,size_t oldsize,size_t size){lua_assert(oldsize==0||
- checkblocksize(block,oldsize));lua_assert(block!=NULL||size>0);if(size==0){
- freeblock(block,oldsize);return NULL;}else if(size>oldsize&&memdebug_total+
- size-oldsize>memdebug_memlimit)return NULL;else{void*newblock;int i;size_t
- realsize=HEADER+size+MARKSIZE;size_t commonsize=(oldsize<size)?oldsize:size;if
- (realsize<size)return NULL;newblock=malloc(realsize);if(newblock==NULL)return
- NULL;if(block){memcpy(cast(char*,newblock)+HEADER,block,commonsize);freeblock(
- block,oldsize);}fillmem(cast(char*,newblock)+HEADER+commonsize,size-commonsize
- );memdebug_total+=size;if(memdebug_total>memdebug_maxmem)memdebug_maxmem=
- memdebug_total;memdebug_numblocks++;setsize(newblock,size);for(i=0;i<MARKSIZE;
- i++)*(cast(char*,newblock)+HEADER+size+i)=cast(char,MARK+i);return cast(char*,
- newblock)+HEADER;}}static char*buildop(Proto*p,int pc,char*buff){Instruction i
- =p->code[pc];OpCode o=GET_OPCODE(i);const char*name=luaP_opnames[o];int line=
- getline(p,pc);sprintf(buff,"(%4d) %4d - ",line,pc);switch(getOpMode(o)){case
- iABC:sprintf(buff+strlen(buff),"%-12s%4d %4d %4d",name,GETARG_A(i),GETARG_B(i)
- ,GETARG_C(i));break;case iABx:sprintf(buff+strlen(buff),"%-12s%4d %4d",name,
- GETARG_A(i),GETARG_Bx(i));break;case iAsBx:sprintf(buff+strlen(buff),
- "%-12s%4d %4d",name,GETARG_A(i),GETARG_sBx(i));break;}return buff;}
- #if 0
- void luaI_printcode(Proto*pt,int size){int pc;for(pc=0;pc<size;pc++){char buff
- [100];printf("%s\n",buildop(pt,pc,buff));}printf("-------\n");}
- #endif
- static int listcode(lua_State*L){int pc;Proto*p;luaL_argcheck(L,lua_isfunction
- (L,1)&&!lua_iscfunction(L,1),1,"Lua function expected");p=clvalue(func_at(L,1)
- )->l.p;lua_newtable(L);setnameval(L,"maxstack",p->maxstacksize);setnameval(L,
- "numparams",p->numparams);for(pc=0;pc<p->sizecode;pc++){char buff[100];
- lua_pushintegral(L,pc+1);lua_pushstring(L,buildop(p,pc,buff));lua_settable(L,-
- 3);}return 1;}static int listk(lua_State*L){Proto*p;int i;luaL_argcheck(L,
- lua_isfunction(L,1)&&!lua_iscfunction(L,1),1,"Lua function expected");p=
- clvalue(func_at(L,1))->l.p;lua_newtable(L);for(i=0;i<p->sizek;i++){
- lua_pushintegral(L,i+1);luaA_pushobject(L,p->k+i);lua_settable(L,-3);}return 1
- ;}static int listlocals(lua_State*L){Proto*p;int pc=luaL_checkint(L,2)-1;int i
- =0;const char*name;luaL_argcheck(L,lua_isfunction(L,1)&&!lua_iscfunction(L,1),
- 1,"Lua function expected");p=clvalue(func_at(L,1))->l.p;while((name=
- luaF_getlocalname(p,++i,pc))!=NULL)lua_pushstring(L,name);return i-1;}static
- int get_limits(lua_State*L){lua_newtable(L);setnameval(L,"BITS_INT",BITS_INT);
- setnameval(L,"LFPF",LFIELDS_PER_FLUSH);setnameval(L,"MAXVARS",MAXVARS);
- setnameval(L,"MAXPARAMS",MAXPARAMS);setnameval(L,"MAXSTACK",MAXSTACK);
- setnameval(L,"MAXUPVALUES",MAXUPVALUES);return 1;}static int mem_query(
- lua_State*L){if(lua_isnone(L,1)){lua_pushintegral(L,memdebug_total);
- lua_pushintegral(L,memdebug_numblocks);lua_pushintegral(L,memdebug_maxmem);
- return 3;}else{memdebug_memlimit=luaL_checkint(L,1);return 0;}}static int
- hash_query(lua_State*L){if(lua_isnone(L,2)){luaL_argcheck(L,lua_type(L,1)==
- LUA_TSTRING,1,"string expected");lua_pushintegral(L,tsvalue(func_at(L,1))->tsv
- .hash);}else{TObject*o=func_at(L,1);Table*t;luaL_checktype(L,2,LUA_TTABLE);t=
- hvalue(func_at(L,2));lua_pushintegral(L,luaH_mainposition(t,o)-t->node);}
- return 1;}static int stacklevel(lua_State*L){unsigned long a=0;
- lua_pushintegral(L,(int)(L->top-L->stack));lua_pushintegral(L,(int)(L->
- stack_last-L->stack));lua_pushintegral(L,(int)(L->ci-L->base_ci));
- lua_pushintegral(L,(int)(L->end_ci-L->base_ci));lua_pushintegral(L,(unsigned
- long)&a);return 5;}static int table_query(lua_State*L){const Table*t;int i=
- luaL_optint(L,2,-1);luaL_checktype(L,1,LUA_TTABLE);t=hvalue(func_at(L,1));if(i
- ==-1){lua_pushintegral(L,t->sizearray);lua_pushintegral(L,sizenode(t));
- lua_pushintegral(L,t->firstfree-t->node);}else if(i<t->sizearray){
- lua_pushintegral(L,i);luaA_pushobject(L,&t->array[i]);lua_pushnil(L);}else if(
- (i-=t->sizearray)<sizenode(t)){if(!ttisnil(gval(gnode(t,i)))||ttisnil(gkey(
- gnode(t,i)))||ttisnumber(gkey(gnode(t,i)))){luaA_pushobject(L,gkey(gnode(t,i))
- );}else lua_pushstring(L,"<undef>");luaA_pushobject(L,gval(gnode(t,i)));if(t->
- node[i].next)lua_pushintegral(L,t->node[i].next-t->node);else lua_pushnil(L);}
- return 3;}static int string_query(lua_State*L){stringtable*tb=&G(L)->strt;int
- s=luaL_optint(L,2,0)-1;if(s==-1){lua_pushintegral(L,tb->nuse);lua_pushintegral
- (L,tb->size);return 2;}else if(s<tb->size){GCObject*ts;int n=0;for(ts=tb->hash
- [s];ts;ts=ts->gch.next){setsvalue2s(L->top,gcotots(ts));incr_top(L);n++;}
- return n;}return 0;}static int tref(lua_State*L){int level=lua_gettop(L);int
- lock=luaL_optint(L,2,1);luaL_checkany(L,1);lua_pushvalue(L,1);lua_pushintegral
- (L,lua_ref(L,lock));assert(lua_gettop(L)==level+1);return 1;}static int getref
- (lua_State*L){int level=lua_gettop(L);lua_getref(L,luaL_checkint(L,1));assert(
- lua_gettop(L)==level+1);return 1;}static int unref(lua_State*L){int level=
- lua_gettop(L);lua_unref(L,luaL_checkint(L,1));assert(lua_gettop(L)==level);
- return 0;}static int metatable(lua_State*L){luaL_checkany(L,1);if(lua_isnone(L
- ,2)){if(lua_getmetatable(L,1)==0)lua_pushnil(L);}else{lua_settop(L,2);
- luaL_checktype(L,2,LUA_TTABLE);lua_setmetatable(L,1);}return 1;}static int
- upvalue(lua_State*L){int n=luaL_checkint(L,2);luaL_checktype(L,1,LUA_TFUNCTION
- );if(lua_isnone(L,3)){const char*name=lua_getupvalue(L,1,n);if(name==NULL)
- return 0;lua_pushstring(L,name);return 2;}else{const char*name=lua_setupvalue(
- L,1,n);lua_pushstring(L,name);return 1;}}static int newuserdata(lua_State*L){
- size_t size=luaL_checkint(L,1);char*p=cast(char*,lua_newuserdata(L,size));
- while(size--)*p++='\0';return 1;}static int pushuserdata(lua_State*L){
- lua_pushlightuserdata(L,cast(void*,luaL_checkint(L,1)));return 1;}static int
- udataval(lua_State*L){lua_pushintegral(L,cast(int,lua_touserdata(L,1)));return
- 1;}static int doonnewstack(lua_State*L){lua_State*L1=lua_newthread(L);size_t
- l;const char*s=luaL_checklstring(L,1,&l);int status=luaL_loadbuffer(L1,s,l,s);
- if(status==0)status=lua_pcall(L1,0,0,0);lua_pushintegral(L,status);return 1;}
- static int s2d(lua_State*L){lua_pushnumber(L,*cast(const double*,
- luaL_checkstring(L,1)));return 1;}static int d2s(lua_State*L){double d=
- luaL_checknumber(L,1);lua_pushlstring(L,cast(char*,&d),sizeof(d));return 1;}
- static int newstate(lua_State*L){lua_State*L1=lua_open();if(L1){
- lua_userstateopen(L1);lua_pushintegral(L,(unsigned long)L1);}else lua_pushnil(
- L);return 1;}static int loadlib(lua_State*L){static const luaL_reg libs[]={{
- "mathlibopen",luaopen_math},{"strlibopen",luaopen_string},{"iolibopen",
- luaopen_io},{"tablibopen",luaopen_table},{"dblibopen",luaopen_debug},{
- "baselibopen",luaopen_base},{NULL,NULL}};lua_State*L1=cast(lua_State*,cast(
- unsigned long,luaL_checknumber(L,1)));lua_pushvalue(L1,LUA_GLOBALSINDEX);
- luaL_openlib(L1,NULL,libs,0);return 0;}static int closestate(lua_State*L){
- lua_State*L1=cast(lua_State*,cast(unsigned long,luaL_checknumber(L,1)));
- lua_close(L1);lua_unlock(L);return 0;}static int doremote(lua_State*L){
- lua_State*L1=cast(lua_State*,cast(unsigned long,luaL_checknumber(L,1)));size_t
- lcode;const char*code=luaL_checklstring(L,2,&lcode);int status;lua_settop(L1,
- 0);status=luaL_loadbuffer(L1,code,lcode,code);if(status==0)status=lua_pcall(L1
- ,0,LUA_MULTRET,0);if(status!=0){lua_pushnil(L);lua_pushintegral(L,status);
- lua_pushstring(L,lua_tostring(L1,-1));return 3;}else{int i=0;while(!lua_isnone
- (L1,++i))lua_pushstring(L,lua_tostring(L1,i));lua_pop(L1,i-1);return i-1;}}
- static int log2_aux(lua_State*L){lua_pushintegral(L,luaO_log2(luaL_checkint(L,
- 1)));return 1;}static int int2fb_aux(lua_State*L){int b=luaO_int2fb(
- luaL_checkint(L,1));lua_pushintegral(L,b);lua_pushintegral(L,fb2int(b));return
- 2;}static int test_do(lua_State*L){const char*p=luaL_checkstring(L,1);if(*p==
- '@')lua_dofile(L,p+1);else lua_dostring(L,p);return lua_gettop(L);}static
- const char*const delimits=" \t\n,;";static void skip(const char**pc){while(**
- pc!='\0'&&strchr(delimits,**pc))(*pc)++;}static int getnum_aux(lua_State*L,
- const char**pc){int res=0;int sig=1;skip(pc);if(**pc=='.'){res=cast(int,
- lua_tonumber(L,-1));lua_pop(L,1);(*pc)++;return res;}else if(**pc=='-'){sig=-1
- ;(*pc)++;}while(isdigit(cast(int,**pc)))res=res*10+(*(*pc)++)-'0';return sig*
- res;}static const char*getname_aux(char*buff,const char**pc){int i=0;skip(pc);
- while(**pc!='\0'&&!strchr(delimits,**pc))buff[i++]=*(*pc)++;buff[i]='\0';
- return buff;}
- #define EQ(s1) (strcmp(s1,inst)==0)
- #define getnum (getnum_aux(L,&pc))
- #define getname (getname_aux(buff,&pc))
- static int testC(lua_State*L){char buff[30];const char*pc=luaL_checkstring(L,1
- );for(;;){const char*inst=getname;if EQ("")return 0;else if EQ("isnumber"){
- lua_pushintegral(L,lua_isnumber(L,getnum));}else if EQ("isstring"){
- lua_pushintegral(L,lua_isstring(L,getnum));}else if EQ("istable"){
- lua_pushintegral(L,lua_istable(L,getnum));}else if EQ("iscfunction"){
- lua_pushintegral(L,lua_iscfunction(L,getnum));}else if EQ("isfunction"){
- lua_pushintegral(L,lua_isfunction(L,getnum));}else if EQ("isuserdata"){
- lua_pushintegral(L,lua_isuserdata(L,getnum));}else if EQ("isudataval"){
- lua_pushintegral(L,lua_islightuserdata(L,getnum));}else if EQ("isnil"){
- lua_pushintegral(L,lua_isnil(L,getnum));}else if EQ("isnull"){lua_pushintegral
- (L,lua_isnone(L,getnum));}else if EQ("tonumber"){lua_pushnumber(L,lua_tonumber
- (L,getnum));}else if EQ("tostring"){const char*s=lua_tostring(L,getnum);
- lua_pushstring(L,s);}else if EQ("strlen"){lua_pushintegral(L,lua_strlen(L,
- getnum));}else if EQ("tocfunction"){lua_pushcfunction(L,lua_tocfunction(L,
- getnum));}else if EQ("return"){return getnum;}else if EQ("gettop"){
- lua_pushintegral(L,lua_gettop(L));}else if EQ("settop"){lua_settop(L,getnum);}
- else if EQ("pop"){lua_pop(L,getnum);}else if EQ("pushnum"){lua_pushintegral(L,
- getnum);}else if EQ("pushnil"){lua_pushnil(L);}else if EQ("pushbool"){
- lua_pushboolean(L,getnum);}else if EQ("tobool"){lua_pushintegral(L,
- lua_toboolean(L,getnum));}else if EQ("pushvalue"){lua_pushvalue(L,getnum);}
- else if EQ("pushcclosure"){lua_pushcclosure(L,testC,getnum);}else if EQ(
- "pushupvalues"){lua_pushupvalues(L);}else if EQ("remove"){lua_remove(L,getnum)
- ;}else if EQ("insert"){lua_insert(L,getnum);}else if EQ("replace"){lua_replace
- (L,getnum);}else if EQ("gettable"){lua_gettable(L,getnum);}else if EQ(
- "settable"){lua_settable(L,getnum);}else if EQ("next"){lua_next(L,-2);}else if
- EQ("concat"){lua_concat(L,getnum);}else if EQ("lessthan"){int a=getnum;
- lua_pushboolean(L,lua_lessthan(L,a,getnum));}else if EQ("equal"){int a=getnum;
- lua_pushboolean(L,lua_equal(L,a,getnum));}else if EQ("rawcall"){int narg=
- getnum;int nres=getnum;lua_call(L,narg,nres);}else if EQ("call"){int narg=
- getnum;int nres=getnum;lua_pcall(L,narg,nres,0);}else if EQ("loadstring"){
- size_t sl;const char*s=luaL_checklstring(L,getnum,&sl);luaL_loadbuffer(L,s,sl,
- s);}else if EQ("loadfile"){luaL_loadfile(L,luaL_checkstring(L,getnum));}else
- if EQ("setmetatable"){lua_setmetatable(L,getnum);}else if EQ("getmetatable"){
- if(lua_getmetatable(L,getnum)==0)lua_pushnil(L);}else if EQ("type"){
- lua_pushstring(L,lua_typename(L,lua_type(L,getnum)));}else if EQ("getn"){int i
- =getnum;lua_pushintegral(L,luaL_getn(L,i));}else if EQ("setn"){int i=getnum;
- int n=cast(int,lua_tonumber(L,-1));luaL_setn(L,i,n);lua_pop(L,1);}else
- luaL_error(L,"unknown instruction %s",buff);}return 0;}static void yieldf(
- lua_State*L,lua_Debug*ar){lua_yield(L,0);}static int setyhook(lua_State*L){if(
- lua_isnoneornil(L,1))lua_sethook(L,NULL,0,0);else{const char*smask=
- luaL_checkstring(L,1);int count=luaL_optint(L,2,0);int mask=0;if(strchr(smask,
- 'l'))mask|=LUA_MASKLINE;if(count>0)mask|=LUA_MASKCOUNT;lua_sethook(L,yieldf,
- mask,count);}return 0;}static int coresume(lua_State*L){int status;lua_State*
- co=lua_tothread(L,1);luaL_argcheck(L,co,1,"coroutine expected");status=
- lua_resume(co,0);if(status!=0){lua_pushboolean(L,0);lua_insert(L,-2);return 2;
- }else{lua_pushboolean(L,1);return 1;}}static const struct luaL_reg tests_funcs
- []={{"hash",hash_query},{"limits",get_limits},{"listcode",listcode},{"listk",
- listk},{"listlocals",listlocals},{"loadlib",loadlib},{"stacklevel",stacklevel}
- ,{"querystr",string_query},{"querytab",table_query},{"doit",test_do},{"testC",
- testC},{"ref",tref},{"getref",getref},{"unref",unref},{"d2s",d2s},{"s2d",s2d},
- {"metatable",metatable},{"upvalue",upvalue},{"newuserdata",newuserdata},{
- "pushuserdata",pushuserdata},{"udataval",udataval},{"doonnewstack",
- doonnewstack},{"newstate",newstate},{"closestate",closestate},{"doremote",
- doremote},{"log2",log2_aux},{"int2fb",int2fb_aux},{"totalmem",mem_query},{
- "resume",coresume},{"setyhook",setyhook},{NULL,NULL}};static void fim(void){if
- (!islocked)lua_close(lua_state);lua_assert(memdebug_numblocks==0);lua_assert(
- memdebug_total==0);}static int l_panic(lua_State*L){UNUSED(L);fprintf(stderr,
- "unable to recover; exiting\n");return 0;}int luaB_opentests(lua_State*L){
- lua_atpanic(L,l_panic);lua_userstateopen(L);lua_state=L;luaL_openlib(L,"T",
- tests_funcs,0);atexit(fim);return 0;}
- #undef main
- int main(int argc,char*argv[]){char*limit=getenv("MEMLIMIT");if(limit)
- memdebug_memlimit=strtoul(limit,NULL,10);l_main(argc,argv);return 0;}
- #endif
- #line 1 "ltm.c"
- #define ltm_c
- const char*const luaT_typenames[]={"nil","boolean","userdata","number",
- "string","table","function","userdata","thread"};void luaT_init(lua_State*L){
- static const char*const luaT_eventname[]={"__index","__newindex","__gc",
- "__mode","__eq","__add","__sub","__mul","__div","__pow","__unm","__lt","__le",
- "__concat","__call"};int i;for(i=0;i<TM_N;i++){G(L)->tmname[i]=luaS_new(L,
- luaT_eventname[i]);luaS_fix(G(L)->tmname[i]);}}const TObject*luaT_gettm(Table*
- events,TMS event,TString*ename){const TObject*tm=luaH_getstr(events,ename);
- lua_assert(event<=TM_EQ);if(ttisnil(tm)){events->flags|=cast(lu_byte,1u<<event
- );return NULL;}else return tm;}const TObject*luaT_gettmbyobj(lua_State*L,const
- TObject*o,TMS event){TString*ename=G(L)->tmname[event];switch(ttype(o)){case
- LUA_TTABLE:return luaH_getstr(hvalue(o)->metatable,ename);case LUA_TUSERDATA:
- return luaH_getstr(uvalue(o)->uv.metatable,ename);default:return&
- luaO_nilobject;}}
- #line 1 "lua.c"
- #define lua_c
- #ifdef LUA_USERCONFIG
- #include LUA_USERCONFIG
- #endif
- #ifdef _POSIX_C_SOURCE
- #define stdin_is_tty() isatty(0)
- #else
- #define stdin_is_tty() 1
- #endif
- #ifndef PROMPT
- #define PROMPT "> "
- #endif
- #ifndef PROMPT2
- #define PROMPT2 ">> "
- #endif
- #ifndef PROGNAME
- #define PROGNAME "lua"
- #endif
- #ifndef lua_userinit
- #define lua_userinit(L) openstdlibs(L)
- #endif
- #ifndef LUA_EXTRALIBS
- #define LUA_EXTRALIBS
- #endif
- static lua_State*L=NULL;static const char*progname=PROGNAME;LUALIB_API int
- luaopen_posix(lua_State*L);static const luaL_reg lualibs[]={{"base",
- luaopen_base},{"table",luaopen_table},{"io",luaopen_io},{"string",
- luaopen_string},{"debug",luaopen_debug},{"loadlib",luaopen_loadlib},{"posix",
- luaopen_posix},LUA_EXTRALIBS{NULL,NULL}};static void lstop(lua_State*l,
- lua_Debug*ar){(void)ar;lua_sethook(l,NULL,0,0);luaL_error(l,"interrupted!");}
- static void laction(int i){signal(i,SIG_DFL);lua_sethook(L,lstop,LUA_MASKCALL|
- LUA_MASKRET|LUA_MASKCOUNT,1);}static void print_usage(void){fprintf(stderr,
- "usage: %s [options] [script [args]].\n""Available options are:\n"
- " - execute stdin as a file\n"" -e stat execute string `stat'\n"
- " -i enter interactive mode after executing `script'\n"
- " -l name load and run library `name'\n"
- " -v show version information\n"" -- stop handling options\n",
- progname);}static void l_message(const char*pname,const char*msg){if(pname)
- fprintf(stderr,"%s: ",pname);fprintf(stderr,"%s\n",msg);}static int report(int
- status){const char*msg;if(status){msg=lua_tostring(L,-1);if(msg==NULL)msg=
- "(error with no message)";l_message(progname,msg);lua_pop(L,1);}return status;
- }static int lcall(int narg,int clear){int status;int base=lua_gettop(L)-narg;
- lua_pushliteral(L,"_TRACEBACK");lua_rawget(L,LUA_GLOBALSINDEX);lua_insert(L,
- base);signal(SIGINT,laction);status=lua_pcall(L,narg,(clear?0:LUA_MULTRET),
- base);signal(SIGINT,SIG_DFL);lua_remove(L,base);return status;}static void
- print_version(void){l_message(NULL,LUA_VERSION" "LUA_COPYRIGHT);}static void
- getargs(char*argv[],int n){int i;lua_newtable(L);for(i=0;argv[i];i++){
- lua_pushnumber(L,i-n);lua_pushstring(L,argv[i]);lua_rawset(L,-3);}
- lua_pushliteral(L,"n");lua_pushnumber(L,i-n-1);lua_rawset(L,-3);}static int
- docall(int status){if(status==0)status=lcall(0,1);return report(status);}
- static int file_input(const char*name){return docall(luaL_loadfile(L,name));}
- static int dostring(const char*s,const char*name){return docall(
- luaL_loadbuffer(L,s,strlen(s),name));}static int load_file(const char*name){
- lua_pushliteral(L,"require");lua_rawget(L,LUA_GLOBALSINDEX);if(!lua_isfunction
- (L,-1)){lua_pop(L,1);return file_input(name);}else{lua_pushstring(L,name);
- return report(lcall(1,1));}}
- #ifndef lua_saveline
- #define lua_saveline(L,line)
- #endif
- #ifndef lua_readline
- #define lua_readline(L,prompt) readline(L,prompt)
- #ifndef MAXINPUT
- #define MAXINPUT 512
- #endif
- static int readline(lua_State*l,const char*prompt){static char buffer[MAXINPUT
- ];if(prompt){fputs(prompt,stdout);fflush(stdout);}if(fgets(buffer,sizeof(
- buffer),stdin)==NULL)return 0;else{lua_pushstring(l,buffer);return 1;}}
- #endif
- static const char*get_prompt(int firstline){const char*p=NULL;lua_pushstring(L
- ,firstline?"_PROMPT":"_PROMPT2");lua_rawget(L,LUA_GLOBALSINDEX);p=lua_tostring
- (L,-1);if(p==NULL)p=(firstline?PROMPT:PROMPT2);lua_pop(L,1);return p;}static
- int incomplete(int status){if(status==LUA_ERRSYNTAX&&strstr(lua_tostring(L,-1)
- ,"near `<eof>'")!=NULL){lua_pop(L,1);return 1;}else return 0;}static int
- load_string(void){int status;lua_settop(L,0);if(lua_readline(L,get_prompt(1))
- ==0)return-1;if(lua_tostring(L,-1)[0]=='='){lua_pushfstring(L,"return %s",
- lua_tostring(L,-1)+1);lua_remove(L,-2);}for(;;){status=luaL_loadbuffer(L,
- lua_tostring(L,1),lua_strlen(L,1),"=stdin");if(!incomplete(status))break;if(
- lua_readline(L,get_prompt(0))==0)return-1;lua_concat(L,lua_gettop(L));}
- lua_saveline(L,lua_tostring(L,1));lua_remove(L,1);return status;}static void
- manual_input(void){int status;const char*oldprogname=progname;progname=NULL;
- while((status=load_string())!=-1){if(status==0)status=lcall(0,0);report(status
- );if(status==0&&lua_gettop(L)>0){lua_getglobal(L,"print");lua_insert(L,1);if(
- lua_pcall(L,lua_gettop(L)-1,0,0)!=0)l_message(progname,lua_pushfstring(L,
- "error calling `print' (%s)",lua_tostring(L,-1)));}}lua_settop(L,0);fputs("\n"
- ,stdout);progname=oldprogname;}static int handle_argv(char*argv[],int*
- interactive){if(argv[1]==NULL){if(stdin_is_tty()){print_version();manual_input
- ();}else file_input(NULL);}else{int i;for(i=1;argv[i]!=NULL;i++){if(argv[i][0]
- !='-')break;switch(argv[i][1]){case'-':{if(argv[i][2]!='\0'){print_usage();
- return 1;}i++;goto endloop;}case'\0':{file_input(NULL);break;}case'i':{*
- interactive=1;break;}case'v':{print_version();break;}case'e':{const char*chunk
- =argv[i]+2;if(*chunk=='\0')chunk=argv[++i];if(chunk==NULL){print_usage();
- return 1;}if(dostring(chunk,"=<command line>")!=0)return 1;break;}case'l':{
- const char*filename=argv[i]+2;if(*filename=='\0')filename=argv[++i];if(
- filename==NULL){print_usage();return 1;}if(load_file(filename))return 1;break;
- }case'c':{l_message(progname,"option `-c' is deprecated");break;}case's':{
- l_message(progname,"option `-s' is deprecated");break;}default:{print_usage();
- return 1;}}}endloop:if(argv[i]!=NULL){const char*filename=argv[i];getargs(argv
- ,i);lua_setglobal(L,"arg");if(strcmp(filename,"/dev/stdin")==0)filename=NULL;
- return file_input(filename);}}return 0;}static void openstdlibs(lua_State*l){
- const luaL_reg*lib=lualibs;for(;lib->func;lib++){lib->func(l);lua_settop(l,0);
- }}static int handle_luainit(void){const char*init=getenv("LUA_INIT");if(init==
- NULL)return 0;else if(init[0]=='@')return file_input(init+1);else return
- dostring(init,"=LUA_INIT");}struct Smain{int argc;char**argv;int status;};
- static int pmain(lua_State*l){struct Smain*s=(struct Smain*)lua_touserdata(l,1
- );int status;int interactive=0;if(s->argv[0]&&s->argv[0][0])progname=s->argv[0
- ];L=l;lua_userinit(l);status=handle_luainit();if(status==0){status=handle_argv
- (s->argv,&interactive);if(status==0&&interactive)manual_input();}s->status=
- status;return 0;}int main(int argc,char*argv[]){int status;struct Smain s;
- lua_State*l=lua_open();if(l==NULL){l_message(argv[0],
- "cannot create state: not enough memory");return EXIT_FAILURE;}s.argc=argc;s.
- argv=argv;status=lua_cpcall(l,&pmain,&s);report(status);lua_close(l);return(
- status||s.status)?EXIT_FAILURE:EXIT_SUCCESS;}
- #line 1 "lundump.c"
- #define lundump_c
- #define LoadByte (lu_byte)ezgetc
- typedef struct{lua_State*L;ZIO*Z;Mbuffer*b;int swap;const char*name;}LoadState
- ;static void unexpectedEOZ(LoadState*S){luaG_runerror(S->L,
- "unexpected end of file in %s",S->name);}static int ezgetc(LoadState*S){int c=
- zgetc(S->Z);if(c==EOZ)unexpectedEOZ(S);return c;}static void ezread(LoadState*
- S,void*b,int n){int r=luaZ_read(S->Z,b,n);if(r!=0)unexpectedEOZ(S);}static
- void LoadBlock(LoadState*S,void*b,size_t size){if(S->swap){char*p=(char*)b+
- size-1;int n=size;while(n--)*p--=(char)ezgetc(S);}else ezread(S,b,size);}
- static void LoadVector(LoadState*S,void*b,int m,size_t size){if(S->swap){char*
- q=(char*)b;while(m--){char*p=q+size-1;int n=size;while(n--)*p--=(char)ezgetc(S
- );q+=size;}}else ezread(S,b,m*size);}static int LoadInt(LoadState*S){int x;
- LoadBlock(S,&x,sizeof(x));if(x<0)luaG_runerror(S->L,"bad integer in %s",S->
- name);return x;}static size_t LoadSize(LoadState*S){size_t x;LoadBlock(S,&x,
- sizeof(x));return x;}static lua_Number LoadNumber(LoadState*S){lua_Number x;
- LoadBlock(S,&x,sizeof(x));return x;}static TString*LoadString(LoadState*S){
- size_t size=LoadSize(S);if(size==0)return NULL;else{char*s=luaZ_openspace(S->L
- ,S->b,size);ezread(S,s,size);return luaS_newlstr(S->L,s,size-1);}}static void
- LoadCode(LoadState*S,Proto*f){int size=LoadInt(S);f->code=luaM_newvector(S->L,
- size,Instruction);f->sizecode=size;LoadVector(S,f->code,size,sizeof(*f->code))
- ;}static void LoadLocals(LoadState*S,Proto*f){int i,n;n=LoadInt(S);f->locvars=
- luaM_newvector(S->L,n,LocVar);f->sizelocvars=n;for(i=0;i<n;i++){f->locvars[i].
- varname=LoadString(S);f->locvars[i].startpc=LoadInt(S);f->locvars[i].endpc=
- LoadInt(S);}}static void LoadLines(LoadState*S,Proto*f){int size=LoadInt(S);f
- ->lineinfo=luaM_newvector(S->L,size,int);f->sizelineinfo=size;LoadVector(S,f->
- lineinfo,size,sizeof(*f->lineinfo));}static void LoadUpvalues(LoadState*S,
- Proto*f){int i,n;n=LoadInt(S);if(n!=0&&n!=f->nups)luaG_runerror(S->L,
- "bad nupvalues in %s: read %d; expected %d",S->name,n,f->nups);f->upvalues=
- luaM_newvector(S->L,n,TString*);f->sizeupvalues=n;for(i=0;i<n;i++)f->upvalues[
- i]=LoadString(S);}static Proto*LoadFunction(LoadState*S,TString*p);static void
- LoadConstants(LoadState*S,Proto*f){int i,n;n=LoadInt(S);f->k=luaM_newvector(S
- ->L,n,TObject);f->sizek=n;for(i=0;i<n;i++){TObject*o=&f->k[i];int t=LoadByte(S
- );switch(t){case LUA_TNUMBER:setnvalue(o,LoadNumber(S));break;case LUA_TSTRING
- :setsvalue2n(o,LoadString(S));break;case LUA_TNIL:setnilvalue(o);break;default
- :luaG_runerror(S->L,"bad constant type (%d) in %s",t,S->name);break;}}n=
- LoadInt(S);f->p=luaM_newvector(S->L,n,Proto*);f->sizep=n;for(i=0;i<n;i++)f->p[
- i]=LoadFunction(S,f->source);}static Proto*LoadFunction(LoadState*S,TString*p)
- {Proto*f=luaF_newproto(S->L);f->source=LoadString(S);if(f->source==NULL)f->
- source=p;f->lineDefined=LoadInt(S);f->nups=LoadByte(S);f->numparams=LoadByte(S
- );f->is_vararg=LoadByte(S);f->maxstacksize=LoadByte(S);LoadLines(S,f);
- LoadLocals(S,f);LoadUpvalues(S,f);LoadConstants(S,f);LoadCode(S,f);
- #ifndef TRUST_BINARIES
- if(!luaG_checkcode(f))luaG_runerror(S->L,"bad code in %s",S->name);
- #endif
- return f;}static void LoadSignature(LoadState*S){const char*s=LUA_SIGNATURE;
- while(*s!=0&&ezgetc(S)==*s)++s;if(*s!=0)luaG_runerror(S->L,
- "bad signature in %s",S->name);}static void TestSize(LoadState*S,int s,const
- char*what){int r=LoadByte(S);if(r!=s)luaG_runerror(S->L,
- "virtual machine mismatch in %s: ""size of %s is %d but read %d",S->name,what,
- s,r);}
- #define TESTSIZE(s,w) TestSize(S,s,w)
- #define V(v) v/16,v%16
- static void LoadHeader(LoadState*S){int version;lua_Number x,tx=TEST_NUMBER;
- LoadSignature(S);version=LoadByte(S);if(version>VERSION)luaG_runerror(S->L,
- "%s too new: ""read version %d.%d; expected at most %d.%d",S->name,V(version),
- V(VERSION));if(version<VERSION0)luaG_runerror(S->L,"%s too old: "
- "read version %d.%d; expected at least %d.%d",S->name,V(version),V(VERSION0));
- S->swap=(luaU_endianness()!=LoadByte(S));TESTSIZE(sizeof(int),"int");TESTSIZE(
- sizeof(size_t),"size_t");TESTSIZE(sizeof(Instruction),"Instruction");TESTSIZE(
- SIZE_OP,"OP");TESTSIZE(SIZE_A,"A");TESTSIZE(SIZE_B,"B");TESTSIZE(SIZE_C,"C");
- TESTSIZE(sizeof(lua_Number),"number");x=LoadNumber(S);if((long)x!=(long)tx)
- luaG_runerror(S->L,"unknown number format in %s",S->name);}static Proto*
- LoadChunk(LoadState*S){LoadHeader(S);return LoadFunction(S,NULL);}Proto*
- luaU_undump(lua_State*L,ZIO*Z,Mbuffer*buff){LoadState S;const char*s=zname(Z);
- if(*s=='@'||*s=='=')S.name=s+1;else if(*s==LUA_SIGNATURE[0])S.name=
- "binary string";else S.name=s;S.L=L;S.Z=Z;S.b=buff;return LoadChunk(&S);}int
- luaU_endianness(void){int x=1;return*(char*)&x;}
- #line 1 "lvm.c"
- #define lvm_c
- #ifndef lua_number2str
- #define lua_number2str(s,n) sprintf((s),LUA_NUMBER_FMT,(n))
- #endif
- #define MAXTAGLOOP 100
- const TObject*luaV_tonumber(const TObject*obj,TObject*n){lua_Number num;if(
- ttisnumber(obj))return obj;if(ttisstring(obj)&&luaO_str2d(svalue(obj),&num)){
- setnvalue(n,num);return n;}else return NULL;}int luaV_tostring(lua_State*L,
- StkId obj){if(!ttisnumber(obj))return 0;else{char s[32];lua_number2str(s,
- nvalue(obj));setsvalue2s(obj,luaS_new(L,s));return 1;}}static void traceexec(
- lua_State*L){lu_byte mask=L->hookmask;if(mask&LUA_MASKCOUNT){if(L->hookcount==
- 0){resethookcount(L);luaD_callhook(L,LUA_HOOKCOUNT,-1);return;}}if(mask&
- LUA_MASKLINE){CallInfo*ci=L->ci;Proto*p=ci_func(ci)->l.p;int newline=getline(p
- ,pcRel(*ci->u.l.pc,p));if(!L->hookinit){luaG_inithooks(L);return;}lua_assert(
- ci->state&CI_HASFRAME);if(pcRel(*ci->u.l.pc,p)==0)ci->u.l.savedpc=*ci->u.l.pc;
- if(*ci->u.l.pc<=ci->u.l.savedpc||newline!=getline(p,pcRel(ci->u.l.savedpc,p)))
- {luaD_callhook(L,LUA_HOOKLINE,newline);ci=L->ci;}ci->u.l.savedpc=*ci->u.l.pc;}
- }static void callTMres(lua_State*L,const TObject*f,const TObject*p1,const
- TObject*p2){setobj2s(L->top,f);setobj2s(L->top+1,p1);setobj2s(L->top+2,p2);
- luaD_checkstack(L,3);L->top+=3;luaD_call(L,L->top-3,1);L->top--;}static void
- callTM(lua_State*L,const TObject*f,const TObject*p1,const TObject*p2,const
- TObject*p3){setobj2s(L->top,f);setobj2s(L->top+1,p1);setobj2s(L->top+2,p2);
- setobj2s(L->top+3,p3);luaD_checkstack(L,4);L->top+=4;luaD_call(L,L->top-4,0);}
- static const TObject*luaV_index(lua_State*L,const TObject*t,TObject*key,int
- loop){const TObject*tm=fasttm(L,hvalue(t)->metatable,TM_INDEX);if(tm==NULL)
- return&luaO_nilobject;if(ttisfunction(tm)){callTMres(L,tm,t,key);return L->top
- ;}else return luaV_gettable(L,tm,key,loop);}static const TObject*
- luaV_getnotable(lua_State*L,const TObject*t,TObject*key,int loop){const
- TObject*tm=luaT_gettmbyobj(L,t,TM_INDEX);if(ttisnil(tm))luaG_typeerror(L,t,
- "index");if(ttisfunction(tm)){callTMres(L,tm,t,key);return L->top;}else return
- luaV_gettable(L,tm,key,loop);}const TObject*luaV_gettable(lua_State*L,const
- TObject*t,TObject*key,int loop){if(loop>MAXTAGLOOP)luaG_runerror(L,
- "loop in gettable");if(ttistable(t)){Table*h=hvalue(t);const TObject*v=
- luaH_get(h,key);if(!ttisnil(v))return v;else return luaV_index(L,t,key,loop+1)
- ;}else return luaV_getnotable(L,t,key,loop+1);}void luaV_settable(lua_State*L,
- const TObject*t,TObject*key,StkId val){const TObject*tm;int loop=0;do{if(
- ttistable(t)){Table*h=hvalue(t);TObject*oldval=luaH_set(L,h,key);if(!ttisnil(
- oldval)||(tm=fasttm(L,h->metatable,TM_NEWINDEX))==NULL){setobj2t(oldval,val);
- return;}}else if(ttisnil(tm=luaT_gettmbyobj(L,t,TM_NEWINDEX)))luaG_typeerror(L
- ,t,"index");if(ttisfunction(tm)){callTM(L,tm,t,key,val);return;}t=tm;}while(++
- loop<=MAXTAGLOOP);luaG_runerror(L,"loop in settable");}static int call_binTM(
- lua_State*L,const TObject*p1,const TObject*p2,StkId res,TMS event){ptrdiff_t
- result=savestack(L,res);const TObject*tm=luaT_gettmbyobj(L,p1,event);if(
- ttisnil(tm))tm=luaT_gettmbyobj(L,p2,event);if(!ttisfunction(tm))return 0;
- callTMres(L,tm,p1,p2);res=restorestack(L,result);setobjs2s(res,L->top);return
- 1;}static const TObject*get_compTM(lua_State*L,Table*mt1,Table*mt2,TMS event){
- const TObject*tm1=fasttm(L,mt1,event);const TObject*tm2;if(tm1==NULL)return
- NULL;if(mt1==mt2)return tm1;tm2=fasttm(L,mt2,event);if(tm2==NULL)return NULL;
- if(luaO_rawequalObj(tm1,tm2))return tm1;return NULL;}static int call_orderTM(
- lua_State*L,const TObject*p1,const TObject*p2,TMS event){const TObject*tm1=
- luaT_gettmbyobj(L,p1,event);const TObject*tm2;if(ttisnil(tm1))return-1;tm2=
- luaT_gettmbyobj(L,p2,event);if(!luaO_rawequalObj(tm1,tm2))return-1;callTMres(L
- ,tm1,p1,p2);return!l_isfalse(L->top);}static int luaV_strcmp(const TString*ls,
- const TString*rs){const char*l=getstr(ls);size_t ll=ls->tsv.len;const char*r=
- getstr(rs);size_t lr=rs->tsv.len;for(;;){int temp=strcoll(l,r);if(temp!=0)
- return temp;else{size_t len=strlen(l);if(len==lr)return(len==ll)?0:1;else if(
- len==ll)return-1;len++;l+=len;ll-=len;r+=len;lr-=len;}}}int luaV_lessthan(
- lua_State*L,const TObject*l,const TObject*r){int res;if(ttype(l)!=ttype(r))
- return luaG_ordererror(L,l,r);else if(ttisnumber(l))return nvalue(l)<nvalue(r)
- ;else if(ttisstring(l))return luaV_strcmp(tsvalue(l),tsvalue(r))<0;else if((
- res=call_orderTM(L,l,r,TM_LT))!=-1)return res;return luaG_ordererror(L,l,r);}
- static int luaV_lessequal(lua_State*L,const TObject*l,const TObject*r){int res
- ;if(ttype(l)!=ttype(r))return luaG_ordererror(L,l,r);else if(ttisnumber(l))
- return nvalue(l)<=nvalue(r);else if(ttisstring(l))return luaV_strcmp(tsvalue(l
- ),tsvalue(r))<=0;else if((res=call_orderTM(L,l,r,TM_LE))!=-1)return res;else
- if((res=call_orderTM(L,r,l,TM_LT))!=-1)return!res;return luaG_ordererror(L,l,r
- );}int luaV_equalval(lua_State*L,const TObject*t1,const TObject*t2){const
- TObject*tm;lua_assert(ttype(t1)==ttype(t2));switch(ttype(t1)){case LUA_TNIL:
- return 1;case LUA_TNUMBER:return nvalue(t1)==nvalue(t2);case LUA_TBOOLEAN:
- return bvalue(t1)==bvalue(t2);case LUA_TLIGHTUSERDATA:return pvalue(t1)==
- pvalue(t2);case LUA_TUSERDATA:{if(uvalue(t1)==uvalue(t2))return 1;tm=
- get_compTM(L,uvalue(t1)->uv.metatable,uvalue(t2)->uv.metatable,TM_EQ);break;}
- case LUA_TTABLE:{if(hvalue(t1)==hvalue(t2))return 1;tm=get_compTM(L,hvalue(t1)
- ->metatable,hvalue(t2)->metatable,TM_EQ);break;}default:return gcvalue(t1)==
- gcvalue(t2);}if(tm==NULL)return 0;callTMres(L,tm,t1,t2);return!l_isfalse(L->
- top);}void luaV_concat(lua_State*L,int total,int last){do{StkId top=L->base+
- last+1;int n=2;if(!tostring(L,top-2)||!tostring(L,top-1)){if(!call_binTM(L,top
- -2,top-1,top-2,TM_CONCAT))luaG_concaterror(L,top-2,top-1);}else if(tsvalue(top
- -1)->tsv.len>0){lu_mem tl=cast(lu_mem,tsvalue(top-1)->tsv.len)+cast(lu_mem,
- tsvalue(top-2)->tsv.len);char*buffer;int i;while(n<total&&tostring(L,top-n-1))
- {tl+=tsvalue(top-n-1)->tsv.len;n++;}if(tl>MAX_SIZET)luaG_runerror(L,
- "string size overflow");buffer=luaZ_openspace(L,&G(L)->buff,tl);tl=0;for(i=n;i
- >0;i--){size_t l=tsvalue(top-i)->tsv.len;memcpy(buffer+tl,svalue(top-i),l);tl
- +=l;}setsvalue2s(top-n,luaS_newlstr(L,buffer,tl));}total-=n-1;last-=n-1;}while
- (total>1);}static void Arith(lua_State*L,StkId ra,const TObject*rb,const
- TObject*rc,TMS op){TObject tempb,tempc;const TObject*b,*c;if((b=luaV_tonumber(
- rb,&tempb))!=NULL&&(c=luaV_tonumber(rc,&tempc))!=NULL){switch(op){case TM_ADD:
- setnvalue(ra,nvalue(b)+nvalue(c));break;case TM_SUB:setnvalue(ra,nvalue(b)-
- nvalue(c));break;case TM_MUL:setnvalue(ra,nvalue(b)*nvalue(c));break;case
- TM_DIV:setnvalue(ra,nvalue(b)/nvalue(c));break;case TM_POW:{const TObject*f=
- luaH_getstr(hvalue(gt(L)),G(L)->tmname[TM_POW]);ptrdiff_t res=savestack(L,ra);
- if(!ttisfunction(f))luaG_runerror(L,"`__pow' (`^' operator) is not a function"
- );callTMres(L,f,b,c);ra=restorestack(L,res);setobjs2s(ra,L->top);break;}
- default:lua_assert(0);break;}}else if(!call_binTM(L,rb,rc,ra,op))
- luaG_aritherror(L,rb,rc);}
- #define runtime_check(L, c){if(!(c))return 0;}
- #define RA(i) (base+GETARG_A(i))
- #define XRA(i) (L->base+GETARG_A(i))
- #define RB(i) (base+GETARG_B(i))
- #define RKB(i) ((GETARG_B(i)<MAXSTACK)?RB(i):k+GETARG_B(i)-MAXSTACK)
- #define RC(i) (base+GETARG_C(i))
- #define RKC(i) ((GETARG_C(i)<MAXSTACK)?RC(i):k+GETARG_C(i)-MAXSTACK)
- #define KBx(i) (k+GETARG_Bx(i))
- #define dojump(pc, i)((pc)+=(i))
- StkId luaV_execute(lua_State*L){LClosure*cl;TObject*k;const Instruction*pc;
- callentry:if(L->hookmask&LUA_MASKCALL){L->ci->u.l.pc=&pc;luaD_callhook(L,
- LUA_HOOKCALL,-1);}retentry:L->ci->u.l.pc=&pc;lua_assert(L->ci->state==
- CI_SAVEDPC||L->ci->state==(CI_SAVEDPC|CI_CALLING));L->ci->state=CI_HASFRAME;pc
- =L->ci->u.l.savedpc;cl=&clvalue(L->base-1)->l;k=cl->p->k;for(;;){const
- Instruction i=*pc++;StkId base,ra;if((L->hookmask&(LUA_MASKLINE|LUA_MASKCOUNT)
- )&&(--L->hookcount==0||L->hookmask&LUA_MASKLINE)){traceexec(L);if(L->ci->state
- &CI_YIELD){L->ci->u.l.savedpc=pc-1;L->ci->state=CI_YIELD|CI_SAVEDPC;return
- NULL;}}base=L->base;ra=RA(i);lua_assert(L->ci->state&CI_HASFRAME);lua_assert(
- base==L->ci->base);lua_assert(L->top<=L->stack+L->stacksize&&L->top>=base);
- lua_assert(L->top==L->ci->top||GET_OPCODE(i)==OP_CALL||GET_OPCODE(i)==
- OP_TAILCALL||GET_OPCODE(i)==OP_RETURN||GET_OPCODE(i)==OP_SETLISTO);switch(
- GET_OPCODE(i)){case OP_MOVE:{setobjs2s(ra,RB(i));break;}case OP_LOADK:{
- setobj2s(ra,KBx(i));break;}case OP_LOADBOOL:{setbvalue(ra,GETARG_B(i));if(
- GETARG_C(i))pc++;break;}case OP_LOADNIL:{TObject*rb=RB(i);do{setnilvalue(rb--)
- ;}while(rb>=ra);break;}case OP_GETUPVAL:{int b=GETARG_B(i);setobj2s(ra,cl->
- upvals[b]->v);break;}case OP_GETGLOBAL:{TObject*rb=KBx(i);const TObject*v;
- lua_assert(ttisstring(rb)&&ttistable(&cl->g));v=luaH_getstr(hvalue(&cl->g),
- tsvalue(rb));if(!ttisnil(v)){setobj2s(ra,v);}else setobj2s(XRA(i),luaV_index(L
- ,&cl->g,rb,0));break;}case OP_GETTABLE:{StkId rb=RB(i);TObject*rc=RKC(i);if(
- ttistable(rb)){const TObject*v=luaH_get(hvalue(rb),rc);if(!ttisnil(v)){
- setobj2s(ra,v);}else setobj2s(XRA(i),luaV_index(L,rb,rc,0));}else setobj2s(XRA
- (i),luaV_getnotable(L,rb,rc,0));break;}case OP_SETGLOBAL:{lua_assert(
- ttisstring(KBx(i))&&ttistable(&cl->g));luaV_settable(L,&cl->g,KBx(i),ra);break
- ;}case OP_SETUPVAL:{int b=GETARG_B(i);setobj(cl->upvals[b]->v,ra);break;}case
- OP_SETTABLE:{luaV_settable(L,ra,RKB(i),RKC(i));break;}case OP_NEWTABLE:{int b=
- GETARG_B(i);b=fb2int(b);sethvalue(ra,luaH_new(L,b,GETARG_C(i)));luaC_checkGC(L
- );break;}case OP_SELF:{StkId rb=RB(i);TObject*rc=RKC(i);runtime_check(L,
- ttisstring(rc));setobjs2s(ra+1,rb);if(ttistable(rb)){const TObject*v=
- luaH_getstr(hvalue(rb),tsvalue(rc));if(!ttisnil(v)){setobj2s(ra,v);}else
- setobj2s(XRA(i),luaV_index(L,rb,rc,0));}else setobj2s(XRA(i),luaV_getnotable(L
- ,rb,rc,0));break;}case OP_ADD:{TObject*rb=RKB(i);TObject*rc=RKC(i);if(
- ttisnumber(rb)&&ttisnumber(rc)){setnvalue(ra,nvalue(rb)+nvalue(rc));}else
- Arith(L,ra,rb,rc,TM_ADD);break;}case OP_SUB:{TObject*rb=RKB(i);TObject*rc=RKC(
- i);if(ttisnumber(rb)&&ttisnumber(rc)){setnvalue(ra,nvalue(rb)-nvalue(rc));}
- else Arith(L,ra,rb,rc,TM_SUB);break;}case OP_MUL:{TObject*rb=RKB(i);TObject*rc
- =RKC(i);if(ttisnumber(rb)&&ttisnumber(rc)){setnvalue(ra,nvalue(rb)*nvalue(rc))
- ;}else Arith(L,ra,rb,rc,TM_MUL);break;}case OP_DIV:{TObject*rb=RKB(i);TObject*
- rc=RKC(i);if(ttisnumber(rb)&&ttisnumber(rc)){setnvalue(ra,nvalue(rb)/nvalue(rc
- ));}else Arith(L,ra,rb,rc,TM_DIV);break;}case OP_POW:{Arith(L,ra,RKB(i),RKC(i)
- ,TM_POW);break;}case OP_UNM:{const TObject*rb=RB(i);TObject temp;if(tonumber(
- rb,&temp)){setnvalue(ra,-nvalue(rb));}else{setnilvalue(&temp);if(!call_binTM(L
- ,RB(i),&temp,ra,TM_UNM))luaG_aritherror(L,RB(i),&temp);}break;}case OP_NOT:{
- int res=l_isfalse(RB(i));setbvalue(ra,res);break;}case OP_CONCAT:{int b=
- GETARG_B(i);int c=GETARG_C(i);luaV_concat(L,c-b+1,c);base=L->base;setobjs2s(RA
- (i),base+b);luaC_checkGC(L);break;}case OP_JMP:{dojump(pc,GETARG_sBx(i));break
- ;}case OP_EQ:{if(equalobj(L,RKB(i),RKC(i))!=GETARG_A(i))pc++;else dojump(pc,
- GETARG_sBx(*pc)+1);break;}case OP_LT:{if(luaV_lessthan(L,RKB(i),RKC(i))!=
- GETARG_A(i))pc++;else dojump(pc,GETARG_sBx(*pc)+1);break;}case OP_LE:{if(
- luaV_lessequal(L,RKB(i),RKC(i))!=GETARG_A(i))pc++;else dojump(pc,GETARG_sBx(*
- pc)+1);break;}case OP_TEST:{TObject*rb=RB(i);if(l_isfalse(rb)==GETARG_C(i))pc
- ++;else{setobjs2s(ra,rb);dojump(pc,GETARG_sBx(*pc)+1);}break;}case OP_CALL:
- case OP_TAILCALL:{StkId firstResult;int b=GETARG_B(i);int nresults;if(b!=0)L->
- top=ra+b;nresults=GETARG_C(i)-1;firstResult=luaD_precall(L,ra);if(firstResult)
- {if(firstResult>L->top){lua_assert(L->ci->state==(CI_C|CI_YIELD));(L->ci-1)->u
- .l.savedpc=pc;(L->ci-1)->state=CI_SAVEDPC;return NULL;}luaD_poscall(L,nresults
- ,firstResult);if(nresults>=0)L->top=L->ci->top;}else{if(GET_OPCODE(i)==OP_CALL
- ){(L->ci-1)->u.l.savedpc=pc;(L->ci-1)->state=(CI_SAVEDPC|CI_CALLING);}else{int
- aux;base=(L->ci-1)->base;ra=RA(i);if(L->openupval)luaF_close(L,base);for(aux=
- 0;ra+aux<L->top;aux++)setobjs2s(base+aux-1,ra+aux);(L->ci-1)->top=L->top=base+
- aux;lua_assert(L->ci->state&CI_SAVEDPC);(L->ci-1)->u.l.savedpc=L->ci->u.l.
- savedpc;(L->ci-1)->u.l.tailcalls++;(L->ci-1)->state=CI_SAVEDPC;L->ci--;L->base
- =L->ci->base;}goto callentry;}break;}case OP_RETURN:{CallInfo*ci=L->ci-1;int b
- =GETARG_B(i);if(b!=0)L->top=ra+b-1;lua_assert(L->ci->state&CI_HASFRAME);if(L->
- openupval)luaF_close(L,base);L->ci->state=CI_SAVEDPC;L->ci->u.l.savedpc=pc;if(
- !(ci->state&CI_CALLING)){lua_assert((ci->state&CI_C)||ci->u.l.pc!=&pc);return
- ra;}else{int nresults;lua_assert(ttisfunction(ci->base-1)&&(ci->state&
- CI_SAVEDPC));lua_assert(GET_OPCODE(*(ci->u.l.savedpc-1))==OP_CALL);nresults=
- GETARG_C(*(ci->u.l.savedpc-1))-1;luaD_poscall(L,nresults,ra);if(nresults>=0)L
- ->top=L->ci->top;goto retentry;}}case OP_FORLOOP:{lua_Number step,idx,limit;
- const TObject*plimit=ra+1;const TObject*pstep=ra+2;if(!ttisnumber(ra))
- luaG_runerror(L,"`for' initial value must be a number");if(!tonumber(plimit,ra
- +1))luaG_runerror(L,"`for' limit must be a number");if(!tonumber(pstep,ra+2))
- luaG_runerror(L,"`for' step must be a number");step=nvalue(pstep);idx=nvalue(
- ra)+step;limit=nvalue(plimit);if(step>0?idx<=limit:idx>=limit){dojump(pc,
- GETARG_sBx(i));chgnvalue(ra,idx);}break;}case OP_TFORLOOP:{int nvar=GETARG_C(i
- )+1;StkId cb=ra+nvar+2;setobjs2s(cb,ra);setobjs2s(cb+1,ra+1);setobjs2s(cb+2,ra
- +2);L->top=cb+3;luaD_call(L,cb,nvar);L->top=L->ci->top;ra=XRA(i)+2;cb=ra+nvar;
- do{nvar--;setobjs2s(ra+nvar,cb+nvar);}while(nvar>0);if(ttisnil(ra))pc++;else
- dojump(pc,GETARG_sBx(*pc)+1);break;}case OP_TFORPREP:{if(ttistable(ra)){
- setobjs2s(ra+1,ra);setobj2s(ra,luaH_getstr(hvalue(gt(L)),luaS_new(L,"next")));
- }dojump(pc,GETARG_sBx(i));break;}case OP_SETLIST:case OP_SETLISTO:{int bc;int
- n;Table*h;runtime_check(L,ttistable(ra));h=hvalue(ra);bc=GETARG_Bx(i);if(
- GET_OPCODE(i)==OP_SETLIST)n=(bc&(LFIELDS_PER_FLUSH-1))+1;else{n=L->top-ra-1;L
- ->top=L->ci->top;}bc&=~(LFIELDS_PER_FLUSH-1);for(;n>0;n--)setobj2t(luaH_setnum
- (L,h,bc+n),ra+n);break;}case OP_CLOSE:{luaF_close(L,ra);break;}case OP_CLOSURE
- :{Proto*p;Closure*ncl;int nup,j;p=cl->p->p[GETARG_Bx(i)];nup=p->nups;ncl=
- luaF_newLclosure(L,nup,&cl->g);ncl->l.p=p;for(j=0;j<nup;j++,pc++){if(
- GET_OPCODE(*pc)==OP_GETUPVAL)ncl->l.upvals[j]=cl->upvals[GETARG_B(*pc)];else{
- lua_assert(GET_OPCODE(*pc)==OP_MOVE);ncl->l.upvals[j]=luaF_findupval(L,base+
- GETARG_B(*pc));}}setclvalue(ra,ncl);luaC_checkGC(L);break;}}}}
- #line 1 "lzio.c"
- #define lzio_c
- int luaZ_fill(ZIO*z){size_t size;const char*buff=z->reader(NULL,z->data,&size)
- ;if(buff==NULL||size==0)return EOZ;z->n=size-1;z->p=buff;return char2int(*(z->
- p++));}int luaZ_lookahead(ZIO*z){if(z->n==0){int c=luaZ_fill(z);if(c==EOZ)
- return c;z->n++;z->p--;}return char2int(*z->p);}void luaZ_init(ZIO*z,
- lua_Chunkreader reader,void*data,const char*name){z->reader=reader;z->data=
- data;z->name=name;z->n=0;z->p=NULL;}size_t luaZ_read(ZIO*z,void*b,size_t n){
- while(n){size_t m;if(z->n==0){if(luaZ_fill(z)==EOZ)return n;else{++z->n;--z->p
- ;}}m=(n<=z->n)?n:z->n;memcpy(b,z->p,m);z->n-=m;z->p+=m;b=(char*)b+m;n-=m;}
- return 0;}char*luaZ_openspace(lua_State*L,Mbuffer*buff,size_t n){if(n>buff->
- buffsize){if(n<LUA_MINBUFFER)n=LUA_MINBUFFER;luaM_reallocvector(L,buff->buffer
- ,buff->buffsize,n,char);buff->buffsize=n;}return buff->buffer;}
|