123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128 |
- /*-
- * 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 "libmime/message.h"
- #include "libutil/expression.h"
- #include "libserver/composites.h"
- #include "libserver/cfg_file_private.h"
- #include "libmime/lang_detection.h"
- #include "lua/lua_map.h"
- #include "lua/lua_thread_pool.h"
- #include "utlist.h"
- #include <math.h>
-
- /***
- * This module is used to configure rspamd and is normally available as global
- * variable named `rspamd_config`. Unlike other modules, it is not necessary to
- * require it before usage.
- * @module rspamd_config
- * @example
- -- Register some callback symbol
- local function foo(task)
- -- do something
- end
- rspamd_config:register_symbol('SYMBOL', 1.0, foo)
-
- -- Get configuration
- local tab = rspamd_config:get_all_opt('module') -- get table for module's options
- local opts = rspamd_config:get_key('options') -- get content of the specified key in rspamd configuration
- */
-
- /* Config file methods */
- /***
- * @method rspamd_config:get_module_opt(mname, optname)
- * Returns value of specified option `optname` for a module `mname`,
- * @param {string} mname name of module
- * @param {string} optname option to get
- * @return {string or table} value of the option or `nil` if option is not found
- */
- LUA_FUNCTION_DEF (config, get_module_opt);
- /***
- * @method rspamd_config:get_all_opt(mname)
- * Returns value of all options for a module `mname`, flattening values into a single table consisting
- * of all sections with such a name.
- * @param {string} mname name of module
- * @return {table} table of all options for `mname` or `nil` if a module's configuration is not found
- */
- LUA_FUNCTION_DEF (config, get_all_opt);
-
- /***
- * @method rspamd_config:get_ucl()
- * Returns full configuration as a native Lua object (ucl to lua conversion).
- * This method uses caching if possible.
- * @return {table} table of all options in the configuration
- */
- LUA_FUNCTION_DEF (config, get_ucl);
- /***
- * @method rspamd_config:get_mempool()
- * Returns static configuration memory pool.
- * @return {mempool} [memory pool](mempool.md) object
- */
- LUA_FUNCTION_DEF (config, get_mempool);
- /***
- * @method rspamd_config:get_resolver()
- * Returns DNS resolver.
- * @return {dns_resolver} opaque DNS resolver pointer if any
- */
- LUA_FUNCTION_DEF (config, get_resolver);
- /***
- * @method rspamd_config:add_radix_map(mapline[, description])
- * Creates new dynamic map of IP/mask addresses.
- * @param {string} mapline URL for a map
- * @param {string} description optional map description
- * @return {map} radix tree object
- * @example
- local ip_map = rspamd_config:add_radix_map ('file:///path/to/file', 'my radix map')
- ...
- local function foo(task)
- local ip = task:get_from_ip()
- if ip_map:get_key(ip) then
- return true
- end
- return false
- end
- */
-
- /***
- * @method rspamd_config:radix_from_config(mname, optname)
- * Creates new embedded map of IP/mask addresses from config.
- * @param {string} mname name of module
- * @param {string} optname option to get
- * @return {map} radix tree object
- * @example
- local ip_map = rspamd_config:radix_from_config ('mymodule', 'ips')
- ...
- local function foo(task)
- local ip = task:get_from_ip()
- if ip_map:get_key(ip) then
- return true
- end
- return false
- end
- */
- /***
- * @method rspamd_config:radix_from_ucl(obj)
- * Creates new embedded map of IP/mask addresses from object.
- * @param {ucl} obj object
- * @return {map} radix tree object
- */
- /***
- * @method rspamd_config:add_hash_map(mapline[, description])
- * Creates new dynamic map string objects.
- * @param {string} mapline URL for a map
- * @param {string} description optional map description
- * @return {map} hash set object
- * @example
- local hash_map = rspamd_config:add_hash_map ('file:///path/to/file', 'my hash map')
- ...
- local function foo(task)
- local from = task:get_from()
- if hash_map:get_key(from['user']) then
- return true
- end
- return false
- end
- */
- /***
- * @method rspamd_config:add_kv_map(mapline[, description])
- * Creates new dynamic map of key/values associations.
- * @param {string} mapline URL for a map
- * @param {string} description optional map description
- * @return {map} hash table object
- * @example
- local kv_map = rspamd_config:add_kv_map ('file:///path/to/file', 'my kv map')
- ...
- local function foo(task)
- local from = task:get_from()
- if from then
- local value = kv_map:get_key(from['user'])
- if value then
- return true,value
- end
- end
- return false
- end
- */
- /***
- * @method rspamd_config:add_map({args})
- * Creates new dynamic map according to the attributes passed.
- *
- * - `type`: type of map to be created, can be one of the following set:
- * + `set`: set of strings
- * + `radix`: map of IP addresses to strings
- * + `map`: map of strings to strings
- * + `regexp`: map of regexps to strings
- * + `callback`: map processed by lua callback
- * - `url`: url to load map from
- * - `description`: map's description
- * - `callback`: lua callback for the map
- *
- * @return {map} `true` if map has been added
- * @example
-
- local str = ''
- local function process_map(in)
- str = in
- end
-
- rspamd_config:add_map('http://example.com/map', "settings map", process_map)
- */
- /***
- * @method rspamd_config:get_maps()
- * Get all maps defined as an array of rspamd{map} objects
- *
- * @return {table|rspamd{map}}
- */
- /***
- * @method rspamd_config:get_classifier(name)
- * Returns classifier config.
- * @param {string} name name of classifier (e.g. `bayes`)
- * @return {classifier} classifier object or `nil`
- */
- LUA_FUNCTION_DEF (config, get_classifier);
- /***
- * @method rspamd_config:register_symbol(table)
- * Register symbol of a specified type in rspamd. This function accepts table of arguments:
- *
- * - `name`: name of symbol (can be missing for callback symbols)
- * - `callback`: function to be called for symbol's check (can be absent for virtual symbols)
- * - `weight`: weight of symbol (should normally be 1 or missing)
- * - `priority`: priority of symbol (normally 0 or missing)
- * - `type`: type of symbol: `normal` (default), `virtual` or `callback`
- * - `flags`: various flags split by commas or spaces:
- * + `nice` if symbol can produce negative score;
- * + `empty` if symbol can be called for empty messages
- * + `skip` if symbol should be skipped now
- * + `nostat` if symbol should be excluded from stat tokens
- * + `trivial` symbol is trivial (e.g. no network requests)
- * + `explicit_disable` requires explicit disabling (e.g. via settings)
- * + `ignore_passthrough` executed even if passthrough result has been set
- * - `parent`: id of parent symbol (useful for virtual symbols)
- *
- * @return {number} id of symbol registered
- */
- LUA_FUNCTION_DEF (config, register_symbol);
- /***
- * @method rspamd_config:register_symbols(callback, [weight], callback_name, [, symbol, ...])
- * Register callback function to be called for a set of symbols with initial weight.
- * @param {function} callback callback function to be called for a specified symbol
- * @param {number} weight initial weight of symbol (can be less than zero to specify non-spam symbols)
- * @param {string} callback_name symbolic name of callback
- * @param {list of strings} symbol list of symbols registered by this function
- */
- LUA_FUNCTION_DEF (config, register_symbols);
- /***
- * @method rspamd_config:register_virtual_symbol(name, weight,)
- * Register virtual symbol that is not associated with any callback.
- *
- * **This method is deprecated and should not be used in newly written code **
- * @param {string} virtual name symbol's name
- * @param {number} weight initial weight of symbol (can be less than zero to specify non-spam symbols)
- */
- LUA_FUNCTION_DEF (config, register_virtual_symbol);
- /***
- * @method rspamd_config:register_callback_symbol(name, weight, callback)
- * Register callback function to be called for a specified symbol with initial weight. Symbol itself is
- * not registered in the metric and is not intended to be visible by a user.
- *
- * **This method is deprecated and should not be used in newly written code **
- * @param {string} name symbol's name (just for unique id purposes)
- * @param {number} weight initial weight of symbol (can be less than zero to specify non-spam symbols)
- * @param {function} callback callback function to be called for a specified symbol
- */
- LUA_FUNCTION_DEF (config, register_callback_symbol);
- LUA_FUNCTION_DEF (config, register_callback_symbol_priority);
-
- /***
- * @method rspamd_config:register_dependency(id, dep)
- * Create a dependency between symbol identified by `id` and a symbol identified
- * by some symbolic name `dep`
- * @param {number|string} id id or name of source (numeric id is returned by all register_*_symbol)
- * @param {string} dep dependency name
- * @example
- local function cb(task)
- ...
- end
-
- local id = rspamd_config:register_symbol('SYM', 1.0, cb)
- rspamd_config:register_dependency(id, 'OTHER_SYM')
- -- Alternative form
- rspamd_config:register_dependency('SYMBOL_FROM', 'SYMBOL_TO')
- */
- LUA_FUNCTION_DEF (config, register_dependency);
-
- /***
- * @method rspamd_config:get_symbol_flags(name)
- * Returns symbol flags
- * @param {string} name symbols's name
- * @return {table|string} list of flags for symbol or nil
- */
- LUA_FUNCTION_DEF (config, get_symbol_flags);
-
- /***
- * @method rspamd_config:add_symbol_flags(name, flags)
- * Adds flags to a symbol
- * @param {string} name symbols's name
- * @param {table|string} flags flags to add
- * @return {table|string} new set of flags
- */
- LUA_FUNCTION_DEF (config, add_symbol_flags);
-
- /**
- * @method rspamd_config:register_re_selector(name, selector_str)
- * Registers selector with the specific name to use in regular expressions in form
- * name=/re/$ or name=/re/{selector}
- * @param {string} name name of the selector
- * @param {selector_str} selector string
- * @return true if selector has been registered
- */
- LUA_FUNCTION_DEF (config, register_re_selector);
-
- /**
- * @method rspamd_config:set_symbol({table})
- * Sets the value of a specified symbol in a metric. This function accepts table with the following elements:
- *
- * - `name`: name of symbol (string)
- * - `score`: score for symbol (number)
- * - `metric`: name of metric (string, optional)
- * - `description`: description of symbol (string, optional)
- * - `group`: name of group for symbol (string, optional)
- * - `one_shot`: turn off multiple hits for a symbol (boolean, optional)
- * - `one_param`: turn off multiple options for a symbol (boolean, optional)
- * - `flags`: comma separated string of flags:
- * + `ignore`: do not strictly check validity of symbol and corresponding rule
- * + `one_shot`: turn off multiple hits for a symbol
- * + `one_param`: allow only one parameter for a symbol
- * - `priority`: priority of symbol's definition
- */
- LUA_FUNCTION_DEF (config, set_metric_symbol);
-
- /**
- * @method rspamd_config:set_action({table})
- * Sets the score of a specified action in a metric. This function accepts table with the following elements:
- *
- * - `action`: name of action (string)
- * - `score`: score for action (number)
- * - `metric`: name of metric (string, optional)
- * - `priority`: priority of action's definition
- */
- LUA_FUNCTION_DEF (config, set_metric_action);
-
- /**
- * @method rspamd_config:get_symbol(name)
- * Gets metric data for a specific symbol identified by `name`:
- *
- * - `score`: score for symbol (number)
- * - `description`: description of symbol (string, optional)
- * - `group`: name of group for symbol (string, optional)
- * - `one_shot`: turn off multiple hits for a symbol (boolean, optional)
- * - `flags`: comma separated string of flags:
- * + `ignore`: do not strictly check validity of symbol and corresponding rule
- * + `one_shot`: turn off multiple hits for a symbol
- *
- * @param {string} name name of symbol
- * @return {table} symbol's definition or nil in case of undefined symbol
- */
- LUA_FUNCTION_DEF (config, get_metric_symbol);
-
- /**
- * @method rspamd_config:get_action(name)
- * Gets data for a specific action in config. This function returns number reperesenting action's score
- *
- * @param {string} name name of action
- * @return {number} action's score or nil in case of undefined score or action
- */
- LUA_FUNCTION_DEF (config, get_metric_action);
-
- /**
- * @method rspamd_config:get_all_actions()
- * Gets data for all action in config
- * @return {table|str->num} action's score or nil in case of undefined score or action
- */
- LUA_FUNCTION_DEF (config, get_all_actions);
-
- /**
- * @method rspamd_config:add_composite(name, expression)
- * @param {string} name name of composite symbol
- * @param {string} expression symbolic expression of the composite rule
- * @return {bool} true if a composite has been added successfully
- */
- LUA_FUNCTION_DEF (config, add_composite);
- /***
- * @method rspamd_config:register_pre_filter(callback[, order])
- * Register function to be called prior to symbols processing.
- * @param {function} callback callback function
- * @param {number} order filters are called from lower orders to higher orders, order is equal to 0 by default
- * @example
- local function check_function(task)
- -- It is possible to manipulate the task object here: set settings, set pre-action and so on
- ...
- end
-
- rspamd_config:register_pre_filter(check_function)
- */
- LUA_FUNCTION_DEF (config, register_pre_filter);
- /***
- * @method rspamd_config:register_post_filter(callback[, order])
- * Register function to be called after symbols are processed.
- *
- * @param {function} callback callback function
- * @param {number} order filters are called from lower orders to higher orders, order is equal to 0 by default
- */
- LUA_FUNCTION_DEF (config, register_post_filter);
- /* XXX: obsoleted */
- LUA_FUNCTION_DEF (config, register_module_option);
- /* XXX: not needed now */
- LUA_FUNCTION_DEF (config, get_api_version);
- /***
- * @method rspamd_config:get_key(name)
- * Returns configuration section with the specified `name`.
- * @param {string} name name of config section
- * @return {variant} specific value of section
- * @example
-
- local set_section = rspamd_config:get_key("settings")
- if type(set_section) == "string" then
- -- Just a map of ucl
- if rspamd_config:add_map(set_section, "settings map", process_settings_map) then
- rspamd_config:register_pre_filter(check_settings)
- end
- elseif type(set_section) == "table" then
- if process_settings_table(set_section) then
- rspamd_config:register_pre_filter(check_settings)
- end
- end
- */
- LUA_FUNCTION_DEF (config, get_key);
-
- /***
- * @method rspamd_config:add_condition(symbol, condition)
- * Adds condition callback for specified symbol
- * @param {string} symbol symbol's name
- * @param {function} condition condition callback
- * @return {boolean} true if condition has been added
- * @example
-
- rspamd_config:add_condition('FUZZY_DENIED', function(task)
- if some_map:find_key(task:get_from()) then return false end
- return true
- end)
- */
- LUA_FUNCTION_DEF (config, add_condition);
-
- /***
- * @method rspamd_config:enable_symbol(symbol)
- * Enables execution for the specified symbol
- * @param {string} symbol symbol's name
- */
- LUA_FUNCTION_DEF (config, enable_symbol);
-
- /***
- * @method rspamd_config:disable_symbol(symbol)
- * Disables execution for the specified symbol
- * @param {string} symbol symbol's name
- */
- LUA_FUNCTION_DEF (config, disable_symbol);
-
- /***
- * @method rspamd_config:__newindex(name, callback)
- * This metamethod is called if new indicies are added to the `rspamd_config` object.
- * Technically, it is the equivalent of @see rspamd_config:register_symbol where `weight` is 1.0.
- * There is also table form invocation that allows to control more things:
- *
- * - `callback`: has the same meaning and acts as function of task
- * - `score`: default score for a symbol
- * - `group`: default group for a symbol
- * - `description`: default symbol's description
- * - `priority`: additional priority value
- * - `one_shot`: default value for one shot attribute
- * - `condition`: function of task that can enable or disable this specific rule's execution
- * @param {string} name index name
- * @param {function/table} callback callback to be called
- * @return {number} id of the new symbol added
- * @example
- rspamd_config.R_EMPTY_IMAGE = function (task)
- parts = task:get_text_parts()
- if parts then
- for _,part in ipairs(parts) do
- if part:is_empty() then
- images = task:get_images()
- if images then
- -- Symbol `R_EMPTY_IMAGE` is inserted
- return true
- end
- return false
- end
- end
- end
- return false
- end
-
- rspamd_config.SYMBOL = {
- callback = function(task)
- ...
- end,
- score = 5.1,
- description = 'sample symbol',
- group = 'sample symbols',
- condition = function(task)
- if task:get_from()[1]['addr'] == 'user@example.com' then
- return false
- end
- return true
- end
- }
- */
- LUA_FUNCTION_DEF (config, newindex);
-
- /***
- * @method rspamd_config:register_regexp(params)
- * Registers new re for further cached usage
- * 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
- * + `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
- * - `pcre_only`: flag regexp as pcre only regexp
- */
- LUA_FUNCTION_DEF (config, register_regexp);
-
- /***
- * @method rspamd_config:replace_regexp(params)
- * Replaces regexp with a new one
- * Params is the table with the following fields (mandatory fields are marked with `*`):
- * - `old_re`* : old regular expression object (must be in the cache)
- * - `new_re`* : old regular expression object (must not be in the cache)
- */
- LUA_FUNCTION_DEF (config, replace_regexp);
-
- /***
- * @method rspamd_config:register_worker_script(worker_type, script)
- * Registers the following script for workers of a specified type. The exact type
- * of script function depends on worker type
- * @param {string} worker_type worker type (e.g. "normal")
- * @param {function} script script for a worker
- * @return {boolean} `true` if a script has been registered
- */
- LUA_FUNCTION_DEF (config, register_worker_script);
-
- /***
- * @method rspamd_config:add_on_load(function(cfg, ev_base, worker) ... end)
- * Registers the following script to be executed when configuration is completely loaded
- * @param {function} script function to be executed
- * @example
- rspamd_config:add_on_load(function(cfg, ev_base, worker)
- rspamd_config:add_periodic(ev_base, 1.0, function(cfg, ev_base)
- local logger = require "rspamd_logger"
- logger.infox(cfg, "periodic function in worker %s", worker:get_name())
- return true
- end)
- end)
- */
- LUA_FUNCTION_DEF (config, add_on_load);
-
- /***
- * @method rspamd_config:add_periodic(event_base, timeout, function(cfg, ev_base) ... end, [jitter = false])
- * Registers function to be periodically executed by Rspamd
- * @param {ev_base} event_base event base that is needed for async events
- * @param {number} timeout time in seconds (could be fractional)
- * @param {function} script function to be executed
- * @param {boolean} jitter `true` if timeout jittering is needed
- * @example
- rspamd_config:add_on_load(function(cfg, ev_base)
- rspamd_config:add_periodic(ev_base, 1.0, function(cfg, ev_base)
- local logger = require "rspamd_logger"
- logger.infox(cfg, "periodic function")
- return true -- if return false, then the periodic event is removed
- end)
- end)
- */
- LUA_FUNCTION_DEF (config, add_periodic);
-
- /***
- * @method rspamd_config:get_symbols_count()
- * Returns number of symbols registered in rspamd configuration
- * @return {number} number of symbols registered in the configuration
- */
- LUA_FUNCTION_DEF (config, get_symbols_count);
-
- /***
- * @method rspamd_config:get_symbols_cksum()
- * Returns checksum for all symbols in the cache
- * @return {int64} boxed value of the 64 bit checksum
- */
- LUA_FUNCTION_DEF (config, get_symbols_cksum);
-
- /***
- * @method rspamd_config:get_symbols_counters()
- * Returns table of all counters in the cache (weights, frequencies etc)
- * @return {table|tables} all symbols indexed by name
- */
- LUA_FUNCTION_DEF (config, get_symbols_counters);
-
- /***
- * @method rspamd_config:get_symbols_scores()
- * Returns table of all scores defined in config
- * @return {table|tables} all symbols indexed by name
- */
- LUA_FUNCTION_DEF (config, get_symbols_scores);
-
- /***
- * @method rspamd_config:get_symbol_callback(name)
- * Returns callback function for the specified symbol if it is a lua registered callback
- * @return {function} callback function or nil
- */
- LUA_FUNCTION_DEF (config, get_symbol_callback);
-
- /***
- * @method rspamd_config:get_symbol_stat(name)
- * Returns table with statistics for a specific symbol:
- * - `frequency`: frequency for symbol's hits
- * - `stddev`: standard deviation of `frequency`
- * - `time`: average time in seconds (floating point)
- * - `count`: total number of hits
- * @return {table} symbol stats
- */
- LUA_FUNCTION_DEF (config, get_symbol_stat);
-
- /***
- * @method rspamd_config:set_symbol_callback(name, callback)
- * Sets callback for the specified symbol
- * @return {boolean} true if function has been replaced
- */
- LUA_FUNCTION_DEF (config, set_symbol_callback);
-
- /***
- * @method rspamd_config:register_finish_script(callback)
- * Adds new callback that is called on worker process termination when all
- * tasks pending are processed
- *
- * @param callback {function} a function with one argument (rspamd_task)
- */
- LUA_FUNCTION_DEF (config, register_finish_script);
-
- /***
- * @method rspamd_config:register_monitored(url, type, [{params}])
- * Registers monitored resource to watch its availability. Supported types:
- *
- * - `dns`: DNS monitored object
- *
- * Params are optional table specific for each type. For DNS it supports the
- * following options:
- *
- * - `prefix`: prefix to add before making request
- * - `type`: type of request (e.g. 'a' or 'txt')
- * - `ipnet`: array of ip/networks to expect on reply
- * - `rcode`: expected return code (e.g. `nxdomain`)
- *
- * Returned object has the following methods:
- *
- * - `alive`: returns `true` if monitored resource is alive
- * - `offline`: returns number of seconds of the current offline period (or 0 if alive)
- * - `total_offline`: returns number of seconds of the overall offline
- * - `latency`: returns the current average latency in seconds (or 0 if offline)
- *
- * @param {string} url resource to monitor
- * @param {string} type type of monitoring
- * @param {table} opts optional parameters
- * @return {rspamd_monitored} rspamd monitored object
- */
- LUA_FUNCTION_DEF (config, register_monitored);
-
- /***
- * @method rspamd_config:add_doc(path, option, doc_string, [{params}])
- * Adds new documentation string for an option `option` at path `path`
- * Options defines optional params, such as:
- *
- * - `default`: default option value
- * - `type`: type of an option (`string`, `number`, `object`, `array` etc)
- * - `reqired`: if an option is required
- *
- * @param {string} path documentation path (e.g. module name)
- * @param {string} option name of the option
- * @param {string} doc_string documentation string
- * @param {table} params optional parameters
- */
- LUA_FUNCTION_DEF (config, add_doc);
-
- /***
- * @method rspamd_config:add_example(path, option, doc_string, example)
- * Adds new documentation
- *
- * @param {string} path documentation path (e.g. module name or nil for top)
- * @param {string} option name of the option
- * @param {string} doc_string documentation string
- * @param {string} example example in ucl format, comments are also parsed
- */
- LUA_FUNCTION_DEF (config, add_example);
-
- /***
- * @method rspamd_config:set_peak_cb(function)
- * Sets a function that will be called when frequency of some symbol goes out of
- * stddev * 2 over the last period of refreshment.
- *
- * @example
- rspamd_config:set_peak_cb(function(ev_base, sym, mean, stddev, value, error)
- -- ev_base: event base for async events (e.g. redis)
- -- sym: symbol's name
- -- mean: mean frequency value
- -- stddev: standard deviation of frequency
- -- value: current frequency value
- -- error: squared error
- local logger = require "rspamd_logger"
- logger.infox(rspamd_config, "symbol %s has changed frequency significantly: %s(%s) over %s(%s)",
- sym, value, error, mean, stddev)
- end)
- */
- LUA_FUNCTION_DEF (config, set_peak_cb);
- /***
- * @method rspamd_config:get_cpu_flags()
- * Returns architecture dependent flags supported by the CPU
- * Currently, only x86 flags are supported:
- * - 'ssse3'
- * - 'sse42'
- * - 'avx'
- * - 'avx2'
- * @return {table} flag -> true table
- */
- LUA_FUNCTION_DEF (config, get_cpu_flags);
-
- /***
- * @method rspamd_config:has_torch()
- * Returns true if Rspamd is compiled with torch support and the runtime CPU
- * supports sse4.2 required for torch.
- * @return {boolean} true if torch is compiled and supported
- */
- LUA_FUNCTION_DEF (config, has_torch);
-
- /***
- * @method rspamd_config:experimental_enabled()
- * Returns true if experimental plugins are enabled
- * @return {boolean} true if experimental plugins are enabled
- */
- LUA_FUNCTION_DEF (config, experimental_enabled);
-
- /***
- * @method rspamd_config:load_ucl(filename[, include_trace])
- * Loads config from the UCL file (but does not perform parsing using rcl)
- * @param {string} filename file to load
- * @return true or false + error message
- */
- LUA_FUNCTION_DEF (config, load_ucl);
-
- /***
- * @method rspamd_config:parse_rcl([skip_sections])
- * Parses RCL using loaded ucl file
- * @param {table|string} sections to skip
- * @return true or false + error message
- */
- LUA_FUNCTION_DEF (config, parse_rcl);
-
- /***
- * @method rspamd_config:init_modules()
- * Initialize lua and internal modules
- * @return true or false
- */
- LUA_FUNCTION_DEF (config, init_modules);
-
- /***
- * @method rspamd_config:init_subsystem(str)
- * Initialize config subsystem from a comma separated list:
- * - `modules` - init modules
- * - `langdet` - language detector
- * - `dns` - DNS resolver
- * - TODO: add more
- */
- LUA_FUNCTION_DEF (config, init_subsystem);
-
- /***
- * @method rspamd_config:get_tld_path()
- * Returns path to TLD file
- * @return {string} path to tld file
- */
- LUA_FUNCTION_DEF (config, get_tld_path);
-
- static const struct luaL_reg configlib_m[] = {
- LUA_INTERFACE_DEF (config, get_module_opt),
- LUA_INTERFACE_DEF (config, get_mempool),
- LUA_INTERFACE_DEF (config, get_resolver),
- LUA_INTERFACE_DEF (config, get_all_opt),
- LUA_INTERFACE_DEF (config, get_ucl),
- LUA_INTERFACE_DEF (config, add_radix_map),
- LUA_INTERFACE_DEF (config, radix_from_config),
- LUA_INTERFACE_DEF (config, radix_from_ucl),
- LUA_INTERFACE_DEF (config, add_hash_map),
- LUA_INTERFACE_DEF (config, add_kv_map),
- LUA_INTERFACE_DEF (config, add_map),
- LUA_INTERFACE_DEF (config, get_maps),
- LUA_INTERFACE_DEF (config, get_classifier),
- LUA_INTERFACE_DEF (config, register_symbol),
- LUA_INTERFACE_DEF (config, register_symbols),
- LUA_INTERFACE_DEF (config, register_virtual_symbol),
- LUA_INTERFACE_DEF (config, register_callback_symbol),
- LUA_INTERFACE_DEF (config, register_callback_symbol_priority),
- LUA_INTERFACE_DEF (config, register_dependency),
- LUA_INTERFACE_DEF (config, get_symbol_flags),
- LUA_INTERFACE_DEF (config, add_symbol_flags),
- LUA_INTERFACE_DEF (config, set_metric_symbol),
- {"set_symbol", lua_config_set_metric_symbol},
- LUA_INTERFACE_DEF (config, set_metric_action),
- {"set_action", lua_config_set_metric_action},
- LUA_INTERFACE_DEF (config, get_metric_symbol),
- {"get_symbol", lua_config_get_metric_symbol},
- LUA_INTERFACE_DEF (config, get_metric_action),
- {"get_action", lua_config_get_metric_action},
- LUA_INTERFACE_DEF (config, get_all_actions),
- LUA_INTERFACE_DEF (config, add_composite),
- LUA_INTERFACE_DEF (config, register_module_option),
- LUA_INTERFACE_DEF (config, register_pre_filter),
- LUA_INTERFACE_DEF (config, register_post_filter),
- LUA_INTERFACE_DEF (config, get_api_version),
- LUA_INTERFACE_DEF (config, get_key),
- LUA_INTERFACE_DEF (config, add_condition),
- LUA_INTERFACE_DEF (config, enable_symbol),
- LUA_INTERFACE_DEF (config, disable_symbol),
- LUA_INTERFACE_DEF (config, register_regexp),
- LUA_INTERFACE_DEF (config, replace_regexp),
- LUA_INTERFACE_DEF (config, register_worker_script),
- LUA_INTERFACE_DEF (config, register_re_selector),
- LUA_INTERFACE_DEF (config, add_on_load),
- LUA_INTERFACE_DEF (config, add_periodic),
- LUA_INTERFACE_DEF (config, get_symbols_count),
- LUA_INTERFACE_DEF (config, get_symbols_cksum),
- LUA_INTERFACE_DEF (config, get_symbols_counters),
- LUA_INTERFACE_DEF (config, get_symbols_scores),
- LUA_INTERFACE_DEF (config, get_symbol_callback),
- LUA_INTERFACE_DEF (config, set_symbol_callback),
- LUA_INTERFACE_DEF (config, get_symbol_stat),
- LUA_INTERFACE_DEF (config, register_finish_script),
- LUA_INTERFACE_DEF (config, register_monitored),
- LUA_INTERFACE_DEF (config, add_doc),
- LUA_INTERFACE_DEF (config, add_example),
- LUA_INTERFACE_DEF (config, set_peak_cb),
- LUA_INTERFACE_DEF (config, get_cpu_flags),
- LUA_INTERFACE_DEF (config, has_torch),
- LUA_INTERFACE_DEF (config, experimental_enabled),
- LUA_INTERFACE_DEF (config, load_ucl),
- LUA_INTERFACE_DEF (config, parse_rcl),
- LUA_INTERFACE_DEF (config, init_modules),
- LUA_INTERFACE_DEF (config, init_subsystem),
- LUA_INTERFACE_DEF (config, get_tld_path),
- {"__tostring", rspamd_lua_class_tostring},
- {"__newindex", lua_config_newindex},
- {NULL, NULL}
- };
-
- LUA_FUNCTION_DEF (monitored, alive);
- LUA_FUNCTION_DEF (monitored, latency);
- LUA_FUNCTION_DEF (monitored, offline);
- LUA_FUNCTION_DEF (monitored, total_offline);
-
- static const struct luaL_reg monitoredlib_m[] = {
- LUA_INTERFACE_DEF (monitored, alive),
- LUA_INTERFACE_DEF (monitored, latency),
- LUA_INTERFACE_DEF (monitored, offline),
- LUA_INTERFACE_DEF (monitored, total_offline),
- {"__tostring", rspamd_lua_class_tostring},
- {NULL, NULL}
- };
-
- static const guint64 rspamd_lua_callback_magic = 0x32c118af1e3263c7ULL;
-
- struct rspamd_config *
- lua_check_config (lua_State * L, gint pos)
- {
- void *ud = rspamd_lua_check_udata (L, pos, "rspamd{config}");
- luaL_argcheck (L, ud != NULL, pos, "'config' expected");
- return ud ? *((struct rspamd_config **)ud) : NULL;
- }
-
- static struct rspamd_monitored *
- lua_check_monitored (lua_State * L, gint pos)
- {
- void *ud = rspamd_lua_check_udata (L, pos, "rspamd{monitored}");
- luaL_argcheck (L, ud != NULL, pos, "'monitored' expected");
- return ud ? *((struct rspamd_monitored **)ud) : NULL;
- }
-
- /*** Config functions ***/
- static gint
- lua_config_get_api_version (lua_State *L)
- {
- msg_warn ("get_api_version is deprecated, do not use it");
- lua_pushnumber (L, 100);
-
- return 1;
- }
-
- static gint
- lua_config_get_module_opt (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *mname, *optname;
- const ucl_object_t *obj;
-
- if (cfg) {
- mname = luaL_checkstring (L, 2);
- optname = luaL_checkstring (L, 3);
-
- if (mname && optname) {
- obj = rspamd_config_get_module_opt (cfg, mname, optname);
- if (obj) {
- return ucl_object_push_lua (L, obj, TRUE);
- }
- }
- }
- lua_pushnil (L);
- return 1;
- }
-
- static int
- lua_config_get_mempool (lua_State * L)
- {
- LUA_TRACE_POINT;
- rspamd_mempool_t **ppool;
- struct rspamd_config *cfg = lua_check_config (L, 1);
-
- if (cfg != NULL) {
- ppool = lua_newuserdata (L, sizeof (rspamd_mempool_t *));
- rspamd_lua_setclass (L, "rspamd{mempool}", -1);
- *ppool = cfg->cfg_pool;
- }
- else {
- lua_pushnil (L);
- }
- return 1;
- }
-
- static int
- lua_config_get_resolver (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_dns_resolver **pres;
- struct rspamd_config *cfg = lua_check_config (L, 1);
-
- if (cfg != NULL && cfg->dns_resolver) {
- pres = lua_newuserdata (L, sizeof (*pres));
- rspamd_lua_setclass (L, "rspamd{resolver}", -1);
- *pres = cfg->dns_resolver;
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- static gint
- lua_config_get_all_opt (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *mname;
- const ucl_object_t *obj, *cur, *cur_elt;
- ucl_object_iter_t it = NULL;
- gint i;
-
- if (cfg) {
- mname = luaL_checkstring (L, 2);
-
- if (mname) {
- obj = ucl_obj_get_key (cfg->rcl_obj, mname);
- /* Flatten object */
- if (obj != NULL && (ucl_object_type (obj) == UCL_OBJECT ||
- ucl_object_type (obj) == UCL_ARRAY)) {
-
- lua_newtable (L);
- it = ucl_object_iterate_new (obj);
-
- LL_FOREACH (obj, cur) {
- it = ucl_object_iterate_reset (it, cur);
-
- while ((cur_elt = ucl_object_iterate_safe (it, true))) {
- lua_pushstring (L, ucl_object_key (cur_elt));
- ucl_object_push_lua (L, cur_elt, true);
- lua_settable (L, -3);
- }
- }
-
- ucl_object_iterate_free (it);
-
- return 1;
- }
- else if (obj != NULL) {
- lua_newtable (L);
- i = 1;
-
- LL_FOREACH (obj, cur) {
- lua_pushinteger (L, i++);
- ucl_object_push_lua (L, cur, true);
- lua_settable (L, -3);
- }
-
- return 1;
- }
- }
- }
- lua_pushnil (L);
-
- return 1;
- }
-
- struct rspamd_lua_cached_config {
- lua_State *L;
- gint ref;
- };
-
- static void
- lua_config_ucl_dtor (gpointer p)
- {
- struct rspamd_lua_cached_config *cached = p;
-
- luaL_unref (cached->L, LUA_REGISTRYINDEX, cached->ref);
- }
-
- static gint
- lua_config_get_ucl (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_lua_cached_config *cached;
-
- if (cfg) {
- cached = rspamd_mempool_get_variable (cfg->cfg_pool, "ucl_cached");
-
- if (cached) {
- lua_rawgeti (L, LUA_REGISTRYINDEX, cached->ref);
- }
- else {
- ucl_object_push_lua (L, cfg->rcl_obj, true);
- lua_pushvalue (L, -1);
- cached = rspamd_mempool_alloc (cfg->cfg_pool, sizeof (*cached));
- cached->L = L;
- cached->ref = luaL_ref (L, LUA_REGISTRYINDEX);
- rspamd_mempool_set_variable (cfg->cfg_pool, "ucl_cached",
- cached, lua_config_ucl_dtor);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
-
- static gint
- lua_config_get_classifier (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_classifier_config *clc = NULL, **pclc = NULL;
- const gchar *name;
- GList *cur;
-
- if (cfg) {
- name = luaL_checkstring (L, 2);
-
- cur = g_list_first (cfg->classifiers);
- while (cur) {
- clc = cur->data;
- if (g_ascii_strcasecmp (clc->name, name) == 0) {
- pclc = &clc;
- break;
- }
- cur = g_list_next (cur);
- }
- if (pclc) {
- pclc = lua_newuserdata (L,
- sizeof (struct rspamd_classifier_config *));
- rspamd_lua_setclass (L, "rspamd{classifier}", -1);
- *pclc = clc;
- return 1;
- }
- }
-
- lua_pushnil (L);
- return 1;
-
- }
-
- struct lua_callback_data {
- guint64 magic;
- lua_State *L;
- gchar *symbol;
-
- union {
- gchar *name;
- gint ref;
- } callback;
- gboolean cb_is_ref;
-
- /* Dynamic data */
- gint stack_level;
- gint order;
- struct rspamd_symcache_item *item;
- };
-
- /*
- * Unref symbol if it is local reference
- */
- static void
- lua_destroy_cfg_symbol (gpointer ud)
- {
- struct lua_callback_data *cd = ud;
-
- /* Unref callback */
- if (cd->cb_is_ref) {
- luaL_unref (cd->L, LUA_REGISTRYINDEX, cd->callback.ref);
- }
- }
-
- static gint
- lua_config_register_module_option (lua_State *L)
- {
- return 0;
- }
-
- static gint
- rspamd_compare_order_func (gconstpointer a, gconstpointer b)
- {
- const struct lua_callback_data *cb1 = a, *cb2 = b;
-
- /* order of call goes from lower to higher */
- return cb2->order - cb1->order;
- }
-
- static void
- lua_metric_symbol_callback (struct rspamd_task *task,
- struct rspamd_symcache_item *item,
- gpointer ud)
- {
- struct lua_callback_data *cd = ud;
- struct rspamd_task **ptask;
- gint level = lua_gettop (cd->L), nresults, err_idx, ret;
- lua_State *L = cd->L;
- GString *tb;
- struct rspamd_symbol_result *s;
-
- cd->item = item;
- rspamd_symcache_item_async_inc (task, item, "lua symbol");
- lua_pushcfunction (L, &rspamd_lua_traceback);
- err_idx = lua_gettop (L);
-
- level ++;
-
- if (cd->cb_is_ref) {
- lua_rawgeti (L, LUA_REGISTRYINDEX, cd->callback.ref);
- }
- else {
- lua_getglobal (L, cd->callback.name);
- }
-
- ptask = lua_newuserdata (L, sizeof (struct rspamd_task *));
- rspamd_lua_setclass (L, "rspamd{task}", -1);
- *ptask = task;
-
- if ((ret = lua_pcall (L, 1, LUA_MULTRET, err_idx)) != 0) {
- tb = lua_touserdata (L, -1);
- msg_err_task ("call to (%s) failed (%d): %v", cd->symbol, ret, tb);
-
- if (tb) {
- g_string_free (tb, TRUE);
- lua_pop (L, 1);
- }
- }
- else {
- nresults = lua_gettop (L) - level;
-
- if (nresults >= 1) {
- /* Function returned boolean, so maybe we need to insert result? */
- gint res = 0;
- gint i;
- gdouble flag = 1.0;
- gint type;
-
- type = lua_type (cd->L, level + 1);
-
- if (type == LUA_TBOOLEAN) {
- res = lua_toboolean (L, level + 1);
- }
- else if (type == LUA_TNUMBER) {
- res = lua_tonumber (L, level + 1);
- }
- else if (type == LUA_TNIL) {
- /* Can happen sometimes... */
- res = FALSE;
- }
- else {
- g_assert_not_reached ();
- }
-
- if (res) {
- gint first_opt = 2;
-
- if (lua_type (L, level + 2) == LUA_TNUMBER) {
- flag = lua_tonumber (L, level + 2);
- /* Shift opt index */
- first_opt = 3;
- }
- else {
- flag = res;
- }
-
- s = rspamd_task_insert_result (task, cd->symbol, flag, NULL);
-
- if (s) {
- guint last_pos = lua_gettop (L);
-
- for (i = level + first_opt; i <= last_pos; i++) {
- if (lua_type (L, i) == LUA_TSTRING) {
- const char *opt = lua_tostring (L, i);
-
- rspamd_task_add_result_option (task, s, opt);
- }
- else if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
-
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- const char *opt = lua_tostring (L, -1);
-
- rspamd_task_add_result_option (task, s, opt);
- }
-
- lua_pop (L, 1);
- }
- }
- }
-
- }
-
- lua_pop (L, nresults);
- }
- }
-
- lua_pop (L, 1); /* Error function */
- rspamd_symcache_item_async_dec_check (task, cd->item, "lua symbol");
- g_assert (lua_gettop (L) == level - 1);
- }
-
- static void lua_metric_symbol_callback_return (struct thread_entry *thread_entry,
- int ret);
-
- static void lua_metric_symbol_callback_error (struct thread_entry *thread_entry,
- int ret,
- const char *msg);
-
- static void
- lua_metric_symbol_callback_coro (struct rspamd_task *task,
- struct rspamd_symcache_item *item,
- gpointer ud)
- {
- struct lua_callback_data *cd = ud;
- struct rspamd_task **ptask;
- struct thread_entry *thread_entry;
-
- rspamd_symcache_item_async_inc (task, item, "lua coro symbol");
- thread_entry = lua_thread_pool_get_for_task (task);
-
- g_assert(thread_entry->cd == NULL);
- thread_entry->cd = cd;
-
- lua_State *thread = thread_entry->lua_state;
- cd->stack_level = lua_gettop (thread);
- cd->item = item;
-
- if (cd->cb_is_ref) {
- lua_rawgeti (thread, LUA_REGISTRYINDEX, cd->callback.ref);
- }
- else {
- lua_getglobal (thread, cd->callback.name);
- }
-
- ptask = lua_newuserdata (thread, sizeof (struct rspamd_task *));
- rspamd_lua_setclass (thread, "rspamd{task}", -1);
- *ptask = task;
-
- thread_entry->finish_callback = lua_metric_symbol_callback_return;
- thread_entry->error_callback = lua_metric_symbol_callback_error;
-
- lua_thread_call (thread_entry, 1);
- }
-
- static void
- lua_metric_symbol_callback_error (struct thread_entry *thread_entry,
- int ret,
- const char *msg)
- {
- struct lua_callback_data *cd = thread_entry->cd;
- struct rspamd_task *task = thread_entry->task;
- msg_err_task ("call to coroutine (%s) failed (%d): %s", cd->symbol, ret, msg);
-
- rspamd_symcache_item_async_dec_check (task, cd->item, "lua coro symbol");
- }
-
- static void
- lua_metric_symbol_callback_return (struct thread_entry *thread_entry, int ret)
- {
- struct lua_callback_data *cd = thread_entry->cd;
- struct rspamd_task *task = thread_entry->task;
- int nresults;
- struct rspamd_symbol_result *s;
-
- (void)ret;
-
- lua_State *L = thread_entry->lua_state;
-
- nresults = lua_gettop (L) - cd->stack_level;
-
- if (nresults >= 1) {
- /* Function returned boolean, so maybe we need to insert result? */
- gint res = 0;
- gint i;
- gdouble flag = 1.0;
- gint type;
-
- type = lua_type (L, cd->stack_level + 1);
-
- if (type == LUA_TBOOLEAN) {
- res = lua_toboolean (L, cd->stack_level + 1);
- }
- else if (type == LUA_TFUNCTION) {
- g_assert_not_reached ();
- }
- else {
- res = lua_tonumber (L, cd->stack_level + 1);
- }
-
- if (res) {
- gint first_opt = 2;
-
- if (lua_type (L, cd->stack_level + 2) == LUA_TNUMBER) {
- flag = lua_tonumber (L, cd->stack_level + 2);
- /* Shift opt index */
- first_opt = 3;
- }
- else {
- flag = res;
- }
-
- s = rspamd_task_insert_result (task, cd->symbol, flag, NULL);
-
- if (s) {
- guint last_pos = lua_gettop (L);
-
- for (i = cd->stack_level + first_opt; i <= last_pos; i++) {
- if (lua_type (L, i) == LUA_TSTRING) {
- const char *opt = lua_tostring (L, i);
-
- rspamd_task_add_result_option (task, s, opt);
- }
- else if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
-
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- const char *opt = lua_tostring (L, -1);
-
- rspamd_task_add_result_option (task, s, opt);
- }
-
- lua_pop (L, 1);
- }
- }
- }
-
- }
-
- lua_pop (L, nresults);
- }
-
- g_assert (lua_gettop (L) == cd->stack_level); /* we properly cleaned up the stack */
-
- cd->stack_level = 0;
- rspamd_symcache_item_async_dec_check (task, cd->item, "lua coro symbol");
- }
-
- static gint
- rspamd_register_symbol_fromlua (lua_State *L,
- struct rspamd_config *cfg,
- const gchar *name,
- gint ref,
- gdouble weight,
- gint priority,
- enum rspamd_symbol_type type,
- gint parent,
- gboolean optional,
- gboolean no_squeeze)
- {
- struct lua_callback_data *cd;
- gint ret = -1, err_idx;
-
- if (priority == 0 && weight < 0) {
- priority = 1;
- }
-
- if ((ret = rspamd_symcache_find_symbol (cfg->cache, name)) != -1) {
- if (optional) {
- msg_debug_config ("duplicate symbol: %s, skip registering", name);
-
- return ret;
- }
- else {
- msg_err_config ("duplicate symbol: %s, skip registering", name);
-
- return -1;
- }
- }
-
- if (ref != -1) {
- if (type & SYMBOL_TYPE_USE_CORO) {
- /* Coroutines are incompatible with squeezing */
- no_squeeze = TRUE;
- }
- /*
- * We call for routine called lua_squeeze_rules.squeeze_rule if it exists
- */
- lua_pushcfunction (L, &rspamd_lua_traceback);
- err_idx = lua_gettop (L);
-
- if (!no_squeeze && (type & (SYMBOL_TYPE_CALLBACK|SYMBOL_TYPE_NORMAL)) &&
- rspamd_lua_require_function (L, "lua_squeeze_rules", "squeeze_rule")) {
- if (name) {
- lua_pushstring (L, name);
- }
- else {
- lua_pushnil (L);
- }
-
- /* Push function reference */
- lua_rawgeti (L, LUA_REGISTRYINDEX, ref);
-
- /* Flags */
- lua_createtable (L, 0, 0);
-
- if (type & SYMBOL_TYPE_MIME_ONLY) {
- lua_pushstring (L, "mime");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (type & SYMBOL_TYPE_FINE) {
- lua_pushstring (L, "fine");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (type & SYMBOL_TYPE_NOSTAT) {
- lua_pushstring (L, "nostat");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (type & SYMBOL_TYPE_EXPLICIT_DISABLE) {
- lua_pushstring (L, "explicit_disable");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
-
- /* Now call for squeeze function */
- if (lua_pcall (L, 3, 1, err_idx) != 0) {
- GString *tb = lua_touserdata (L, -1);
- msg_err_config ("call to squeeze_rule failed: %v", tb);
-
- if (tb) {
- g_string_free (tb, TRUE);
- }
- }
-
- ret = lua_tonumber (L, -1);
-
- if (ret == -1) {
- /* Do direct registration */
- cd = rspamd_mempool_alloc0 (cfg->cfg_pool,
- sizeof (struct lua_callback_data));
- cd->magic = rspamd_lua_callback_magic;
- cd->cb_is_ref = TRUE;
- cd->callback.ref = ref;
- cd->L = L;
-
- if (name) {
- cd->symbol = rspamd_mempool_strdup (cfg->cfg_pool, name);
- }
-
- if (type & SYMBOL_TYPE_USE_CORO) {
- ret = rspamd_symcache_add_symbol (cfg->cache,
- name,
- priority,
- lua_metric_symbol_callback_coro,
- cd,
- type,
- parent);
- }
- else {
- ret = rspamd_symcache_add_symbol (cfg->cache,
- name,
- priority,
- lua_metric_symbol_callback,
- cd,
- type,
- parent);
- }
-
- rspamd_mempool_add_destructor (cfg->cfg_pool,
- (rspamd_mempool_destruct_t) lua_destroy_cfg_symbol,
- cd);
- }
- }
- else {
- cd = rspamd_mempool_alloc0 (cfg->cfg_pool,
- sizeof (struct lua_callback_data));
- cd->magic = rspamd_lua_callback_magic;
- cd->cb_is_ref = TRUE;
- cd->callback.ref = ref;
- cd->L = L;
-
- if (name) {
- cd->symbol = rspamd_mempool_strdup (cfg->cfg_pool, name);
- }
-
- if (type & SYMBOL_TYPE_USE_CORO) {
- ret = rspamd_symcache_add_symbol (cfg->cache,
- name,
- priority,
- lua_metric_symbol_callback_coro,
- cd,
- type,
- parent);
- }
- else {
- ret = rspamd_symcache_add_symbol (cfg->cache,
- name,
- priority,
- lua_metric_symbol_callback,
- cd,
- type,
- parent);
- }
- rspamd_mempool_add_destructor (cfg->cfg_pool,
- (rspamd_mempool_destruct_t)lua_destroy_cfg_symbol,
- cd);
- }
-
- /* Cleanup lua stack */
- lua_settop (L, err_idx - 1);
- }
- else {
- ret = rspamd_symcache_add_symbol (cfg->cache,
- name,
- priority,
- NULL,
- NULL,
- type,
- parent);
- }
-
- return ret;
- }
-
- static gint
- lua_config_register_post_filter (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- gint order = 0, cbref, ret;
-
- if (cfg) {
- if (lua_type (L, 3) == LUA_TNUMBER) {
- order = lua_tonumber (L, 3);
- }
-
- if (lua_type (L, 2) == LUA_TFUNCTION) {
- lua_pushvalue (L, 2);
- /* Get a reference */
- cbref = luaL_ref (L, LUA_REGISTRYINDEX);
- }
- else {
- return luaL_error (L, "invalid type for callback: %s",
- lua_typename (L, lua_type (L, 2)));
- }
-
- msg_warn_config ("register_post_filter function is deprecated, "
- "use register_symbol instead");
-
- ret = rspamd_register_symbol_fromlua (L,
- cfg,
- NULL,
- cbref,
- 1.0,
- order,
- SYMBOL_TYPE_POSTFILTER|SYMBOL_TYPE_CALLBACK,
- -1,
- FALSE,
- TRUE);
-
- lua_pushboolean (L, ret);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_register_pre_filter (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- gint order = 0, cbref, ret;
-
- if (cfg) {
- if (lua_type (L, 3) == LUA_TNUMBER) {
- order = lua_tonumber (L, 3);
- }
-
- if (lua_type (L, 2) == LUA_TFUNCTION) {
- lua_pushvalue (L, 2);
- /* Get a reference */
- cbref = luaL_ref (L, LUA_REGISTRYINDEX);
- }
- else {
- return luaL_error (L, "invalid type for callback: %s",
- lua_typename (L, lua_type (L, 2)));
- }
-
- msg_warn_config ("register_pre_filter function is deprecated, "
- "use register_symbol instead");
-
- ret = rspamd_register_symbol_fromlua (L,
- cfg,
- NULL,
- cbref,
- 1.0,
- order,
- SYMBOL_TYPE_PREFILTER|SYMBOL_TYPE_CALLBACK,
- -1,
- FALSE,
- TRUE);
-
- lua_pushboolean (L, ret);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_get_key (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name;
- size_t namelen;
- const ucl_object_t *val;
-
- name = luaL_checklstring(L, 2, &namelen);
- if (name && cfg) {
- val = ucl_object_lookup_len(cfg->rcl_obj, name, namelen);
- if (val != NULL) {
- ucl_object_push_lua (L, val, val->type != UCL_ARRAY);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_parse_symbol_flags (const gchar *str)
- {
- int ret = 0;
-
- if (str) {
- if (strstr (str, "fine") != NULL) {
- ret |= SYMBOL_TYPE_FINE;
- }
- if (strstr (str, "nice") != NULL) {
- ret |= SYMBOL_TYPE_FINE;
- }
- if (strstr (str, "empty") != NULL) {
- ret |= SYMBOL_TYPE_EMPTY;
- }
- if (strstr (str, "skip") != NULL) {
- ret |= SYMBOL_TYPE_SKIPPED;
- }
- if (strstr (str, "nostat") != NULL) {
- ret |= SYMBOL_TYPE_NOSTAT;
- }
- if (strstr (str, "idempotent") != NULL) {
- ret |= SYMBOL_TYPE_IDEMPOTENT;
- }
- if (strstr (str, "squeezed") != NULL) {
- ret |= SYMBOL_TYPE_SQUEEZED;
- }
- if (strstr (str, "trivial") != NULL) {
- ret |= SYMBOL_TYPE_TRIVIAL;
- }
- if (strstr (str, "mime") != NULL) {
- ret |= SYMBOL_TYPE_MIME_ONLY;
- }
- if (strstr (str, "ignore_passthrough") != NULL) {
- ret |= SYMBOL_TYPE_IGNORE_PASSTHROUGH;
- }
- if (strstr (str, "explicit_disable") != NULL) {
- ret |= SYMBOL_TYPE_EXPLICIT_DISABLE;
- }
- if (strstr (str, "coro") != NULL) {
- ret |= SYMBOL_TYPE_USE_CORO;
- }
- }
-
- return ret;
- }
-
- static gint
- lua_parse_symbol_type (const gchar *str)
- {
- gint ret = SYMBOL_TYPE_NORMAL;
- gchar **vec;
- guint i, l;
-
- if (str) {
- vec = g_strsplit_set (str, ",;", -1);
-
- if (vec) {
- l = g_strv_length (vec);
-
- for (i = 0; i < l; i ++) {
- str = vec[i];
-
- if (g_ascii_strcasecmp (str, "virtual") == 0) {
- ret = SYMBOL_TYPE_VIRTUAL;
- } else if (g_ascii_strcasecmp (str, "callback") == 0) {
- ret = SYMBOL_TYPE_CALLBACK;
- } else if (g_ascii_strcasecmp (str, "normal") == 0) {
- ret = SYMBOL_TYPE_NORMAL;
- } else if (g_ascii_strcasecmp (str, "prefilter") == 0) {
- ret = SYMBOL_TYPE_PREFILTER | SYMBOL_TYPE_GHOST;
- } else if (g_ascii_strcasecmp (str, "postfilter") == 0) {
- ret = SYMBOL_TYPE_POSTFILTER | SYMBOL_TYPE_GHOST;
- } else if (g_ascii_strcasecmp (str, "idempotent") == 0) {
- ret = SYMBOL_TYPE_POSTFILTER | SYMBOL_TYPE_GHOST |
- SYMBOL_TYPE_IDEMPOTENT;
- } else {
- gint fl = 0;
-
- fl = lua_parse_symbol_flags (str);
-
- if (fl == 0) {
- msg_warn ("bad type: %s", str);
- }
- else {
- ret |= fl;
- }
- }
- }
-
- g_strfreev (vec);
- }
- }
-
- return ret;
- }
-
- static void
- lua_push_symbol_flags (lua_State *L, guint flags)
- {
- guint i = 1;
-
- lua_newtable (L);
-
- if (flags & SYMBOL_TYPE_FINE) {
- lua_pushstring (L, "fine");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_EMPTY) {
- lua_pushstring (L, "empty");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_SQUEEZED) {
- lua_pushstring (L, "squeezed");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_EXPLICIT_DISABLE) {
- lua_pushstring (L, "explicit_disable");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_IGNORE_PASSTHROUGH) {
- lua_pushstring (L, "ignore_passthrough");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_NOSTAT) {
- lua_pushstring (L, "nostat");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_IDEMPOTENT) {
- lua_pushstring (L, "idempotent");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_MIME_ONLY) {
- lua_pushstring (L, "mime");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_TRIVIAL) {
- lua_pushstring (L, "trivial");
- lua_rawseti (L, -2, i++);
- }
-
- if (flags & SYMBOL_TYPE_SKIPPED) {
- lua_pushstring (L, "skip");
- lua_rawseti (L, -2, i++);
- }
- }
-
- static gint
- lua_config_get_symbol_flags (lua_State *L)
- {
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name = luaL_checkstring (L, 2);
- guint flags;
-
- if (cfg && name) {
- flags = rspamd_symcache_get_symbol_flags (cfg->cache,
- name);
-
- if (flags != 0) {
- lua_push_symbol_flags (L, flags);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_add_symbol_flags (lua_State *L)
- {
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name = luaL_checkstring (L, 2);
- guint flags, new_flags = 0;
-
- if (cfg && name && lua_istable (L, 3)) {
-
- for (lua_pushnil (L); lua_next (L, 3); lua_pop (L, 1)) {
- new_flags |= lua_parse_symbol_flags (lua_tostring (L, -1));
- }
-
- flags = rspamd_symcache_get_symbol_flags (cfg->cache,
- name);
-
- if (flags != 0) {
- rspamd_symcache_add_symbol_flags (cfg->cache, name, new_flags);
- /* Push old flags */
- lua_push_symbol_flags (L, flags);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_register_symbol (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name = NULL, *flags_str = NULL, *type_str = NULL,
- *description = NULL, *group = NULL;
- double weight = 0, score = NAN, parent_float = NAN;
- gboolean one_shot = FALSE, no_squeeze = FALSE;
- gint ret = -1, cbref = -1, type, flags = 0;
- gint64 parent = 0, priority = 0, nshots = 0;
- GError *err = NULL;
-
- if (cfg) {
- if (!rspamd_lua_parse_table_arguments (L, 2, &err,
- "name=S;weight=N;callback=F;flags=S;type=S;priority=I;parent=D;"
- "score=D;description=S;group=S;one_shot=B;nshots=I;no_squeeze=B",
- &name, &weight, &cbref, &flags_str, &type_str,
- &priority, &parent_float,
- &score, &description, &group, &one_shot, &nshots, &no_squeeze)) {
- msg_err_config ("bad arguments: %e", err);
- g_error_free (err);
-
- return luaL_error (L, "invalid arguments");
- }
-
- if (!no_squeeze) {
- no_squeeze = cfg->disable_lua_squeeze;
- }
-
- if (nshots == 0) {
- nshots = cfg->default_max_shots;
- }
-
- type = lua_parse_symbol_type (type_str);
-
- if (!name && !(type & SYMBOL_TYPE_CALLBACK)) {
- return luaL_error (L, "no symbol name but type is not callback");
- }
- else if (!(type & SYMBOL_TYPE_VIRTUAL) && cbref == -1) {
- return luaL_error (L, "no callback for symbol %s", name);
- }
-
- if (flags_str) {
- /* Turn off squeezing as well for now */
- /* TODO: deal with it */
- no_squeeze = TRUE;
- type |= lua_parse_symbol_flags (flags_str);
- }
-
- if (isnan (parent_float)) {
- parent = -1;
- }
- else {
- parent = parent_float;
- }
-
- ret = rspamd_register_symbol_fromlua (L,
- cfg,
- name,
- cbref,
- weight == 0 ? 1.0 : weight,
- priority,
- type,
- parent,
- FALSE,
- no_squeeze);
-
- if (!isnan (score) || group) {
- if (one_shot) {
- nshots = 1;
- }
-
- if (!isnan (score)) {
- rspamd_config_add_symbol (cfg, name,
- score, description, group, flags,
- (guint) priority, nshots);
- }
- else {
- rspamd_config_add_symbol (cfg, name,
- 0.0, description, group, flags,
- (guint) priority, nshots);
- }
-
- lua_pushstring (L, "groups");
- lua_gettable (L, 2);
-
- if (lua_istable (L, -1)) {
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- if (lua_isstring (L, -1)) {
- rspamd_config_add_symbol_group (cfg, name,
- lua_tostring (L, -1));
- }
- else {
- return luaL_error (L, "invalid groups element");
- }
- }
- }
-
- lua_pop (L, 1);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushinteger (L, ret);
-
- return 1;
- }
-
- static gint
- lua_config_register_symbols (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- gint i, top, idx, ret = -1;
- const gchar *sym;
- gdouble weight = 1.0;
-
- if (lua_gettop (L) < 3) {
- if (cfg) {
- msg_err_config ("not enough arguments to register a function");
- }
-
- lua_error (L);
-
- return 0;
- }
- if (cfg) {
- if (lua_type (L, 2) == LUA_TSTRING) {
- lua_getglobal (L, luaL_checkstring (L, 2));
- }
- else {
- lua_pushvalue (L, 2);
- }
- idx = luaL_ref (L, LUA_REGISTRYINDEX);
-
- if (lua_type (L, 3) == LUA_TNUMBER) {
- weight = lua_tonumber (L, 3);
- top = 4;
- }
- else {
- top = 3;
- }
- sym = luaL_checkstring (L, top ++);
- ret = rspamd_register_symbol_fromlua (L,
- cfg,
- sym,
- idx,
- weight,
- 0,
- SYMBOL_TYPE_CALLBACK,
- -1,
- FALSE,
- FALSE);
-
- for (i = top; i <= lua_gettop (L); i++) {
- if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
- lua_pushnil (L);
- while (lua_next (L, -2)) {
- lua_pushvalue (L, -2);
- sym = luaL_checkstring (L, -2);
- rspamd_symcache_add_symbol (cfg->cache, sym,
- 0, NULL, NULL,
- SYMBOL_TYPE_VIRTUAL, ret);
- lua_pop (L, 2);
- }
- lua_pop (L, 1);
- }
- else if (lua_type (L, i) == LUA_TSTRING) {
- sym = luaL_checkstring (L, i);
- rspamd_symcache_add_symbol (cfg->cache, sym,
- 0, NULL, NULL,
- SYMBOL_TYPE_VIRTUAL, ret);
- }
- }
- }
-
- lua_pushinteger (L, ret);
-
- return 1;
- }
-
- static gint
- lua_config_register_virtual_symbol (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name;
- double weight;
- gint ret = -1, parent = -1;
-
- if (cfg) {
- name = luaL_checkstring (L, 2);
- weight = luaL_checknumber (L, 3);
-
- if (lua_gettop (L) > 3) {
- parent = lua_tonumber (L, 4);
- }
-
- if (name) {
- ret = rspamd_symcache_add_symbol (cfg->cache, name,
- weight > 0 ? 0 : -1, NULL, NULL,
- SYMBOL_TYPE_VIRTUAL, parent);
- }
- }
-
- lua_pushinteger (L, ret);
-
- return 1;
- }
-
- static gint
- lua_config_register_callback_symbol (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name = NULL;
- double weight;
- gint ret = -1, top = 2;
-
- if (cfg) {
- if (lua_type (L, 2) == LUA_TSTRING) {
- /* Legacy syntax */
- name = luaL_checkstring (L, 2);
- top ++;
- }
-
- weight = luaL_checknumber (L, top);
-
- if (lua_type (L, top + 1) == LUA_TSTRING) {
- lua_getglobal (L, luaL_checkstring (L, top + 1));
- }
- else {
- lua_pushvalue (L, top + 1);
- }
- ret = rspamd_register_symbol_fromlua (L,
- cfg,
- name,
- luaL_ref (L, LUA_REGISTRYINDEX),
- weight,
- 0,
- SYMBOL_TYPE_CALLBACK,
- -1,
- FALSE,
- lua_type (L, top + 1) == LUA_TSTRING);
- }
-
- lua_pushinteger (L, ret);
-
- return 1;
- }
-
- static gint
- lua_config_register_callback_symbol_priority (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name = NULL;
- double weight;
- gint priority, ret = -1, top = 2;
-
- if (cfg) {
- if (lua_type (L, 2) == LUA_TSTRING) {
- /* Legacy syntax */
- name = luaL_checkstring (L, 2);
- top ++;
- }
-
- weight = luaL_checknumber (L, top);
- priority = luaL_checknumber (L, top + 1);
-
- if (lua_type (L, top + 2) == LUA_TSTRING) {
- lua_getglobal (L, luaL_checkstring (L, top + 2));
- }
- else {
- lua_pushvalue (L, top + 2);
- }
-
- ret = rspamd_register_symbol_fromlua (L,
- cfg,
- name,
- luaL_ref (L, LUA_REGISTRYINDEX),
- weight,
- priority,
- SYMBOL_TYPE_CALLBACK,
- -1,
- FALSE,
- lua_type (L, top + 2) == LUA_TSTRING);
- }
-
- lua_pushinteger (L, ret);
-
- return 1;
- }
-
- static gboolean
- rspamd_lua_squeeze_dependency (lua_State *L, struct rspamd_config *cfg,
- const gchar *child,
- const gchar *parent)
- {
- gint err_idx;
- gboolean ret = FALSE;
-
- g_assert (parent != NULL);
- g_assert (child != NULL);
-
- lua_pushcfunction (L, &rspamd_lua_traceback);
- err_idx = lua_gettop (L);
-
- if (rspamd_lua_require_function (L, "lua_squeeze_rules", "squeeze_dependency")) {
- lua_pushstring (L, child);
- lua_pushstring (L, parent);
-
- if (lua_pcall (L, 2, 1, err_idx) != 0) {
- GString *tb = lua_touserdata (L, -1);
- msg_err_config ("call to squeeze_dependency script failed: %v", tb);
-
- if (tb) {
- g_string_free (tb, TRUE);
- }
- }
- else {
- ret = lua_toboolean (L, -1);
- }
- }
-
- lua_settop (L, err_idx - 1);
-
- return ret;
- }
-
- static gint
- lua_config_register_dependency (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *parent = NULL, *child = NULL;
- gint child_id;
- gboolean skip_squeeze;
-
- if (cfg == NULL) {
- lua_error (L);
- return 0;
- }
-
- skip_squeeze = cfg->disable_lua_squeeze;
-
- if (lua_type (L, 2) == LUA_TNUMBER) {
- child_id = luaL_checknumber (L, 2);
- parent = luaL_checkstring (L, 3);
-
- if (!skip_squeeze && lua_isboolean (L, 4)) {
- skip_squeeze = lua_toboolean (L, 4);
- }
-
- msg_warn_config ("calling for obsolete method to register deps for symbol %d->%s",
- child_id, parent);
-
- if (child_id > 0 && parent != NULL) {
-
- if (skip_squeeze || !rspamd_lua_squeeze_dependency (L, cfg,
- rspamd_symcache_symbol_by_id (cfg->cache, child_id),
- parent)) {
- rspamd_symcache_add_dependency (cfg->cache, child_id, parent);
- }
- }
- }
- else {
- child = luaL_checkstring (L,2);
- parent = luaL_checkstring (L, 3);
-
- if (!skip_squeeze && lua_isboolean (L, 4)) {
- skip_squeeze = lua_toboolean (L, 4);
- }
-
- if (child != NULL && parent != NULL) {
-
- if (skip_squeeze || !rspamd_lua_squeeze_dependency (L, cfg, child, parent)) {
- rspamd_symcache_add_delayed_dependency (cfg->cache, child,
- parent);
- }
-
- }
- }
-
- return 0;
- }
-
- static gint
- lua_config_set_metric_symbol (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *description = NULL,
- *group = NULL, *name = NULL, *flags_str = NULL;
- double weight;
- gboolean one_shot = FALSE, one_param = FALSE;
- GError *err = NULL;
- gdouble priority = 0.0;
- guint flags = 0;
- gint64 nshots = 0;
-
- if (cfg) {
-
- if (lua_type (L, 2) == LUA_TTABLE) {
- if (!rspamd_lua_parse_table_arguments (L, 2, &err,
- "*name=S;score=N;description=S;"
- "group=S;one_shot=B;one_param=B;priority=N;flags=S;"
- "nshots=I",
- &name, &weight, &description,
- &group, &one_shot, &one_param,
- &priority, &flags_str, &nshots)) {
- msg_err_config ("bad arguments: %e", err);
- g_error_free (err);
-
- return 0;
- }
- }
- else {
- name = luaL_checkstring (L, 2);
- weight = luaL_checknumber (L, 3);
-
- if (lua_gettop (L) > 3 && lua_type (L, 4) == LUA_TSTRING) {
- description = luaL_checkstring (L, 4);
- }
- if (lua_gettop (L) > 4 && lua_type (L, 5) == LUA_TSTRING) {
- /* XXX: metrics */
- }
- if (lua_gettop (L) > 5 && lua_type (L, 6) == LUA_TSTRING) {
- group = luaL_checkstring (L, 6);
- }
- if (lua_gettop (L) > 6 && lua_type (L, 7) == LUA_TBOOLEAN) {
- one_shot = lua_toboolean (L, 7);
- }
- }
-
- if (nshots == 0) {
- nshots = cfg->default_max_shots;
- }
-
- if (one_shot) {
- nshots = 1;
- }
- if (one_param) {
- flags |= RSPAMD_SYMBOL_FLAG_ONEPARAM;
- }
-
- if (flags_str) {
- if (strstr (flags_str, "one_shot") != NULL) {
- nshots = 1;
- }
- if (strstr (flags_str, "ignore") != NULL) {
- flags |= RSPAMD_SYMBOL_FLAG_IGNORE;
- }
- if (strstr (flags_str, "one_param") != NULL) {
- flags |= RSPAMD_SYMBOL_FLAG_ONEPARAM;
- }
- }
-
- rspamd_config_add_symbol (cfg, name,
- weight, description, group, flags, (guint) priority, nshots);
-
-
- if (lua_type (L, 2) == LUA_TTABLE) {
- lua_pushstring (L, "groups");
- lua_gettable (L, 2);
-
- if (lua_istable (L, -1)) {
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- if (lua_isstring (L, -1)) {
- rspamd_config_add_symbol_group (cfg, name,
- lua_tostring (L, -1));
- } else {
- return luaL_error (L, "invalid groups element");
- }
- }
- }
-
- lua_pop (L, 1);
- }
- }
- else {
- return luaL_error (L, "invalid arguments, rspamd_config expected");
- }
-
- return 0;
- }
-
- static gint
- lua_config_get_metric_symbol (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *sym_name = luaL_checkstring (L, 2);
- struct rspamd_symbol *sym_def;
- struct rspamd_symbols_group *sym_group;
- guint i;
-
- if (cfg && sym_name) {
- sym_def = g_hash_table_lookup (cfg->symbols, sym_name);
-
- if (sym_def == NULL) {
- lua_pushnil (L);
- }
- else {
- lua_createtable (L, 0, 3);
- lua_pushstring (L, "score");
- lua_pushnumber (L, sym_def->score);
- lua_settable (L, -3);
-
- if (sym_def->description) {
- lua_pushstring (L, "description");
- lua_pushstring (L, sym_def->description);
- lua_settable (L, -3);
- }
-
- if (sym_def->gr) {
- lua_pushstring (L, "group");
- lua_pushstring (L, sym_def->gr->name);
- lua_settable (L, -3);
- }
-
- lua_pushstring (L, "groups");
- lua_createtable (L, sym_def->groups->len, 0);
-
- PTR_ARRAY_FOREACH (sym_def->groups, i, sym_group) {
- lua_pushstring (L, sym_group->name);
- lua_rawseti (L, -2, i + 1);
- }
-
- lua_settable (L, -3);
- }
- }
- else {
- luaL_error (L, "Invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_set_metric_action (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name = NULL;
- double threshold = NAN;
- GError *err = NULL;
- gdouble priority = 0.0;
- ucl_object_t *obj_tbl = NULL;
-
- if (cfg) {
-
- if (lua_type (L, 2) == LUA_TTABLE) {
- if (!rspamd_lua_parse_table_arguments (L, 2, &err,
- "*action=S;score=N;"
- "priority=N",
- &name, &threshold,
- &priority)) {
- msg_err_config ("bad arguments: %e", err);
- g_error_free (err);
-
- return 0;
- }
- }
- else if (lua_type (L, 2) == LUA_TSTRING && lua_type (L, 3) == LUA_TTABLE) {
- name = lua_tostring (L, 2);
- obj_tbl = ucl_object_lua_import (L, 3);
-
- if (obj_tbl) {
- if (name) {
- rspamd_config_set_action_score (cfg, name, obj_tbl);
- ucl_object_unref (obj_tbl);
- }
- else {
- ucl_object_unref (obj_tbl);
- return luaL_error (L, "invalid first argument, action name expected");
- }
- }
- else {
- return luaL_error (L, "invalid second argument, table expected");
- }
- }
- else {
- return luaL_error (L, "invalid arguments, table expected");
- }
-
- if (name != NULL && !isnan (threshold) && threshold != 0) {
- obj_tbl = ucl_object_typed_new (UCL_OBJECT);
- ucl_object_insert_key (obj_tbl, ucl_object_fromdouble (threshold),
- "score", 0, false);
- ucl_object_insert_key (obj_tbl, ucl_object_fromdouble (priority),
- "priority", 0, false);
- rspamd_config_set_action_score (cfg, name, obj_tbl);
- ucl_object_unref (obj_tbl);
- }
- }
- else {
- return luaL_error (L, "invalid arguments, rspamd_config expected");
- }
-
- return 0;
- }
-
- static gint
- lua_config_get_metric_action (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *act_name = luaL_checkstring (L, 2);
- struct rspamd_action *act;
-
- if (cfg && act_name) {
- act = rspamd_config_get_action (cfg, act_name);
-
- if (act) {
- if (!isnan (act->threshold)) {
- lua_pushnumber (L, act->threshold);
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- lua_pushnil (L);
- }
- }
- else {
- return luaL_error (L, "invalid arguments, rspamd_config expected");
- }
-
- return 1;
- }
-
- static gint
- lua_config_get_all_actions (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_action *act, *tmp;
-
- if (cfg) {
- lua_createtable (L, 0, HASH_COUNT (cfg->actions));
-
- HASH_ITER (hh, cfg->actions, act, tmp) {
- if (!isnan (act->threshold)) {
- lua_pushstring (L, act->name);
- lua_pushnumber (L, act->threshold);
- lua_settable (L, -3);
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments, rspamd_config expected");
- }
-
- return 1;
- }
-
- static gint
- lua_config_add_composite (lua_State * L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_expression *expr;
- gchar *name;
- const gchar *expr_str;
- struct rspamd_composite *composite;
- gboolean ret = FALSE, new = TRUE;
- GError *err = NULL;
-
- if (cfg) {
- name = rspamd_mempool_strdup (cfg->cfg_pool, luaL_checkstring (L, 2));
- expr_str = luaL_checkstring (L, 3);
-
- if (name && expr_str) {
- if (!rspamd_parse_expression (expr_str, 0, &composite_expr_subr,
- NULL, cfg->cfg_pool, &err, &expr)) {
- msg_err_config ("cannot parse composite expression %s: %e",
- expr_str,
- err);
- g_error_free (err);
- }
- else {
- if (g_hash_table_lookup (cfg->composite_symbols, name) != NULL) {
- msg_warn_config ("composite %s is redefined", name);
- new = FALSE;
- }
-
- composite = rspamd_mempool_alloc0 (cfg->cfg_pool,
- sizeof (struct rspamd_composite));
- composite->expr = expr;
- composite->id = g_hash_table_size (cfg->composite_symbols);
- composite->str_expr = rspamd_mempool_strdup (cfg->cfg_pool,
- expr_str);
- composite->sym = name;
- g_hash_table_insert (cfg->composite_symbols,
- (gpointer)name,
- composite);
-
- if (new) {
- rspamd_symcache_add_symbol (cfg->cache, name,
- 0, NULL, composite, SYMBOL_TYPE_COMPOSITE, -1);
- }
-
- ret = TRUE;
- }
- }
- }
-
- lua_pushboolean (L, ret);
-
- return 1;
- }
-
- static gint
- lua_config_newindex (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name;
- gint id, nshots, flags = 0;
- gboolean optional = FALSE, no_squeeze = FALSE;
-
- name = luaL_checkstring (L, 2);
-
- if (cfg != NULL && name != NULL && lua_gettop (L) == 3) {
- no_squeeze = cfg->disable_lua_squeeze;
-
- if (lua_type (L, 3) == LUA_TFUNCTION) {
- /* Normal symbol from just a function */
- lua_pushvalue (L, 3);
- rspamd_register_symbol_fromlua (L,
- cfg,
- name,
- luaL_ref (L, LUA_REGISTRYINDEX),
- 1.0,
- 0,
- SYMBOL_TYPE_NORMAL,
- -1,
- FALSE,
- no_squeeze);
- }
- else if (lua_type (L, 3) == LUA_TTABLE) {
- gint type = SYMBOL_TYPE_NORMAL, priority = 0, idx;
- gdouble weight = 1.0, score = NAN;
- const char *type_str, *group = NULL, *description = NULL;
-
- no_squeeze = cfg->disable_lua_squeeze;
- /*
- * Table can have the following attributes:
- * "callback" - should be a callback function
- * "weight" - optional weight
- * "priority" - optional priority
- * "type" - optional type (normal, virtual, callback)
- * "flags" - optional flags
- * -- Metric options
- * "score" - optional default score (overridden by metric)
- * "group" - optional default group
- * "one_shot" - optional one shot mode
- * "description" - optional description
- */
- lua_pushvalue (L, 3);
- lua_pushstring (L, "callback");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) != LUA_TFUNCTION) {
- lua_pop (L, 2);
- msg_info_config ("cannot find callback definition for %s",
- name);
- return 0;
- }
- idx = luaL_ref (L, LUA_REGISTRYINDEX);
-
- /* Optional fields */
- lua_pushstring (L, "weight");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TNUMBER) {
- weight = lua_tonumber (L, -1);
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "priority");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TNUMBER) {
- priority = lua_tonumber (L, -1);
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "optional");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- optional = lua_toboolean (L, -1);
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "type");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- type_str = lua_tostring (L, -1);
- type = lua_parse_symbol_type (type_str);
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "flags");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- type_str = lua_tostring (L, -1);
- type |= lua_parse_symbol_flags (type_str);
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "condition");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TFUNCTION) {
- no_squeeze = TRUE;
- }
- lua_pop (L, 1);
-
- if (!no_squeeze) {
- lua_pushstring (L, "no_squeeze");
- lua_gettable (L, -2);
-
- if (lua_toboolean (L, -1)) {
- no_squeeze = TRUE;
- }
- lua_pop (L, 1);
- }
-
- id = rspamd_register_symbol_fromlua (L,
- cfg,
- name,
- idx,
- weight,
- priority,
- type,
- -1,
- optional,
- no_squeeze);
-
- if (id != -1) {
- /* Check for condition */
- lua_pushstring (L, "condition");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TFUNCTION) {
- gint condref;
-
- /* Here we pop function from the stack, so no lua_pop is required */
- condref = luaL_ref (L, LUA_REGISTRYINDEX);
- g_assert (name != NULL);
- rspamd_symcache_add_condition_delayed (cfg->cache,
- name, L, condref);
- }
- else {
- lua_pop (L, 1);
- }
- }
-
- /*
- * Now check if a symbol has not been registered in any metric and
- * insert default value if applicable
- */
- if (g_hash_table_lookup (cfg->symbols, name) == NULL) {
- nshots = cfg->default_max_shots;
-
- lua_pushstring (L, "score");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TNUMBER) {
- score = lua_tonumber (L, -1);
-
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "group");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TSTRING) {
- group = lua_tostring (L, -1);
- }
- lua_pop (L, 1);
-
- if (!isnan (score) || group != NULL) {
- lua_pushstring (L, "description");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- description = lua_tostring (L, -1);
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "one_shot");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- if (lua_toboolean (L, -1)) {
- nshots = 1;
- }
- }
- lua_pop (L, 1);
-
- lua_pushstring (L, "one_param");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- if (lua_toboolean (L, -1)) {
- flags |= RSPAMD_SYMBOL_FLAG_ONEPARAM;
- }
- }
- lua_pop (L, 1);
-
- /*
- * Do not override the existing symbols (using zero priority),
- * since we are defining default values here
- */
- if (!isnan (score)) {
- rspamd_config_add_symbol (cfg, name, score,
- description, group, flags, 0, nshots);
- }
- else if (group) {
- /* Add with zero score */
- rspamd_config_add_symbol (cfg, name, 0.0,
- description, group, flags, 0, nshots);
- }
-
- lua_pushstring (L, "groups");
- lua_gettable (L, -2);
-
- if (lua_istable (L, -1)) {
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- if (lua_isstring (L, -1)) {
- rspamd_config_add_symbol_group (cfg, name,
- lua_tostring (L, -1));
- }
- else {
- return luaL_error (L, "invalid groups element");
- }
- }
- }
-
- lua_pop (L, 1);
- }
- }
-
- /* Remove table from stack */
- lua_pop (L, 1);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_config_add_condition (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *sym = luaL_checkstring (L, 2);
- gboolean ret = FALSE;
- gint condref;
-
- if (cfg && sym && lua_type (L, 3) == LUA_TFUNCTION) {
- lua_pushvalue (L, 3);
- condref = luaL_ref (L, LUA_REGISTRYINDEX);
-
- ret = rspamd_symcache_add_condition_delayed (cfg->cache, sym, L,
- condref);
-
- if (!ret) {
- luaL_unref (L, LUA_REGISTRYINDEX, condref);
- }
- }
-
- lua_pushboolean (L, ret);
- return 1;
- }
-
- static gint
- lua_config_set_peak_cb (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- gint condref;
-
- if (cfg && lua_type (L, 2) == LUA_TFUNCTION) {
- lua_pushvalue (L, 2);
- condref = luaL_ref (L, LUA_REGISTRYINDEX);
- rspamd_symcache_set_peak_callback (cfg->cache,
- condref);
- }
-
- return 0;
- }
-
- static gint
- lua_config_enable_symbol (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *sym = luaL_checkstring (L, 2);
-
- if (cfg && sym) {
- rspamd_symcache_enable_symbol_perm (cfg->cache, sym);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_config_disable_symbol (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *sym = luaL_checkstring (L, 2);
-
- if (cfg && sym) {
- rspamd_symcache_disable_symbol_perm (cfg->cache, sym);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_config_register_regexp (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_lua_regexp *re = NULL;
- rspamd_regexp_t *cache_re;
- const gchar *type_str = NULL, *header_str = NULL;
- gsize header_len = 0;
- GError *err = NULL;
- enum rspamd_re_type type = RSPAMD_RE_BODY;
- gboolean pcre_only = FALSE;
- guint old_flags;
-
- /*
- * - `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
- * - `pcre_only`: allow merely pcre for this regexp
- */
- if (cfg != NULL) {
- if (!rspamd_lua_parse_table_arguments (L, 2, &err,
- "*re=U{regexp};*type=S;header=S;pcre_only=B",
- &re, &type_str, &header_str, &pcre_only)) {
- msg_err_config ("cannot get parameters list: %e", err);
-
- if (err) {
- g_error_free (err);
- }
- }
- else {
- type = rspamd_re_cache_type_from_string (type_str);
-
- if ((type == RSPAMD_RE_HEADER ||
- type == RSPAMD_RE_RAWHEADER ||
- type == RSPAMD_RE_MIMEHEADER) &&
- header_str == NULL) {
- msg_err_config (
- "header argument is mandatory for header/rawheader regexps");
- }
- else {
- if (pcre_only) {
- old_flags = rspamd_regexp_get_flags (re->re);
- old_flags |= RSPAMD_REGEXP_FLAG_PCRE_ONLY;
- rspamd_regexp_set_flags (re->re, old_flags);
- }
-
- if (header_str != NULL) {
- /* Include the last \0 */
- header_len = strlen (header_str) + 1;
- }
-
- cache_re = rspamd_re_cache_add (cfg->re_cache, re->re, type,
- (gpointer) header_str, header_len);
-
- /*
- * XXX: here are dragons!
- * Actually, lua regexp contains internal rspamd_regexp_t
- * and it owns it.
- * However, after this operation we have some OTHER regexp,
- * which we really would like to use.
- * So we do the following:
- * 1) Remove old re and unref it
- * 2) Replace the internal re with cached one
- * 3) Increase its refcount to share ownership between cache and
- * lua object
- */
- if (cache_re != re->re) {
- rspamd_regexp_unref (re->re);
- re->re = rspamd_regexp_ref (cache_re);
- }
- }
- }
- }
-
- return 0;
- }
-
- static gint
- lua_config_replace_regexp (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_lua_regexp *old_re = NULL, *new_re = NULL;
- GError *err = NULL;
-
- if (cfg != NULL) {
- if (!rspamd_lua_parse_table_arguments (L, 2, &err,
- "*old_re=U{regexp};*new_re=U{regexp}",
- &old_re, &new_re)) {
- msg_err_config ("cannot get parameters list: %e", err);
-
- if (err) {
- g_error_free (err);
- }
- }
- else {
- rspamd_re_cache_replace (cfg->re_cache, old_re->re, new_re->re);
- }
- }
-
- return 0;
- }
-
- static gint
- lua_config_register_worker_script (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *worker_type = luaL_checkstring (L, 2), *wtype;
- struct rspamd_worker_conf *cf;
- GList *cur;
- struct rspamd_worker_lua_script *sc;
- gboolean found = FALSE;
-
- if (cfg == NULL || worker_type == NULL || lua_type (L, 3) != LUA_TFUNCTION) {
- return luaL_error (L, "invalid arguments");
- }
-
- for (cur = g_list_first (cfg->workers); cur != NULL; cur = g_list_next (cur)) {
- cf = cur->data;
- wtype = g_quark_to_string (cf->type);
-
- if (g_ascii_strcasecmp (wtype, worker_type) == 0) {
- sc = rspamd_mempool_alloc0 (cfg->cfg_pool, sizeof (*sc));
- lua_pushvalue (L, 3);
- sc->cbref = luaL_ref (L, LUA_REGISTRYINDEX);
- DL_APPEND (cf->scripts, sc);
- found = TRUE;
- }
- }
-
- lua_pushboolean (L, found);
-
- return 1;
- }
-
- static gint
- lua_config_add_on_load (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_config_post_load_script *sc;
-
- if (cfg == NULL || lua_type (L, 2) != LUA_TFUNCTION) {
- return luaL_error (L, "invalid arguments");
- }
-
- sc = g_malloc0 (sizeof (*sc));
- lua_pushvalue (L, 2);
- sc->cbref = luaL_ref (L, LUA_REGISTRYINDEX);
- DL_APPEND (cfg->on_load, sc);
-
- return 0;
- }
-
- static void lua_periodic_callback_finish (struct thread_entry *thread, int ret);
- static void lua_periodic_callback_error (struct thread_entry *thread, int ret, const char *msg);
-
- struct rspamd_lua_periodic {
- struct event_base *ev_base;
- struct rspamd_config *cfg;
- lua_State *L;
- gdouble timeout;
- struct event ev;
- gint cbref;
- gboolean need_jitter;
- };
-
- static void
- lua_periodic_callback (gint unused_fd, short what, gpointer ud)
- {
- struct rspamd_lua_periodic *periodic = ud;
- struct rspamd_config **pcfg, *cfg;
- struct event_base **pev_base;
- struct thread_entry *thread;
- lua_State *L;
-
- thread = lua_thread_pool_get_for_config (periodic->cfg);
- thread->cd = periodic;
- thread->finish_callback = lua_periodic_callback_finish;
- thread->error_callback = lua_periodic_callback_error;
-
- L = thread->lua_state;
-
- lua_rawgeti (L, LUA_REGISTRYINDEX, periodic->cbref);
- pcfg = lua_newuserdata (L, sizeof (*pcfg));
- rspamd_lua_setclass (L, "rspamd{config}", -1);
- cfg = periodic->cfg;
- *pcfg = cfg;
- pev_base = lua_newuserdata (L, sizeof (*pev_base));
- rspamd_lua_setclass (L, "rspamd{ev_base}", -1);
- *pev_base = periodic->ev_base;
-
- event_del (&periodic->ev);
- lua_thread_call (thread, 2);
- }
-
- static void
- lua_periodic_callback_finish (struct thread_entry *thread, int ret)
- {
- lua_State *L;
- struct rspamd_lua_periodic *periodic = thread->cd;
- gboolean plan_more = FALSE;
- struct timeval tv;
- gdouble timeout = 0.0;
-
- L = thread->lua_state;
-
- #ifdef HAVE_EVENT_NO_CACHE_TIME_FUNC
- event_base_update_cache_time (periodic->ev_base);
- #endif
-
- if (ret == 0) {
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- plan_more = lua_toboolean (L, -1);
- timeout = periodic->timeout;
- }
- else if (lua_type (L, -1) == LUA_TNUMBER) {
- timeout = lua_tonumber (L, -1);
- plan_more = timeout > 0 ? TRUE : FALSE;
- }
-
- lua_pop (L, 1); /* Return value */
- }
- if (plan_more) {
- if (periodic->need_jitter) {
- timeout = rspamd_time_jitter (timeout, 0.0);
- }
-
- double_to_tv (timeout, &tv);
- event_add (&periodic->ev, &tv);
- }
- else {
- luaL_unref (L, LUA_REGISTRYINDEX, periodic->cbref);
- g_free (periodic);
- }
- }
-
- static void
- lua_periodic_callback_error (struct thread_entry *thread, int ret, const char *msg)
- {
- struct rspamd_config *cfg;
- struct rspamd_lua_periodic *periodic = thread->cd;
- cfg = periodic->cfg;
-
- msg_err_config ("call to finishing script failed: %s", msg);
-
- lua_periodic_callback_finish(thread, ret);
- }
-
-
- static gint
- lua_config_add_periodic (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct event_base *ev_base = lua_check_ev_base (L, 2);
- gdouble timeout = lua_tonumber (L, 3);
- struct timeval tv;
- struct rspamd_lua_periodic *periodic;
- gboolean need_jitter = FALSE;
-
- if (cfg == NULL || timeout < 0 || lua_type (L, 4) != LUA_TFUNCTION) {
- return luaL_error (L, "invalid arguments");
- }
-
- if (lua_type (L, 5) == LUA_TBOOLEAN) {
- need_jitter = lua_toboolean (L, 5);
- }
-
- periodic = g_malloc0 (sizeof (*periodic));
- periodic->timeout = timeout;
- periodic->L = L;
- periodic->cfg = cfg;
- periodic->ev_base = ev_base;
- periodic->need_jitter = need_jitter;
- lua_pushvalue (L, 4);
- periodic->cbref = luaL_ref (L, LUA_REGISTRYINDEX);
- event_set (&periodic->ev, -1, EV_TIMEOUT, lua_periodic_callback, periodic);
- event_base_set (ev_base, &periodic->ev);
-
- if (need_jitter) {
- timeout = rspamd_time_jitter (timeout, 0.0);
- }
-
- double_to_tv (timeout, &tv);
- event_add (&periodic->ev, &tv);
-
- return 0;
- }
-
- static gint
- lua_config_get_symbols_count (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- guint res = 0;
-
- if (cfg != NULL) {
- res = rspamd_symcache_stats_symbols_count (cfg->cache);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_pushinteger (L, res);
-
- return 1;
- }
-
- static gint
- lua_config_get_symbols_cksum (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- guint64 res = 0, *pres;
-
- if (cfg != NULL) {
- res = rspamd_symcache_get_cksum (cfg->cache);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- pres = lua_newuserdata (L, sizeof (res));
- *pres = res;
- rspamd_lua_setclass (L, "rspamd{int64}", -1);
-
- return 1;
- }
-
- static gint
- lua_config_get_symbols_counters (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- ucl_object_t *counters;
-
- if (cfg != NULL) {
- counters = rspamd_symcache_counters (cfg->cache);
- ucl_object_push_lua (L, counters, true);
- ucl_object_unref (counters);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
- static void
- lua_metric_symbol_inserter (gpointer k, gpointer v, gpointer ud)
- {
- lua_State *L = (lua_State *) ud;
- const gchar *sym = k;
- struct rspamd_symbol *s = (struct rspamd_symbol *) v;
-
- lua_pushstring (L, sym);
-
- lua_createtable (L, 0, 3); /* TODO: add more if needed */
- lua_pushstring (L, "score");
- lua_pushnumber (L, s->score);
- lua_settable (L, -3);
- lua_pushstring (L, "description");
- lua_pushstring (L, s->description);
- lua_settable (L, -3);
-
- lua_settable (L, -3);
- }
-
- static gint
- lua_config_get_symbols_scores (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
-
- if (cfg != NULL) {
- lua_createtable (L, 0, g_hash_table_size (cfg->symbols));
- g_hash_table_foreach (cfg->symbols,
- lua_metric_symbol_inserter,
- L);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
-
- static gint
- lua_config_get_symbol_callback (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *sym = luaL_checkstring (L, 2);
- struct rspamd_abstract_callback_data *abs_cbdata;
- struct lua_callback_data *cbd;
-
- if (cfg != NULL && sym != NULL) {
- abs_cbdata = rspamd_symcache_get_cbdata (cfg->cache, sym);
-
- if (abs_cbdata == NULL || abs_cbdata->magic != rspamd_lua_callback_magic) {
- lua_pushnil (L);
- }
- else {
- cbd = (struct lua_callback_data *)abs_cbdata;
-
- if (cbd->cb_is_ref) {
- lua_rawgeti (L, LUA_REGISTRYINDEX, cbd->callback.ref);
- }
- else {
- lua_getglobal (L, cbd->callback.name);
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_set_symbol_callback (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *sym = luaL_checkstring (L, 2);
- struct rspamd_abstract_callback_data *abs_cbdata;
- struct lua_callback_data *cbd;
-
- if (cfg != NULL && sym != NULL && lua_type (L, 3) == LUA_TFUNCTION) {
- abs_cbdata = rspamd_symcache_get_cbdata (cfg->cache, sym);
-
- if (abs_cbdata == NULL || abs_cbdata->magic != rspamd_lua_callback_magic) {
- lua_pushboolean (L, FALSE);
- }
- else {
- cbd = (struct lua_callback_data *)abs_cbdata;
-
- if (cbd->cb_is_ref) {
- luaL_unref (L, LUA_REGISTRYINDEX, cbd->callback.ref);
- }
- else {
- cbd->cb_is_ref = TRUE;
- }
-
- lua_pushvalue (L, 3);
- cbd->callback.ref = luaL_ref (L, LUA_REGISTRYINDEX);
- lua_pushboolean (L, TRUE);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_get_symbol_stat (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *sym = luaL_checkstring (L, 2);
- gdouble freq, stddev, tm;
- guint hits;
-
- if (cfg != NULL && sym != NULL) {
- if (!rspamd_symcache_stat_symbol (cfg->cache, sym, &freq,
- &stddev, &tm, &hits)) {
- lua_pushnil (L);
- }
- else {
- lua_createtable (L, 0, 4);
- lua_pushstring (L, "frequency");
- lua_pushnumber (L, freq);
- lua_settable (L, -3);
- lua_pushstring (L, "sttdev");
- lua_pushnumber (L, stddev);
- lua_settable (L, -3);
- lua_pushstring (L, "time");
- lua_pushnumber (L, tm);
- lua_settable (L, -3);
- lua_pushstring (L, "hits");
- lua_pushinteger (L, hits);
- lua_settable (L, -3);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
-
- static gint
- lua_config_register_finish_script (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_config_post_load_script *sc;
-
- if (cfg != NULL && lua_type (L, 2) == LUA_TFUNCTION) {
- sc = g_malloc0 (sizeof (*sc));
- lua_pushvalue (L, 2);
- sc->cbref = luaL_ref (L, LUA_REGISTRYINDEX);
- DL_APPEND (cfg->finish_callbacks, sc);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_config_register_monitored (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_monitored *m, **pm;
- const gchar *url, *type;
- ucl_object_t *params = NULL;
-
- url = lua_tostring (L, 2);
- type = lua_tostring (L, 3);
-
- if (cfg != NULL && url != NULL && type != NULL) {
- if (g_ascii_strcasecmp (type, "dns") == 0) {
- lua_Debug ar;
-
- if (lua_type (L, 4) == LUA_TTABLE) {
- params = ucl_object_lua_import (L, 4);
- }
-
- /* Get lua line and source */
- lua_getstack (L, 1, &ar);
- lua_getinfo (L, "nSl", &ar);
-
- m = rspamd_monitored_create_ (cfg->monitored_ctx, url,
- RSPAMD_MONITORED_DNS, RSPAMD_MONITORED_DEFAULT,
- params, ar.short_src);
-
- if (m) {
- pm = lua_newuserdata (L, sizeof (*pm));
- *pm = m;
- rspamd_lua_setclass (L, "rspamd{monitored}", -1);
- }
- else {
- lua_pushnil (L);
- }
-
- if (params) {
- ucl_object_unref (params);
- }
- }
- else {
- return luaL_error (L, "invalid monitored type: %s", type);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_add_doc (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg;
- const gchar *path = NULL, *option, *doc_string;
- const gchar *type_str = NULL, *default_value = NULL;
- ucl_type_t type = UCL_NULL;
- gboolean required = FALSE;
- GError *err = NULL;
-
- cfg = lua_check_config (L, 1);
-
- if (lua_type (L, 2 ) == LUA_TSTRING) {
- path = luaL_checkstring (L, 2);
- }
-
- option = luaL_checkstring (L, 3);
- doc_string = luaL_checkstring (L, 4);
-
- if (cfg && option && doc_string) {
- if (lua_type (L, 5) == LUA_TTABLE) {
- if (!rspamd_lua_parse_table_arguments (L, 5, &err,
- "type=S;default=S;required=B",
- &type_str, &default_value, &required)) {
- msg_err_config ("cannot get parameters list: %e", err);
-
- if (err) {
- g_error_free (err);
- }
-
- if (type_str) {
- if (!ucl_object_string_to_type (type_str, &type)) {
- msg_err_config ("invalid type: %s", type_str);
- }
- }
- }
- }
-
- rspamd_rcl_add_doc_by_path (cfg, path, doc_string, option,
- type, NULL, 0, default_value, required);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_config_add_example (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg;
- const gchar *path = NULL, *option, *doc_string, *example;
- gsize example_len;
-
- cfg = lua_check_config (L, 1);
-
- if (lua_type (L, 2 ) == LUA_TSTRING) {
- path = luaL_checkstring (L, 2);
- }
-
- option = luaL_checkstring (L, 3);
- doc_string = luaL_checkstring (L, 4);
- example = luaL_checklstring (L, 5, &example_len);
-
- if (cfg && option && doc_string && example) {
-
- rspamd_rcl_add_doc_by_example (cfg, path, doc_string, option,
- example, example_len);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_config_get_cpu_flags (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_cryptobox_library_ctx *crypto_ctx;
-
- if (cfg != NULL) {
- crypto_ctx = cfg->libs_ctx->crypto_ctx;
- lua_newtable (L);
-
- if (crypto_ctx->cpu_config & CPUID_SSSE3) {
- lua_pushstring (L, "ssse3");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (crypto_ctx->cpu_config & CPUID_SSE41) {
- lua_pushstring (L, "sse41");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (crypto_ctx->cpu_config & CPUID_SSE42) {
- lua_pushstring (L, "sse42");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (crypto_ctx->cpu_config & CPUID_SSE2) {
- lua_pushstring (L, "sse2");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (crypto_ctx->cpu_config & CPUID_SSE3) {
- lua_pushstring (L, "sse3");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (crypto_ctx->cpu_config & CPUID_AVX) {
- lua_pushstring (L, "avx");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- if (crypto_ctx->cpu_config & CPUID_AVX2) {
- lua_pushstring (L, "avx2");
- lua_pushboolean (L, true);
- lua_settable (L, -3);
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_has_torch (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- struct rspamd_cryptobox_library_ctx *crypto_ctx;
-
- if (cfg != NULL) {
- crypto_ctx = cfg->libs_ctx->crypto_ctx;
- #if !defined(WITH_TORCH) || !defined(WITH_LUAJIT)
- lua_pushboolean (L, false);
- (void)crypto_ctx;
- #else
- if (crypto_ctx->cpu_config & CPUID_SSE2) {
- lua_pushboolean (L, true);
- }
- else {
- lua_pushboolean (L, false);
- }
- #endif
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_experimental_enabled (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
-
- if (cfg != NULL) {
- lua_pushboolean (L, cfg->enable_experimental);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- struct rspamd_lua_include_trace_cbdata {
- lua_State *L;
- gint cbref;
- };
-
- static void
- lua_include_trace_cb (struct ucl_parser *parser,
- const ucl_object_t *parent,
- const ucl_object_t *args,
- const char *path,
- size_t pathlen,
- void *user_data)
- {
- struct rspamd_lua_include_trace_cbdata *cbdata =
- (struct rspamd_lua_include_trace_cbdata *)user_data;
- gint err_idx;
- GString *tb;
- lua_State *L;
-
- L = cbdata->L;
- lua_pushcfunction (L, &rspamd_lua_traceback);
- err_idx = lua_gettop (L);
-
- lua_rawgeti (L, LUA_REGISTRYINDEX, cbdata->cbref);
- /* Current filename */
- lua_pushstring (L, ucl_parser_get_cur_file (parser));
- /* Included filename */
- lua_pushlstring (L, path, pathlen);
- /* Params */
- if (args) {
- ucl_object_push_lua (L, args, true);
- }
- else {
- lua_newtable (L);
- }
- /* Parent */
- if (parent) {
- lua_pushstring (L, ucl_object_key (parent));
- }
- else {
- lua_pushnil (L);
- }
-
- if (lua_pcall (L, 4, 0, err_idx) != 0) {
- tb = lua_touserdata (L, -1);
- msg_err ("lua call to local include trace failed: %v", tb);
- if (tb) {
- g_string_free (tb, TRUE);
- }
- }
-
- lua_settop (L, err_idx - 1);
- }
-
- #define LUA_TABLE_TO_HASH(htb, idx) do { \
- lua_pushstring (L, (idx)); \
- lua_gettable (L, -2); \
- if (lua_isstring (L, -1)) { \
- g_hash_table_insert ((htb), (idx), g_strdup (lua_tostring (L, -1))); \
- } \
- lua_pop (L, 1); \
- } while(0)
-
- static gint
- lua_config_load_ucl (lua_State *L)
- {
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *filename;
- GHashTable *paths = g_hash_table_new_full (rspamd_str_hash, rspamd_str_equal,
- NULL, g_free);
- GError *err = NULL;
-
- if (cfg) {
- if (lua_isstring (L, 2)) {
- filename = lua_tostring (L, 2);
- }
- else {
- filename = RSPAMD_CONFDIR "/rspamd.conf";
- }
-
- /* Convert rspamd_paths */
- lua_getglobal (L, "rspamd_paths");
-
- if (lua_istable (L, -1)) {
- LUA_TABLE_TO_HASH(paths, RSPAMD_CONFDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_LOCAL_CONFDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_RUNDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_DBDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_LOGDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_WWWDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_PLUGINSDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_RULESDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_LUALIBDIR_INDEX);
- LUA_TABLE_TO_HASH(paths, RSPAMD_PREFIX_INDEX);
- }
-
- lua_pop (L, 1);
-
- if (lua_isfunction (L, 3)) {
- struct rspamd_lua_include_trace_cbdata cbd;
-
- lua_pushvalue (L, 3);
- cbd.cbref = luaL_ref (L, LUA_REGISTRYINDEX);
- cbd.L = L;
-
- if (!rspamd_config_parse_ucl (cfg, filename, paths,
- lua_include_trace_cb, &cbd, &err)) {
- luaL_unref (L, LUA_REGISTRYINDEX, cbd.cbref);
- lua_pushboolean (L, false);
- lua_pushfstring (L, "failed to load config: %s", err->message);
- g_error_free (err);
- g_hash_table_unref (paths);
-
- return 2;
- }
-
- luaL_unref (L, LUA_REGISTRYINDEX, cbd.cbref);
- }
- else {
- if (!rspamd_config_parse_ucl (cfg, filename, paths, NULL, NULL, &err)) {
- lua_pushboolean (L, false);
- lua_pushfstring (L, "failed to load config: %s", err->message);
- g_error_free (err);
- g_hash_table_unref (paths);
-
- return 2;
- }
- }
-
- rspamd_rcl_maybe_apply_lua_transform (cfg);
- rspamd_config_calculate_cksum (cfg);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- g_hash_table_unref (paths);
- lua_pushboolean (L, true);
-
- return 1;
- }
-
- #undef IDX_TO_HASH
-
- static gint
- lua_config_parse_rcl (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- GHashTable *excluded = g_hash_table_new_full (rspamd_str_hash, rspamd_str_equal,
- g_free, NULL);
- GError *err = NULL;
- struct rspamd_rcl_section *top;
-
- if (cfg) {
- if (lua_istable (L, 2)) {
- lua_pushvalue (L, 2);
-
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- g_hash_table_insert (excluded, g_strdup (lua_tostring (L, -1)),
- GINT_TO_POINTER (-1));
- }
-
- lua_pop (L, 1);
- }
-
- top = rspamd_rcl_config_init (cfg, excluded);
-
- if (!rspamd_rcl_parse (top, cfg, cfg, cfg->cfg_pool, cfg->rcl_obj, &err)) {
- lua_pushboolean (L, false);
- lua_pushfstring (L, "failed to load config: %s", err->message);
- g_error_free (err);
- g_hash_table_unref (excluded);
- rspamd_rcl_section_free (top);
-
- return 2;
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- g_hash_table_unref (excluded);
- rspamd_rcl_section_free (top);
- lua_pushboolean (L, true);
-
- return 1;
- }
-
- static gint
- lua_config_init_modules (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
-
- if (cfg != NULL) {
- rspamd_lua_post_load_config (cfg);
- lua_pushboolean (L, rspamd_init_filters (cfg, FALSE));
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_config_init_subsystem (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *subsystem = luaL_checkstring (L, 2);
- gchar **parts;
- guint nparts, i;
-
- if (cfg != NULL && subsystem != NULL) {
- parts = g_strsplit_set (subsystem, ";,", -1);
- nparts = g_strv_length (parts);
-
- for (i = 0; i < nparts; i ++) {
- if (strcmp (parts[i], "filters") == 0) {
- rspamd_lua_post_load_config (cfg);
- rspamd_init_filters (cfg, FALSE);
- }
- else if (strcmp (parts[i], "langdet") == 0) {
- cfg->lang_det = rspamd_language_detector_init (cfg);
- }
- else if (strcmp (parts[i], "stat") == 0) {
- rspamd_stat_init (cfg, NULL);
- }
- else if (strcmp (parts[i], "dns") == 0) {
- struct event_base *ev_base = lua_check_ev_base (L, 3);
-
- if (ev_base) {
- cfg->dns_resolver = dns_resolver_init (rspamd_logger_get_singleton(),
- ev_base,
- cfg);
- }
- else {
- g_strfreev (parts);
-
- return luaL_error (L, "no event base specified");
- }
- }
- else {
- g_strfreev (parts);
-
- return luaL_error (L, "invalid param: %s", parts[i]);
- }
- }
-
- g_strfreev (parts);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 0;
- }
-
- static gint
- lua_config_register_re_selector (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
- const gchar *name = luaL_checkstring (L, 2);
- const gchar *selector_str = luaL_checkstring (L, 3);
- const gchar *delimiter = "";
- gint top = lua_gettop (L);
- bool res = false;
-
- if (cfg && name && selector_str) {
- if (lua_gettop (L) >= 4) {
- delimiter = luaL_checkstring (L, 4);
- }
-
- if (luaL_dostring (L, "return require \"lua_selectors\"") != 0) {
- msg_warn_config ("cannot require lua_selectors: %s",
- lua_tostring (L, -1));
- }
- else {
- if (lua_type (L, -1) != LUA_TTABLE) {
- msg_warn_config ("lua selectors must return "
- "table and not %s",
- lua_typename (L, lua_type (L, -1)));
- }
- else {
- lua_pushstring (L, "create_selector_closure");
- lua_gettable (L, -2);
-
- if (lua_type (L, -1) != LUA_TFUNCTION) {
- msg_warn_config ("create_selector_closure must return "
- "function and not %s",
- lua_typename (L, lua_type (L, -1)));
- }
- else {
- gint err_idx, ret;
- GString *tb;
- struct rspamd_config **pcfg;
-
- lua_pushcfunction (L, &rspamd_lua_traceback);
- err_idx = lua_gettop (L);
-
- /* Push function */
- lua_pushvalue (L, -2);
-
- pcfg = lua_newuserdata (L, sizeof (*pcfg));
- rspamd_lua_setclass (L, "rspamd{config}", -1);
- *pcfg = cfg;
- lua_pushstring (L, selector_str);
- lua_pushstring (L, delimiter);
-
- if ((ret = lua_pcall (L, 3, 1, err_idx)) != 0) {
- tb = lua_touserdata (L, -1);
- msg_err_config ("call to create_selector_closure lua "
- "script failed (%d): %v", ret, tb);
-
- if (tb) {
- g_string_free (tb, TRUE);
- }
- }
- else {
- if (lua_type (L, -1) != LUA_TFUNCTION) {
- msg_warn_config ("create_selector_closure "
- "invocation must return "
- "function and not %s",
- lua_typename (L, lua_type (L, -1)));
- }
- else {
- ret = luaL_ref (L, LUA_REGISTRYINDEX);
- rspamd_re_cache_add_selector (cfg->re_cache,
- name, ret);
- res = true;
- }
- }
- }
- }
- }
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- lua_settop (L, top);
- lua_pushboolean (L, res);
-
- if (res) {
- msg_info_config ("registered regexp selector %s", name);
- }
-
- return 1;
- }
-
- static gint
- lua_config_get_tld_path (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_config *cfg = lua_check_config (L, 1);
-
- if (cfg != NULL) {
- lua_pushstring (L, cfg->tld_file);
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_monitored_alive (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_monitored *m = lua_check_monitored (L, 1);
-
- if (m) {
- lua_pushboolean (L, rspamd_monitored_alive (m));
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_monitored_offline (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_monitored *m = lua_check_monitored (L, 1);
-
- if (m) {
- lua_pushnumber (L, rspamd_monitored_offline_time (m));
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_monitored_total_offline (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_monitored *m = lua_check_monitored (L, 1);
-
- if (m) {
- lua_pushnumber (L, rspamd_monitored_total_offline_time (m));
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- static gint
- lua_monitored_latency (lua_State *L)
- {
- LUA_TRACE_POINT;
- struct rspamd_monitored *m = lua_check_monitored (L, 1);
-
- if (m) {
- lua_pushnumber (L, rspamd_monitored_latency (m));
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return 1;
- }
-
- void
- luaopen_config (lua_State * L)
- {
- rspamd_lua_new_class (L, "rspamd{config}", configlib_m);
-
- lua_pop (L, 1);
-
- rspamd_lua_new_class (L, "rspamd{monitored}", monitoredlib_m);
-
- lua_pop (L, 1);
- }
-
- void
- lua_call_finish_script (struct rspamd_config_post_load_script *sc,
- struct rspamd_task *task) {
-
- struct rspamd_task **ptask;
- struct thread_entry *thread;
-
- thread = lua_thread_pool_get_for_task (task);
- thread->task = task;
-
- lua_State *L = thread->lua_state;
-
- lua_rawgeti (L, LUA_REGISTRYINDEX, sc->cbref);
-
- ptask = lua_newuserdata (L, sizeof (struct rspamd_task *));
- rspamd_lua_setclass (L, "rspamd{task}", - 1);
- *ptask = task;
-
- lua_thread_call (thread, 1);
- }
|