bfq

bfq

关注
发信
关注(28)粉丝(399)

实现一个跨平台的mysock库（windows、linux）



源码下载


1.首先确定自己的操作系统为32位还是64位：

root@bfq:~/mysock# uname -a

Linux bfq 3.11.0-26-generic#45~precise1-Ubuntu SMP Tue Jul 15 04:02:35 UTC 2014x86_64x86_64 x86_64 GNU/Linux

2.编写測试程序：

root@bfq:~/mysock# vim test.c

#include <stdio.h>

int main()

{

        printf("sizeof(char *)=%d\n",sizeof(char *));

        return 0;

}

编译、运行：

root@bfq:~/mysock# gcc test.c

root@bfq:~/mysock# file a.out

a.out: ELF64-bitLSB executable, x86-64, version 1 (SYSV), dynamically linked (uses sharedlibs), for GNU/Linux 2.6.24,BuildID[sha1]=0x69d02236045c81b597e24cb143cfca5909987b80,
not stripped

root@bfq:~/mysock# ./a.out

sizeof(char *)=8

root@bfq:~/mysock# gcc test.c-m32

root@bfq:~/mysock# file a.out

a.out: ELF32-bitLSB executable, Intel 80386, version 1 (SYSV), dynamically linked (uses sharedlibs), for GNU/Linux 2.6.24,BuildID[sha1]=0x668431a373c94667857a83c2d92929e8b57cff58,
not stripped

root@bfq:~/mysock# ./a.out

sizeof(char *)=4

使用-m32来使得gcc编译32位程序（在x86_64系统上），使用-m elf_i386參数可以使得64位的ld可以兼容32位的库

3.跨平台代码分析

编译：

root@bfq:~/mysock#ls

libmysock64.so makefile-lib32 makefile-lib64  makefile-win mysock.c  mysock.h

root@bfq:~/mysock#make -f makefile-lib32

gcc-Wall -D__UNIX -g -m32 -fPIC -o mysock.o -c mysock.c

gcc -m32 -shared -o libmysock32.so mysock.o

root@bfq:~/mysock#file libmysock32.so

libmysock32.so:ELF32-bit LSB shared object, Intel 80386,version 1 (SYSV), dynamically linked,BuildID[sha1]=0x18b8a6de746679cccfb10306d6bc62955e3566f4,
not stripped

root@bfq:~/mysock#

root@bfq:~/mysock#make -f makefile-lib64

gcc-Wall -D__UNIX -g -fPIC -o mysock.o -c mysock.c

gcc-shared -o libmysock64.so mysock.o

root@bfq:~/mysock#ls

libmysock64.so makefile-lib32 makefile-lib64  makefile-win mysock.c  mysock.h mysock.o

root@bfq:~/mysock#file libmysock64.so

libmysock64.so:ELF64-bit LSB shared object, x86-64, version 1(SYSV), dynamically linked,BuildID[sha1]=0xbec5736775c33f27fe81e88c98f1c26a1edf97b9,
not stripped

源代码结构：

linux

实现一个跨平台的mysock库（windows、linux）-LMLPHP

windows

实现一个跨平台的mysock库（windows、linux）-LMLPHP

各种平台的makefile

.SUFFIXES: .c .o

CC=gcc

SRCS=mysock.c

OBJS=$(SRCS:.c=.o)

EXEC=libmysock64.so

.c.o:

	$(CC) -Wall -D__UNIX -g -fPIC -o $@ -c $<

all: $(OBJS)

	$(CC) -shared -o $(EXEC) $(OBJS)

clean:

	rm -rf $(OBJS) $(EXEC)
.SUFFIXES: .c .o

CC=gcc

SRCS=mysock.c

OBJS=$(SRCS:.c=.o)

EXEC=libmysock32.so

.c.o:

	$(CC) -Wall -D__UNIX -g -m32 -fPIC -o $@ -c $<

all: $(OBJS)

	$(CC)  -m32 -shared -o $(EXEC) $(OBJS)

clean:

	rm -rf $(OBJS) $(EXEC)

利用vs开发工具本身能够编译出lib和dll。

实现一个跨平台的mysock库（windows、linux）-LMLPHP

实现一个跨平台的mysock库（windows、linux）-LMLPHP

mysock.h

#ifndef __MYSOCK_H

#define __MYSOCK_H

#ifdef __cplusplus

