12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021 |
- /*-
- * 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 "config.h"
- #include "libutil/util.h"
- #include "libutil/http.h"
- #include "libutil/http_private.h"
- #include "rspamdclient.h"
- #include "utlist.h"
- #include "unix-std.h"
- #ifdef HAVE_SYS_WAIT_H
- #include <sys/wait.h>
- #endif
-
- #define DEFAULT_PORT 11333
- #define DEFAULT_CONTROL_PORT 11334
-
- static gchar *connect_str = "localhost";
- static gchar *password = NULL;
- static gchar *ip = NULL;
- static gchar *from = NULL;
- static gchar *deliver_to = NULL;
- static gchar **rcpts = NULL;
- static gchar *user = NULL;
- static gchar *helo = NULL;
- static gchar *hostname = NULL;
- static gchar *classifier = NULL;
- static gchar *local_addr = NULL;
- static gchar *execute = NULL;
- static gchar *sort = NULL;
- static gchar **http_headers = NULL;
- static gchar **exclude_patterns = NULL;
- static gint weight = 0;
- static gint flag = 0;
- static gchar *fuzzy_symbol = NULL;
- static gchar *dictionary = NULL;
- static gint max_requests = 8;
- static gdouble timeout = 10.0;
- static gboolean pass_all;
- static gboolean tty = FALSE;
- static gboolean verbose = FALSE;
- static gboolean print_commands = FALSE;
- static gboolean json = FALSE;
- static gboolean compact = FALSE;
- static gboolean headers = FALSE;
- static gboolean raw = FALSE;
- static gboolean extended_urls = FALSE;
- static gboolean mime_output = FALSE;
- static gboolean empty_input = FALSE;
- static gboolean compressed = FALSE;
- static gboolean profile = FALSE;
- static gboolean skip_images = FALSE;
- static gboolean skip_attachments = FALSE;
- static gchar *key = NULL;
- static gchar *user_agent = "rspamc";
- static GList *children;
- static GPatternSpec **exclude_compiled = NULL;
-
- #define ADD_CLIENT_HEADER(o, n, v) do { \
- struct rspamd_http_client_header *nh; \
- nh = g_malloc (sizeof (*nh)); \
- nh->name = (n); \
- nh->value = (v); \
- g_queue_push_tail ((o), nh); \
- } while (0)
-
- static gboolean rspamc_password_callback (const gchar *option_name,
- const gchar *value,
- gpointer data,
- GError **error);
-
- static GOptionEntry entries[] =
- {
- { "connect", 'h', 0, G_OPTION_ARG_STRING, &connect_str,
- "Specify host and port", NULL },
- { "password", 'P', G_OPTION_FLAG_OPTIONAL_ARG, G_OPTION_ARG_CALLBACK,
- &rspamc_password_callback, "Specify control password", NULL },
- { "classifier", 'c', 0, G_OPTION_ARG_STRING, &classifier,
- "Classifier to learn spam or ham", NULL },
- { "weight", 'w', 0, G_OPTION_ARG_INT, &weight,
- "Weight for fuzzy operations", NULL },
- { "flag", 'f', 0, G_OPTION_ARG_INT, &flag, "Flag for fuzzy operations",
- NULL },
- { "pass-all", 'p', 0, G_OPTION_ARG_NONE, &pass_all, "Pass all filters",
- NULL },
- { "verbose", 'v', 0, G_OPTION_ARG_NONE, &verbose, "More verbose output",
- NULL },
- { "ip", 'i', 0, G_OPTION_ARG_STRING, &ip,
- "Emulate that message was received from specified ip address",
- NULL },
- { "user", 'u', 0, G_OPTION_ARG_STRING, &user,
- "Emulate that message was received from specified authenticated user", NULL },
- { "deliver", 'd', 0, G_OPTION_ARG_STRING, &deliver_to,
- "Emulate that message is delivered to specified user (for LDA/statistics)", NULL },
- { "from", 'F', 0, G_OPTION_ARG_STRING, &from,
- "Emulate that message has specified SMTP FROM address", NULL },
- { "rcpt", 'r', 0, G_OPTION_ARG_STRING_ARRAY, &rcpts,
- "Emulate that message has specified SMTP RCPT address", NULL },
- { "helo", 0, 0, G_OPTION_ARG_STRING, &helo,
- "Imitate SMTP HELO passing from MTA", NULL },
- { "hostname", 0, 0, G_OPTION_ARG_STRING, &hostname,
- "Imitate hostname passing from MTA", NULL },
- { "timeout", 't', 0, G_OPTION_ARG_DOUBLE, &timeout,
- "Time in seconds to wait for a reply", NULL },
- { "bind", 'b', 0, G_OPTION_ARG_STRING, &local_addr,
- "Bind to specified ip address", NULL },
- { "commands", 0, 0, G_OPTION_ARG_NONE, &print_commands,
- "List available commands", NULL },
- { "json", 'j', 0, G_OPTION_ARG_NONE, &json, "Output json reply", NULL },
- { "compact", '\0', 0, G_OPTION_ARG_NONE, &compact, "Output compact json reply", NULL},
- { "headers", 0, 0, G_OPTION_ARG_NONE, &headers, "Output HTTP headers",
- NULL },
- { "raw", 0, 0, G_OPTION_ARG_NONE, &raw, "Output raw reply from rspamd",
- NULL },
- { "ucl", 0, 0, G_OPTION_ARG_NONE, &raw, "Output ucl reply from rspamd",
- NULL },
- { "max-requests", 'n', 0, G_OPTION_ARG_INT, &max_requests,
- "Maximum count of parallel requests to rspamd", NULL },
- { "extended-urls", 0, 0, G_OPTION_ARG_NONE, &extended_urls,
- "Output urls in extended format", NULL },
- { "key", 0, 0, G_OPTION_ARG_STRING, &key,
- "Use specified pubkey to encrypt request", NULL },
- { "exec", 'e', 0, G_OPTION_ARG_STRING, &execute,
- "Execute the specified command and pass output to it", NULL },
- { "mime", 'm', 0, G_OPTION_ARG_NONE, &mime_output,
- "Write mime body of message with headers instead of just a scan's result", NULL },
- {"header", 0, 0, G_OPTION_ARG_STRING_ARRAY, &http_headers,
- "Add custom HTTP header to query (can be repeated)", NULL},
- {"exclude", 0, 0, G_OPTION_ARG_STRING_ARRAY, &exclude_patterns,
- "Exclude specific glob patterns in file names (can be repeated)", NULL},
- {"sort", 0, 0, G_OPTION_ARG_STRING, &sort,
- "Sort output in a specific order (name, weight, frequency, hits)", NULL},
- { "empty", 'E', 0, G_OPTION_ARG_NONE, &empty_input,
- "Allow empty input instead of reading from stdin", NULL },
- { "fuzzy-symbol", 'S', 0, G_OPTION_ARG_STRING, &fuzzy_symbol,
- "Learn the specified fuzzy symbol", NULL },
- { "compressed", 'z', 0, G_OPTION_ARG_NONE, &compressed,
- "Enable zstd compression", NULL },
- { "profile", '\0', 0, G_OPTION_ARG_NONE, &profile,
- "Profile symbols execution time", NULL },
- { "dictionary", 'D', 0, G_OPTION_ARG_FILENAME, &dictionary,
- "Use dictionary to compress data", NULL },
- { "skip-images", '\0', 0, G_OPTION_ARG_NONE, &skip_images,
- "Skip images when learning/unlearning fuzzy", NULL },
- { "skip-attachments", '\0', 0, G_OPTION_ARG_NONE, &skip_attachments,
- "Skip attachments when learning/unlearning fuzzy", NULL },
- { "user-agent", 'U', 0, G_OPTION_ARG_STRING, &user_agent,
- "Use specific User-Agent instead of \"rspamc\"", NULL },
- { NULL, 0, 0, G_OPTION_ARG_NONE, NULL, NULL, NULL }
- };
-
- /* Copy to avoid linking with librspamdserver */
- enum rspamd_action_type {
- METRIC_ACTION_REJECT = 0,
- METRIC_ACTION_SOFT_REJECT,
- METRIC_ACTION_REWRITE_SUBJECT,
- METRIC_ACTION_ADD_HEADER,
- METRIC_ACTION_GREYLIST,
- METRIC_ACTION_NOACTION,
- METRIC_ACTION_MAX
- };
-
- static void rspamc_symbols_output (FILE *out, ucl_object_t *obj);
- static void rspamc_uptime_output (FILE *out, ucl_object_t *obj);
- static void rspamc_counters_output (FILE *out, ucl_object_t *obj);
- static void rspamc_stat_output (FILE *out, ucl_object_t *obj);
-
- enum rspamc_command_type {
- RSPAMC_COMMAND_UNKNOWN = 0,
- RSPAMC_COMMAND_CHECK,
- RSPAMC_COMMAND_SYMBOLS,
- RSPAMC_COMMAND_LEARN_SPAM,
- RSPAMC_COMMAND_LEARN_HAM,
- RSPAMC_COMMAND_FUZZY_ADD,
- RSPAMC_COMMAND_FUZZY_DEL,
- RSPAMC_COMMAND_FUZZY_DELHASH,
- RSPAMC_COMMAND_STAT,
- RSPAMC_COMMAND_STAT_RESET,
- RSPAMC_COMMAND_COUNTERS,
- RSPAMC_COMMAND_UPTIME,
- RSPAMC_COMMAND_ADD_SYMBOL,
- RSPAMC_COMMAND_ADD_ACTION
- };
-
- struct rspamc_command {
- enum rspamc_command_type cmd;
- const char *name;
- const char *description;
- const char *path;
- gboolean is_controller;
- gboolean is_privileged;
- gboolean need_input;
- void (*command_output_func)(FILE *, ucl_object_t *obj);
- } rspamc_commands[] = {
- {
- .cmd = RSPAMC_COMMAND_SYMBOLS,
- .name = "symbols",
- .path = "checkv2",
- .description = "scan message and show symbols (default command)",
- .is_controller = FALSE,
- .is_privileged = FALSE,
- .need_input = TRUE,
- .command_output_func = rspamc_symbols_output
- },
- {
- .cmd = RSPAMC_COMMAND_LEARN_SPAM,
- .name = "learn_spam",
- .path = "learnspam",
- .description = "learn message as spam",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = TRUE,
- .command_output_func = NULL
- },
- {
- .cmd = RSPAMC_COMMAND_LEARN_HAM,
- .name = "learn_ham",
- .path = "learnham",
- .description = "learn message as ham",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = TRUE,
- .command_output_func = NULL
- },
- {
- .cmd = RSPAMC_COMMAND_FUZZY_ADD,
- .name = "fuzzy_add",
- .path = "fuzzyadd",
- .description =
- "add hashes from a message to the fuzzy storage (check -f and -w options for this command)",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = TRUE,
- .command_output_func = NULL
- },
- {
- .cmd = RSPAMC_COMMAND_FUZZY_DEL,
- .name = "fuzzy_del",
- .path = "fuzzydel",
- .description =
- "delete hashes from a message from the fuzzy storage (check -f option for this command)",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = TRUE,
- .command_output_func = NULL
- },
- {
- .cmd = RSPAMC_COMMAND_FUZZY_DELHASH,
- .name = "fuzzy_delhash",
- .path = "fuzzydelhash",
- .description =
- "delete a hash from fuzzy storage (check -f option for this command)",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = FALSE,
- .command_output_func = NULL
- },
- {
- .cmd = RSPAMC_COMMAND_STAT,
- .name = "stat",
- .path = "stat",
- .description = "show rspamd statistics",
- .is_controller = TRUE,
- .is_privileged = FALSE,
- .need_input = FALSE,
- .command_output_func = rspamc_stat_output,
- },
- {
- .cmd = RSPAMC_COMMAND_STAT_RESET,
- .name = "stat_reset",
- .path = "statreset",
- .description = "show and reset rspamd statistics (useful for graphs)",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = FALSE,
- .command_output_func = rspamc_stat_output
- },
- {
- .cmd = RSPAMC_COMMAND_COUNTERS,
- .name = "counters",
- .path = "counters",
- .description = "display rspamd symbols statistics",
- .is_controller = TRUE,
- .is_privileged = FALSE,
- .need_input = FALSE,
- .command_output_func = rspamc_counters_output
- },
- {
- .cmd = RSPAMC_COMMAND_UPTIME,
- .name = "uptime",
- .path = "auth",
- .description = "show rspamd uptime",
- .is_controller = TRUE,
- .is_privileged = FALSE,
- .need_input = FALSE,
- .command_output_func = rspamc_uptime_output
- },
- {
- .cmd = RSPAMC_COMMAND_ADD_SYMBOL,
- .name = "add_symbol",
- .path = "addsymbol",
- .description = "add or modify symbol settings in rspamd",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = FALSE,
- .command_output_func = NULL
- },
- {
- .cmd = RSPAMC_COMMAND_ADD_ACTION,
- .name = "add_action",
- .path = "addaction",
- .description = "add or modify action settings",
- .is_controller = TRUE,
- .is_privileged = TRUE,
- .need_input = FALSE,
- .command_output_func = NULL
- }
- };
-
- struct rspamc_callback_data {
- struct rspamc_command *cmd;
- gchar *filename;
- };
-
- gboolean
- rspamc_password_callback (const gchar *option_name,
- const gchar *value,
- gpointer data,
- GError **error)
- {
- guint plen = 8192;
- guint8 *map, *end;
- gsize sz;
-
- if (value != NULL) {
- if (value[0] == '/' || value[0] == '.') {
- /* Try to open file */
- map = rspamd_file_xmap (value, PROT_READ, &sz, 0);
-
- if (map == NULL) {
- /* Just use it as a string */
- password = g_strdup (value);
- }
- else {
- /* Strip trailing spaces */
- g_assert (sz > 0);
- end = map + sz - 1;
-
- while (g_ascii_isspace (*end) && end > map) {
- end --;
- }
-
- end ++;
- password = g_malloc (end - map + 1);
- rspamd_strlcpy (password, map, end - map + 1);
- munmap (map, sz);
- }
- }
- else {
- password = g_strdup (value);
- }
- }
- else {
- /* Read password from console */
- password = g_malloc0 (plen);
- plen = rspamd_read_passphrase (password, plen, 0, NULL);
- }
-
- if (plen == 0) {
- rspamd_fprintf (stderr, "Invalid password\n");
- exit (EXIT_FAILURE);
- }
-
- return TRUE;
- }
-
- /*
- * Parse command line
- */
- static void
- read_cmd_line (gint *argc, gchar ***argv)
- {
- GError *error = NULL;
- GOptionContext *context;
-
- /* Prepare parser */
- context = g_option_context_new ("- run rspamc client");
- g_option_context_set_summary (context,
- "Summary:\n Rspamd client version " RVERSION "\n Release id: " RID);
- g_option_context_add_main_entries (context, entries, NULL);
-
- /* Parse options */
- if (!g_option_context_parse (context, argc, argv, &error)) {
- fprintf (stderr, "option parsing failed: %s\n", error->message);
- exit (EXIT_FAILURE);
- }
-
- if (json || compact) {
- raw = TRUE;
- }
- /* Argc and argv are shifted after this function */
- }
-
- static gboolean
- rspamd_action_from_str_rspamc (const gchar *data, gint *result)
- {
- if (strcmp (data, "reject") == 0) {
- *result = METRIC_ACTION_REJECT;
- }
- else if (strcmp (data, "greylist") == 0) {
- *result = METRIC_ACTION_GREYLIST;
- }
- else if (strcmp (data, "add_header") == 0) {
- *result = METRIC_ACTION_ADD_HEADER;
- }
- else if (strcmp (data, "rewrite_subject") == 0) {
- *result = METRIC_ACTION_REWRITE_SUBJECT;
- }
- else if (strcmp (data, "add header") == 0) {
- *result = METRIC_ACTION_ADD_HEADER;
- }
- else if (strcmp (data, "rewrite subject") == 0) {
- *result = METRIC_ACTION_REWRITE_SUBJECT;
- }
- else if (strcmp (data, "soft_reject") == 0) {
- *result = METRIC_ACTION_SOFT_REJECT;
- }
- else if (strcmp (data, "soft reject") == 0) {
- *result = METRIC_ACTION_SOFT_REJECT;
- }
- else if (strcmp (data, "no_action") == 0) {
- *result = METRIC_ACTION_NOACTION;
- }
- else if (strcmp (data, "no action") == 0) {
- *result = METRIC_ACTION_NOACTION;
- }
- else {
- return FALSE;
- }
- return TRUE;
- }
-
- /*
- * Check rspamc command from string (used for arguments parsing)
- */
- static struct rspamc_command *
- check_rspamc_command (const gchar *cmd)
- {
- enum rspamc_command_type ct = 0;
- guint i;
-
- if (g_ascii_strcasecmp (cmd, "SYMBOLS") == 0 ||
- g_ascii_strcasecmp (cmd, "CHECK") == 0 ||
- g_ascii_strcasecmp (cmd, "REPORT") == 0) {
- /* These all are symbols, don't use other commands */
- ct = RSPAMC_COMMAND_SYMBOLS;
- }
- else if (g_ascii_strcasecmp (cmd, "LEARN_SPAM") == 0) {
- ct = RSPAMC_COMMAND_LEARN_SPAM;
- }
- else if (g_ascii_strcasecmp (cmd, "LEARN_HAM") == 0) {
- ct = RSPAMC_COMMAND_LEARN_HAM;
- }
- else if (g_ascii_strcasecmp (cmd, "FUZZY_ADD") == 0) {
- ct = RSPAMC_COMMAND_FUZZY_ADD;
- }
- else if (g_ascii_strcasecmp (cmd, "FUZZY_DEL") == 0) {
- ct = RSPAMC_COMMAND_FUZZY_DEL;
- }
- else if (g_ascii_strcasecmp (cmd, "FUZZY_DELHASH") == 0) {
- ct = RSPAMC_COMMAND_FUZZY_DELHASH;
- }
- else if (g_ascii_strcasecmp (cmd, "STAT") == 0) {
- ct = RSPAMC_COMMAND_STAT;
- }
- else if (g_ascii_strcasecmp (cmd, "STAT_RESET") == 0) {
- ct = RSPAMC_COMMAND_STAT_RESET;
- }
- else if (g_ascii_strcasecmp (cmd, "COUNTERS") == 0) {
- ct = RSPAMC_COMMAND_COUNTERS;
- }
- else if (g_ascii_strcasecmp (cmd, "UPTIME") == 0) {
- ct = RSPAMC_COMMAND_UPTIME;
- }
- else if (g_ascii_strcasecmp (cmd, "ADD_SYMBOL") == 0) {
- ct = RSPAMC_COMMAND_ADD_SYMBOL;
- }
- else if (g_ascii_strcasecmp (cmd, "ADD_ACTION") == 0) {
- ct = RSPAMC_COMMAND_ADD_ACTION;
- }
-
- for (i = 0; i < G_N_ELEMENTS (rspamc_commands); i++) {
- if (rspamc_commands[i].cmd == ct) {
- return &rspamc_commands[i];
- }
- }
-
- return NULL;
- }
-
- static void
- print_commands_list (void)
- {
- guint i;
- guint cmd_len = 0;
- gchar fmt_str[32];
-
- rspamd_fprintf (stdout, "Rspamc commands summary:\n");
-
- for (i = 0; i < G_N_ELEMENTS (rspamc_commands); i++) {
- gsize clen = strlen (rspamc_commands[i].name);
-
- if (clen > cmd_len) {
- cmd_len = clen;
- }
- }
-
- rspamd_snprintf (fmt_str, sizeof (fmt_str), " %%%ds (%%7s%%1s)\t%%s\n",
- cmd_len);
-
- for (i = 0; i < G_N_ELEMENTS (rspamc_commands); i++) {
- fprintf (stdout,
- fmt_str,
- rspamc_commands[i].name,
- rspamc_commands[i].is_controller ? "control" : "normal",
- rspamc_commands[i].is_privileged ? "*" : "",
- rspamc_commands[i].description);
- }
-
- rspamd_fprintf (stdout,
- "\n* is for privileged commands that may need password (see -P option)\n");
- rspamd_fprintf (stdout,
- "control commands use port 11334 while normal use 11333 by default (see -h option)\n");
- }
-
- static void
- add_options (GQueue *opts)
- {
- GString *numbuf;
- gchar **hdr, **rcpt;
-
- if (user_agent) {
- ADD_CLIENT_HEADER (opts, "User-Agent", user_agent);
- }
-
- if (ip != NULL) {
- rspamd_inet_addr_t *addr = NULL;
-
- if (!rspamd_parse_inet_address (&addr, ip, strlen (ip))) {
- /* Try to resolve */
- struct addrinfo hints, *res, *cur;
- gint r;
-
- memset (&hints, 0, sizeof (hints));
- hints.ai_socktype = SOCK_STREAM; /* Type of the socket */
- #ifdef AI_IDN
- hints.ai_flags = AI_NUMERICSERV|AI_IDN;
- #else
- hints.ai_flags = AI_NUMERICSERV;
- #endif
- hints.ai_family = AF_UNSPEC;
-
- if ((r = getaddrinfo (ip, "25", &hints, &res)) == 0) {
-
- cur = res;
- while (cur) {
- addr = rspamd_inet_address_from_sa (cur->ai_addr,
- cur->ai_addrlen);
-
- if (addr != NULL) {
- ip = g_strdup (rspamd_inet_address_to_string (addr));
- rspamd_inet_address_free (addr);
- break;
- }
-
- cur = cur->ai_next;
- }
-
- freeaddrinfo (res);
- }
- else {
- rspamd_fprintf (stderr, "address resolution for %s failed: %s\n",
- ip,
- gai_strerror (r));
- }
- }
- else {
- rspamd_inet_address_free (addr);
- }
-
- ADD_CLIENT_HEADER (opts, "Ip", ip);
- }
-
- if (from != NULL) {
- ADD_CLIENT_HEADER (opts, "From", from);
- }
-
- if (user != NULL) {
- ADD_CLIENT_HEADER (opts, "User", user);
- }
-
- if (rcpts != NULL) {
-
- for (rcpt = rcpts; *rcpt != NULL; rcpt ++) {
- ADD_CLIENT_HEADER (opts, "Rcpt", *rcpt);
- }
- }
-
- if (deliver_to != NULL) {
- ADD_CLIENT_HEADER (opts, "Deliver-To", deliver_to);
- }
-
- if (helo != NULL) {
- ADD_CLIENT_HEADER (opts, "Helo", helo);
- }
-
- if (hostname != NULL) {
- ADD_CLIENT_HEADER (opts, "Hostname", hostname);
- }
-
- if (password != NULL) {
- ADD_CLIENT_HEADER (opts, "Password", password);
- }
-
- if (pass_all) {
- ADD_CLIENT_HEADER (opts, "Pass", "all");
- }
-
- if (classifier) {
- ADD_CLIENT_HEADER (opts, "Classifier", classifier);
- }
-
- if (weight != 0) {
- numbuf = g_string_sized_new (8);
- rspamd_printf_gstring (numbuf, "%d", weight);
- ADD_CLIENT_HEADER (opts, "Weight", numbuf->str);
- }
-
- if (fuzzy_symbol != NULL) {
- ADD_CLIENT_HEADER (opts, "Symbol", fuzzy_symbol);
- }
-
- if (flag != 0) {
- numbuf = g_string_sized_new (8);
- rspamd_printf_gstring (numbuf, "%d", flag);
- ADD_CLIENT_HEADER (opts, "Flag", numbuf->str);
- }
-
- if (extended_urls) {
- ADD_CLIENT_HEADER (opts, "URL-Format", "extended");
- }
-
- if (profile) {
- ADD_CLIENT_HEADER (opts, "Profile", "true");
- }
-
- if (skip_images) {
- ADD_CLIENT_HEADER (opts, "Skip-Images", "true");
- }
-
- if (skip_attachments) {
- ADD_CLIENT_HEADER (opts, "Skip-Attachments", "true");
- }
-
- hdr = http_headers;
-
- while (hdr != NULL && *hdr != NULL) {
- gchar **kv = g_strsplit_set (*hdr, ":=", 2);
-
- if (kv == NULL || kv[1] == NULL) {
- ADD_CLIENT_HEADER (opts, *hdr, "");
-
- if (kv) {
- g_strfreev (kv);
- }
- }
- else {
- ADD_CLIENT_HEADER (opts, kv[0], kv[1]);
- }
-
- hdr ++;
- }
- }
-
- static void
- rspamc_symbol_output (FILE *out, const ucl_object_t *obj)
- {
- const ucl_object_t *val, *cur;
- ucl_object_iter_t it = NULL;
- gboolean first = TRUE;
-
- rspamd_fprintf (out, "Symbol: %s ", ucl_object_key (obj));
- val = ucl_object_lookup (obj, "score");
-
- if (val != NULL) {
- rspamd_fprintf (out, "(%.2f)", ucl_object_todouble (val));
- }
- val = ucl_object_lookup (obj, "options");
- if (val != NULL && val->type == UCL_ARRAY) {
- rspamd_fprintf (out, "[");
-
- while ((cur = ucl_object_iterate (val, &it, TRUE)) != NULL) {
- if (first) {
- rspamd_fprintf (out, "%s", ucl_object_tostring (cur));
- first = FALSE;
- }
- else {
- rspamd_fprintf (out, ", %s", ucl_object_tostring (cur));
- }
- }
- rspamd_fprintf (out, "]");
- }
- rspamd_fprintf (out, "\n");
- }
-
- static gint
- rspamc_symbols_sort_func (gconstpointer a, gconstpointer b)
- {
- ucl_object_t * const *ua = a, * const *ub = b;
-
- return strcmp (ucl_object_key (*ua), ucl_object_key (*ub));
- }
-
- #define PRINT_PROTOCOL_STRING(ucl_name, output_message) do { \
- elt = ucl_object_lookup (obj, (ucl_name)); \
- if (elt) { \
- rspamd_fprintf (out, output_message ": %s\n", ucl_object_tostring (elt)); \
- } \
- } while (0)
-
- static void
- rspamc_metric_output (FILE *out, const ucl_object_t *obj)
- {
- ucl_object_iter_t it = NULL;
- const ucl_object_t *cur, *elt;
- gdouble score = 0, required_score = 0;
- gint got_scores = 0, action = METRIC_ACTION_MAX;
- GPtrArray *sym_ptr;
- guint i;
-
- sym_ptr = g_ptr_array_new ();
- rspamd_fprintf (out, "[Metric: default]\n");
-
- elt = ucl_object_lookup (obj, "required_score");
-
- if (elt) {
- required_score = ucl_object_todouble (elt);
- got_scores++;
- }
-
- elt = ucl_object_lookup (obj, "score");
-
- if (elt) {
- score = ucl_object_todouble (elt);
- got_scores++;
- }
-
- PRINT_PROTOCOL_STRING ("action", "Action");
- /* Defined by previous macro */
- if (elt && rspamd_action_from_str_rspamc (ucl_object_tostring (elt), &action)) {
- rspamd_fprintf (out, "Spam: %s\n", action < METRIC_ACTION_GREYLIST ?
- "true" : "false");
- }
-
- PRINT_PROTOCOL_STRING ("subject", "Subject");
-
- if (got_scores == 2) {
- rspamd_fprintf (out,
- "Score: %.2f / %.2f\n",
- score,
- required_score);
- }
-
- elt = ucl_object_lookup (obj, "symbols");
-
- while (elt && (cur = ucl_object_iterate (elt, &it, true)) != NULL) {
- if (cur->type == UCL_OBJECT) {
- g_ptr_array_add (sym_ptr, (void *)cur);
- }
- }
-
- g_ptr_array_sort (sym_ptr, rspamc_symbols_sort_func);
-
- for (i = 0; i < sym_ptr->len; i ++) {
- cur = (const ucl_object_t *)g_ptr_array_index (sym_ptr, i);
- rspamc_symbol_output (out, cur);
- }
-
- g_ptr_array_free (sym_ptr, TRUE);
- }
-
- static gint
- rspamc_profile_sort_func (gconstpointer a, gconstpointer b)
- {
- ucl_object_t * const *ua = a, * const *ub = b;
-
- return ucl_object_compare (*ua, *ub);
- }
-
- static void
- rspamc_profile_output (FILE *out, const ucl_object_t *obj)
- {
- ucl_object_iter_t it = NULL;
- const ucl_object_t *cur;
- guint i;
- GPtrArray *ar;
-
- ar = g_ptr_array_sized_new (obj->len);
-
- while ((cur = ucl_object_iterate (obj, &it, true)) != NULL) {
- g_ptr_array_add (ar, (void *)cur);
- }
-
- g_ptr_array_sort (ar, rspamc_profile_sort_func);
-
- for (i = 0; i < ar->len; i ++) {
- cur = (const ucl_object_t *)g_ptr_array_index (ar, i);
- rspamd_fprintf (out, "\t%s: %.3f usec\n",
- ucl_object_key (cur), ucl_object_todouble (cur));
- }
-
- g_ptr_array_free (ar, TRUE);
- }
-
- static void
- rspamc_symbols_output (FILE *out, ucl_object_t *obj)
- {
- ucl_object_iter_t mit = NULL;
- const ucl_object_t *cmesg, *elt;
- gchar *emitted;
-
- rspamc_metric_output (out, obj);
-
- PRINT_PROTOCOL_STRING ("message-id", "Message-ID");
- PRINT_PROTOCOL_STRING ("queue-id", "Queue-ID");
-
- elt = ucl_object_lookup (obj, "urls");
-
- if (elt) {
- if (!extended_urls || compact) {
- emitted = ucl_object_emit (elt, UCL_EMIT_JSON_COMPACT);
- }
- else {
- emitted = ucl_object_emit (elt, UCL_EMIT_JSON);
- }
-
- rspamd_fprintf (out, "Urls: %s\n", emitted);
- free (emitted);
- }
-
- elt = ucl_object_lookup (obj, "emails");
-
- if (elt) {
- if (!extended_urls || compact) {
- emitted = ucl_object_emit (elt, UCL_EMIT_JSON_COMPACT);
- }
- else {
- emitted = ucl_object_emit (elt, UCL_EMIT_JSON);
- }
-
- rspamd_fprintf (out, "Emails: %s\n", emitted);
- free (emitted);
- }
-
- PRINT_PROTOCOL_STRING ("error", "Scan error");
-
- elt = ucl_object_lookup (obj, "messages");
- if (elt && elt->type == UCL_OBJECT) {
- mit = NULL;
- while ((cmesg = ucl_object_iterate (elt, &mit, true)) != NULL) {
- rspamd_fprintf (out, "Message - %s: %s\n",
- ucl_object_key (cmesg), ucl_object_tostring (cmesg));
- }
- }
-
- PRINT_PROTOCOL_STRING ("dkim-signature", "DKIM-Signature");
-
- elt = ucl_object_lookup (obj, "profile");
-
- if (elt) {
- rspamd_fprintf (out, "Profile data:\n");
- rspamc_profile_output (out, elt);
- }
- }
-
- static void
- rspamc_uptime_output (FILE *out, ucl_object_t *obj)
- {
- const ucl_object_t *elt;
- int64_t seconds, days, hours, minutes;
-
- elt = ucl_object_lookup (obj, "version");
- if (elt != NULL) {
- rspamd_fprintf (out, "Rspamd version: %s\n", ucl_object_tostring (
- elt));
- }
-
- elt = ucl_object_lookup (obj, "uptime");
- if (elt != NULL) {
- rspamd_printf ("Uptime: ");
- seconds = ucl_object_toint (elt);
- if (seconds >= 2 * 3600) {
- days = seconds / 86400;
- hours = seconds / 3600 - days * 24;
- minutes = seconds / 60 - hours * 60 - days * 1440;
- rspamd_printf ("%L day%s %L hour%s %L minute%s\n", days,
- days > 1 ? "s" : "", hours, hours > 1 ? "s" : "",
- minutes, minutes > 1 ? "s" : "");
- }
- /* If uptime is less than 1 minute print only seconds */
- else if (seconds / 60 == 0) {
- rspamd_printf ("%L second%s\n", seconds,
- (gint)seconds > 1 ? "s" : "");
- }
- /* Else print the minutes and seconds. */
- else {
- hours = seconds / 3600;
- minutes = seconds / 60 - hours * 60;
- seconds -= hours * 3600 + minutes * 60;
- rspamd_printf ("%L hour %L minute%s %L second%s\n", hours,
- minutes, minutes > 1 ? "s" : "",
- seconds, seconds > 1 ? "s" : "");
- }
- }
- }
-
- static gint
- rspamc_counters_sort (const ucl_object_t **o1, const ucl_object_t **o2)
- {
- gint order1 = 0, order2 = 0, c;
- const ucl_object_t *elt1, *elt2;
- gboolean inverse = FALSE;
- gchar **args;
-
- if (sort != NULL) {
- args = g_strsplit_set (sort, ":", 2);
- if (args && args[0]) {
- if (args[1] && g_ascii_strcasecmp (args[1], "desc") == 0) {
- inverse = TRUE;
- }
-
- if (g_ascii_strcasecmp (args[0], "name") == 0) {
- elt1 = ucl_object_lookup (*o1, "symbol");
- elt2 = ucl_object_lookup (*o2, "symbol");
-
- if (elt1 && elt2) {
- c = strcmp (ucl_object_tostring (elt1),
- ucl_object_tostring (elt2));
-
- order1 = c > 0 ? 1 : 0;
- order2 = c < 0 ? 1 : 0;
- }
- }
- else if (g_ascii_strcasecmp (args[0], "weight") == 0) {
- elt1 = ucl_object_lookup (*o1, "weight");
- elt2 = ucl_object_lookup (*o2, "weight");
-
- if (elt1 && elt2) {
- order1 = ucl_object_todouble (elt1) * 1000.0;
- order2 = ucl_object_todouble (elt2) * 1000.0;
- }
- }
- else if (g_ascii_strcasecmp (args[0], "frequency") == 0) {
- elt1 = ucl_object_lookup (*o1, "frequency");
- elt2 = ucl_object_lookup (*o2, "frequency");
-
- if (elt1 && elt2) {
- order1 = ucl_object_todouble (elt1) * 100000;
- order2 = ucl_object_todouble (elt2) * 100000;
- }
- }
- else if (g_ascii_strcasecmp (args[0], "time") == 0) {
- elt1 = ucl_object_lookup (*o1, "time");
- elt2 = ucl_object_lookup (*o2, "time");
-
- if (elt1 && elt2) {
- order1 = ucl_object_todouble (elt1) * 1000000;
- order2 = ucl_object_todouble (elt2) * 1000000;
- }
- }
- else if (g_ascii_strcasecmp (args[0], "hits") == 0) {
- elt1 = ucl_object_lookup (*o1, "hits");
- elt2 = ucl_object_lookup (*o2, "hits");
-
- if (elt1 && elt2) {
- order1 = ucl_object_toint (elt1);
- order2 = ucl_object_toint (elt2);
- }
- }
-
- g_strfreev (args);
- }
- }
-
- return (inverse ? (order2 - order1) : (order1 - order2));
- }
-
- static void
- rspamc_counters_output (FILE *out, ucl_object_t *obj)
- {
- const ucl_object_t *cur, *sym, *weight, *freq, *freq_dev, *nhits;
- ucl_object_iter_t iter = NULL;
- gchar fmt_buf[64], dash_buf[82], sym_buf[82];
- gint l, max_len = INT_MIN, i;
- static const gint dashes = 44;
-
- if (obj->type != UCL_ARRAY) {
- rspamd_printf ("Bad output\n");
- return;
- }
-
- /* Sort symbols by their order */
- if (sort != NULL) {
- ucl_object_array_sort (obj, rspamc_counters_sort);
- }
-
- /* Find maximum width of symbol's name */
- while ((cur = ucl_object_iterate (obj, &iter, true)) != NULL) {
- sym = ucl_object_lookup (cur, "symbol");
- if (sym != NULL) {
- l = sym->len;
- if (l > max_len) {
- max_len = MIN (sizeof (dash_buf) - dashes - 1, l);
- }
- }
- }
-
- rspamd_snprintf (fmt_buf, sizeof (fmt_buf),
- "| %%3s | %%%ds | %%7s | %%13s | %%7s |\n", max_len);
- memset (dash_buf, '-', dashes + max_len);
- dash_buf[dashes + max_len] = '\0';
-
- printf ("Symbols cache\n");
- printf (" %s \n", dash_buf);
- if (tty) {
- printf ("\033[1m");
- }
- printf (fmt_buf, "Pri", "Symbol", "Weight", "Frequency", "Hits");
- printf (" %s \n", dash_buf);
- printf (fmt_buf, "", "", "", "hits/min", "");
- if (tty) {
- printf ("\033[0m");
- }
- rspamd_snprintf (fmt_buf, sizeof (fmt_buf),
- "| %%3d | %%%ds | %%7.1f | %%6.3f(%%5.3f) | %%7ju |\n", max_len);
-
- iter = NULL;
- i = 0;
- while ((cur = ucl_object_iterate (obj, &iter, true)) != NULL) {
- printf (" %s \n", dash_buf);
- sym = ucl_object_lookup (cur, "symbol");
- weight = ucl_object_lookup (cur, "weight");
- freq = ucl_object_lookup (cur, "frequency");
- freq_dev = ucl_object_lookup (cur, "frequency_stddev");
- nhits = ucl_object_lookup (cur, "hits");
-
- if (sym && weight && freq && nhits) {
- const gchar *sym_name;
-
- if (sym->len > max_len) {
- rspamd_snprintf (sym_buf, sizeof (sym_buf), "%*s...",
- (max_len - 3), ucl_object_tostring (sym));
- sym_name = sym_buf;
- }
- else {
- sym_name = ucl_object_tostring (sym);
- }
-
- printf (fmt_buf, i,
- sym_name,
- ucl_object_todouble (weight),
- ucl_object_todouble (freq) * 60.0,
- ucl_object_todouble (freq_dev) * 60.0,
- (uintmax_t)ucl_object_toint (nhits));
- }
- i++;
- }
- printf (" %s \n", dash_buf);
- }
-
- static void
- rspamc_stat_actions (ucl_object_t *obj, GString *out, gint64 scanned)
- {
- const ucl_object_t *actions = ucl_object_lookup (obj, "actions"), *cur;
- ucl_object_iter_t iter = NULL;
- gint64 spam, ham;
-
- if (actions && ucl_object_type (actions) == UCL_OBJECT) {
- while ((cur = ucl_object_iterate (actions, &iter, true)) != NULL) {
- gint64 cnt = ucl_object_toint (cur);
- rspamd_printf_gstring (out, "Messages with action %s: %L"
- ", %.2f%%\n", ucl_object_key (cur), cnt,
- ((gdouble)cnt / (gdouble)scanned) * 100.);
- }
- }
-
- spam = ucl_object_toint (ucl_object_lookup (obj, "spam_count"));
- ham = ucl_object_toint (ucl_object_lookup (obj, "ham_count"));
- rspamd_printf_gstring (out, "Messages treated as spam: %L, %.2f%%\n", spam,
- ((gdouble)spam / (gdouble)scanned) * 100.);
- rspamd_printf_gstring (out, "Messages treated as ham: %L, %.2f%%\n", ham,
- ((gdouble)ham / (gdouble)scanned) * 100.);
- }
-
- static void
- rspamc_stat_statfile (const ucl_object_t *obj, GString *out)
- {
- gint64 version, size, blocks, used_blocks, nlanguages, nusers;
- const gchar *label, *symbol, *type;
-
- version = ucl_object_toint (ucl_object_lookup (obj, "revision"));
- size = ucl_object_toint (ucl_object_lookup (obj, "size"));
- blocks = ucl_object_toint (ucl_object_lookup (obj, "total"));
- used_blocks = ucl_object_toint (ucl_object_lookup (obj, "used"));
- label = ucl_object_tostring (ucl_object_lookup (obj, "label"));
- symbol = ucl_object_tostring (ucl_object_lookup (obj, "symbol"));
- type = ucl_object_tostring (ucl_object_lookup (obj, "type"));
- nlanguages = ucl_object_toint (ucl_object_lookup (obj, "languages"));
- nusers = ucl_object_toint (ucl_object_lookup (obj, "users"));
-
- if (label) {
- rspamd_printf_gstring (out, "Statfile: %s <%s> type: %s; ", symbol,
- label, type);
- }
- else {
- rspamd_printf_gstring (out, "Statfile: %s type: %s; ", symbol, type);
- }
- rspamd_printf_gstring (out, "length: %hL; free blocks: %hL; total blocks: %hL; "
- "free: %.2f%%; learned: %L; users: %L; languages: %L\n",
- size,
- blocks - used_blocks, blocks,
- blocks > 0 ? (blocks - used_blocks) * 100.0 / (gdouble)blocks : 0,
- version,
- nusers, nlanguages);
- }
-
- static void
- rspamc_stat_output (FILE *out, ucl_object_t *obj)
- {
- GString *out_str;
- ucl_object_iter_t iter = NULL;
- const ucl_object_t *st, *cur;
- gint64 scanned;
-
- out_str = g_string_sized_new (BUFSIZ);
-
- scanned = ucl_object_toint (ucl_object_lookup (obj, "scanned"));
- rspamd_printf_gstring (out_str, "Messages scanned: %L\n",
- scanned);
-
- if (scanned > 0) {
- rspamc_stat_actions (obj, out_str, scanned);
- }
-
- rspamd_printf_gstring (out_str, "Messages learned: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "learned")));
- rspamd_printf_gstring (out_str, "Connections count: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "connections")));
- rspamd_printf_gstring (out_str, "Control connections count: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "control_connections")));
- /* Pools */
- rspamd_printf_gstring (out_str, "Pools allocated: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "pools_allocated")));
- rspamd_printf_gstring (out_str, "Pools freed: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "pools_freed")));
- rspamd_printf_gstring (out_str, "Bytes allocated: %HL\n",
- ucl_object_toint (ucl_object_lookup (obj, "bytes_allocated")));
- rspamd_printf_gstring (out_str, "Memory chunks allocated: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "chunks_allocated")));
- rspamd_printf_gstring (out_str, "Shared chunks allocated: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "shared_chunks_allocated")));
- rspamd_printf_gstring (out_str, "Chunks freed: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "chunks_freed")));
- rspamd_printf_gstring (out_str, "Oversized chunks: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "chunks_oversized")));
- /* Fuzzy */
-
- st = ucl_object_lookup (obj, "fuzzy_hashes");
- if (st) {
- ucl_object_iter_t it = NULL;
- const ucl_object_t *cur;
- gint64 stored = 0;
-
- while ((cur = ucl_iterate_object (st, &it, true)) != NULL) {
- rspamd_printf_gstring (out_str, "Fuzzy hashes in storage \"%s\": %L\n",
- ucl_object_key (cur),
- ucl_object_toint (cur));
- stored += ucl_object_toint (cur);
- }
-
- rspamd_printf_gstring (out_str, "Fuzzy hashes stored: %L\n",
- stored);
- }
-
- st = ucl_object_lookup (obj, "fuzzy_checked");
- if (st != NULL && ucl_object_type (st) == UCL_ARRAY) {
- rspamd_printf_gstring (out_str, "Fuzzy hashes checked: ");
- iter = NULL;
-
- while ((cur = ucl_object_iterate (st, &iter, true)) != NULL) {
- rspamd_printf_gstring (out_str, "%hL ", ucl_object_toint (cur));
- }
-
- rspamd_printf_gstring (out_str, "\n");
- }
-
- st = ucl_object_lookup (obj, "fuzzy_found");
- if (st != NULL && ucl_object_type (st) == UCL_ARRAY) {
- rspamd_printf_gstring (out_str, "Fuzzy hashes found: ");
- iter = NULL;
-
- while ((cur = ucl_object_iterate (st, &iter, true)) != NULL) {
- rspamd_printf_gstring (out_str, "%hL ", ucl_object_toint (cur));
- }
-
- rspamd_printf_gstring (out_str, "\n");
- }
-
- st = ucl_object_lookup (obj, "statfiles");
- if (st != NULL && ucl_object_type (st) == UCL_ARRAY) {
- iter = NULL;
-
- while ((cur = ucl_object_iterate (st, &iter, true)) != NULL) {
- rspamc_stat_statfile (cur, out_str);
- }
- }
- rspamd_printf_gstring (out_str, "Total learns: %L\n",
- ucl_object_toint (ucl_object_lookup (obj, "total_learns")));
-
- rspamd_fprintf (out, "%v", out_str);
- }
-
- static void
- rspamc_output_headers (FILE *out, struct rspamd_http_message *msg)
- {
- struct rspamd_http_header *h, *htmp;
-
- HASH_ITER (hh, msg->headers, h, htmp) {
- rspamd_fprintf (out, "%T: %T\n", &h->name, &h->value);
- }
-
- rspamd_fprintf (out, "\n");
- }
-
- static void
- rspamc_mime_output (FILE *out, ucl_object_t *result, GString *input,
- gdouble time, GError *err)
- {
- const ucl_object_t *cur, *res, *syms;
- ucl_object_iter_t it = NULL;
- const gchar *action = "no action", *line_end = "\r\n", *p;
- gchar scorebuf[32];
- GString *symbuf, *folded_symbuf, *added_headers;
- gint act = 0;
- goffset headers_pos;
- gdouble score = 0.0, required_score = 0.0;
- gboolean is_spam = FALSE;
- gchar *json_header, *json_header_encoded, *sc;
- enum rspamd_newlines_type nl_type = RSPAMD_TASK_NEWLINES_CRLF;
-
- headers_pos = rspamd_string_find_eoh (input, NULL);
-
- if (headers_pos == -1) {
- rspamd_fprintf (stderr,"cannot find end of headers position");
- return;
- }
-
- p = input->str + headers_pos;
-
- if (headers_pos > 1 && *(p - 1) == '\n') {
- if (headers_pos > 2 && *(p - 2) == '\r') {
- line_end = "\r\n";
- nl_type = RSPAMD_TASK_NEWLINES_CRLF;
- }
- else {
- line_end = "\n";
- nl_type = RSPAMD_TASK_NEWLINES_LF;
- }
- }
- else if (headers_pos > 1 && *(p - 1) == '\r') {
- line_end = "\r";
- nl_type = RSPAMD_TASK_NEWLINES_CR;
- }
-
- added_headers = g_string_sized_new (127);
-
- if (result) {
- res = ucl_object_lookup (result, "action");
-
- if (res) {
- action = ucl_object_tostring (res);
- }
-
- res = ucl_object_lookup (result, "score");
- if (res) {
- score = ucl_object_todouble (res);
- }
-
- res = ucl_object_lookup (result, "required_score");
- if (res) {
- required_score = ucl_object_todouble (res);
- }
-
- rspamd_action_from_str_rspamc (action, &act);
-
- if (act < METRIC_ACTION_GREYLIST) {
- is_spam = TRUE;
- }
-
- rspamd_printf_gstring (added_headers, "X-Spam-Scanner: %s%s",
- "rspamc " RVERSION, line_end);
- rspamd_printf_gstring (added_headers, "X-Spam-Scan-Time: %.3f%s",
- time, line_end);
-
- /*
- * TODO: add rmilter_headers support here
- */
- if (is_spam) {
- rspamd_printf_gstring (added_headers, "X-Spam: yes%s", line_end);
- }
-
- rspamd_printf_gstring (added_headers, "X-Spam-Action: %s%s",
- action, line_end);
- rspamd_printf_gstring (added_headers, "X-Spam-Score: %.2f / %.2f%s",
- score, required_score, line_end);
-
- /* SA style stars header */
- for (sc = scorebuf; sc < scorebuf + sizeof (scorebuf) - 1 && score > 0;
- sc ++, score -= 1.0) {
- *sc = '*';
- }
-
- *sc = '\0';
- rspamd_printf_gstring (added_headers, "X-Spam-Level: %s%s",
- scorebuf, line_end);
-
- /* Short description of all symbols */
- symbuf = g_string_sized_new (64);
- syms = ucl_object_lookup (result, "symbols");
-
- while (syms && (cur = ucl_object_iterate (syms, &it, true)) != NULL) {
-
- if (ucl_object_type (cur) == UCL_OBJECT) {
- rspamd_printf_gstring (symbuf, "%s,", ucl_object_key (cur));
- }
- }
- /* Trim the last comma */
- if (symbuf->str[symbuf->len - 1] == ',') {
- g_string_erase (symbuf, symbuf->len - 1, 1);
- }
-
- folded_symbuf = rspamd_header_value_fold ("X-Spam-Symbols",
- symbuf->str,
- 0, nl_type, ",");
- rspamd_printf_gstring (added_headers, "X-Spam-Symbols: %v%s",
- folded_symbuf, line_end, ",");
-
- g_string_free (folded_symbuf, TRUE);
- g_string_free (symbuf, TRUE);
-
- if (ucl_object_lookup (result, "dkim-signature")) {
- rspamd_printf_gstring (added_headers, "DKIM-Signature: %s%s",
- ucl_object_tostring (
- ucl_object_lookup (result, "dkim-signature")),
- line_end);
- }
-
- if (json || raw || compact) {
- /* We also append json data as a specific header */
- if (json) {
- json_header = ucl_object_emit (result,
- compact ? UCL_EMIT_JSON_COMPACT : UCL_EMIT_JSON);
- }
- else {
- json_header = ucl_object_emit (result,
- compact ? UCL_EMIT_JSON_COMPACT : UCL_EMIT_CONFIG);
- }
-
- json_header_encoded = rspamd_encode_base64_fold (json_header,
- strlen (json_header), 60, NULL, nl_type);
- free (json_header);
- rspamd_printf_gstring (added_headers,
- "X-Spam-Result: %s%s",
- json_header_encoded, line_end);
- g_free (json_header_encoded);
- }
-
- ucl_object_unref (result);
- }
- else {
- rspamd_printf_gstring (added_headers, "X-Spam-Scanner: %s%s",
- "rspamc " RVERSION, line_end);
- rspamd_printf_gstring (added_headers, "X-Spam-Scan-Time: %.3f%s",
- time, line_end);
- rspamd_printf_gstring (added_headers, "X-Spam-Error: %e%s",
- err, line_end);
- }
-
- /* Write message */
- if (rspamd_fprintf (out, "%*s", (gint)headers_pos, input->str)
- == headers_pos) {
- if (rspamd_fprintf (out, "%v", added_headers)
- == (gint)added_headers->len) {
- rspamd_fprintf (out, "%s", input->str + headers_pos);
- }
- }
-
- g_string_free (added_headers, TRUE);
- }
-
- static void
- rspamc_client_execute_cmd (struct rspamc_command *cmd, ucl_object_t *result,
- GString *input, gdouble time, GError *err)
- {
- gchar **eargv;
- gint eargc, infd, outfd, errfd;
- GError *exec_err = NULL;
- GPid cld;
- FILE *out;
- gchar *ucl_out;
-
- if (!g_shell_parse_argv (execute, &eargc, &eargv, &err)) {
- rspamd_fprintf (stderr, "Cannot execute %s: %e", execute, err);
- g_error_free (err);
-
- return;
- }
-
- if (!g_spawn_async_with_pipes (NULL, eargv, NULL,
- G_SPAWN_SEARCH_PATH|G_SPAWN_DO_NOT_REAP_CHILD, NULL, NULL, &cld,
- &infd, &outfd, &errfd, &exec_err)) {
-
- rspamd_fprintf (stderr, "Cannot execute %s: %e", execute, exec_err);
- g_error_free (exec_err);
-
- exit (EXIT_FAILURE);
- }
- else {
- children = g_list_prepend (children, GSIZE_TO_POINTER (cld));
- out = fdopen (infd, "w");
-
- if (cmd->cmd == RSPAMC_COMMAND_SYMBOLS && mime_output && input) {
- rspamc_mime_output (out, result, input, time, err);
- }
- else if (result) {
- if (raw || cmd->command_output_func == NULL) {
- if (json) {
- ucl_out = ucl_object_emit (result,
- compact ? UCL_EMIT_JSON_COMPACT : UCL_EMIT_JSON);
- }
- else {
- ucl_out = ucl_object_emit (result,
- compact ? UCL_EMIT_JSON_COMPACT : UCL_EMIT_CONFIG);
- }
- rspamd_fprintf (out, "%s", ucl_out);
- free (ucl_out);
- }
- else {
- cmd->command_output_func (out, result);
- }
-
- ucl_object_unref (result);
- }
- else {
- rspamd_fprintf (out, "%e\n", err);
- }
-
- fflush (out);
-
- fclose (out);
- }
-
- g_strfreev (eargv);
- }
-
- static void
- rspamc_client_cb (struct rspamd_client_connection *conn,
- struct rspamd_http_message *msg,
- const gchar *name, ucl_object_t *result, GString *input,
- gpointer ud, gdouble start_time, gdouble send_time,
- GError *err)
- {
- gchar *ucl_out;
- struct rspamc_callback_data *cbdata = (struct rspamc_callback_data *)ud;
- struct rspamc_command *cmd;
- FILE *out = stdout;
- gdouble finish = rspamd_get_ticks (FALSE), diff;
- const gchar *body;
- gsize body_len;
-
- cmd = cbdata->cmd;
-
- if (send_time > 0) {
- diff = finish - send_time;
- }
- else {
- diff = finish - start_time;
- }
-
- if (execute) {
- /* Pass all to the external command */
- rspamc_client_execute_cmd (cmd, result, input, diff, err);
- }
- else {
-
- if (cmd->cmd == RSPAMC_COMMAND_SYMBOLS && mime_output && input) {
- rspamc_mime_output (out, result, input, diff, err);
- }
- else {
- if (cmd->need_input) {
- if (!compact) {
- rspamd_fprintf (out, "Results for file: %s (%.3f seconds)\n",
- cbdata->filename, diff);
- }
- }
- else {
- if (!compact) {
- rspamd_fprintf (out, "Results for command: %s (%.3f seconds)\n",
- cmd->name, diff);
- }
- }
-
- if (result != NULL) {
- if (headers && msg != NULL) {
- rspamc_output_headers (out, msg);
- }
- if (raw || cmd->command_output_func == NULL) {
- if (cmd->need_input) {
- ucl_object_insert_key (result,
- ucl_object_fromstring (cbdata->filename),
- "filename", 0,
- false);
- }
-
- ucl_object_insert_key (result,
- ucl_object_fromdouble (diff),
- "scan_time", 0,
- false);
-
- if (json) {
- ucl_out = ucl_object_emit (result,
- compact ? UCL_EMIT_JSON_COMPACT : UCL_EMIT_JSON);
- }
- else {
- ucl_out = ucl_object_emit (result,
- compact ? UCL_EMIT_JSON_COMPACT : UCL_EMIT_CONFIG);
- }
-
- rspamd_fprintf (out, "%s", ucl_out);
- free (ucl_out);
- }
- else {
- cmd->command_output_func (out, result);
- }
-
- ucl_object_unref (result);
- }
- else if (err != NULL) {
- rspamd_fprintf (out, "%s\n", err->message);
-
- if (json && msg != NULL) {
- body = rspamd_http_message_get_body (msg, &body_len);
-
- if (body) {
- /* We can also output the resulting json */
- rspamd_fprintf (out, "%*s\n", (gint)body_len, body);
- }
- }
- }
- rspamd_fprintf (out, "\n");
- }
-
- fflush (out);
- }
-
- rspamd_client_destroy (conn);
- g_free (cbdata->filename);
- g_free (cbdata);
- }
-
- static void
- rspamc_process_input (struct event_base *ev_base, struct rspamc_command *cmd,
- FILE *in, const gchar *name, GQueue *attrs)
- {
- struct rspamd_client_connection *conn;
- gchar *hostbuf = NULL, *p;
- guint16 port;
- GError *err = NULL;
- struct rspamc_callback_data *cbdata;
-
- if (connect_str[0] == '[') {
- p = strrchr (connect_str, ']');
-
- if (p != NULL) {
- hostbuf = g_malloc (p - connect_str);
- rspamd_strlcpy (hostbuf, connect_str + 1, p - connect_str);
- p ++;
- }
- else {
- p = connect_str;
- }
- }
- else {
- p = connect_str;
- }
-
- p = strrchr (p, ':');
-
- if (!hostbuf) {
- if (p != NULL) {
- hostbuf = g_malloc (p - connect_str + 1);
- rspamd_strlcpy (hostbuf, connect_str, p - connect_str + 1);
- }
- else {
- hostbuf = g_strdup (connect_str);
- }
- }
-
- if (p != NULL) {
- port = strtoul (p + 1, NULL, 10);
- }
- else {
- /*
- * If we connect to localhost, 127.0.0.1 or ::1, then try controller
- * port first
- */
-
- if (strcmp (hostbuf, "localhost") == 0 ||
- strcmp (hostbuf, "127.0.0.1") == 0 ||
- strcmp (hostbuf, "::1") == 0 ||
- strcmp (hostbuf, "[::1]") == 0) {
- port = DEFAULT_CONTROL_PORT;
- }
- else {
- port = cmd->is_controller ? DEFAULT_CONTROL_PORT : DEFAULT_PORT;
- }
-
- }
-
- conn = rspamd_client_init (ev_base, hostbuf, port, timeout, key);
-
- if (conn != NULL) {
- cbdata = g_malloc0 (sizeof (struct rspamc_callback_data));
- cbdata->cmd = cmd;
-
- if (name) {
- cbdata->filename = g_strdup (name);
- }
-
- if (cmd->need_input) {
- rspamd_client_command (conn, cmd->path, attrs, in, rspamc_client_cb,
- cbdata, compressed, dictionary, cbdata->filename, &err);
- }
- else {
- rspamd_client_command (conn,
- cmd->path,
- attrs,
- NULL,
- rspamc_client_cb,
- cbdata,
- compressed,
- dictionary,
- cbdata->filename,
- &err);
- }
- }
- else {
- rspamd_fprintf (stderr, "cannot connect to %s\n", connect_str);
- exit (EXIT_FAILURE);
- }
-
- g_free (hostbuf);
- }
-
- static gsize
- rspamd_dirent_size (DIR * dirp)
- {
- goffset name_max;
- gsize name_end;
-
- #if defined(HAVE_FPATHCONF) && defined(HAVE_DIRFD) \
- && defined(_PC_NAME_MAX)
- name_max = fpathconf (dirfd (dirp), _PC_NAME_MAX);
-
-
- # if defined(NAME_MAX)
- if (name_max == -1) {
- name_max = (NAME_MAX > 255) ? NAME_MAX : 255;
- }
- # else
- if (name_max == -1) {
- return (size_t)(-1);
- }
- # endif
- #else
- # if defined(NAME_MAX)
- name_max = (NAME_MAX > 255) ? NAME_MAX : 255;
- # else
- # error "buffer size for readdir_r cannot be determined"
- # endif
- #endif
-
- name_end = G_STRUCT_OFFSET (struct dirent, d_name) + name_max + 1;
-
- return (name_end > sizeof (struct dirent) ? name_end : sizeof(struct dirent));
- }
-
- static void
- rspamc_process_dir (struct event_base *ev_base, struct rspamc_command *cmd,
- const gchar *name, GQueue *attrs)
- {
- DIR *d;
- GPatternSpec **ex;
- struct dirent *pentry;
- gint cur_req = 0, r;
- gchar fpath[PATH_MAX];
- FILE *in;
- struct stat st;
- gboolean is_reg, is_dir, skip;
-
- d = opendir (name);
-
- if (d != NULL) {
- while ((pentry = readdir (d))!= NULL) {
-
- if (pentry->d_name[0] == '.') {
- continue;
- }
-
- r = rspamd_snprintf (fpath, sizeof (fpath), "%s%c%s",
- name, G_DIR_SEPARATOR,
- pentry->d_name);
-
- /* Check exclude */
- ex = exclude_compiled;
- skip = FALSE;
- while (ex != NULL && *ex != NULL) {
- if (g_pattern_match (*ex, r, fpath, NULL)) {
- skip = TRUE;
- break;
- }
-
- ex ++;
- }
-
- if (skip) {
- continue;
- }
-
- is_reg = FALSE;
- is_dir = FALSE;
-
- #if (defined(_DIRENT_HAVE_D_TYPE) || defined(__APPLE__)) && defined(DT_UNKNOWN)
- if (pentry->d_type == DT_UNKNOWN) {
- /* Fallback to lstat */
- if (lstat (fpath, &st) == -1) {
- rspamd_fprintf (stderr, "cannot stat file %s: %s\n",
- fpath, strerror (errno));
- continue;
- }
-
- is_dir = S_ISDIR (st.st_mode);
- is_reg = S_ISREG (st.st_mode);
- }
- else {
- if (pentry->d_type == DT_REG) {
- is_reg = TRUE;
- }
- else if (pentry->d_type == DT_DIR) {
- is_dir = TRUE;
- }
- }
- #else
- if (lstat (fpath, &st) == -1) {
- rspamd_fprintf (stderr, "cannot stat file %s: %s\n",
- fpath, strerror (errno));
- continue;
- }
-
- is_dir = S_ISDIR (st.st_mode);
- is_reg = S_ISREG (st.st_mode);
- #endif
- if (is_dir) {
- rspamc_process_dir (ev_base, cmd, fpath, attrs);
- continue;
- }
- else if (is_reg) {
- in = fopen (fpath, "r");
- if (in == NULL) {
- rspamd_fprintf (stderr, "cannot open file %s: %s\n",
- fpath, strerror (errno));
- continue;
- }
-
- rspamc_process_input (ev_base, cmd, in, fpath, attrs);
- cur_req++;
- fclose (in);
-
- if (cur_req >= max_requests) {
- cur_req = 0;
- /* Wait for completion */
- event_base_loop (ev_base, 0);
- }
- }
- }
- }
- else {
- fprintf (stderr, "cannot open directory %s: %s\n", name, strerror (errno));
- exit (EXIT_FAILURE);
- }
-
- closedir (d);
- event_base_loop (ev_base, 0);
- }
-
- gint
- main (gint argc, gchar **argv, gchar **env)
- {
- gint i, start_argc, cur_req = 0, res, ret, npatterns;
- GQueue *kwattrs;
- GList *cur;
- GPid cld;
- struct rspamc_command *cmd;
- FILE *in = NULL;
- struct event_base *ev_base;
- struct stat st;
- struct sigaction sigpipe_act;
- gchar **exclude_pattern;
-
- kwattrs = g_queue_new ();
-
- read_cmd_line (&argc, &argv);
-
- tty = isatty (STDOUT_FILENO);
-
- if (print_commands) {
- print_commands_list ();
- exit (EXIT_SUCCESS);
- }
-
- /* Deal with exclude patterns */
- exclude_pattern = exclude_patterns;
- npatterns = 0;
-
- while (exclude_pattern && *exclude_pattern) {
- npatterns ++;
- }
-
- if (npatterns > 0) {
- exclude_compiled = g_malloc0 (sizeof (*exclude_compiled) * (npatterns + 1));
-
- for (i = 0; i < npatterns; i ++) {
- exclude_compiled[i] = g_pattern_spec_new (exclude_patterns[i]);
-
- if (exclude_compiled[i] == NULL) {
- rspamd_fprintf (stderr, "Invalid glob pattern: %s\n",
- exclude_patterns[i]);
- exit (EXIT_FAILURE);
- }
- }
- }
-
- rspamd_init_libs ();
- ev_base = event_base_new ();
-
- /* Ignore sigpipe */
- sigemptyset (&sigpipe_act.sa_mask);
- sigaddset (&sigpipe_act.sa_mask, SIGPIPE);
- sigpipe_act.sa_handler = SIG_IGN;
- sigpipe_act.sa_flags = 0;
- sigaction (SIGPIPE, &sigpipe_act, NULL);
-
- /* Now read other args from argc and argv */
- if (argc == 1) {
- start_argc = argc;
- in = stdin;
- cmd = check_rspamc_command ("symbols");
- }
- else if (argc == 2) {
- /* One argument is whether command or filename */
- if ((cmd = check_rspamc_command (argv[1])) != NULL) {
- start_argc = argc;
- in = stdin;
- }
- else {
- cmd = check_rspamc_command ("symbols"); /* Symbols command */
- start_argc = 1;
- }
- }
- else {
- if ((cmd = check_rspamc_command (argv[1])) != NULL) {
- /* In case of command read arguments starting from 2 */
- if (cmd->cmd == RSPAMC_COMMAND_ADD_SYMBOL || cmd->cmd ==
- RSPAMC_COMMAND_ADD_ACTION) {
- if (argc < 4 || argc > 5) {
- fprintf (stderr, "invalid arguments\n");
- exit (EXIT_FAILURE);
- }
- if (argc == 5) {
- ADD_CLIENT_HEADER (kwattrs, "metric", argv[2]);
- ADD_CLIENT_HEADER (kwattrs, "name", argv[3]);
- ADD_CLIENT_HEADER (kwattrs, "value", argv[4]);
- }
- else {
- ADD_CLIENT_HEADER (kwattrs, "name", argv[2]);
- ADD_CLIENT_HEADER (kwattrs, "value", argv[3]);
- }
- start_argc = argc;
- }
- else {
- start_argc = 2;
- }
- }
- else {
- cmd = check_rspamc_command ("symbols");
- start_argc = 1;
- }
- }
-
- add_options (kwattrs);
-
- if (start_argc == argc) {
- /* Do command without input or with stdin */
- if (empty_input) {
- rspamc_process_input (ev_base, cmd, NULL, "empty", kwattrs);
- }
- else {
- rspamc_process_input (ev_base, cmd, in, "stdin", kwattrs);
- }
- }
- else {
- for (i = start_argc; i < argc; i++) {
- if (cmd->cmd == RSPAMC_COMMAND_FUZZY_DELHASH) {
- ADD_CLIENT_HEADER (kwattrs, "Hash", argv[i]);
- }
- else {
- if (stat (argv[i], &st) == -1) {
- fprintf (stderr, "cannot stat file %s\n", argv[i]);
- exit (EXIT_FAILURE);
- }
- if (S_ISDIR (st.st_mode)) {
- /* Directories are processed with a separate limit */
- rspamc_process_dir (ev_base, cmd, argv[i], kwattrs);
- cur_req = 0;
- }
- else {
- in = fopen (argv[i], "r");
- if (in == NULL) {
- fprintf (stderr, "cannot open file %s\n", argv[i]);
- exit (EXIT_FAILURE);
- }
- rspamc_process_input (ev_base, cmd, in, argv[i], kwattrs);
- cur_req++;
- fclose (in);
- }
- if (cur_req >= max_requests) {
- cur_req = 0;
- /* Wait for completion */
- event_base_loop (ev_base, 0);
- }
- }
- }
-
- if (cmd->cmd == RSPAMC_COMMAND_FUZZY_DELHASH) {
- rspamc_process_input (ev_base, cmd, NULL, "hashes", kwattrs);
- }
- }
-
- event_base_loop (ev_base, 0);
-
- g_queue_free_full (kwattrs, g_free);
-
- /* Wait for children processes */
- cur = g_list_first (children);
- ret = 0;
-
- while (cur) {
- cld = GPOINTER_TO_SIZE (cur->data);
-
- if (waitpid (cld, &res, 0) == -1) {
- fprintf (stderr, "Cannot wait for %d: %s", (gint)cld,
- strerror (errno));
-
- ret = errno;
- }
-
- if (ret == 0) {
- /* Check return code */
- if (WIFSIGNALED (res)) {
- ret = WTERMSIG (res);
- }
- else if (WIFEXITED (res)) {
- ret = WEXITSTATUS (res);
- }
- }
-
- cur = g_list_next (cur);
- }
-
- if (children != NULL) {
- g_list_free (children);
- }
-
- for (i = 0; i < npatterns; i ++) {
- g_pattern_spec_free (exclude_compiled[i]);
- }
-
- return ret;
- }
|