Fixed html insert

Signed-off-by: tyrolyean <tyrolyean@tyrolyean.net>
This commit is contained in:
tyrolyean 2020-05-01 11:24:42 +02:00
parent 38c07ded99
commit 48d3990043
No known key found for this signature in database
GPG key ID: EDD105663B707C62

View file

@ -15,11 +15,13 @@
* under the License. * under the License.
*/ */
#define _GNU_SOURCE
#include <string.h>
#include "mail.h" #include "mail.h"
#include "tools.h" #include "tools.h"
#include "attach.h" #include "attach.h"
#include <string.h>
#include <stddef.h> #include <stddef.h>
#include <stdint.h> #include <stdint.h>
#include <stdlib.h> #include <stdlib.h>
@ -128,6 +130,63 @@ int append_to_body(struct email_t* mail, const char* text){
} }
/* Append the string text DIRECTLY before the last closing </html> tag inside
* the body.
*/
int append_to_html_body(struct email_t* mail, const char* text){
if(text == NULL || mail == NULL){
return -1;
}
char *buffer = malloc(strlen(text) + 3);
memset(buffer, 0, strlen(text) + 3);
buffer[0] = '\r';
buffer[1] = '\n';
strcat(buffer, text);
struct email_t* root = get_root_mail(mail);
char* body_end = mail->message_length +
mail->message;
char* html_end = mail->message;
for(;;){
char * html_end_int = strcasestr(html_end+1, "</html>");
if(html_end_int == NULL || html_end_int >= body_end){
break;
}else{
html_end = html_end_int;
}
}
size_t root_offset = 0;
if(html_end == mail->message){
root_offset = mail->message_length +
(mail->message - root->message);
}else{
root_offset = html_end - root->message;
}
char* old_root = root->message;
char * new_root = insert_string(root->message, buffer,
root->message_length, root_offset);
if(new_root == NULL){
return -1;
}
propagate_size_change(mail, strlen(buffer));
propagate_root_pointer(root, new_root, old_root);
propagate_insert_delete(root, root->message+root_offset,
strlen(buffer));
free(buffer);
redetect_body_head(mail);
return 0;
}
int remove_mail(struct email_t* mail){ int remove_mail(struct email_t* mail){
if(mail == NULL || mail->parent == NULL || !mail->parent->is_multipart){ if(mail == NULL || mail->parent == NULL || !mail->parent->is_multipart){