プログラマ専用SNS ミクプラ

コード:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <pthread.h>

#define BUFSIZE 1024
#define LISTENQ 1024

void *thread(void *arg);

struct ThreadArgs {
  int sock;
  int cnt;
  struct sockaddr_in sa;
};

int main(int argc, char *argv[])
{
  int listen_sock, comm_sock, sa_len, cnt;
  struct sockaddr_in sa, new_sa;
  pthread_t thread_id;
  struct ThreadArgs *args;

  listen_sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);

  memset(&sa, 0, sizeof(sa));
  sa.sin_family = AF_INET;
  sa.sin_port = htons(49152);
  sa.sin_addr.s_addr = htonl(INADDR_ANY);

  if (bind(listen_sock, (struct sockaddr *)&sa, sizeof(sa)) < 0) {
    fprintf(stderr, "bind() failure\n");
    exit(1);
  }

  if (listen(listen_sock, LISTENQ) < 0) {
    fprintf(stderr, "listen() failure\n");
    exit(1);
  }

  cnt = 0;
  while (1) {
    printf("Waiting for a client\n");
    sa_len = sizeof(new_sa);
    if ((comm_sock = accept(listen_sock, (struct sockaddr *)&new_sa, &sa_len)) < 0) {
      fprintf(stderr, "accept() failure\n");
      exit(1);
    }
    ++cnt;

    args = (struct ThreadArgs *)malloc(sizeof(struct ThreadArgs));
    args->sock = listen_sock;
    args->cnt  = cnt;
    args->sa   = sa;

    if (pthread_create(&thread_id, NULL, (void *)thread, (void *)comm_sock) != 0) {
      fprintf(stderr, "pthread_create() failure\n");
      exit(1);
    }
  }
  return 0;
}

void *thread(void *arg)
{
  int sock, cnt, n, msg_len;
  struct sockaddr_in sa;
  char buf[BUFSIZE];

  // 1. pthread_detach()でメモリを再利用できるようにしておく
  // 2. argから通信用ソケットディスクリプタ, クライアント番号, アドレス構造体
  //    を取得し、それぞれsock, cnt, saにセット

  printf("Client %d (%s) connected\n", cnt, inet_ntoa(sa.sin_addr));

  if ((n = recv(sock, buf, sizeof(buf), 0)) > 0) {
    buf[n] = '\0';
    msg_len = strlen(buf);
    printf("Client %d: %s\n", cnt, buf);
    if (send(sock, buf, msg_len, 0) != msg_len) {
      fprintf(stderr, "send() failure\n");
      exit(1);
    }
  }
  close(sock);

  printf("Client %d disconnecter\n", cnt);

  free(arg);
  return;
}