TEST_P(extractMultipartParamTest, extract_multipart)
{
size_t consumed = 0;
multipartParam const & p = GetParam();
parser = multipart_parser_create(p.boundary, strlen(p.boundary), &setting);
ASSERT_NE(parser, (multipart_parser*)NULL);
multipart_parser_set_data(parser, this);
for(int i=0; i < p.bufcnt; i++){
ASSERT_TRUE(bufused + strlen(p.bufs[i]) < buflen);
memcpy(buf+bufused, p.bufs[i], strlen(p.bufs[i]));
bufused += strlen(p.bufs[i]);
consumed = multipart_parser_execute(parser, buf + total_consumed, bufused - total_consumed);
total_consumed += consumed;
}
ASSERT_EQ(cksum, p.expected_cksum);
}
int MultipartFormCreate(HTTPConnectionRef const conn, uv_buf_t const *const boundary, MultipartFormRef *const out) {
assert(out);
if(!conn || !boundary->base || !boundary->len) return UV_EINVAL;
MultipartFormRef form = calloc(1, sizeof(struct MultipartForm));
if(!form) return UV_ENOMEM;
form->conn = conn;
form->parser = multipart_parser_init(boundary->base, boundary->len, &callbacks);
if(!form->parser) {
MultipartFormFree(&form);
return UV_ENOMEM;
}
multipart_parser_set_data(form->parser, form);
form->type = MultipartNothing;
*form->out = uv_buf_init(NULL, 0);
form->flags = 0;
*out = form;
return 0;
}
int multipart_parse(evhtp_request_t *req, const char *content_type, const char *address, const char *buff, int post_size)
{
int err_no = 0;
char *boundary = NULL, *boundary_end = NULL;
char *boundaryPattern = NULL;
int boundary_len = 0;
mp_arg_t *mp_arg = NULL;
evbuffer_add_printf(req->buffer_out,
"<html>\n<head>\n"
"<title>Upload Result</title>\n"
"</head>\n"
"<body>\n"
);
if(strstr(content_type, "boundary") == 0)
{
LOG_PRINT(LOG_DEBUG, "boundary NOT found!");
LOG_PRINT(LOG_ERROR, "%s fail post parse", address);
err_no = 6;
goto done;
}
boundary = strchr(content_type, '=');
boundary++;
boundary_len = strlen(boundary);
if(boundary[0] == '"')
{
boundary++;
boundary_end = strchr(boundary, '"');
if (!boundary_end)
{
LOG_PRINT(LOG_DEBUG, "Invalid boundary in multipart/form-data POST data");
LOG_PRINT(LOG_ERROR, "%s fail post parse", address);
err_no = 6;
goto done;
}
}
else
{
/* search for the end of the boundary */
boundary_end = strpbrk(boundary, ",;");
}
if (boundary_end)
{
boundary_end[0] = '\0';
boundary_len = boundary_end-boundary;
}
LOG_PRINT(LOG_DEBUG, "boundary Find. boundary = %s", boundary);
boundaryPattern = (char *)malloc(boundary_len + 3);
if(boundaryPattern == NULL)
{
LOG_PRINT(LOG_DEBUG, "boundarypattern malloc failed!");
LOG_PRINT(LOG_ERROR, "%s fail malloc", address);
err_no = 0;
goto done;
}
snprintf(boundaryPattern, boundary_len + 3, "--%s", boundary);
LOG_PRINT(LOG_DEBUG, "boundaryPattern = %s, strlen = %d", boundaryPattern, (int)strlen(boundaryPattern));
multipart_parser* parser = multipart_parser_init(boundaryPattern);
if(!parser)
{
LOG_PRINT(LOG_DEBUG, "Multipart_parser Init Failed!");
LOG_PRINT(LOG_ERROR, "%s fail post save", address);
err_no = 0;
goto done;
}
mp_arg = (mp_arg_t *)malloc(sizeof(mp_arg_t));
if(!mp_arg)
{
LOG_PRINT(LOG_DEBUG, "Multipart_parser Arg Malloc Failed!");
LOG_PRINT(LOG_ERROR, "%s fail post save", address);
err_no = 0;
goto done;
}
evthr_t *thread = get_request_thr(req);
thr_arg_t *thr_arg = (thr_arg_t *)evthr_get_aux(thread);
mp_arg->req = req;
mp_arg->thr_arg = thr_arg;
str_lcpy(mp_arg->address, address, 16);
mp_arg->partno = 0;
mp_arg->succno = 0;
mp_arg->check_name = 0;
multipart_parser_set_data(parser, mp_arg);
multipart_parser_execute(parser, buff, post_size);
multipart_parser_free(parser);
if(mp_arg->succno == 0)
{
evbuffer_add_printf(req->buffer_out, "<h1>Upload Failed!</h1>\n");
}
evbuffer_add_printf(req->buffer_out, "</body>\n</html>\n");
evhtp_headers_add_header(req->headers_out, evhtp_header_new("Content-Type", "text/html", 0, 0));
err_no = -1;
done:
free(boundaryPattern);
free(mp_arg);
return err_no;
}
