libhttp服务端ipv6兼容ipv4代码抽取重组

13 篇文章 0 订阅
3 篇文章 0 订阅

简单分析

https://github.com/lammertb/libhttp
c语言  WIN/UNIX
httplib_set_ports_option.c 线程:根据串创定[仅ipv4]或[仅ipv6]或[兼容4_6]的服务端监听,监听套接字入监听缓冲区
master_thread_run(void*thread_func_param)线程:监听缓冲区中的有效socket轮询poll查看,并
XX_httplib_accept_new_connection,掩码和属性设置,新连接套接字入客户端缓冲区
https://tools.ietf.org/html/rfc3513#section-2.2
so.sock = socket( so.lsa.sa.sa_family, SOCK_STREAM, 6 )
//the same port can be used again in the same process  /linux和WIN不一样
 setsockopt( so.sock, SOL_SOCKET, SO_REUSEADDR, (SOCK_OPT_TYPE)&on, sizeof(on) ) != 0 
only  ipv4========
so.lsa.sa.sa_family == AF_INET
len = sizeof(so.lsa.sin);
bind( so.sock, &so.lsa.sa, len ) != 0
only  ipv6========
int    off         = 0;
so.lsa.sa.sa_family == AF_INET6
  setsockopt( so.sock, IPPROTO_IPV6, IPV6_V6ONLY, (void *)&off, sizeof(off)
len = sizeof(so.lsa.sin6);
both  ipv4-ipv6====
 sizeof(so.lsa.sin6);
bind( so.sock, &so.lsa.sa, len ) != 0 )
----------------------------------------------------------
 listen( so.sock, SOMAXCONN )
getsockname( so.sock, &(usa.sa), &len ) != 0  ||  usa.sa.sa_family  !=  so.lsa.sa.sa_family
if ( so.lsa.sa.sa_family == AF_INET6 ) so.lsa.sin6.sin6_port = usa.sin6.sin6_port;
		else             so.lsa.sin.sin_port   = usa.sin.sin_port;
//
httplib_master_thread.c
select   accept--

 

 connect() returns “invalid argument” with ipv6 address

https://stackoverflow.com/questions/12260003/connect-returns-invalid-argument-with-ipv6-address
即图部分文字,作为客户端使用getaddrinfo,使用本地连接【fe80..】需要指定接口id

Addresses starting with ff... are multicast addresses. Connecting a stream to a multicast address does not work.

Addresses starting with fe80... are link-local addresses, which have an interface identifier associated with them. Try looking at the sockaddr returned from getaddrinfo, is the scope field filled out?
share edit
answered Sep 4 '12 at 10:01
Simon Richter
24.6k11 gold badge3737 silver badges5757 bronze badges

    2
    This field should contain the interface number. Have you specified the scope with the %eth0 or %1 (1 being the interface number, which you can look up with the ip tool) notation?\ – Simon Richter Sep 4 '12 at 10:55
    Nope. When I do that, this scope field is filled with some value and it seems to work (although I need to configure my specific FW, but I get no invalid argument anymore). But I seem to miss something and in all the materials connected with ipv6 I was reading I didn't come across this - do I need to specify the interface whenever I use link-local address? – flyjohny Sep 4 '12 at 12:43
    4
    Yes, you do. Link-local address means precisely that uniqueness of the address is only guaranteed on this particular link, so the system has no way of finding out which interface to send the data to unless you tell it. The alternative is to assign addresses with global scope and use a routing table. – Simon Richter Sep 4 '12 at 15:01
    @SimonRichter - If I am using the local link address, does it mean that the scope_id field will be always 0 and we need to explicitly specify it ? Can system pick this automatically with any config in /etc/hosts ot any other config ? (Note : I have no DNS in place in my network) – kaps Jan 17 '19 at 5:26
    1
    @kaps, no, there is no way to have the system auto-select the interface. It is perfectly legal to have ppp0, ppp1 and ppp2 all configured to use a local address of fe80::1 and a peer address of fe80::2, so the OS could not decide which interface to use for a destination address of fe80::2. So the program needs to fill in the scope_id. When the addresses are configured by the user, getaddrinfo will fill it out when the user uses the % notation, when you use multicast for discovery, you can use the scope ID from the discovery packet's origin. – Simon Richter Jan 18 '19 at 2:07

Makefile编译make

obj= c2.o   ipv4_6s.o  ci6.o
CC=gcc  
CFLAGS= -Wall -g
CPPFLAGS= -lpthread
.PHONY:th
th:$(obj)
$(obj):%.o:%.c
	$(CC)  $(CFLAGS) $(<) $(CPPFLAGS) -o $(@)
clean:
	rm *.o

运行

 

 

./ipv4_6.s.o
./c2.o
./ci6.o  ::1/128  8080

 

c2.c  ipv4 客户端

 

