up.c

Go to the documentation of this file.

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>

#include "ev3_io.h"

#include "defs.h"
#include "packets.h"
#include "error.h"
#include "funcs.h"

#define CHUNK_SIZE 1000 // The EV3 doesn't do packets > 1024B

int up(FILE *fp, const char *dst)
{
    int res;

    fseek(fp, 0, SEEK_END);
    long fsize = ftell(fp);
    fseek(fp, 0, SEEK_SET);

    if ((unsigned long)fsize > (u32)-1)
        return ERR_FTOOBIG;   

    unsigned chunks   = fsize / CHUNK_SIZE;
    unsigned final_chunk_sz = fsize % CHUNK_SIZE;

    //fprintf(stderr, "Attempting file upload (%ldb total; %u chunks): \n", fsize, chunks + 1);

    size_t dst_len = strlen(dst) + 1;
    BEGIN_DOWNLOAD *bd = packet_alloc(BEGIN_DOWNLOAD, dst_len);
    bd->fileSize = fsize;
    memcpy(bd->fileName, dst, dst_len);

    res = ev3_write(handle, (u8 *)bd, bd->packetLen + PREFIX_SIZE);
    if (res < 0)
    {
        errmsg = "Unable to write BEGIN_DOWNLOAD.";
        return ERR_COMM;
    }

    BEGIN_DOWNLOAD_REPLY bdrep;
    res = ev3_read_timeout(handle, (u8 *)&bdrep, sizeof bdrep, TIMEOUT);
    if (res <= 0)
    {
        errmsg = "Unable to read BEGIN_DOWNLOAD";
        return ERR_COMM;
    }

    if (bdrep.type == VM_ERROR)
    {
        errno = bdrep.ret;
        errmsg = "BEGIN_DOWNLOAD was denied.";
        return ERR_VM;
    }

    CONTINUE_DOWNLOAD *cd = packet_alloc(CONTINUE_DOWNLOAD, CHUNK_SIZE);
    for (size_t i = 0; i < chunks; ++i)
    {
        cd->fileHandle = bdrep.fileHandle;
        fread(cd->fileChunk, 1, CHUNK_SIZE, fp);
        res = ev3_write(handle, (u8 *)cd, cd->packetLen + PREFIX_SIZE);

        if (res < 0)
        {
            errmsg = "Unable to write CONTINUE_DOWNLOAD.";
            return ERR_COMM;
        }
        res = ev3_read_timeout(handle, (u8 *)&bdrep, sizeof bdrep, TIMEOUT);
        if (res <= 0)
        {
            errmsg = "Unable to read CONTINUE_DOWNLOAD";
            return ERR_COMM;
        }
    }
    cd->packetLen = sizeof(CONTINUE_DOWNLOAD) + final_chunk_sz - PREFIX_SIZE;
    fread(cd->fileChunk, 1, final_chunk_sz, fp);
    ev3_write(handle, (u8*)cd, cd->packetLen + PREFIX_SIZE);
    if (res < 0)
    {
        errmsg = "Unable to write CONTINUE_DOWNLOAD.";
        return ERR_COMM;
    }
    res = ev3_read_timeout(handle, (u8 *)&bdrep, sizeof bdrep, TIMEOUT);
    if (res <= 0)
    {
        errmsg = "Unable to read CONTINUE_DOWNLOAD";
        return ERR_COMM;
    }
    if (bdrep.type == VM_ERROR)
    {
        errno = bdrep.ret;

        fputs("Transfer failed.\nlast_reply=", stderr);

        print_bytes(&bdrep, bdrep.packetLen);

        errmsg = "CONTINUE_DOWNLOAD was denied.";
        return ERR_VM;
    }

    errmsg = "`upload` was successful.";
    return ERR_UNK;
}