/*

CPC xfer.... transfer files to/from M4 board sd card, via command line, useful for ie. crossdev

Duke 2016/2017

*/

#include <stdlib.h>

#include <stdio.h>

#include <stdint.h>

#include "http.h"

#include "parse.h"

#include "cpc.h"

#ifndef __WIN32__

#include <sys/types.h>

#include <sys/stat.h>

#include <unistd.h>

#define strnicmp strncasecmp

#endif

// mres = m4 reboot

// cres = cpc reset

// chlt = cpc pause/unpause (BUSRQ)

// run = run file on sd at startup

void dispInfo(char *exename)

{

printf("CPC M4 xfer tool v2.0.3 - Duke 2016/2017\r\n");

printf("%s -u [ipaddr] file path opt\t\t- Upload file, opt 0: no header add, 1: add ascii header 2: binary header\r\n", exename);

printf("%s -d [ipaddr] file path opt\t\t- Download file, opt 0: leave header, 1: remove header\r\n", exename);

printf("%s -f [ipaddr] file slot name\t\t- Upload rom\r\n", exename);

printf("%s -c [ipaddr] file\t\t\t- Upload cartridge image (.CPR/.BIN)\r\n", exename);

printf("%s -x [ipaddr] path+file\t\t- Execute file on CPC\r\n", exename);

printf("%s -y [ipaddr] local_file\t\t- Upload file on CPC and execute it immediatly (the sd card must contain folder '/tmp')\r\n", exename);

printf("%s -p [ipaddr]\t\t\t\t- Start (plus) cartridge\r\n", exename);

printf("%s -s [ipaddr]\t\t\t\t- Reset CPC\r\n", exename);

printf("%s -r [ipaddr]\t\t\t\t- Reboot M4\r\n", exename);

}

void reset(char *ip)

{ char httpReq[128];

int sd;

volatile int n;

sd = httpConnect(ip);

if ( sd > 0 )

{

n = sprintf(httpReq, "GET /config.cgi?mres HTTP/1.0\r\nHost: %s\r\nUser-Agent: cpcxfer\r\n\r\n", ip);

send(sd, httpReq, n, 0);

while (n > 0) // ignore response... a 200 OK check would be a good idea....

n = recv(sd, httpReq, sizeof(httpReq),0 );

#ifdef __WIN32__

closesocket(sd);

#else

close(sd);

#endif

printf("M4 Reset request sent.\r\n");

}

}

void startCard(char *ip)

{ char httpReq[128];

int sd;

volatile int n;

sd = httpConnect(ip);

if ( sd > 0 )

{

n = sprintf(httpReq, "GET /config.cgi?cctr HTTP/1.0\r\nHost: %s\r\nUser-Agent: cpcxfer\r\n\r\n", ip);

send(sd, httpReq, n, 0);

while (n > 0) // ignore response... a 200 OK check would be a good idea....

n = recv(sd, httpReq, sizeof(httpReq),0 );

#ifdef __WIN32__

closesocket(sd);

#else

close(sd);

#endif

printf("M4 Reset request sent.\r\n");

}

}

void resetCPC(char *ip)

{ char httpReq[128];

int sd;

volatile int n;

sd = httpConnect(ip);

if ( sd > 0 )

{

n = sprintf(httpReq, "GET /config.cgi?cres HTTP/1.0\r\nHost: %s\r\nUser-Agent: cpcxfer\r\n\r\n", ip);

send(sd, httpReq, n, 0);

while (n > 0) // ignore response... a 200 OK check would be a good idea....

n = recv(sd, httpReq, sizeof(httpReq),0 );

#ifdef __WIN32__

closesocket(sd);

#else

close(sd);

#endif

printf("M4 Reset request sent.\r\n");

}

}

void run(char *ip, char *path)

{ char httpReq[128];

int sd;

volatile int n;

sd = httpConnect(ip);

if ( sd > 0 )

{

n = sprintf(httpReq, "GET /config.cgi?run=%s HTTP/1.0\r\nHost: %s\r\nUser-Agent: cpcxfer\r\n\r\n", path, ip);

send(sd, httpReq, n, 0);

while (n > 0) // ignore response... a 200 OK check would be a good idea....

n = recv(sd, httpReq, sizeof(httpReq),0 );

#ifdef __WIN32__

closesocket(sd);

#else

close(sd);

#endif

printf("Running %s\r\n", path);

}

}

#ifndef __WIN32__

int is_regular_file(const char *path)

{

struct stat path_stat;

stat(path, &path_stat);

return S_ISREG(path_stat.st_mode);

}

#endif

void upload(char *filename, char *path, char *ip, int opt, unsigned short start, unsigned short exec)

