模拟服务器

开发平台：
C/C++

zbuffer.cpp：源码内容
							#include <stdafx.h>
#include "zbuffer.h"
#ifdef ZBUFFER_H
#include <string.h>
#include <stdlib.h>
ZPerf::ZPerf() {
	packet_number = remain_number = 0;
	total_size = remain_size = 0;
	max_size = min_size = -1;
	max_interval = min_interval = -1;
}
void ZPerf::start() {
#ifdef WIN32
	tick_count = ZTimer::now();
#endif
}
void ZPerf::stop() {
#ifdef WIN32
	tick_count = ZTimer::now() - tick_count;
#endif
}
void ZPerf::perfPacket(int size, connection_info *connection) {					//发送或者接收一个数据包
	if(size > max_size || max_size == -1)
		max_size = size;
	if(size < min_size || min_size == -1)
		min_size = size;
	total_size += size;
	
#ifdef WIN32
	unsigned long interval = ZTimer::now();
	if(connection->packet_time == 0)
		connection->packet_time = interval;
	else {
		interval -= connection->packet_time;
		if(interval > max_interval || max_interval == -1)
			max_interval = (short)interval;
		if(interval < min_interval || min_interval == -1)
			min_interval = (short)interval;
		connection->packet_time += interval;
	}
#endif
}
void ZPerf::useBuffer(int size, connection_info *connection)
{
	remain_size += size;
	remain_number++;
	connection->used_bufsize += size;
}
void ZPerf::freeBuffer(int size, connection_info *connection)
{
	remain_size -= size;
	remain_number--;
	connection->used_bufsize -= size;
}
bool ZBuffer::getNode(int index, int size) {
	int data_size = size;
	if (size >= 64 * 1024)
	{
		printf("too large blockn");
		return false;
	}
	bool bDebug = false;
	size += sizeof(packet_info);
	unsigned long align = size % MINIMIZE_BLOCK_SIZE;
	if(align)
		size += MINIMIZE_BLOCK_SIZE - align;
	if(size > buffer_size)
	{
		printf("size too large... <%08X> %d > %dn", this, size, buffer_size);
		return false;
	}
	packet_info *next;
	int nDeadLock = 0;
	while(free_packet->size < size) {										//删除后面的项目直到有足够的空间为止
		if(bDebug) {
			printf("free %d %dn", (char *)free_packet - buffer, free_packet->size);
		}
		next = (packet_info *)((char *)free_packet + free_packet->size);	//下一个节点
		if ((char *)next - buffer + sizeof(packet_info) >= buffer_size) {							//已经到了结束的位置，回第一个
			free_packet = (packet_info *)buffer;
			if (free_packet->state != STATE_FREE)
			{
				printf("free_packet at head not freen");
				while (free_packet->state != STATE_FREE)
					free_packet = (packet_info *)((char *)free_packet + free_packet->size);
			}
			printf("buffer <%08X> recycle...n", this);
			printf("free_packet %d size = %dn", (char *)free_packet - buffer, free_packet->size);
			continue;
		}
		else {
			if (next->state != STATE_FREE) {									//需要删除最老的节点
				printf("node alloc failed....request size = %d.n", size);
				printf("skip for free node...n");
				while (next->state != STATE_FREE && nDeadLock < 1024)
				{
					next = (packet_info *)((char *)next + next->size);
					if ((char *)next - buffer + sizeof(packet_info) >= buffer_size) {							//已经到了结束的位置，回第一个
						next = (packet_info *)buffer;
						printf("abnormal buffer <%08X> recycle...n", this);
					}
					nDeadLock++;
				}
				if(nDeadLock == 1024) {
					printf("Dead lockn");
				}
				if (next->state == STATE_FREE)
				{
					free_packet = next;
					printf("skip for free node success %d...n", (char *)free_packet - buffer);
					bDebug = true;
					continue;
				}
/*				printf("force recoveryn");
				int i;
				for (i = 0; i < max_connection; i++)
				{
					if (connections[i].state != CONNECTION_FREE &&
						connections[i].head_offset == (char*)next - buffer)
					{
						packet_info *packet = (packet_info *)(buffer + connections[i].head_offset);
						if (packet->next == -1)
							connections[i].head_offset = connections[i].tail_offset = -1;
						else
							connections[i].head_offset = packet->next;
						next->state = STATE_FREE;
						break;
					}
				}
				if (i >= max_connection)*/
				{
					printf("skip for free node failed...n");
					printf("node pos %d | %dn", (char*)free_packet - buffer, (char*)next - buffer);
					printf("status: total->%d | send-> %d / %d| recv-> %d / %dn", buffer_size, send_perf.remain_size, send_perf.remain_number, receive_perf.remain_size, receive_perf.remain_number);
/*					for (int i = 0; i < max_connection; i++)
					{
						if (connections[i].state != CONNECTION_FREE && connections[i].used_bufsize)
						{
							printf("    [%d] connection use %dn", i, connections[i].used_bufsize);
						}
					}*/
					return false;
				}
			}
			free_packet->size += next->size;
			free_packet->next = -1;
		}
	}
	next = (packet_info *)((char *)free_packet + size);
	if ((char *)next - buffer + sizeof(packet_info) >= buffer_size)
	{
		printf("buffer <%08X> recycle...n", this);
		next = (packet_info *)buffer;
		nDeadLock = 0;
		if (next->state != STATE_FREE)
		{
			printf("recycle failed....n");
			printf("skip for free node...n");
			while (next->state != STATE_FREE && nDeadLock < 1024)
			{
				next = (packet_info *)((char *)next + next->size);
				if ((char *)next - buffer + sizeof(packet_info) >= buffer_size) {							//已经到了结束的位置，回第一个
					next = (packet_info *)buffer;
					printf("abnormal buffer <%08X> recycle...n", this);
				}
				nDeadLock++;
			}
		if(nDeadLock == 1024) {
				printf("Dead lockn");
			}
			if (next->state != STATE_FREE)
			{
				printf("skip for free node failed...n");
				return false;
			}
			else {
				printf("skip for free node success...n");
			}
		}
	}
	else if (free_packet->size > size) {										//切分当前空闲块为占用和非占用两块
		next->size = free_packet->size - size;
		next->data_size = 0;
		next->current_size = 0;
		next->state = STATE_FREE;
		next->next = -1;
	}
	free_packet->size = size;
	free_packet->data_size = data_size;
	free_packet->current_size = 0;
	free_packet->next = -1;
	free_packet->state = STATE_IDLE;
	connection_info *current = &connections[index];
	if(current->head_offset == -1) {										//没有数据
		current->head_offset = current->tail_offset = (char *)free_packet - buffer;
	}
	else {
		packet_info *tail = (packet_info *)(buffer + current->tail_offset);
		tail->next = (char *)free_packet - buffer;
		current->tail_offset = tail->next;
	}
	free_packet = next;														//重新设置空闲块
	return true;
}
ZBuffer::ZBuffer(long size, int number) {
	printf("alloc..ZBuffer<%08X>..", this);
	buffer_size = max_connection = 0;
	connections = new connection_info[number];
	if(!connections)
		return;
	max_connection = number;
	size += sizeof(packet_info);
	unsigned long align = size % MINIMIZE_BLOCK_SIZE;
	if(align)
		size += MINIMIZE_BLOCK_SIZE - align;
	buffer = new char[size];
	if(!buffer) {
		delete[] connections;
		return;
	}
	buffer_size = size;
	printf("size = %dn", buffer_size);
	free_packet = (packet_info *)buffer;
	free_packet->size = buffer_size;
	free_packet->data_size = 0;
	free_packet->current_size = 0;
	free_packet->next = -1;
	bPerf = false;
	recv_buffers = new char*[max_connection];
	memset(recv_buffers, 0, sizeof(recv_buffers[0]) * max_connection);
	recv_buffer_size = new int[max_connection];
	memset(recv_buffer_size, 0, sizeof(recv_buffer_size[0]) * max_connection);
}
ZBuffer::~ZBuffer() {
	if(buffer_size) {
		delete[] buffer;
		delete[] connections;
	}
	if (recv_buffers)
	{
		for (int i = 0; i < max_connection; i++)
		{
			if (recv_buffers[i])
				delete recv_buffers[i];
		}
		delete recv_buffers;
	}
	if (recv_buffer_size)
		delete recv_buffer_size;
}
int ZBuffer::getConnection() {
	int index;
	for(index = 0; index < max_connection; index++) {
		if(connections[index].state == CONNECTION_FREE) {
			connections[index].state = CONNECTION_USED;
			return index;
		}
	}
	return -1;
}
long ZBuffer::getUsedBufferSize(int index)
{
	return connections[index].used_bufsize;
}
void ZBuffer::clear(int index, bool bSendOrRecv) {
	mutex.lock();
	connection_info *current = &connections[index];
	if(current->head_offset >= 0 ) {
		packet_info *packet = (packet_info *)(buffer + current->head_offset);
		while(packet) {
			packet->state = STATE_FREE;
			if (bPerf)
			{
				if (bSendOrRecv)
					send_perf.freeBuffer(packet->size, current);
				else
					receive_perf.freeBuffer(packet->size, current);
			}
			if(packet->next == -1) break;
			packet = (packet_info *)(buffer + packet->next);
		}
		current->head_offset = current->tail_offset = -1;
	}
	connections[index].reset();
	mutex.unlock();
}
bool ZBuffer::receiveData(int index, const char *data, int actsize) {				//接收到一块数据，可能不是正好在边界上，可能有几个数据包
	//接收到一块数据，可能不是正好在边界上，可能有几个数据包
	int size = actsize;
	mutex.lock();
	//printf("receiving buffer %d - %dn", index, actsize);
	connection_info *current = &connections[index];
	const char *data_ptr = data;
	packet_info *packet = NULL;
	unsigned short packet_size = 0xFFFF;
	while (size > 0) {
		if (current->tail_offset >= 0) {
			packet = (packet_info *)(buffer + current->tail_offset);
		}
		//正好收到末尾一个字节的边界处理 [wxb 2003-8-5]
		if (1 == size &&
			(packet && packet->current_size == packet->data_size))
		{
			//ASSERT(current->halfword == 0);
			current->halfword = (((unsigned short)0x8000) | (unsigned short)(*((unsigned char*)data_ptr)));
			//printf("Half word received..[%d]..tail<%02X>n", index, *((unsigned char*)data_ptr));
			break;
		}
		if (current->head_offset == -1 || 
			(packet && packet->current_size == packet->data_size)) {
			//没有数据包或者已经是完整的数据包了，需要分配
			if (current->halfword)
			{
				//处理上一个包尾的单个字节 [wxb 2003-8-5]
				packet_size = (unsigned short)((current->halfword & 0x7FFF) | ((unsigned short)(*((unsigned char*)data_ptr))) << 8) - sizeof(unsigned short);
				//printf("Half word received..[%d]..head<%02X>...processing<%02X>...yes perfect<%d>n", index, *((unsigned char*)data_ptr), (current->halfword & 0x7FFF), packet_size);
				current->halfword = 0;
				data_ptr += 1;
				size -= 1;
			}
			else
			{
				packet_size = *(unsigned short *)data_ptr - sizeof(unsigned short);
				data_ptr += sizeof(unsigned short);
				size -= sizeof(unsigned short);
			}
			//printf("packet size = %dn", packet_size);
			if (!getNode(index, packet_size)) {
 				mutex.unlock();
				printf("getNode fail while receiveData()n");
				exit(0);
				return false;
			}
			packet = (packet_info *)(buffer + current->tail_offset);	//重新获取尾部包的指针
			packet->state = STATE_RECEIVING;							//正在接收数据
			//得到一个新的数据包了,进行性能分析
			if (bPerf) {
				receive_perf.perfPacket(packet_size, current);
				receive_perf.packet_number++;
				receive_perf.useBuffer(packet->size, current);
			}
		}
		int copy_size = packet->data_size - packet->current_size;
		if(copy_size > size)
			copy_size = size;
		if (copy_size < 0)
		{
			mutex.unlock();
			printf("copy_size negative %08X, %d, %d, %d. <?> %d, %d | pos = %dn", this, copy_size, packet->data_size, packet->current_size, actsize, data_ptr - data, (char*)packet - buffer);
			return false;
		}
		memcpy((char *)packet + sizeof(packet_info) + packet->current_size, data_ptr, copy_size);
		size -= copy_size;
		data_ptr += copy_size;
		packet->current_size += copy_size;
		if(packet->current_size == packet->data_size) {
			packet->state = STATE_WAITING;
		}
	}
	mutex.unlock();
	return true;
}
	
