dialog.c

这是一个关于linux下上外网的drcom1.4.4

/*
	libdrcom - Library for communicating with DrCOM 2133 Broadband Access Server
	Copyright (C) 2005 William Poetra Yoga Hadisoeseno <[e-mail]>

	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation; either version 2 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.	See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program; if not, write to the Free Software
	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA	02111-1307	USA
*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <ctype.h>
#include <sys/ioctl.h>
#include <netinet/in.h>

#include "drcomd.h"
#include "daemon_server.h"
#include "log.h"

/*
	type is one of: (all LSB)
	- 0x0001 request
	- 0x0002 challenge
	- 0x0103 login
	- 0x0106 logout
	- 0x0109 passwd
	- 0x0004 success
	- 0x0005 failure
	The length of buf is determined by type.

	If an unexpected packet is recv'ed, we just ignore it.
*/
#if 0
static void dump_packet(unsigned char *pkt, int len)
{
	char left[64];
	char right[20];
	char tmp[8];
	int n = 0;
	unsigned char *p=pkt;

	left[0]='\0';
	loginfo("------dump packet(size=%d)------\n", len);
	while (p < pkt+len) {
		if (n%2==1)
			sprintf(tmp, "%02X ", *p);
		else
			sprintf(tmp, "%02X", *p);
		strcat(left, tmp);
		if (isprint(*p))
			right[n] = *p;
		else
			right[n] = '.';
		n = (n+1)%16;
		if (n==0){
			right[16] = '\0';
			loginfo("%s  %s\n", left, right);
			left[0]='\0';
		}
		p++;
	}

	if(n!=0){
		right[n] = '\0';
		loginfo("%s  %s\n", left, right);
	}
}
#endif

int _send_dialog_packet(struct drcom_socks *socks, void *buf, uint16_t type)
{
	struct drcom_request req, *request=&req;
	int len, r;

	switch (type) {
	case PKT_REQUEST:
		len = sizeof(struct drcom_request);
		memset(request, 0, sizeof(struct drcom_request));

		request->host_header.pkt_type = PKT_REQUEST;
		request->host_header.zero = 0;
		request->host_header.len = 4;
		request->host_header.checksum[0] = 1;

		buf = (void *) request;
		loginfo("send: PKT_REQUEST\n");
		break;
	case PKT_LOGIN: 
		len = sizeof(struct drcom_login_packet); 
		loginfo("send: PKT_LOGIN\n");
		break;
	case PKT_LOGOUT: 
		len = sizeof(struct drcom_logout_packet); 
		loginfo("send: PKT_LOGOUT\n");
		break;
	case PKT_PASSWORD_CHANGE: 
		len = sizeof(struct drcom_passwd_packet); 
		loginfo("send: PKT_PASSWD_CHANGE\n");
		break;
	default: 
		loginfo("send: Unknown PKT\n");
		return -2; 
		break;
	}

	r = sendto(socks->sockfd, buf, len, 0,
			(struct sockaddr *) &socks->servaddr_in,
			sizeof(struct sockaddr));

	if (r != len){
		logerr("send:failed\n");
		return -1;
	}

	return 0;
}

int _recv_dialog_packet(struct drcom_socks *socks, unsigned char **pkt, int *pkt_size)
{
	unsigned char *p=NULL;
	fd_set readfds;
	int r;
	int size;

	*pkt = NULL;
	*pkt_size = 0;

	while(1){
		struct timeval t;
		int addrlen;

		FD_ZERO(&readfds);
		FD_SET(socks->sockfd, &readfds);
		t.tv_sec = 2;
		t.tv_usec = 0;
		r = select(socks->sockfd+1, &readfds, NULL, NULL, &t);
		if(r < 0){
			if(errno == EINTR)
				continue;
			logerr("select() returned negative(errno=%d)\n", errno);
			return r;
		}
		if(r==0){
			logerr("select() timeout\n");
			return -1;
		}

		if(!FD_ISSET(socks->sockfd, &readfds))
			continue;

		r = ioctl(socks->sockfd, FIONREAD, &size);
		if(r != 0){
	                logerr("_recv_dialog_packet: ioctl()");
			return -1;
		}
		if(size == 0)
			continue;

		p = (unsigned char*)malloc(size);
		if (p==NULL){
			logerr("malloc failed\n");
			return -1;
		}

		addrlen = sizeof(struct sockaddr);
		r = recvfrom(socks->sockfd, p, size, 0,
			(struct sockaddr *) &socks->servaddr_in,
	 		(unsigned int *) &addrlen);

		if (r < 0){
			logerr("recvfrom: %s\n", strerror(errno));
			free(p);
			return r;
		}

		*pkt = p;
		*pkt_size = size;

		return 0;
	}
}