{

int ret, size, p, n, k, i, j;

char fullpath[256]; // don't exceed this, the cpc wouldn't be able |cd it anyway

FILE *fd;

SOCKET sd;

unsigned char *buf;

fd = fopen(filename, "rb");

if ( fd == NULL )

{ printf("file %s not found\n", filename);

return;

}

#ifndef __WIN32__

if (!is_regular_file(filename))

{ printf("file %s is not a regular file\n", filename);

fclose(fd);

return;

}

#endif

fseek(fd, 0, SEEK_END);

size = ftell(fd);

fseek(fd, 0, SEEK_SET);

buf = malloc(size+0x80);

if ( buf == NULL )

{

printf("Not enough memory!\n");

fclose(fd);

return;

}

if ( opt != 0 )

{ _cpchead *cpcheader = (_cpchead *)buf;

memset(cpcheader, 0, 0x80);

// set up cleaned filename and extionsion

formatfn(cpcheader->filename, filename);

getExtension(cpcheader->extension, filename);

// add more file types here... and more paramters....

switch (opt)

{

case 1:

cpcheader->addr = 0x172; // protext uses this

cpcheader->type = 10;

cpcheader->size = size;

cpcheader->size2 = size;

cpcheader->checksum = checksum16(buf, 66);

break;

default:

case 2: // binary header

cpcheader->addr = start;

cpcheader->exec = exec;

cpcheader->type = 2;

cpcheader->size = size;

cpcheader->size2 = size;

cpcheader->checksum = checksum16(buf, 66);

break;

}

fread(&buf[0x80], size, 1, fd);

size+=0x80;

}

else

fread(buf, size, 1, fd);

fclose(fd);

ret = httpConnect(ip);

if ( ret >= 0 )

{ sd = ret; // connect socket

p = pathPos(filename, strlen(filename)); // strip any PC path from filename

sprintf(fullpath, "%s/%s", path, &filename[p]);

k = strlen(fullpath);

n = 0;

while ( n < k ) // remove leading / from path

{

if ( fullpath[n] != '/' )

break;

n++;

}

// remove duplicate /'s

k = strlen(fullpath);

j = 0;

for (i=n; i < (k+1); i++)

{

if ( (fullpath[i] == '/') && (fullpath[i+1] == '/') )

i++;

fullpath[j++] = fullpath[i];

}

if ( httpSend(sd, fullpath, buf, size, "upfile", "/upload.html", ip) >= 0 )

ret = httpResponse(sd);

httpClose(sd);

if ( ret == 200 )

printf("Upload OK!\r\n");

else

printf("Upload error code. %i\r\n", ret);

}

else

printf("Connect to %s failed\n", ip);

free(buf);

}

void uploadctr(char *filename, char *ip)

{

int ret, size, p, n, k, i, j;

FILE *fd;

SOCKET sd;

unsigned char *buf;

fd = fopen(filename, "rb");

if ( fd == NULL )

{ printf("file %s not found\n", filename);

return;

}

#ifndef __WIN32__

if (!is_regular_file(filename))

{ printf("file %s is not a regular file\n", filename);

fclose(fd);

return;

}

#endif

fseek(fd, 0, SEEK_END);

size = ftell(fd);

fseek(fd, 0, SEEK_SET);

buf = malloc(size);

if ( buf == NULL )

{

printf("Not enough memory!\n");

fclose(fd);

return;

}

fread(buf, size, 1, fd); // gah no validation, M4 will verify some...

fclose(fd);

ret = httpConnect(ip);

if ( ret >= 0 )

{ sd = ret; // connect socket

if ( httpSend(sd, "/CARTIMG.BIN", buf, size, "upfile", "/upload.html", ip) >= 0 )

ret = httpResponse(sd);

httpClose(sd);

if ( ret == 200 )

printf("Upload OK!\r\n");

else

printf("Upload error code. %i\r\n", ret);

}

else

printf("Connect to %s failed\n", ip);

free(buf);

}

void uploadRom(char *filename, char *ip, int slot, char *slotname)

{

int ret, size, p, n, k, i, j;

char fullpath[256]; // don't exceed this, the cpc wouldn't be able |cd it anyway

FILE *fd;

SOCKET sd;

unsigned char *buf;

fd = fopen(filename, "rb");

if ( fd == NULL )

{ printf("file %s not found\n", filename);

return;

}

fseek(fd, 0, SEEK_END);

size = ftell(fd);

fseek(fd, 0, SEEK_SET);

buf = malloc(size);

if ( buf == NULL )

{

printf("Not enough memory!\n");

fclose(fd);

return;

}

fread(buf, size, 1, fd);

fclose(fd);

ret = httpConnect(ip);

if ( ret >= 0 )

{ sd = ret; // connect socket

if ( httpSendRom(sd, "rom.bin", buf, size, slot, "/roms.shtml", ip, slotname) >= 0 )

ret = httpResponse(sd);

httpClose(sd);

if ( ret == 200 )

printf("Upload OK!\r\n");

else

printf("Upload error code. %i\r\n", ret);

}

else

printf("Connect to %s failed\n", ip);

free(buf);

}

void download(char *filename, char *path, char *ip, int opt)