char *ZBuffer::getPacket(int index, int &size) {									//获取一个完整的数据包
	mutex.lock();
	connection_info *current = &connections[index];
	if(current->head_offset == -1) {
		mutex.unlock();
		return NULL;
	}
	// flying comment here, just to make sure that this argument
	// is correct.
	if (index == -1)
	{
		printf("call getPacket with illegal index!n");
		mutex.unlock();
		return NULL;
	}
	packet_info *packet = (packet_info *)(buffer + current->head_offset);
	char *result = (char *)packet + sizeof(packet_info);
	if(current->head_offset == current->tail_offset) {					//只有一个数据包
		if(packet->current_size != packet->data_size) {
			mutex.unlock();
			return NULL;		//不完整
		}
		current->head_offset = current->tail_offset = -1;
	}
	else current->head_offset = packet->next;
	size = packet->data_size;
	packet->current_size = 0;
	packet->data_size = 0;
	packet->state = STATE_FREE;
	if(bPerf)
	{
		receive_perf.freeBuffer(packet->size, current);
	}
	result = getRecvBuffer(index, size);
	memcpy(result, (char *)packet + sizeof(packet_info), size);
	mutex.unlock();
	return result;
}
//发送数据的接口分成两个函数，一个是获取当前发送的缓冲区和长度，另外一个是从缓冲区清除指定大小的数据
//还有就是同时发送很多数据包的功能。
char *ZBuffer::getSendData(int index, int &size) {
	connection_info *current = &connections[index];
	if(current->head_offset == -1) return NULL;					//没有数据
	mutex.lock();
	char *result = NULL;
	packet_info *packet = (packet_info *)(buffer + current->head_offset);
	if(packet->state == STATE_SENDING) {
		size = packet->data_size - packet->current_size;
		if(size == 0) {
			result = NULL;
		}
		else {
			result = (char *)packet + sizeof(packet_info) + packet->current_size;
		}
	}
	mutex.unlock();
	return result;
}
void ZBuffer::sendData(int index, int size) {
	//一定是获取发送缓冲区成功之后才调用，可以省掉一些错误处理
	connection_info *current = &connections[index];
	mutex.lock();
	packet_info *packet = (packet_info *)(buffer + current->head_offset);
	if(packet->state == STATE_SENDING) {
		packet->current_size += size;
		if(packet->current_size == packet->data_size) {			//一个数据包发送完成
			if(bPerf) {
				send_perf.freeBuffer(packet->size, current);
				send_perf.perfPacket(packet->data_size, current);	//统计性能	
			}
			packet->state = STATE_FREE;
			current->head_offset = packet->next;				
			if(current->head_offset < 0) 
				current->tail_offset = -1;
		}
	}
	//printf("data sent: %d - %dn", index, size);
	mutex.unlock();
}
bool ZBuffer::packData(int index, const char *data, int size) {		//发送一个数据包的数据
	mutex.lock();
	connection_info *current = &connections[index];
	packet_info *packet = NULL;
	if(current->tail_offset >= 0 ) packet = (packet_info *)(buffer + current->tail_offset);
	if(!packet || packet->state != STATE_PACKING) {
//记录当前尾部的偏移量
		if(!getNode(index, MAX_PACKET_SIZE)) {
			mutex.unlock();
			printf("getNode fail while packData()n");
			exit(0);
			return false;
		}
		packet = (packet_info *)(buffer + current->tail_offset);
		packet->state = STATE_PACKING;
		packet->current_size += sizeof(unsigned short);
		if (bPerf)
		{
			send_perf.useBuffer(packet->size, current);
		}
	}
	if(size + packet->current_size + sizeof(packet_info) > packet->size) {		//超出了一个块的大小
		//printf("packing size a bit large...%08X, %d,%d,%dn", this, size,packet->current_size,packet->size);
		packet_info *old_packet = packet;
		packet_info *previous = (packet_info *)(buffer + current->head_offset);
		unsigned long old_tail = current->tail_offset;
		while(previous) {
			if(current->tail_offset == (char *)previous - buffer || previous->next == (char *)old_packet - buffer) {							//找到尾巴了
				if(current->tail_offset == (char *)previous - buffer)
					current->head_offset = current->tail_offset = -1;
				else
					current->tail_offset = (char *)previous - buffer;
				if(!getNode(index, old_packet->size + MAX_PACKET_SIZE)) {				//增加大小
					mutex.unlock();
					printf("getNode fail while packData() 2n");
					exit(0);
					return false;
				}
				old_packet->state = STATE_FREE;
				packet = (packet_info *)(buffer + current->tail_offset);
				if (bPerf)
				{
					send_perf.useBuffer(packet->size, current);
					send_perf.freeBuffer(old_packet->size, current);
				}
				memcpy((char *)packet + sizeof(packet_info),
					(char *)old_packet + sizeof(packet_info), old_packet->current_size);
				packet->current_size = old_packet->current_size;
				packet->state = STATE_PACKING;
				break;
			}
			if(previous->next == -1) {
				mutex.unlock();
				return false;
			}
			previous = (packet_info *)(buffer + previous->next);
		}
	}
	char *data_ptr = (char *)packet + sizeof(packet_info) + packet->current_size;
	memcpy(data_ptr, data, size);
	packet->current_size += size;
	mutex.unlock();
	return true;
}
char *ZBuffer::sendPacket(int index, int &datalength) {
	char *result = NULL;
	mutex.lock();
	connection_info *current = &connections[index];
	if(current->tail_offset >= 0) {
		packet_info *packet = (packet_info *)(buffer + current->tail_offset);
		if(packet->current_size > 0 && packet->state != STATE_SENDING) {
			packet->data_size = packet->current_size;
			datalength = packet->current_size - sizeof(unsigned short);
			result = (char *)packet + sizeof(packet_info) + sizeof(unsigned short);
			*(unsigned short *)((char *)packet + sizeof(packet_info)) = packet->data_size;
			packet->state = STATE_SENDING;
			packet->current_size = 0;
			if(bPerf) {
				send_perf.packet_number++;
			}
		}
	}
	mutex.unlock();
	//	if (datalength > 0)
	//printf("prepare to send %d - %dn", index, datalength);
	return result;
}
#endif

						