implemented file API in general protocol

This commit is contained in:
Bogdan Pilyugin 2024-03-24 20:10:58 +02:00
parent 5aaa2004c3
commit 9b11542f1a
2 changed files with 89 additions and 136 deletions

View File

@ -40,21 +40,30 @@
#define READ_ORERATION 1 #define READ_ORERATION 1
#define DELETE_ORERATION 2 #define DELETE_ORERATION 2
#define APPEND_ORERATION 3 #define WRITE_ORERATION 3
#define REPLACE_ORERATION 4
static const char *dirpath = "/data/"; static const char *dirpath = "/data/";
#define MEM_OBLECT_MAX_LENGTH 32 #define MEM_OBLECT_MAX_LENGTH 32
void RawDataHandler(char *argres, int rw) typedef struct
{ {
int operation = 0; int opertype;
int offset; int operphase;
//int offset;
int size; int size;
char mem_object[MEM_OBLECT_MAX_LENGTH]; char mem_object[MEM_OBLECT_MAX_LENGTH];
char filepath[FILE_PATH_MAX]; char filepath[FILE_PATH_MAX];
struct stat file_stat; struct stat file_stat;
FILE *f;
int open_file_timeout;
} file_transaction_t;
static file_transaction_t FileTransaction = {
.opertype = 0
};
void RawDataHandler(char *argres, int rw)
{
struct jReadElement result; struct jReadElement result;
jRead(argres, "", &result); jRead(argres, "", &result);
if (result.dataType != JREAD_OBJECT) if (result.dataType != JREAD_OBJECT)
@ -63,13 +72,13 @@ void RawDataHandler(char *argres, int rw)
return; return;
} }
jRead(argres, "{'operation'", &result); jRead(argres, "{'opertype'", &result);
if (result.elements == 1) if (result.elements == 1)
{ {
operation = atoi((char*) result.pValue); FileTransaction.opertype = atoi((char*) result.pValue);
if (operation < 1 || operation > 4) if (FileTransaction.opertype < 1 || FileTransaction.opertype > 3)
{ {
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:'operation' value not in [1...4]\""); snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:'operation' value not in [1...3]\"");
return; return;
} }
} }
@ -79,13 +88,29 @@ void RawDataHandler(char *argres, int rw)
return; return;
} }
jRead(argres, "{'operphase'", &result);
if (result.elements == 1)
{
FileTransaction.operphase = atoi((char*) result.pValue);
if (FileTransaction.operphase < 0 || FileTransaction.operphase > 3)
{
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:'file_operation' value not in [0...3]\"");
return;
}
}
else
{
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:key 'file_operation' not found\"");
return;
}
jRead(argres, "{'mem_object'", &result); jRead(argres, "{'mem_object'", &result);
if (result.elements == 1) if (result.elements == 1)
{ {
if (result.bytelen > 0 && result.bytelen < MEM_OBLECT_MAX_LENGTH) if (result.bytelen > 0 && result.bytelen < MEM_OBLECT_MAX_LENGTH)
{ {
memcpy(mem_object, (char*) result.pValue, result.bytelen); memcpy(FileTransaction.mem_object, (char*) result.pValue, result.bytelen);
mem_object[result.bytelen] = 0x00; FileTransaction.mem_object[result.bytelen] = 0x00;
} }
else else
{ {
@ -99,21 +124,10 @@ void RawDataHandler(char *argres, int rw)
return; return;
} }
jRead(argres, "{'offset'", &result);
if (result.elements == 1)
{
offset = atoi((char*) result.pValue);
}
else
{
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:key 'offset' not found\"");
return;
}
jRead(argres, "{'size'", &result); jRead(argres, "{'size'", &result);
if (result.elements == 1) if (result.elements == 1)
{ {
size = atoi((char*) result.pValue); FileTransaction.size = atoi((char*) result.pValue);
} }
else else
{ {
@ -122,46 +136,58 @@ void RawDataHandler(char *argres, int rw)
} }
//ESP_LOGI(TAG, "Got memory object %s, with offest %d and size %d", mem_object, offset, size); //ESP_LOGI(TAG, "Got memory object %s, with offest %d and size %d", mem_object, offset, size);
strcpy(filepath, dirpath); strcpy(FileTransaction.filepath, dirpath);
strcat(filepath, mem_object); strcat(FileTransaction.filepath, FileTransaction.mem_object);
if (operation == READ_ORERATION || operation == DELETE_ORERATION) if (FileTransaction.operphase == 1 || FileTransaction.operphase == 3)
{ {
if (stat(filepath, &file_stat) == -1) if (FileTransaction.opertype == READ_ORERATION || FileTransaction.opertype == DELETE_ORERATION)
{ {
ESP_LOGE("FILE_API", "File does not exist : %s", mem_object); if (stat(FileTransaction.filepath, &FileTransaction.file_stat) == -1)
{
ESP_LOGE("FILE_API", "File does not exist : %s", FileTransaction.mem_object);
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\""); snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\"");
return; return;
} }
} }
}
if (operation == DELETE_ORERATION) if (FileTransaction.opertype == DELETE_ORERATION)
{ {
unlink(filepath); unlink(FileTransaction.filepath);
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"DELETED OK\""); snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"DELETED OK\"");
return; return;
} }
else if (operation == READ_ORERATION) else if (FileTransaction.opertype == READ_ORERATION)
{ {
FILE *f = fopen(filepath, "r"); if (FileTransaction.operphase == 1 || FileTransaction.operphase == 3)
{
FileTransaction.f = fopen(FileTransaction.filepath, "r");
ESP_LOGI("FILE_API", "Open file for read : %s", FileTransaction.mem_object);
}
unsigned char *scr, *dst; unsigned char *scr, *dst;
size_t dlen, olen, slen; size_t dlen, olen, slen;
if (f == NULL) if (FileTransaction.f == NULL)
{ {
ESP_LOGE(TAG, "Failed to open file %s for writing", mem_object); ESP_LOGE(TAG, "Failed to open file %s for writing", FileTransaction.mem_object);
return; return;
} }
scr = (unsigned char*) malloc(size + 1); scr = (unsigned char*) malloc(FileTransaction.size + 1);
if (scr == NULL) if (scr == NULL)
{ {
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR: no memory to handle request\""); snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR: no memory to handle request\"");
return; return;
} }
fseek(f, offset, SEEK_SET);
int read = fread(scr, 1, size, f); int read = fread(scr, 1, FileTransaction.size, FileTransaction.f);
scr[read] = 0x00; scr[read] = 0x00;
slen = read; slen = read;
fclose(f); if (FileTransaction.operphase == 2 || FileTransaction.operphase == 3)
{
fclose(FileTransaction.f);
ESP_LOGI("FILE_API", "Close file for read : %s", FileTransaction.mem_object);
}
dlen = 0; dlen = 0;
mbedtls_base64_encode(NULL, dlen, &olen, scr, slen); mbedtls_base64_encode(NULL, dlen, &olen, scr, slen);
dst = (unsigned char*) malloc(olen); dst = (unsigned char*) malloc(olen);
@ -170,49 +196,52 @@ void RawDataHandler(char *argres, int rw)
struct jWriteControl jwc; struct jWriteControl jwc;
jwOpen(&jwc, argres, VAR_MAX_VALUE_LENGTH, JW_OBJECT, JW_COMPACT); jwOpen(&jwc, argres, VAR_MAX_VALUE_LENGTH, JW_OBJECT, JW_COMPACT);
jwObj_int(&jwc, "operation", operation); jwObj_int(&jwc, "operation", FileTransaction.opertype);
jwObj_string(&jwc, "mem_object", mem_object); jwObj_string(&jwc, "mem_object", FileTransaction.mem_object);
jwObj_int(&jwc, "offset", offset);
jwObj_int(&jwc, "size", read); jwObj_int(&jwc, "size", read);
jwObj_string(&jwc, "dat", (char*) dst); jwObj_string(&jwc, "dat", (char*) dst);
jwClose(&jwc); jwClose(&jwc);
free(scr); free(scr);
free(dst); free(dst);
} }
else if (operation == APPEND_ORERATION || operation == REPLACE_ORERATION) else if (FileTransaction.opertype == WRITE_ORERATION)
{ {
jRead(argres, "{'dat'", &result); jRead(argres, "{'dat'", &result);
if (result.elements == 1) if (result.elements == 1)
{ {
if (result.bytelen > 0 && result.bytelen <= size) if (result.bytelen > 0 && result.bytelen <= FileTransaction.size)
{ {
unsigned char *dst; unsigned char *dst;
size_t dlen, olen; size_t dlen, olen;
FILE *f; if (FileTransaction.operphase == 1 || FileTransaction.operphase == 3)
if (operation == REPLACE_ORERATION)
f = fopen(filepath, "w");
else
f = fopen(filepath, "a");
if (f == NULL)
{ {
ESP_LOGE(TAG, "Failed to open file %s for writing", mem_object); FileTransaction.f = fopen(FileTransaction.filepath, "a");
ESP_LOGI("FILE_API", "Open file for write : %s", FileTransaction.mem_object);
}
if (FileTransaction.f == NULL)
{
ESP_LOGE(TAG, "Failed to open file %s for writing", FileTransaction.mem_object);
return; return;
} }
fseek(f, offset, SEEK_SET);
dlen = 0; dlen = 0;
mbedtls_base64_decode(NULL, dlen, &olen, (unsigned char*) result.pValue, (size_t) result.bytelen); mbedtls_base64_decode(NULL, dlen, &olen, (unsigned char*) result.pValue, (size_t) result.bytelen);
dst = (unsigned char*) malloc(olen); dst = (unsigned char*) malloc(olen);
dlen = olen; dlen = olen;
mbedtls_base64_decode(dst, dlen, &olen, (unsigned char*) result.pValue, (size_t) result.bytelen); mbedtls_base64_decode(dst, dlen, &olen, (unsigned char*) result.pValue, (size_t) result.bytelen);
int write = fwrite((char*) dst, olen, 1, f); //ESP_LOGI("FILE_API", "File write operation BEGIN");
fclose(f); int write = fwrite((char*) dst, olen, 1, FileTransaction.f);
//ESP_LOGI("FILE_API", "File write operation END");
if (FileTransaction.operphase == 2 || FileTransaction.operphase == 3)
{
fclose(FileTransaction.f);
ESP_LOGI("FILE_API", "Close file for write : %s", FileTransaction.mem_object);
}
free(dst); free(dst);
struct jWriteControl jwc; struct jWriteControl jwc;
jwOpen(&jwc, argres, VAR_MAX_VALUE_LENGTH, JW_OBJECT, JW_COMPACT); jwOpen(&jwc, argres, VAR_MAX_VALUE_LENGTH, JW_OBJECT, JW_COMPACT);
jwObj_int(&jwc, "operation", operation); jwObj_int(&jwc, "operation", FileTransaction.opertype);
jwObj_string(&jwc, "mem_object", mem_object); jwObj_string(&jwc, "mem_object", FileTransaction.mem_object);
jwObj_int(&jwc, "offset", offset);
jwObj_int(&jwc, "size", write); jwObj_int(&jwc, "size", write);
jwClose(&jwc); jwClose(&jwc);
} }

View File

@ -407,85 +407,13 @@ static void funct_file_list(char *argres, int rw)
} }
jwArr_object(&jwc); jwArr_object(&jwc);
jwObj_raw(&jwc, "sel", "false");
jwObj_string(&jwc, "name", (char*) entry->d_name); jwObj_string(&jwc, "name", (char*) entry->d_name);
jwObj_string(&jwc, "type", entrytype);
jwObj_int(&jwc, "size", entry_stat.st_size); jwObj_int(&jwc, "size", entry_stat.st_size);
jwEnd(&jwc); jwEnd(&jwc);
} }
jwClose(&jwc); jwClose(&jwc);
}
static void funct_file_delete(char *argres, int rw)
{
char filepath[FILE_PATH_MAX];
struct stat file_stat;
const char *filename = argres;
if (!filename)
{
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\"");
return;
}
if (filename[strlen(filename) - 1] == '/')
{
ESP_LOGE("FILE_API", "Invalid filename : %s", filename);
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\"");
return;
}
strcpy(filepath, dirpath);
strcat(filepath, filename);
ESP_LOGI("FILE_API", " filepath to delete : %s", filepath);
if (stat(filepath, &file_stat) == -1)
{
ESP_LOGE("FILE_API", "File does not exist : %s", filename);
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\"");
/* Respond with 400 Bad Request */
return;
}
unlink(filepath);
}
static void funct_file_get(char *argres, int rw)
{
char filepath[FILE_PATH_MAX];
struct stat file_stat;
const char *filename = argres;
if (!filename)
{
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\"");
return;
}
if (filename[strlen(filename) - 1] == '/')
{
ESP_LOGE("FILE_API", "Invalid filename : %s", filename);
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\"");
return;
}
strcpy(filepath, dirpath);
strcat(filepath, filename);
ESP_LOGI("FILE_API", " filepath to delete : %s", filepath);
if (stat(filepath, &file_stat) == -1)
{
ESP_LOGE("FILE_API", "File does not exist : %s", filename);
snprintf(argres, VAR_MAX_VALUE_LENGTH, "\"ERROR:DIR_NOT_FOUND\"");
/* Respond with 400 Bad Request */
return;
}
}
static void funct_file_put(char *argres, int rw)
{
} }
static void funct_raw_data(char *argres, int rw) static void funct_raw_data(char *argres, int rw)
@ -680,10 +608,6 @@ const rest_var_t SystemVariables[] =
{ 0, "objsinfo", &funct_objsinfo, VAR_FUNCT, R, 0, 0 }, { 0, "objsinfo", &funct_objsinfo, VAR_FUNCT, R, 0, 0 },
{ 0, "file_list", &funct_file_list, VAR_FUNCT, R, 0, 0 }, { 0, "file_list", &funct_file_list, VAR_FUNCT, R, 0, 0 },
{ 0, "file_delete", &funct_file_delete, VAR_FUNCT, R, 0, 0 },
{ 0, "file_get", &funct_file_get, VAR_FUNCT, R, 0, 0 },
{ 0, "file_put", &funct_file_put, VAR_FUNCT, R, 0, 0 },
{ 0, "raw_data", &funct_raw_data, VAR_FUNCT, R, 0, 0 }, { 0, "raw_data", &funct_raw_data, VAR_FUNCT, R, 0, 0 },
}; };