#include<linux/sockios.h>
#include<sys/ioctl.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<string.h>
#include<stdio.h>
#include<errno.h>
void printfd(int sock)
{
	//thread  safe  C99
	socklen_t						peer_addr_size,selflen;
	struct sockaddr_in 			peer_addr,self;
	unsigned int				esme_port=0;
	char						esme_ip[200];
	int ern;
	int ret=0;
	peer_addr_size=sizeof(struct sockaddr);
	ret=getpeername(sock,(struct sockaddr*)&peer_addr,&peer_addr_size);
	ern=errno;
	if(!ret){
	printf("peer[%s]\n",inet_ntop(peer_addr.sin_family,&peer_addr.sin_addr,esme_ip,sizeof(esme_ip))); //ip  线程   inet_ntoa不安全  
	esme_port = ntohs(peer_addr.sin_port);
	printf("peer%d**[sock=%d] port[%d] ip[%s]\n",ret,sock,esme_port,esme_ip);
	}else
	{
		printf("peer %s\n",strerror(ern));
	}
	selflen=sizeof(struct sockaddr);
	ret=getsockname(sock, (struct sockaddr*)&self,&selflen);
	ern=errno;
	if(!ret){
	printf("self[%s]\n",inet_ntop(self.sin_family,&self.sin_addr,esme_ip,sizeof(esme_ip)));
	esme_port = ntohs(self.sin_port);
	printf("self%d**[sock=%d] port[%d] ip[%s]\n",ret,sock,esme_port,esme_ip);
	}else
	{
		printf("self %s\n",strerror(ern));
	}
}


int  main()
{
 
int sockClient;//客户端Socket
struct sockaddr_in addrServer;//服务端地址

//新建客户端socket
sockClient=socket(AF_INET,SOCK_STREAM,0);
if(sockClient<0)
{
	printf("socket errno\n");
	return 1;
}
//定义要连接的服务端地址
inet_pton(AF_INET, "127.0.0.1", &addrServer.sin_addr.s_addr);
//addrServer.sin_addr.s_addr=inet_addr("127.0.0.1");//目标IP(127.0.0.1是回送地址)
addrServer.sin_family=AF_INET;
addrServer.sin_port=htons(8080);//连接端口6000

//连接到服务端
if(connect(sockClient,(struct sockaddr*)&addrServer,sizeof(struct sockaddr))<0)
{
	printf("connect errno\n");
	return 1;
}

//发送数据
char message[100]="HelloSocket!";
send(sockClient,message,strlen(message)+1,0);
 printfd(sockClient);
//接受服务器返回,也可以用recv();并打印
int  bytes=0;
int ret;
sleep(1);
//ret=recv(sockClient,&bytes,sizeof(bytes),MSG_TRUNC );
//ret=ioctl(sockClient,SIOCINQ,&bytes);
//bytes=bytes>100?100:bytes;

//printf("ret%d Q-recv[%d]\n",ret,bytes);
//read(sockClient,message,bytes);
// printfd(sockClient);
//printf("%s",message);
 int sockfd=sockClient;


write(sockfd, "123", 3);  
    printf("wrote 3 bytes of normal data\n");  
    sleep(1);  
  
    send(sockfd, "4", 1, MSG_OOB);  
    printf("wrote 1 byte of OOB data\n");  
    sleep(1);  
  
    write(sockfd, "56", 2);  
    printf("wrote 2 bytes of normal data\n");  
    sleep(1);  
  
    send(sockfd, "mc7", 3, MSG_OOB);  
    printf("wrote 1 byte of OOB data\n");  
    sleep(1);  
  
    write(sockfd, "89", 2);  
    printf("wrote 2 bytes of normal data\n");  
    sleep(1);  
  
    exit(0);  



//关闭socket
close(sockClient);
return 0;}

 

ci6.c  ipv6客户端

 

