client_asm.c

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

#include "ascii.h"
#include "client.h"
#include "options.h"


int sockfd = 0;

void sigint_handler(int sigint) {
    (void) (sigint);
    ascii_write_destroy();
    printf("Closing connection and exiting . . .\n");
    close(sockfd);
    exit(0);
}
//----------------------ASM goes here-------------
 .MODEL small, c

        .DATA

arg1    DW     1234
arg2    DW     4321
format1 DB     "Arg1: %d", 10, 0      ; Format string for printf
format2 DB     "Arg2: %d", 10, 10, 0  ; Format string for printf

EXTRN  _acrtused:abs              ; Bring in C startup

        .CODE

EXTRN  ptrswap:proc               ; External C routine
EXTRN  printf:proc                ; External C run-time routine
PUBLIC main                       ; C startup requires the name _main

main:
        MOV    ax, arg1
        PUSH   ax                 ; Push 2nd argument (C convention)
        MOV    bx, offset format1
        PUSH   bx                 ; Push 1st argument (C convention)
        CALL   printf             ; Call C run-time routine

        MOV    ax, arg2
        PUSH   ax                 ; Push 2nd argument (C convention)
        MOV    bx, offset format2
        PUSH   bx                 ; Push 1st argument (C convention)
        CALL   printf             ; Call C run-time routine

        MOV    bx, offset arg2
        PUSH   bx                 ; Push 2nd argument (C convention)
        MOV    bx, offset arg1
        PUSH   bx                 ; Push 1st argument (C convention)
        CALL   ptrswap            ; Call C routine from module

        MOV    ax, arg1
        PUSH   ax                 ; Push 2nd argument (C convention)
        MOV    bx, offset format1
        PUSH   bx                 ; Push 1st argument (C convention)
        CALL   printf             ; Call C run-time routine

        MOV    ax, arg2
        PUSH   ax                 ; Push 2nd argument (C convention)
        MOV    bx, offset format2
        PUSH   bx                 ; Push 1st argument (C convention)
        CALL   printf             ; Call C run-time routine

        MOV    ah, 4ch            ; Terminate program
        int    21h

        END                       ; Entry point will be specified by
                                  ;    the C startup code


//------------------------------------------------

int main(int argc, char *argv[]) {
    options_init(argc, argv);
    char *address = opt_address;
    int   port    = strtoint(opt_port);

    char recvBuff[40000];
    struct sockaddr_in serv_addr;

    struct timespec
    sleep_start = {
        .tv_sec  = 3,
        .tv_nsec = 0
    },
    sleep_stop = {
        .tv_sec  = 0,
        .tv_nsec = 0
    };

    // Cleanup nicely on Ctrl+C.
    signal(SIGINT, sigint_handler);

    /* read from the socket as long as the size of the read is > 0 */
    while(1) {
        // try to open a socket
        memset(recvBuff, '0',sizeof(recvBuff));
        // error creating socket
        if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
            printf("\n Error: could not create socket \n");
            return 1;
        }

        // reserve memory space to store IP address
        memset(&serv_addr, '0', sizeof(serv_addr));

        // set type of address to IPV4 and port to 5000
        printf("Connecting on port %d\n", port);

        serv_addr.sin_family = AF_INET;
        serv_addr.sin_port = htons(port);

        // an error occurred when trying to set server address and port number
        if(inet_pton(AF_INET, address, &serv_addr.sin_addr) <= 0) {
            printf("\n Error: inet_pton \n");
            return 1;
        }

        int read_result = 0;
        ascii_write_init();

        printf("Attempting to connect...\n");
        // failed when trying to connect to the server
        if(connect(sockfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) {
            printf("\n Error: connect failed \n");
        }

        while (0 < (read_result = read(sockfd, recvBuff, sizeof(recvBuff)-1))) {
            recvBuff[read_result] = 0;
            ascii_write(recvBuff);
        }

        console_clear();

        nanosleep((struct timespec *)&sleep_start,(struct timespec *)&sleep_stop);
    }

    return 0;
}