#include "network.h"
#include <unistd.h>

#include <stdio.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <stdlib.h>
//Tcp 专用
NetWork* accept_network(NetWork* nw)
{
	if(SOCK_STREAM != nw->type)
	{
		printf("类型错误\n");
		return NULL;
	}
	NetWork* clinw =malloc(sizeof(NetWork));
	if(NULL == clinw)
	{
		perror("network accept malloc");
		return NULL;
	}
	clinw->type =nw->type;
	clinw->len =sizeof(clinw->addr);
	clinw->fd = accept(nw->fd,(SP)&clinw->addr,&clinw->len);
	if(0>clinw->fd)
	{
		perror("network accept");
		free(clinw);
		return NULL;

	}
	return clinw;
}
//创建网络连接
NetWork* open_network(char c_or_s,int type,char* ip,uint16_t port)
{
	NetWork* nw = malloc(sizeof(NetWork));
	if(NULL == nw)
	{
		perror("net malloc");
		return NULL;
	}
	nw->fd = socket(AF_INET,type,0);
	if(0>nw->fd)
	{
		perror("network socket");
		return NULL;
	}
	//准备通信地址
	nw->type =type;
	nw->addr.sin_family =AF_INET;
	nw->addr.sin_port = htons(port);
	nw->addr.sin_addr.s_addr = inet_addr(ip);
	nw->len=sizeof(nw->addr);
	
	if('s' == c_or_s)
	{
		if(bind(nw->fd,(SP)&nw->addr,nw->len))
		{
			perror("network bind");
			return NULL;
		}
		if(SOCK_STREAM == type)
		{
			if(listen(nw->fd,50))
			{
				perror("network listen");
				free(nw);
				return NULL;
			}
		}
	}
	else if(SOCK_STREAM ==type)
	{
		if(connect(nw->fd,(SP)&nw->addr,nw->len))
		{
			perror("network connect");
			return NULL;
		}
	}
	return nw;

}
//发送
int nsend(NetWork* nw,void* buf,uint32_t len)
{
	if(SOCK_STREAM == nw->type)
	{
		return send(nw->fd,buf,len,0);
	}
	else if(SOCK_DGRAM== nw->type)
	{
		return sendto(nw->fd,buf,len,0,(SP)&nw->addr,nw->len);
	}
	return -1;
}

//接收
int nrecv(NetWork* nw,void* buf,uint32_t len)
{
	
	if(SOCK_STREAM == nw->type)
	{
		return recv(nw->fd,buf,len,0);
	}
	else if(SOCK_DGRAM== nw->type)
	{
		return recvfrom(nw->fd,buf,len,0,(SP)&nw->addr,&nw->len);
	}
	return -1;
}
//关闭
void close_network(NetWork* nw)
{
	close(nw->fd);
	free(nw);
}