server.cpp

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#ifdef __MINGW32__
#include <windows.h>
#include <ws2tcpip.h>
#include <winsock2.h>
#else
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#endif
#include <pthread.h>

#include "isleep.h"
#include "ikcp.c"

#define SERVER_PORT 8888
#define BUFF_LEN    1024

int server_fd;
ikcpcb *kcp2;

void * handle_udp_msg(void *argp)
{
	int fd = (int) argp;
	socklen_t len;
	int hr;
	char buffer[BUFF_LEN];
	struct sockaddr_in clent_addr;

	while (1) {
		memset(buffer, 0, BUFF_LEN);
		hr = recvfrom(fd, buffer, BUFF_LEN, 0, (struct sockaddr*)&clent_addr, &len);

		if (hr == -1) {
			//printf("recieve data fail!\n");
			continue;
		}

		//printf("ikcp_input: hr: %d\n", hr);
		// 如果 p2收到udp，则作为下层协议输入到kcp2
		ikcp_input(kcp2, buffer, hr);
	}
}

void * handle_update(void *argp)
{
	while (1) {
		isleep(1);
		ikcp_update(kcp2, iclock());
	}
}

// 模拟网络：模拟发送一个 udp包
int udp_output(const char *buf, int len, ikcpcb *kcp, void *user)
{
	//union { int id; void *ptr; } parameter;
	//parameter.ptr = user;

	struct sockaddr_in ser_addr;
	memset(&ser_addr, 0, sizeof(ser_addr));
	ser_addr.sin_family      = AF_INET;
	ser_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
	ser_addr.sin_port        = htons(8889);

	//printf("%s call udp sendto\n", __func__);
	sendto(server_fd, buf, len, 0, (struct sockaddr*)&ser_addr, sizeof(struct sockaddr_in));

	return 0;
}

int main(int argc, char* argv[])
{
	int ret;
	struct sockaddr_in ser_addr;

#ifdef __MINGW32__
	WORD sockVersion = MAKEWORD(2,2);
	WSADATA data;

	if (WSAStartup(sockVersion, &data) != 0) {
		return -1;
	}
#endif

	server_fd = socket(AF_INET, SOCK_DGRAM, 0);
	if (server_fd < 0) {
		printf("create socket fail!\n");
		return -1;
	}

	memset(&ser_addr, 0, sizeof(ser_addr));
	ser_addr.sin_family      = AF_INET;
	ser_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	ser_addr.sin_port        = htons(8888);

	ret = bind(server_fd, (struct sockaddr*)&ser_addr, sizeof(ser_addr));
	if (ret < 0) {
		printf("socket bind fail!\n");
		return -1;
	}

	// 创建两个端点的 kcp对象，第一个参数 conv是会话编号，同一个会话需要相同
	// 最后一个是 user参数，用来传递标识
	kcp2 = ikcp_create(0x11223344, (void*)1);

	// 设置kcp的下层输出，这里为 udp_output，模拟udp网络输出函数
	kcp2->output = udp_output;

	// 配置窗口大小：平均延迟200ms，每20ms发送一个包，
	// 而考虑到丢包重发，设置最大收发窗口为128
	ikcp_wndsize(kcp2, 2048, 2048);

	// 启动快速模式
	// 第二个参数 nodelay-启用以后若干常规加速将启动
	// 第三个参数 interval为内部处理时钟，默认设置为 10ms
	// 第四个参数 resend为快速重传指标，设置为2
	// 第五个参数 为是否禁用常规流控，这里禁止
	ikcp_nodelay(kcp2, 1, 10, 2, 1);

	//handle_udp_msg(server_fd);
	pthread_t recvdata_id;
	pthread_t update_id;
	pthread_create(&recvdata_id, NULL, handle_udp_msg, (void*)server_fd);
	pthread_create(&update_id, NULL, handle_update, NULL);

	IUINT32 current = iclock();
	IUINT32 slap = current + 20;
	IUINT32 index = 0;
	IUINT32 next = 0;
	IINT64 sumrtt = 0;
	int count = 0;
	int maxrtt = 0;

	char buffer[2000];
	int hr;

	IUINT32 ts1 = iclock();

	while (1) {
#if 0
		hr = ikcp_recv(kcp2, buffer, 10);
		// 没有收到包就退出
		if (hr < 0) continue;
		// 如果收到包就打印
		IUINT32 sn = *(IUINT32*)(buffer + 0);
		IUINT32 ts = *(IUINT32*)(buffer + 4);

		static int recv_index = 0;
		recv_index++;
		if (recv_index % 100 == 0) {
			printf("[ikcp_recv]: sn: %d ts: %d\n", sn, ts);
		}
#endif
#if 1
		isleep(1);
		// kcp2接收到任何包都返回回去
		while (1) {
			hr = ikcp_recv(kcp2, buffer, 10);
			// 没有收到包就退出
			if (hr < 0) break;
			// 如果收到包就回射
			ikcp_send(kcp2, buffer, hr);
		}
#endif
	}

	ikcp_release(kcp2);

	pthread_join(recvdata_id, NULL);
	pthread_join(update_id, NULL);

#ifdef __MINGW32__
	closesocket(server_fd);
	WSACleanup();
#else
	close(server_fd);
#endif

	return 0;
}