Network Programming - Simple Socket Echo Server

這學期做網程設的助教，我想應該作業要寫一遍的前提下，紀錄一下也讓以後如果自己要寫的話不用再從頭抓資料多麻煩XD

Client.c

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

void error(const char *msg)
{
    perror(msg);
    exit(0);
}

int main(int argc, char *argv[])
{
    int sockfd, n;
    struct sockaddr_in serv_addr;
    struct hostent *server;

    char buffer[256];
    if (argc < 3) {
       fprintf(stderr,"usage: %s [hostname] [port]\n", argv[0]);
       exit(0);
    }
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
        error("ERROR opening socket");
    server = gethostbyname(argv[1]);

    if (server == NULL) {
        fprintf(stderr,"ERROR, no such host\n");
        exit(0);
    }
    bzero((char *) &serv_addr, sizeof(serv_addr));

    serv_addr.sin_family = AF_INET;
    bcopy((char *)server->h_addr,
         (char *)&serv_addr.sin_addr.s_addr,
         server->h_length);
    serv_addr.sin_port = htons(atoi(argv[2]));

    if (connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr)) < 0)
        error("ERROR connecting");

    printf("Connected!\necho-Server >");
    fflush(stdout);
    bzero(buffer,256);
    fgets(buffer,255,stdin);

    n = write(sockfd,buffer,strlen(buffer));
    if (n < 0)
         error("ERROR writing to socket");

    bzero(buffer,256);

    n = read(sockfd,buffer,255);
    if (n < 0)
         error("ERROR reading from socket");

    printf("%s\n",buffer);

    close(sockfd);
    return 0;
}

Server.c

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

#define LISTENQLEN 10

void errmsg(const char *msg)
{
    perror(msg);
    exit(1);
}

int main(int argc, char *argv[])
{
    int sockfd, newsockfd, portno;
    socklen_t clilen;
    char buffer[256];
    struct sockaddr_in serv_addr, cli_addr;
    int n;

    if (argc != 2) {
        fprintf(stderr,"ERROR, no port provided\n");
        exit(1);
    }

    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    //socket():
    //argument 1:                       argument 2:
    //AF_INET : IPv4                    SOCK_STREAM : stream socket
    //AF_INET6: IPv6                    SOCK_DGRAM  : datagram socket
    //AF_LOCAL: Unix domain protocols   SOCK_SEQPACKET: sequenced packet socket
    //AF_ROUTE: Routing socket          SOCK_RAW    : raw socket
    //AF_KEY  : Key socket
    //
    //argument 3:
    // IPPROTO_TCP : TCP
    // IPPROTO_UDP : UDP
    // IPPROTO_SCTP: SCTP

    if (sockfd < 0)
        errmsg("ERROR opening socket");

    bzero((char *) &serv_addr, sizeof(serv_addr));

    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = INADDR_ANY;   // listen where
    serv_addr.sin_port = htons(atoi(argv[1]));// port number

    if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0)
        errmsg("ERROR on binding");

    listen(sockfd,LISTENQLEN);

    clilen = sizeof(cli_addr);

    newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);

    if (newsockfd < 0)
        errmsg("ERROR on accept");

    bzero(buffer,256);
    n = read(newsockfd,buffer,255);
    if (n < 0)
        errmsg("ERROR reading from socket");

    printf("Recieved string: %s\n",buffer);
    n = write(newsockfd,buffer,strlen(buffer));
    if (n < 0)
        errmsg("ERROR writing to socket");

    close(newsockfd);
    close(sockfd);

    return 0;
}

Makefile

all : client.c server.c
 gcc -Wall -g -o client client.c
 gcc -Wall -g -o server server.c
 
clean : 
 rm client
 rm server

直接make完以後就可以使用。