Meu primeiro cliente/servidor

Aula de Redes de Computadores. Começando a trabalhar com sockets, primeiros programas estabelecendo comunicação...

Cliente:

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

/* função para imprimir a mensagem de erro */
void error(const char *msg) {
    printf("%s\n", msg);
    exit(0);
}

/* função principal */
int main(int argc, char *argv[]) {
    int sockfd, portno, n;
    struct sockaddr_in serv_addr;
    struct hostent *server;
    char buffer[256];


    if (argc < 3) {
        fprintf(stderr, "Use: %s hostname port\n", argv[0]);
        exit(0);
    }

    portno = atoi(argv[2]);

    /* socket = primeira função chamada */
    sockfd = socket(AF_INET, SOCK_STREAM, 0);

    if (sockfd < 0)
        error("Falha ao criar o socket");

    /* Inicializando a estrutura */
    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);

    // O que htons faz?!?
    serv_addr.sin_port = htons(portno);

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

    printf("Please enter the message: ");
    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\n", buffer);

    close(sockfd);

    return 0;
}

Servidor:

 /* A simple server in the internet domain using TCP

   The port number is passed as an argument */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <arpa/inet.h>

#include <unistd.h>

/* função para imprimir a mensagem de erro */

void error(const char *msg) {

    perror(msg);

    exit(1);

}

/* função principal */

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);

    }

    /* socket = primeira função chamada */

    sockfd = socket(AF_INET, SOCK_STREAM, 0);

    if (sockfd < 0)

        error("ERROR opening socket");

    /* bzero limpa, colocando /o em todas as posições */

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

    /* portno recebe o argumento do teclado com o numero da porta */

    portno = atoi(argv[1]);

    serv_addr.sin_family = AF_INET; /* Usaremos a família de protocolos */

    /* correspondente à Internet */

    serv_addr.sin_addr.s_addr = INADDR_ANY;

    /* Usaremos a porta passada como parâmetro */

    serv_addr.sin_port = htons(portno); 

    /* bind = segunda função chamada */

    if (bind(sockfd, (struct sockaddr *) &serv_addr,

            sizeof (serv_addr)) < 0)

        error("ERROR on binding");

    /* listen = função que cria fila de conexões */

    if (listen(sockfd, 5) < 0)

        error("ERROR on waiting");

    clilen = sizeof (cli_addr);

    printf("Waiting for connection\n");

    fflush(NULL);

    /* accept = quarta função chamada */

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

    if (newsockfd < 0)

        error("ERROR on accept");

    bzero(buffer, 256);

    /* Receber mensagem através de newsockfd */

    n = recv(newsockfd, buffer, 256, 0);

    if (n < 0) error("ERROR reading from socket");

    printf("Here is the message: %s\n", buffer);

    /* Enviar mensagem através de newsockfd */

    n = send(newsockfd, buffer, n, 0);

    if (n < 0) error("ERROR writing to socket");

    if ((n = recv(newsockfd, buffer, 256, 0)) < 0)

        error("Falhou ao receber os dados restantes do cliente");

    close(newsockfd);

    close(sockfd);

    return 0;

}