FS-4614 --resolve update to dynamic line buffer and use internal buffering to boot for file I/O on preprocessing XML

This commit is contained in:
Anthony Minessale 2012-09-18 15:39:23 -05:00
parent 1b2b456510
commit c5204c2128
4 changed files with 163 additions and 37 deletions

View File

@ -234,7 +234,8 @@ static inline uint32_t switch_known_bitrate(switch_payload_t payload)
} }
SWITCH_DECLARE(switch_size_t) switch_fd_read_line(int fd, char *buf, switch_size_t len); SWITCH_DECLARE(switch_size_t) switch_fd_read_line(int fd, char *buf, switch_size_t len);
SWITCH_DECLARE(switch_size_t) switch_fd_read_dline(int fd, char **buf, switch_size_t *len);
SWITCH_DECLARE(switch_size_t) switch_fp_read_dline(FILE *fd, char **buf, switch_size_t *len);
SWITCH_DECLARE(switch_status_t) switch_frame_alloc(switch_frame_t **frame, switch_size_t size); SWITCH_DECLARE(switch_status_t) switch_frame_alloc(switch_frame_t **frame, switch_size_t size);
SWITCH_DECLARE(switch_status_t) switch_frame_dup(switch_frame_t *orig, switch_frame_t **clone); SWITCH_DECLARE(switch_status_t) switch_frame_dup(switch_frame_t *orig, switch_frame_t **clone);

View File

