C++ socket程序

C++ socket程序

 

下面是一个C++调用windows API实现有通信程序,您可以据此进行修改,封装成类.

 

// *******************************************************************

// client.cpp : Defines the entry point for the console application.
// author: hnynes

// MSN:

// *******************************************************************

#include "stdafx.h"
#include
#include
#define PORT_SERVER 6656

int main(int argc, char* argv[])
{
 std::cout << "Start up tcp client." << std::endl;
 
 WSADATA wsaData;
 WORD sockVersion = MAKEWORD(2, 0);
 if (0 != WSAStartup(sockVersion, &wsaData))
 {
  std::cout << "Failed to retrive socket version."
   << std::endl;
  return 0;
 }
 
 SOCKET sock_client;
 sock_client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 if (INVALID_SOCKET == sock_client)
 {
  std::cout << "Invalid socket." << std::endl;
  WSACleanup();
  return 0;
 }
 sockaddr_in addr_sev;
 addr_sev.sin_family = AF_INET;
 addr_sev.sin_port = htons(PORT_SERVER);
 addr_sev.sin_addr.s_addr = inet_addr("127.0.0.1");
 if (SOCKET_ERROR == connect(sock_client, (sockaddr *)&addr_sev, sizeof(addr_sev)))
 {
  std::cout << "Failed to connect." << std::endl;
  WSACleanup();
  return 0;
 }

 char buf[512];
 int recv_size = 0;
 recv_size = recv(sock_client, buf, 512, 0);
 if (512 > recv_size)
 {
  buf[recv_size] = '\0';
  std::cout << "receive data is :" << buf << std::endl;
  memset(buf, 0, sizeof(buf));
 }
 else
 {
  std::cout << "receive data is overflow" << std::endl;
 }
 

 closesocket(sock_client);
 WSACleanup();
 system("PAUSE");
 return 0;
}

 

// *******************************************************************

// server.cpp : Defines the entry point for the console application.
// author: hnynes

// MSN:

// *******************************************************************

#include "stdafx.h"
#include
#include
#define PORT_SERVER 6656
#define NUM_CLIENTS 10

int main(int argc, char* argv[])
{
 std::cout << "Start up tcp server." << std::endl;
 
 WSADATA wsaData;
 WORD sockVersion = MAKEWORD(2, 0);
 if (0 != WSAStartup(sockVersion, &wsaData))
 {
  std::cout << "Failed to retrive socket version."
         << std::endl;
  return 0;
 } 

 SOCKET sock_sev;
 sock_sev = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 if (INVALID_SOCKET == sock_sev)
 {
  std::cout << "Invalid socket." << std::endl;
  WSACleanup();
  return 0;
 }

 sockaddr_in addr_sev;
 addr_sev.sin_family = AF_INET;
 addr_sev.sin_port = htons(PORT_SERVER);
 addr_sev.sin_addr.s_addr = INADDR_ANY;
 if (SOCKET_ERROR == bind(sock_sev, (sockaddr *)&addr_sev, sizeof(addr_sev)))
 {
  std::cout << "Failed to bind." << std::endl;
  WSACleanup();
  return 0;
 }
 if (SOCKET_ERROR == listen(sock_sev, NUM_CLIENTS))
 {
  std::cout << "Failed to listen." << std::endl;
  WSACleanup();
  return 0;
 }
 

 // 多路复用,暂时还未实现.后期会对此给出示例.
 //fd_set reads;
 //fd_set writes;
 //fd_set accepts;
 //FD_ZERO(&reads);
 //FD_ZERO(&writes);
 //FD_ZERO(&accepts);
 SOCKET sock_client;
 sockaddr_in addr_client;
 int nAddrLen = sizeof(addr_client);
 while (true)
 {
  char tmp[512];
  sock_client = accept(sock_sev, (sockaddr *)&addr_client, &nAddrLen);
  
  if (INVALID_SOCKET == sock_client)
  {
   std::cout << "Failed to accept." << std::endl;
   continue;
  }
  sprintf(tmp, "Your IP is %s\n", inet_ntoa(addr_client.sin_addr));
  send(sock_client, tmp, strlen(tmp), 0);
  std::cout << "Connection from " << inet_ntoa(addr_client.sin_addr)
         << std::endl;

  closesocket(sock_client);
 }
 
 return 0;
}

 

注:本程序使用的是VC6 IDE,编译时请在link中加入对ws2_32.lib库的引用.

这个程序只是一个雏形,也作学习吧.