smtp: fix test

pull/2028/head
Victor Julien 10 years ago
parent c4a9580fce
commit 7f700a137c

@ -5040,35 +5040,28 @@ int SMTPProcessDataChunkTest05(void){
0x41, 0x3D, 0x3D, 0x0D, 0x0A,}; 0x41, 0x3D, 0x3D, 0x0D, 0x0A,};
Flow f; Flow f;
int ret;
FLOW_INITIALIZE(&f); FLOW_INITIALIZE(&f);
f.alstate = SMTPStateAlloc(); f.alstate = SMTPStateAlloc();
FAIL_IF(f.alstate == NULL);
MimeDecParseState *state = MimeDecInitParser(&f, NULL); MimeDecParseState *state = MimeDecInitParser(&f, NULL);
((MimeDecEntity *)state->stack->top->data)->ctnt_flags = CTNT_IS_ATTACHMENT; ((MimeDecEntity *)state->stack->top->data)->ctnt_flags = CTNT_IS_ATTACHMENT;
FAIL_IF(state == NULL);
state->body_begin = 1; state->body_begin = 1;
int ret;
uint64_t file_size = 0;
ret = SMTPProcessDataChunk((uint8_t *)mimemsg, sizeof(mimemsg), state); ret = SMTPProcessDataChunk((uint8_t *)mimemsg, sizeof(mimemsg), state);
FAIL_IF(ret != 0);
state->body_begin = 0; state->body_begin = 0;
if(ret){goto end;}
SMTPState *smtp_state = (SMTPState *)((Flow *)state->data)->alstate; SMTPState *smtp_state = (SMTPState *)((Flow *)state->data)->alstate;
FileContainer *files = smtp_state->files_ts; FileContainer *files = smtp_state->files_ts;
FAIL_IF(files == NULL);
File *file = files->head; File *file = files->head;
file_size = file->size; FAIL_IF(file == NULL);
FileDisableStoring(&f, STREAM_TOSERVER);
FileDisableMagic(&f, STREAM_TOSERVER);
FileDisableMd5(&f, STREAM_TOSERVER);
ret = SMTPProcessDataChunk((uint8_t *)mimemsg, sizeof(mimemsg), state); ret = SMTPProcessDataChunk((uint8_t *)mimemsg, sizeof(mimemsg), state);
if(ret){goto end;} FAIL_IF(ret != 0);
printf("%u\t%u\n", (uint32_t) file->size, (uint32_t) file_size); FAIL_IF(file->size != 106);
if(file->size == file_size){ SMTPStateFree(smtp_state);
return 1; FLOW_DESTROY(&f);
}else{ PASS;
return 0;
}
end:
return ret == 0;
} }
#endif /* UNITTESTS */ #endif /* UNITTESTS */

Loading…
Cancel
Save