X-Git-Url: http://www.privoxy.org/gitweb/?p=privoxy.git;a=blobdiff_plain;f=loaders.c;h=65aba0acd724d355cae6c2ebcf8540e497122d18;hp=f30d9ddc2eaec3fea70552cd4fe0f70f41cf4b42;hb=e03b2db8ef7bd0138b41b1ab6f4361cf1710e217;hpb=9ff91f223c567dfb54c405915507e52f93741137 diff --git a/loaders.c b/loaders.c index f30d9ddc..65aba0ac 100644 --- a/loaders.c +++ b/loaders.c @@ -1,4 +1,3 @@ -const char loaders_rcs[] = "$Id: loaders.c,v 1.99 2014/06/02 06:22:21 fabiankeil Exp $"; /********************************************************************* * * File : $Source: /cvsroot/ijbswa/current/loaders.c,v $ @@ -63,8 +62,6 @@ const char loaders_rcs[] = "$Id: loaders.c,v 1.99 2014/06/02 06:22:21 fabiankeil #include "urlmatch.h" #include "encode.h" -const char loaders_h_rcs[] = LOADERS_H_VERSION; - /* * Currently active files. * These are also entered in the main linked list of files. @@ -74,13 +71,51 @@ const char loaders_h_rcs[] = LOADERS_H_VERSION; static struct file_list *current_trustfile = NULL; #endif /* def FEATURE_TRUST */ +#ifndef FUZZ static int load_one_re_filterfile(struct client_state *csp, int fileid); +#endif static struct file_list *current_re_filterfile[MAX_AF_FILES] = { NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL }; +/********************************************************************* + * + * Function : free_csp_resources + * + * Description : Frees memory referenced by the csp that isn't + * shared with other csps. + * + * Parameters : + * 1 : csp = Current client state (buffers, headers, etc...) + * + * Returns : N/A + * + *********************************************************************/ +void free_csp_resources(struct client_state *csp) +{ + freez(csp->ip_addr_str); +#ifdef FEATURE_CLIENT_TAGS + freez(csp->client_address); +#endif + freez(csp->listen_addr_str); + freez(csp->client_iob->buf); + freez(csp->iob->buf); + freez(csp->error_message); + + if (csp->action->flags & ACTION_FORWARD_OVERRIDE && + NULL != csp->fwd) + { + unload_forward_spec(csp->fwd); + } + free_http_request(csp->http); + + destroy_list(csp->headers); + destroy_list(csp->tags); + + free_current_action(csp->action); +} /********************************************************************* * @@ -181,23 +216,6 @@ unsigned int sweep(void) { last_active->next = client_list->next; - freez(csp->ip_addr_str); - freez(csp->client_iob->buf); - freez(csp->iob->buf); - freez(csp->error_message); - - if (csp->action->flags & ACTION_FORWARD_OVERRIDE && - NULL != csp->fwd) - { - unload_forward_spec(csp->fwd); - } - free_http_request(csp->http); - - destroy_list(csp->headers); - destroy_list(csp->tags); - - free_current_action(csp->action); - #ifdef FEATURE_STATISTICS urls_read++; if (csp->flags & CSP_FLAG_REJECTED) @@ -288,23 +306,10 @@ int check_file_changed(const struct file_list * current, return 0; } - fs = (struct file_list *)zalloc(sizeof(struct file_list)); - if (fs == NULL) - { - /* Out of memory error */ - return 1; - } - - - fs->filename = strdup(filename); + fs = zalloc_or_die(sizeof(struct file_list)); + fs->filename = strdup_or_die(filename); fs->lastmodified = statbuf->st_mtime; - if (fs->filename == NULL) - { - /* Out of memory error */ - freez (fs); - return 1; - } *newfl = fs; return 1; } @@ -364,6 +369,7 @@ jb_err simple_read_line(FILE *fp, char **dest, int *newline) for (;;) { ch = getc(fp); + if (ch == EOF) { if (len > 0) @@ -420,6 +426,7 @@ jb_err simple_read_line(FILE *fp, char **dest, int *newline) } else if (ch == 0) { + /* XXX: Why do we allow this anyway? */ *p = '\0'; *dest = buf; return JB_ERR_OK; @@ -537,30 +544,15 @@ jb_err edit_read_line(FILE *fp, if (raw_out) { - raw = strdup(""); - if (NULL == raw) - { - return JB_ERR_MEMORY; - } + raw = strdup_or_die(""); } if (prefix_out) { - prefix = strdup(""); - if (NULL == prefix) - { - freez(raw); - return JB_ERR_MEMORY; - } + prefix = strdup_or_die(""); } if (data_out) { - data = strdup(""); - if (NULL == data) - { - freez(raw); - freez(prefix); - return JB_ERR_MEMORY; - } + data = strdup_or_die(""); } /* Main loop. Loop while we need more data & it's not EOF. */ @@ -652,15 +644,12 @@ jb_err edit_read_line(FILE *fp, if (*linestart) { is_empty = 0; - if (data) + if (string_append(&data, linestart)) { - if (string_append(&data, linestart)) - { - freez(raw); - freez(prefix); - free(linebuf); - return JB_ERR_MEMORY; - } + freez(raw); + freez(prefix); + free(linebuf); + return JB_ERR_MEMORY; } } @@ -848,11 +837,7 @@ int load_trustfile(struct client_state *csp) goto load_trustfile_error; } - fs->f = bl = (struct block_spec *)zalloc(sizeof(*bl)); - if (bl == NULL) - { - goto load_trustfile_error; - } + fs->f = bl = zalloc_or_die(sizeof(*bl)); if ((fp = fopen(csp->config->trustfile, "r")) == NULL) { @@ -895,11 +880,7 @@ int load_trustfile(struct client_state *csp) } /* allocate a new node */ - if ((b = zalloc(sizeof(*b))) == NULL) - { - fclose(fp); - goto load_trustfile_error; - } + b = zalloc_or_die(sizeof(*b)); /* add it to the list */ b->next = bl->next; @@ -1116,7 +1097,6 @@ int load_one_re_filterfile(struct client_state *csp, int fileid) struct file_list *fs; char *buf = NULL; - int error; unsigned long linenum = 0; pcrs_job *dummy, *lastjob = NULL; @@ -1183,11 +1163,7 @@ int load_one_re_filterfile(struct client_state *csp, int fileid) */ if (new_filter != FT_INVALID_FILTER) { - new_bl = (struct re_filterfile_spec *)zalloc(sizeof(*bl)); - if (new_bl == NULL) - { - goto load_re_filterfile_error; - } + new_bl = zalloc_or_die(sizeof(*bl)); if (new_filter == FT_CONTENT_FILTER) { new_bl->name = chomp(buf + 7); @@ -1214,15 +1190,16 @@ int load_one_re_filterfile(struct client_state *csp, int fileid) new_bl->description = html_encode(chomp(new_bl->description)); if (NULL == new_bl->description) { - new_bl->description = strdup("Out of memory while encoding this filter's description to HTML"); + new_bl->description = strdup_or_die("Out of memory while " + "encoding filter description to HTML"); } } else { - new_bl->description = strdup("No description available for this filter"); + new_bl->description = strdup_or_die("No description available"); } - new_bl->name = strdup(chomp(new_bl->name)); + new_bl->name = strdup_or_die(chomp(new_bl->name)); /* * If this is the first filter block, chain it @@ -1249,6 +1226,7 @@ int load_one_re_filterfile(struct client_state *csp, int fileid) #ifdef FEATURE_EXTERNAL_FILTERS if ((bl != NULL) && (bl->type == FT_EXTERNAL_CONTENT_FILTER)) { + jb_err jb_error; /* Save the code as "pattern", but do not compile anything. */ if (bl->patterns->first != NULL) { @@ -1256,8 +1234,8 @@ int load_one_re_filterfile(struct client_state *csp, int fileid) "Did you forget to escape a line break?", bl->name); } - error = enlist(bl->patterns, buf); - if (JB_ERR_MEMORY == error) + jb_error = enlist(bl->patterns, buf); + if (JB_ERR_MEMORY == jb_error) { log_error(LOG_LEVEL_FATAL, "Out of memory while enlisting external filter code \'%s\' for filter %s.", @@ -1269,17 +1247,19 @@ int load_one_re_filterfile(struct client_state *csp, int fileid) #endif if (bl != NULL) { + int pcrs_error; + jb_err jb_error; /* * Save the expression, make it a pcrs_job * and chain it into the current filter's joblist */ - error = enlist(bl->patterns, buf); - if (JB_ERR_MEMORY == error) + jb_error = enlist(bl->patterns, buf); + if (JB_ERR_MEMORY == jb_error) { log_error(LOG_LEVEL_FATAL, "Out of memory while enlisting re_filter job \'%s\' for filter %s.", buf, bl->name); } - assert(JB_ERR_OK == error); + assert(JB_ERR_OK == jb_error); if (pcrs_job_is_dynamic(buf)) { @@ -1311,11 +1291,11 @@ int load_one_re_filterfile(struct client_state *csp, int fileid) continue; } - if ((dummy = pcrs_compile_command(buf, &error)) == NULL) + if ((dummy = pcrs_compile_command(buf, &pcrs_error)) == NULL) { log_error(LOG_LEVEL_ERROR, "Adding re_filter job \'%s\' to filter %s failed: %s", - buf, bl->name, pcrs_strerror(error)); + buf, bl->name, pcrs_strerror(pcrs_error)); freez(buf); continue; }