{

SOCKET sd;

int ret, k, i, j;

char fullpath[256];

sprintf(fullpath, "sd/%s/%s", path, filename);

// remove duplicate /'s

k = strlen(fullpath);

j = 0;

for (i=0; i < (k+1); i++)

{

if ( (fullpath[i] == '/') && (fullpath[i+1] == '/') )

i++;

fullpath[j++] = fullpath[i];

}

ret = httpConnect(ip);

sd = ret;

if ( ret >= 0 )

{

if ( httpGet(sd, ip, fullpath, opt) == 0)

printf("Downloaded succesfully.\r\n");

else

printf("Download failed.\r\n");

}

else

printf("Connect failed, wrong ip?\r\n");

}

#define GET_CPCIP(argpos, argmax) \

{ \

printf("%d, %d, %d\n", argc, argmax, argpos); \

cpcip = NULL; \

cpcip = getenv("CPCIP"); \

printf("IP1=%s\n", cpcip); \

/*if (cpcip != NULL) \

{ \

while (*cpcip != '=') {++cpcip;}; \

++cpcip; \

} else{printf("not found");}*/\

printf("IP2=%s\n", cpcip); \

if (argc == (argmax)+1 || cpcip==NULL) { \

printf("IP2=%s\n", cpcip); \

cpcip = argv[(argpos)]; \

printf("IP2=%s\n", cpcip); \

} \

else { \

argdelta = -1; \

} \

if (NULL == cpcip) { \

fprintf(stderr, "[ERROR] Unable to retreive the IP of the M4.\n Please provide it as an argument or through CPCIP variable.\n"); \

exit(-1); \

} \

printf("IP3=%s\n", cpcip); \

}

int main(int argc, char *argv[])

{ int opt = 0;

// Retreive the IP of the M4. Can be overriden by the option managment

const char * cpcip=NULL;

// Contains -1 if the ip is not provided in the command line

int argdelta = 0;

#ifdef __WIN32__

WSADATA WsaDat;

#endif

if ( argc < 2 )

{

dispInfo(argv[0]);

exit(0);

}

#ifdef __WIN32__

WSAStartup(MAKEWORD(2,2),&WsaDat);

#endif

if ( !strnicmp(argv[1], "-p", 2) ) // reset cpc

{

GET_CPCIP(2, 2);

startCard(cpcip);

}

else

if ( !strnicmp(argv[1], "-r", 2) ) // reboot M4

{

GET_CPCIP(2, 2);

reset(cpcip);

}

else if ( !strnicmp(argv[1], "-s", 2) ) // reset cpc

{

GET_CPCIP(2, 2);

resetCPC(cpcip);

}

else if ( !strnicmp(argv[1], "-x", 2) && (argc>=3) ) // execute

{

GET_CPCIP(2, 2);

run(cpcip, argv[3 + argdelta]);

}

else if ( !strnicmp(argv[1], "-f", 2) && (argc>=4) ) // upload (flash) rom

{

GET_CPCIP(2, 4);

uploadRom(argv[3+argdelta], cpcip, atoi(argv[4+argdelta]), argv[5+argdelta] );

}

else if ( !strnicmp(argv[1], "-u", 2) && (argc>=3) ) // upload

{ unsigned short start = 0;

unsigned short exec = 0;

GET_CPCIP(2, 5);

if ( argc < 5+argdelta && cpcip == NULL)

{

dispInfo(argv[0]);

exit(0);

}

if ( argc>5+argdelta )

{ opt = atoi(argv[5+argdelta]);

if ( opt == 2 ) // add binary header ?

{

// we need more parameters then

if ( argc < 8 )

{

printf("Please add start address and exec. address for binary header\r\n");

exit(0);

}

start = atoh(argv[6+argdelta]);

exec = atoh(argv[7+argdelta]);

}

}

upload(argv[3+argdelta], argv[4+argdelta], cpcip, opt, start, exec);

}

else if ( !strnicmp(argv[1], "-c", 2) && (argc>=4) ) // upload cartridge

{

GET_CPCIP(2, 3);

uploadctr(argv[3+argdelta], cpcip);

}

else if ( !strnicmp(argv[1], "-y", 2) && (argc>=3) )

{

GET_CPCIP(2, 3);

// Prepare the file manipulation variables

char fullpath[256];

char path[] = "/tmp"; // TODO Find a way to use /tmp or something like that

char * filename = argv[3+argdelta];

int p = pathPos(filename, strlen(filename)); // strip any PC path from filename

sprintf(fullpath, "%s/%s", path, &filename[p]);

upload(filename, path, cpcip, 0, 0, 0 ); // Upload the file on CPC

run(cpcip, fullpath); // Execute the file on CPC

}

else if ( !strnicmp(argv[1], "-d", 2) && (argc>=4) ) // download

{

GET_CPCIP(2, 5);

// TODO properlly manage the case when not provided

if ( argc>5 )

opt = atoi(argv[5+argdelta]);

download(argv[3+argdelta], argv[4+argdelta], argv[2], opt);

}

else

{

dispInfo(argv[0]);

exit(0);

}

#ifdef __WIN32__

WSACleanup();

#endif

return 0;

}