123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075 |
- /*-
- * Copyright 2016 Vsevolod Stakhov
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include "lua_common.h"
- #include "message.h"
- #include "images.h"
- #include "archives.h"
- #include "utlist.h"
- #include "unix-std.h"
- #include "libmime/smtp_parsers.h"
- #include "libserver/mempool_vars_internal.h"
- #include "libserver/dkim.h"
- #include "libserver/task.h"
- #include "libserver/cfg_file_private.h"
- #include "libmime/filter_private.h"
- #include "libstat/stat_api.h"
- #include "libutil/map_helpers.h"
-
- #include <math.h>
-
- /***
- * @module rspamd_task
- * This module provides routines for tasks manipulation in rspamd. Tasks usually
- * represent messages being scanned, and this API provides access to such elements
- * as headers, symbols, metrics and so on and so forth. Normally, task objects
- * are passed to the lua callbacks allowing to check specific properties of messages
- * and add the corresponding symbols to the scan's results.
- @example
- rspamd_config.DATE_IN_PAST = function(task)
- local dm = task:get_date{format = 'message', gmt = true}
- local dt = task:get_date{format = 'connect', gmt = true}
- -- A day
- if dt - dm > 86400 then
- return true
- end
-
- return false
- end
- */
-
- /* Task methods */
- /***
- * @function rspamd_task.load_from_file(filename[, cfg])
- * Loads a message from specific file
- * @return {boolean,rspamd_task|error} status + new task or error message
- */
- LUA_FUNCTION_DEF (task, load_from_file);
- /***
- * @function rspamd_task.load_from_string(message[, cfg])
- * Loads a message from specific file
- * @return {boolean,rspamd_task|error} status + new task or error message
- */
- LUA_FUNCTION_DEF (task, load_from_string);
-
- LUA_FUNCTION_DEF (task, get_message);
- /***
- * @method task:process_message()
- * Parses message
- */
- LUA_FUNCTION_DEF (task, process_message);
- /***
- * @method task:get_cfg()
- * Get configuration object for a task.
- * @return {rspamd_config} (config.md)[configuration object] for the task
- */
- LUA_FUNCTION_DEF (task, get_cfg);
- LUA_FUNCTION_DEF (task, set_cfg);
- LUA_FUNCTION_DEF (task, destroy);
- /***
- * @method task:get_mempool()
- * Returns memory pool valid for a lifetime of task. It is used internally by
- * many rspamd routines.
- * @return {rspamd_mempool} memory pool object
- */
- LUA_FUNCTION_DEF (task, get_mempool);
- /***
- * @method task:get_session()
- * Returns asynchronous session object that is used by many rspamd asynchronous
- * utilities internally.
- * @return {rspamd_session} session object
- */
- LUA_FUNCTION_DEF (task, get_session);
- /***
- * @method task:get_ev_base()
- * Return asynchronous event base for using in callbacks and resolver.
- * @return {rspamd_ev_base} event base
- */
- LUA_FUNCTION_DEF (task, get_ev_base);
- /***
- * @method task:get_worker()
- * Returns a worker object associated with the task
- * @return {rspamd_worker} worker object
- */
- LUA_FUNCTION_DEF (task, get_worker);
- /***
- * @method task:insert_result([enforce_symbol,]symbol, weight[, option1, ...])
- * Insert specific symbol to the tasks scanning results assigning the initial
- * weight to it.
- * @param {boolean} enforce_symbol if represented and true, then insert symbol even if it is not registered in the metric
- * @param {string} symbol symbol to insert
- * @param {number} weight initial weight (this weight is multiplied by the metric weight)
- * @param {string} options list of optional options attached to a symbol inserted
- @example
- local function cb(task)
- if task:get_header('Some header') then
- task:insert_result('SOME_HEADER', 1.0, 'Got some header')
- end
- end
- */
- LUA_FUNCTION_DEF (task, insert_result);
-
- /***
- * @method task:adjust_result(symbol, score[, option1, ...])
- * Alters the existing symbol's score to a new score. It is not affected by
- * metric score or grow factor. You can also add new options
- * using this method. Symbol must be already inserted into metric or an error
- * will be emitted.
- * @param {string} symbol symbol to adjust
- * @param {number} score this value is NOT multiplied by the metric score
- * @param {string/table} options list of optional options attached to a symbol adjusted
- */
- LUA_FUNCTION_DEF (task, adjust_result);
- /***
- * @method task:set_pre_result(action, description)
- * Sets pre-result for a task. It is used in pre-filters to specify early result
- * of the task scanned. If a pre-filter sets some result, then further processing
- * may be skipped. For selecting action it is possible to use global table
- * `rspamd_actions` or a string value:
- *
- * - `reject`: reject message permanently
- * - `add header`: add spam header
- * - `rewrite subject`: rewrite subject to spam subject
- * - `greylist`: greylist message
- * - `accept` or `no action`: whitelist message
- * @param {rspamd_action or string} action a numeric or string action value
- * @param {string} description optional description
- @example
- local function cb(task)
- local gr = task:get_header('Greylist')
- if gr and gr == 'greylist' then
- task:set_pre_result(rspamd_actions['greylist'], 'Greylisting required')
- end
- end
- */
- LUA_FUNCTION_DEF (task, set_pre_result);
-
- /***
- * @method task:has_pre_result()
- * Returns true if task has some pre-result being set
- *
- * @return {boolean} true if task has some pre-result being set
- */
- LUA_FUNCTION_DEF (task, has_pre_result);
- /***
- * @method task:append_message(message)
- * Adds a message to scanning output.
- * @param {string} message
- @example
- local function cb(task)
- task:append_message('Example message')
- end
- */
- LUA_FUNCTION_DEF (task, append_message);
- /***
- * @method task:get_urls([need_emails|list_protos])
- * Get all URLs found in a message. Telephone urls and emails are not included unless explicitly asked in `list_protos`
- * @param {boolean} need_emails if `true` then reutrn also email urls, this can be a comma separated string of protocols desired or a table (e.g. `mailto` or `telephone`)
- * @return {table rspamd_url} list of all urls found
- @example
- local function phishing_cb(task)
- local urls = task:get_urls({'https', 'http'});
-
- if urls then
- for _,url in ipairs(urls) do
- if url:is_phished() then
- return true
- end
- end
- end
- return false
- end
- */
- LUA_FUNCTION_DEF (task, get_urls);
- /***
- * @method task:has_urls([need_emails])
- * Returns 'true' if a task has urls listed
- * @param {boolean} need_emails if `true` then reutrn also email urls
- * @return {boolean} true if a task has urls (urls or emails if `need_emails` is true)
- */
- LUA_FUNCTION_DEF (task, has_urls);
- /***
- * @method task:get_content()
- * Get raw content for the specified task
- * @return {text} the data contained in the task
- */
- LUA_FUNCTION_DEF (task, get_content);
-
- /***
- * @method task:get_filename()
- * Returns filename for a specific task
- * @return {string|nil} filename or nil if unknown
- */
- LUA_FUNCTION_DEF (task, get_filename);
-
- /***
- * @method task:get_rawbody()
- * Get raw body for the specified task
- * @return {text} the data contained in the task
- */
- LUA_FUNCTION_DEF (task, get_rawbody);
-
- /***
- * @method task:get_emails()
- * Get all email addresses found in a message.
- * @return {table rspamd_url} list of all email addresses found
- */
- LUA_FUNCTION_DEF (task, get_emails);
- /***
- * @method task:get_text_parts()
- * Get all text (and HTML) parts found in a message
- * @return {table rspamd_text_part} list of text parts
- */
- LUA_FUNCTION_DEF (task, get_text_parts);
- /***
- * @method task:get_parts()
- * Get all mime parts found in a message
- * @return {table rspamd_mime_part} list of mime parts
- */
- LUA_FUNCTION_DEF (task, get_parts);
- /***
- * @method task:get_meta_words([how='stem'])
- * Get meta words from task (subject and displayed names)
- * - `stem`: stemmed words (default)
- * - `norm`: normalised words (utf normalised + lowercased)
- * - `raw`: raw words in utf (if possible)
- * - `full`: list of tables, each table has the following fields:
- * - [1] - stemmed word
- * - [2] - normalised word
- * - [3] - raw word
- * - [4] - flags (table of strings)
- */
- LUA_FUNCTION_DEF (task, get_meta_words);
- /***
- * @method task:get_request_header(name)
- * Get value of a HTTP request header.
- * @param {string} name name of header to get
- * @return {rspamd_text} value of an HTTP header
- */
- LUA_FUNCTION_DEF (task, get_request_header);
- /***
- * @method task:set_request_header(name, value)
- * Set value of a HTTP request header. If value is omitted, then a header is removed
- * @param {string} name name of header to get
- * @param {rspamd_text/string} value new header's value
- */
- LUA_FUNCTION_DEF (task, set_request_header);
- /***
- * @method task:get_subject()
- * Returns task subject (either from the protocol override or from a header)
- * @return {string} value of a subject (decoded)
- */
- LUA_FUNCTION_DEF (task, get_subject);
- /***
- * @method task:get_header(name[, case_sensitive])
- * Get decoded value of a header specified with optional case_sensitive flag.
- * By default headers are searched in caseless matter.
- * @param {string} name name of header to get
- * @param {boolean} case_sensitive case sensitiveness flag to search for a header
- * @return {string} decoded value of a header
- */
- LUA_FUNCTION_DEF (task, get_header);
- /***
- * @method task:get_header_raw(name[, case_sensitive])
- * Get raw value of a header specified with optional case_sensitive flag.
- * By default headers are searched in caseless matter.
- * @param {string} name name of header to get
- * @param {boolean} case_sensitive case sensitiveness flag to search for a header
- * @return {string} raw value of a header
- */
- LUA_FUNCTION_DEF (task, get_header_raw);
- /***
- * @method task:get_header_full(name[, case_sensitive])
- * Get raw value of a header specified with optional case_sensitive flag.
- * By default headers are searched in caseless matter. This method returns more
- * information about the header as a list of tables with the following structure:
- *
- * - `name` - name of a header
- * - `value` - raw value of a header
- * - `decoded` - decoded value of a header
- * - `tab_separated` - `true` if a header and a value are separated by `tab` character
- * - `empty_separator` - `true` if there are no separator between a header and a value
- * @param {string} name name of header to get
- * @param {boolean} case_sensitive case sensitiveness flag to search for a header
- * @return {list of tables} all values of a header as specified above
- @example
- function check_header_delimiter_tab(task, header_name)
- for _,rh in ipairs(task:get_header_full(header_name)) do
- if rh['tab_separated'] then return true end
- end
- return false
- end
- */
- LUA_FUNCTION_DEF (task, get_header_full);
- /***
- * @method task:get_header_count(name[, case_sensitive])
- * Lightweight version if you need just a header's count
- * * By default headers are searched in caseless matter.
- * @param {string} name name of header to get
- * @param {boolean} case_sensitive case sensitiveness flag to search for a header
- * @return {number} number of header's occurrencies or 0 if not found
- */
- LUA_FUNCTION_DEF (task, get_header_count);
- /***
- * @method task:get_raw_headers()
- * Get all undecoded headers of a message as a string
- * @return {rspamd_text} all raw headers for a message as opaque text
- */
- LUA_FUNCTION_DEF (task, get_raw_headers);
-
- /***
- * @method task:get_received_headers()
- * Returns a list of tables of parsed received headers. A tables returned have
- * the following structure:
- *
- * - `from_hostname` - string that represents hostname provided by a peer
- * - `from_ip` - string representation of IP address as provided by a peer
- * - `real_hostname` - hostname as resolved by MTA
- * - `real_ip` - string representation of IP as resolved by PTR request of MTA
- * - `by_hostname` - MTA hostname
- * - `proto` - protocol, e.g. ESMTP or ESMTPS
- * - `timestamp` - received timestamp
- * - `for` - for value (unparsed mailbox)
- *
- * Please note that in some situations rspamd cannot parse all the fields of received headers.
- * In that case you should check all strings for validity.
- * @return {table of tables} list of received headers described above
- */
- LUA_FUNCTION_DEF (task, get_received_headers);
- /***
- * @method task:get_queue_id()
- * Returns queue ID of the message being processed.
- */
- LUA_FUNCTION_DEF (task, get_queue_id);
- /***
- * @method task:get_uid()
- * Returns ID of the task being processed.
- */
- LUA_FUNCTION_DEF (task, get_uid);
- /***
- * @method task:get_resolver()
- * Returns ready to use rspamd_resolver object suitable for making asynchronous DNS requests.
- * @return {rspamd_resolver} resolver object associated with the task's session
- * @example
- local logger = require "rspamd_logger"
-
- local function task_cb(task)
- local function dns_cb(resolver, to_resolve, results, err)
- -- task object is available due to closure
- task:inc_dns_req()
- if results then
- logger.info(string.format('<%s> [%s] resolved for symbol: %s',
- task:get_message_id(), to_resolve, 'EXAMPLE_SYMBOL'))
- task:insert_result('EXAMPLE_SYMBOL', 1)
- end
- end
- local r = task:get_resolver()
- r:resolve_a(task:get_session(), task:get_mempool(), 'example.com', dns_cb)
- end
- */
- LUA_FUNCTION_DEF (task, get_resolver);
- /***
- * @method task:inc_dns_req()
- * Increment number of DNS requests for the task. Is used just for logging purposes.
- */
- LUA_FUNCTION_DEF (task, inc_dns_req);
- /***
- * @method task:get_dns_req()
- * Get number of dns requests being sent in the task
- * @return {number} number of DNS requests
- */
- LUA_FUNCTION_DEF (task, get_dns_req);
-
- /***
- * @method task:has_recipients([type])
- * Return true if there are SMTP or MIME recipients for a task.
- * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP recipients and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP recipients and `2` or `mime` means MIME recipients only
- * @return {bool} `true` if there are recipients of the following type
- */
- LUA_FUNCTION_DEF (task, has_recipients);
-
- /***
- * @method task:get_recipients([type])
- * Return SMTP or MIME recipients for a task. This function returns list of internet addresses each one is a table with the following structure:
- *
- * - `name` - name of internet address in UTF8, e.g. for `Vsevolod Stakhov <blah@foo.com>` it returns `Vsevolod Stakhov`
- * - `addr` - address part of the address
- * - `user` - user part (if present) of the address, e.g. `blah`
- * - `domain` - domain part (if present), e.g. `foo.com`
- * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP recipients and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP recipients and `2` or `mime` means MIME recipients only
- * @return {list of addresses} list of recipients or `nil`
- */
- LUA_FUNCTION_DEF (task, get_recipients);
-
- /***
- * @method task:get_principal_recipient()
- * Returns a single string with so called `principal recipient` for a message. The order
- * of check is the following:
- *
- * - deliver-to request header
- * - the first recipient (envelope)
- * - the first recipient (mime)
- * @return {string} principal recipient
- */
- LUA_FUNCTION_DEF (task, get_principal_recipient);
-
- /***
- * @method task:set_recipients([type], {rcpt1, rcpt2...})
- * Sets sender for a task. This function accepts table that will be converted to the address.
- * If some fields are missing they are subsequently reconstructed by this function. E.g. if you
- * specify 'user' and 'domain', then address and raw string will be reconstructed
- *
- * - `name` - name of internet address in UTF8, e.g. for `Vsevolod Stakhov <blah@foo.com>` it returns `Vsevolod Stakhov`
- * - `addr` - address part of the address
- * - `user` - user part (if present) of the address, e.g. `blah`
- * - `domain` - domain part (if present), e.g. `foo.com`
- * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP recipients and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP recipients and `2` or `mime` means MIME recipients only
- * @param {list of tables} recipients recipients to set
- * @return {boolean} result of the operation
- */
- LUA_FUNCTION_DEF (task, set_recipients);
-
- /***
- * @method task:has_from([type])
- * Return true if there is SMTP or MIME sender for a task.
- * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP recipients and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP recipients and `2` or `mime` means MIME recipients only
- * @return {bool} `true` if there is sender of the following type
- */
- LUA_FUNCTION_DEF (task, has_from);
-
- /***
- * @method task:get_from([type])
- * Return SMTP or MIME sender for a task. This function returns an internet address which one is a table with the following structure:
- *
- * - `name` - name of internet address in UTF8, e.g. for `Vsevolod Stakhov <blah@foo.com>` it returns `Vsevolod Stakhov`
- * - `addr` - address part of the address
- * - `user` - user part (if present) of the address, e.g. `blah`
- * - `domain` - domain part (if present), e.g. `foo.com`
- * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP sender and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP sender and `2` or `mime` means MIME `From:` only
- * @return {address} sender or `nil`
- */
- LUA_FUNCTION_DEF (task, get_from);
-
- /***
- * @method task:set_from(type, addr)
- * Sets sender for a task. This function accepts table that will be converted to the address.
- * If some fields are missing they are subsequently reconstructed by this function. E.g. if you
- * specify 'user' and 'domain', then address and raw string will be reconstructed
- *
- * - `name` - name of internet address in UTF8, e.g. for `Vsevolod Stakhov <blah@foo.com>` it returns `Vsevolod Stakhov`
- * - `addr` - address part of the address
- * - `user` - user part (if present) of the address, e.g. `blah`
- * - `domain` - domain part (if present), e.g. `foo.com`
- * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP sender and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP sender and `2` or `mime` means MIME `From:` only
- * @param {table
- * @return {boolean} success or not
- */
- LUA_FUNCTION_DEF (task, set_from);
- /***
- * @method task:get_user()
- * Returns authenticated user name for this task if specified by an MTA.
- * @return {string} username or nil
- */
- LUA_FUNCTION_DEF (task, get_user);
- /***
- * @method task:set_user([username])
- * Sets or resets (if username is not specified) authenticated user name for this task.
- * @return {string} the previously set username or nil
- */
- LUA_FUNCTION_DEF (task, set_user);
- /***
- * @method task:get_from_ip()
- * Returns [ip_addr](ip.md) object of a sender that is provided by MTA
- * @return {rspamd_ip} ip address object
- */
- LUA_FUNCTION_DEF (task, get_from_ip);
- /***
- * @method task:set_from_ip(str)
- * Set tasks's IP address based on the passed string
- * @param {string} str string representation of ip
- */
- LUA_FUNCTION_DEF (task, set_from_ip);
- LUA_FUNCTION_DEF (task, get_from_ip_num);
- /***
- * @method task:get_client_ip()
- * Returns [ip_addr](ip.md) object of a client connected to rspamd (normally, it is an IP address of MTA)
- * @return {rspamd_ip} ip address object
- */
- LUA_FUNCTION_DEF (task, get_client_ip);
- /***
- * @method task:get_helo()
- * Returns the value of SMTP helo provided by MTA.
- * @return {string} HELO value
- */
- LUA_FUNCTION_DEF (task, get_helo);
- LUA_FUNCTION_DEF (task, set_helo);
- /***
- * @method task:get_hostname()
- * Returns the value of sender's hostname provided by MTA
- * @return {string} hostname value
- */
- LUA_FUNCTION_DEF (task, get_hostname);
- LUA_FUNCTION_DEF (task, set_hostname);
- /***
- * @method task:get_images()
- * Returns list of all images found in a task as a table of `rspamd_image`.
- * Each image has the following methods:
- *
- * * `get_width` - return width of an image in pixels
- * * `get_height` - return height of an image in pixels
- * * `get_type` - return string representation of image's type (e.g. 'jpeg')
- * * `get_filename` - return string with image's file name
- * * `get_size` - return size in bytes
- * @return {list of rspamd_image} images found in a message
- */
- LUA_FUNCTION_DEF (task, get_images);
- /***
- * @method task:get_archives()
- * Returns list of all archives found in a task as a table of `rspamd_archive`.
- * Each archive has the following methods available:
- *
- * * `get_files` - return list of strings with filenames inside archive
- * * `get_files_full` - return list of tables with all information about files
- * * `is_encrypted` - return true if an archive is encrypted
- * * `get_type` - return string representation of image's type (e.g. 'zip')
- * * `get_filename` - return string with archive's file name
- * * `get_size` - return size in bytes
- * @return {list of rspamd_archive} archives found in a message
- */
- LUA_FUNCTION_DEF (task, get_archives);
- /***
- * @method task:get_dkim_results()
- * Returns list of all dkim check results as table of maps. Callee must ensure that
- * dkim checks have been completed by adding dependency on `DKIM_TRACE` symbol.
- * Fields in map:
- *
- * * `result` - string result of check:
- * - `reject`
- * - `allow`
- * - `tempfail`
- * - `permfail`
- * - `not found`
- * - `bad record`
- * * `domain` - dkim domain
- * * `selector` - dkim selector
- * * `bhash` - short version of b tag (8 base64 symbols)
- * * `fail_reason` - reason of failure (if applicable)
- * @return {list of maps} dkim check results
- */
- LUA_FUNCTION_DEF (task, get_dkim_results);
- /***
- * @method task:get_symbol(name)
- * Searches for a symbol `name` in all metrics results and returns a list of tables
- * one per metric that describes the symbol inserted. Please note that this function
- * is intended to return values for **inserted** symbols, so if this symbol was not
- * inserted it won't be in the function's output. This method is useful for post-filters mainly.
- * The symbols are returned as the list of the following tables:
- *
- * - `metric` - name of metric
- * - `score` - score of a symbol in that metric
- * - `options` - a table of strings representing options of a symbol
- * - `group` - a group of symbol (or 'ungrouped')
- * @param {string} name symbol's name
- * @return {list of tables} list of tables or nil if symbol was not found in any metric
- */
- LUA_FUNCTION_DEF (task, get_symbol);
- /***
- * @method task:get_symbols_all()
- * Returns array of symbols matched in default metric with all metadata
- * @return {table} table of tables formatted as in `task:get_symbol()` except that `metric` is absent and `name` is added
- */
- LUA_FUNCTION_DEF (task, get_symbols_all);
- /***
- * @method task:get_symbols()
- * Returns array of all symbols matched for this task
- * @return {table, table} table of strings with symbols names + table of theirs scores
- */
- LUA_FUNCTION_DEF (task, get_symbols);
-
- /***
- * @method task:get_symbols_numeric()
- * Returns array of all symbols matched for this task
- * @return {table|number, table|number} table of numbers with symbols ids + table of theirs scores
- */
- LUA_FUNCTION_DEF (task, get_symbols_numeric);
-
- /***
- * @method task:get_symbols_tokens()
- * Returns array of all symbols as statistical tokens
- * @return {table|number} table of numbers
- */
- LUA_FUNCTION_DEF (task, get_symbols_tokens);
-
- /***
- * @method task:has_symbol(name)
- * Fast path to check if a specified symbol is in the task's results
- * @param {string} name symbol's name
- * @return {boolean} `true` if symbol has been found
- */
- LUA_FUNCTION_DEF (task, has_symbol);
- /***
- * @method task:has_symbol(name)
- * Fast path to check if a specified symbol is in the task's results
- * @param {string} name symbol's name
- * @return {boolean} `true` if symbol has been found
- */
- LUA_FUNCTION_DEF (task, enable_symbol);
- /***
- * @method task:has_symbol(name)
- * Fast path to check if a specified symbol is in the task's results
- * @param {string} name symbol's name
- * @return {boolean} `true` if symbol has been found
- */
- LUA_FUNCTION_DEF (task, disable_symbol);
- /***
- * @method task:get_date(type[, gmt])
- * Returns timestamp for a connection or for a MIME message. This function can be called with a
- * single table arguments with the following fields:
- *
- * * `format` - a format of date returned:
- * - `message` - returns a mime date as integer (unix timestamp)
- * - `message_str` - returns a mime date as string (UTC format)
- * - `connect` - returns a unix timestamp of a connection to rspamd
- * - `connect_str` - returns connection time in UTC format
- * * `gmt` - returns date in `GMT` timezone (normal for unix timestamps)
- *
- * By default this function returns connection time in numeric format.
- * @param {string} type date format as described above
- * @param {boolean} gmt gmt flag as described above
- * @return {string/number} date representation according to format
- * @example
- rspamd_config.DATE_IN_PAST = function(task)
- local dm = task:get_date{format = 'message', gmt = true}
- local dt = task:get_date{format = 'connect', gmt = true}
- -- A day
- if dt - dm > 86400 then
- return true
- end
-
- return false
- end
- */
- LUA_FUNCTION_DEF (task, get_date);
- /***
- * @method task:get_message_id()
- * Returns message id of the specified task
- * @return {string} if of a message
- */
- LUA_FUNCTION_DEF (task, get_message_id);
- LUA_FUNCTION_DEF (task, get_timeval);
- /***
- * @method task:get_metric_result()
- * Get full result of a metric as a table:
- * - `score`: current score
- * - `action`: current action as a string
- * - `nnegative`: number of negative rules matched
- * - `npositive`: number of positive rules matched
- * - `positive_score`: total score for positive rules
- * - `negative_score`: total score for negative rules
- * - `passthrough`: set to true if message has a passthrough result
- * @return {table} metric result
- */
- LUA_FUNCTION_DEF (task, get_metric_result);
- /***
- * @method task:get_metric_score(name)
- * Get the current score of metric `name` (must be nil or 'default') . Should be used in idempotent filters only.
- * @param {string} name name of a metric
- * @return {number,number} 2 numbers containing the current score and required score of the metric
- */
- LUA_FUNCTION_DEF (task, get_metric_score);
- /***
- * @method task:get_metric_action(name)
- * Get the current action of metric `name` (must be nil or 'default'). Should be used in idempotent filters only.
- * @param {string} name name of a metric
- * @return {string} the current action of the metric as a string
- */
- LUA_FUNCTION_DEF (task, get_metric_action);
- /***
- * @method task:set_metric_score(name, score)
- * Set the current score of metric `name`. Should be used in post-filters only.
- * @param {string} name name of a metric
- * @param {number} score the current score of the metric
- */
- LUA_FUNCTION_DEF (task, set_metric_score);
- /***
- * @method task:set_metric_subject(subject)
- * Set the subject in the default metric
- * @param {string} subject subject to set
- */
- LUA_FUNCTION_DEF (task, set_metric_subject);
-
- /***
- * @method task:learn(is_spam[, classifier)
- * Learn classifier `classifier` with the task. If `is_spam` is true then message
- * is learnt as spam. Otherwise HAM is learnt. By default, this function learns
- * `bayes` classifier.
- * @param {boolean} is_spam learn spam or ham
- * @param {string} classifier classifier's name
- * @return {boolean} `true` if classifier has been learnt successfully
- */
- LUA_FUNCTION_DEF (task, learn);
- /***
- * @method task:set_settings(obj)
- * Set users settings object for a task. The format of this object is described
- * [here](https://rspamd.com/doc/configuration/settings.html).
- * @param {any} obj any lua object that corresponds to the settings format
- */
- LUA_FUNCTION_DEF (task, set_settings);
-
- /***
- * @method task:get_settings()
- * Gets users settings object for a task. The format of this object is described
- * [here](https://rspamd.com/doc/configuration/settings.html).
- * @return {lua object} lua object generated from UCL
- */
- LUA_FUNCTION_DEF (task, get_settings);
-
- /***
- * @method task:lookup_settings(key)
- * Gets users settings object with the specified key for a task.
- * @param {string} key key to lookup
- * @return {lua object} lua object generated from UCL
- */
- LUA_FUNCTION_DEF (task, lookup_settings);
-
- /***
- * @method task:get_settings_id()
- * Get numeric hash of settings id if specified for this task. 0 is returned otherwise.
- * @return {number} settings-id hash
- */
- LUA_FUNCTION_DEF (task, get_settings_id);
-
- /***
- * @method task:set_milter_reply(obj)
- * Set special reply for milter
- * @param {any} obj any lua object that corresponds to the settings format
- * @example
- task:set_milter_reply({
- add_headers = {['X-Lua'] = 'test'},
- -- 1 is the position of header to remove
- remove_headers = {['DKIM-Signature'] = 1},
- })
- */
- LUA_FUNCTION_DEF (task, set_milter_reply);
-
- /***
- * @method task:process_re(params)
- * Processes the specified regexp and returns number of captures (cached or new)
- * Params is the table with the following fields (mandatory fields are marked with `*`):
- * - `re`* : regular expression object
- * - `type`*: type of regular expression:
- * + `mime`: mime regexp
- * + `header`: header regexp
- * + `rawheader`: raw header expression
- * + `rawmime`: raw mime regexp
- * + `body`: raw body regexp
- * + `url`: url regexp
- * - `header`: for header and rawheader regexp means the name of header
- * - `strong`: case sensitive match for headers
- * @return {number} number of regexp occurrences in the task (limited by 255 so far)
- */
- LUA_FUNCTION_DEF (task, process_regexp);
-
- /***
- * @method task:cache_set(key, value)
- * Store some value to the task cache
- * @param {string} key key to use
- * @param {any} value any value (including functions and tables)
- */
- LUA_FUNCTION_DEF (task, cache_set);
- /***
- * @method task:cache_get(key)
- * Returns cached value or nil if nothing is cached
- * @param {string} key key to use
- * @return {any} cached value
- */
- LUA_FUNCTION_DEF (task, cache_get);
-
- /***
- * @method task:get_size()
- * Returns size of the task in bytes (that includes headers + parts size)
- * @return {number} size in bytes
- */
- LUA_FUNCTION_DEF (task, get_size);
-
- /***
- * @method task:set_flag(flag_name[, set])
- * Set specific flag for task:
- *
- * - `no_log`: do not log task summary
- * - `no_stat`: do not include task into scanned stats
- * - `pass_all`: check all filters for task
- * - `extended_urls`: output extended info about urls
- * - `skip`: skip task processing
- * - `learn_spam`: learn message as spam
- * - `learn_ham`: learn message as ham
- * - `broken_headers`: header data is broken for a message
- * @param {string} flag to set
- * @param {boolean} set set or clear flag (default is set)
- @example
- --[[
- For messages with undefined queue ID (scanned with rspamc or WebUI)
- do not include results into statistics and do not log task summary
- (it will not appear in the WebUI history as well).
- ]]--
-
- -- Callback function to set flags
- local function no_log_stat_cb(task)
- if not task:get_queue_id() then
- task:set_flag('no_log')
- task:set_flag('no_stat')
- end
- end
-
- rspamd_config:register_symbol({
- name = 'LOCAL_NO_LOG_STAT',
- type = 'postfilter',
- callback = no_log_stat_cb
- })
- */
- LUA_FUNCTION_DEF (task, set_flag);
-
-
- /***
- * @method task:has_flag(flag_name)
- * Checks for a specific flag in task:
- *
- * - `no_log`: do not log task summary
- * - `no_stat`: do not include task into scanned stats
- * - `pass_all`: check all filters for task
- * - `extended_urls`: output extended info about urls
- * - `skip`: skip task processing
- * - `learn_spam`: learn message as spam
- * - `learn_ham`: learn message as ham
- * - `broken_headers`: header data is broken for a message
- * @param {string} flag to check
- * @return {boolean} true if flags is set
- */
- LUA_FUNCTION_DEF (task, has_flag);
-
-
- /***
- * @method task:get_flags()
- * Get list of flags for task:
- *
- * - `no_log`: do not log task summary
- * - `no_stat`: do not include task into scanned stats
- * - `pass_all`: check all filters for task
- * - `extended_urls`: output extended info about urls
- * - `skip`: skip task processing
- * - `learn_spam`: learn message as spam
- * - `learn_ham`: learn message as ham
- * - `broken_headers`: header data is broken for a message
- * - `milter`: task is initiated by milter connection
- * @return {array of strings} table with all flags as strings
- */
- LUA_FUNCTION_DEF (task, get_flags);
-
- /***
- * @method task:get_digest()
- * Returns message's unique digest (32 hex symbols)
- * @return {string} hex digest
- */
- LUA_FUNCTION_DEF (task, get_digest);
-
- /***
- * @method task:store_in_file([mode|table])
- * If task was loaded using file scan, then this method just returns its name,
- * otherwise, a fresh temporary file is created and its name is returned. Default
- * mode is 0600. To convert lua number to the octal mode you can use the following
- * trick: `tonumber("0644", 8)`. The file is automatically removed when task is
- * destroyed.
- *
- * If table argument is specified, the following extra fields are allowed:
- *
- * - `mode`: same as mode argument
- * - `force_new`: always create a new file
- * - `filename`: use specific filename instead of a temporary one
- * - `tmpmask`: use specific tempmask, e.g. '/tmp/file-XXXXX', where XXXX will be replaced by some random letters
- * - `keep`: do not remove file after task is processed
- *
- * @param {number} mode mode for new file
- * @return {string} file name with task content
- */
- LUA_FUNCTION_DEF (task, store_in_file);
-
- /***
- * @method task:get_protocol_reply([flags])
- * This method being called from a **postfilter** will return reply for a message
- * as it is returned to a client. This method returns the Lua table corresponding
- * to the UCL object. Flags is a table that specify which information should be
- * there in a reply:
- *
- * - `basic`: basic info, such as message-id
- * - `metrics`: metrics and symbols
- * - `messages`: messages
- * - `dkim`: dkim signature
- * - `milter`: milter control block
- * - `extra`: extra data, such as profiling
- * - `urls`: list of all urls in a message
- *
- * @param {table} flags table of flags (default is all flags but `urls`)
- * @return {table} ucl object corresponding to the reply
- */
- LUA_FUNCTION_DEF (task, get_protocol_reply);
-
- /***
- * @method task:headers_foreach(callback, [params])
- * This method calls `callback` for each header that satisfies some condition.
- * By default, all headers are iterated unless `callback` returns `true`. Nil or
- * false means continue of iterations.
- * Params could be as following:
- *
- * - `full`: header value is full table of all attributes @see task:get_header_full for details
- * - `regexp`: return headers that satisfies the specified regexp
- * @param {function} callback function from header name and header value
- * @param {table} params optional parameters
- */
- LUA_FUNCTION_DEF (task, headers_foreach);
-
- /***
- * @method task:disable_action(action)
- * Disables some action for this task (e.g. 'greylist')
- *
- * @param {string} action action to disable
- * @return {boolean} true if an action was enabled and is disabled after the method call
- */
- LUA_FUNCTION_DEF (task, disable_action);
-
- /***
- * @method task:get_newlines_type()
- * Returns the most frequent newlines type met in a task
- *
- * @return {string} "cr" for \r, "lf" for \n, "crlf" for \r\n
- */
- LUA_FUNCTION_DEF (task, get_newlines_type);
-
- /***
- * @method task:get_stat_tokens()
- * Returns list of tables the statistical tokens:
- * - `data`: 64 bit number encoded as a string
- * - `t1`: the first token (if any)
- * - `t2`: the second token (if any)
- * - `win`: window index
- * - `flag`: table of strings:
- * - `text`: text token
- * - `meta`: meta token
- * - `lua`: lua meta token
- * - `exception`: exception
- * - `subject`: subject token
- * - `unigram`: unigram token
- *
- * @return {table of tables}
- */
- LUA_FUNCTION_DEF (task, get_stat_tokens);
-
- /***
- * @method task:lookup_words(map, function({o, n, s, f}) ... end)
- * Matches words in a task (including meta words) against some map (set, regexp and so on)
- * and call the specified function with a table containing 4 values:
- * - [1] - stemmed word
- * - [2] - normalised word
- * - [3] - raw word
- * - [4] - flags (table of strings)
- */
- LUA_FUNCTION_DEF (task, lookup_words);
-
- /**
- * @method task:topointer()
- *
- * Returns raw C pointer (lightuserdata) associated with task
- */
- LUA_FUNCTION_DEF (task, topointer);
-
- static const struct luaL_reg tasklib_f[] = {
- LUA_INTERFACE_DEF (task, load_from_file),
- LUA_INTERFACE_DEF (task, load_from_string),
- {NULL, NULL}
- };
-
- static const struct luaL_reg tasklib_m[] = {
- LUA_INTERFACE_DEF (task, get_message),
- LUA_INTERFACE_DEF (task, destroy),
- LUA_INTERFACE_DEF (task, process_message),
- LUA_INTERFACE_DEF (task, set_cfg),
- LUA_INTERFACE_DEF (task, get_cfg),
- LUA_INTERFACE_DEF (task, get_mempool),
- LUA_INTERFACE_DEF (task, get_session),
- LUA_INTERFACE_DEF (task, get_ev_base),
- LUA_INTERFACE_DEF (task, get_worker),
- LUA_INTERFACE_DEF (task, insert_result),
- LUA_INTERFACE_DEF (task, adjust_result),
- LUA_INTERFACE_DEF (task, set_pre_result),
- LUA_INTERFACE_DEF (task, has_pre_result),
- LUA_INTERFACE_DEF (task, append_message),
- LUA_INTERFACE_DEF (task, has_urls),
- LUA_INTERFACE_DEF (task, get_urls),
- LUA_INTERFACE_DEF (task, get_content),
- LUA_INTERFACE_DEF (task, get_filename),
- LUA_INTERFACE_DEF (task, get_rawbody),
- LUA_INTERFACE_DEF (task, get_emails),
- LUA_INTERFACE_DEF (task, get_text_parts),
- LUA_INTERFACE_DEF (task, get_parts),
- LUA_INTERFACE_DEF (task, get_request_header),
- LUA_INTERFACE_DEF (task, set_request_header),
- LUA_INTERFACE_DEF (task, get_header),
- LUA_INTERFACE_DEF (task, get_header_raw),
- LUA_INTERFACE_DEF (task, get_header_full),
- LUA_INTERFACE_DEF (task, get_header_count),
- LUA_INTERFACE_DEF (task, get_raw_headers),
- LUA_INTERFACE_DEF (task, get_received_headers),
- LUA_INTERFACE_DEF (task, get_queue_id),
- LUA_INTERFACE_DEF (task, get_uid),
- LUA_INTERFACE_DEF (task, get_resolver),
- LUA_INTERFACE_DEF (task, inc_dns_req),
- LUA_INTERFACE_DEF (task, get_dns_req),
- LUA_INTERFACE_DEF (task, has_recipients),
- LUA_INTERFACE_DEF (task, get_recipients),
- LUA_INTERFACE_DEF (task, set_recipients),
- LUA_INTERFACE_DEF (task, get_principal_recipient),
- LUA_INTERFACE_DEF (task, has_from),
- LUA_INTERFACE_DEF (task, get_from),
- LUA_INTERFACE_DEF (task, set_from),
- LUA_INTERFACE_DEF (task, get_user),
- LUA_INTERFACE_DEF (task, set_user),
- {"get_addr", lua_task_get_from_ip},
- {"get_ip", lua_task_get_from_ip},
- {"get_from_addr", lua_task_get_from_ip},
- LUA_INTERFACE_DEF (task, get_from_ip),
- LUA_INTERFACE_DEF (task, set_from_ip),
- LUA_INTERFACE_DEF (task, get_from_ip_num),
- LUA_INTERFACE_DEF (task, get_client_ip),
- LUA_INTERFACE_DEF (task, get_subject),
- LUA_INTERFACE_DEF (task, get_helo),
- LUA_INTERFACE_DEF (task, set_helo),
- LUA_INTERFACE_DEF (task, get_hostname),
- LUA_INTERFACE_DEF (task, set_hostname),
- LUA_INTERFACE_DEF (task, get_images),
- LUA_INTERFACE_DEF (task, get_archives),
- LUA_INTERFACE_DEF (task, get_dkim_results),
- LUA_INTERFACE_DEF (task, get_symbol),
- LUA_INTERFACE_DEF (task, get_symbols),
- LUA_INTERFACE_DEF (task, get_symbols_all),
- LUA_INTERFACE_DEF (task, get_symbols_numeric),
- LUA_INTERFACE_DEF (task, get_symbols_tokens),
- LUA_INTERFACE_DEF (task, has_symbol),
- LUA_INTERFACE_DEF (task, enable_symbol),
- LUA_INTERFACE_DEF (task, disable_symbol),
- LUA_INTERFACE_DEF (task, get_date),
- LUA_INTERFACE_DEF (task, get_message_id),
- LUA_INTERFACE_DEF (task, get_timeval),
- LUA_INTERFACE_DEF (task, get_metric_result),
- LUA_INTERFACE_DEF (task, get_metric_score),
- LUA_INTERFACE_DEF (task, get_metric_action),
- LUA_INTERFACE_DEF (task, set_metric_score),
- LUA_INTERFACE_DEF (task, set_metric_subject),
- LUA_INTERFACE_DEF (task, learn),
- LUA_INTERFACE_DEF (task, set_settings),
- LUA_INTERFACE_DEF (task, get_settings),
- LUA_INTERFACE_DEF (task, lookup_settings),
- LUA_INTERFACE_DEF (task, get_settings_id),
- LUA_INTERFACE_DEF (task, cache_get),
- LUA_INTERFACE_DEF (task, cache_set),
- LUA_INTERFACE_DEF (task, process_regexp),
- LUA_INTERFACE_DEF (task, get_size),
- LUA_INTERFACE_DEF (task, set_flag),
- LUA_INTERFACE_DEF (task, get_flags),
- LUA_INTERFACE_DEF (task, has_flag),
- {"set_rmilter_reply", lua_task_set_milter_reply},
- LUA_INTERFACE_DEF (task, set_milter_reply),
- LUA_INTERFACE_DEF (task, get_digest),
- LUA_INTERFACE_DEF (task, store_in_file),
- LUA_INTERFACE_DEF (task, get_protocol_reply),
- LUA_INTERFACE_DEF (task, headers_foreach),
- LUA_INTERFACE_DEF (task, disable_action),
- LUA_INTERFACE_DEF (task, get_newlines_type),
- LUA_INTERFACE_DEF (task, get_stat_tokens),
- LUA_INTERFACE_DEF (task, get_meta_words),
- LUA_INTERFACE_DEF (task, lookup_words),
- LUA_INTERFACE_DEF (task, topointer),
- {"__tostring", rspamd_lua_class_tostring},
- {NULL, NULL}
- };
-
- /* Image methods */
- LUA_FUNCTION_DEF (image, get_width);
- LUA_FUNCTION_DEF (image, get_height);
- LUA_FUNCTION_DEF (image, get_type);
- LUA_FUNCTION_DEF (image, get_filename);
- LUA_FUNCTION_DEF (image, get_size);
-
- static const struct luaL_reg imagelib_m[] = {
- LUA_INTERFACE_DEF (image, get_width),
- LUA_INTERFACE_DEF (image, get_height),
- LUA_INTERFACE_DEF (image, get_type),
- LUA_INTERFACE_DEF (image, get_filename),
- LUA_INTERFACE_DEF (image, get_size),
- {"__tostring", rspamd_lua_class_tostring},
- {NULL, NULL}
- };
-
- /* Archive methods */
- LUA_FUNCTION_DEF (archive, get_type);
- LUA_FUNCTION_DEF (archive, get_files);
- LUA_FUNCTION_DEF (archive, get_files_full);
- LUA_FUNCTION_DEF (archive, is_encrypted);
- LUA_FUNCTION_DEF (archive, get_filename);
- LUA_FUNCTION_DEF (archive, get_size);
-
- static const struct luaL_reg archivelib_m[] = {
- LUA_INTERFACE_DEF (archive, get_type),
- LUA_INTERFACE_DEF (archive, get_files),
- LUA_INTERFACE_DEF (archive, get_files_full),
- LUA_INTERFACE_DEF (archive, is_encrypted),
- LUA_INTERFACE_DEF (archive, get_filename),
- LUA_INTERFACE_DEF (archive, get_size),
- {"__tostring", rspamd_lua_class_tostring},
- {NULL, NULL}
- };
-
- /* Blob methods */
- LUA_FUNCTION_DEF (text, fromstring);
- LUA_FUNCTION_DEF (text, fromtable);
- LUA_FUNCTION_DEF (text, len);
- LUA_FUNCTION_DEF (text, str);
- LUA_FUNCTION_DEF (text, ptr);
- LUA_FUNCTION_DEF (text, save_in_file);
- LUA_FUNCTION_DEF (text, take_ownership);
- LUA_FUNCTION_DEF (text, gc);
-
- static const struct luaL_reg textlib_f[] = {
- LUA_INTERFACE_DEF (text, fromstring),
- LUA_INTERFACE_DEF (text, fromtable),
- {NULL, NULL}
- };
-
- static const struct luaL_reg textlib_m[] = {
- LUA_INTERFACE_DEF (text, len),
- LUA_INTERFACE_DEF (text, str),
- LUA_INTERFACE_DEF (text, ptr),
- LUA_INTERFACE_DEF (text, take_ownership),
- LUA_INTERFACE_DEF (text, save_in_file),
- {"write", lua_text_save_in_file},
- {"__len", lua_text_len},
- {"__tostring", lua_text_str},
- {"__gc", lua_text_gc},
- {NULL, NULL}
- };
-
- /* Utility functions */
- struct rspamd_task *
- lua_check_task (lua_State * L, gint pos)
- {
- void *ud = rspamd_lua_check_udata (L, pos, "rspamd{task}");
- luaL_argcheck (L, ud != NULL, pos, "'task' expected");
- return ud ? *((struct rspamd_task **)ud) : NULL;
- }
-
- struct rspamd_task *
- lua_check_task_maybe (lua_State * L, gint pos)
- {
- void *ud = rspamd_lua_check_udata_maybe (L, pos, "rspamd{task}");
-
- return ud ? *((struct rspamd_task **)ud) : NULL;
- }
-
- static struct rspamd_image *
- lua_check_image (lua_State * L)
- {
- void *ud = rspamd_lua_check_udata (L, 1, "rspamd{image}");
- luaL_argcheck (L, ud != NULL, 1, "'image' expected");
- return ud ? *((struct rspamd_image **)ud) : NULL;
- }
-
- static struct rspamd_archive *
- lua_check_archive (lua_State * L)
- {
- void *ud = rspamd_lua_check_udata (L, 1, "rspamd{archive}");
- luaL_argcheck (L, ud != NULL, 1, "'archive' expected");
- return ud ? *((struct rspamd_archive **)ud) : NULL;
- }
-
- struct rspamd_lua_text *
- lua_check_text (lua_State * L, gint pos)
- {
- void *ud = rspamd_lua_check_udata (L, pos, "rspamd{text}");
- luaL_argcheck (L, ud != NULL, pos, "'text' expected");
- return ud ? (struct rspamd_lua_text *)ud : NULL;
- }
-
- static void
- lua_task_set_cached (lua_State *L, struct rspamd_task *task, const gchar *key,
- gint pos, guint id)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_cached_entry *entry;
-
- lua_pushvalue (L, pos);
-
- entry = g_hash_table_lookup (task->lua_cache, key);
-
- if (G_UNLIKELY (entry != NULL)) {
- /* Unref previous value */
- luaL_unref (L, LUA_REGISTRYINDEX, entry->ref);
- }
- else {
- entry = rspamd_mempool_alloc (task->task_pool, sizeof (*entry));
- g_hash_table_insert (task->lua_cache,
- rspamd_mempool_strdup (task->task_pool, key), entry);
- }
-
- entry->ref = luaL_ref (L, LUA_REGISTRYINDEX);
- entry->id = id;
- }
-
-
- static gboolean
- lua_task_get_cached (lua_State *L, struct rspamd_task *task, const gchar *key,
- guint id)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_cached_entry *entry;
-
- entry = g_hash_table_lookup (task->lua_cache, key);
-
- if (entry != NULL && entry->id == id) {
- lua_rawgeti (L, LUA_REGISTRYINDEX, entry->ref);
-
- return TRUE;
- }
-
- return FALSE;
- }
-
- /* Task methods */
- static int
- lua_task_process_message (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- if (task->msg.len > 0) {
- if (rspamd_message_parse (task)) {
- lua_pushboolean (L, TRUE);
- rspamd_message_process (task);
- }
- else {
- lua_pushboolean (L, FALSE);
- }
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static int
- lua_task_get_cfg (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_config **pcfg;
-
- if (task) {
- pcfg = lua_newuserdata (L, sizeof (gpointer));
- rspamd_lua_setclass (L, "rspamd{config}", -1);
- *pcfg = task->cfg;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static int
- lua_task_set_cfg (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- void *ud = rspamd_lua_check_udata (L, 2, "rspamd{config}");
-
- if (task) {
- luaL_argcheck (L, ud != NULL, 1, "'config' expected");
- task->cfg = ud ? *((struct rspamd_config **)ud) : NULL;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static int
- lua_task_destroy (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- rspamd_task_free (task);
- }
-
- return 0;
- }
-
- static int
- lua_task_get_message (lua_State * L)
- {
- LUA_TRACE_POINT;
- return luaL_error (L, "task:get_message is no longer supported");
- }
-
- static void
- lua_task_unmap_dtor (gpointer p)
- {
- struct rspamd_task *task = (struct rspamd_task *)p;
-
- if (task->msg.begin) {
- munmap ((gpointer)task->msg.begin, task->msg.len);
- }
- }
-
- static void
- lua_task_free_dtor (gpointer p)
- {
- struct rspamd_task *task = (struct rspamd_task *)p;
-
- if (task->msg.begin) {
- g_free ((gpointer)task->msg.begin);
- }
- }
-
- static gint
- lua_task_load_from_file (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = NULL, **ptask;
- const gchar *fname = luaL_checkstring (L, 1), *err = NULL;
- struct rspamd_config *cfg = NULL;
- gboolean res = FALSE;
- gpointer map;
- gsize sz;
-
- if (fname) {
-
- if (lua_type (L, 2) == LUA_TUSERDATA) {
- gpointer p;
- p = rspamd_lua_check_udata_maybe (L, 2, "rspamd{config}");
-
- if (p) {
- cfg = *(struct rspamd_config **)p;
- }
- }
-
- if (strcmp (fname, "-") == 0) {
- /* Read from stdin */
- gint fd = STDIN_FILENO;
- GString *data = g_string_sized_new (BUFSIZ);
- gchar buf[BUFSIZ];
- gssize r;
-
- for (;;) {
- r = read (fd, buf, sizeof (buf));
-
- if (r == -1) {
- err = strerror (errno);
- break;
- }
- else if (r == 0) {
- break;
- }
- else {
- g_string_append_len (data, buf, r);
- }
- }
-
- task = rspamd_task_new (NULL, cfg, NULL, NULL, NULL);
- task->msg.begin = data->str;
- task->msg.len = data->len;
- rspamd_mempool_add_destructor (task->task_pool,
- lua_task_free_dtor, task);
- res = TRUE;
- g_string_free (data, FALSE); /* Buffer is still valid */
- }
- else {
- map = rspamd_file_xmap (fname, PROT_READ, &sz, TRUE);
-
- if (!map) {
- err = strerror (errno);
- } else {
- task = rspamd_task_new (NULL, cfg, NULL, NULL, NULL);
- task->msg.begin = map;
- task->msg.len = sz;
- rspamd_mempool_add_destructor (task->task_pool,
- lua_task_unmap_dtor, task);
- res = TRUE;
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushboolean (L, res);
-
- if (res) {
- ptask = lua_newuserdata (L, sizeof (*ptask));
- *ptask = task;
- rspamd_lua_setclass (L, "rspamd{task}", -1);
- }
- else {
- if (err) {
- lua_pushstring (L, err);
- }
- else {
- lua_pushnil (L);
- }
- }
-
- return 2;
- }
-
- static gint
- lua_task_load_from_string (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = NULL, **ptask;
- const gchar *str_message;
- gsize message_len;
- struct rspamd_config *cfg = NULL;
-
- str_message = luaL_checklstring (L, 1, &message_len);
-
- if (str_message) {
-
- if (lua_type (L, 2) == LUA_TUSERDATA) {
- gpointer p;
- p = rspamd_lua_check_udata_maybe (L, 2, "rspamd{config}");
-
- if (p) {
- cfg = *(struct rspamd_config **)p;
- }
- }
-
- task = rspamd_task_new (NULL, cfg, NULL, NULL, NULL);
- task->msg.begin = g_strdup (str_message);
- task->msg.len = message_len;
- rspamd_mempool_add_destructor (task->task_pool, lua_task_free_dtor, task);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushboolean (L, true);
-
- ptask = lua_newuserdata (L, sizeof (*ptask));
- *ptask = task;
- rspamd_lua_setclass (L, "rspamd{task}", -1);
-
- return 2;
- }
-
- static int
- lua_task_get_mempool (lua_State * L)
- {
- LUA_TRACE_POINT;
- rspamd_mempool_t **ppool;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- ppool = lua_newuserdata (L, sizeof (rspamd_mempool_t *));
- rspamd_lua_setclass (L, "rspamd{mempool}", -1);
- *ppool = task->task_pool;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static int
- lua_task_get_session (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_async_session **psession;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- psession = lua_newuserdata (L, sizeof (void *));
- rspamd_lua_setclass (L, "rspamd{session}", -1);
- *psession = task->s;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
- return 1;
- }
-
- static int
- lua_task_get_ev_base (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct event_base **pbase;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- pbase = lua_newuserdata (L, sizeof (struct event_base *));
- rspamd_lua_setclass (L, "rspamd{ev_base}", -1);
- *pbase = task->ev_base;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
- return 1;
- }
-
- static int
- lua_task_get_worker (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_worker **pworker;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- if (task->worker) {
- pworker = lua_newuserdata (L, sizeof (struct rspamd_worker *));
- rspamd_lua_setclass (L, "rspamd{worker}", -1);
- *pworker = task->worker;
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
- return 1;
- }
-
-
- static gint
- lua_task_insert_result (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol_name, *param;
- double weight;
- struct rspamd_symbol_result *s;
- enum rspamd_symbol_insert_flags flags = RSPAMD_SYMBOL_INSERT_DEFAULT;
- gint i, top, args_start;
-
- if (task != NULL) {
- if (lua_isboolean (L, 2)) {
- args_start = 3;
-
- if (lua_toboolean (L, 2)) {
- flags |= RSPAMD_SYMBOL_INSERT_ENFORCE;
- }
- }
- else {
- args_start = 2;
- }
-
- symbol_name = rspamd_mempool_strdup (task->task_pool,
- luaL_checkstring (L, args_start));
- weight = luaL_checknumber (L, args_start + 1);
- top = lua_gettop (L);
- s = rspamd_task_insert_result_full (task, symbol_name, weight, NULL, flags);
-
- /* Get additional options */
- if (s) {
- for (i = args_start + 2; i <= top; i++) {
- if (lua_type (L, i) == LUA_TSTRING) {
- param = luaL_checkstring (L, i);
- rspamd_task_add_result_option (task, s, param);
- }
- else if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
- lua_pushnil (L);
-
- while (lua_next (L, -2)) {
- param = lua_tostring (L, -1);
- rspamd_task_add_result_option (task, s, param);
- lua_pop (L, 1);
- }
-
- lua_pop (L, 1);
- }
- }
- }
-
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_adjust_result (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol_name, *param;
- struct rspamd_metric_result *metric_res;
- struct rspamd_symbol_result *s = NULL;
- double weight;
- gint i, top;
-
- if (task != NULL) {
-
- symbol_name = luaL_checkstring (L, 2);
- weight = luaL_checknumber (L, 3);
- top = lua_gettop (L);
- metric_res = task->result;
-
- if (metric_res) {
- s = rspamd_task_find_symbol_result (task, symbol_name);
- }
- else {
- return luaL_error (L, "no metric result");
- }
-
- if (s) {
- metric_res->score -= s->score;
- s->score = weight;
- metric_res->score += s->score;
- }
- else {
- return luaL_error (L, "symbol not found: %s", symbol_name);
- }
-
- /* Get additional options */
- if (s) {
- for (i = 4; i <= top; i++) {
- if (lua_type (L, i) == LUA_TSTRING) {
- param = luaL_checkstring (L, i);
- rspamd_task_add_result_option (task, s, param);
- }
- else if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
- lua_pushnil (L);
-
- while (lua_next (L, -2)) {
- param = lua_tostring (L, -1);
- rspamd_task_add_result_option (task, s, param);
- lua_pop (L, 1);
- }
-
- lua_pop (L, 1);
- }
- }
- }
-
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_set_pre_result (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *message = NULL, *module = NULL;
- gdouble score = NAN;
- struct rspamd_action *action;
- guint priority = RSPAMD_PASSTHROUGH_NORMAL;
-
- if (task != NULL) {
-
- if (RSPAMD_TASK_IS_SKIPPED (task)) {
- /* Do not set pre-result for a skipped task */
- return 0;
- }
-
- if (lua_type (L, 2) == LUA_TSTRING) {
- const gchar *act_name = lua_tostring (L, 2);
- gint internal_type;
-
- if (strcmp (act_name, "accept") == 0) {
- /* Compatibility! */
- act_name = "no action";
- }
- else if (rspamd_action_from_str (act_name, &internal_type)) {
- /* Compatibility! */
- act_name = rspamd_action_to_str (internal_type);
- }
-
- action = rspamd_config_get_action (task->cfg, act_name);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- if (action == NULL) {
- struct rspamd_action *tmp;
-
- HASH_ITER (hh, task->cfg->actions, action, tmp) {
- msg_err_task ("known defined action: %s = %f",
- action->name, action->threshold);
- }
-
- return luaL_error (L, "unknown action %s", lua_tostring (L, 2));
- }
-
- if (lua_type (L, 3) == LUA_TSTRING) {
- message = lua_tostring (L, 3);
-
- /* Keep compatibility here :( */
- ucl_object_replace_key (task->messages,
- ucl_object_fromstring_common (message, 0, UCL_STRING_RAW),
- "smtp_message", 0,
- false);
- }
- else {
- message = "unknown reason";
- }
-
- if (lua_type (L, 4) == LUA_TSTRING) {
- module = lua_tostring (L, 4);
- }
- else {
- module = "Unknown lua";
- }
-
- if (lua_type (L, 5) == LUA_TNUMBER) {
- score = lua_tonumber (L, 5);
- }
-
- if (lua_type (L, 6) == LUA_TNUMBER) {
- priority = lua_tonumber (L, 6);
- }
-
-
- rspamd_add_passthrough_result (task, action, priority,
- score, rspamd_mempool_strdup (task->task_pool, message),
- rspamd_mempool_strdup (task->task_pool, module));
-
- /* Don't classify or filter message if pre-filter sets results */
- task->processed_stages |= (RSPAMD_TASK_STAGE_CLASSIFIERS |
- RSPAMD_TASK_STAGE_CLASSIFIERS_PRE |
- RSPAMD_TASK_STAGE_CLASSIFIERS_POST);
- rspamd_symcache_disable_all_symbols (task, task->cfg->cache,
- SYMBOL_TYPE_IDEMPOTENT|SYMBOL_TYPE_IGNORE_PASSTHROUGH|
- SYMBOL_TYPE_POSTFILTER);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_has_pre_result (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- lua_pushboolean (L, task->result->passthrough_result != NULL);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_append_message (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *message = luaL_checkstring (L, 2), *category;
-
- if (task != NULL) {
- if (lua_type (L, 3) == LUA_TSTRING) {
- category = luaL_checkstring (L, 3);
- }
- else {
- category = "unknown";
- }
-
- ucl_object_insert_key (task->messages,
- ucl_object_fromstring_common (message, 0, UCL_STRING_RAW),
- category, 0,
- true);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- struct lua_tree_cb_data {
- lua_State *L;
- int i;
- gint mask;
- };
-
- static void
- lua_tree_url_callback (gpointer key, gpointer value, gpointer ud)
- {
- struct rspamd_lua_url *lua_url;
- struct rspamd_url *url = (struct rspamd_url *)value;
- struct lua_tree_cb_data *cb = ud;
-
- if (url->protocol & cb->mask) {
- lua_url = lua_newuserdata (cb->L, sizeof (struct rspamd_lua_url));
- rspamd_lua_setclass (cb->L, "rspamd{url}", -1);
- lua_url->url = url;
- lua_rawseti (cb->L, -2, cb->i++);
- }
- }
-
- static gint
- lua_task_get_urls (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct lua_tree_cb_data cb;
- gint protocols_mask = 0;
- static const gint default_mask = PROTOCOL_HTTP|PROTOCOL_HTTPS|
- PROTOCOL_FILE|PROTOCOL_FTP;
- gsize sz;
-
- if (task) {
- if (lua_gettop (L) >= 2) {
- if (lua_type (L, 2) == LUA_TBOOLEAN) {
- protocols_mask = default_mask;
- if (lua_toboolean (L, 2)) {
- protocols_mask |= PROTOCOL_MAILTO;
- }
- }
- else if (lua_type (L, 2) == LUA_TTABLE) {
- for (lua_pushnil (L); lua_next (L, 2); lua_pop (L, 1)) {
- int nmask;
- const gchar *pname = lua_tostring (L, -1);
-
- nmask = rspamd_url_protocol_from_string (pname);
-
- if (nmask != PROTOCOL_UNKNOWN) {
- protocols_mask |= nmask;
- }
- else {
- msg_info ("bad url protocol: %s", pname);
- }
- }
- }
- else if (lua_type (L, 2) == LUA_TSTRING) {
- const gchar *plist = lua_tostring (L, 2);
- gchar **strvec;
- gchar * const *cvec;
-
- strvec = g_strsplit_set (plist, ",;", -1);
- cvec = strvec;
-
- while (*cvec) {
- int nmask;
-
- nmask = rspamd_url_protocol_from_string (*cvec);
-
- if (nmask != PROTOCOL_UNKNOWN) {
- protocols_mask |= nmask;
- }
- else {
- msg_info ("bad url protocol: %s", *cvec);
- }
-
- cvec ++;
- }
-
- g_strfreev (strvec);
- }
- }
- else {
- protocols_mask = default_mask;
- }
-
- cb.i = 1;
- cb.L = L;
- cb.mask = protocols_mask;
-
- if (protocols_mask & PROTOCOL_MAILTO) {
- sz = g_hash_table_size (task->urls) + g_hash_table_size (task->emails);
-
- if (protocols_mask == (default_mask|PROTOCOL_MAILTO)) {
- /* Can use cached version */
- if (!lua_task_get_cached (L, task, "emails+urls", sz)) {
- lua_createtable (L, sz, 0);
- g_hash_table_foreach (task->urls, lua_tree_url_callback, &cb);
- g_hash_table_foreach (task->emails, lua_tree_url_callback, &cb);
-
- lua_task_set_cached (L, task, "emails+urls", -1, sz);
- }
- }
- else {
- lua_createtable (L, sz, 0);
- g_hash_table_foreach (task->urls, lua_tree_url_callback, &cb);
- g_hash_table_foreach (task->emails, lua_tree_url_callback, &cb);
- }
-
- }
- else {
- sz = g_hash_table_size (task->urls);
-
- if (protocols_mask == (default_mask)) {
- if (!lua_task_get_cached (L, task, "urls", sz)) {
- lua_createtable (L, sz, 0);
- g_hash_table_foreach (task->urls, lua_tree_url_callback, &cb);
- lua_task_set_cached (L, task, "urls", -1, sz);
- }
- }
- else {
- lua_createtable (L, sz, 0);
- g_hash_table_foreach (task->urls, lua_tree_url_callback, &cb);
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_has_urls (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gboolean need_emails = FALSE, ret = FALSE;
-
- if (task) {
- if (lua_gettop (L) >= 2) {
- need_emails = lua_toboolean (L, 2);
- }
-
- if (g_hash_table_size (task->urls) > 0) {
- ret = TRUE;
- }
-
- if (need_emails && g_hash_table_size (task->emails) > 0) {
- ret = TRUE;
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushboolean (L, ret);
-
- return 1;
- }
-
- static gint
- lua_task_get_content (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_lua_text *t;
-
- if (task) {
- t = lua_newuserdata (L, sizeof (*t));
- rspamd_lua_setclass (L, "rspamd{text}", -1);
- t->len = task->msg.len;
- t->start = task->msg.begin;
- t->flags = 0;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_filename (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- if (task->msg.fpath) {
- lua_pushstring (L, task->msg.fpath);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_rawbody (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_lua_text *t;
-
- if (task) {
- t = lua_newuserdata (L, sizeof (*t));
- rspamd_lua_setclass (L, "rspamd{text}", -1);
-
- if (task->raw_headers_content.len > 0) {
- g_assert (task->raw_headers_content.len <= task->msg.len);
- t->start = task->msg.begin + task->raw_headers_content.len;
- t->len = task->msg.len - task->raw_headers_content.len;
- }
- else {
- t->len = task->msg.len;
- t->start = task->msg.begin;
- }
-
- t->flags = 0;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_emails (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct lua_tree_cb_data cb;
-
- if (task) {
- lua_createtable (L, g_hash_table_size (task->emails), 0);
- cb.i = 1;
- cb.L = L;
- cb.mask = PROTOCOL_MAILTO;
- g_hash_table_foreach (task->emails, lua_tree_url_callback, &cb);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_text_parts (lua_State * L)
- {
- LUA_TRACE_POINT;
- guint i;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_mime_text_part *part, **ppart;
-
- if (task != NULL) {
-
- if (!lua_task_get_cached (L, task, "text_parts", task->text_parts->len)) {
- lua_createtable (L, task->text_parts->len, 0);
-
- for (i = 0; i < task->text_parts->len; i ++) {
- part = g_ptr_array_index (task->text_parts, i);
- ppart = lua_newuserdata (L, sizeof (struct rspamd_mime_text_part *));
- *ppart = part;
- rspamd_lua_setclass (L, "rspamd{textpart}", -1);
- /* Make it array */
- lua_rawseti (L, -2, i + 1);
- }
-
- lua_task_set_cached (L, task, "text_parts", -1, task->text_parts->len);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_parts (lua_State * L)
- {
- LUA_TRACE_POINT;
- guint i;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_mime_part *part, **ppart;
-
- if (task != NULL) {
- if (!lua_task_get_cached (L, task, "mime_parts", task->parts->len)) {
- lua_createtable (L, task->parts->len, 0);
-
- for (i = 0; i < task->parts->len; i ++) {
- part = g_ptr_array_index (task->parts, i);
- ppart = lua_newuserdata (L, sizeof (struct rspamd_mime_part *));
- *ppart = part;
- rspamd_lua_setclass (L, "rspamd{mimepart}", -1);
- /* Make it array */
- lua_rawseti (L, -2, i + 1);
- }
-
- lua_task_set_cached (L, task, "mime_parts", -1, task->parts->len);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_request_header (lua_State *L)
- {
- LUA_TRACE_POINT;
- rspamd_ftok_t *hdr;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *s;
- struct rspamd_lua_text *t;
-
- s = luaL_checkstring (L, 2);
-
- if (s && task) {
- hdr = rspamd_task_get_request_header (task, s);
-
- if (hdr) {
- t = lua_newuserdata (L, sizeof (*t));
- rspamd_lua_setclass (L, "rspamd{text}", -1);
- t->start = hdr->begin;
- t->len = hdr->len;
- t->flags = 0;
-
- return 1;
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_request_header (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *s, *v = NULL;
- rspamd_fstring_t *buf;
- struct rspamd_lua_text *t;
- rspamd_ftok_t *hdr, *new_name;
- gsize len, vlen;
-
- s = luaL_checklstring (L, 2, &len);
-
- if (s && task) {
- if (lua_type (L, 3) == LUA_TSTRING) {
- v = luaL_checklstring (L, 3, &vlen);
- }
- else if (lua_type (L, 3) == LUA_TUSERDATA) {
- t = lua_check_text (L, 3);
-
- if (t != NULL) {
- v = t->start;
- vlen = t->len;
- }
- }
-
- if (v != NULL) {
- buf = rspamd_fstring_new_init (v, vlen);
- hdr = rspamd_ftok_map (buf);
- buf = rspamd_fstring_new_init (s, len);
- new_name = rspamd_ftok_map (buf);
-
- rspamd_task_add_request_header (task, new_name, hdr);
- }
-
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
-
- return 0;
- }
-
-
-
- gint
- rspamd_lua_push_header (lua_State *L, struct rspamd_mime_header *rh,
- enum rspamd_lua_task_header_type how)
- {
- LUA_TRACE_POINT;
- switch (how) {
- case RSPAMD_TASK_HEADER_PUSH_FULL:
- /* Create new associated table for a header */
- lua_createtable (L, 0, 7);
- rspamd_lua_table_set (L, "name", rh->name);
-
- if (rh->value) {
- rspamd_lua_table_set (L, "value", rh->value);
- }
-
- if (rh->raw_len > 0) {
- lua_pushstring (L, "raw");
- lua_pushlstring (L, rh->raw_value, rh->raw_len);
- lua_settable (L, -3);
- }
-
- if (rh->decoded) {
- rspamd_lua_table_set (L, "decoded", rh->decoded);
- }
-
- lua_pushstring (L, "tab_separated");
- lua_pushboolean (L, rh->tab_separated);
- lua_settable (L, -3);
- lua_pushstring (L, "empty_separator");
- lua_pushboolean (L, rh->empty_separator);
- lua_settable (L, -3);
- rspamd_lua_table_set (L, "separator", rh->separator);
- lua_pushstring (L, "order");
- lua_pushinteger (L, rh->order);
- lua_settable (L, -3);
- break;
- case RSPAMD_TASK_HEADER_PUSH_RAW:
- if (rh->value) {
- lua_pushstring (L, rh->value);
- }
- else {
- lua_pushnil (L);
- }
- break;
- case RSPAMD_TASK_HEADER_PUSH_SIMPLE:
- if (rh->decoded) {
- lua_pushstring (L, rh->decoded);
- }
- else {
- lua_pushnil (L);
- }
- break;
- case RSPAMD_TASK_HEADER_PUSH_COUNT:
- default:
- g_assert_not_reached ();
- break;
- }
-
- return 1;
- }
-
- gint
- rspamd_lua_push_header_array (lua_State * L,
- GPtrArray *ar,
- enum rspamd_lua_task_header_type how)
- {
- LUA_TRACE_POINT;
- struct rspamd_mime_header *rh;
- guint i;
-
- if (ar == NULL || ar->len == 0) {
- if (how == RSPAMD_TASK_HEADER_PUSH_COUNT) {
- lua_pushnumber (L, 0);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- if (how == RSPAMD_TASK_HEADER_PUSH_FULL) {
- lua_createtable (L, ar->len, 0);
- PTR_ARRAY_FOREACH (ar, i, rh) {
- rspamd_lua_push_header (L, rh, how);
- lua_rawseti (L, -2, i + 1);
- }
- }
- else if (how == RSPAMD_TASK_HEADER_PUSH_COUNT) {
- lua_pushinteger (L, ar->len);
- }
- else {
- rh = g_ptr_array_index (ar, 0);
-
- return rspamd_lua_push_header (L, rh, how);
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_header_common (lua_State *L, enum rspamd_lua_task_header_type how)
- {
- LUA_TRACE_POINT;
- gboolean strong = FALSE;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *name;
- GPtrArray *ar;
-
- name = luaL_checkstring (L, 2);
-
- if (name && task) {
- if (lua_gettop (L) == 3) {
- strong = lua_toboolean (L, 3);
- }
-
- ar = rspamd_message_get_header_array (task, name, strong);
-
- return rspamd_lua_push_header_array (L, ar, how);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
- }
-
- static gint
- lua_task_get_header_full (lua_State * L)
- {
- return lua_task_get_header_common (L, RSPAMD_TASK_HEADER_PUSH_FULL);
- }
-
- static gint
- lua_task_get_header (lua_State * L)
- {
- return lua_task_get_header_common (L, RSPAMD_TASK_HEADER_PUSH_SIMPLE);
- }
-
- static gint
- lua_task_get_header_raw (lua_State * L)
- {
- return lua_task_get_header_common (L, RSPAMD_TASK_HEADER_PUSH_RAW);
- }
-
- static gint
- lua_task_get_header_count (lua_State * L)
- {
- return lua_task_get_header_common (L, RSPAMD_TASK_HEADER_PUSH_COUNT);
- }
-
- static gint
- lua_task_get_raw_headers (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_lua_text *t;
-
- if (task) {
- t = lua_newuserdata (L, sizeof (*t));
- rspamd_lua_setclass (L, "rspamd{text}", -1);
- t->start = task->raw_headers_content.begin;
- t->len = task->raw_headers_content.len;
- t->flags = 0;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
-
- return 1;
- }
-
- static gint
- lua_task_get_received_headers (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct received_header *rh;
- const gchar *proto;
- guint i, k = 1;
-
- if (task) {
- if (!lua_task_get_cached (L, task, "received", task->received->len)) {
- lua_createtable (L, task->received->len, 0);
-
- for (i = 0; i < task->received->len; i ++) {
- rh = g_ptr_array_index (task->received, i);
-
- lua_createtable (L, 0, 10);
-
- if (rh->hdr && rh->hdr->decoded) {
- rspamd_lua_table_set (L, "raw", rh->hdr->decoded);
- }
-
- lua_pushstring (L, "flags");
- lua_createtable (L, 0, 3);
-
- lua_pushstring (L, "artificial");
- if (rh->flags & RSPAMD_RECEIVED_FLAG_ARTIFICIAL) {
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- lua_settable (L, -3);
-
- lua_pushstring (L, "authenticated");
- if (rh->flags & RSPAMD_RECEIVED_FLAG_AUTHENTICATED) {
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- lua_settable (L, -3);
-
- lua_pushstring (L, "ssl");
- if (rh->flags & RSPAMD_RECEIVED_FLAG_SSL) {
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- lua_settable (L, -3);
-
- lua_settable (L, -3);
-
- if (G_UNLIKELY (rh->from_ip == NULL &&
- rh->real_ip == NULL &&
- rh->real_hostname == NULL &&
- rh->by_hostname == NULL && rh->timestamp == 0 &&
- rh->for_mbox == NULL)) {
- lua_rawseti (L, -2, k ++);
-
- continue;
- }
-
- rspamd_lua_table_set (L, "from_hostname", rh->from_hostname);
- rspamd_lua_table_set (L, "from_ip", rh->from_ip);
- rspamd_lua_table_set (L, "real_hostname", rh->real_hostname);
- lua_pushstring (L, "real_ip");
- rspamd_lua_ip_push (L, rh->addr);
- lua_settable (L, -3);
- lua_pushstring (L, "proto");
-
- switch (rh->type) {
- case RSPAMD_RECEIVED_SMTP:
- proto = "smtp";
- break;
- case RSPAMD_RECEIVED_ESMTP:
- proto = "esmtp";
- break;
- case RSPAMD_RECEIVED_ESMTPS:
- proto = "esmtps";
- break;
- case RSPAMD_RECEIVED_ESMTPA:
- proto = "esmtpa";
- break;
- case RSPAMD_RECEIVED_ESMTPSA:
- proto = "esmtpsa";
- break;
- case RSPAMD_RECEIVED_LMTP:
- proto = "lmtp";
- break;
- case RSPAMD_RECEIVED_IMAP:
- proto = "imap";
- break;
- case RSPAMD_RECEIVED_HTTP:
- proto = "http";
- break;
- case RSPAMD_RECEIVED_LOCAL:
- proto = "local";
- break;
- case RSPAMD_RECEIVED_MAPI:
- proto = "mapi";
- break;
- case RSPAMD_RECEIVED_UNKNOWN:
- default:
- proto = "unknown";
- break;
- }
-
- lua_pushstring (L, proto);
- lua_settable (L, -3);
-
- lua_pushstring (L, "timestamp");
- lua_pushinteger (L, rh->timestamp);
- lua_settable (L, -3);
-
- rspamd_lua_table_set (L, "by_hostname", rh->by_hostname);
- rspamd_lua_table_set (L, "for", rh->for_mbox);
- lua_rawseti (L, -2, k ++);
- }
-
- lua_task_set_cached (L, task, "received", -1, task->received->len);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_queue_id (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- if (task->queue_id != NULL && strcmp (task->queue_id, "undef") != 0) {
- lua_pushstring (L, task->queue_id);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_uid (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- lua_pushstring (L, task->task_pool->tag.uid);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_resolver (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_dns_resolver **presolver;
-
- if (task != NULL && task->resolver != NULL) {
- presolver = lua_newuserdata (L, sizeof (void *));
- rspamd_lua_setclass (L, "rspamd{resolver}", -1);
- *presolver = task->resolver;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_inc_dns_req (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- static guint warning_shown = 0;
-
- if (warning_shown < 100) {
- warning_shown ++;
- msg_warn_task_check ("task:inc_dns_req is deprecated and should not be used");
- }
-
- if (task != NULL) {
- /* Deprecation: already done in make_dns_request */
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_get_dns_req (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- lua_pushinteger (L, task->dns_requests);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- enum rspamd_address_type {
- RSPAMD_ADDRESS_ANY = 0,
- RSPAMD_ADDRESS_SMTP = 1,
- RSPAMD_ADDRESS_MIME = 2,
- RSPAMD_ADDRESS_RAW_ANY = 3,
- RSPAMD_ADDRESS_RAW_SMTP = 4,
- RSPAMD_ADDRESS_RAW_MIME = 5,
- RSPAMD_ADDRESS_MAX
- };
-
- /*
- * Convert element at the specified position to the type
- * for get_from/get_recipients
- */
- static enum rspamd_address_type
- lua_task_str_to_get_type (lua_State *L, gint pos)
- {
- const gchar *type = NULL;
- gint ret = RSPAMD_ADDRESS_ANY;
- guint64 h;
- gsize sz;
-
- /* Get what value */
-
- if (lua_type (L, pos) == LUA_TNUMBER) {
- ret = lua_tonumber (L, pos);
-
- if (ret >= RSPAMD_ADDRESS_ANY && ret < RSPAMD_ADDRESS_MAX) {
- return ret;
- }
-
- return RSPAMD_ADDRESS_ANY;
- }
- else if (lua_type (L, pos) == LUA_TSTRING) {
- type = lua_tolstring (L, pos, &sz);
-
- if (type && sz > 0) {
- h = rspamd_cryptobox_fast_hash_specific (RSPAMD_CRYPTOBOX_XXHASH64,
- type, sz, 0xdeadbabe);
-
- switch (h) {
- case 0xDA081341FB600389ULL: /* mime */
- ret = RSPAMD_ADDRESS_MIME;
- break;
- case 0xEEC8A7832F8C43ACULL: /* any */
- ret = RSPAMD_ADDRESS_ANY;
- break;
- case 0x472274D5193B2A80ULL: /* smtp */
- case 0xEFE0F586CC9F14A9ULL: /* envelope */
- ret = RSPAMD_ADDRESS_SMTP;
- break;
- case 0x9DA887501690DE20ULL: /* raw_mime */
- ret = RSPAMD_ADDRESS_RAW_MIME;
- break;
- case 0x6B54FE02DEB595A4ULL: /* raw_smtp */
- case 0xE0E596C861777B02ULL: /* raw_envelope */
- ret = RSPAMD_ADDRESS_RAW_SMTP;
- break;
- case 0x2C49DBE3A10A0197ULL: /* raw_any */
- ret = RSPAMD_ADDRESS_RAW_ANY;
- break;
- }
- }
- }
-
- return ret;
- }
-
- #define EMAIL_CHECK_FLAG(fl, str) do { \
- if (addr->flags & (fl)) { \
- lua_pushstring (L, (str)); \
- lua_pushboolean (L, true); \
- lua_settable (L, -3); \
- } \
- } while(0)
-
- static void
- lua_push_email_address (lua_State *L, struct rspamd_email_address *addr)
- {
- if (addr) {
- lua_createtable (L, 0, 4);
-
- if (addr->addr_len > 0) {
- lua_pushstring (L, "addr");
- lua_pushlstring (L, addr->addr, addr->addr_len);
- lua_settable (L, -3);
- }
- else {
- lua_pushstring (L, "addr");
- lua_pushstring (L, "");
- lua_settable (L, -3);
- }
- if (addr->domain_len > 0) {
- lua_pushstring (L, "domain");
- lua_pushlstring (L, addr->domain, addr->domain_len);
- lua_settable (L, -3);
- }
- else {
- lua_pushstring (L, "domain");
- lua_pushstring (L, "");
- lua_settable (L, -3);
- }
- if (addr->user_len > 0) {
- lua_pushstring (L, "user");
- lua_pushlstring (L, addr->user, addr->user_len);
- lua_settable (L, -3);
- }
- else {
- lua_pushstring (L, "user");
- lua_pushstring (L, "");
- lua_settable (L, -3);
- }
-
- if (addr->name) {
- lua_pushstring (L, "name");
- lua_pushstring (L, addr->name);
- lua_settable (L, -3);
- }
- else {
- lua_pushstring (L, "name");
- lua_pushstring (L, "");
- lua_settable (L, -3);
- }
-
- lua_pushstring (L, "flags");
- lua_createtable (L, 0, 7);
-
- EMAIL_CHECK_FLAG (RSPAMD_EMAIL_ADDR_VALID, "valid");
- EMAIL_CHECK_FLAG (RSPAMD_EMAIL_ADDR_IP, "ip");
- EMAIL_CHECK_FLAG (RSPAMD_EMAIL_ADDR_BRACED, "braced");
- EMAIL_CHECK_FLAG (RSPAMD_EMAIL_ADDR_QUOTED, "quoted");
- EMAIL_CHECK_FLAG (RSPAMD_EMAIL_ADDR_EMPTY, "empty");
- EMAIL_CHECK_FLAG (RSPAMD_EMAIL_ADDR_HAS_BACKSLASH, "backslash");
- EMAIL_CHECK_FLAG (RSPAMD_EMAIL_ADDR_HAS_8BIT, "8bit");
-
- lua_settable (L, -3);
- }
- }
-
- void
- lua_push_emails_address_list (lua_State *L, GPtrArray *addrs)
- {
- struct rspamd_email_address *addr;
- guint i;
-
- lua_createtable (L, addrs->len, 0);
-
- for (i = 0; i < addrs->len; i ++) {
- addr = g_ptr_array_index (addrs, i);
- lua_push_email_address (L, addr);
- lua_rawseti (L, -2, i + 1);
- }
- }
-
- static gboolean
- lua_import_email_address (lua_State *L, struct rspamd_task *task,
- gint pos,
- struct rspamd_email_address **paddr)
- {
- struct rspamd_email_address *addr;
- const gchar *p;
- gchar *dst;
- gsize len;
-
- g_assert (paddr != NULL);
-
- if (!lua_istable (L, pos)) {
- return FALSE;
- }
-
- addr = g_malloc0 (sizeof (*addr));
-
- lua_pushstring (L, "name");
- lua_gettable (L, pos);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- p = lua_tolstring (L, -1, &len);
- dst = rspamd_mempool_alloc (task->task_pool, len + 1);
- rspamd_strlcpy (dst, p, len + 1);
- addr->name = dst;
- }
-
- lua_pop (L, 1);
-
- lua_pushstring (L, "user");
- lua_gettable (L, pos);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- p = lua_tolstring (L, -1, &len);
- addr->user = (const gchar *)rspamd_mempool_alloc (task->task_pool, len);
- memcpy ((gchar *)addr->user, p, len);
- addr->user_len = len;
- }
-
- lua_pop (L, 1);
-
- lua_pushstring (L, "domain");
- lua_gettable (L, pos);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- p = lua_tolstring (L, -1, &len);
- addr->domain = (const gchar *)rspamd_mempool_alloc (task->task_pool, len);
- memcpy ((gchar *)addr->domain, p, len);
- addr->domain_len = len;
- }
-
- lua_pop (L, 1);
-
- lua_pushstring (L, "addr");
- lua_gettable (L, pos);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- p = lua_tolstring (L, -1, &len);
- addr->addr = (const gchar *)rspamd_mempool_alloc (task->task_pool, len);
- memcpy ((gchar *)addr->addr, p, len);
- addr->addr_len = len;
- }
- else {
- /* Construct addr */
- len = addr->domain_len + addr->user_len + 1;
- addr->addr = (const gchar *)rspamd_mempool_alloc (task->task_pool, len);
- addr->addr_len = rspamd_snprintf ((gchar *)addr->addr, len, "%*s@%*s",
- (int)addr->user_len, addr->user,
- (int)addr->domain_len, addr->domain);
- }
-
- lua_pop (L, 1);
-
- lua_pushstring (L, "raw");
- lua_gettable (L, pos);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- p = lua_tolstring (L, -1, &len);
- addr->raw = (const gchar *)rspamd_mempool_alloc (task->task_pool, len);
- memcpy ((gchar *)addr->raw, p, len);
- addr->raw_len = len;
- }
- else {
- /* Construct raw addr */
- len = addr->addr_len + 3;
-
- if (addr->name) {
- len += strlen (addr->name) + 1;
- dst = rspamd_mempool_alloc (task->task_pool, len + 1);
-
- addr->raw_len = rspamd_snprintf (dst, len, "%s <%*s>",
- addr->name,
- (int)addr->addr_len, addr->addr);
-
- }
- else {
- dst = rspamd_mempool_alloc (task->task_pool, len + 1);
-
- addr->raw_len = rspamd_snprintf (dst, len, "<%*s@%*s>",
- (int)addr->user_len, addr->user,
- (int)addr->domain_len, addr->domain);
- }
-
- addr->raw = dst;
- }
-
- lua_pop (L, 1);
- addr->flags = RSPAMD_EMAIL_ADDR_VALID;
-
- *paddr = addr;
-
- return TRUE;
- }
-
- static gint
- lua_task_get_recipients (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- GPtrArray *ptrs = NULL;
- gint what = 0;
-
- if (task) {
- if (lua_gettop (L) == 2) {
- /* Get what value */
- what = lua_task_str_to_get_type (L, 2);
- }
-
- switch (what) {
- case RSPAMD_ADDRESS_SMTP:
- /* Here we check merely envelope rcpt */
- ptrs = task->rcpt_envelope;
- break;
- case RSPAMD_ADDRESS_MIME:
- /* Here we check merely mime rcpt */
- ptrs = task->rcpt_mime;
- break;
- case RSPAMD_ADDRESS_ANY:
- default:
- if (task->rcpt_envelope) {
- ptrs = task->rcpt_envelope;
- }
- else {
- ptrs = task->rcpt_mime;
- }
- break;
- }
- if (ptrs) {
- lua_push_emails_address_list (L, ptrs);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_recipients (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- GPtrArray *ptrs = NULL;
- struct rspamd_email_address *addr = NULL;
- gint what = 0, pos = 3;
-
- if (task) {
- if (lua_isstring (L, 2) || lua_isnumber (L, 2)) {
- /* Get what value */
- what = lua_task_str_to_get_type (L, 2);
- }
- else if (lua_istable (L, 2)) {
- pos = 2;
- }
-
- switch (what) {
- case RSPAMD_ADDRESS_SMTP:
- /* Here we check merely envelope rcpt */
- ptrs = task->rcpt_envelope;
- break;
- case RSPAMD_ADDRESS_MIME:
- /* Here we check merely mime rcpt */
- ptrs = task->rcpt_mime;
- break;
- case RSPAMD_ADDRESS_ANY:
- default:
- if (task->rcpt_envelope) {
- ptrs = task->rcpt_envelope;
- }
- else {
- ptrs = task->rcpt_mime;
- }
- break;
- }
- if (ptrs) {
- guint i;
- struct rspamd_email_address *tmp;
-
- PTR_ARRAY_FOREACH (ptrs, i, tmp) {
- rspamd_email_address_free (tmp);
- }
-
- g_ptr_array_set_size (ptrs, 0);
- lua_pushvalue (L, pos);
-
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- if (lua_import_email_address (L, task, lua_gettop (L), &addr)) {
- g_ptr_array_add (ptrs, addr);
- }
- }
-
- lua_pop (L, 1);
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
-
- #define CHECK_EMAIL_ADDR(addr) do { \
- if (addr == NULL) { \
- ret = 0; \
- } \
- else { \
- ret = addr->flags & RSPAMD_EMAIL_ADDR_VALID; \
- } \
- } while (0)
-
- #define CHECK_EMAIL_ADDR_LIST(addr) do { \
- if (addr == NULL) { \
- ret = 0; \
- } \
- else { \
- ret = addr->len > 0; \
- } \
- } while (0)
-
- static gint
- lua_task_has_from (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gint what = 0;
- gboolean ret = FALSE;
-
- if (task) {
- if (lua_gettop (L) == 2) {
- /* Get what value */
- what = lua_task_str_to_get_type (L, 2);
- }
-
- switch (what) {
- case RSPAMD_ADDRESS_SMTP:
- /* Here we check merely envelope rcpt */
- CHECK_EMAIL_ADDR (task->from_envelope);
- break;
- case RSPAMD_ADDRESS_MIME:
- /* Here we check merely mime rcpt */
- CHECK_EMAIL_ADDR_LIST (task->from_mime);
- break;
- case RSPAMD_ADDRESS_ANY:
- default:
- CHECK_EMAIL_ADDR (task->from_envelope);
-
- if (!ret) {
- CHECK_EMAIL_ADDR_LIST (task->from_mime);
- }
- break;
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushboolean (L, ret);
-
- return 1;
- }
-
- static gint
- lua_task_has_recipients (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gint what = 0;
- gboolean ret = FALSE;
-
- if (task) {
- if (lua_gettop (L) == 2) {
- /* Get what value */
- what = lua_task_str_to_get_type (L, 2);
- }
-
- switch (what) {
- case RSPAMD_ADDRESS_SMTP:
- /* Here we check merely envelope rcpt */
- CHECK_EMAIL_ADDR_LIST (task->rcpt_envelope);
- break;
- case RSPAMD_ADDRESS_MIME:
- /* Here we check merely mime rcpt */
- CHECK_EMAIL_ADDR_LIST (task->rcpt_mime);
- break;
- case RSPAMD_ADDRESS_ANY:
- default:
- CHECK_EMAIL_ADDR_LIST (task->rcpt_envelope);
-
- if (!ret) {
- CHECK_EMAIL_ADDR_LIST (task->rcpt_mime);
- }
- break;
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushboolean (L, ret);
-
- return 1;
- }
-
- static gint
- lua_task_get_from (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- GPtrArray *addrs = NULL;
- struct rspamd_email_address *addr = NULL;
- gint what = 0;
-
- if (task) {
- if (lua_gettop (L) == 2) {
- /* Get what value */
- what = lua_task_str_to_get_type (L, 2);
- }
-
- switch (what) {
- case RSPAMD_ADDRESS_SMTP:
- /* Here we check merely envelope rcpt */
- addr = task->from_envelope;
- break;
- case RSPAMD_ADDRESS_MIME:
- /* Here we check merely mime rcpt */
- addrs = task->from_mime;
- break;
- case RSPAMD_ADDRESS_ANY:
- default:
- if (task->from_envelope) {
- addr = task->from_envelope;
- }
- else {
- addrs = task->from_mime;
- }
- break;
- }
-
- if (addrs) {
- lua_push_emails_address_list (L, addrs);
- }
- else if (addr) {
- /* Create table to preserve compatibility */
- if (addr->addr) {
- lua_createtable (L, 1, 0);
- lua_push_email_address (L, addr);
- lua_rawseti (L, -2, 1);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_from (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- GPtrArray *addrs = NULL;
- struct rspamd_email_address **paddr = NULL, *addr;
- gint what = 0, pos = 3;
-
- if (task) {
- if (lua_isstring (L, 2) || lua_isnumber (L, 2)) {
- /* Get what value */
- what = lua_task_str_to_get_type (L, 2);
- }
- else if (lua_istable (L, 2)) {
- pos = 2;
- }
-
- switch (what) {
- case RSPAMD_ADDRESS_SMTP:
- /* Here we check merely envelope rcpt */
- paddr = &task->from_envelope;
- break;
- case RSPAMD_ADDRESS_MIME:
- /* Here we check merely mime rcpt */
- addrs = task->from_mime;
- break;
- case RSPAMD_ADDRESS_ANY:
- default:
- if (task->from_envelope) {
- paddr = &task->from_envelope;
- }
- else {
- addrs = task->from_mime;
- }
- break;
- }
-
- if (addrs) {
- if (lua_import_email_address (L, task, pos, &addr)) {
- guint i;
- struct rspamd_email_address *tmp;
-
- PTR_ARRAY_FOREACH (addrs, i, tmp) {
- rspamd_email_address_free (tmp);
- }
-
- g_ptr_array_set_size (addrs, 0);
- g_ptr_array_add (addrs, addr);
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- }
- else if (paddr) {
- if (lua_import_email_address (L, task, pos, &addr)) {
-
- if (paddr) {
- rspamd_email_address_free (*paddr);
- }
-
- *paddr = addr;
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- }
- else {
- lua_pushboolean (L, false);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_principal_recipient (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *r;
-
- if (task) {
- r = rspamd_task_get_principal_recipient (task);
- if (r != NULL) {
- lua_pushstring (L, r);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_user (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- if (task->user != NULL) {
- lua_pushstring (L, task->user);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_user (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *new_user;
-
- if (task) {
-
- if (lua_type (L, 2) == LUA_TSTRING) {
- new_user = lua_tostring (L, 2);
-
- if (task->user) {
- /* Push old user */
- lua_pushstring (L, task->user);
- }
- else {
- lua_pushnil (L);
- }
-
- task->user = rspamd_mempool_strdup (task->task_pool, new_user);
- }
- else {
- /* Reset user */
- if (task->user) {
- /* Push old user */
- lua_pushstring (L, task->user);
- }
- else {
- lua_pushnil (L);
- }
-
- task->user = NULL;
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_from_ip (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- rspamd_lua_ip_push (L, task->from_addr);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_from_ip (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *ip_str = luaL_checkstring (L, 2);
- rspamd_inet_addr_t *addr = NULL;
-
- if (!task || !ip_str) {
- lua_pushstring (L, "invalid parameters");
- return lua_error (L);
- }
- else {
- if (!rspamd_parse_inet_address (&addr,
- ip_str,
- 0)) {
- msg_warn_task ("cannot get IP from received header: '%s'",
- ip_str);
- }
- else {
- if (task->from_addr) {
- rspamd_inet_address_free (task->from_addr);
- }
-
- task->from_addr = addr;
- }
- }
-
- return 0;
- }
-
- static gint
- lua_task_get_from_ip_num (lua_State *L)
- {
- LUA_TRACE_POINT;
- msg_err ("this function is deprecated and should no longer be used");
- lua_pushnil (L);
- return 1;
- }
-
- static gint
- lua_task_get_client_ip (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- rspamd_lua_ip_push (L, task->client_addr);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_helo (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- if (task->helo != NULL) {
- lua_pushstring (L, task->helo);
- return 1;
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_subject (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- if (task->subject != NULL) {
- lua_pushstring (L, task->subject);
- return 1;
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_helo (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *new_helo;
-
- if (task) {
- new_helo = luaL_checkstring (L, 2);
- if (new_helo) {
- task->helo = rspamd_mempool_strdup (task->task_pool, new_helo);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_get_hostname (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- if (task->hostname != NULL) {
- /* Check whether it looks like an IP address */
- if (*task->hostname == '[') {
- /*
- * From the milter documentation:
- * If the reverse lookup fails or if none of the IP
- * addresses of the resolved host name matches the
- * original IP address, hostname will contain the
- * message sender's IP address enclosed in square
- * brackets (e.g. `[a.b.c.d]')
- */
- lua_pushnil (L);
- }
- else {
- lua_pushstring (L, task->hostname);
- }
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_hostname (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *new_hostname;
-
- if (task) {
- new_hostname = luaL_checkstring (L, 2);
- if (new_hostname) {
- task->hostname = rspamd_mempool_strdup (task->task_pool,
- new_hostname);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_get_images (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- guint nelt = 0, i;
- struct rspamd_mime_part *part;
- struct rspamd_image **pimg;
-
- if (task) {
- if (!lua_task_get_cached (L, task, "images", task->parts->len)) {
- lua_createtable (L, task->parts->len, 0);
-
- for (i = 0; i < task->parts->len; i ++) {
- part = g_ptr_array_index (task->parts, i);
-
- if (part->flags & RSPAMD_MIME_PART_IMAGE) {
- pimg = lua_newuserdata (L, sizeof (struct rspamd_image *));
- rspamd_lua_setclass (L, "rspamd{image}", -1);
- *pimg = part->specific.img;
- lua_rawseti (L, -2, ++nelt);
- }
- }
-
- lua_task_set_cached (L, task, "images", -1, task->parts->len);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_archives (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- guint nelt = 0, i;
- struct rspamd_mime_part *part;
- struct rspamd_archive **parch;
-
- if (task) {
- if (!lua_task_get_cached (L, task, "archives", task->parts->len)) {
- lua_createtable (L, task->parts->len, 0);
-
- for (i = 0; i < task->parts->len; i ++) {
- part = g_ptr_array_index (task->parts, i);
-
- if (part->flags & RSPAMD_MIME_PART_ARCHIVE) {
- parch = lua_newuserdata (L, sizeof (struct rspamd_archive *));
- rspamd_lua_setclass (L, "rspamd{archive}", -1);
- *parch = part->specific.arch;
- lua_rawseti (L, -2, ++nelt);
- }
- }
-
- lua_task_set_cached (L, task, "archives", -1, task->parts->len);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_dkim_results (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- guint nelt = 0, i;
- struct rspamd_dkim_check_result **pres, **cur;
-
- if (task) {
- if (!lua_task_get_cached (L, task, "dkim_results", 0)) {
- pres = rspamd_mempool_get_variable (task->task_pool,
- RSPAMD_MEMPOOL_DKIM_CHECK_RESULTS);
-
- if (pres == NULL) {
- lua_newtable (L);
- }
- else {
- for (cur = pres; *cur != NULL; cur ++) {
- nelt ++;
- }
-
- lua_createtable (L, nelt, 0);
-
- for (i = 0; i < nelt; i ++) {
- struct rspamd_dkim_check_result *res = pres[i];
- const gchar *result_str = "unknown";
-
- lua_createtable (L, 0, 4);
-
- switch (res->rcode) {
- case DKIM_CONTINUE:
- result_str = "allow";
- break;
- case DKIM_REJECT:
- result_str = "reject";
- break;
- case DKIM_TRYAGAIN:
- result_str = "tempfail";
- break;
- case DKIM_NOTFOUND:
- result_str = "not found";
- break;
- case DKIM_RECORD_ERROR:
- result_str = "bad record";
- break;
- case DKIM_PERM_ERROR:
- result_str = "permanent error";
- break;
- default:
- break;
- }
-
- rspamd_lua_table_set (L, "result", result_str);
-
- if (res->domain) {
- rspamd_lua_table_set (L, "domain", res->domain);
- }
-
- if (res->selector) {
- rspamd_lua_table_set (L, "selector", res->selector);
- }
-
- if (res->short_b) {
- rspamd_lua_table_set (L, "bhash", res->short_b);
- }
-
- if (res->fail_reason) {
- rspamd_lua_table_set (L, "fail_reason", res->fail_reason);
- }
-
- lua_rawseti (L, -2, i + 1);
- }
- }
-
- lua_task_set_cached (L, task, "dkim_results", -1, 0);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static inline gboolean
- lua_push_symbol_result (lua_State *L,
- struct rspamd_task *task,
- const gchar *symbol,
- struct rspamd_symbol_result *symbol_result,
- gboolean add_metric,
- gboolean add_name)
- {
- struct rspamd_metric_result *metric_res;
- struct rspamd_symbol_result *s = NULL;
- struct rspamd_symbol_option *opt;
- struct rspamd_symbols_group *sym_group;
- guint i;
- gint j = 1, e = 4;
-
- if (!symbol_result) {
- metric_res = task->result;
-
- if (metric_res) {
- s = rspamd_task_find_symbol_result (task, symbol);
- }
- }
- else {
- s = symbol_result;
- }
-
- if (s) {
- if (add_metric) {
- e++;
- }
- if (add_name) {
- e++;
- }
-
- lua_createtable (L, 0, e);
-
- if (add_name) {
- lua_pushstring (L, "name");
- lua_pushstring (L, symbol);
- lua_settable (L, -3);
- }
- lua_pushstring (L, "score");
- lua_pushnumber (L, s->score);
- lua_settable (L, -3);
-
- if (s->sym && s->sym->gr) {
- lua_pushstring (L, "group");
- lua_pushstring (L, s->sym->gr->name);
- lua_settable (L, -3);
-
- lua_pushstring (L, "groups");
- lua_createtable (L, s->sym->groups->len, 0);
-
- PTR_ARRAY_FOREACH (s->sym->groups, i, sym_group) {
- lua_pushstring (L, sym_group->name);
- lua_rawseti (L, -2, i + 1);
- }
-
- lua_settable (L, -3);
- }
- else {
- lua_pushstring (L, "group");
- lua_pushstring (L, "ungrouped");
- lua_settable (L, -3);
- }
-
- if (s->options) {
- lua_pushstring (L, "options");
- lua_createtable (L, kh_size (s->options), 0);
-
- DL_FOREACH (s->opts_head, opt) {
- lua_pushstring (L, (const char*)opt->option);
- lua_rawseti (L, -2, j++);
- }
-
- lua_settable (L, -3);
- }
-
- return TRUE;
- }
-
- return FALSE;
- }
-
- static gint
- lua_task_get_symbol (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol;
- gboolean found = FALSE;
- gint i = 1;
-
- symbol = luaL_checkstring (L, 2);
-
- if (task && symbol) {
- lua_createtable (L, 1, 0);
-
- if ((found = lua_push_symbol_result (L, task, symbol,
- NULL, TRUE, FALSE))) {
- lua_rawseti (L, -2, i++);
- }
- else {
- /* Pop table */
- lua_pop (L, 1);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- if (!found) {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- static gint
- lua_task_has_symbol (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol;
- gboolean found = FALSE;
-
- symbol = luaL_checkstring (L, 2);
-
- if (task && symbol) {
- found = (rspamd_task_find_symbol_result (task, symbol) != NULL);
- lua_pushboolean (L, found);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_enable_symbol (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol;
- gboolean found = FALSE;
-
- symbol = luaL_checkstring (L, 2);
-
- if (task && symbol) {
- found = rspamd_symcache_enable_symbol (task, task->cfg->cache, symbol);
- lua_pushboolean (L, found);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_disable_symbol (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol;
- gboolean found = FALSE;
-
- symbol = luaL_checkstring (L, 2);
-
- if (task && symbol) {
- found = rspamd_symcache_disable_symbol (task, task->cfg->cache, symbol);
- lua_pushboolean (L, found);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_symbols (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_metric_result *mres;
- gint i = 1;
- struct rspamd_symbol_result *s;
-
- if (task) {
- mres = task->result;
-
- if (mres) {
- lua_createtable (L, kh_size (mres->symbols), 0);
- lua_createtable (L, kh_size (mres->symbols), 0);
-
- kh_foreach_value_ptr (mres->symbols, s, {
- if (!(s->flags & RSPAMD_SYMBOL_RESULT_IGNORED)) {
- lua_pushstring (L, s->name);
- lua_rawseti (L, -3, i);
- lua_pushnumber (L, s->score);
- lua_rawseti (L, -2, i);
- i++;
- }
- });
- }
- else {
- lua_createtable (L, 0, 0);
- lua_createtable (L, 0, 0);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 2;
- }
-
- static gint
- lua_task_get_symbols_all (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_metric_result *mres;
- struct rspamd_symbol_result *s;
- gboolean found = FALSE;
- gint i = 1;
-
- if (task) {
- mres = task->result;
-
- if (mres) {
- found = TRUE;
- lua_createtable (L, kh_size (mres->symbols), 0);
-
- kh_foreach_value_ptr (mres->symbols, s, {
- lua_push_symbol_result (L, task, s->name, s, FALSE, TRUE);
- lua_rawseti (L, -2, i++);
- });
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- if (!found) {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
-
- static gint
- lua_task_get_symbols_numeric (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_metric_result *mres;
- gint i = 1, id;
- struct rspamd_symbol_result *s;
-
- if (task) {
- mres = task->result;
-
- if (mres) {
- lua_createtable (L, kh_size (mres->symbols), 0);
- lua_createtable (L, kh_size (mres->symbols), 0);
-
- lua_createtable (L, kh_size (mres->symbols), 0);
-
- kh_foreach_value_ptr (mres->symbols, s, {
- if (!(s->flags & RSPAMD_SYMBOL_RESULT_IGNORED)) {
- id = rspamd_symcache_find_symbol (task->cfg->cache,
- s->name);
- lua_pushinteger (L, id);
- lua_rawseti (L, -3, i);
- lua_pushnumber (L, s->score);
- lua_rawseti (L, -2, i);
- i++;
- }
- });
- }
- else {
- lua_createtable (L, 0, 0);
- lua_createtable (L, 0, 0);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 2;
- }
-
- struct tokens_foreach_cbdata {
- struct rspamd_task *task;
- lua_State *L;
- gint idx;
- gboolean normalize;
- };
-
- static void
- tokens_foreach_cb (gint id, const gchar *sym, gint flags, gpointer ud)
- {
- struct tokens_foreach_cbdata *cbd = ud;
- struct rspamd_symbol_result *s;
-
- if (flags & SYMBOL_TYPE_NOSTAT) {
- return;
- }
-
- if ((s = rspamd_task_find_symbol_result (cbd->task, sym)) != NULL) {
- if (cbd->normalize) {
- lua_pushnumber (cbd->L, tanh (s->score));
- }
- else {
- lua_pushnumber (cbd->L, s->score);
- }
- }
- else {
- lua_pushnumber (cbd->L, 0.0);
- }
-
- lua_rawseti (cbd->L, -2, cbd->idx++);
- }
-
- static gint
- lua_task_get_symbols_tokens (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct tokens_foreach_cbdata cbd;
-
- cbd.task = task;
- cbd.L = L;
- cbd.idx = 1;
- cbd.normalize = TRUE;
-
- if (lua_type (L, 2) == LUA_TBOOLEAN) {
- cbd.normalize = lua_toboolean (L, 2);
- }
- else {
- cbd.normalize = TRUE;
- }
-
- lua_createtable (L,
- rspamd_symcache_stats_symbols_count (task->cfg->cache), 0);
- rspamd_symcache_foreach (task->cfg->cache, tokens_foreach_cb, &cbd);
-
- return 1;
- }
-
- enum lua_date_type {
- DATE_CONNECT = 0,
- DATE_MESSAGE,
- DATE_CONNECT_STRING,
- DATE_MESSAGE_STRING
- };
-
- static enum lua_date_type
- lua_task_detect_date_type (struct rspamd_task *task,
- lua_State *L, gint idx, gboolean *gmt)
- {
- enum lua_date_type type = DATE_CONNECT;
-
- if (lua_type (L, idx) == LUA_TNUMBER) {
- gint num = lua_tonumber (L, idx);
- if (num >= DATE_CONNECT && num <= DATE_MESSAGE_STRING) {
- return num;
- }
- }
- else if (lua_type (L, idx) == LUA_TTABLE) {
- const gchar *str;
-
- lua_pushvalue (L, idx);
- lua_pushstring (L, "format");
- lua_gettable (L, -2);
-
- str = lua_tostring (L, -1);
-
- if (str) {
- if (g_ascii_strcasecmp (str, "message") == 0) {
- type = DATE_MESSAGE;
- } else if (g_ascii_strcasecmp (str, "connect_str") == 0) {
- type = DATE_CONNECT_STRING;
- } else if (g_ascii_strcasecmp (str, "message_str") == 0) {
- type = DATE_MESSAGE_STRING;
- }
- }
- else {
- msg_warn_task ("date format has not been specified");
- }
-
- lua_pop (L, 1);
-
- lua_pushstring (L, "gmt");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- *gmt = lua_toboolean (L, -1);
- }
-
- /* Value and table */
- lua_pop (L, 2);
- }
-
- return type;
- }
-
- static gint
- lua_task_get_date (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- GPtrArray *hdrs;
- gdouble tim;
- enum lua_date_type type = DATE_CONNECT;
- gboolean gmt = TRUE;
-
- if (task != NULL) {
- if (lua_gettop (L) > 1) {
- type = lua_task_detect_date_type (task, L, 2, &gmt);
- }
- /* Get GMT date and store it to time_t */
- if (type == DATE_CONNECT || type == DATE_CONNECT_STRING) {
- tim = (tv_to_msec (&task->tv)) / 1000.;
-
- if (!gmt) {
- struct tm t;
- time_t tt;
-
- tt = tim;
- rspamd_localtime (tt, &t);
- #if !defined(__sun)
- t.tm_gmtoff = 0;
- #endif
- t.tm_isdst = 0;
- tim = mktime (&t);
- }
- }
- else {
- hdrs = rspamd_message_get_header_array (task, "Date",
- FALSE);
-
- if (hdrs && hdrs->len > 0) {
- time_t tt;
- struct tm t;
- struct rspamd_mime_header *h;
-
- h = g_ptr_array_index (hdrs, 0);
- tt = rspamd_parse_smtp_date (h->decoded, strlen (h->decoded));
-
- if (!gmt) {
- rspamd_localtime (tt, &t);
- #if !defined(__sun)
- t.tm_gmtoff = 0;
- #endif
- t.tm_isdst = 0;
- tim = mktime (&t);
- }
- else {
- tim = tt;
- }
- }
- else {
- tim = 0.0;
- }
- }
-
- if (type == DATE_CONNECT || type == DATE_MESSAGE) {
- lua_pushnumber (L, tim);
- }
- else {
- GTimeVal tv;
- gchar *out;
-
- double_to_tv (tim, &tv);
- out = g_time_val_to_iso8601 (&tv);
- lua_pushstring (L, out);
- g_free (out);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_message_id (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- if (task->message_id != NULL) {
- lua_pushstring (L, task->message_id);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_timeval (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- lua_createtable (L, 0, 2);
- lua_pushstring (L, "tv_sec");
- lua_pushinteger (L, (lua_Integer)task->tv.tv_sec);
- lua_settable (L, -3);
- lua_pushstring (L, "tv_usec");
- lua_pushinteger (L, (lua_Integer)task->tv.tv_usec);
- lua_settable (L, -3);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_size (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
- lua_pushinteger (L, task->msg.len);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- /**
- * - `no_log`: do not log task summary
- * - `no_stat`: do not include task into scanned stats
- * - `pass_all`: check all filters for task
- * - `extended_urls`: output extended info about urls
- * - `skip`: skip task processing
- */
-
- #define LUA_TASK_FLAG_WRITE(flag, set) do { \
- task->flags = (set) ? (task->flags | (flag)) : (task->flags & ~(flag)); \
- } while(0)
-
- #define LUA_TASK_SET_FLAG(flag, strname, macro, set) do { \
- if (!found && strcmp ((flag), strname) == 0) { \
- LUA_TASK_FLAG_WRITE((macro), set); \
- found = TRUE; \
- } \
- } while(0)
-
- #define LUA_TASK_FLAG_READ(flag) do { \
- lua_pushboolean(L, !!(task->flags & (flag))); \
- } while(0)
-
- #define LUA_TASK_GET_FLAG(flag, strname, macro) do { \
- if (!found && strcmp ((flag), strname) == 0) { \
- LUA_TASK_FLAG_READ((macro)); \
- found = TRUE; \
- } \
- } while(0)
-
- static gint
- lua_task_set_flag (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *flag = luaL_checkstring (L, 2);
- gboolean set = TRUE, found = FALSE;
-
- if (lua_gettop (L) >= 3) {
- set = lua_toboolean (L, 3);
- }
-
- if (task != NULL && flag != NULL) {
- LUA_TASK_SET_FLAG (flag, "pass_all", RSPAMD_TASK_FLAG_PASS_ALL, set);
- LUA_TASK_SET_FLAG (flag, "no_log", RSPAMD_TASK_FLAG_NO_LOG, set);
- LUA_TASK_SET_FLAG (flag, "no_stat", RSPAMD_TASK_FLAG_NO_STAT, set);
- LUA_TASK_SET_FLAG (flag, "skip", RSPAMD_TASK_FLAG_SKIP, set);
- LUA_TASK_SET_FLAG (flag, "extended_urls", RSPAMD_TASK_FLAG_EXT_URLS, set);
- LUA_TASK_SET_FLAG (flag, "learn_spam", RSPAMD_TASK_FLAG_LEARN_SPAM, set);
- LUA_TASK_SET_FLAG (flag, "learn_ham", RSPAMD_TASK_FLAG_LEARN_HAM, set);
- LUA_TASK_SET_FLAG (flag, "broken_headers",
- RSPAMD_TASK_FLAG_BROKEN_HEADERS, set);
- LUA_TASK_SET_FLAG (flag, "greylisted", RSPAMD_TASK_FLAG_GREYLISTED, set);
- LUA_TASK_SET_FLAG (flag, "skip_process", RSPAMD_TASK_FLAG_SKIP_PROCESS, set);
-
- if (!found) {
- msg_warn_task ("unknown flag requested: %s", flag);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_has_flag (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *flag = luaL_checkstring (L, 2);
- gboolean found = FALSE;
-
- if (task != NULL && flag != NULL) {
- LUA_TASK_GET_FLAG (flag, "pass_all", RSPAMD_TASK_FLAG_PASS_ALL);
- LUA_TASK_GET_FLAG (flag, "no_log", RSPAMD_TASK_FLAG_NO_LOG);
- LUA_TASK_GET_FLAG (flag, "no_stat", RSPAMD_TASK_FLAG_NO_STAT);
- LUA_TASK_GET_FLAG (flag, "skip", RSPAMD_TASK_FLAG_SKIP);
- LUA_TASK_GET_FLAG (flag, "extended_urls", RSPAMD_TASK_FLAG_EXT_URLS);
- LUA_TASK_GET_FLAG (flag, "learn_spam", RSPAMD_TASK_FLAG_LEARN_SPAM);
- LUA_TASK_GET_FLAG (flag, "learn_ham", RSPAMD_TASK_FLAG_LEARN_HAM);
- LUA_TASK_GET_FLAG (flag, "greylisted", RSPAMD_TASK_FLAG_GREYLISTED);
- LUA_TASK_GET_FLAG (flag, "broken_headers",
- RSPAMD_TASK_FLAG_BROKEN_HEADERS);
- LUA_TASK_GET_FLAG (flag, "skip_process",
- RSPAMD_TASK_FLAG_SKIP_PROCESS);
- LUA_TASK_GET_FLAG (flag, "milter",
- RSPAMD_TASK_FLAG_MILTER);
- LUA_TASK_GET_FLAG (flag, "bad_unicode",
- RSPAMD_TASK_FLAG_BAD_UNICODE);
- LUA_TASK_GET_FLAG (flag, "mime",
- RSPAMD_TASK_FLAG_MIME);
-
- if (!found) {
- msg_warn_task ("unknown flag requested: %s", flag);
- lua_pushboolean (L, 0);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_flags (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gint idx = 1;
- guint flags, bit, i;
-
- if (task) {
- lua_createtable (L, 8, 0);
-
- flags = task->flags;
-
- for (i = 0; i < sizeof (task->flags) * NBBY; i ++) {
- bit = (1U << i);
-
- if (flags & bit) {
- switch (bit) {
- case RSPAMD_TASK_FLAG_PASS_ALL:
- lua_pushstring (L, "pass_all");
- lua_rawseti (L, -2, idx ++);
- break;
- case RSPAMD_TASK_FLAG_NO_LOG:
- lua_pushstring (L, "no_log");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_NO_STAT:
- lua_pushstring (L, "no_stat");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_SKIP:
- lua_pushstring (L, "skip");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_EXT_URLS:
- lua_pushstring (L, "extended_urls");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_BROKEN_HEADERS:
- lua_pushstring (L, "broken_headers");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_LEARN_SPAM:
- lua_pushstring (L, "learn_spam");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_LEARN_HAM:
- lua_pushstring (L, "learn_ham");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_GREYLISTED:
- lua_pushstring (L, "greylisted");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_SKIP_PROCESS:
- lua_pushstring (L, "skip_process");
- lua_rawseti (L, -2, idx++);
- break;
- case RSPAMD_TASK_FLAG_MILTER:
- lua_pushstring (L, "milter");
- lua_rawseti (L, -2, idx++);
- break;
- default:
- break;
- }
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_digest (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gchar hexbuf[33];
- gint r;
-
- if (task) {
- r = rspamd_encode_hex_buf (task->digest, sizeof (task->digest),
- hexbuf, sizeof (hexbuf) - 1);
-
- if (r > 0) {
- hexbuf[r] = '\0';
- lua_pushstring (L, hexbuf);
- }
- else {
- lua_pushnil (L);
- }
-
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_learn (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gboolean is_spam = FALSE;
- const gchar *clname = NULL;
- GError *err = NULL;
- int ret = 1;
-
- if (task == NULL) {
- return luaL_error (L, "invalid arguments");
- }
-
- is_spam = lua_toboolean(L, 2);
- if (lua_gettop (L) > 2) {
- clname = luaL_checkstring (L, 3);
- }
-
- if (!rspamd_learn_task_spam (task, is_spam, clname, &err)) {
- lua_pushboolean (L, FALSE);
- if (err != NULL) {
- lua_pushstring (L, err->message);
- ret = 2;
- }
- }
- else {
- lua_pushboolean (L, TRUE);
- }
-
- return ret;
- }
-
- static gint
- lua_task_set_settings (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- ucl_object_t *settings;
- const ucl_object_t *act, *elt, *metric_elt, *vars, *cur;
- ucl_object_iter_t it = NULL;
- struct rspamd_metric_result *mres;
- guint i;
-
- settings = ucl_object_lua_import (L, 2);
-
- if (settings != NULL && task != NULL) {
-
- metric_elt = ucl_object_lookup (settings, DEFAULT_METRIC);
-
- if (metric_elt) {
- task->settings = ucl_object_ref (metric_elt);
- ucl_object_unref (settings);
- }
- else {
- task->settings = settings;
- }
-
- act = ucl_object_lookup (task->settings, "actions");
-
- if (act) {
- /* Adjust desired actions */
- mres = task->result;
-
- for (i = 0; i < mres->nactions; i++) {
- struct rspamd_action_result *act_res = &mres->actions_limits[i];
- elt = ucl_object_lookup (act, act_res->action->name);
-
- if (elt) {
- if (ucl_object_type (elt) == UCL_FLOAT ||
- ucl_object_type (elt) == UCL_INT) {
- gdouble nscore = ucl_object_todouble (elt);
-
- msg_debug_task ("adjusted action %s: %.2f -> %.2f",
- ucl_object_key (elt),
- act_res->cur_limit,
- nscore);
- act_res->cur_limit = nscore;
- }
- else if (ucl_object_type (elt) == UCL_NULL) {
- act_res->cur_limit = NAN;
- msg_info_task ("disabled action %s due to settings",
- ucl_object_key (elt));
- }
- }
- }
- }
-
- vars = ucl_object_lookup (task->settings, "variables");
- if (vars && ucl_object_type (vars) == UCL_OBJECT) {
- /* Set memory pool variables */
- it = NULL;
-
- while ((cur = ucl_object_iterate (vars, &it, true)) != NULL) {
- if (ucl_object_type (cur) == UCL_STRING) {
- rspamd_mempool_set_variable (task->task_pool,
- ucl_object_key (cur), rspamd_mempool_strdup (
- task->task_pool,
- ucl_object_tostring (cur)
- ), NULL);
- }
- }
- }
-
- rspamd_symcache_process_settings (task, task->cfg->cache);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_set_milter_reply (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- ucl_object_t *reply, *prev;
-
- reply = ucl_object_lua_import (L, 2);
-
- if (reply != NULL && task != NULL) {
- prev = rspamd_mempool_get_variable (task->task_pool,
- RSPAMD_MEMPOOL_MILTER_REPLY);
-
- if (prev) {
- ucl_object_merge (prev, reply, false);
- ucl_object_unref (reply);
- }
- else {
- rspamd_mempool_set_variable (task->task_pool,
- RSPAMD_MEMPOOL_MILTER_REPLY,
- reply, (rspamd_mempool_destruct_t) ucl_object_unref);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_task_get_settings (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task != NULL) {
-
- if (task->settings) {
- return ucl_object_push_lua (L, task->settings, true);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_lookup_settings (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *key = NULL;
- const ucl_object_t *elt;
-
- if (task != NULL) {
-
- if (lua_isstring (L, 2)) {
- key = lua_tostring (L, 2);
- }
-
- if (task->settings) {
- if (key == NULL) {
- return ucl_object_push_lua (L, task->settings, true);
- }
- else {
- elt = ucl_object_lookup (task->settings, key);
-
- if (elt) {
- return ucl_object_push_lua (L, elt, true);
- }
- else {
- lua_pushnil (L);
- }
- }
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_settings_id (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- guint32 *hp;
-
- if (task != NULL) {
- hp = rspamd_mempool_get_variable (task->task_pool, "settings_hash");
-
- if (hp) {
- lua_pushnumber (L, *hp);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_cache_get (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *key = luaL_checkstring (L, 2);
- guint id = 0;
-
- if (task && key) {
- if (lua_type (L, 3) == LUA_TNUMBER) {
- id = lua_tonumber (L, 3);
- }
-
- if (!lua_task_get_cached (L, task, key, id)) {
- lua_pushnil (L);
- }
- }
- else {
- luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_cache_set (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *key = luaL_checkstring (L, 2);
- guint id = 0;
-
- if (task && key && lua_gettop (L) >= 3) {
- if (lua_type (L, 4) == LUA_TNUMBER) {
- id = lua_tonumber (L, 4);
- }
-
- lua_task_set_cached (L, task, key, 3, id);
- }
- else {
- luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- struct lua_file_cbdata {
- gchar *fname;
- gint fd;
- gboolean keep;
- };
-
- static void
- lua_tmp_file_dtor (gpointer p)
- {
- struct lua_file_cbdata *cbdata = p;
-
- if (!cbdata->keep) {
- unlink (cbdata->fname);
- }
-
- close (cbdata->fd);
- }
-
- static gint
- lua_task_store_in_file (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gboolean force_new = FALSE, keep = FALSE;
- gchar fpath[PATH_MAX];
- const gchar *tmpmask = NULL, *fname = NULL;
- guint64 mode = 00600;
- gint fd;
- struct lua_file_cbdata *cbdata;
- GError *err = NULL;
-
- if (task) {
- if (lua_istable (L, 2)) {
- if (!rspamd_lua_parse_table_arguments (L, 2, &err,
- "filename=S;tmpmask=S;mode=I;force_new=B;keep=B",
- &fname, &tmpmask, &mode, &force_new, &keep)) {
- msg_err_task ("cannot get parameters list: %e", err);
-
- if (err) {
- g_error_free (err);
- }
-
- return luaL_error (L, "invalid arguments");
- }
- }
- else if (lua_isnumber (L, 2)) {
- mode = lua_tonumber (L, 2);
- }
-
- if (!force_new && (task->flags & RSPAMD_TASK_FLAG_FILE) &&
- task->msg.fpath) {
- lua_pushstring (L, task->msg.fpath);
- }
- else {
- if (fname == NULL) {
- if (tmpmask == NULL) {
- rspamd_snprintf (fpath, sizeof (fpath), "%s%c%s",
- task->cfg->temp_dir,
- G_DIR_SEPARATOR, "rmsg-XXXXXXXXXX");
- }
- else {
- rspamd_snprintf (fpath, sizeof (fpath), "%s", tmpmask);
- }
-
- fd = mkstemp (fpath);
- fname = fpath;
-
- if (fd != -1) {
- fchmod (fd, mode);
- }
- }
- else {
- fd = rspamd_file_xopen (fname, O_WRONLY|O_CREAT|O_EXCL,
- (guint)mode, FALSE);
- }
-
- if (fd == -1) {
- msg_err_task ("cannot save file: %s", strerror (errno));
- lua_pushnil (L);
- }
- else {
- if (write (fd, task->msg.begin, task->msg.len) == -1) {
- msg_err_task ("cannot write file %s: %s", fpath,
- strerror (errno));
- unlink (fname);
- close (fd);
- lua_pushnil (L);
-
- return 1;
- }
-
- cbdata = rspamd_mempool_alloc (task->task_pool, sizeof (*cbdata));
- cbdata->fd = fd;
- cbdata->fname = rspamd_mempool_strdup (task->task_pool, fname);
- cbdata->keep = keep;
- lua_pushstring (L, cbdata->fname);
- rspamd_mempool_add_destructor (task->task_pool,
- lua_tmp_file_dtor, cbdata);
- }
- }
- }
- else {
- luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_process_regexp (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_lua_regexp *re = NULL;
- gboolean strong = FALSE;
- const gchar *type_str = NULL, *header_str = NULL;
- gsize header_len = 0;
- GError *err = NULL;
- gint ret = 0;
- enum rspamd_re_type type = RSPAMD_RE_BODY;
-
- /*
- * - `re`* : regular expression object
- * - `type`*: type of regular expression:
- * + `mime`: mime regexp
- * + `rawmime`: raw mime regexp
- * + `header`: header regexp
- * + `rawheader`: raw header expression
- * + `body`: raw body regexp
- * + `url`: url regexp
- * - `header`: for header and rawheader regexp means the name of header
- * - `strong`: case sensitive match for headers
- */
- if (task != NULL) {
- if (!rspamd_lua_parse_table_arguments (L, 2, &err,
- "*re=U{regexp};*type=S;header=V;strong=B",
- &re, &type_str, &header_len, &header_str,
- &strong)) {
- msg_err_task ("cannot get parameters list: %e", err);
-
- if (err) {
- g_error_free (err);
- }
-
- return luaL_error (L, "invalid arguments");
- }
- else {
- type = rspamd_re_cache_type_from_string (type_str);
-
- if ((type == RSPAMD_RE_HEADER || type == RSPAMD_RE_RAWHEADER)
- && header_str == NULL) {
- msg_err_task (
- "header argument is mandatory for header/rawheader regexps");
- }
- else {
- ret = rspamd_re_cache_process (task, re->re, type,
- (gpointer) header_str, header_len, strong);
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushinteger (L, ret);
-
- return 1;
- }
-
- static gint
- lua_task_get_metric_result (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_metric_result *metric_res;
- struct rspamd_action *action;
-
- if (task) {
- metric_res = task->result;
-
- /* Fields added:
- * - `score`: current score
- * - `action`: current action as a string
- * - `nnegative`: number of negative rules matched
- * - `npositive`: number of positive rules matched
- * - `positive_score`: total score for positive rules
- * - `negative_score`: total score for negative rules
- * - `passthrough`: set to true if message has a passthrough result
- */
- lua_createtable (L, 0, 7);
-
- lua_pushstring (L, "score");
- lua_pushnumber (L, metric_res->score);
- lua_settable (L, -3);
-
- action = rspamd_check_action_metric (task);
-
- if (action) {
- lua_pushstring (L, "action");
- lua_pushstring (L, action->name);
- lua_settable (L, -3);
- }
-
- lua_pushstring (L, "nnegative");
- lua_pushnumber (L, metric_res->nnegative);
- lua_settable (L, -3);
-
- lua_pushstring (L, "npositive");
- lua_pushnumber (L, metric_res->npositive);
- lua_settable (L, -3);
-
- lua_pushstring (L, "positive_score");
- lua_pushnumber (L, metric_res->positive_score);
- lua_settable (L, -3);
-
- lua_pushstring (L, "negative_score");
- lua_pushnumber (L, metric_res->negative_score);
- lua_settable (L, -3);
-
- lua_pushstring (L, "passthrough");
- lua_pushboolean (L, !!(metric_res->passthrough_result != NULL));
- lua_settable (L, -3);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_metric_score (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- gdouble rs;
- struct rspamd_metric_result *metric_res;
-
- if (task) {
- if ((metric_res = task->result) != NULL) {
- lua_createtable (L, 2, 0);
- lua_pushnumber (L, isnan (metric_res->score) ? 0.0 : metric_res->score);
- rs = rspamd_task_get_required_score (task, metric_res);
- lua_rawseti (L, -2, 1);
- lua_pushnumber (L, rs);
- lua_rawseti (L, -2, 2);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_metric_action (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_metric_result *metric_res;
- struct rspamd_action *action;
-
- if (task) {
- metric_res = task->result;
-
- action = rspamd_check_action_metric (task);
- lua_pushstring (L, action->name);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_metric_score (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_metric_result *metric_res;
- gdouble nscore;
-
- if (lua_isnumber (L, 2)) {
- nscore = luaL_checknumber (L, 2);
- }
- else {
- nscore = luaL_checknumber (L, 3);
- }
-
- if (task) {
- if ((metric_res = task->result) != NULL) {
- msg_debug_task ("set metric score from %.2f to %.2f",
- metric_res->score, nscore);
- metric_res->score = nscore;
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_disable_action (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *action_name;
- struct rspamd_action_result *action_res;
-
- action_name = luaL_checkstring (L, 2);
-
- if (task && action_name) {
-
- for (guint i = 0; i < task->result->nactions; i ++) {
- action_res = &task->result->actions_limits[i];
-
- if (strcmp (action_name, action_res->action->name) == 0) {
- if (isnan (action_res->cur_limit)) {
- lua_pushboolean (L, false);
- }
- else {
- action_res->cur_limit = NAN;
- lua_pushboolean (L, true);
- }
-
- break;
- }
- }
-
-
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_newlines_type (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- switch (task->nlines_type) {
- case RSPAMD_TASK_NEWLINES_CR:
- lua_pushstring (L, "cr");
- break;
- case RSPAMD_TASK_NEWLINES_LF:
- lua_pushstring (L, "lf");
- break;
- case RSPAMD_TASK_NEWLINES_CRLF:
- default:
- lua_pushstring (L, "crlf");
- break;
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static void
- lua_push_stat_token (lua_State *L, rspamd_token_t *tok)
- {
- gchar numbuf[64];
-
- /* Table values
- * - `data`: 64 bit number encoded as a string
- * - `t1`: the first token (if any)
- * - `t2`: the second token (if any)
- * - `win`: window index
- * - `flag`: table of strings:
- * - `text`: text token
- * - `meta`: meta token
- * - `lua`: lua meta token
- * - `exception`: exception
- * - `subject`: subject token
- * - `unigram`: unigram token
- */
- lua_createtable (L, 0, 5);
-
- rspamd_snprintf (numbuf, sizeof (numbuf), "%uL", tok->data);
- lua_pushstring (L, "data");
- lua_pushstring (L, numbuf);
- lua_settable (L, -3);
-
- if (tok->t1) {
- lua_pushstring (L, "t1");
- lua_pushlstring (L, tok->t1->stemmed.begin, tok->t1->stemmed.len);
- lua_settable (L, -3);
- }
-
- if (tok->t2) {
- lua_pushstring (L, "t2");
- lua_pushlstring (L, tok->t2->stemmed.begin, tok->t2->stemmed.len);
- lua_settable (L, -3);
- }
-
- lua_pushstring (L, "win");
- lua_pushinteger (L, tok->window_idx);
- lua_settable (L, -3);
-
- lua_pushstring (L, "flags");
- lua_createtable (L, 0, 5);
-
- /* Flags */
- {
- if (tok->flags & RSPAMD_STAT_TOKEN_FLAG_TEXT) {
- lua_pushstring (L, "text");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (tok->flags & RSPAMD_STAT_TOKEN_FLAG_META) {
- lua_pushstring (L, "meta");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (tok->flags & RSPAMD_STAT_TOKEN_FLAG_LUA_META) {
- lua_pushstring (L, "lua");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (tok->flags & RSPAMD_STAT_TOKEN_FLAG_EXCEPTION) {
- lua_pushstring (L, "exception");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (tok->flags & RSPAMD_STAT_TOKEN_FLAG_HEADER) {
- lua_pushstring (L, "header");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- }
- lua_settable (L, -3);
- }
-
- static gint
- lua_task_get_stat_tokens (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- guint i;
- rspamd_token_t *tok;
-
- if (task) {
- if (!task->tokens) {
- rspamd_stat_process_tokenize (NULL, task);
- }
-
- if (!task->tokens) {
- lua_pushnil (L);
- }
- else {
- lua_createtable (L, task->tokens->len, 0);
-
- PTR_ARRAY_FOREACH (task->tokens, i, tok) {
- lua_push_stat_token (L, tok);
- lua_rawseti (L, -2, i + 1);
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_set_metric_subject (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *subject;
-
- subject = luaL_checkstring (L, 2);
-
- if (task && subject) {
- rspamd_mempool_set_variable (task->task_pool, "metric_subject",
- rspamd_mempool_strdup(task->task_pool, subject), NULL);
- lua_pushboolean (L, true);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_task_get_protocol_reply (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- guint flags = 0;
- ucl_object_t *obj;
-
- if (!task) {
- return luaL_error (L, "invalid arguments");
- }
-
- if (!(task->processed_stages & (RSPAMD_TASK_STAGE_POST_FILTERS >> 1))) {
- return luaL_error (L, "must not be called before post-filters");
- }
-
- if (lua_istable (L, 2)) {
- for (lua_pushnil (L); lua_next (L, 2); lua_pop (L, 1)) {
- if (lua_isstring (L, -1)) {
- const gchar *str = lua_tostring (L, -1);
-
- if (strcmp (str, "default") == 0) {
- flags |= RSPAMD_PROTOCOL_DEFAULT;
- }
- else if (strcmp (str, "basic") == 0) {
- flags |= RSPAMD_PROTOCOL_BASIC;
- }
- else if (strcmp (str, "metrics") == 0) {
- flags |= RSPAMD_PROTOCOL_METRICS;
- }
- else if (strcmp (str, "messages") == 0) {
- flags |= RSPAMD_PROTOCOL_MESSAGES;
- }
- else if (strcmp (str, "rmilter") == 0) {
- flags |= RSPAMD_PROTOCOL_RMILTER;
- }
- else if (strcmp (str, "dkim") == 0) {
- flags |= RSPAMD_PROTOCOL_DKIM;
- }
- else if (strcmp (str, "extra") == 0) {
- flags |= RSPAMD_PROTOCOL_EXTRA;
- }
- else {
- msg_err_task ("invalid protocol flag: %s", str);
- }
- }
- }
- }
- else {
- flags = RSPAMD_PROTOCOL_DEFAULT;
- }
-
- obj = rspamd_protocol_write_ucl (task, flags);
-
- if (obj) {
- ucl_object_push_lua (L, obj, true);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- static gint
- lua_task_headers_foreach (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- enum rspamd_lua_task_header_type how = RSPAMD_TASK_HEADER_PUSH_SIMPLE;
- struct rspamd_lua_regexp *re = NULL;
- GList *cur;
- struct rspamd_mime_header *hdr;
- gint old_top;
-
- if (task && lua_isfunction (L, 2)) {
- if (lua_istable (L, 3)) {
- lua_pushstring (L, "full");
- lua_gettable (L, 3);
-
- if (lua_isboolean (L, -1) && lua_toboolean (L, -1)) {
- how = RSPAMD_TASK_HEADER_PUSH_FULL;
- }
-
- lua_pop (L, 1);
-
- lua_pushstring (L, "raw");
- lua_gettable (L, 3);
-
- if (lua_isboolean (L, -1) && lua_toboolean (L, -1)) {
- how = RSPAMD_TASK_HEADER_PUSH_RAW;
- }
-
- lua_pop (L, 1);
-
- lua_pushstring (L, "regexp");
- lua_gettable (L, 3);
-
- if (lua_isuserdata (L, -1)) {
- re = *(struct rspamd_lua_regexp **)
- rspamd_lua_check_udata (L, -1, "rspamd{regexp}");
- }
-
- lua_pop (L, 1);
- }
-
- if (task->headers_order) {
- cur = task->headers_order->head;
-
- while (cur) {
- hdr = cur->data;
-
- if (re && re->re) {
- if (!rspamd_regexp_match (re->re, hdr->name,
- strlen (hdr->name),FALSE)) {
- cur = g_list_next (cur);
- continue;
- }
- }
-
- old_top = lua_gettop (L);
- lua_pushvalue (L, 2);
- lua_pushstring (L, hdr->name);
- rspamd_lua_push_header (L, hdr, how);
-
- if (lua_pcall (L, 2, LUA_MULTRET, 0) != 0) {
- msg_err ("call to header_foreach failed: %s",
- lua_tostring (L, -1));
- lua_settop (L, old_top);
- break;
- }
- else {
- if (lua_gettop (L) > old_top) {
- if (lua_isboolean (L, old_top + 1)) {
- if (lua_toboolean (L, old_top + 1)) {
- lua_settop (L, old_top);
- break;
- }
- }
- }
- }
-
- lua_settop (L, old_top);
- cur = g_list_next (cur);
- }
- }
- }
-
- return 0;
- }
-
- static gint
- lua_task_get_meta_words (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- enum rspamd_lua_words_type how = RSPAMD_LUA_WORDS_STEM;
-
- if (task == NULL) {
- return luaL_error (L, "invalid arguments");
- }
-
- if (task->meta_words == NULL) {
- lua_createtable (L, 0, 0);
- }
- else {
- if (lua_type (L, 2) == LUA_TSTRING) {
- const gchar *how_str = lua_tostring (L, 2);
-
- if (strcmp (how_str, "stem") == 0) {
- how = RSPAMD_LUA_WORDS_STEM;
- }
- else if (strcmp (how_str, "norm") == 0) {
- how = RSPAMD_LUA_WORDS_NORM;
- }
- else if (strcmp (how_str, "raw") == 0) {
- how = RSPAMD_LUA_WORDS_RAW;
- }
- else if (strcmp (how_str, "full") == 0) {
- how = RSPAMD_LUA_WORDS_FULL;
- }
- else {
- return luaL_error (L, "unknown words type: %s", how_str);
- }
- }
-
- return rspamd_lua_push_words (L, task->meta_words, how);
- }
-
- return 1;
- }
-
- static guint
- lua_lookup_words_array (lua_State *L,
- gint cbpos,
- struct rspamd_task *task,
- struct rspamd_lua_map *map,
- GArray *words)
- {
- rspamd_stat_token_t *tok;
- guint i, nmatched = 0;
- gint err_idx;
- gboolean matched;
- const gchar *key;
- gsize keylen;
-
- for (i = 0; i < words->len; i ++) {
- tok = &g_array_index (words, rspamd_stat_token_t, i);
-
- matched = FALSE;
-
- if (tok->normalized.len == 0) {
- continue;
- }
-
- key = tok->normalized.begin;
- keylen = tok->normalized.len;
-
- switch (map->type) {
- case RSPAMD_LUA_MAP_SET:
- case RSPAMD_LUA_MAP_HASH:
- /* We know that tok->normalized is zero terminated in fact */
- if (rspamd_match_hash_map (map->data.hash, key)) {
- matched = TRUE;
- }
- break;
- case RSPAMD_LUA_MAP_REGEXP:
- case RSPAMD_LUA_MAP_REGEXP_MULTIPLE:
- if (rspamd_match_regexp_map_single (map->data.re_map, key,
- keylen)) {
- matched = TRUE;
- }
- break;
- default:
- g_assert_not_reached ();
- break;
- }
-
- if (matched) {
- nmatched ++;
-
- lua_pushcfunction (L, &rspamd_lua_traceback);
- err_idx = lua_gettop (L);
- lua_pushvalue (L, cbpos); /* Function */
- rspamd_lua_push_full_word (L, tok);
-
- if (lua_pcall (L, 1, 0, err_idx) != 0) {
- GString *tb = lua_touserdata (L, -1);
- msg_err_task ("cannot call callback function for lookup words: %s",
- tb->str);
- g_string_free (tb, TRUE);
- }
-
- lua_settop (L, err_idx - 1);
- }
- }
-
- return nmatched;
- }
-
- static gint
- lua_task_lookup_words (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
- struct rspamd_lua_map *map = lua_check_map (L, 2);
- struct rspamd_mime_text_part *tp;
-
- guint i, matches = 0;
-
- if (task == NULL || map == NULL || lua_type (L, 3) != LUA_TFUNCTION) {
- return luaL_error (L, "invalid arguments");
- }
-
- if (map->type != RSPAMD_LUA_MAP_SET &&
- map->type != RSPAMD_LUA_MAP_REGEXP &&
- map->type != RSPAMD_LUA_MAP_HASH &&
- map->type != RSPAMD_LUA_MAP_REGEXP_MULTIPLE) {
- return luaL_error (L, "invalid map type");
- }
-
- PTR_ARRAY_FOREACH (task->text_parts, i, tp) {
- if (tp->utf_words) {
- matches += lua_lookup_words_array (L, 3, task, map, tp->utf_words);
- }
- }
-
- if (task->meta_words) {
- matches += lua_lookup_words_array (L, 3, task, map, task->meta_words);
- }
-
- lua_pushinteger (L, matches);
-
- return 1;
- }
-
- static gint
- lua_task_topointer (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_task *task = lua_check_task (L, 1);
-
- if (task) {
- lua_pushlightuserdata (L, task);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
-
- /* Image functions */
- static gint
- lua_image_get_width (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_image *img = lua_check_image (L);
-
- if (img != NULL) {
- lua_pushinteger (L, img->width);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_image_get_height (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_image *img = lua_check_image (L);
-
- if (img != NULL) {
- lua_pushinteger (L, img->height);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_image_get_type (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_image *img = lua_check_image (L);
-
- if (img != NULL) {
- lua_pushstring (L, rspamd_image_type_str (img->type));
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_image_get_size (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_image *img = lua_check_image (L);
-
- if (img != NULL) {
- lua_pushinteger (L, img->data->len);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_image_get_filename (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_image *img = lua_check_image (L);
-
- if (img != NULL && img->filename != NULL) {
- lua_pushlstring (L, img->filename->begin, img->filename->len);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- /* Arvhive methods */
- static gint
- lua_archive_get_type (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_archive *arch = lua_check_archive (L);
-
- if (arch != NULL) {
- lua_pushstring (L, rspamd_archive_type_str (arch->type));
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_archive_get_files (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_archive *arch = lua_check_archive (L);
- guint i;
- struct rspamd_archive_file *f;
-
- if (arch != NULL) {
- lua_createtable (L, arch->files->len, 0);
-
- for (i = 0; i < arch->files->len; i ++) {
- f = g_ptr_array_index (arch->files, i);
-
- lua_pushlstring (L, f->fname->str, f->fname->len);
- lua_rawseti (L, -2, i + 1);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_archive_get_files_full (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_archive *arch = lua_check_archive (L);
- guint i;
- struct rspamd_archive_file *f;
-
- if (arch != NULL) {
- lua_createtable (L, arch->files->len, 0);
-
- for (i = 0; i < arch->files->len; i ++) {
- f = g_ptr_array_index (arch->files, i);
-
- lua_createtable (L, 0, 4);
-
- lua_pushstring (L, "name");
- lua_pushlstring (L, f->fname->str, f->fname->len);
- lua_settable (L, -3);
-
- lua_pushstring (L, "compressed_size");
- lua_pushinteger (L, f->compressed_size);
- lua_settable (L, -3);
-
- lua_pushstring (L, "uncompressed_size");
- lua_pushinteger (L, f->uncompressed_size);
- lua_settable (L, -3);
-
- lua_pushstring (L, "encrypted");
- lua_pushboolean (L, (f->flags & RSPAMD_ARCHIVE_FILE_ENCRYPTED) ? true : false);
- lua_settable (L, -3);
-
- lua_rawseti (L, -2, i + 1);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_archive_is_encrypted (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_archive *arch = lua_check_archive (L);
-
- if (arch != NULL) {
- lua_pushboolean (L, (arch->flags & RSPAMD_ARCHIVE_ENCRYPTED) ? true : false);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_archive_get_size (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_archive *arch = lua_check_archive (L);
-
- if (arch != NULL) {
- lua_pushinteger (L, arch->size);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_archive_get_filename (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_archive *arch = lua_check_archive (L);
-
- if (arch != NULL) {
- lua_pushlstring (L, arch->archive_name->begin, arch->archive_name->len);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- /* Text methods */
- static gint
- lua_text_fromstring (lua_State *L)
- {
- LUA_TRACE_POINT;
- const gchar *str;
- gsize l = 0;
- struct rspamd_lua_text *t;
-
- str = luaL_checklstring (L, 1, &l);
-
- if (str) {
- t = lua_newuserdata (L, sizeof (*t));
- t->start = g_malloc (l + 1);
- rspamd_strlcpy ((char *)t->start, str, l + 1);
- t->len = l;
- t->flags = RSPAMD_TEXT_FLAG_OWN;
- rspamd_lua_setclass (L, "rspamd{text}", -1);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
-
- return 1;
- }
-
- static gint
- lua_text_fromtable (lua_State *L)
- {
- LUA_TRACE_POINT;
- const gchar *delim = "", *st;
- struct rspamd_lua_text *t, *elt;
- gsize textlen = 0, dlen, stlen, tblen;
- gchar *dest;
-
- if (!lua_istable (L, 1)) {
- return luaL_error (L, "invalid arguments");
- }
-
- if (lua_type (L, 2) == LUA_TSTRING) {
- delim = lua_tolstring (L, 2, &dlen);
- }
- else {
- dlen = strlen (delim);
- }
-
- /* Calculate length needed */
- tblen = rspamd_lua_table_size (L, 1);
-
- for (guint i = 0; i < tblen; i ++) {
- lua_rawgeti (L, 1, i + 1);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- #if LUA_VERSION_NUM >= 502
- stlen = lua_rawlen (L, -1);
- #else
- stlen = lua_objlen (L, -1);
- #endif
- textlen += stlen;
- }
- else {
- elt = lua_check_text (L, -1);
-
- if (elt) {
- textlen += elt->len;
- }
- }
-
- lua_pop (L, 1);
- textlen += dlen;
- }
-
- /* Allocate new text */
- t = lua_newuserdata (L, sizeof (*t));
- dest = g_malloc (textlen);
- t->start = dest;
- t->len = textlen;
- t->flags = RSPAMD_TEXT_FLAG_OWN;
- rspamd_lua_setclass (L, "rspamd{text}", -1);
-
- for (guint i = 0; i < tblen; i ++) {
- lua_rawgeti (L, 1, i + 1);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- st = lua_tolstring (L, -1, &stlen);
- memcpy (dest, st, stlen);
- dest += stlen;
- }
- else {
- elt = lua_check_text (L, -1);
-
- if (elt) {
- memcpy (dest, elt->start, elt->len);
- }
- }
-
- memcpy (dest, delim, dlen);
- lua_pop (L, 1);
- }
-
- return 1;
- }
-
- static gint
- lua_text_len (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_text *t = lua_check_text (L, 1);
- gsize l = 0;
-
- if (t != NULL) {
- l = t->len;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushinteger (L, l);
-
- return 1;
- }
-
- static gint
- lua_text_str (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_text *t = lua_check_text (L, 1);
-
- if (t != NULL) {
- lua_pushlstring (L, t->start, t->len);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_text_ptr (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_text *t = lua_check_text (L, 1);
-
- if (t != NULL) {
- lua_pushlightuserdata (L, (gpointer)t->start);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_text_take_ownership (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_text *t = lua_check_text (L, 1);
- gchar *dest;
-
- if (t != NULL) {
- if (t->flags & RSPAMD_TEXT_FLAG_OWN) {
- /* We already own it */
- lua_pushboolean (L, true);
- }
- else {
- dest = g_malloc (t->len);
- memcpy (dest, t->start, t->len);
- t->start = dest;
- t->flags |= RSPAMD_TEXT_FLAG_OWN;
- lua_pushboolean (L, true);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_text_save_in_file (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_text *t = lua_check_text (L, 1);
- const gchar *fname = NULL;
- guint mode = 00644;
- gint fd = -1;
- gboolean need_close = FALSE;
-
- if (t != NULL) {
- if (lua_type (L, 2) == LUA_TSTRING) {
- fname = luaL_checkstring (L, 2);
-
- if (lua_type (L, 3) == LUA_TNUMBER) {
- mode = lua_tonumber (L, 3);
- }
- }
- else if (lua_type (L, 2) == LUA_TNUMBER) {
- /* Created fd */
- fd = lua_tonumber (L, 2);
- }
-
- if (fd == -1) {
- if (fname) {
- fd = rspamd_file_xopen (fname, O_CREAT | O_WRONLY | O_EXCL, mode, 0);
-
- if (fd == -1) {
- lua_pushboolean (L, false);
- lua_pushstring (L, strerror (errno));
-
- return 2;
- }
- need_close = TRUE;
- }
- else {
- fd = STDOUT_FILENO;
- }
- }
-
- if (write (fd, t->start, t->len) == -1) {
- if (fd != STDOUT_FILENO) {
- close (fd);
- }
-
- lua_pushboolean (L, false);
- lua_pushstring (L, strerror (errno));
-
- return 2;
- }
-
- if (need_close) {
- close (fd);
- }
-
- lua_pushboolean (L, true);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_text_gc (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_lua_text *t = lua_check_text (L, 1);
-
- if (t != NULL) {
- if (t->flags & RSPAMD_TEXT_FLAG_OWN) {
- if (t->flags & RSPAMD_TEXT_FLAG_MMAPED) {
- munmap ((gpointer)t->start, t->len);
- }
- else {
- g_free ((gpointer)t->start);
- }
- }
-
- }
-
- return 0;
- }
-
- /* Init part */
-
- static gint
- lua_load_task (lua_State * L)
- {
- lua_newtable (L);
- luaL_register (L, NULL, tasklib_f);
-
- return 1;
- }
-
- static gint
- lua_load_text (lua_State * L)
- {
- lua_newtable (L);
- luaL_register (L, NULL, textlib_f);
-
- return 1;
- }
-
-
- static void
- luaopen_archive (lua_State * L)
- {
- rspamd_lua_new_class (L, "rspamd{archive}", archivelib_m);
- lua_pop (L, 1);
- }
-
- void
- luaopen_task (lua_State * L)
- {
- rspamd_lua_new_class (L, "rspamd{task}", tasklib_m);
- lua_pop (L, 1);
-
- rspamd_lua_add_preload (L, "rspamd_task", lua_load_task);
-
- luaopen_archive (L);
- }
-
- void
- luaopen_image (lua_State * L)
- {
- rspamd_lua_new_class (L, "rspamd{image}", imagelib_m);
- lua_pop (L, 1);
- }
-
- void
- luaopen_text (lua_State *L)
- {
- rspamd_lua_new_class (L, "rspamd{text}", textlib_m);
- lua_pop (L, 1);
-
- rspamd_lua_add_preload (L, "rspamd_text", lua_load_text);
- }
-
- void
- rspamd_lua_task_push (lua_State *L, struct rspamd_task *task)
- {
- struct rspamd_task **ptask;
-
- ptask = lua_newuserdata (L, sizeof (gpointer));
- rspamd_lua_setclass (L, "rspamd{task}", -1);
- *ptask = task;
- }
|