eloop.c

802.1x 源码,基于linux平台开发

/*
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation. See README and COPYING for
 * more details.

	Module Name:
	eloop.c

	Revision History:
	Who         When          What
	--------    ----------    ----------------------------------------------
	Jan, Lee    Dec --2003    modified

*/

#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/time.h>
#include <sys/stat.h>
#include <unistd.h>
#include <errno.h>
#include <signal.h>
#include <fcntl.h>
#include <limits.h>
#include <netinet/in.h>

#include "rt61apd.h"
#include "eloop.h"
#include "switch_if.h"


extern void Recv_Eapol(char * pcBuf, int lPacketLength, int lSrcPort, void * p);


//#define RC_DEBUG
#ifdef RC_DEBUG
#define DEBUG(x...) fprintf(x)
#else
#define DEBUG(x...)
#endif


struct eloop_sock {
	int sock;
	void *eloop_data;
	void *user_data;
	void (*handler)(int sock, void *eloop_ctx, void *sock_ctx);
};

struct eloop_timeout {
	struct timeval time;
	void *eloop_data;
	void *user_data;
	void (*handler)(void *eloop_ctx, void *sock_ctx);
	struct eloop_timeout *next;
};

struct eloop_signal {
	int sig;
	void *user_data;
	void (*handler)(int sig, void *eloop_ctx, void *signal_ctx);
};

struct eloop_data {
	void *user_data;

	int max_sock, reader_count;
	struct eloop_sock *readers;

	struct eloop_timeout *timeout;

	int signal_count;
	struct eloop_signal *signals;

	int terminate;
    int reload;
};

static struct eloop_data eloop;


void eloop_init(void *user_data)
{
	memset(&eloop, 0, sizeof(eloop));
	eloop.user_data = user_data;
}


int eloop_register_read_sock(int sock, void (*handler)(int sock, void *eloop_ctx, void *sock_ctx),
                             void *eloop_data, void *user_data)
{
	struct eloop_sock *tmp;

	tmp = (struct eloop_sock *)	realloc(eloop.readers, (eloop.reader_count + 1) * sizeof(struct eloop_sock));
	if (tmp == NULL)
		return -1;

	tmp[eloop.reader_count].sock = sock;
	tmp[eloop.reader_count].eloop_data = eloop_data;
	tmp[eloop.reader_count].user_data = user_data;
	tmp[eloop.reader_count].handler = handler;
	eloop.reader_count++;
	eloop.readers = tmp;
	DBGPRINT(RT_DEBUG_INFO,"eloop.reader_count(%d)\n",eloop.reader_count);
	
	if (sock > eloop.max_sock)
		eloop.max_sock = sock;

	return 0;
}

int eloop_register_timeout(unsigned int secs, unsigned int usecs,
			   void (*handler)(void *eloop_ctx, void *timeout_ctx),
			   void *eloop_data, void *user_data)
{
	struct eloop_timeout *timeout, *tmp, *prev;

	timeout = (struct eloop_timeout *) malloc(sizeof(*timeout));
	if (timeout == NULL)
		return -1;

	gettimeofday(&timeout->time, NULL);
	timeout->time.tv_sec += secs;
	timeout->time.tv_usec += usecs;

	while (timeout->time.tv_usec >= 1000000)
    {
		timeout->time.tv_sec++;
		timeout->time.tv_usec -= 1000000;
	}
	timeout->eloop_data = eloop_data;
	timeout->user_data = user_data;
	timeout->handler = handler;
	timeout->next = NULL;

	if (eloop.timeout == NULL) {
		eloop.timeout = timeout;
		return 0;
	}

	prev = NULL;
	tmp = eloop.timeout;
	while (tmp != NULL)
    {
		if (timercmp(&timeout->time, &tmp->time, <))
			break;
		prev = tmp;
		tmp = tmp->next;
	}

	if (prev == NULL)
    {
		timeout->next = eloop.timeout;
		eloop.timeout = timeout;
	}
    else
    {
		timeout->next = prev->next;
		prev->next = timeout;
	}

	return 0;
}

