重写socket类

     记得自己以前写了个socket类，自己也一直在用，但是这个类一直有个bug，就是accept后的socket没办法获得远程端的ip和port，所以自己改了一下，把代码发了上来.

mysocket.h:

// mysocket.h: interface for the mysocket class. // // #if !defined(AFX_MYSOCKET_H__C142FE73_FC40_4235_8C9D_5E7493376017__INCLUDED_) #define AFX_MYSOCKET_H__C142FE73_FC40_4235_8C9D_5E7493376017__INCLUDED_ #if _MSC_VER > 1000 #pragma once #endif // _MSC_VER > 1000 #define WM_SOCKET WM_USER+1 class mysocket { public: mysocket(); virtual ~mysocket(); void init(int type=0,int protocol=1); int send(char *buf,int len,int flags=0); int recv(char *buf,int len,int flags=0); int bind(int port); int listen(int max=5); void accept(mysocket& s1); int connect(char *addr1,int port); void attach(SOCKET s1); SOCKET getsocket(); void asyncselect(HWND d,int m=FD_ACCEPT|FD_READ|FD_WRITE|FD_CONNECT|FD_CLOSE); int getport(); char *getaddr(); void close(); int sendto(char *buf,int len,char *addr,int port,int flags=0); int recvfrom(char *buf,int len,int flags=0); char* getpeer(); int getpeerport(); protected: sockaddr_in addr; SOCKET s; }; #endif // !defined(AFX_MYSOCKET_H__C142FE73_FC40_4235_8C9D_5E7493376017__INCLUDED_)

 

 mysocket.cpp

// mysocket.cpp: implementation of the mysocket class. // // #include<WINSOCK2.H> #include "mysocket.h" #pragma comment(lib,"ws2_32"); #ifdef _DEBUG #undef THIS_FILE static char THIS_FILE[]=__FILE__; #define new DEBUG_NEW #endif // // Construction/Destruction // mysocket::mysocket() { } mysocket::~mysocket() { } void mysocket::init(int type,int protocol){ if(type==0){ s=::socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); }else if(type==1){ s=::socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP); }else{ s=::socket(AF_INET,SOCK_RAW,protocol); } } int mysocket::send(char *buf,int len,int flags){ return ::send(s,buf,len,flags); } int mysocket::recv(char *buf,int len,int flags){ return ::recv(s,buf,len,flags); } int mysocket::bind(int port){ addr.sin_family=AF_INET; addr.sin_port=::htons(port); addr.sin_addr.S_un.S_addr=INADDR_ANY; return ::bind(s,(LPSOCKADDR)&addr,sizeof(addr)); } int mysocket::listen(int max){ return ::listen(s,max); } void mysocket::accept(mysocket& s1){ int n=sizeof(s1.addr); s1.s=::accept(s,(SOCKADDR*)&s1.addr,&n); } int mysocket::connect(char *addr1,int port){ addr.sin_family=AF_INET; addr.sin_port=::htons(port); addr.sin_addr.S_un.S_addr=::inet_addr(addr1); return ::connect(s,(sockaddr*)&addr,sizeof(addr)); } void mysocket::attach(SOCKET s1){ s=s1; } SOCKET mysocket::getsocket(){ return s; } void mysocket::asyncselect(HWND d,int m){ ::WSAAsyncSelect(s,d,WM_SOCKET,m); } int mysocket::getport(){ return ::ntohl(addr.sin_port); } char *mysocket::getaddr(){ return ::inet_ntoa(addr.sin_addr); } void mysocket::close(){ ::closesocket(s); } int mysocket::sendto(char *buf,int len,char *addr1,int port,int flags){ addr.sin_family=AF_INET; addr.sin_port=::htons(port); addr.sin_addr.S_un.S_addr=::inet_addr(addr1); int l=::sendto(s,buf,len,flags,(sockaddr*)&addr,sizeof(addr)); return l; } int mysocket::recvfrom(char *buf,int len,int flags){ sockaddr_in addr1; int n=sizeof(addr1); int l=::recvfrom(s,buf,len,flags,(sockaddr*)&addr,&n); addr=addr1; return l; } char* mysocket::getpeer(){ sockaddr_in addr; int n=sizeof(addr); ::getpeername(getsocket(),(sockaddr*)&addr,&n); // int peerport=::ntohs(addr.sin_port); return ::inet_ntoa(addr.sin_addr); } int mysocket::getpeerport(){ sockaddr_in addr; int n=sizeof(addr); ::getpeername(getsocket(),(sockaddr*)&addr,&n); return ::ntohs(addr.sin_port); }

使用方法和以前那个类一样，可以参考下。

       自己最近一直在研究网络编程和多线程应用，感觉这两块是目前最热门也最实用的地方，所以自己学习有什么心得，会和大家一起分享的。

       本文有什么不足之处，还望大家多多指正。