#include <stdio.h>  
#include <string.h>  
#include <errno.h>  
#include <sys/socket.h>  
#include <resolv.h>  
#include <stdlib.h>  
#include <netinet/in.h>  
#include <arpa/inet.h>  
#include <unistd.h>  
#define MAXBUF 1024 
//源代码来自: https://blog.csdn.net/hepeng597/article/details/7803277
//仅ipv6  客户端
//ifconfig  inet6 看ipv6地址
//gcc ci6.c
//#运行  程序名  地址    端口
//run :./ci6.o  ::1/128  8080
//run :./ci6.o  fe40::20c:29ff:eec1:9896%eth0 8080 
int main(int argc, char **argv)  
{  
    int sockfd, len;  
    /* struct sockaddr_in dest; */ // IPv4  
    struct sockaddr_in6 dest;      // IPv6  
    char buffer[MAXBUF + 1];  
  
    if (argc != 3) {  
        printf  
            ("参数格式错误!正确用法如下:\n\t\t%s IP地址 端口\n\t比如:\t%s ::1/128 8080\n此程序用来从某个 IP 地址的服务器某个端口接收最多 MAXBUF 个字节的消息",  
             argv[0], argv[0]);  
        exit(0);  
    }  
    /* 创建一个 socket 用于 tcp 通信 */  
    /* if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) { */ // IPv4  
    if ((sockfd = socket(AF_INET6, SOCK_STREAM, 0)) < 0) {      // IPv6  
        perror("Socket");  
        exit(errno);  
    }  
    printf("socket created\n");  
  
    /* 初始化服务器端(对方)的地址和端口信息 */  
    bzero(&dest, sizeof(dest));  
    /* dest.sin_family = AF_INET; */  // IPv4  
    dest.sin6_family = AF_INET6;     // IPv6  
    /* dest.sin_port = htons(atoi(argv[2])); */ // IPv4  
    dest.sin6_port = htons(atoi(argv[2]));     // IPv6  
    /* if (inet_aton(argv[1], (struct in_addr *) &dest.sin_addr.s_addr) == 0) { */ // IPv4  
    if ( inet_pton(AF_INET6, argv[1], &dest.sin6_addr) < 0 ) {                 // IPv6  
        perror(argv[1]);  
        exit(errno);  
    }  
    printf("address created\n");  
  
    /* 连接服务器 */  
    if (connect(sockfd, (struct sockaddr *) &dest, sizeof(dest)) != 0) {  
        perror("Connect ");  
        exit(errno);  
    }  
    printf("server connected\n");  
  
    /* 接收对方发过来的消息,最多接收 MAXBUF 个字节 */  
    bzero(buffer, MAXBUF + 1);  
    /* 接收服务器来的消息 */  
    len = recv(sockfd, buffer, MAXBUF, 0);  
    if (len > 0)  
        printf("接收消息成功:'%s',共%d个字节的数据\n",  
               buffer, len);  
    else  
        printf  
            ("消息接收失败!错误代码是%d,错误信息是'%s'\n",  
             errno, strerror(errno));  
  
    bzero(buffer, MAXBUF + 1);  
    strcpy(buffer, "这是客户端发给服务器端的消息\n");  
    /* 发消息给服务器 */  
    len = send(sockfd, buffer, strlen(buffer), 0);  
    if (len < 0)  
        printf  
            ("消息'%s'发送失败!错误代码是%d,错误信息是'%s'\n",  
             buffer, errno, strerror(errno));  
    else  
        printf("消息'%s'发送成功,共发送了%d个字节!\n",  
               buffer, len);  
  
    /* 关闭连接 */  
    close(sockfd);  
    return 0;  
}  

 

ipv4_6s.c改造的服务端

 

#include<sys/socket.h>
#include<string.h>
#include<stdio.h>
#include <netdb.h>
#include <netinet/tcp.h>

#include<stdint.h>//uint64_t
#include <fcntl.h>
#include <poll.h>
#include<stdlib.h>//exit
#include<errno.h>
#include <stdbool.h>//bool
#include <unistd.h>//close
#include<malloc.h>
#include<pthread.h>

#if !defined(SOMAXCONN)
#define SOMAXCONN (100)
#endif

#define INVALID_SOCKET (-1)
#define IP_ADDR_STR_LEN (50) /* IPv6 hex string is 46 chars */
#define ERROR_STRING_LEN		(256)
#define ERRNO		errno
#define closesocket(fd)  close(fd)
#define IMGAPISTACKSIZE   10485760//10*1024*1024  线程栈  10MB

#define STRX(x) #x
#define STR(x) STRX(x)
#define __func__ __FILE__ ":" STR(__LINE__)
//
#define httplib_cry( a, ctx, b, format,...)  printf(format,##__VA_ARGS__)  

typedef int SOCKET;
/* Unified socket address. For IPv6 support, add IPv6 address structure in the
 * union u. */
union usa {
	struct sockaddr sa;
	struct sockaddr_in sin;
	struct sockaddr_in6 sin6;
};
/* Describes listening socket, or socket which was accept()-ed by the master
 * thread and queued for future handling by the worker thread. */
struct socket {
	SOCKET sock;			// Listening socket					
	union usa lsa;			// Local socket address				
	union usa rsa;			//Remote socket address				
	//bool has_ssl;			Is port SSL-ed					
	//bool has_redir;			Is port supposed to redirect everything to SSL port	
	unsigned char in_use;		// Is valid	
	
};

typedef const void *SOCK_OPT_TYPE;

/* Describes a string (chunk of memory). */
struct vec {
	const char *	ptr;
	size_t		len;
};
struct demolh_ctx_t {
	struct socket *listening_sockets;//malloc  relloc  free
	struct pollfd *listening_socket_fds;//malloc  relloc  free
	unsigned int num_listening_sockets;
	char *	listening_ports;//eg.+8080  #服务端ip46兼容   192.168.10.23:8080 #ipv4   [::1]:8080 #ipv6   192.168.10.23:6000,[::1]:8080 #多个 
};