extern "C" {

#endif

	//linux下设置进程进入daemon状态。windows无效

	void set_daemon();

	//windows下调用的初始化socket环境。linux不须要

	void init_socket();

	//windows下调用的释放socket环境。linux不须要

	void free_socket();

	//将域名转化为IP，domain代表域名，返回值为struct in_addr的s_addr成员

	unsigned int domain2ip(const char *domain);

	//建立一个socket。

	//status = 1 is TCP, other is UDP。返回值>=0代表成功建立的socket句柄。-1失败

	int create_socket(int status);

	//关闭由create_socket创建的socket,

	//sock为create_socket函数返回的socket句柄

	void close_socket(int sock);

	//为socket绑定一个端口号

	//sock为create_socket函数返回的socket句柄。port为要绑定的端口号。返回值=0成功。-1失败

	int bind_socket(int sock, unsigned short port);

	//使用UDP协议接收数据

	//sock为create_socket函数返回的socket句柄，buf为接收数据缓冲区。 bufsize为缓冲区大小（单位：字节）, srcIP为数据来源IP地址。返回值>0成功接收数据字节数，=0连接正常关闭，-1失败

	int udp_recv(int sock, char *buf, size_t bufsize, char *srcIP);

	//使用UDP协议发送数据

	//sock为create_socket函数返回的socket句柄。buf为发送数据缓冲区。 bufsize为发送数据大小（单位：字节）。destIP为发送目的IP地址，port为发送目的端口号 。返回值>0成功发送数据字节数，=0连接正常关闭，-1失败

	int udp_send(int sock, const char *destIP, unsigned short port, const char *buf, size_t bufsize);

	//使用TCP协议接收数据之前须要先listen

	//sock为create_socket函数返回的socket句柄。返回值=0成功,-1失败

	int tcp_listen(int sock);

	//使用TCP协议接收来自client的链接

	//sock为create_socket函数返回的socket句柄， srcIP为来自client的IP地址。返回值>0代表来自client的socket句柄,-1失败

	int tcp_accept(int sock, char *srcIP);

	//使用TCP协议连接到指定的服务器

	//sock为create_socket函数返回的socket句柄，destIP为来自服务端端的IP地址。port为服务端的端口号。返回值=0成功,-1失败

	int tcp_connect(int sock, const char *destIP, unsigned short port);

	//使用TCP协议接收数据

	//sock为create_socket函数返回的socket句柄。buf为接收数据缓冲区。 bufsize为缓冲区大小（单位：字节）。返回值>0成功接收数据字节数。=0连接正常关闭，-1失败

	int tcp_recv(int sock, char *buf, size_t bufsize);

	//使用TCP协议发送数据

	//sock为create_socket函数返回的socket句柄，buf为发送数据缓冲区。 bufsize为发送数据大小哦（单位：字节）。返回值>0成功发送数据字节数，=0连接正常关闭，-1失败

	int tcp_send(int sock, const char *buf, size_t bufsize);

	////////////////////////////////下面为app函数//////////////////////////

	//使用UDP协议接收数据

	//buf为接收数据缓冲区， bufsize为缓冲区大小（单位：字节）, port为接收数据的端口号，srcIP为数据来源IP地址。返回值>0成功接收数据字节数。=0连接正常关闭。-1失败

	int app_udp_recv(char *buf, size_t bufsize, unsigned short port, char *srcIP);

	//使用UDP协议发送数据

	//buf为发送数据缓冲区。 bufsize为发送数据大小（单位：字节），destIP为发送目的IP地址，port为发送目的端口号 。返回值>0成功发送数据字节数，=0连接正常关闭。-1失败

	int app_udp_send(const char *destIP, unsigned short port, const char *buf, size_t bufsize);

	//使用TCP协议client连接到指定的服务端

	//destIP为来自服务端的IP地址，port为服务端的端口号。返回值=0成功,-1失败

	int app_client_connect(const char *destIP, unsigned short port);

	//断开clientTCP连接

	void app_client_close();

	//断开服务端TCP连接

	void app_server_close();

	//使用TCP协议client接收数据

	//buf为接收数据缓冲区。bufsize为缓冲区大小（单位：字节）。返回值>0成功接收数据字节数，=0连接正常关闭。-1失败

	int app_client_recv(char *buf, size_t bufsize);

	//使用TCP协议client发送数据

	//buf为发送数据缓冲区。 bufsize为发送数据大小哦（单位：字节）。

返回值>0成功发送数据字节数，=0连接正常关闭，-1失败

	int app_client_send(const char *buf, size_t bufsize);

	//使用TCP协议。建立服务端socket

	//參数port为服务端端口号，返回值=0成功,-1失败

	int app_server_create(unsigned short port);

	//使用TCP协议服务端接收数据

	//buf为接收数据缓冲区。bufsize为缓冲区大小（单位：字节）,srcIP为clientIP地址。

返回值>0成功接收数据字节数。=0连接正常关闭。-1失败

	int app_server_recv(char *buf, size_t bufsize, char *srcIP);

	//使用TCP协议服务端发送数据

	//buf为发送数据缓冲区， bufsize为发送数据大小哦（单位：字节）。返回值>0成功发送数据字节数。=0连接正常关闭，-1失败

	int app_server_send(const char *buf, size_t bufsize);

#ifdef __cplusplus

}