@ -930,19 +930,21 @@ SWITCH_DECLARE(switch_status_t) switch_core_mime_add_type(const char *type, cons
static void load_mime_types(void) static void load_mime_types(void)
{ {
char *cf = "mime.types"; char *cf = "mime.types";
int fd = -1; FILE *fd = NULL;
char line_buf[1024] = ""; char *line_buf;
switch_size_t llen = 0;
char *mime_path = NULL; char *mime_path = NULL;
mime_path = switch_mprintf("%s/%s", SWITCH_GLOBAL_dirs.conf_dir, cf); mime_path = switch_mprintf("%s/%s", SWITCH_GLOBAL_dirs.conf_dir, cf);
switch_assert(mime_path); switch_assert(mime_path);
fd = open(mime_path, O_RDONLY | O_BINARY); fd = fopen(mime_path, "rb");
if (fd <= 0) {
if (fd == NULL) {
goto end; goto end;
} }
while ((switch_fd_read_line(fd, line_buf, sizeof(line_buf)))) { while ((switch_fp_read_dline(fd, &line_buf, &llen))) {
char *p; char *p;
char *type = line_buf; char *type = line_buf;
@ -966,9 +968,11 @@ static void load_mime_types(void)
} }
if (fd > -1) { switch_safe_free(line_buf);
close(fd);
fd = -1; if (fd) {
fclose(fd);
fd = NULL;
} }
end: end:

View File

@ -453,6 +453,103 @@ SWITCH_DECLARE(switch_size_t) switch_fd_read_line(int fd, char *buf, switch_size
return total; return total;
} }
#define DLINE_BLOCK_SIZE 1024
#define DLINE_MAX_SIZE 1048576
SWITCH_DECLARE(switch_size_t) switch_fd_read_dline(int fd, char **buf, switch_size_t *len)
{
char c, *p;
int cur;
switch_size_t total = 0;
char *data = *buf;
switch_size_t ilen = *len;
if (!data) {
*len = ilen = DLINE_BLOCK_SIZE;
data = malloc(ilen);
memset(data, 0, ilen);
}
p = data;
while ((cur = read(fd, &c, 1)) == 1) {
if (total + 2 >= ilen) {
if (ilen + DLINE_BLOCK_SIZE > DLINE_MAX_SIZE) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_CRIT, "Single line limit reached!\n");
break;
}
ilen += DLINE_BLOCK_SIZE;
data = realloc(data, ilen);
switch_assert(data);
p = data + total;
}
total += cur;
*p++ = c;
if (c == '\r' || c == '\n') {
break;
}
}
*p++ = '\0';
*len = ilen;
*buf = data;
return total;
}
SWITCH_DECLARE(switch_size_t) switch_fp_read_dline(FILE *fd, char **buf, switch_size_t *len)
{
char c, *p;
switch_size_t total = 0;
char *data = *buf;
switch_size_t ilen = *len;
if (!data) {
*len = ilen = DLINE_BLOCK_SIZE;
data = malloc(ilen);
memset(data, 0, ilen);
}
p = data;
//while ((c = fgetc(fd)) != EOF) {
while (fread(&c, 1, 1, fd) == 1) {
if (total + 2 >= ilen) {
if (ilen + DLINE_BLOCK_SIZE > DLINE_MAX_SIZE) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_CRIT, "Single line limit reached!\n");
break;
}
ilen += DLINE_BLOCK_SIZE;
data = realloc(data, ilen);
switch_assert(data);
p = data + total;
}
total++;
*p++ = c;
if (c == '\r' || c == '\n') {
break;
}
}
*p++ = '\0';
*len = ilen;
*buf = data;
return total;
}
SWITCH_DECLARE(char *) switch_amp_encode(char *s, char *buf, switch_size_t len) SWITCH_DECLARE(char *) switch_amp_encode(char *s, char *buf, switch_size_t len)
{ {
char *p, *q; char *p, *q;

View File

@ -104,7 +104,7 @@ void globfree(glob_t *);
/* Use UTF-8 as the general encoding */ /* Use UTF-8 as the general encoding */
static switch_bool_t USE_UTF_8_ENCODING = SWITCH_TRUE; static switch_bool_t USE_UTF_8_ENCODING = SWITCH_TRUE;
static int preprocess(const char *cwd, const char *file, int write_fd, int rlevel); static int preprocess(const char *cwd, const char *file, FILE *write_fd, int rlevel);
typedef struct switch_xml_root *switch_xml_root_t; typedef struct switch_xml_root *switch_xml_root_t;
struct switch_xml_root { /* additional data for the root tag */ struct switch_xml_root { /* additional data for the root tag */
@ -1224,12 +1224,12 @@ static char *expand_vars(char *buf, char *ebuf, switch_size_t elen, switch_size_
return ebuf; return ebuf;
} }
static int preprocess_exec(const char *cwd, const char *command, int write_fd, int rlevel) static FILE *preprocess_exec(const char *cwd, const char *command, FILE *write_fd, int rlevel)
{ {
#ifdef WIN32 #ifdef WIN32
char message[] = "<!-- exec not implemented in windows yet -->"; char message[] = "<!-- exec not implemented in windows yet -->";
if (write(write_fd, message, sizeof(message)) < 0) { if (fwrite( message, 1, sizeof(message), write_fd) < 0) {
goto end; goto end;
} }
#else #else
@ -1249,7 +1249,7 @@ static int preprocess_exec(const char *cwd, const char *command, int write_fd, i
int bytes; int bytes;
close(fds[1]); close(fds[1]);
while ((bytes = read(fds[0], buf, sizeof(buf))) > 0) { while ((bytes = read(fds[0], buf, sizeof(buf))) > 0) {
if (write(write_fd, buf, bytes) <= 0) { if (fwrite(buf, 1, bytes, write_fd) <= 0) {
break; break;
} }
} }
@ -1271,7 +1271,7 @@ static int preprocess_exec(const char *cwd, const char *command, int write_fd, i
} }
static int preprocess_glob(const char *cwd, const char *pattern, int write_fd, int rlevel) static FILE *preprocess_glob(const char *cwd, const char *pattern, FILE *write_fd, int rlevel)
{ {
char *full_path = NULL; char *full_path = NULL;
char *dir_path = NULL, *e = NULL; char *dir_path = NULL, *e = NULL;
@ -1310,28 +1310,43 @@ static int preprocess_glob(const char *cwd, const char *pattern, int write_fd, i
return write_fd; return write_fd;
} }
static int preprocess(const char *cwd, const char *file, int write_fd, int rlevel) static int preprocess(const char *cwd, const char *file, FILE *write_fd, int rlevel)
{ {
int read_fd = -1; FILE *read_fd = NULL;
switch_size_t cur = 0, ml = 0; switch_size_t cur = 0, ml = 0;
char *q, *cmd, buf[2048], ebuf[8192]; char *q, *cmd, *buf = NULL, *ebuf = NULL;
char *tcmd, *targ; char *tcmd, *targ;
int line = 0; int line = 0;
switch_size_t len = 0, eblen = 0;
if ((read_fd = open(file, O_RDONLY, 0)) < 0) {
const char *reason = strerror(errno);
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldnt open %s (%s)\n", file, reason);
return read_fd;
}
if (rlevel > 100) { if (rlevel > 100) {
return -1; return -1;
} }
while ((cur = switch_fd_read_line(read_fd, buf, sizeof(buf))) > 0) { if (!(read_fd = fopen(file, "r"))) {
const char *reason = strerror(errno);
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldnt open %s (%s)\n", file, reason);
return -1;
}
setvbuf(read_fd, (char *) NULL, _IOFBF, 65536);
for(;;) {
char *arg, *e; char *arg, *e;
const char *err = NULL; const char *err = NULL;
char *bp = expand_vars(buf, ebuf, sizeof(ebuf), &cur, &err); char *bp;
switch_safe_free(ebuf);
if ((cur = switch_fp_read_dline(read_fd, &buf, &len)) <= 0) {
break;
}
eblen = len *2;
ebuf = malloc(eblen);
memset(ebuf, 0, eblen);
bp = expand_vars(buf, ebuf, eblen, &cur, &err);
line++; line++;
if (err) { if (err) {
@ -1360,7 +1375,7 @@ static int preprocess(const char *cwd, const char *file, int write_fd, int rleve
if ((e = strstr(tcmd, "/>"))) { if ((e = strstr(tcmd, "/>"))) {
*e += 2; *e += 2;
*e = '\0'; *e = '\0';
if (write(write_fd, e, (unsigned) strlen(e)) != (int) strlen(e)) { if (fwrite(e, 1, (unsigned) strlen(e), write_fd) != (int) strlen(e)) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n");
} }
} }
@ -1431,13 +1446,13 @@ static int preprocess(const char *cwd, const char *file, int write_fd, int rleve
} }
if ((cmd = strstr(bp, "<!--#"))) { if ((cmd = strstr(bp, "<!--#"))) {
if (write(write_fd, bp, (unsigned) (cmd - bp)) != (cmd - bp)) { if (fwrite(bp, 1, (unsigned) (cmd - bp), write_fd) != (cmd - bp)) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n");
} }
if ((e = strstr(cmd, "-->"))) { if ((e = strstr(cmd, "-->"))) {
*e = '\0'; *e = '\0';
e += 3; e += 3;
if (write(write_fd, e, (unsigned) strlen(e)) != (int) strlen(e)) { if (fwrite(e, 1, (unsigned) strlen(e), write_fd) != (int) strlen(e)) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n");
} }
} else { } else {
@ -1489,13 +1504,18 @@ static int preprocess(const char *cwd, const char *file, int write_fd, int rleve
continue; continue;
} }
if (write(write_fd, bp, (unsigned) cur) != (int) cur) { if (fwrite(bp, 1, (unsigned) cur, write_fd) != (int) cur) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Short write!\n");
} }
} }
close(read_fd); switch_safe_free(buf);
return write_fd; switch_safe_free(ebuf);
fclose(read_fd);
return 0;
} }
SWITCH_DECLARE(switch_xml_t) switch_xml_parse_file_simple(const char *file) SWITCH_DECLARE(switch_xml_t) switch_xml_parse_file_simple(const char *file)
@ -1527,7 +1547,8 @@ SWITCH_DECLARE(switch_xml_t) switch_xml_parse_file_simple(const char *file)
SWITCH_DECLARE(switch_xml_t) switch_xml_parse_file(const char *file) SWITCH_DECLARE(switch_xml_t) switch_xml_parse_file(const char *file)
{ {
int fd = -1, write_fd = -1; int fd = -1;
FILE *write_fd = NULL;
switch_xml_t xml = NULL; switch_xml_t xml = NULL;
char *new_file = NULL; char *new_file = NULL;
const char *abs, *absw; const char *abs, *absw;
@ -1546,13 +1567,15 @@ SWITCH_DECLARE(switch_xml_t) switch_xml_parse_file(const char *file)
goto done; goto done;
} }
if ((write_fd = open(new_file, O_WRONLY | O_CREAT | O_TRUNC, S_IRUSR | S_IWUSR)) < 0) { if ((write_fd = fopen(new_file, "w+")) == NULL) {
goto done; goto done;
} }
setvbuf(write_fd, (char *) NULL, _IOFBF, 65536);
if (preprocess(SWITCH_GLOBAL_dirs.conf_dir, file, write_fd, 0) > -1) { if (preprocess(SWITCH_GLOBAL_dirs.conf_dir, file, write_fd, 0) > -1) {
close(write_fd); fclose(write_fd);
write_fd = -1; write_fd = NULL;
if ((fd = open(new_file, O_RDONLY, 0)) > -1) { if ((fd = open(new_file, O_RDONLY, 0)) > -1) {
if ((xml = switch_xml_parse_fd(fd))) { if ((xml = switch_xml_parse_fd(fd))) {
xml->free_path = new_file; xml->free_path = new_file;
@ -1567,8 +1590,9 @@ SWITCH_DECLARE(switch_xml_t) switch_xml_parse_file(const char *file)
switch_mutex_unlock(FILE_LOCK); switch_mutex_unlock(FILE_LOCK);
if (write_fd > -1) { if (write_fd) {
close(write_fd); fclose(write_fd);
write_fd = NULL;
} }
if (fd > -1) { if (fd > -1) {