//用一个size_t标记存的数据,因此a-1后申请空间
static int64_t				httplib_memory_blocks_used	= 0;
static int64_t				httplib_memory_bytes_used	= 0;
typedef void (*httplib_alloc_callback_func)( const char *file, unsigned line, const char *action, int64_t current_bytes, int64_t total_blocks, int64_t total_bytes );
static httplib_alloc_callback_func	alloc_log_func			= NULL;
#define					httplib_calloc(a, b) XX_httplib_calloc_ex(a, b, __FILE__, __LINE__)
#define					httplib_free(a) XX_httplib_free_ex(a, __FILE__, __LINE__)
#define					httplib_malloc(a) XX_httplib_malloc_ex(a, __FILE__, __LINE__)
#define					httplib_realloc(a, b) XX_httplib_realloc_ex(a, b, __FILE__, __LINE__)

void *			XX_httplib_calloc_ex( size_t count, size_t size, const char *file, unsigned line );
void *			XX_httplib_free_ex( void *memory, const char *file, unsigned line );
void *			XX_httplib_malloc_ex( size_t size, const char *file, unsigned line );
void *			XX_httplib_realloc_ex( void *memory, size_t newsize, const char *file, unsigned line );

void XX_httplib_close_all_listening_sockets( struct demolh_ctx_t *ctx ) {

	unsigned int i;

	if ( ctx == NULL ) return;

	for (i=0; i<ctx->num_listening_sockets; i++) {

		closesocket( ctx->listening_sockets[i].sock );
		ctx->listening_sockets[i].sock = INVALID_SOCKET;
	}

	ctx->listening_sockets    = httplib_free( ctx->listening_sockets    );
	ctx->listening_socket_fds = httplib_free( ctx->listening_socket_fds );

}  /* XX_close_all_listening_sockets */

void XX_httplib_set_close_on_exec( SOCKET fd ) { 
	fcntl( fd, F_SETFD, FD_CLOEXEC );
}  /* XX_httplib_set_close_on_exec */
int XX_httplib_inet_pton( int af, const char *src, void *dst, size_t dstlen ) {

	struct addrinfo hints;
	struct addrinfo *res;
	struct addrinfo *ressave;
	int func_ret;
	int gai_ret;

	func_ret = 0;

	memset( & hints, 0, sizeof(struct addrinfo) );
	hints.ai_family = af;

	gai_ret = getaddrinfo( src, NULL, &hints, &res );

	if ( gai_ret != 0 ) {

		/*
		 * gai_strerror could be used to convert gai_ret to a string
		 * POSIX return values: see
		 * http://pubs.opengroup.org/onlinepubs/9699919799/functions/freeaddrinfo.html
		 *
		 * Windows return values: see
		 * https://msdn.microsoft.com/en-us/library/windows/desktop/ms738520%28v=vs.85%29.aspx
		 */

		return 0;
	}
	ressave = res;
	while ( res ) {

		if ( dstlen >= res->ai_addrlen ) {

			memcpy( dst, res->ai_addr, res->ai_addrlen );
			func_ret = 1;
		}
		res = res->ai_next;
	}
	freeaddrinfo( ressave );

	return func_ret;

}  /* XX_httplib_inet_pton */
int XX_httplib_is_valid_port( unsigned long port ) {

	return ( port < 0xffff );

}  
/*略过LWS 
 *,分隔
 *eg:
 *list:80, 443s, 127.0.0.1:3128, 192.0.2.3:8080s
 *list:[::]:80, [::1]:80, [2001:0db8:7654:3210:FEDC:BA98:7654:3210]:443s
 *list:+8080
 */