int eloop_cancel_timeout(void (*handler)(void *eloop_ctx, void *sock_ctx),
			 void *eloop_data, void *user_data)
{
	struct eloop_timeout *timeout, *prev, *next;
	int removed = 0;

	prev = NULL;
	timeout = eloop.timeout;
	while (timeout != NULL)
    {
		next = timeout->next;

		if (timeout->handler == handler && (timeout->eloop_data == eloop_data || eloop_data == ELOOP_ALL_CTX) 
            && (timeout->user_data == user_data || user_data == ELOOP_ALL_CTX))
        {
			if (prev == NULL)
				eloop.timeout = next;
			else
				prev->next = next;
			free(timeout);
			removed++;
		} else
			prev = timeout;

		timeout = next;
	}

	return removed;
}

static void eloop_handle_signal(int sig)
{
	int i;
	for (i = 0; i < eloop.signal_count; i++)
    {
		if (eloop.signals[i].sig == sig)
        {
			eloop.signals[i].handler(eloop.signals[i].sig, eloop.user_data, eloop.signals[i].user_data);
			break;
		}
	}
}

int eloop_register_signal(int sig, void (*handler)(int sig, void *eloop_ctx, void *signal_ctx), void *user_data)
{
	struct eloop_signal *tmp;

	tmp = (struct eloop_signal *) realloc(eloop.signals, (eloop.signal_count + 1) *	sizeof(struct eloop_signal));
	if (tmp == NULL)
		return -1;

	tmp[eloop.signal_count].sig = sig;
	tmp[eloop.signal_count].user_data = user_data;
	tmp[eloop.signal_count].handler = handler;
	eloop.signal_count++;
	eloop.signals = tmp;
	signal(sig, eloop_handle_signal);

	return 0;
}

void eloop_dump_packet(char * pBuf, int lLength)
{
    int i;
    for(i = 0; i< lLength; i++)
    {
        fprintf(stderr, "%02x ", pBuf[i]);
    }
    fprintf(stderr, "\n");
}

