END_TEST
START_TEST(test_tcpv4_inet)
{
int server_root_socket, server_client_socket, client_socket;
char buffer[BUFFERSIZE];
char *msg;
char ipbuffer[BUFFERSIZE];
ck_assert_int_ge((server_root_socket = OS_Bindporttcp(PORT, NULL, 0)), 0);
ck_assert_int_ge((client_socket = OS_ConnectTCP(PORT, IPV4, 0, NULL)) , 0);
ck_assert_int_ge((server_client_socket = OS_AcceptTCP(server_root_socket, ipbuffer, BUFFERSIZE)), 0);
ck_assert_str_eq(ipbuffer, IPV4);
ck_assert_int_eq(OS_SendTCP(client_socket, SENDSTRING), 0);
ck_assert_int_eq(OS_RecvTCPBuffer(server_client_socket, buffer, BUFFERSIZE), 0);
ck_assert_str_eq(buffer, SENDSTRING);
ck_assert_int_eq(OS_SendTCPbySize(server_client_socket, 5, SENDSTRING), 0);
ck_assert_ptr_ne((msg = OS_RecvTCP(client_socket, BUFFERSIZE)), NULL);
ck_assert_str_eq(msg, "Hello"); /* only 5 bytes send */
free(msg);
OS_CloseSocket(client_socket);
OS_CloseSocket(server_client_socket);
OS_CloseSocket(server_root_socket);
}
END_TEST
START_TEST(test_tcpinvalidsockets)
{
char buffer[BUFFERSIZE];
ck_assert_int_eq(OS_SendTCP(-1, SENDSTRING), OS_SOCKTERR);
ck_assert_int_eq(OS_SendTCPbySize(-1, strlen(SENDSTRING), SENDSTRING), OS_SOCKTERR);
ck_assert_ptr_eq(OS_RecvTCP(-1, BUFFERSIZE), NULL);
ck_assert_int_eq(OS_RecvTCPBuffer(-1, buffer, BUFFERSIZE), -1);
ck_assert_int_eq(OS_AcceptTCP(-1, buffer, BUFFERSIZE), -1);
}
/* OS_Sendsms.
*/
int OS_Sendsms(MailConfig *mail, struct tm *p, MailMsg *sms_msg)
{
int socket, i = 0, final_to_sz;
char *msg;
char snd_msg[128];
char final_to[512];
/* Connecting to the smtp server */
socket = OS_ConnectTCP(SMTP_DEFAULT_PORT, mail->smtpserver);
if(socket < 0)
{
return(socket);
}
/* Receiving the banner */
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDBANNER, msg)))
{
merror(BANNER_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Received banner: '%s' %s", msg, "");
free(msg);
/* Sending HELO message */
OS_SendTCP(socket,HELOMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
if(msg)
{
/* Ugly fix warning :) */
/* In some cases (with virus scans in the middle)
* we may get two banners. Check for that in here.
*/
if(OS_Match(VALIDBANNER, msg))
{
free(msg);
/* Try again */
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror("%s:%s",HELO_ERROR,msg!= NULL?msg:"null");
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
}
else
{
merror("%s:%s",HELO_ERROR,msg);
free(msg);
close(socket);
return(OS_INVALID);
}
}
else
{
merror("%s:%s",HELO_ERROR,"null");
close(socket);
return(OS_INVALID);
}
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", HELOMSG, msg);
free(msg);
/* Building "Mail from" msg */
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, MAILFROM, mail->from);
OS_SendTCP(socket, snd_msg);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror(FROM_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", snd_msg, msg);
free(msg);
/* Additional RCPT to */
final_to[0] = '\0';
final_to_sz = sizeof(final_to) -2;
if(mail->gran_to)
{
i = 0;
while(mail->gran_to[i] != NULL)
{
if(mail->gran_set[i] != SMS_FORMAT)
{
i++;
continue;
}
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, RCPTTO, mail->gran_to[i]);
OS_SendTCP(socket, snd_msg);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror(TO_ERROR, mail->gran_to[i]);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", snd_msg, msg);
free(msg);
/* Creating header for to */
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, TO, mail->gran_to[i]);
strncat(final_to, snd_msg, final_to_sz);
final_to_sz -= strlen(snd_msg) +2;
i++;
continue;
}
}
/* Sending the "DATA" msg */
OS_SendTCP(socket,DATAMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDDATA, msg)))
{
merror(DATA_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", DATAMSG, msg);
free(msg);
/* Building "From" and "To" in the e-mail header */
OS_SendTCP(socket, final_to);
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, FROM, mail->from);
OS_SendTCP(socket, snd_msg);
/* Sending date */
memset(snd_msg,'\0',128);
/* Solaris doesn't have the "%z", so we set the timezone to 0. */
#ifdef SOLARIS
strftime(snd_msg, 127, "Date: %a, %d %b %Y %T -0000\r\n",p);
#else
strftime(snd_msg, 127, "Date: %a, %d %b %Y %T %z\r\n",p);
#endif
OS_SendTCP(socket,snd_msg);
/* Sending subject */
memset(snd_msg,'\0',128);
snprintf(snd_msg, 127, SUBJECT, sms_msg->subject);
OS_SendTCP(socket,snd_msg);
/* Sending body */
OS_SendTCP(socket, sms_msg->body);
/* Sending end of data \r\n.\r\n */
OS_SendTCP(socket,ENDDATA);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if(mail->strict_checking && ((msg == NULL)||(!OS_Match(VALIDMAIL, msg))))
{
merror(END_DATA_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
/* Checking msg in here, since it may be null */
if(msg)
free(msg);
/* quitting and closing socket */
OS_SendTCP(socket,QUITMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if(msg)
free(msg);
memset(snd_msg,'\0',128);
/* Returning 0 (success) */
close(socket);
return(0);
}
/* OS_Sendmail v0.1: 2005/03/18
*/
int OS_Sendmail(MailConfig *mail, struct tm *p)
{
int socket,i=0;
char *msg;
char snd_msg[128];
char additional_to[512];
MailNode *mailmsg;
additional_to[0] = '\0';
/* If there is no sms message, we attempt to get from the
* email list.
*/
mailmsg = OS_PopLastMail();
if(mailmsg == NULL)
{
merror("%s: No email to be sent. Inconsistent state.",ARGV0);
}
/* Connecting to the smtp server */
socket = OS_ConnectTCP(SMTP_DEFAULT_PORT, mail->smtpserver);
if(socket < 0)
{
return(socket);
}
/* Receiving the banner */
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDBANNER, msg)))
{
merror(BANNER_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Received banner: '%s' %s", msg, "");
free(msg);
/* Sending HELO message */
OS_SendTCP(socket,HELOMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
if(msg)
{
/* Ugly fix warning :) */
/* In some cases (with virus scans in the middle)
* we may get two banners. Check for that in here.
*/
if(OS_Match(VALIDBANNER, msg))
{
free(msg);
/* Try again */
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror("%s:%s",HELO_ERROR,msg!= NULL?msg:"null");
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
}
else
{
merror("%s:%s",HELO_ERROR,msg);
free(msg);
close(socket);
return(OS_INVALID);
}
}
else
{
merror("%s:%s",HELO_ERROR,"null");
close(socket);
return(OS_INVALID);
}
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", HELOMSG, msg);
free(msg);
/* Building "Mail from" msg */
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, MAILFROM, mail->from);
OS_SendTCP(socket, snd_msg);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror(FROM_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", snd_msg, msg);
free(msg);
/* Building "RCPT TO" msg */
while(1)
{
if(mail->to[i] == NULL)
{
if(i == 0)
{
merror(INTERNAL_ERROR);
close(socket);
return(OS_INVALID);
}
break;
}
memset(snd_msg,'\0',128);
snprintf(snd_msg,127,RCPTTO, mail->to[i++]);
OS_SendTCP(socket,snd_msg);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror(TO_ERROR, mail->to[i -1]);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", snd_msg, msg);
free(msg);
}
/* Additional RCPT to */
if(mail->gran_to)
{
i = 0;
while(mail->gran_to[i] != NULL)
{
if(mail->gran_set[i] != FULL_FORMAT)
{
i++;
continue;
}
memset(snd_msg,'\0',128);
snprintf(snd_msg,127,RCPTTO, mail->gran_to[i]);
OS_SendTCP(socket,snd_msg);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror(TO_ERROR, mail->gran_to[i]);
if(msg)
free(msg);
i++;
continue;
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", snd_msg, msg);
free(msg);
i++;
continue;
}
}
/* Sending the "DATA" msg */
OS_SendTCP(socket,DATAMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDDATA, msg)))
{
merror(DATA_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", DATAMSG, msg);
free(msg);
/* Building "From" and "To" in the e-mail header */
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, TO, mail->to[0]);
OS_SendTCP(socket, snd_msg);
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, FROM, mail->from);
OS_SendTCP(socket, snd_msg);
/* Adding CCs */
if(mail->to[1])
{
i = 1;
while(1)
{
if(mail->to[i] == NULL)
{
break;
}
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, TO, mail->to[i]);
OS_SendTCP(socket,snd_msg);
i++;
}
}
/* More CCs - from granular options */
if(mail->gran_to)
{
i = 0;
while(mail->gran_to[i] != NULL)
{
if(mail->gran_set[i] != FULL_FORMAT)
{
i++;
continue;
}
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, TO, mail->gran_to[i]);
OS_SendTCP(socket, snd_msg);
i++;
continue;
}
}
/* Sending date */
memset(snd_msg,'\0',128);
/* Solaris doesn't have the "%z", so we set the timezone to 0. */
#ifdef SOLARIS
strftime(snd_msg, 127, "Date: %a, %d %b %Y %T -0000\r\n",p);
#else
strftime(snd_msg, 127, "Date: %a, %d %b %Y %T %z\r\n",p);
#endif
OS_SendTCP(socket,snd_msg);
/* Sending subject */
memset(snd_msg,'\0',128);
/* Checking if global subject is available */
if((_g_subject_level != 0) && (_g_subject[0] != '\0'))
{
snprintf(snd_msg, 127, SUBJECT, _g_subject);
/* Clearing global values */
_g_subject[0] = '\0';
_g_subject_level = 0;
}
else
{
snprintf(snd_msg, 127, SUBJECT, mailmsg->mail->subject);
}
OS_SendTCP(socket,snd_msg);
/* Sending body */
/* Sending multiple emails together if we have to */
do
{
OS_SendTCP(socket, mailmsg->mail->body);
mailmsg = OS_PopLastMail();
}while(mailmsg);
/* Sending end of data \r\n.\r\n */
OS_SendTCP(socket,ENDDATA);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if(mail->strict_checking && ((msg == NULL)||(!OS_Match(VALIDMAIL, msg))))
{
merror(END_DATA_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
/* Checking msg in here, since it may be null */
if(msg)
free(msg);
/* quitting and closing socket */
OS_SendTCP(socket,QUITMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if(msg)
free(msg);
memset(snd_msg,'\0',128);
/* Returning 0 (success) */
close(socket);
return(0);
}
/* OS_SendCustomEmail
*/
int OS_SendCustomEmail(char **to, char *subject, char *smtpserver, char *from, FILE *fp, struct tm *p)
{
int socket,i = 0;
char *msg;
char snd_msg[128];
char buffer[2049];
buffer[2048] = '\0';
/* Connecting to the smtp server */
socket = OS_ConnectTCP(SMTP_DEFAULT_PORT, smtpserver, 0);
if(socket < 0)
{
return(socket);
}
/* Receiving the banner */
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDBANNER, msg)))
{
merror(BANNER_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Received banner: '%s' %s", msg, "");
free(msg);
/* Sending HELO message */
OS_SendTCP(socket,HELOMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
if(msg)
{
/* Ugly fix warning :) */
/* In some cases (with virus scans in the middle)
* we may get two banners. Check for that in here.
*/
if(OS_Match(VALIDBANNER, msg))
{
free(msg);
/* Try again */
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror("%s:%s",HELO_ERROR,msg!= NULL?msg:"null");
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
}
else
{
merror("%s:%s",HELO_ERROR,msg);
free(msg);
close(socket);
return(OS_INVALID);
}
}
else
{
merror("%s:%s",HELO_ERROR,"null");
close(socket);
return(OS_INVALID);
}
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", HELOMSG, msg);
free(msg);
/* Building "Mail from" msg */
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, MAILFROM, from);
OS_SendTCP(socket, snd_msg);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror(FROM_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", snd_msg, msg);
free(msg);
/* Building "RCPT TO" msg */
while(to[i])
{
memset(snd_msg,'\0',128);
snprintf(snd_msg,127,RCPTTO, to[i]);
OS_SendTCP(socket,snd_msg);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDMAIL, msg)))
{
merror(TO_ERROR, to[i]);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", snd_msg, msg);
free(msg);
i++;
}
/* Sending the "DATA" msg */
OS_SendTCP(socket,DATAMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if((msg == NULL)||(!OS_Match(VALIDDATA, msg)))
{
merror(DATA_ERROR);
if(msg)
free(msg);
close(socket);
return(OS_INVALID);
}
MAIL_DEBUG("DEBUG: Sent '%s', received: '%s'", DATAMSG, msg);
free(msg);
/* Building "From" and "To" in the e-mail header */
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, TO, to[0]);
OS_SendTCP(socket, snd_msg);
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, FROM, from);
OS_SendTCP(socket, snd_msg);
/* Adding CCs */
if(to[1])
{
i = 1;
while(1)
{
if(to[i] == NULL)
{
break;
}
memset(snd_msg,'\0',128);
snprintf(snd_msg,127, TO, to[i]);
OS_SendTCP(socket,snd_msg);
i++;
}
}
/* Sending date */
memset(snd_msg,'\0',128);
/* Solaris doesn't have the "%z", so we set the timezone to 0. */
#ifdef SOLARIS
strftime(snd_msg, 127, "Date: %a, %d %b %Y %T -0000\r\n",p);
#else
strftime(snd_msg, 127, "Date: %a, %d %b %Y %T %z\r\n",p);
#endif
OS_SendTCP(socket,snd_msg);
/* Sending subject */
memset(snd_msg, '\0', 128);
snprintf(snd_msg, 127, SUBJECT, subject);
OS_SendTCP(socket, snd_msg);
OS_SendTCP(socket,ENDHEADER);
/* Sending body */
fseek(fp, 0, SEEK_SET);
while(fgets(buffer, 2048, fp) != NULL)
{
OS_SendTCP(socket,buffer);
}
/* Sending end of data \r\n.\r\n */
OS_SendTCP(socket,ENDDATA);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
/* Checking msg in here, since it may be null */
if(msg)
free(msg);
/* quitting and closing socket */
OS_SendTCP(socket,QUITMSG);
msg = OS_RecvTCP(socket, OS_SIZE_1024);
if(msg)
free(msg);
memset(snd_msg,'\0',128);
/* Returning 0 (success) */
close(socket);
return(0);
}