/* XX_httplib_is_valid_port */
const char *XX_httplib_next_option( const char *list, struct vec *val, struct vec *eq_val ) {

	int end;

reparse:
	if ( val == NULL  ||  list == NULL  ||  *list == '\0' ) return NULL;
	
	/*
	 * Skip over leading LWS
	 */

	while ( *list == ' '  ||  *list == '\t' ) list++;

	val->ptr = list;
	if ( (list = strchr( val->ptr, ',' )) != NULL ) {

		/*
		 * Comma found. Store length and shift the list ptr
		 */

		val->len = ((size_t)(list - val->ptr));
		list++;
	}
	
	else {

		/*
		 * This value is the last one
		 */

		list     = val->ptr + strlen(val->ptr);
		val->len = ((size_t)(list - val->ptr));
	}

	/*
	 * Adjust length for trailing LWS
	 */

	end = (int)val->len - 1;
	while ( end >= 0  &&  ( val->ptr[end] == ' ' || val->ptr[end] == '\t' ) ) end--;

	val->len = (size_t)(end + 1);

	if ( val->len == 0 ) goto reparse; /* Ignore any empty entries. */

	if ( eq_val != NULL ) {

		/*
		 * Value has form "x=y", adjust pointers and lengths
		 * so that val points to "x", and eq_val points to "y".
		 */

		eq_val->len = 0;
		eq_val->ptr = (const char *)memchr( val->ptr, '=', val->len );

		if ( eq_val->ptr != NULL ) {

			eq_val->ptr++; /* Skip over '=' character */
			eq_val->len = ((size_t)(val->ptr - eq_val->ptr)) + val->len;
			val->len    = ((size_t)(eq_val->ptr - val->ptr)) - 1;
		}
	}

	return list;

}  /* XX_httplib_next_option */
char *httplib_error_string( int error_code, char *buf, size_t buf_len ) {

	if ( buf == NULL  ||  buf_len < 1 ) return NULL;
	strerror_r( error_code, buf, buf_len );
	return buf;

}  /* httplib_error_string */
static bool parse_port_string( const struct vec *vec, struct socket *so, int *ip_version ) {

	unsigned int a;
	unsigned int b;
	unsigned int c;
	unsigned int d;
	unsigned int port;
	int ch;
	int len;
	char buf[100] = {0};
	/*
	 * MacOS needs that. If we do not zero it, subsequent bind() will fail.
	 * Also, all-zeroes in the socket address means binding to all addresses
	 * for both IPv4 and IPv6 (INADDR_ANY and IN6ADDR_ANY_INIT).
	 */

	memset( so, 0, sizeof(*so) );
	so->lsa.sin.sin_family = AF_INET;
	*ip_version            = 0;
	if ( sscanf( vec->ptr, "%u.%u.%u.%u:%u%n", &a, &b, &c, &d, &port, &len ) == 5 ) {

		/*
		 * Bind to a specific IPv4 address, e.g. 192.168.1.5:8080
		 */

		so->lsa.sin.sin_addr.s_addr = htonl( (a << 24) | (b << 16) | (c << 8) | d );
		so->lsa.sin.sin_port        = htons( (uint16_t)port );

		*ip_version = 4;
	}
	else if ( sscanf( vec->ptr, "[%49[^]]]:%u%n", buf, &port, &len ) == 2  &&  XX_httplib_inet_pton( AF_INET6, buf, &so->lsa.sin6, sizeof(so->lsa.sin6) ) ) {

		/*
		 * IPv6 address, examples: see above
		 * so->lsa.sin6.sin6_family = AF_INET6; already set by httplib_inet_pton
		 */

		so->lsa.sin6.sin6_port = htons( (uint16_t)port );
		*ip_version = 6;
	}
	else if ( vec->ptr[0] == '+'  &&  sscanf( vec->ptr + 1, "%u%n", &port, &len ) == 1 ) {

		/*
		 * Port is specified with a +, bind to IPv6 and IPv4, INADDR_ANY
		 * Add 1 to len for the + character we skipped before
		 */

		len++;

		/*
		 * Set socket family to IPv6, do not use IPV6_V6ONLY
		 */
		so->lsa.sin6.sin6_family = AF_INET6;
		so->lsa.sin6.sin6_port   = htons(( uint16_t)port );

		*ip_version = 4 + 6;
	}
	else if ( sscanf( vec->ptr, "%u%n", &port, &len ) == 1 ) {

		/*
		 * If only port is specified, bind to IPv4, INADDR_ANY
		 */

		so->lsa.sin.sin_port = htons( (uint16_t)port );
		*ip_version = 4;

	}
	else {
		/*
		 * Parsing failure. Make port invalid.
		 */

		port = 0;
		len  = 0;
	}
	/*
	 * sscanf and the option splitting code ensure the following condition
	 */

	if ( len < 0 && ((unsigned)len) > ((unsigned)vec->len) ) {

		*ip_version = 0;
		return false;
	}

	ch            = vec->ptr[len]; /* Next character after the port number */
	//so->has_ssl   = ( ch == 's' );
	//so->has_redir = ( ch == 'r' );

	/*
	 * Make sure the port is valid and vector ends with 's', 'r' or ','
	 */

	if ( XX_httplib_is_valid_port( port )  &&  ( ch == '\0'  ||  ch == 's'  ||  ch == 'r'  ||  ch == ',' ) ) return true;

	/*
	 * Reset ip_version to 0 if there is an error
	 */

	*ip_version = 0;
	return false;

}  /* parse_port_string */