#endif

#endif

mysock.c

#include <string.h>

#include <stdlib.h>

#include <stdio.h>

#ifndef __UNIX

#include <WinSock2.h>

#pragma comment(lib, "Ws2_32.lib")

#pragma warning(disable:4996)

#else

#include <sys/types.h>

#include <sys/socket.h>

#include <unistd.h>

#include <arpa/inet.h>

#include <netinet/in.h>

#include <netdb.h>

#endif

//假设编译动态库。须要把#define STATIC_LIB凝视

#define STATIC_LIB

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

void set_daemon()

{

#ifdef __UNIX

	pid_t pid, sid;

	pid = fork();

	if (pid < 0)

	{

		exit(-1);

	}

	if (pid > 0)

	{

		exit(0);

	}

	if ((sid = setsid()) < 0)

	{

		exit(-1);

	}

#endif

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

void init_socket()

{

#ifndef __UNIX

	unsigned short ver;

	WSADATA wsaData;

	ver = MAKEWORD(1, 1);

	WSAStartup(ver, &wsaData);

#endif

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

void free_socket()

{

#ifndef __UNIX

	WSACleanup();

#endif

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

unsigned int domain2ip(const char *domain)

{

	struct hostent *iphost = gethostbyname(domain);

	if (iphost)

	{

		struct in_addr *p = (struct in_addr *)iphost->h_addr_list[0];

		if (p)

			return p->s_addr;

	}

	return 0;

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int create_socket(int status)

{

#ifndef __UNIX

	char on = 1;

#else

	int on = 1;

#endif

	int st = 0;

	if (status == 1)

	{

		st = socket(AF_INET, SOCK_STREAM, 0);//建立TCP socket

		setsockopt(st, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));//设置socket地址可重用

	}

	else

	{

		st = socket(AF_INET, SOCK_DGRAM, 0);//建立UDP socket

		setsockopt(st, SOL_SOCKET, SO_BROADCAST, &on, sizeof(on));//设置能够发送udp广播消息

	}

	return st;

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

void close_socket(int sock)

{

#ifndef __UNIX

	closesocket(sock);

#else

	close(sock);

#endif

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int bind_socket(int sock, unsigned short port)

{

	struct sockaddr_in recv_addr;

	memset(&recv_addr, 0, sizeof(recv_addr));

	recv_addr.sin_family = AF_INET;

	recv_addr.sin_port = htons(port);//指定port为要连接的端口号

	recv_addr.sin_addr.s_addr = htonl(INADDR_ANY);//指定接收端IP地址为随意

	return bind(sock, (struct sockaddr *) &recv_addr, sizeof(recv_addr));

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int udp_recv(int sock, char *buf, size_t bufsize, char *srcIP)

{

	struct sockaddr_in addr;

	memset(&addr, 0, sizeof(addr));

#ifndef __UNIX

	int len = sizeof(addr);

#else

	socklen_t len = sizeof(addr);

#endif

	unsigned int rc = recvfrom(sock, buf, bufsize, 0, (struct sockaddr *) &addr, &len);//接收到来自srcIP的消息

	//strcpy(srcIP, inet_ntoa(addr.sin_addr));

	unsigned char *c = (unsigned char *)&addr.sin_addr.s_addr;

	sprintf(srcIP, "%u.%u.%u.%u", c[0], c[1], c[2], c[3]);

	return rc;

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int udp_send(int sock, const char *destIP, unsigned short port, const char *buf, size_t bufsize)

{

	struct sockaddr_in addr;

	memset(&addr, 0, sizeof(addr));

	addr.sin_family = AF_INET;

	addr.sin_port = htons(port);//指定port为要连接的端口号

	//addr.sin_addr.s_addr = inet_addr(destIP);//指定IP为要连接的IP地址

	addr.sin_addr.s_addr = domain2ip(destIP);

	return sendto(sock, buf, bufsize, 0, (struct sockaddr *) &addr, sizeof(addr));//向指定IP发送消息

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int tcp_listen(int sock)

{

	return listen(sock, 20);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int tcp_accept(int sock, char *srcIP)

{

	struct sockaddr_in addr;

	memset(&addr, 0, sizeof(addr));

#ifndef __UNIX

	int len = sizeof(addr);

#else

	socklen_t len = sizeof(addr);

#endif

	int st = accept(sock, (struct sockaddr *) &addr, &len);

	//strcpy(srcIP, inet_ntoa(addr.sin_addr));

	unsigned char *c = (unsigned char *)&addr.sin_addr.s_addr;

	sprintf(srcIP, "%u.%u.%u.%u", c[0], c[1], c[2], c[3]);

	return st;

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int tcp_connect(int sock, const char *destIP, unsigned short port)

{

	struct sockaddr_in addr;

	memset(&addr, 0, sizeof(addr));

	addr.sin_family = AF_INET;

	addr.sin_port = htons(port);//指定port为要连接的端口号

	//addr.sin_addr.s_addr = inet_addr(destIP);//指定IP为要连接的IP地址

	addr.sin_addr.s_addr = domain2ip(destIP);

	return connect(sock, (struct sockaddr *) &addr, sizeof(addr));//连接到指定IP

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int tcp_recv(int sock, char *buf, size_t bufsize)

{

	return recv(sock, buf, bufsize, 0);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int tcp_send(int sock, const char *buf, size_t bufsize)

{

	return send(sock, buf, bufsize, 0);

}

////////////////////////////////下面为app函数//////////////////////////

static int is_init = 0;

static int udp_recv_sock = -1;

static int udp_send_sock = -1;

static int tcp_client_sock = -1;

static int tcp_server_sock = -1;

static int tcp_list_sock = -1;

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_udp_recv(char *buf, size_t bufsize, unsigned short port, char *srcIP)

{

	if (is_init == 0)

	{

		init_socket();

		is_init = 1;

	}

	if (udp_recv_sock != -1)

		close_socket(udp_recv_sock);

	udp_recv_sock = create_socket(0);

	int rc = bind_socket(udp_recv_sock, port);

	if (rc < 0)

		return rc;

	return udp_recv(udp_recv_sock, buf, bufsize, srcIP);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_udp_send(const char *destIP, unsigned short port, const char *buf, size_t bufsize)

{

	if (is_init == 0)

	{

		init_socket();

		is_init = 1;

	}

	if (udp_send_sock == -1)

	{

		udp_send_sock = create_socket(0);

	}

	return udp_send(udp_send_sock, destIP, port, buf, bufsize);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_client_connect(const char *destIP, unsigned short port)

{

	if (is_init == 0)

	{

		init_socket();

		is_init = 1;

	}

	if (tcp_client_sock != -1)

		close_socket(tcp_client_sock);

	tcp_client_sock = create_socket(1);

	return tcp_connect(tcp_client_sock, destIP, port);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

void app_client_close()

{

	if (tcp_client_sock != -1)

	{

		close_socket(tcp_client_sock);

		tcp_client_sock = -1;

	}

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

void app_server_close()

{

	if (tcp_server_sock != -1)

	{

		close_socket(tcp_server_sock);

		tcp_server_sock = -1;

	}

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_client_recv(char *buf, size_t bufsize)

{

	return tcp_recv(tcp_client_sock, buf, bufsize);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_client_send(const char *buf, size_t bufsize)

{

	return tcp_send(tcp_client_sock, buf, bufsize);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_server_create(unsigned short port)

{

	if (is_init == 0)

	{

		init_socket();

		is_init = 1;

	}

	if (tcp_list_sock != -1)

	{

		close_socket(tcp_list_sock);

	}

	tcp_list_sock = create_socket(1);

	int rc = bind_socket(tcp_list_sock, port);

	if (rc == -1)

		return rc;

	return tcp_listen(tcp_list_sock);

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_server_recv(char *buf, size_t bufsize, char *srcIP)

{

	static char IP[100];

	if (tcp_server_sock <= 0)

	{

		memset(IP, 0, sizeof(IP));

		tcp_server_sock = tcp_accept(tcp_list_sock, IP);

	}

	strcpy(srcIP, IP);

	int rc = tcp_recv(tcp_server_sock, buf, bufsize);

	if (rc <= 0)

		app_server_close();

	return rc;

}

#ifndef __UNIX

#ifndef STATIC_LIB

__declspec(dllexport)

#endif

#endif

int app_server_send(const char *buf, size_t bufsize)

{

	return tcp_send(tcp_server_sock, buf, bufsize);

}

05-01 22:44
Powered by LMLPHP ©2024 bfq 0.075561