/**************************************************************
*    FUNCTION : eloop_run
***************************************************************
* descript:
* input param:
* 1. rtapd - 
* output param:
* return :
* create by leoncao 2007年1月19日
* modify record:
**************************************************************/
void eloop_run(rtapd * rtapd)
{
/*-----------------------function point-------------------------*
*1. init service
*2. settimer
*3. socket select only for radius socket
*4. timer process
*5. fifo process
*6. socket rcv
*-------------------------------------------------------------*/
    fd_set rfds;
    int i, res;
    struct timeval tv, now;
/*    int fd;
//    int read_byte_num = 0;
    SWITCH_8021X_PACKET_S *pPacket;
    char cBuf[2000];
    int lOffset = 0;*/
    
/*1*/
#if 0    
    tv.tv_sec = 0;
    tv.tv_usec = 50000;
    memset(cBuf, 0, sizeof(cBuf));
    if((mkfifo(SWITCH_8021X_PIPE, 0777) == -1) && (errno != EEXIST))
    {
        fprintf(stderr, "SNMP TRAP: mkfifo %s failed (errno=%d)\r\n", SWITCH_8021X_PIPE, errno);
        /* delete named pipe */
        /*unlink(pipepath);*/
        return;
    }
    
    fd = open(SWITCH_8021X_PIPE, O_NONBLOCK | O_RDONLY);
    if(fd <= 0)
     {
        perror("open error\n");
        //fprintf(stderr, "SWITCH_CLI_Pipe_Task file open error (errorno=%d)\n", errno);
        return;
    }
#endif

    while (!eloop.terminate && (eloop.timeout || eloop.reader_count > 0))
    {
/*2*/    
#if 0
        if (eloop.timeout)
        {
            gettimeofday(&now, NULL);
            if (timercmp(&now, &eloop.timeout->time, >=))
            {
                tv.tv_sec = 0;
                tv.tv_usec = 50000;
            }
            else
            {
                timersub(&eloop.timeout->time, &now, &tv);
                if(tv.tv_sec > 1)
                {
                    tv.tv_sec = 0;
                    tv.tv_usec = 50000;
                }
            }
        }
#else
        if (eloop.timeout)
        {
            gettimeofday(&now, NULL);
            if (timercmp(&now, &eloop.timeout->time, >=))
                tv.tv_sec = tv.tv_usec = 0;
            else
                timersub(&eloop.timeout->time, &now, &tv);
        }
#endif
        
/*3*/
        FD_ZERO(&rfds);
        for (i = 0; i < eloop.reader_count; i++)
            FD_SET(eloop.readers[i].sock, &rfds);

        /*res = select(eloop.max_sock + 1, &rfds, NULL, NULL,
             eloop.timeout ? &tv : NULL);*/
        res = select(eloop.max_sock + 1, &rfds, NULL, NULL,
             &tv);
        if (res < 0 && errno != EINTR)
        {
            perror("select");
            return ;
        }

/*4*/
        /* check if some registered timeouts have occurred */
        if (eloop.timeout)
        {
            struct eloop_timeout *tmp;

            gettimeofday(&now, NULL);
            if (timercmp(&now, &eloop.timeout->time, >=))
            {
                tmp = eloop.timeout;
                eloop.timeout = eloop.timeout->next;
//fprintf(stderr, " eloop timer hanlder\n");
                tmp->handler(tmp->eloop_data, tmp->user_data);
//fprintf(stderr, " eloop timer hanlder over\n");
                free(tmp);
            }
        }
#if 0
/*5*/
        /*read date from switch*/
        read_byte_num = read(fd, cBuf, sizeof(cBuf));
        lOffset = 0;
        while((read_byte_num - lOffset) > 0)
        {
//fprintf(stderr, "Rx a Packet, len=%d\n",read_byte_num);
            pPacket = (SWITCH_8021X_PACKET_S *)&cBuf[lOffset];
//fprintf(stderr, "Rx a Packet, len=%d, offset=%d, port=%d\n",pPacket->lLength, lOffset, pPacket->lPort);
//eloop_dump_packet(pPacket, pPacket->lLength + 8);
            if(pPacket->lLength > (read_byte_num - lOffset - 8))
            {
//fprintf(stderr, " continue 2\n");
                memset(cBuf, 0, sizeof(cBuf));
                break;
            }
//fprintf(stderr, "  Rx a Packet, len=%d, offset=%d, port=%d\n",pPacket->lLength, lOffset, pPacket->lPort);
            Recv_Eapol(pPacket->cPacket, pPacket->lLength, pPacket->lPort, rtapd);
//fprintf(stderr, "  Rx a Packet, Rcv_Eapol process completed\n");
            lOffset += pPacket->lLength + 8;
        }
#endif
/*6*/
        if (res <= 0)
            continue;

        for (i = 0; i < eloop.reader_count; i++)
        {
            if (FD_ISSET(eloop.readers[i].sock, &rfds))
            {
//printf("   eloop.readers[%d].handler\n",i);
                eloop.readers[i].handler(eloop.readers[i].sock,	eloop.readers[i].eloop_data, eloop.readers[i].user_data);
//fprintf(stderr, "   eloop.readers[i].handler process completed\n");
            }
        }
        
    }
//    close(fd);
}

void eloop_terminate(void)
{
	eloop.terminate = 1;
}

void eloop_reload(void)
{
	eloop.reload = 1;
}

void eloop_destroy(void)
{
	struct eloop_timeout *timeout, *prev;

	timeout = eloop.timeout;
	while (timeout != NULL)
    {
		prev = timeout;
		timeout = timeout->next;
		free(prev);
	}
	free(eloop.readers);
	free(eloop.signals);
}

int eloop_terminated(void)
{
	return eloop.terminate;
}