Windows平台下MingGW的网络socket编程模型

Windows平台下MingGW的网络socket编程模型

1、TCP服务器

#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <winsock2.h>
#include <pthread.h>

#define LOCAL_IP      "192.168.31.254"
#define LOCAL_PORT    ((uint16_t)10126)

#define debug(...)                                     \
            do                             \
            {                              \
              printf(__VA_ARGS__);       \
              fflush(stdout);            \
            }while(0)



int main(void)
{
  int ret = 0;
  WSADATA ws;
  SOCKET sock, newsock;
  struct sockaddr_in local;
  struct sockaddr_in client;
  int addrlen = 0;
  int nrecv   = 0;
  char buf[512];

  if(WSAStartup(MAKEWORD(2, 2), &ws) != 0)
  {
    ret = -1;
    goto __exit;
  }

  sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
  if(sock == INVALID_SOCKET)
  {
    ret = -2;
    goto __exit;
  }

  local.sin_family = AF_INET;
  local.sin_addr.S_un.S_addr = inet_addr(LOCAL_IP);
  local.sin_port = htons(LOCAL_PORT);
  memset(local.sin_zero, 0X00, sizeof(local.sin_zero));

  if(bind(sock, (const struct sockaddr *)&local, sizeof(local)) == SOCKET_ERROR)
  {
    ret = -3;
    goto __exit;
  }

  listen(sock, 5);

  addrlen = sizeof(local);

  debug(" accept %s:%d...\r\n", LOCAL_IP, LOCAL_PORT);

  if((newsock = accept(sock, (struct sockaddr *)&client, &addrlen)) == SOCKET_ERROR)
  {
    ret = -4;
    goto __exit;
  }

  debug(" %s:%d connected\r\n", inet_ntoa(client.sin_addr), client.sin_port);

  while(1)
  {
    if((nrecv = recv(newsock, buf, sizeof(buf), 0)) <= 0)
      break;

    if(strncmp(buf, "quit", 4) == 0)
      break;

    if(send(newsock, buf, nrecv, 0) < 0)
      break;
  }

__exit:
  closesocket(sock);
  closesocket(newsock);
  return ret;
}

2、TCP客户端

#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <winsock2.h>
#include <pthread.h>

#define LOCAL_IP      "192.168.31.254"
#define LOCAL_PORT    ((uint16_t)10126)

#define debug(...)                                     \
            do                             \
            {                              \
              printf(__VA_ARGS__);       \
              fflush(stdout);            \
            }while(0)


int main(void)
{
  int ret = 0;
  WSADATA ws;
  SOCKET sock;
  struct sockaddr_in server;
  int nrecv   = 0;
  char buf[100];

  if(WSAStartup(MAKEWORD(2, 2), &ws) != 0)
  {
    ret = -1;
    goto __exit;
  }

  sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
  if(sock == INVALID_SOCKET)
  {
    ret = -2;
    goto __exit;
  }

  server.sin_family = AF_INET;
  server.sin_addr.S_un.S_addr = inet_addr(LOCAL_IP);
  server.sin_port = htons(LOCAL_PORT);

  if(connect(sock, (struct sockaddr *)&server, sizeof(struct sockaddr_in)) < 0)
  {
    ret = -3;
    goto __exit;
  }

  memset(buf, 'A', sizeof(buf));

  while(1)
  {
    if(send(sock, buf, sizeof(buf), 0) < 0)
      break;

    if((nrecv = recv(sock, buf, sizeof(buf), 0)) <= 0)
      break;

    if(strncmp(buf, "quit", 4) == 0)
      break;
  }

__exit:
  closesocket(sock);
  return ret;
}

ends…