int demoXX_httplib_set_ports_option( struct demolh_ctx_t *ctx ) {

	const char *list;
	char error_string[ERROR_STRING_LEN];
	int on;
	int off;
	struct vec vec;
	struct socket so;
	struct socket *ptr;
	struct pollfd *pfd;
	union usa usa;
	socklen_t len;
	int ip_version;
	int ports_total;
	int ports_ok;

	if ( ctx == NULL ) return 0;

	on          = 1;
	off         = 0;
	ports_total = 0;
	ports_ok    = 0;

	memset( & so,  0, sizeof(so)  );
	memset( & usa, 0, sizeof(usa) );

	len  = sizeof(usa);
	list = ctx->listening_ports;

	while ( (list = XX_httplib_next_option( list, &vec, NULL )) != NULL ) {

		ports_total++;

		if ( ! parse_port_string( &vec, &so, &ip_version ) ) {

			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: %.*s: invalid port spec (entry %i). Expecting list of: %s", __func__, (int)vec.len, vec.ptr, ports_total, "[IP_ADDRESS:]PORT[s|r]" );
			continue;
		}
		if ( ( so.sock = socket( so.lsa.sa.sa_family, SOCK_STREAM, 6 ) ) == INVALID_SOCKET ) {

			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: cannot create socket (entry %i)", __func__, ports_total );
			continue;
		}

		if ( setsockopt( so.sock, SOL_SOCKET, SO_REUSEADDR, (SOCK_OPT_TYPE)&on, sizeof(on) ) != 0 ) {

			/*
			 * Set reuse option, but don't abort on errors.
			 */

			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: cannot set socket option SO_REUSEADDR (entry %i)", __func__, ports_total );
		}

		if ( ip_version > 4 ) {

			if ( ip_version == 6 ) {

				if ( so.lsa.sa.sa_family == AF_INET6  &&  setsockopt( so.sock, IPPROTO_IPV6, IPV6_V6ONLY, (void *)&off, sizeof(off) ) != 0 ) {

					/*
					 * Set IPv6 only option, but don't abort on errors.
					 */

					httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: cannot set socket option IPV6_V6ONLY (entry %i)", __func__, ports_total );
				}
			}
		}

		if ( so.lsa.sa.sa_family == AF_INET ) {

			len = sizeof(so.lsa.sin);

			if ( bind( so.sock, &so.lsa.sa, len ) != 0 ) {

				httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: cannot bind to %.*s: %d (%s)", __func__, (int)vec.len, vec.ptr, (int)ERRNO, httplib_error_string( ERRNO, error_string, ERROR_STRING_LEN ) );
				closesocket( so.sock );
				so.sock = INVALID_SOCKET;
				continue;
			}
		}

		else if ( so.lsa.sa.sa_family == AF_INET6 ) {

			len = sizeof(so.lsa.sin6);

			if ( bind( so.sock, &so.lsa.sa, len ) != 0 ) {

				httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: cannot bind to IPv6 %.*s: %d (%s)", __func__, (int)vec.len, vec.ptr, (int)ERRNO, httplib_error_string( ERRNO, error_string, ERROR_STRING_LEN ) );
				closesocket( so.sock );
				so.sock = INVALID_SOCKET;
				continue;
			}
		}

		else {
			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: cannot bind: address family not supported (entry %i)", __func__, ports_total );
			continue;
		}

		if ( listen( so.sock, SOMAXCONN ) != 0 ) {

			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: cannot listen to %.*s: %d (%s)", __func__, (int)vec.len, vec.ptr, (int)ERRNO, httplib_error_string( ERRNO, error_string, ERROR_STRING_LEN ) );
			closesocket( so.sock );
			so.sock = INVALID_SOCKET;
			continue;
		}

		if ( getsockname( so.sock, &(usa.sa), &len ) != 0  ||  usa.sa.sa_family  !=  so.lsa.sa.sa_family ) {

			int err = (int)ERRNO;
			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: call to getsockname failed %.*s: %d (%s)", __func__, (int)vec.len, vec.ptr, err, httplib_error_string( ERRNO, error_string, ERROR_STRING_LEN ) );
			closesocket( so.sock );
			so.sock = INVALID_SOCKET;
			continue;
		}

/*
 * Update lsa port in case of random free ports
 */

		if ( so.lsa.sa.sa_family == AF_INET6 ) so.lsa.sin6.sin6_port = usa.sin6.sin6_port;
		else                                   so.lsa.sin.sin_port   = usa.sin.sin_port;

		ptr = httplib_realloc( ctx->listening_sockets, (ctx->num_listening_sockets+1) * sizeof(ctx->listening_sockets[0]) );

		if ( ptr != NULL ) ctx->listening_sockets = ptr;
		else {
			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: out of memory on listening sockets", __func__ );
			closesocket( so.sock );
			so.sock = INVALID_SOCKET;
			continue;
		}


		pfd = httplib_realloc( ctx->listening_socket_fds, (ctx->num_listening_sockets+1) * sizeof(ctx->listening_socket_fds[0]) );

		if ( pfd != NULL ) ctx->listening_socket_fds = pfd;
		else {
			httplib_cry( LH_DEBUG_CRASH, ctx, NULL, "%s: out of memory on fds", __func__ );
			closesocket( so.sock );
			so.sock = INVALID_SOCKET;
			continue;
		}


		XX_httplib_set_close_on_exec( so.sock );

		ctx->listening_sockets[ctx->num_listening_sockets] = so;
		ctx->num_listening_sockets++;

		ports_ok++;
	}

	if ( ports_ok != ports_total ) {

		XX_httplib_close_all_listening_sockets( ctx );
		ports_ok = 0;
	}

	return ports_ok;

}  /* XX_httplib_set_ports_option */


 void *XX_httplib_realloc_ex( void *memory, size_t newsize, const char *file, unsigned line ) {

	size_t *olddata;
	size_t *newdata;
	size_t oldsize;
	int64_t diff;

	if ( newsize == 0 ) {

		if ( memory != NULL ) XX_httplib_free_ex( memory, file, line );
		return NULL;
	}

	if ( memory == NULL ) return XX_httplib_malloc_ex( newsize, file, line );

	olddata = ((size_t *)memory) - 1;
	oldsize = *olddata;
	newdata = realloc( olddata, newsize + sizeof(size_t) );
	if ( newdata == NULL ) {
		
		if ( alloc_log_func != NULL ) alloc_log_func( file, line, "realloc", 0, httplib_memory_blocks_used, httplib_memory_bytes_used );
		return NULL;
	}

	httplib_memory_bytes_used -= oldsize;
	httplib_memory_bytes_used += newsize;

	*newdata = newsize;
	diff     = ((int64_t)newsize) - ((int64_t)oldsize);

	if ( alloc_log_func != NULL ) alloc_log_func( file, line, "realloc", diff, httplib_memory_blocks_used, httplib_memory_bytes_used );

	return (newdata+1);

}  /* demoXX_httplib_realloc_ex */
void *XX_httplib_malloc_ex( size_t size, const char *file, unsigned line ) {

	size_t *data;

	if ( size == 0 ) {

		if ( alloc_log_func != NULL ) alloc_log_func( file, line, "malloc", 0, httplib_memory_blocks_used, httplib_memory_bytes_used );
		return NULL;
	}

	data = malloc( size + sizeof(size_t) );

	if ( data == NULL ) {
	
		if ( alloc_log_func != NULL ) alloc_log_func( file, line, "malloc", 0, httplib_memory_blocks_used, httplib_memory_bytes_used );
		return NULL;
	}

	httplib_memory_bytes_used += size;
	httplib_memory_blocks_used++;

	*data = size;

	if ( alloc_log_func != NULL ) alloc_log_func( file, line, "malloc", size, httplib_memory_blocks_used, httplib_memory_bytes_used );

	return (data+1);

}  /* XX_httplib_malloc_ex */
void *XX_httplib_free_ex( void *memory, const char *file, unsigned line ) {

	size_t *data;

	if ( memory == NULL ) return NULL;

	data = ((size_t *)memory) - 1;

	httplib_memory_bytes_used -= *data;
	httplib_memory_blocks_used--;

	if ( alloc_log_func != NULL ) alloc_log_func( file, line, "free", - ((int64_t)*data), httplib_memory_blocks_used, httplib_memory_bytes_used );

	free( data );

	return NULL;

}  /* XX_httplib_free_ex */
void *XX_httplib_calloc_ex( size_t count, size_t size, const char *file, unsigned line ) {

	void *data;

	data = XX_httplib_malloc_ex( size*count, file, line );
	if ( data == NULL ) return NULL;

	memset( data, 0x00, size*count );

	return data;

}  /* XX_httplib_calloc_ex */


