#include <freeradius-devel/ident.h>
RCSID("$Id$")
#include "std-includes.h"
#include "rfcnb-priv.h"
#include "rfcnb-util.h"
#include "rfcnb-io.h"
#ifdef HAVE_SYS_UIO_H
#include <sys/uio.h>
#endif
#ifdef HAVE_SYS_SIGNAL_H
#include <sys/signal.h>
#endif
#include <string.h>
int RFCNB_Timeout = 0;
void rfcnb_alarm(int sig)
{
fprintf(stderr, "IO Timed out ...\n");
}
int RFCNB_Set_Timeout(int seconds)
{
#ifndef SA_RESTART
struct sigvec invec, outvec;
#else
struct sigaction inact, outact;
#endif
RFCNB_Timeout = seconds;
if (RFCNB_Timeout > 0) {
#ifndef SA_RESTART
invec.sv_handler = (void (*)())rfcnb_alarm;
invec.sv_mask = 0;
invec.sv_flags = SV_INTERRUPT;
if (sigvec(SIGALRM, &invec, &outvec) < 0)
return(-1);
#else
inact.sa_handler = (void (*)())rfcnb_alarm;
sigemptyset(&inact.sa_mask);
inact.sa_flags = 0;
if (sigaction(SIGALRM, &inact, &outact) < 0)
return(-1);
#endif
}
return(0);
}
int RFCNB_Discard_Rest(struct RFCNB_Con *con, int len)
{ char temp[100];
int rest, this_read, bytes_read;
#ifdef RFCNB_DEBUG
fprintf(stderr, "Discard_Rest called to discard: %i\n", len);
#endif
rest = len;
while (rest > 0) {
this_read = (rest > sizeof(temp)?sizeof(temp):rest);
bytes_read = read(con -> fd, temp, this_read);
if (bytes_read <= 0) {
if (bytes_read < 0)
RFCNB_errno = RFCNBE_BadRead;
else
RFCNB_errno = RFCNBE_ConGone;
RFCNB_saved_errno = errno;
return(RFCNBE_Bad);
}
rest = rest - bytes_read;
}
return(0);
}
int RFCNB_Put_Pkt(struct RFCNB_Con *con, struct RFCNB_Pkt *pkt, int len)
{ int len_sent, tot_sent, this_len;
struct RFCNB_Pkt *pkt_ptr;
char *this_data;
int i;
struct iovec io_list[10];
pkt_ptr = pkt;
len_sent = tot_sent = 0;
i = 0;
while ((pkt_ptr != NULL) & (i < 10)) {
this_len = pkt_ptr -> len;
this_data = pkt_ptr -> data;
if ((tot_sent + this_len) > len)
this_len = len - tot_sent;
io_list[i].iov_len = this_len;
io_list[i].iov_base = this_data;
i++;
tot_sent += this_len;
if (tot_sent == len) break;
pkt_ptr = pkt_ptr -> next;
}
#ifdef RFCNB_DEBUG
fprintf(stderr, "Frags = %i, tot_sent = %i\n", i, tot_sent);
#endif
if (RFCNB_Timeout > 0)
alarm(RFCNB_Timeout);
if ((len_sent = writev(con -> fd, io_list, i)) < 0) {
con -> rfc_errno = errno;
if (errno == EINTR)
RFCNB_errno = RFCNBE_Timeout;
else
RFCNB_errno = RFCNBE_BadWrite;
RFCNB_saved_errno = errno;
return(RFCNBE_Bad);
}
if (len_sent < tot_sent) {
if (errno == EINTR)
RFCNB_errno = RFCNBE_Timeout;
else
RFCNB_errno = RFCNBE_BadWrite;
RFCNB_saved_errno = errno;
return(RFCNBE_Bad);
}
if (RFCNB_Timeout > 0)
alarm(0);
#ifdef RFCNB_DEBUG
fprintf(stderr, "Len sent = %i ...\n", len_sent);
RFCNB_Print_Pkt(stderr, "sent", pkt, len_sent);
#endif
return(len_sent);
}
int RFCNB_Get_Pkt(struct RFCNB_Con *con, struct RFCNB_Pkt *pkt, int len)
{ int read_len, pkt_len;
char hdr[RFCNB_Pkt_Hdr_Len];
struct RFCNB_Pkt *pkt_frag;
int more, this_time, offset, frag_len, this_len;
BOOL seen_keep_alive = TRUE;
if (len < RFCNB_Pkt_Hdr_Len) {
#ifdef RFCNB_DEBUG
fprintf(stderr, "Trying to read less than a packet:");
perror("");
#endif
RFCNB_errno = RFCNBE_BadParam;
return(RFCNBE_Bad);
}
if (RFCNB_Timeout > 0)
alarm(RFCNB_Timeout);
while (seen_keep_alive) {
if ((read_len = read(con -> fd, hdr, sizeof(hdr))) < 0) {
#ifdef RFCNB_DEBUG
fprintf(stderr, "Reading the packet, we got:");
perror("");
#endif
if (errno == EINTR)
RFCNB_errno = RFCNBE_Timeout;
else
RFCNB_errno = RFCNBE_BadRead;
RFCNB_saved_errno = errno;
return(RFCNBE_Bad);
}
if (read_len == 0) {
#ifdef RFCNB_DEBUG
fprintf(stderr, "Connection closed reading\n");
#endif
if (errno == EINTR)
RFCNB_errno = RFCNBE_Timeout;
else
RFCNB_errno = RFCNBE_ConGone;
RFCNB_saved_errno = errno;
return(RFCNBE_Bad);
}
if (RFCNB_Pkt_Type(hdr) == RFCNB_SESSION_KEEP_ALIVE) {
#ifdef RFCNB_DEBUG
fprintf(stderr, "RFCNB KEEP ALIVE received\n");
#endif
}
else {
seen_keep_alive = FALSE;
}
}
if (read_len < sizeof(hdr)) {
memcpy(pkt -> data, hdr, read_len);
#ifdef RFCNB_DEBUG
RFCNB_Print_Pkt(stderr, "rcvd", pkt, read_len);
#endif
return(read_len);
}
pkt_len = RFCNB_Pkt_Len(hdr);
#ifdef RFCNB_DEBUG
fprintf(stderr, "Reading Pkt: Length = %i\n", pkt_len);
#endif
memcpy(pkt -> data, hdr, sizeof(hdr));
if (len < pkt_len)
more = len - RFCNB_Pkt_Hdr_Len;
else
more = pkt_len;
this_time = 0;
if (pkt -> len == read_len){
pkt_frag = pkt -> next;
offset = 0;
}
else {
pkt_frag = pkt;
offset = RFCNB_Pkt_Hdr_Len;
}
frag_len = pkt_frag -> len;
if (more <= frag_len)
this_len = more;
else
this_len = frag_len - offset;
while (more > 0) {
if ((this_time = read(con -> fd, (pkt_frag -> data) + offset, this_len)) <= 0) {
if (errno == EINTR) {
RFCNB_errno = RFCNB_Timeout;
}
else {
if (this_time < 0)
RFCNB_errno = RFCNBE_BadRead;
else
RFCNB_errno = RFCNBE_ConGone;
}
RFCNB_saved_errno = errno;
return(RFCNBE_Bad);
}
#ifdef RFCNB_DEBUG
fprintf(stderr, "Frag_Len = %i, this_time = %i, this_len = %i, more = %i\n", frag_len,
this_time, this_len, more);
#endif
read_len = read_len + this_time;
if (pkt_frag -> next == NULL) break;
pkt_frag = pkt_frag -> next;
this_len = pkt_frag -> len;
offset = 0;
more = more - this_time;
}
#ifdef RFCNB_DEBUG
fprintf(stderr,"Pkt Len = %i, read_len = %i\n", pkt_len, read_len);
RFCNB_Print_Pkt(stderr, "rcvd", pkt, read_len + sizeof(hdr));
#endif
if (read_len < (pkt_len + sizeof(hdr))) {
return(RFCNB_Discard_Rest(con, (pkt_len + sizeof(hdr)) - read_len));
}
if (RFCNB_Timeout > 0)
alarm(0);
return(read_len + sizeof(RFCNB_Hdr));
}