学步园

包括TCP 和UDP 以及客户端的 简单socket通讯实现。

#ifndef SERVERSOCKET__H<br /> #define SERVERSOCKET__H </p> <p>#include <sys/types.h><br /> #include <sys/socket.h><br /> #include <netinet/in.h><br /> #include <unistd.h><br /> #include <sys/uio.h><br /> #include <sys/socket.h><br /> #include <arpa/inet.h><br /> #include <iostream><br /> #include <string.h><br /> #include <stdlib.h><br /> #include <sys/wait.h><br /> #include <cstdio><br /> #include <fcntl.h><br /> #include <ext/stdio_filebuf.h><br /> #include <netdb.h><br /> #include <wait.h><br /> using namespace std;<br /> class ServerSocket<br /> {<br /> protected:<br /> int sin_size;<br /> int sockfd,new_fd;<br /> unsigned short port;<br /> struct sockaddr_in my_addr,their_addr,get_addr; </p> <p> public:<br /> ServerSocket();<br /> virtual ~ServerSocket();<br /> virtual int Socket();<br /> void SetPort(unsigned short in_port);<br /> bool Bind();<br /> void SetRemotePort(unsigned short in_port);<br /> unsigned short GetRemotePort();<br /> // void GetRemoteAddrees(char *addrees);<br /> bool Connect();<br /> void Listen();<br /> void Accept();<br /> int Send(const char *buf,int len);<br /> int Sendto(const char *buf,int len);<br /> int Recv(char *buf,int len);<br /> int Recvfrom(char *buf,int len);<br /> void Close(); </p> <p>};<br /> class TCPserver:public ServerSocket<br /> {<br /> public:<br /> int Socket();<br /> };<br /> class UDPserver:public ServerSocket<br /> {<br /> public:<br /> int Socket();<br /> };<br /> class UDPclient:public UDPserver<br /> {<br /> public:<br /> int SetServer(char* server_addr, unsigned short in_port);<br /> };<br /> #endif<br />

函数实现：

#include "server.h"<br /> ServerSocket::ServerSocket(){ </p> <p>}<br /> ServerSocket::~ServerSocket(){ </p> <p> Close(); </p> <p> }<br /> int UDPserver::Socket(){<br /> sockfd = socket(AF_INET ,SOCK_DGRAM, 0);<br /> return sockfd; </p> <p> }<br /> int TCPserver::Socket(){ </p> <p> sockfd = socket(AF_INET,SOCK_STREAM, 0);<br /> return sockfd; </p> <p> }<br /> int ServerSocket::Socket(){<br /> int i;<br /> i= sockfd = socket(AF_INET,SOCK_STREAM, 0); </p> <p> return sockfd; </p> <p> } </p> <p> void ServerSocket::SetPort(unsigned short in_port){<br /> struct hostent *host;<br /> host=gethostbyname("192.168.1.135");<br /> my_addr.sin_family = AF_INET;<br /> port = in_port;<br /> my_addr.sin_port = htons(port);<br /> my_addr.sin_addr = *((struct in_addr *)host->h_addr);<br /> // my_addr.sin_addr.s_addr=INADDR_ANY;<br /> bzero(&(my_addr.sin_zero),8);</p> <p> }<br /> void ServerSocket::SetRemotePort(unsigned short in_port){ </p> <p> their_addr.sin_port=htons(in_port); </p> <p> } </p> <p> unsigned short ServerSocket::GetRemotePort(){ </p> <p> return ntohs(get_addr.sin_port); </p> <p> } </p> <p> bool ServerSocket::Bind(){<br /> cout<<my_addr.sin_addr.s_addr<<endl;<br /> if(bind(sockfd,(struct sockaddr *)&my_addr,sizeof(struct sockaddr))==-1) {<br /> perror("bind err/n");<br /> return false;<br /> }<br /> return true; </p> <p> } </p> <p> int ServerSocket::Recvfrom(char *buf,int len){<br /> char str[24] ;<br /> int numbytes;<br /> socklen_t addr_len = sizeof(their_addr);<br /> numbytes = recvfrom(sockfd,buf,len,0,(struct sockaddr *)&their_addr,&addr_len);<br /> printf("you ip is %s at port %d:%s/n",<br /> inet_ntop(AF_INET, &their_addr.sin_addr,str,24),<br /> ntohs(their_addr.sin_port),buf);<br /> cout<<strlen(buf)<<endl;<br /> if(numbytes == -1)<br /> {<br /> return false;<br /> }<br /> buf[numbytes] = '/0'; </p> <p> return numbytes; </p> <p> }<br /> int ServerSocket::Sendto(const char *buf,int len){ </p> <p> return sendto(sockfd,buf,len,0,(struct sockaddr *)&their_addr,sizeof(struct sockaddr)); </p> <p> } </p> <p> void ServerSocket::Close(){ </p> <p> close(sockfd);<br /> close(new_fd); </p> <p> } </p> <p> void ServerSocket::Listen(){ </p> <p> if(listen(sockfd,10)==-1)<br /> cout<<"listen err!/n";</p> <p> } </p> <p>void ServerSocket::Accept(){ </p> <p> socklen_t size = sizeof(struct sockaddr_in);<br /> new_fd = accept(sockfd,(struct sockaddr *)&get_addr,&size);<br /> if(new_fd==-1){<br /> perror("accept err!/n");<br /> exit(1);<br /> }<br /> } </p> <p> int ServerSocket::Recv(char *buf,int len){<br /> return recv(new_fd,buf,len,0); </p> <p> } </p> <p> int ServerSocket::Send(const char * buf, int len){ </p> <p> return send(new_fd,buf,len,0); </p> <p> } </p> <p> bool ServerSocket::Connect(){ </p> <p> if(connect(sockfd,(struct sockaddr *)&their_addr, sizeof(struct sockaddr))==-1)<br /> return false; </p> <p> return true; </p> <p> }<br /> int UDPclient::SetServer(char* server_addr, unsigned short in_port)<br /> {<br /> struct hostent *host;<br /> host=gethostbyname(server_addr);<br /> their_addr.sin_family = AF_INET;<br /> port = in_port;<br /> their_addr.sin_port = htons(port);<br /> their_addr.sin_addr = *((struct in_addr *)host->h_addr);<br /> bzero(&(their_addr.sin_zero),8);<br /> return 0;<br />