int httplib_poll( struct pollfd *pfd, unsigned int n, int milliseconds ) 
{
		return poll( pfd, n, milliseconds );
}


void XX_httplib_sockaddr_to_string( char *buf, size_t len, const union usa *usa ) {

	if ( usa == NULL  ||  buf == NULL  ||  len < 1 ) return;

	buf[0] = '\0';

	if      ( usa->sa.sa_family == AF_INET  ) getnameinfo(&usa->sa, sizeof(usa->sin),  buf, (unsigned)len, NULL, 0, NI_NUMERICHOST );
	else if ( usa->sa.sa_family == AF_INET6 ) getnameinfo(&usa->sa, sizeof(usa->sin6), buf, (unsigned)len, NULL, 0, NI_NUMERICHOST );

}  /* XX_httplib_sockaddr_to_string */

void*PostRoutine(void*pram) 
{//Linux
	struct socket *psock=(struct socket *)pram;
	int connfd=psock->sock;
	int ret;
	free(pram);
	psock=pram=NULL;
	write(connfd,"you link libhttp ok !",24);
     char buf[1024];
    fd_set read_fds;
    fd_set exception_fds;
    FD_ZERO(&read_fds);
    FD_ZERO(&exception_fds);

    while(1)
    {
        memset(buf, '\0', sizeof(buf));
        /* 每次调用select前都要重新在read_fds和exception_fds中设置文件描述符connfd,因为事件发生之后,文件描述符集合将被内核修改 */
        FD_SET(connfd, &read_fds);
        FD_SET(connfd, &exception_fds);
        ret = select(connfd + 1, &read_fds, NULL, &exception_fds, NULL);
        if(ret < 0)
        {
            printf("selection failure\n");
            break;
        }

        /* 对于可读事件,采用普通的recv函数读取数据 */
        if(FD_ISSET(connfd, &read_fds))
        {
            ret = recv(connfd, buf, sizeof(buf) - 1, 0);
            if(ret <= 0)
            {
                break;
            }
            printf("get %d bytes of normal data: %s\n", ret, buf);
        }
        /* 对于异常事件,采用带MSG_OOB标志的recv函数读取带外数据 */
        else if(FD_ISSET(connfd, &exception_fds))
        {
            ret = recv(connfd, buf, sizeof(buf) - 1, MSG_OOB);
            if(ret <= 0)
            {
                break;
            }
            printf("get %d bytes of oob data: %s\n", ret, buf);
        }
    }
    close(connfd);
	pthread_exit(NULL);
}
int pthreadFormPost(void* psock) 
{//Linux
	int ret = 0;
	int eno;
	pthread_attr_t  attr; 
	pthread_t  pthreadid;
	struct socket *pram=NULL;
	ret=pthread_attr_init(&attr);
	if (0 != ret) 
	{
		printf("pthread_attr_init()error\n");
		return ret;
	}
	ret = pthread_attr_setstacksize(&attr, IMGAPISTACKSIZE);
	if (0 != ret)
	{
		printf("pthread_attr_setstacksize()error\n");
		return ret;
	}
	ret= pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);//ChildThread quit and free source
	if (0 != ret)
	{
		printf("pthread_attr_setdetachstate()error\n");
		return ret;
	}
	pram=malloc(sizeof(struct socket));
	if(NULL==pram)
	{
		printf("pthread malloc()error\n");
		return -1;
	}
	
    memcpy(pram,psock,sizeof(struct socket));
	ret = pthread_create(&pthreadid,&attr,PostRoutine,pram);
	eno = errno;
	if (0 != ret) 
	{
		free(pram);
		printf("pthread_create() errno:%d", eno); 
		return ret;
	}
	pthread_attr_destroy(&attr);
	return 0;
}

