tcp_socket.h

来自「封装了linux的tcp协议」· C头文件 代码 · 共 71 行

#ifndef _TCP_SOCKET_H
#define _TCP_SOCKET_H
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <iostream>
#include <string>
using namespace std;

class tcp_socket
{
	private:
	 
	  int conn;
	
	  int lisn;
       

          unsigned short port;
          string _req_file_name;
	  string _file_name;
	  struct sockaddr_in serv_addr;

	  
	public:
	  enum soc_type 
	  {
	  	SOC_CLIN =0,
	  	SOC_SERV
	  } soc_type;
	   
	  
	  tcp_socket();
	  
	  ~tcp_socket();
        
	  //function to deal with connection
	  int CreatServ(char* ipstr, char* portstr);
	  int TcpAccept();
	  int TcpRecv(int socfd, void* recv_buf, int size);
	  int TcpSend(int socfd, void* send_buf, int size);
	  int TcpSend(int socfd, string str);
	  void CloseLisn();
	  
	  int CreatClin(char* ipstr, char* portstr);
	  int TcpConn();
	  int TcpRecv(void* recv_buf, int size);
	  int TcpSend(void* send_buf, int size);
	  int TcpSend(string str);
	  void CloseClin();
	  
	  
	  //function to deal with file
	  void SetFile(char* req_file_name, char* file_name);
	  
	  int SendComd(int socfd);
	  int SendComd();
	  
	  int ReadComd(int socfd);
	  int ReadComd();
	  
	  int SaveFile(int socfd);
	  int SaveFile();
	  
	  int ReadFile(int socfd);
	  int ReadFile();
	  
} ;

#endif