void demoXX_httplib_accept_new_connection( const struct socket *listener, struct demolh_ctx_t*ctx ) {

	struct socket so;
	//char src_addr[IP_ADDR_STR_LEN];
	char error_string[ERROR_STRING_LEN];
	socklen_t len;
	int on;

	if ( listener == NULL ) return;

	on  = 1;
	len = sizeof(so.rsa);

	so.sock = accept( listener->sock, &so.rsa.sa, &len );
	if ( so.sock == INVALID_SOCKET ) return;
		/*
		 * Put so socket structure into the queue
		 */
	//原来码此处 检验IP是否符合掩码要求
		XX_httplib_set_close_on_exec( so.sock );


		if ( getsockname( so.sock, &so.lsa.sa, &len ) != 0 ) {

			httplib_cry( LH_DEBUG_ERROR, ctx, NULL, "%s: getsockname() failed: %s", __func__, httplib_error_string( ERRNO, error_string, ERROR_STRING_LEN ) );
		}

	//原代码此处 开启TCP socket的 保活、 TCP Nagle's algorithm、收发超时设置 --此处删掉
	if(pthreadFormPost(&so)!=0)
	printf("pthreadFormPost error\n");

}  /* XX_httplib_accept_new_connection */
 void demomaster_thread_run(void*thread_func_param) 
{
	int i;
	struct demolh_ctx_t*ctx = (struct demolh_ctx_t*)thread_func_param;
	struct pollfd *pfd;
		pfd = ctx->listening_socket_fds;

	while (1) {

		for (i=0; i<(int)ctx->num_listening_sockets; i++) {

			pfd[i].fd     = ctx->listening_sockets[i].sock;
			pfd[i].events = POLLIN;
		}

		if ( httplib_poll( pfd, ctx->num_listening_sockets, 200 ) > 0 ) {

			for (i=0; i<(int)ctx->num_listening_sockets; i++) {

				/*
				 * NOTE(lsm): on QNX, poll() returns POLLRDNORM after the
				 * successful poll, and POLLIN is defined as
				 * (POLLRDNORM | POLLRDBAND)
				 * Therefore, we're checking pfd[i].revents & POLLIN, not
				 * pfd[i].revents == POLLIN.
				 */

				if ((pfd[i].revents & POLLIN)) 
				{	
					demoXX_httplib_accept_new_connection( & ctx->listening_sockets[i], ctx ); 
				}
			}
		}
	}
	
}
int  main()
{
	struct demolh_ctx_t ctx;
	memset(&ctx,0,sizeof(struct demolh_ctx_t));
	ctx.listening_ports="+8080";//sever  ipv4+ipv6  可监听多种端口
	  if(0==demoXX_httplib_set_ports_option(&ctx))//socket-bind->listen   
	  {
		  printf("demoXX_httplib_set_ports_option error\n");
		  exit(1);
	  }
	  
      demomaster_thread_run(&ctx);//poll --accept--
	 
	  /*
	 * Stop signal received: somebody called httplib_stop. Quit.
	 */
	XX_httplib_close_all_listening_sockets( &ctx );
		return 0;
}        
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值