Compare commits
1 Commits
flatbuffer
...
ws-server
| Author | SHA1 | Date | |
|---|---|---|---|
| 165a96aa3c |
34
Makefile
34
Makefile
@@ -1,10 +1,12 @@
|
||||
OPENSSL_INCLUDE ?= /usr/include/openssl/
|
||||
BUILD_DIR=out
|
||||
SRC_DIR=src
|
||||
INC_DIR=inc
|
||||
CC=gcc
|
||||
OBJDUMP=objdump
|
||||
LIBS=deps/tomlc99/libtoml.a deps/ptQueue/out/ptQueue.a deps/wsServer/libws.a
|
||||
LIBS_HEADERS=deps/
|
||||
LIBS_HEADERS=deps/ $(OPENSSL_INCLUDE)
|
||||
STATIC_LIBS=crypto
|
||||
STANDART=c23
|
||||
OPTIMIZE=-Og
|
||||
TARGET=main
|
||||
@@ -12,38 +14,41 @@ TARGET=main
|
||||
|
||||
DEFINES=#-DOPCODE_WORDSIZE=2 -DMEM_CELL_WORDS=1 -DPC_WORDSIZE=2 -DGP_REG_CELL_WORDS=1 -DIO_REG_CELL_WORDS=1
|
||||
|
||||
C_SOURCES=$(wildcard $(SRC_DIR)/*.c)
|
||||
# C_SOURCES=$(wildcard $(SRC_DIR)/*.c)
|
||||
C_HEADERS=$(wildcard $(INC_DIR)/*.h)
|
||||
C_SOURCES := $(shell find $(SRC_DIR) -type f -name '*.c')
|
||||
|
||||
C_INCLUDES=-I$(INC_DIR)/ $(addprefix -I,$(LIBS_HEADERS))
|
||||
|
||||
C_DEFS=-D_POSIX_C_SOURCE=199309L
|
||||
|
||||
DISABLE_FLAGS=-Wno-unused-variable -Wno-unused-parameter -Wno-write-strings -Wno-pointer-arith
|
||||
DISABLE_FLAGS=-Wno-unused-variable -Wno-unused-parameter -Wno-write-strings -Wno-pointer-arith -Wno-analyzer-infinite-loop
|
||||
PEDANTIC_FLAGS=-pedantic -pedantic-errors $(DISABLE_FLAGS) -Wall -Wcast-align -Wcast-qual -Wconversion -Wduplicated-branches -Wduplicated-cond -Werror -Wextra -Wfloat-equal -Wlogical-op -Wpedantic -Wredundant-decls -Wsign-conversion
|
||||
ANALYZER_FLAGS=-fanalyzer -fdiagnostics-show-option -fdiagnostics-color=always
|
||||
LSECTIONS=-ffunction-sections -fdata-sections -Wl,--gc-sections
|
||||
CFLAGS=$(C_DEFS) -g $(C_INCLUDES) $(DEFINES) $(OPTIMIZE) --std=$(STANDART) $(PEDANTIC_FLAGS) $(ANALYZER_FLAGS)
|
||||
LFLAGS=$(OPTIMIZE) -g $(PEDANTIC_FLAGS) $(DEFINES) -flto -fuse-linker-plugin $(LSECTIONS) -lm
|
||||
CFLAGS=$(C_DEFS) -g $(C_INCLUDES) $(DEFINES) $(OPTIMIZE) --std=$(STANDART) $(PEDANTIC_FLAGS) $(ANALYZER_FLAGS) -MMD -MP
|
||||
STATICLIBS_FLAGS=$(addprefix -l,$(STATIC_LIBS))
|
||||
LFLAGS=$(OPTIMIZE) -g $(PEDANTIC_FLAGS) $(DEFINES) $(STATICLIBS_FLAGS) -flto -fuse-linker-plugin $(LSECTIONS) -lm
|
||||
|
||||
|
||||
|
||||
OBJECTS = $(addprefix $(BUILD_DIR)/,$(notdir $(C_SOURCES:.c=.o))) $(LIBS)
|
||||
OBJECTS = $(patsubst $(SRC_DIR)/%.c, $(BUILD_DIR)/%.o, $(C_SOURCES))
|
||||
|
||||
DEP = $(filter %.d, $(OBJECTS:.o=.d))
|
||||
$(info $(DEP))
|
||||
|
||||
OBJECTS += $(LIBS)
|
||||
vpath %.c $(sort $(dir $(C_SOURCES)))
|
||||
vpath %.h $(sort $(dir $(C_HEADERS)))
|
||||
|
||||
# $(info SOURCES=$(C_SOURCES))
|
||||
# $(info HEADERS=$(C_HEADERS))
|
||||
# $(info OBJECTS=$(OBJECTS))
|
||||
|
||||
|
||||
all: build
|
||||
|
||||
build: date deps Dir $(C_HEADERS) target compile_commands
|
||||
build: date deps Dir target compile_commands
|
||||
|
||||
rebuild: clean | build
|
||||
|
||||
$(info $(DEP))
|
||||
|
||||
-include $(DEP)
|
||||
|
||||
$(BUILD_DIR)/%.o: %.c Makefile | $(BUILD_DIR)
|
||||
@echo -e '\033[1;32mCC\t'$<'\t->\t'$@'\033[0m'
|
||||
@$(CC) -c $(CFLAGS) -Wa,-a,-ad,-alms=$(BUILD_DIR)/$(notdir $(<:.c=.lst)) $< -o $@
|
||||
@@ -62,6 +67,7 @@ deps:
|
||||
|
||||
BuildDir:
|
||||
@mkdir -p $(BUILD_DIR)
|
||||
$(shell mkdir -p $(dir $(OBJECTS)))
|
||||
|
||||
SrcDir:
|
||||
@mkdir -p $(SRC_DIR)
|
||||
|
||||
Submodule hmmmm_scripts deleted from 9cc7b36cbb
@@ -1,21 +0,0 @@
|
||||
#ifndef __BASE_DEVICE_H__
|
||||
#define __BASE_DEVICE_H__
|
||||
|
||||
#include "hmmmm.h"
|
||||
|
||||
typedef struct {
|
||||
void* specs;
|
||||
device_lib_t* lib;
|
||||
device_public_context_t* ctx;
|
||||
uint64_t clockCycleLimit;
|
||||
uint64_t clockCycleCounter;
|
||||
} device_handle_t;
|
||||
|
||||
|
||||
device_handle_t* openBaseDeviceFromConfig(const char* configPath, char* errbuf);
|
||||
device_handle_t* openBaseDevice(conf_dev_t* devConf, char* errbuf);
|
||||
conf_dev_t* openBaseDeviceConfig(const char* configPath, char* errbuf);
|
||||
void closeBaseDevice(device_handle_t* devHandle);
|
||||
|
||||
|
||||
#endif // ifndef __BASE_DEVICE_H__
|
||||
26
inc/client.h
Normal file
26
inc/client.h
Normal file
@@ -0,0 +1,26 @@
|
||||
#ifndef __CLIENT_H__
|
||||
#define __CLIENT_H__
|
||||
|
||||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "sized_ptr.h"
|
||||
|
||||
#include "wsServer/include/ws.h"
|
||||
#include "ptQueue/inc/ptQueue.h"
|
||||
|
||||
typedef struct {
|
||||
ws_cli_conn_t clientId;
|
||||
uint64_t seatId;
|
||||
uint8_t isAuthed;
|
||||
ptQueue* incomeQ;
|
||||
// ptQueue* outcomeQ;
|
||||
uint64_t connectedAt;
|
||||
uint32_t streamRegIterator;
|
||||
uint64_t orphanedAt;
|
||||
uint64_t orphanedDeadTimeout;
|
||||
SizedPtr* fallbackOutcomeQ;
|
||||
size_t fallbackOutcomeQPadding;
|
||||
} ClientContext;
|
||||
|
||||
#endif //ifndef __CLIENT_H__
|
||||
@@ -1,41 +0,0 @@
|
||||
#ifndef __COMPOSE_DEVICE_H__
|
||||
#define __COMPOSE_DEVICE_H__
|
||||
#include "base_device.h"
|
||||
|
||||
typedef struct {
|
||||
uint64_t limiter;
|
||||
} clock_conf_t;
|
||||
|
||||
typedef struct {
|
||||
char** baseAt;
|
||||
char* baseSeg;
|
||||
char** target;
|
||||
uint64_t projectionShift;
|
||||
|
||||
} projection_conf_t;
|
||||
|
||||
typedef struct {
|
||||
|
||||
char** baseAt;
|
||||
char** pointAt;
|
||||
uint64_t addr;
|
||||
char* seg;
|
||||
|
||||
} intercept_conf_t;
|
||||
|
||||
|
||||
typedef struct {
|
||||
clock_conf_t clockConf;
|
||||
conf_dev_t** baseConfigs;
|
||||
projection_conf_t** projections;
|
||||
intercept_conf_t** intercepts;
|
||||
} compose_dev_conf_t;
|
||||
|
||||
typedef struct {
|
||||
device_handle_t** devHandlers;
|
||||
} compose_dev_handle_t;
|
||||
|
||||
compose_dev_conf_t* openComposeDeviceConfig(const char* configPath, char* errbuf);
|
||||
device_handle_t** openComposeDevice(compose_dev_conf_t* conf, char* errbuf);
|
||||
|
||||
#endif // ifndef __COMPOSE_DEVICE_H__
|
||||
36
inc/config.h
36
inc/config.h
@@ -1,36 +0,0 @@
|
||||
#ifndef __HMMMM_CONFIG_H__
|
||||
#define __HMMMM_CONFIG_H__
|
||||
#include <unistd.h>
|
||||
#include <stdint.h>
|
||||
|
||||
|
||||
typedef struct {
|
||||
char* name;
|
||||
size_t start;
|
||||
size_t len;
|
||||
uint8_t wordLen;
|
||||
uint8_t isExecutable;
|
||||
} conf_mem_seg_t;
|
||||
|
||||
typedef struct {
|
||||
conf_mem_seg_t** memSegConfs;
|
||||
} conf_mem_t;
|
||||
|
||||
typedef struct {
|
||||
char** id;
|
||||
char** clockId;
|
||||
uint64_t clockDivider;
|
||||
uint64_t clockMultipler;
|
||||
conf_mem_t* memConf;
|
||||
char* libPath;
|
||||
} conf_dev_t;
|
||||
|
||||
|
||||
|
||||
void freeMemSegConf(conf_mem_seg_t* memSegConf);
|
||||
void freeMemConf(conf_mem_t* memConf);
|
||||
void freeConf(conf_dev_t* conf);
|
||||
void freeComposeId(char** id);
|
||||
uint8_t compareComposeId(char** idA, char** idB);
|
||||
|
||||
#endif // ifndef __HMMMM_CONFIG_H__
|
||||
46
inc/context.h
Normal file
46
inc/context.h
Normal file
@@ -0,0 +1,46 @@
|
||||
#ifndef __CONTEXT_H__
|
||||
#define __CONTEXT_H__
|
||||
|
||||
#include <pthread.h>
|
||||
|
||||
#include "ptQueue/inc/ptQueue.h"
|
||||
#include "wsServer/include/ws.h"
|
||||
|
||||
#include "linkedlist.h"
|
||||
#include "sized_ptr.h"
|
||||
|
||||
#include "streamed.h"
|
||||
|
||||
typedef struct {
|
||||
SizedPtr* bufs;
|
||||
uint8_t buffersCount;
|
||||
_Atomic (uint8_t) readRequestIdx;
|
||||
_Atomic (uint8_t) currWritingIdx;
|
||||
} OutgoingBuffers;
|
||||
|
||||
|
||||
|
||||
|
||||
typedef struct {
|
||||
uint8_t* emulState;
|
||||
uint64_t* clockCounter;
|
||||
LinkedListEntry** clientsHead;
|
||||
uint8_t* utilizedFlag;
|
||||
OutgoingBuffers* outBufs;
|
||||
DeviceSegStreamReg** deviceStreamRegs;
|
||||
uint8_t** devicesMem;
|
||||
size_t devicesCount;
|
||||
} EmulContext;
|
||||
|
||||
|
||||
typedef struct {
|
||||
pthread_mutex_t registerMutex;
|
||||
ptQueue* regQueue;
|
||||
uint8_t* accessToken;
|
||||
EmulContext* emulContext;
|
||||
_Atomic (uint64_t) seatCounter;
|
||||
} ServerContext;
|
||||
|
||||
|
||||
|
||||
#endif //ifndef __CONTEXT_H__
|
||||
13
inc/events.h
Normal file
13
inc/events.h
Normal file
@@ -0,0 +1,13 @@
|
||||
#ifndef __EVENTS_H__
|
||||
#define __EVENTS_H__
|
||||
|
||||
#include <stdint.h>
|
||||
#include "context.h"
|
||||
|
||||
typedef struct {
|
||||
uint8_t regType;
|
||||
ClientContext* ctx;
|
||||
} ClientRegistrationEvent;
|
||||
|
||||
|
||||
#endif //ifndef __EVENTS_H__
|
||||
33
inc/hmmmm.h
33
inc/hmmmm.h
@@ -1,33 +0,0 @@
|
||||
#ifndef __HMMMMM__
|
||||
#define __HMMMMM__
|
||||
#include "libhmmmm.h"
|
||||
#include "config.h"
|
||||
|
||||
typedef struct {
|
||||
void* (*extractPcounterPtr)(device_public_context_t* devContext);
|
||||
|
||||
size_t (*extractPcounter)(device_public_context_t* devContext);
|
||||
size_t (*extractOpcode)(device_mem_t* devMem, size_t _programCounter);
|
||||
uint8_t (*extractPcounterSizeWords)();
|
||||
|
||||
|
||||
} instruction_simul_handlers_t;
|
||||
|
||||
typedef struct {
|
||||
uint8_t devType;
|
||||
device_public_context_t* (*init)(void*, char* errbuf);
|
||||
uint8_t (*makeDeviceTick)(device_public_context_t* devInfo);
|
||||
void* extendedHandlers;
|
||||
device_public_context_t* devContext;
|
||||
void* _dlhandl;
|
||||
void* (*parseSpecsFromConfig)(const conf_dev_t* devConf, char* errbuf);
|
||||
void (*freeSpecs)(void* specs);
|
||||
void (*freeDevMem)(device_mem_t* mem);
|
||||
void (*fillSmartReadSpecs)(void* specs, smart_read_spec_t* smartReadSpecs, uint64_t smartReadSpecsCount);
|
||||
void (*fillSmartWriteSpecs)(void* specs, smart_write_spec_t* smartWriteSpecs, uint64_t smartWriteSpecsCount);
|
||||
} device_lib_t;
|
||||
|
||||
device_lib_t* loadDeviceLib(const char *libpath, char* errbuf);
|
||||
|
||||
|
||||
#endif // ifndef __HMMMMM__
|
||||
@@ -1,37 +0,0 @@
|
||||
#ifndef __LIBDEVICE_H__
|
||||
#define __LIBDEVICE_H__
|
||||
|
||||
#include <stdint.h>
|
||||
#include "libmem.h"
|
||||
|
||||
#define SMART_ADDR_TYPE_GLOBAL 1
|
||||
#define SMART_ADDR_TYPE_SEGMENTED 2
|
||||
typedef struct
|
||||
{
|
||||
uint64_t addr;
|
||||
uint8_t segno;
|
||||
uint64_t localAddr;
|
||||
uint8_t addrType;
|
||||
mem_h_read_func* handler;
|
||||
uint64_t ident;
|
||||
} smart_read_spec_t;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
uint64_t addr;
|
||||
uint8_t segno;
|
||||
uint64_t localAddr;
|
||||
uint8_t addrType;
|
||||
mem_h_write_func handler;
|
||||
uint64_t ident;
|
||||
} smart_write_spec_t;
|
||||
|
||||
typedef struct {
|
||||
device_mem_t* deviceMem;
|
||||
void* deviceInfo;
|
||||
} device_public_context_t;
|
||||
|
||||
|
||||
|
||||
#endif // ifndef __LIBDEVICE_H__
|
||||
|
||||
@@ -1,14 +0,0 @@
|
||||
#ifndef __LIB_HMMMMM__
|
||||
#define __LIB_HMMMMM__
|
||||
|
||||
#include "libmem.h"
|
||||
#include "libdevice.h"
|
||||
|
||||
#define EXTENDED_DEVICE_TYPE_DUMMY 0
|
||||
#define EXTENDED_DEVICE_TYPE_INSTR_SIMUL 1
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#endif // ifndef __LIB_HMMMMM__
|
||||
52
inc/libmem.h
52
inc/libmem.h
@@ -1,52 +0,0 @@
|
||||
#ifndef __LIBMEM_H__
|
||||
#define __LIBMEM_H__
|
||||
|
||||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#define GET_BIT(n, b) ((n >> b) & 1)
|
||||
|
||||
// Internal mem handlers
|
||||
typedef void* (*mem_h_read_func)(uint64_t ident, uint64_t addr, void* rawCells);
|
||||
typedef void (*mem_h_write_func)(uint64_t ident, uint64_t addr, void* rawCells, void* data);
|
||||
|
||||
typedef struct {
|
||||
mem_h_read_func func;
|
||||
uint64_t ident;
|
||||
} mem_h_read_handler;
|
||||
typedef struct {
|
||||
mem_h_write_func func;
|
||||
uint64_t ident;
|
||||
} mem_h_write_handler;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
void* rawCells;
|
||||
void** cells;
|
||||
uint64_t smartAddrReadMask;
|
||||
uint64_t smartAddrWriteMask;
|
||||
mem_h_read_handler* smartAddrReadHandlers;
|
||||
mem_h_write_handler* smartAddrWriteHandlers;
|
||||
char** memsegNames;
|
||||
uint64_t* memsegShifts;
|
||||
uint64_t* memreadCellAddrs;
|
||||
uint64_t* memwriteCellAddrs;
|
||||
uint8_t memreadLen;
|
||||
uint8_t memwriteLen;
|
||||
} device_mem_t;
|
||||
|
||||
|
||||
typedef struct
|
||||
{
|
||||
uint64_t start;
|
||||
uint64_t len;
|
||||
uint8_t wordLen;
|
||||
char* name;
|
||||
} memseg_spec_t;
|
||||
|
||||
|
||||
// External handlers
|
||||
typedef void* (*ext_h_read_func)(uint64_t addr, void* rawCells, void* devContext);
|
||||
typedef void (*ext_h_write_func)(uint64_t addr, void* rawCells, void* data, void* devContext);
|
||||
|
||||
#endif // ifndef __LIBMEM_H__
|
||||
13
inc/linkedlist.h
Normal file
13
inc/linkedlist.h
Normal file
@@ -0,0 +1,13 @@
|
||||
#ifndef __LINKEDLIST_H__
|
||||
#define __LINKEDLIST_H__
|
||||
|
||||
|
||||
typedef struct LinkedListEntry {
|
||||
struct LinkedListEntry* prevEntry;
|
||||
struct LinkedListEntry* nextEntry;
|
||||
void* payload;
|
||||
} LinkedListEntry;
|
||||
|
||||
void removeLinkedListEntry(LinkedListEntry** head, LinkedListEntry* entry);
|
||||
|
||||
#endif //ifndef __LINKEDLIST_H__
|
||||
42
inc/panic.h
Normal file
42
inc/panic.h
Normal file
@@ -0,0 +1,42 @@
|
||||
#ifndef __PANIC_H__
|
||||
#define __PANIC_H__
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <execinfo.h>
|
||||
|
||||
#ifndef PANIC_FD
|
||||
#define PANIC_FD stderr
|
||||
#endif //ifndef PANIC_FD
|
||||
|
||||
#ifndef PANIC_STRACE_LEN
|
||||
#define PANIC_STRACE_LEN 32
|
||||
#endif //ifndef PANIC_STRACE_LEN
|
||||
|
||||
void print_stacktrace(FILE* fd);
|
||||
|
||||
#define panic(...) \
|
||||
do { \
|
||||
fprintf(PANIC_FD, "Panic at %s:%d\n", __FILE__, __LINE__); \
|
||||
fprintf(PANIC_FD, "Panic message:\n\t" __VA_ARGS__); \
|
||||
print_stacktrace(PANIC_FD); \
|
||||
abort(); \
|
||||
} while(0)
|
||||
|
||||
#define NULL_GUARD(ptr, ...) \
|
||||
do { \
|
||||
if((ptr) == NULL) \
|
||||
{ \
|
||||
if(sizeof(#__VA_ARGS__) > 1) \
|
||||
{ \
|
||||
panic(__VA_ARGS__); \
|
||||
} \
|
||||
else \
|
||||
{ \
|
||||
panic("NULL detected in %s", #ptr); \
|
||||
} \
|
||||
} \
|
||||
} while(0)
|
||||
|
||||
|
||||
#endif //ifndef __PANIC_H__
|
||||
11
inc/proto/dial.h
Normal file
11
inc/proto/dial.h
Normal file
@@ -0,0 +1,11 @@
|
||||
#ifndef __PROTO_DIAL_H__
|
||||
#define __PROTO_DIAL_H__
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "context.h"
|
||||
|
||||
void broadcastClients(EmulContext* emulContext, uint8_t* msg, size_t msgLen);
|
||||
void dispatchOutgoingMessage(OutgoingBuffers* outBufs, ClientContext* client, uint8_t* msg, size_t msgLen);
|
||||
|
||||
#endif
|
||||
37
inc/proto/enums.h
Normal file
37
inc/proto/enums.h
Normal file
@@ -0,0 +1,37 @@
|
||||
#ifndef __PROTO_ENUMS_H__
|
||||
#define __PROTO_ENUMS_H__
|
||||
|
||||
|
||||
#define REG_EVTYPE_CONNECT 1
|
||||
#define REG_EVTYPE_AUTH 2
|
||||
#define REG_EVTYPE_CLOSE 3
|
||||
|
||||
|
||||
|
||||
|
||||
#define PACKET_TYPE_CTRL 0b0001
|
||||
#define PACKET_TYPE_STREAM 0b0010
|
||||
#define PACKET_TYPE_MEM 0b0011
|
||||
|
||||
|
||||
#define CTRL_TYPE_EXEC 0b0001
|
||||
#define CTRL_TYPE_NOTIF_STATE 0b0010
|
||||
#define CTRL_TYPE_LIST_ORPHANED 0b1001
|
||||
#define CTRL_TYPE_LOAD_FAILED 0b1010
|
||||
#define CTRL_TYPE_SETUP_CONNECTION 0b1011
|
||||
|
||||
#define NOTIF_TYPE_EXEC 0b0000
|
||||
|
||||
#define STREAM_TYPE_REG_REQUEST 0b001
|
||||
#define STREAM_TYPE_REG_DISCARD 0b011
|
||||
#define STREAM_TYPE_REG_CONFIRM 0b101
|
||||
#define STREAM_TYPE_SEND 0b000
|
||||
|
||||
#define MEM_TYPE_READ_REQ 0b0000
|
||||
#define MEM_TYPE_READ_RESP 0b0001
|
||||
#define MEM_TYPE_WRITE_PUSH 0b0010
|
||||
|
||||
|
||||
|
||||
|
||||
#endif //ifndef __PROTO_ENUMS_H__
|
||||
11
inc/proto/handlers.h
Normal file
11
inc/proto/handlers.h
Normal file
@@ -0,0 +1,11 @@
|
||||
#ifndef __PROTO_HANDLERS_H__
|
||||
#define __PROTO_HANDLERS_H__
|
||||
|
||||
#include "events.h"
|
||||
|
||||
|
||||
void handleRegEvent(EmulContext* emulContext, ClientRegistrationEvent* ev);
|
||||
void handleAllClients(EmulContext* emulContext);
|
||||
|
||||
|
||||
#endif //ifndef __PROTO_HANDLERS_H__
|
||||
11
inc/proto/handlers/auth.h
Normal file
11
inc/proto/handlers/auth.h
Normal file
@@ -0,0 +1,11 @@
|
||||
#ifndef __PROTO_HANDLERS_AUTH_H__
|
||||
#define __PROTO_HANDLERS_AUTH_H__
|
||||
|
||||
#include <stdint.h>
|
||||
#include "context.h"
|
||||
|
||||
uint8_t handle_auth(ClientContext* cctx, ws_cli_conn_t client, const uint8_t* msg, uint64_t msgSize, int msgType);
|
||||
LinkedListEntry* disconnectDueTimeout(EmulContext* emulContext, LinkedListEntry* clientEntry);
|
||||
void handleOnClientAuthDone(ClientContext* ctx, EmulContext* emulContext);
|
||||
|
||||
#endif //ifndef __PROTO_HANDLERS_AUTH_H__
|
||||
10
inc/proto/handlers/control.h
Normal file
10
inc/proto/handlers/control.h
Normal file
@@ -0,0 +1,10 @@
|
||||
#ifndef __PROTO_HANDLERS_CONTROL_H__
|
||||
#define __PROTO_HANDLERS_CONTROL_H__
|
||||
|
||||
#include "proto/msg.h"
|
||||
#include "proto/dial.h"
|
||||
|
||||
void handleIncomingControlMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext);
|
||||
|
||||
|
||||
#endif //ifndef __PROTO_HANDLERS_CONTROL_H__
|
||||
10
inc/proto/handlers/mem.h
Normal file
10
inc/proto/handlers/mem.h
Normal file
@@ -0,0 +1,10 @@
|
||||
#ifndef __PROTO_HANDLERS_MEM_H__
|
||||
#define __PROTO_HANDLERS_MEM_H__
|
||||
|
||||
#include "proto/msg.h"
|
||||
#include "proto/dial.h"
|
||||
|
||||
void handleIncomingMemMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext);
|
||||
|
||||
#endif //ifndef __PROTO_HANDLERS_MEM_H__
|
||||
|
||||
12
inc/proto/handlers/stream.h
Normal file
12
inc/proto/handlers/stream.h
Normal file
@@ -0,0 +1,12 @@
|
||||
#ifndef __PROTO_HANDLERS_STREAM_H__
|
||||
#define __PROTO_HANDLERS_STREAM_H__
|
||||
|
||||
#include "proto/msg.h"
|
||||
#include "proto/dial.h"
|
||||
|
||||
|
||||
void unregisterClientStream(EmulContext* emulContext, ClientContext* ctx, uint32_t regId);
|
||||
void unregisterClientStreams(EmulContext* emulContext, ClientContext* ctx);
|
||||
void handleIncomingStreamMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext);
|
||||
|
||||
#endif //ifndef __PROTO_HANDLERS_STREAM_H__
|
||||
10
inc/proto/handlers/ws.h
Normal file
10
inc/proto/handlers/ws.h
Normal file
@@ -0,0 +1,10 @@
|
||||
#ifndef __PROTO_HANDLERS_WS_H__
|
||||
#define __PROTO_HANDLERS_WS_H__
|
||||
|
||||
#include "wsServer/include/ws.h"
|
||||
|
||||
void onWsMessage(ws_cli_conn_t client, const unsigned char *msg, uint64_t size, int type);
|
||||
void onWsClose(ws_cli_conn_t client);
|
||||
void onWsOpen(ws_cli_conn_t client);
|
||||
|
||||
#endif //ifndef __PROTO_HANDLERS_WS_H__
|
||||
29
inc/proto/msg.h
Normal file
29
inc/proto/msg.h
Normal file
@@ -0,0 +1,29 @@
|
||||
#ifndef __PROTO_MSG_H__
|
||||
#define __PROTO_MSG_H__
|
||||
|
||||
#include "wsServer/include/ws.h"
|
||||
#include <stdlib.h>
|
||||
#include "client.h"
|
||||
|
||||
typedef struct {
|
||||
ClientContext* client;
|
||||
uint8_t* msg;
|
||||
size_t msgLen;
|
||||
} OutgoingMessage;
|
||||
|
||||
typedef struct {
|
||||
uint64_t nonce;
|
||||
uint8_t packetType;
|
||||
uint8_t payloadHeader;
|
||||
const void* payload;
|
||||
size_t payloadLen;
|
||||
} BaseMessage;
|
||||
|
||||
BaseMessage* parseMessage(const uint8_t* bytes, size_t size);
|
||||
void fillHead(uint64_t nonce, uint8_t packetType, uint8_t payloadHeader, uint8_t* outmsg);
|
||||
uint8_t* createControlNotifyMessage(uint64_t nonce, uint64_t clockCounter, uint8_t newEmulState, size_t* lenOut);
|
||||
uint8_t* createDoneRegMessage(uint64_t nonce, uint8_t X, uint64_t devId, uint64_t segId, uint64_t startAddr, uint64_t segLength, uint32_t regId, size_t* lenOut);
|
||||
uint8_t* createStreamSegmentPush(uint8_t mode, uint32_t regId, uint64_t clockCounter, uint8_t* payload, size_t payloadLen, size_t* lenOut);
|
||||
uint8_t* createClientSetup(uint64_t nonce, ClientContext* ctx, size_t* lenOut);
|
||||
|
||||
#endif //ifndef __PROTO_MSG_H__
|
||||
30
inc/proto/pack.h
Normal file
30
inc/proto/pack.h
Normal file
@@ -0,0 +1,30 @@
|
||||
#ifndef __PROTO_PACK_H__
|
||||
#define __PROTO_PACK_H__
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
|
||||
|
||||
uint64_t decodeBytesToU64(const uint8_t* bytes);
|
||||
uint32_t decodeBytesToU32(const uint8_t* bytes);
|
||||
uint16_t decodeBytesToU16(const uint8_t* bytes);
|
||||
|
||||
|
||||
|
||||
void encodeUint8ToBytes(uint8_t num, uint8_t* tgt);
|
||||
void encodeUint16ToBytes(uint16_t num, uint8_t* tgt);
|
||||
void encodeUint32ToBytes(uint32_t num, uint8_t* tgt);
|
||||
void encodeUint64ToBytes(uint64_t num, uint8_t* tgt);
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#define encodeUintToBytes(num, tgt) _Generic((num), \
|
||||
uint8_t: encodeUint8ToBytes, \
|
||||
uint16_t: encodeUint16ToBytes, \
|
||||
uint32_t: encodeUint32ToBytes, \
|
||||
uint64_t: encodeUint64ToBytes \
|
||||
)(num, tgt)
|
||||
|
||||
#endif //ifndef __PROTO_PACK_H__
|
||||
13
inc/sized_ptr.h
Normal file
13
inc/sized_ptr.h
Normal file
@@ -0,0 +1,13 @@
|
||||
#ifndef __SIZED_PTR_H__
|
||||
#define __SIZED_PTR_H__
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
typedef struct {
|
||||
void* ptr;
|
||||
size_t size;
|
||||
size_t allocatedSize;
|
||||
} SizedPtr;
|
||||
|
||||
|
||||
#endif //ifndef __SIZED_PTR_H__
|
||||
20
inc/state.h
Normal file
20
inc/state.h
Normal file
@@ -0,0 +1,20 @@
|
||||
#ifndef __EMUL_STATE_H__
|
||||
#define __EMUL_STATE_H__
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#define EMUL_STATE_STILL 0
|
||||
#define EMUL_STATE_EXEC 1
|
||||
#define EMUL_STATE_PAUSE 2
|
||||
#define EMUL_STATE_STOP 3
|
||||
|
||||
#define EMUL_STATE_OP_START 1
|
||||
#define EMUL_STATE_OP_PAUSE 2
|
||||
#define EMUL_STATE_OP_RESUME 3
|
||||
#define EMUL_STATE_OP_RESET 4
|
||||
#define EMUL_STATE_OP_STOP 5
|
||||
|
||||
|
||||
uint8_t switchNewEmulState(const uint8_t currentState, const uint8_t controlOp);
|
||||
|
||||
#endif //ifndef __EMUL_STATE_H__
|
||||
27
inc/streamed.h
Normal file
27
inc/streamed.h
Normal file
@@ -0,0 +1,27 @@
|
||||
#ifndef __STREAMED_H__
|
||||
#define __STREAMED_H__
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdint.h>
|
||||
#include "client.h"
|
||||
|
||||
#define STREAM_MODE_READ 0
|
||||
#define STREAM_MODE_WRITE 1
|
||||
|
||||
|
||||
typedef struct {
|
||||
ClientContext* clientContext;
|
||||
uint32_t regId;
|
||||
uint64_t startAddr;
|
||||
uint64_t segLen;
|
||||
uint8_t mode;
|
||||
} StreamReg;
|
||||
|
||||
typedef struct {
|
||||
StreamReg* regs;
|
||||
size_t regCount;
|
||||
size_t allocatedSize;
|
||||
} DeviceSegStreamReg;
|
||||
|
||||
|
||||
#endif //ifndef __STREAMED_H__
|
||||
@@ -1,276 +0,0 @@
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <dlfcn.h>
|
||||
|
||||
#include "tomlc99/toml.h"
|
||||
|
||||
#include "base_device.h"
|
||||
|
||||
|
||||
|
||||
|
||||
device_handle_t* openBaseDeviceFromConfig(const char* configPath, char* errbuf)
|
||||
{
|
||||
char intErrbuf[1024];
|
||||
|
||||
conf_dev_t *devConf = openBaseDeviceConfig(configPath, intErrbuf);
|
||||
if (devConf == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to read device config: %s", intErrbuf);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
device_handle_t* ret = openBaseDevice(devConf, intErrbuf);
|
||||
|
||||
if(ret == NULL)
|
||||
{
|
||||
freeConf(devConf);
|
||||
free(devConf);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void closeBaseDevice(device_handle_t* devHandle)
|
||||
{
|
||||
if(devHandle->ctx != NULL)
|
||||
{
|
||||
if(devHandle->specs != NULL)
|
||||
{
|
||||
devHandle->lib->freeSpecs(devHandle->specs);
|
||||
}
|
||||
if(devHandle->ctx->deviceMem != NULL)
|
||||
{
|
||||
devHandle->lib->freeDevMem(devHandle->ctx->deviceMem);
|
||||
}
|
||||
free(devHandle->ctx);
|
||||
}
|
||||
if(devHandle->lib->extendedHandlers != NULL)
|
||||
{
|
||||
free(devHandle->lib->extendedHandlers);
|
||||
}
|
||||
dlclose(devHandle->lib->_dlhandl);
|
||||
free(devHandle->lib);
|
||||
}
|
||||
|
||||
device_handle_t* openBaseDevice(conf_dev_t* devConf, char* errbuf)
|
||||
{
|
||||
device_handle_t* ret = malloc(sizeof(device_handle_t));
|
||||
if (ret == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate device handle struct");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char intErrbuf[1024];
|
||||
|
||||
device_lib_t* devLib = loadDeviceLib(devConf->libPath, intErrbuf);
|
||||
if (devLib == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load device library %s: %s", devConf->libPath, intErrbuf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void* devSpecs = devLib->parseSpecsFromConfig(devConf, intErrbuf);
|
||||
|
||||
if (devSpecs == NULL)
|
||||
{
|
||||
sprintf(errbuf, "device config parse error: %s", intErrbuf);
|
||||
dlclose(devLib->_dlhandl);
|
||||
free(ret);
|
||||
free(devLib);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
ret->lib = devLib;
|
||||
ret->specs = devSpecs;
|
||||
ret->clockCycleCounter = 0;
|
||||
ret->clockCycleLimit = 0;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
|
||||
conf_mem_seg_t** parseMemToml(const toml_table_t* memTable)
|
||||
{
|
||||
uint16_t memSegIdx = 0;
|
||||
const char* memSegKey = toml_key_in(memTable, 0);
|
||||
while (memSegKey && memSegIdx < 0xFFFF)
|
||||
{
|
||||
memSegIdx++;
|
||||
memSegKey = toml_key_in(memTable, memSegIdx);
|
||||
}
|
||||
|
||||
conf_mem_seg_t** segConfigs = malloc(sizeof(conf_mem_seg_t*) * ((size_t)memSegIdx + 1));
|
||||
if (segConfigs == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
segConfigs[memSegIdx] = NULL;
|
||||
|
||||
for (size_t i = 0; i < memSegIdx; i++)
|
||||
{
|
||||
segConfigs[i] = (conf_mem_seg_t*)malloc(sizeof(conf_mem_seg_t));
|
||||
memSegKey = toml_key_in(memTable, (int)i);
|
||||
|
||||
if (segConfigs[i] == NULL || memSegKey == NULL)
|
||||
{
|
||||
if (segConfigs[i] != NULL)
|
||||
{
|
||||
free(segConfigs[i]);
|
||||
}
|
||||
else
|
||||
{
|
||||
}
|
||||
for(size_t j = 0; j < i; j++)
|
||||
{
|
||||
free(segConfigs[j]->name);
|
||||
free(segConfigs[j]);
|
||||
}
|
||||
free(segConfigs);
|
||||
return NULL;
|
||||
}
|
||||
segConfigs[i]->name = (char*)malloc(strlen(memSegKey) + 1);
|
||||
if (segConfigs[i]->name == NULL)
|
||||
{
|
||||
for(size_t j = 0; j <= i; j++)
|
||||
{
|
||||
if (j < i)
|
||||
free(segConfigs[j]->name);
|
||||
free(segConfigs[j]);
|
||||
}
|
||||
free(segConfigs);
|
||||
return NULL;
|
||||
}
|
||||
strcpy(segConfigs[i]->name, memSegKey);
|
||||
|
||||
toml_table_t* segTable = toml_table_in(memTable, memSegKey);
|
||||
if (segTable == NULL)
|
||||
{
|
||||
|
||||
for(size_t j = 0; j <= i; j++)
|
||||
{
|
||||
free(segConfigs[j]->name);
|
||||
free(segConfigs[j]);
|
||||
}
|
||||
free(segConfigs);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_datum_t startDatum = toml_int_in(segTable, "start");
|
||||
toml_datum_t lenDatum = toml_int_in(segTable, "len");
|
||||
toml_datum_t wordLenDatum = toml_int_in(segTable, "wordLen");
|
||||
toml_datum_t executableDatum = toml_bool_in(segTable, "executable");
|
||||
|
||||
if (!startDatum.ok || !lenDatum.ok || !wordLenDatum.ok)
|
||||
{
|
||||
for(size_t j = 0; j <= i; j++)
|
||||
{
|
||||
free(segConfigs[j]->name);
|
||||
free(segConfigs[j]);
|
||||
}
|
||||
free(segConfigs);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
segConfigs[i]->start = (size_t)startDatum.u.i;
|
||||
segConfigs[i]->len = (size_t)lenDatum.u.i;
|
||||
segConfigs[i]->wordLen = (uint8_t)wordLenDatum.u.i;
|
||||
segConfigs[i]->isExecutable = 0;
|
||||
if (executableDatum.ok)
|
||||
{
|
||||
segConfigs[i]->isExecutable = executableDatum.u.b != 0;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
return segConfigs;
|
||||
}
|
||||
|
||||
|
||||
conf_dev_t* openBaseDeviceConfig(const char* configPath, char* errbuf)
|
||||
{
|
||||
conf_dev_t* ret = malloc(sizeof(conf_dev_t));
|
||||
if (ret == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate device conf struct");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret->id = NULL;
|
||||
ret->clockDivider = 1;
|
||||
ret->clockMultipler = 1;
|
||||
ret->clockId = NULL;
|
||||
|
||||
|
||||
ret->memConf = malloc(sizeof(conf_mem_t));
|
||||
if (ret->memConf == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate device memory conf struct");
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char intErrbuf[1024];
|
||||
|
||||
|
||||
|
||||
FILE *fp = fopen (configPath, "rb");
|
||||
if (!fp) {
|
||||
sprintf(errbuf, "unable to open config file %s", configPath);
|
||||
free(ret->memConf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
toml_table_t* conf = toml_parse_file(fp, intErrbuf, sizeof(intErrbuf));
|
||||
fclose(fp);
|
||||
|
||||
if (!conf) {
|
||||
sprintf(errbuf, "cannot parse - %s", intErrbuf);
|
||||
free(ret->memConf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_table_t* devBlock = toml_table_in(conf, "dev");
|
||||
if (!devBlock) {
|
||||
sprintf(errbuf, "missing [dev]");
|
||||
free(ret->memConf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_table_t* memTable = toml_table_in(conf, "mem");
|
||||
if (!memTable) {
|
||||
sprintf(errbuf, "missing [mem]");
|
||||
free(ret->memConf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_datum_t libpathBlock = toml_string_in(devBlock, "libpath");
|
||||
if (!libpathBlock.ok) {
|
||||
sprintf(errbuf, "unable to read dev.libpath");
|
||||
free(ret->memConf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
conf_mem_seg_t** memSegConfs = parseMemToml(memTable);
|
||||
if (memSegConfs == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to parse mem segments");
|
||||
free(ret->memConf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret->memConf->memSegConfs = memSegConfs;
|
||||
ret->libPath = libpathBlock.u.s;
|
||||
|
||||
return ret;
|
||||
}
|
||||
@@ -1,529 +0,0 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "tomlc99/toml.h"
|
||||
|
||||
#include "base_device.h"
|
||||
#include "compose_device.h"
|
||||
|
||||
|
||||
// void fillDevClockConfig(const toml_table_t* clockTable, conf_dev_t** devConfs)
|
||||
// {
|
||||
// uint16_t clockSegIdx = 0;
|
||||
// uint16_t clockSegSkips = 0;
|
||||
// const char* clockSegKey = toml_key_in(clockTable, 0);
|
||||
// while (clockSegKey && (clockSegIdx + clockSegSkips) < 0xFFFF)
|
||||
// {
|
||||
// clockSegKey = toml_key_in(clockTable, clockSegIdx + clockSegSkips + 1);
|
||||
// if(strcmp(clockSegKey, "limiter") != 0)
|
||||
// {
|
||||
// clockSegIdx++;
|
||||
// }
|
||||
// else
|
||||
// {
|
||||
// clockSegSkips++;
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
char** appendId(char** prev, const char* cur, char* errbuf)
|
||||
{
|
||||
if(prev == NULL)
|
||||
{
|
||||
prev = malloc(sizeof(char*) * 2);
|
||||
if(prev == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate id");
|
||||
return NULL;
|
||||
}
|
||||
prev[0] = NULL;
|
||||
prev[1] = NULL;
|
||||
}
|
||||
|
||||
size_t clockIdLen = 0;
|
||||
while (prev[clockIdLen] != NULL){clockIdLen++;}
|
||||
|
||||
clockIdLen++;
|
||||
|
||||
char** new = realloc(prev, sizeof(char*) * (clockIdLen + 1));
|
||||
|
||||
if(new == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to reallocate id");
|
||||
freeComposeId(prev);
|
||||
return NULL;
|
||||
}
|
||||
prev = new;
|
||||
|
||||
prev[clockIdLen] = NULL;
|
||||
size_t idLen = strlen(cur);
|
||||
prev[clockIdLen - 1] = malloc(sizeof(char) * (idLen + 1));
|
||||
if(prev[clockIdLen - 1] == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate new id entry");
|
||||
freeComposeId(prev);
|
||||
return NULL;
|
||||
}
|
||||
strcpy(prev[clockIdLen - 1], cur);
|
||||
prev[clockIdLen - 1][idLen] = '\0';
|
||||
return prev;
|
||||
}
|
||||
|
||||
conf_dev_t** parseDevToml(const toml_table_t* devTable, const toml_table_t* clockTable, char* errbuf)
|
||||
{
|
||||
uint16_t devSegIdx = 0;
|
||||
const char* devSegKey = toml_key_in(devTable, 0);
|
||||
while (devSegKey && devSegIdx < 0xFFFF)
|
||||
{
|
||||
devSegIdx++;
|
||||
devSegKey = toml_key_in(devTable, devSegIdx);
|
||||
}
|
||||
|
||||
conf_dev_t** deviceConfigs = malloc(sizeof(conf_dev_t*) * (devSegIdx + 1));
|
||||
if(deviceConfigs == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
deviceConfigs[devSegIdx] = NULL;
|
||||
|
||||
for(size_t i = 0; i < devSegIdx; i++)
|
||||
{
|
||||
devSegKey = toml_key_in(devTable, (int)i);
|
||||
if(devSegKey == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load device key %lu", i);
|
||||
for(size_t j = 0; j < i; j++)
|
||||
{
|
||||
freeConf(deviceConfigs[j]);
|
||||
free(deviceConfigs[j]);
|
||||
}
|
||||
free(deviceConfigs);
|
||||
return NULL;
|
||||
}
|
||||
toml_datum_t devpathDatum = toml_string_in(devTable, devSegKey);
|
||||
if(!devpathDatum.ok || !devpathDatum.u.s)
|
||||
{
|
||||
sprintf(errbuf, "unable to get device path for %s", devSegKey);
|
||||
|
||||
for(size_t j = 0; j < i; j++)
|
||||
{
|
||||
freeConf(deviceConfigs[j]);
|
||||
free(deviceConfigs[j]);
|
||||
}
|
||||
free(deviceConfigs);
|
||||
return NULL;
|
||||
}
|
||||
char intErrbuf[1024];
|
||||
|
||||
conf_dev_t* conf = openBaseDeviceConfig(devpathDatum.u.s, intErrbuf);
|
||||
if(conf == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load config for %s: %s", devSegKey, intErrbuf);
|
||||
|
||||
for(size_t j = 0; j < i; j++)
|
||||
{
|
||||
freeConf(deviceConfigs[j]);
|
||||
free(deviceConfigs[j]);
|
||||
}
|
||||
free(deviceConfigs);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
conf->id = NULL;
|
||||
conf->id = appendId(conf->id, devSegKey, intErrbuf);
|
||||
|
||||
|
||||
if(conf->id == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate device %s name field", devSegKey);
|
||||
for(size_t j = 0; j <= i; j++)
|
||||
{
|
||||
freeConf(deviceConfigs[j]);
|
||||
free(deviceConfigs[j]);
|
||||
}
|
||||
free(deviceConfigs);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
deviceConfigs[i] = conf;
|
||||
}
|
||||
|
||||
if(clockTable != NULL)
|
||||
{
|
||||
for(size_t i = 0; i < devSegIdx; i++)
|
||||
{
|
||||
conf_dev_t* conf = deviceConfigs[i];
|
||||
toml_table_t* devClockTable = toml_table_in(clockTable, conf->id[0]);
|
||||
if(devClockTable == NULL)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
toml_datum_t dividerDatum = toml_int_in(devClockTable, "divider");
|
||||
if(dividerDatum.ok)
|
||||
{
|
||||
conf->clockDivider = (uint64_t)dividerDatum.u.i;
|
||||
}
|
||||
toml_datum_t multiplerDatum = toml_int_in(devClockTable, "multipler");
|
||||
if(multiplerDatum.ok)
|
||||
{
|
||||
conf->clockMultipler = (uint64_t)multiplerDatum.u.i;
|
||||
}
|
||||
|
||||
toml_datum_t srcDatum = toml_string_in(devClockTable, "src");
|
||||
if(srcDatum.ok && srcDatum.u.s != NULL)
|
||||
{
|
||||
char intErrbuf[1024] = {0};
|
||||
conf->clockId = appendId(conf->clockId, srcDatum.u.s, intErrbuf);
|
||||
if(conf->clockId == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to append clock id for %s: %s", conf->id[0], intErrbuf);
|
||||
for(size_t j = 0; j <= i; j++)
|
||||
{
|
||||
freeConf(deviceConfigs[j]);
|
||||
free(deviceConfigs[j]);
|
||||
}
|
||||
free(deviceConfigs);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
return deviceConfigs;
|
||||
}
|
||||
void freeProjectionConfig(projection_conf_t* conf)
|
||||
{
|
||||
if(conf == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
freeComposeId(conf->target);
|
||||
freeComposeId(conf->baseAt);
|
||||
if(conf->baseSeg != NULL)
|
||||
{
|
||||
free(conf->baseSeg);
|
||||
}
|
||||
conf->target = NULL;
|
||||
conf->baseAt = NULL;
|
||||
conf->baseSeg = NULL;
|
||||
free(conf);
|
||||
}
|
||||
void freeProjectionConfigs(projection_conf_t** confs)
|
||||
{
|
||||
if(confs == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
size_t i = 0;
|
||||
while(confs[i] != NULL)
|
||||
{
|
||||
freeProjectionConfig(confs[i]);
|
||||
confs[i] = NULL;
|
||||
}
|
||||
free(confs);
|
||||
}
|
||||
|
||||
|
||||
projection_conf_t* parseDeviceProjectionConf(const toml_table_t* deviceProjectionTable, char* errbuf)
|
||||
{
|
||||
toml_datum_t baseAtDatum = toml_string_in(deviceProjectionTable, "base_at");
|
||||
if(!baseAtDatum.ok || baseAtDatum.u.s == NULL)
|
||||
{
|
||||
sprintf(errbuf, "missing 'base_at'");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_datum_t baseSegDatum = toml_string_in(deviceProjectionTable, "base_seg");
|
||||
if(!baseSegDatum.ok || baseSegDatum.u.s == NULL)
|
||||
{
|
||||
sprintf(errbuf, "missing 'base_seg'");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_datum_t projectionShiftDatum = toml_int_in(deviceProjectionTable, "projection_shift");
|
||||
|
||||
char intErrbuf[1024] = {0};
|
||||
|
||||
projection_conf_t* ret = malloc(sizeof(projection_conf_t));
|
||||
if(ret == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate projection conf");
|
||||
return NULL;
|
||||
}
|
||||
ret->baseAt = NULL;
|
||||
ret->baseSeg = NULL;
|
||||
ret->target = NULL;
|
||||
ret->projectionShift = 0;
|
||||
|
||||
if(projectionShiftDatum.ok)
|
||||
{
|
||||
ret->projectionShift = (uint64_t)projectionShiftDatum.u.i;
|
||||
}
|
||||
|
||||
ret->baseAt = appendId(ret->baseAt, baseAtDatum.u.s, intErrbuf);
|
||||
if(ret->baseAt == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to append base_at id");
|
||||
freeProjectionConfig(ret);
|
||||
return NULL;
|
||||
}
|
||||
ret->baseSeg = malloc(strlen(baseSegDatum.u.s));
|
||||
if(ret->baseSeg == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to fill base_seg");
|
||||
freeProjectionConfig(ret);
|
||||
return NULL;
|
||||
}
|
||||
strcpy(ret->baseSeg, baseSegDatum.u.s);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
projection_conf_t** parseDeviceProjectionConfs(const toml_table_t* deviceProjectionsTable, char* errbuf)
|
||||
{
|
||||
uint16_t projSegIdx = 0;
|
||||
const char* projSegKey = toml_key_in(deviceProjectionsTable, 0);
|
||||
while (projSegKey && projSegIdx < 0xFFFF)
|
||||
{
|
||||
projSegIdx++;
|
||||
projSegKey = toml_key_in(deviceProjectionsTable, projSegIdx);
|
||||
}
|
||||
char intErrbuf[1024] = {0};
|
||||
|
||||
projection_conf_t** ret = malloc(sizeof(projection_conf_t*) * (projSegIdx + 1));
|
||||
if(ret == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate device projections confs");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
for(size_t i = 0; i <= projSegIdx; i++){ret[i] = NULL;}
|
||||
|
||||
for(size_t i = 0; i < projSegIdx; i++)
|
||||
{
|
||||
projSegKey = toml_key_in(deviceProjectionsTable, (int)i);
|
||||
if(projSegKey == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load device projection key %lu", i);
|
||||
freeProjectionConfigs(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_table_t* projTable = toml_table_in(deviceProjectionsTable, projSegKey);
|
||||
if(projTable == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to open device %s projection table", projSegKey);
|
||||
freeProjectionConfigs(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret[i] = parseDeviceProjectionConf(projTable, intErrbuf);
|
||||
if(ret[i] == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load device %s projection configs: %s", projSegKey, intErrbuf);
|
||||
freeProjectionConfigs(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret[i]->target = appendId(ret[i]->target, projSegKey, intErrbuf);
|
||||
if(ret[i]->target == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to append projection target id: %s", intErrbuf);
|
||||
freeProjectionConfigs(ret);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
projection_conf_t** parseProjectionConfs(const toml_table_t* projectionTable, char* errbuf)
|
||||
{
|
||||
uint16_t devicesProjSegIdx = 0;
|
||||
const char* devicesProjSegKey = toml_key_in(projectionTable, 0);
|
||||
while (devicesProjSegKey && devicesProjSegIdx < 0xFFFF)
|
||||
{
|
||||
devicesProjSegIdx++;
|
||||
devicesProjSegKey = toml_key_in(projectionTable, devicesProjSegIdx);
|
||||
}
|
||||
|
||||
|
||||
char intErrbuf[1024] = {0};
|
||||
|
||||
size_t totalRetSize = 0;
|
||||
projection_conf_t** ret = malloc(sizeof(projection_conf_t*));
|
||||
if(ret == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate base projection conf array");
|
||||
return NULL;
|
||||
}
|
||||
ret[0] = NULL;
|
||||
|
||||
for(size_t i = 0; i < devicesProjSegIdx; i++)
|
||||
{
|
||||
devicesProjSegKey = toml_key_in(projectionTable, (int)i);
|
||||
if(devicesProjSegKey == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load proj key %lu", i);
|
||||
freeProjectionConfigs(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
toml_table_t* deviceProjTable = toml_table_in(projectionTable, devicesProjSegKey);
|
||||
if(deviceProjTable == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to open device projection table");
|
||||
freeProjectionConfigs(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
projection_conf_t** devProjections = parseDeviceProjectionConfs(deviceProjTable, intErrbuf);
|
||||
|
||||
if(devProjections == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to parse device %s projection rules: %s", devicesProjSegKey, intErrbuf);
|
||||
freeProjectionConfigs(ret);
|
||||
return NULL;
|
||||
}
|
||||
size_t dlen = 0;
|
||||
while(devProjections[dlen] != NULL)
|
||||
{
|
||||
devProjections[dlen]->target = appendId(devProjections[dlen]->target, devicesProjSegKey, intErrbuf);
|
||||
if(devProjections[dlen]->target == NULL)
|
||||
{
|
||||
freeProjectionConfigs(ret);
|
||||
freeProjectionConfigs(devProjections);
|
||||
return NULL;
|
||||
}
|
||||
dlen++;
|
||||
}
|
||||
ret = realloc(ret, sizeof(projection_conf_t*) * (dlen + totalRetSize + 1));
|
||||
if(ret == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to reallocate full projection array");
|
||||
freeProjectionConfigs(devProjections);
|
||||
return NULL;
|
||||
}
|
||||
for(size_t i = 0; i <= dlen; i++)
|
||||
{
|
||||
ret[totalRetSize + i] = devProjections[i];
|
||||
}
|
||||
totalRetSize += dlen;
|
||||
ret[totalRetSize] = NULL;
|
||||
free(devProjections);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
compose_dev_conf_t* openComposeDeviceConfig(const char* configPath, char* errbuf)
|
||||
{
|
||||
compose_dev_conf_t* ret = malloc(sizeof(compose_dev_conf_t));
|
||||
|
||||
if(ret == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate device struct");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char intErrbuf[1024] = {0};
|
||||
FILE *fp = fopen (configPath, "rb");
|
||||
if (!fp) {
|
||||
sprintf(errbuf, "unable to open config file %s", configPath);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
toml_table_t* conf = toml_parse_file(fp, intErrbuf, sizeof(intErrbuf));
|
||||
fclose(fp);
|
||||
|
||||
|
||||
toml_table_t* devTable = toml_table_in(conf, "dev");
|
||||
if (!devTable) {
|
||||
sprintf(errbuf, "missing [dev]");
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
ret->projections = NULL;
|
||||
|
||||
toml_table_t* memTable = toml_table_in(conf, "mem");
|
||||
if(memTable)
|
||||
{
|
||||
toml_table_t* projectionTable = toml_table_in(memTable, "projection");
|
||||
if(projectionTable)
|
||||
{
|
||||
ret->projections = parseProjectionConfs(projectionTable, intErrbuf);
|
||||
if(ret->projections == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load projections: %s", intErrbuf);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
toml_table_t* clockTable = toml_table_in(conf, "clock");
|
||||
|
||||
|
||||
conf_dev_t** devHandlers = parseDevToml(devTable, clockTable, intErrbuf);
|
||||
|
||||
if(devHandlers == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to load devices configs: %s", intErrbuf);
|
||||
freeProjectionConfigs(ret->projections);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret->baseConfigs = devHandlers;
|
||||
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
device_handle_t** openComposeDevice(compose_dev_conf_t* conf, char* errbuf)
|
||||
{
|
||||
size_t devIdx = 0;
|
||||
while(conf->baseConfigs[devIdx] != NULL){devIdx++;}
|
||||
|
||||
device_handle_t** devHandlers = malloc(sizeof(device_handle_t*) * (devIdx + 1));
|
||||
|
||||
if(devHandlers == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate dev handlers");
|
||||
return NULL;
|
||||
}
|
||||
devHandlers[devIdx] = NULL;
|
||||
|
||||
|
||||
for(size_t i = 0; i < devIdx; i++)
|
||||
{
|
||||
conf_dev_t* devConf = conf->baseConfigs[i];
|
||||
char intErrbuf[1024] = {0};
|
||||
device_handle_t* devHandle = openBaseDevice(devConf, intErrbuf);
|
||||
if(devHandle == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to open base device %s: %s", devConf->id[0], intErrbuf);
|
||||
for(size_t j = 0; j < i; j++)
|
||||
{
|
||||
closeBaseDevice(devHandlers[j]);
|
||||
free(devHandlers[j]);
|
||||
}
|
||||
free(devHandlers);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
devHandlers[i] = devHandle;
|
||||
}
|
||||
|
||||
return devHandlers;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
86
src/config.c
86
src/config.c
@@ -1,86 +0,0 @@
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "config.h"
|
||||
|
||||
|
||||
void xfree(void* p)
|
||||
{
|
||||
if(p != NULL)
|
||||
{
|
||||
free(p);
|
||||
}
|
||||
}
|
||||
|
||||
void freeComposeId(char** id)
|
||||
{
|
||||
if(id == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
size_t i = 0;
|
||||
while(id[i] != NULL)
|
||||
{
|
||||
free(id[i]);
|
||||
i++;
|
||||
}
|
||||
free(id);
|
||||
}
|
||||
|
||||
|
||||
uint8_t compareComposeId(char** idA, char** idB)
|
||||
{
|
||||
size_t i = 0;
|
||||
while(idA[i] != NULL && idB[i] != NULL)
|
||||
{
|
||||
if(strcmp(idA[i], idB[i]) != 0)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
i++;
|
||||
}
|
||||
return idA[i] == NULL && idB[i] == NULL;
|
||||
}
|
||||
|
||||
void freeMemSegConf(conf_mem_seg_t* memSegConf)
|
||||
{
|
||||
if(memSegConf == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
xfree(memSegConf->name);
|
||||
}
|
||||
void freeMemConf(conf_mem_t* memConf)
|
||||
{
|
||||
if(memConf == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
for(size_t i = 0; i < 0xFF && memConf->memSegConfs[i] != NULL; i++)
|
||||
{
|
||||
freeMemSegConf(memConf->memSegConfs[i]);
|
||||
}
|
||||
xfree(memConf->memSegConfs);
|
||||
free(memConf);
|
||||
}
|
||||
void freeConf(conf_dev_t* conf)
|
||||
{
|
||||
if(conf == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
if(conf->clockId != NULL)
|
||||
{
|
||||
size_t i = 0;
|
||||
while(conf->clockId[i] != NULL)
|
||||
{
|
||||
free(conf->clockId[i]);
|
||||
i++;
|
||||
}
|
||||
free(conf->clockId);
|
||||
}
|
||||
freeComposeId(conf->id);
|
||||
xfree(conf->libPath);
|
||||
freeMemConf(conf->memConf);
|
||||
free(conf);
|
||||
}
|
||||
238
src/hmmmm.c
238
src/hmmmm.c
@@ -1,238 +0,0 @@
|
||||
#include "hmmmm.h"
|
||||
|
||||
|
||||
#include <dlfcn.h>
|
||||
#include <stdio.h>
|
||||
|
||||
typedef size_t (*_dlib_pubExtractPcounter_t)(device_public_context_t* devContext);
|
||||
typedef size_t (*_dlib_pubExtractOpcode_t)(device_mem_t* devMem, size_t _programCounter);
|
||||
|
||||
typedef uint8_t(*_dlib_pubExtractPcounterSizeWords_t)();
|
||||
|
||||
typedef device_public_context_t* (*_dlib_dev_init_t)(void* specs, char* errbuf);
|
||||
|
||||
typedef uint8_t (*_dlib_makeDeviceTick_t)(device_public_context_t* devInfo);
|
||||
|
||||
typedef uint8_t (*_dlib_deviceType_t)();
|
||||
|
||||
typedef void* (*_dlib_pubExtractPcounterPtr_t)(device_public_context_t* devContext);
|
||||
|
||||
typedef void* (*_dlib_parseSpecsFromConfig_t)(const conf_dev_t* devConf, char* errbuf);
|
||||
|
||||
typedef void (*_dlib_freeSpecs_t)(void* specs);
|
||||
typedef void (*_dlib_freeDevMem_t)(device_mem_t* mem);
|
||||
|
||||
typedef void (*_dlib_fillSmartReadSpecs_t)(void* specs, smart_read_spec_t* smartReadSpecs, uint64_t smartReadSpecsCount);
|
||||
typedef void (*_dlib_fillSmartWriteSpecs_t)(void* specs, smart_write_spec_t* smartWriteSpecs, uint64_t smartWriteSpecsCount);
|
||||
|
||||
instruction_simul_handlers_t* _fillInstructionSimul(void* handle)
|
||||
{
|
||||
instruction_simul_handlers_t* ret = malloc(sizeof(instruction_simul_handlers_t));
|
||||
|
||||
if (ret == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
_dlib_pubExtractPcounterPtr_t _dlib_pubExtractPcounterPtr = (_dlib_pubExtractPcounterPtr_t)(uintptr_t)dlsym(handle, "pubExtractPcounterPtr");
|
||||
|
||||
const char *dlib_pubExtractPcounterPtr_error = dlerror();
|
||||
if (dlib_pubExtractPcounterPtr_error) {
|
||||
dlclose(handle);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret->extractPcounterPtr = _dlib_pubExtractPcounterPtr;
|
||||
|
||||
|
||||
_dlib_pubExtractPcounter_t _dlib_pubExtractPcounter = (_dlib_pubExtractPcounter_t)(uintptr_t)dlsym(handle, "pubExtractPcounter");
|
||||
|
||||
const char *dlib_pubExtractPcounter_error = dlerror();
|
||||
if (dlib_pubExtractPcounter_error) {
|
||||
dlclose(handle);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret->extractPcounter = _dlib_pubExtractPcounter;
|
||||
|
||||
|
||||
_dlib_pubExtractOpcode_t _dlib_pubExtractOpcode = (_dlib_pubExtractOpcode_t)(uintptr_t)dlsym(handle, "pubExtractOpcode");
|
||||
|
||||
const char *dlib_pubExtractOpcode_error = dlerror();
|
||||
if (dlib_pubExtractOpcode_error) {
|
||||
dlclose(handle);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret->extractOpcode = _dlib_pubExtractOpcode;
|
||||
|
||||
_dlib_pubExtractPcounterSizeWords_t _dlib_pubExtractPcounterSizeWords = (_dlib_pubExtractPcounterSizeWords_t)(uintptr_t)dlsym(handle, "pubExtractPcounterSizeWords");
|
||||
|
||||
const char *dlib_pubExtractPcounterSizeWords_error = dlerror();
|
||||
if (dlib_pubExtractPcounterSizeWords_error) {
|
||||
dlclose(handle);
|
||||
free(ret);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ret->extractPcounterSizeWords = _dlib_pubExtractPcounterSizeWords;
|
||||
|
||||
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
device_lib_t* loadDeviceLib(const char *libpath, char* errbuf)
|
||||
{
|
||||
device_lib_t* dev = malloc(sizeof(device_lib_t));
|
||||
|
||||
if (dev == NULL)
|
||||
{
|
||||
sprintf(errbuf, "unable to allocate device lib struct");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void *handle = dlopen(libpath, RTLD_NOW);
|
||||
|
||||
if (!handle) {
|
||||
sprintf(errbuf, "unable to open dl handle");
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
dlerror();
|
||||
|
||||
_dlib_dev_init_t _dlib_dev_init = (_dlib_dev_init_t)(uintptr_t)dlsym(handle, "init");
|
||||
|
||||
const char *dlsym_init_error = dlerror();
|
||||
if (dlsym_init_error) {
|
||||
sprintf(errbuf, "unable to find init symbol: %s", dlsym_init_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
_dlib_makeDeviceTick_t _dlib_makeDeviceTick = (_dlib_makeDeviceTick_t)(uintptr_t)dlsym(handle, "makeDeviceTick");
|
||||
|
||||
|
||||
const char *dlsym_maketick_error = dlerror();
|
||||
if (dlsym_maketick_error) {
|
||||
sprintf(errbuf, "unable to find makeDeviceTick symbol: %s", dlsym_maketick_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
|
||||
_dlib_parseSpecsFromConfig_t _dlib_parseSpecsFromConfig = (_dlib_parseSpecsFromConfig_t)(uintptr_t)dlsym(handle, "parseSpecsFromConfig");
|
||||
|
||||
const char *_dlib_parseSpecsFromConfig_error = dlerror();
|
||||
if (_dlib_parseSpecsFromConfig_error) {
|
||||
sprintf(errbuf, "unable to find parseSpecsFromConfig symbol: %s", _dlib_parseSpecsFromConfig_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
|
||||
_dlib_fillSmartReadSpecs_t _dlib_fillSmartReadSpecs = (_dlib_fillSmartReadSpecs_t)(uintptr_t)dlsym(handle, "fillSmartReadSpecs");
|
||||
|
||||
const char *_dlib_fillSmartReadSpecs_error = dlerror();
|
||||
if (_dlib_fillSmartReadSpecs_error) {
|
||||
sprintf(errbuf, "unable to find fillSmartReadSpecs symbol: %s", _dlib_fillSmartReadSpecs_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
|
||||
_dlib_fillSmartWriteSpecs_t _dlib_fillSmartWriteSpecs = (_dlib_fillSmartWriteSpecs_t)(uintptr_t)dlsym(handle, "fillSmartWriteSpecs");
|
||||
|
||||
const char *_dlib_fillSmartWriteSpecs_error = dlerror();
|
||||
if (_dlib_fillSmartWriteSpecs_error) {
|
||||
sprintf(errbuf, "unable to find fillSmartWriteSpecs symbol: %s", _dlib_fillSmartWriteSpecs_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
_dlib_deviceType_t _dlib_deviceType = (_dlib_deviceType_t)(uintptr_t)dlsym(handle, "pubDeviceType");
|
||||
|
||||
const char *dlib_deviceType_error = dlerror();
|
||||
if (dlib_deviceType_error) {
|
||||
sprintf(errbuf, "unable to find pubDeviceType symbol: %s", dlib_deviceType_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
_dlib_freeSpecs_t _dlib_freeSpecs = (_dlib_freeSpecs_t)(uintptr_t)dlsym(handle, "freeDevSpecs");
|
||||
|
||||
const char *dlib_freeSpecs_error = dlerror();
|
||||
if (dlib_freeSpecs_error) {
|
||||
sprintf(errbuf, "unable to find freeSpecs symbol: %s", dlib_freeSpecs_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
_dlib_freeDevMem_t _dlib_freeDevMem = (_dlib_freeDevMem_t)(uintptr_t)dlsym(handle, "freeDevSpecs");
|
||||
|
||||
const char *dlib_freeDevMem_error = dlerror();
|
||||
if (dlib_freeDevMem_error) {
|
||||
sprintf(errbuf, "unable to find freeDevMem symbol: %s", dlib_freeDevMem_error);
|
||||
dlclose(handle);
|
||||
free(dev);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
dev->devContext = NULL;
|
||||
dev->init = _dlib_dev_init;
|
||||
dev->makeDeviceTick = _dlib_makeDeviceTick;
|
||||
dev->parseSpecsFromConfig = _dlib_parseSpecsFromConfig;
|
||||
dev->fillSmartReadSpecs = _dlib_fillSmartReadSpecs;
|
||||
dev->fillSmartWriteSpecs = _dlib_fillSmartWriteSpecs;
|
||||
dev->freeSpecs = _dlib_freeSpecs;
|
||||
dev->freeDevMem = _dlib_freeDevMem;
|
||||
|
||||
uint8_t devType = _dlib_deviceType();
|
||||
|
||||
dev->devType = devType;
|
||||
|
||||
if (devType == EXTENDED_DEVICE_TYPE_DUMMY)
|
||||
{
|
||||
dev->extendedHandlers = NULL;
|
||||
}
|
||||
else if (devType == EXTENDED_DEVICE_TYPE_INSTR_SIMUL)
|
||||
{
|
||||
dev->extendedHandlers = _fillInstructionSimul(handle);
|
||||
if (dev->extendedHandlers == NULL)
|
||||
{
|
||||
free(dev);
|
||||
dlclose(handle);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
free(dev);
|
||||
dlclose(handle);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
dev->_dlhandl = handle;
|
||||
|
||||
return dev;
|
||||
}
|
||||
49
src/linkedlist.c
Normal file
49
src/linkedlist.c
Normal file
@@ -0,0 +1,49 @@
|
||||
#include <stdlib.h>
|
||||
#include "linkedlist.h"
|
||||
|
||||
|
||||
void removeLinkedListEntry(LinkedListEntry** head, LinkedListEntry* entry)
|
||||
{
|
||||
if(entry == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
// check for head
|
||||
if(entry->prevEntry != NULL)
|
||||
{
|
||||
entry->prevEntry->nextEntry = entry->nextEntry;
|
||||
}
|
||||
// check for tail
|
||||
if(entry->nextEntry != NULL)
|
||||
{
|
||||
entry->nextEntry->prevEntry = entry->prevEntry;
|
||||
}
|
||||
|
||||
// check for removing head entry
|
||||
if(entry == *head)
|
||||
{
|
||||
if(entry->nextEntry == NULL && entry->prevEntry == NULL)
|
||||
{
|
||||
// removing full list
|
||||
*head = NULL;
|
||||
}
|
||||
else if(entry->nextEntry != NULL)
|
||||
{
|
||||
// moving head to next entry
|
||||
*head = entry->nextEntry;
|
||||
}
|
||||
else if(entry->prevEntry != NULL)
|
||||
{
|
||||
// head that have previous item, must be impossible
|
||||
*head = entry->prevEntry;
|
||||
}
|
||||
|
||||
if(entry == *head)
|
||||
{
|
||||
// if still head => there is a circular list, removing whole
|
||||
*head = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
free(entry);
|
||||
}
|
||||
1323
src/main.c
1323
src/main.c
File diff suppressed because it is too large
Load Diff
21
src/panic.c
Normal file
21
src/panic.c
Normal file
@@ -0,0 +1,21 @@
|
||||
#include "panic.h"
|
||||
|
||||
void print_stacktrace(FILE* fd)
|
||||
{
|
||||
void *buffer[PANIC_STRACE_LEN];
|
||||
int size = backtrace(buffer, PANIC_STRACE_LEN);
|
||||
char **symbols = backtrace_symbols(buffer, size);
|
||||
|
||||
if (symbols == NULL) {
|
||||
fprintf(fd, "backtrace_symbols");
|
||||
abort();
|
||||
}
|
||||
|
||||
fprintf(fd, "\n\nStack trace:\n");
|
||||
for (int i = 0; i < size; i++)
|
||||
{
|
||||
fprintf(fd, "%s\n", symbols[i]);
|
||||
}
|
||||
|
||||
free(symbols);
|
||||
}
|
||||
46
src/proto/dial.c
Normal file
46
src/proto/dial.c
Normal file
@@ -0,0 +1,46 @@
|
||||
#include "proto/dial.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "context.h"
|
||||
#include "panic.h"
|
||||
#include "proto/msg.h"
|
||||
|
||||
|
||||
void broadcastClients(EmulContext* emulContext, uint8_t* msg, size_t msgLen)
|
||||
{
|
||||
LinkedListEntry* clientsHead = *emulContext->clientsHead;
|
||||
while(clientsHead != NULL)
|
||||
{
|
||||
uint8_t* newMsg = malloc(sizeof(uint8_t) * msgLen);
|
||||
NULL_GUARD(newMsg);
|
||||
memcpy(newMsg, msg, msgLen);
|
||||
|
||||
ClientContext* ctx = clientsHead->payload;
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, newMsg, msgLen);
|
||||
clientsHead = clientsHead->nextEntry;
|
||||
}
|
||||
free(msg);
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
void dispatchOutgoingMessage(OutgoingBuffers* outBufs, ClientContext* client, uint8_t* msg, size_t msgLen)
|
||||
{
|
||||
SizedPtr* p = &outBufs->bufs[outBufs->currWritingIdx];
|
||||
if(p->size + 1 >= p->allocatedSize)
|
||||
{
|
||||
printf("\t>>Reallocating buf %d\n", outBufs->currWritingIdx);
|
||||
OutgoingMessage* newPtr = realloc(p->ptr, sizeof(OutgoingMessage) * p->allocatedSize * 2);
|
||||
NULL_GUARD(newPtr);
|
||||
p->ptr = newPtr;
|
||||
p->allocatedSize = p->allocatedSize * 2;
|
||||
}
|
||||
OutgoingMessage* outmsg = &((OutgoingMessage*)p->ptr)[p->size];
|
||||
outmsg->msg = msg;
|
||||
outmsg->msgLen = msgLen;
|
||||
outmsg->client = client;
|
||||
p->size++;
|
||||
}
|
||||
199
src/proto/handlers.c
Normal file
199
src/proto/handlers.c
Normal file
@@ -0,0 +1,199 @@
|
||||
#include "proto/handlers.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include "proto/enums.h"
|
||||
#include "panic.h"
|
||||
|
||||
#include "proto/handlers/auth.h"
|
||||
#include "proto/handlers/stream.h"
|
||||
#include "proto/handlers/control.h"
|
||||
#include "proto/handlers/mem.h"
|
||||
#include "state.h"
|
||||
|
||||
|
||||
void handleCloseClient(EmulContext* emulContext, LinkedListEntry* clientEntry)
|
||||
{
|
||||
ClientContext* ctx = clientEntry->payload;
|
||||
|
||||
if(ctx->orphanedAt == 0)
|
||||
{
|
||||
ctx->orphanedAt = (uint64_t)time(NULL);
|
||||
}
|
||||
|
||||
// if(ctx->streamRegIterator > 0)
|
||||
// {
|
||||
// unregisterClientStreams(emulContext, ctx);
|
||||
// }
|
||||
|
||||
// if(ctx->streamRegIterator == 0)
|
||||
// {
|
||||
// removeLinkedListEntry(emulContext->clientsHead, clientEntry);
|
||||
// ctx->clientId = 0;
|
||||
// free(ctx->fallbackOutcomeQ->ptr);
|
||||
// free(ctx->fallbackOutcomeQ);
|
||||
// ptQueueFree(ctx->incomeQ);
|
||||
// free(ctx);
|
||||
// }
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
void handleRegEvent(EmulContext* emulContext, ClientRegistrationEvent* ev)
|
||||
{
|
||||
if(ev->regType == REG_EVTYPE_CONNECT)
|
||||
{
|
||||
printf("open client %lu\n", ev->ctx->seatId);
|
||||
LinkedListEntry* newClientsLinkedListHead = malloc(sizeof(LinkedListEntry));
|
||||
NULL_GUARD(newClientsLinkedListHead);
|
||||
|
||||
newClientsLinkedListHead->payload = ev->ctx;
|
||||
if(*emulContext->clientsHead != NULL)
|
||||
{
|
||||
(*emulContext->clientsHead)->prevEntry = newClientsLinkedListHead;
|
||||
}
|
||||
newClientsLinkedListHead->prevEntry = NULL;
|
||||
newClientsLinkedListHead->nextEntry = *emulContext->clientsHead;
|
||||
*emulContext->clientsHead = newClientsLinkedListHead;
|
||||
}
|
||||
else if (ev->regType == REG_EVTYPE_CLOSE)
|
||||
{
|
||||
LinkedListEntry* clientEntry = *emulContext->clientsHead;
|
||||
|
||||
while(clientEntry != NULL)
|
||||
{
|
||||
if(clientEntry->payload == ev->ctx)
|
||||
{
|
||||
printf("close client %lu\n", ev->ctx->seatId);
|
||||
handleCloseClient(emulContext, clientEntry);
|
||||
break;
|
||||
}
|
||||
clientEntry = clientEntry->nextEntry;
|
||||
}
|
||||
}
|
||||
else if (ev->regType == REG_EVTYPE_AUTH)
|
||||
{
|
||||
printf("auth client %lu\n", ev->ctx->seatId);
|
||||
ev->ctx->isAuthed = 1;
|
||||
handleOnClientAuthDone(ev->ctx, emulContext);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
void handleIncomingMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
switch (msg->packetType)
|
||||
{
|
||||
case PACKET_TYPE_CTRL:
|
||||
{
|
||||
handleIncomingControlMessage(msg, ctx, emulContext);
|
||||
break;
|
||||
}
|
||||
case PACKET_TYPE_STREAM:
|
||||
{
|
||||
handleIncomingStreamMessage(msg, ctx, emulContext);
|
||||
break;
|
||||
}
|
||||
case PACKET_TYPE_MEM:
|
||||
{
|
||||
handleIncomingMemMessage(msg, ctx, emulContext);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
{
|
||||
printf("Unsupported packet type: %u\n", msg->packetType);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void handleAllClients(EmulContext* emulContext)
|
||||
{
|
||||
LinkedListEntry* clientEntry = *emulContext->clientsHead;
|
||||
|
||||
size_t handleLimit = 128;
|
||||
uint8_t hasAliveClients = 0;
|
||||
|
||||
while(clientEntry != NULL && handleLimit > 0)
|
||||
{
|
||||
LinkedListEntry* currEntry = clientEntry;
|
||||
handleLimit--;
|
||||
ClientContext* ctx = clientEntry->payload;
|
||||
|
||||
|
||||
|
||||
if(ctx->orphanedDeadTimeout != 0 && ctx->orphanedAt != 0 && ctx->isAuthed)
|
||||
{
|
||||
// printf("Found orphaned client\n");
|
||||
if((uint64_t)time(NULL) - ctx->orphanedAt > ctx->orphanedDeadTimeout || ctx->streamRegIterator == 0)
|
||||
{
|
||||
printf("Orphaned client dead timeout exceeded\n");
|
||||
if(ctx->streamRegIterator > 0)
|
||||
{
|
||||
unregisterClientStreams(emulContext, ctx);
|
||||
}
|
||||
clientEntry = clientEntry->nextEntry;
|
||||
removeLinkedListEntry(emulContext->clientsHead, currEntry);
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, NULL, 0);
|
||||
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
if(!ctx->isAuthed)
|
||||
{
|
||||
clientEntry = disconnectDueTimeout(emulContext, clientEntry);
|
||||
if(clientEntry == NULL)
|
||||
{
|
||||
break;
|
||||
}
|
||||
if(*emulContext->utilizedFlag)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if(ctx->orphanedAt == 0)
|
||||
{
|
||||
hasAliveClients = 1;
|
||||
}
|
||||
void* payload = ctx->incomeQ->head->payload;
|
||||
if(payload != NULL)
|
||||
{
|
||||
*emulContext->utilizedFlag = 1;
|
||||
BaseMessage* msg = payload;
|
||||
printf("client %lu sent data: \nnonce %lu, ptype %4u, ph: %u\n", ctx->clientId, msg->nonce, msg->packetType, msg->payloadHeader);
|
||||
|
||||
handleIncomingMessage(msg, ctx, emulContext);
|
||||
|
||||
|
||||
free(payload);
|
||||
ctx->incomeQ->head = ctx->incomeQ->head->nextEl;
|
||||
}
|
||||
clientEntry = clientEntry->nextEntry;
|
||||
}
|
||||
}
|
||||
|
||||
if(hasAliveClients == 0)
|
||||
{
|
||||
uint8_t newEmulState = switchNewEmulState(*emulContext->emulState, EMUL_STATE_OP_PAUSE);
|
||||
if(newEmulState != *emulContext->emulState)
|
||||
{
|
||||
printf("No alive clients, pausing execution\n");
|
||||
*emulContext->emulState = newEmulState;
|
||||
|
||||
size_t len = 0;
|
||||
uint8_t* notify = createControlNotifyMessage((uint64_t)~0, *emulContext->clockCounter, newEmulState, &len);
|
||||
broadcastClients(emulContext, notify, len);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
143
src/proto/handlers/auth.c
Normal file
143
src/proto/handlers/auth.c
Normal file
@@ -0,0 +1,143 @@
|
||||
#include "proto/handlers/auth.h"
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <time.h>
|
||||
#include <openssl/sha.h>
|
||||
|
||||
#include "my_mutex.h"
|
||||
|
||||
#include "proto/enums.h"
|
||||
#include "events.h"
|
||||
#include "panic.h"
|
||||
|
||||
|
||||
#include "proto/pack.h"
|
||||
#include "proto/dial.h"
|
||||
#include "proto/handlers/control.h"
|
||||
|
||||
uint8_t validateAccessTokenDeterministic(const uint8_t* data, const uint8_t* token, uint64_t timestamp)
|
||||
{
|
||||
char buf[1024];
|
||||
sprintf(buf, "%s%lu", token, timestamp);
|
||||
|
||||
uint8_t hash[SHA512_DIGEST_LENGTH];
|
||||
SHA512((uint8_t*)buf, strlen(buf), hash);
|
||||
|
||||
uint8_t valid = 1;
|
||||
|
||||
for(size_t i = 0; i < SHA512_DIGEST_LENGTH; i++)
|
||||
{
|
||||
if (data[i] != hash[i])
|
||||
{
|
||||
valid = 0;
|
||||
}
|
||||
}
|
||||
|
||||
return valid;
|
||||
}
|
||||
|
||||
uint8_t validateAccessToken(const uint8_t* data, const uint8_t* access_token)
|
||||
{
|
||||
uint64_t t = (uint64_t)time(NULL) / 30;
|
||||
|
||||
uint8_t valid1 = validateAccessTokenDeterministic(data, access_token, t);
|
||||
uint8_t valid2 = validateAccessTokenDeterministic(data, access_token, t - 1);
|
||||
|
||||
return valid1 || valid2;
|
||||
}
|
||||
|
||||
|
||||
uint8_t handle_auth(ClientContext* cctx, ws_cli_conn_t client, const uint8_t* msg, uint64_t msgSize, int msgType)
|
||||
{
|
||||
ServerContext* ctx = ws_get_server_context(client);
|
||||
if(msgSize != SHA512_DIGEST_LENGTH)
|
||||
{
|
||||
int ret = ws_close_client(client);
|
||||
if(ret == -1)
|
||||
{
|
||||
printf("Unable to close client %lu\n", client);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
uint8_t isValid = validateAccessToken(msg, (const uint8_t*)ctx->accessToken);
|
||||
if(!isValid)
|
||||
{
|
||||
printf("Auth token invalid\n");
|
||||
int ret = ws_close_client(client);
|
||||
if(ret == -1)
|
||||
{
|
||||
printf("Unable to close client %lu\n", client);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
char errbuf[1024];
|
||||
|
||||
printf("Auth token is valid\n");
|
||||
|
||||
ClientRegistrationEvent* ev = malloc(sizeof(ClientContext));
|
||||
if(ev == NULL)
|
||||
{
|
||||
panic("Unable to allocate register event");
|
||||
}
|
||||
ev->regType = REG_EVTYPE_AUTH;
|
||||
ev->ctx = cctx;
|
||||
|
||||
ws_set_connection_context(client, cctx);
|
||||
|
||||
|
||||
with_lock(&ctx->registerMutex)
|
||||
{
|
||||
printf("Writing auth event\n");
|
||||
int exitCode = ptQueuePush(ctx->regQueue, ev, errbuf);
|
||||
if(exitCode)
|
||||
{
|
||||
panic("Unable to push to reg queue: %s\n", errbuf);
|
||||
}
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
|
||||
LinkedListEntry* disconnectDueTimeout(EmulContext* emulContext, LinkedListEntry* clientEntry)
|
||||
{
|
||||
ClientContext* ctx = clientEntry->payload;
|
||||
|
||||
uint64_t now = (uint64_t)time(NULL);
|
||||
if(now - ctx->connectedAt <= 30)
|
||||
{
|
||||
return clientEntry->nextEntry;
|
||||
}
|
||||
|
||||
printf("Timeout on connection %lu\n", ctx->clientId);
|
||||
int ret = ws_close_client(ctx->clientId);
|
||||
if(ret == -1)
|
||||
{
|
||||
printf("Unable to close client\n");
|
||||
}
|
||||
LinkedListEntry* nextEntry = clientEntry->nextEntry;
|
||||
|
||||
removeLinkedListEntry(emulContext->clientsHead, clientEntry);
|
||||
clientEntry = nextEntry;
|
||||
*emulContext->utilizedFlag = 1;
|
||||
return clientEntry;
|
||||
}
|
||||
|
||||
void handleOnClientAuthDone(ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
uint8_t* framedata = malloc(sizeof(uint8_t) * 8);
|
||||
NULL_GUARD(framedata);
|
||||
|
||||
encodeUintToBytes(ctx->seatId, framedata);
|
||||
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, framedata, 8);
|
||||
|
||||
|
||||
size_t len = 0;
|
||||
uint8_t* msg = createControlNotifyMessage((uint64_t)~0, *emulContext->clockCounter, *emulContext->emulState, &len);
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, msg, len);
|
||||
|
||||
|
||||
uint8_t* outMsg = createClientSetup((uint64_t)~0, ctx, &len);
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, outMsg, len);
|
||||
}
|
||||
199
src/proto/handlers/control.c
Normal file
199
src/proto/handlers/control.c
Normal file
@@ -0,0 +1,199 @@
|
||||
#include "proto/handlers/control.h"
|
||||
|
||||
#include "proto/pack.h"
|
||||
#include "proto/enums.h"
|
||||
#include "panic.h"
|
||||
#include "state.h"
|
||||
|
||||
// #include <alloca.h>
|
||||
|
||||
|
||||
void handleIncomingExecControl(BaseMessage* msg, EmulContext* emulContext)
|
||||
{
|
||||
printf("ctrl exec\n");
|
||||
uint8_t stateOp = ((const uint8_t*)(msg->payload))[0];
|
||||
printf("state operation: %u\n", stateOp);
|
||||
uint8_t newEmulState = switchNewEmulState(*emulContext->emulState, stateOp);
|
||||
if(newEmulState != *emulContext->emulState)
|
||||
{
|
||||
printf("Switch state %u -> %u\n", *emulContext->emulState, newEmulState);
|
||||
*emulContext->emulState = newEmulState;
|
||||
|
||||
size_t len = 0;
|
||||
uint8_t* notify = createControlNotifyMessage(msg->nonce, *emulContext->clockCounter, newEmulState, &len);
|
||||
broadcastClients(emulContext, notify, len);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void handleIncomingEditConnectionSettingsControl(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
uint64_t fallbackQLen = decodeBytesToU64(msg->payload);
|
||||
uint64_t orphanedDeadTimeout = decodeBytesToU64(msg->payload + 8);
|
||||
|
||||
ctx->orphanedDeadTimeout = orphanedDeadTimeout;
|
||||
uint8_t* newFallbackQPtr = realloc(ctx->fallbackOutcomeQ->ptr, sizeof(OutgoingMessage) * fallbackQLen);
|
||||
NULL_GUARD(newFallbackQPtr);
|
||||
ctx->fallbackOutcomeQ->ptr = newFallbackQPtr;
|
||||
ctx->fallbackOutcomeQ->allocatedSize = fallbackQLen;
|
||||
|
||||
size_t len = 0;
|
||||
uint8_t* outMsg = createClientSetup(msg->nonce, ctx, &len);
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, outMsg, len);
|
||||
}
|
||||
|
||||
|
||||
|
||||
size_t fillOrphanedClientReportItem(uint8_t* outmsg, ClientContext* ctx)
|
||||
{
|
||||
printf("filling orphaned %lu with %lu packets\n", ctx->seatId, ctx->fallbackOutcomeQ->size - ctx->fallbackOutcomeQPadding);
|
||||
encodeUintToBytes(ctx->seatId, outmsg);
|
||||
encodeUintToBytes(ctx->orphanedAt, outmsg + 8);
|
||||
encodeUintToBytes((uint64_t)(ctx->fallbackOutcomeQ->size - ctx->fallbackOutcomeQPadding), outmsg + 8 + 8);
|
||||
return 8 * 3;
|
||||
}
|
||||
|
||||
void handleIncomingListOrphaned(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
LinkedListEntry* clientEntry = *emulContext->clientsHead;
|
||||
size_t orphanedCount = 0;
|
||||
size_t orphanedPackets = 0;
|
||||
|
||||
while(clientEntry != NULL)
|
||||
{
|
||||
ClientContext* orphanedCtx = clientEntry->payload;
|
||||
if(orphanedCtx->orphanedAt != 0)
|
||||
{
|
||||
orphanedCount++;
|
||||
orphanedPackets += orphanedCtx->fallbackOutcomeQ->size - ctx->fallbackOutcomeQPadding;
|
||||
}
|
||||
clientEntry = clientEntry->nextEntry;
|
||||
}
|
||||
|
||||
printf("found %lu orphaned clients with %lu packets\n", orphanedCount, orphanedPackets);
|
||||
|
||||
const size_t orphanedListReportItemSize = 8 * 3;
|
||||
const size_t msgLen = ((orphanedListReportItemSize * orphanedCount) + 9 + 8);
|
||||
uint8_t* outMsg = malloc(sizeof(uint8_t) * msgLen);
|
||||
NULL_GUARD(outMsg);
|
||||
fillHead(msg->nonce, PACKET_TYPE_CTRL, CTRL_TYPE_LIST_ORPHANED, outMsg);
|
||||
|
||||
|
||||
clientEntry = *emulContext->clientsHead;
|
||||
size_t filledCount = 0;
|
||||
uint8_t* outMsgCursor = outMsg + 9;
|
||||
encodeUintToBytes((uint64_t)orphanedCount, outMsgCursor);
|
||||
outMsgCursor += 8;
|
||||
|
||||
while(clientEntry != NULL && filledCount < orphanedCount)
|
||||
{
|
||||
ClientContext* orphanedCtx = clientEntry->payload;
|
||||
if(orphanedCtx->orphanedAt != 0)
|
||||
{
|
||||
size_t filled = fillOrphanedClientReportItem(outMsgCursor, orphanedCtx);
|
||||
outMsgCursor += filled;
|
||||
filledCount++;
|
||||
}
|
||||
clientEntry = clientEntry->nextEntry;
|
||||
}
|
||||
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, outMsg, msgLen);
|
||||
}
|
||||
|
||||
void handleIncomingLoadFailed(const BaseMessage* msg, ClientContext* ctx, const EmulContext* emulContext)
|
||||
{
|
||||
const uint8_t readMode = ((const uint8_t*)msg->payload)[0];
|
||||
const uint64_t seatId = decodeBytesToU64(((const uint8_t*)msg->payload) + 1);
|
||||
|
||||
|
||||
LinkedListEntry* clientEntry = *emulContext->clientsHead;
|
||||
ClientContext* orphanedCtx = NULL;
|
||||
|
||||
while(clientEntry != NULL)
|
||||
{
|
||||
orphanedCtx = clientEntry->payload;
|
||||
if(orphanedCtx->orphanedAt != 0)
|
||||
{
|
||||
if(orphanedCtx->seatId == seatId)
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
clientEntry = clientEntry->nextEntry;
|
||||
}
|
||||
|
||||
if(clientEntry == NULL || orphanedCtx == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
size_t payloadSize = 0;
|
||||
OutgoingMessage* failedQueue = (OutgoingMessage*)orphanedCtx->fallbackOutcomeQ->ptr;
|
||||
for(size_t i = orphanedCtx->fallbackOutcomeQPadding; i < orphanedCtx->fallbackOutcomeQ->size; i++)
|
||||
{
|
||||
OutgoingMessage* failedMessage = &failedQueue[i];
|
||||
payloadSize += 8;
|
||||
payloadSize += failedMessage->msgLen;
|
||||
}
|
||||
|
||||
const size_t outLen = 9 + payloadSize;
|
||||
|
||||
printf("preparing failed messages report: %lu messages, %lu payload size, %lu message size\n",
|
||||
orphanedCtx->fallbackOutcomeQ->size - orphanedCtx->fallbackOutcomeQPadding,
|
||||
payloadSize,
|
||||
outLen);
|
||||
|
||||
uint8_t* outMsg = malloc(sizeof(uint8_t) * outLen);
|
||||
fillHead(msg->nonce, PACKET_TYPE_CTRL, CTRL_TYPE_LOAD_FAILED, outMsg);
|
||||
uint8_t* outMsgCursor = outMsg + 9;
|
||||
|
||||
for(size_t i = orphanedCtx->fallbackOutcomeQPadding; i < orphanedCtx->fallbackOutcomeQ->size; i++)
|
||||
{
|
||||
OutgoingMessage* failedMessage = &failedQueue[i];
|
||||
encodeUintToBytes(failedMessage->msgLen, outMsgCursor);
|
||||
outMsgCursor += 8;
|
||||
for(size_t j = 0; j < failedMessage->msgLen; j++)
|
||||
{
|
||||
outMsgCursor[j] = failedMessage->msg[j];
|
||||
}
|
||||
outMsgCursor += failedMessage->msgLen;
|
||||
}
|
||||
|
||||
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, outMsg, outLen);
|
||||
if(readMode == 1)
|
||||
{
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, NULL, 0);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void handleIncomingControlMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
switch (msg->payloadHeader)
|
||||
{
|
||||
case CTRL_TYPE_EXEC:
|
||||
{
|
||||
handleIncomingExecControl(msg, emulContext);
|
||||
break;
|
||||
}
|
||||
case CTRL_TYPE_LIST_ORPHANED:
|
||||
{
|
||||
handleIncomingListOrphaned(msg, ctx, emulContext);
|
||||
break;
|
||||
}
|
||||
case CTRL_TYPE_LOAD_FAILED:
|
||||
{
|
||||
handleIncomingLoadFailed(msg, ctx, emulContext);
|
||||
break;
|
||||
}
|
||||
case CTRL_TYPE_SETUP_CONNECTION:
|
||||
{
|
||||
handleIncomingEditConnectionSettingsControl(msg, ctx, emulContext);
|
||||
break;
|
||||
}
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
84
src/proto/handlers/mem.c
Normal file
84
src/proto/handlers/mem.c
Normal file
@@ -0,0 +1,84 @@
|
||||
#include "proto/handlers/mem.h"
|
||||
#include "proto/enums.h"
|
||||
#include "proto/pack.h"
|
||||
#include "panic.h"
|
||||
#include <string.h>
|
||||
|
||||
|
||||
uint8_t* createMemReadResponseMessage(uint64_t nonce, uint64_t clockCounter, uint8_t* payload, size_t payloadLen, size_t* lenOut)
|
||||
{
|
||||
*lenOut = 9 + 8 + payloadLen;
|
||||
uint8_t* outmsg = malloc(sizeof(uint8_t) * (*lenOut));
|
||||
NULL_GUARD(outmsg);
|
||||
|
||||
encodeUintToBytes(nonce, outmsg);
|
||||
|
||||
outmsg[8] = PACKET_TYPE_MEM << 4;
|
||||
outmsg[8] |= MEM_TYPE_READ_RESP;
|
||||
|
||||
encodeUintToBytes(clockCounter, outmsg + 9);
|
||||
|
||||
memcpy(outmsg + 9 + 8, payload, payloadLen);
|
||||
|
||||
return outmsg;
|
||||
}
|
||||
|
||||
void handleIncomingMemReadReq(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
uint64_t devId = decodeBytesToU64(msg->payload);
|
||||
uint64_t segId = decodeBytesToU64(msg->payload + 8);
|
||||
uint64_t startAddr = decodeBytesToU64(msg->payload + 16);
|
||||
uint64_t readLen = decodeBytesToU64(msg->payload + 24);
|
||||
|
||||
if(devId >= emulContext->devicesCount)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
//TODO: lookup config for global addr conversion
|
||||
uint64_t globalAddr = startAddr + segId * 128;
|
||||
|
||||
uint8_t* readPtr = emulContext->devicesMem[devId] + globalAddr;
|
||||
|
||||
size_t outLen = 0;
|
||||
uint8_t* outMsg = createMemReadResponseMessage(msg->nonce, *emulContext->clockCounter, readPtr, readLen, &outLen);
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, outMsg, outLen);
|
||||
}
|
||||
|
||||
|
||||
void handleIncomingMemWrite(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
uint64_t devId = decodeBytesToU64(msg->payload);
|
||||
uint64_t segId = decodeBytesToU64(msg->payload + 8);
|
||||
uint64_t startAddr = decodeBytesToU64(msg->payload + 16);
|
||||
|
||||
if(devId >= emulContext->devicesCount)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
//TODO: lookup config for global addr conversion
|
||||
uint64_t globalAddr = startAddr + segId * 128;
|
||||
|
||||
uint8_t* writePtr = emulContext->devicesMem[devId] + globalAddr;
|
||||
|
||||
memcpy(writePtr, msg->payload + 24, msg->payloadLen - 24);
|
||||
|
||||
size_t outLen = 0;
|
||||
uint8_t* outMsg = createMemReadResponseMessage(msg->nonce, *emulContext->clockCounter, writePtr, msg->payloadLen - 24, &outLen);
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, outMsg, outLen);
|
||||
}
|
||||
|
||||
|
||||
void handleIncomingMemMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
if(msg->payloadHeader == MEM_TYPE_READ_REQ)
|
||||
{
|
||||
handleIncomingMemReadReq(msg, ctx, emulContext);
|
||||
}
|
||||
else if(msg->payloadHeader == MEM_TYPE_WRITE_PUSH)
|
||||
{
|
||||
handleIncomingMemWrite(msg, ctx, emulContext);
|
||||
}
|
||||
}
|
||||
124
src/proto/handlers/stream.c
Normal file
124
src/proto/handlers/stream.c
Normal file
@@ -0,0 +1,124 @@
|
||||
#include "proto/handlers/stream.h"
|
||||
|
||||
#include "streamed.h"
|
||||
#include "proto/pack.h"
|
||||
#include "proto/enums.h"
|
||||
#include "panic.h"
|
||||
|
||||
|
||||
|
||||
void unregisterClientStream(EmulContext* emulContext, ClientContext* ctx, uint32_t regId)
|
||||
{
|
||||
for(size_t deviceId = 0; deviceId < emulContext->devicesCount; deviceId++)
|
||||
{
|
||||
DeviceSegStreamReg* deviceRegs = emulContext->deviceStreamRegs[deviceId];
|
||||
for(size_t i = 0; i < deviceRegs->regCount; i++)
|
||||
{
|
||||
StreamReg* iReg = &deviceRegs->regs[i];
|
||||
if(iReg->clientContext->clientId == ctx->clientId && iReg->regId == regId)
|
||||
{
|
||||
printf("Discard stream %u register for client %lu\n", regId, ctx->clientId);
|
||||
deviceRegs->regCount -= 1;
|
||||
for(size_t j = i; j < deviceRegs->regCount; j++)
|
||||
{
|
||||
deviceRegs->regs[j] = deviceRegs->regs[j + 1];
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void unregisterClientStreams(EmulContext* emulContext, ClientContext* ctx)
|
||||
{
|
||||
for(size_t deviceId = 0; deviceId < emulContext->devicesCount; deviceId++)
|
||||
{
|
||||
DeviceSegStreamReg* deviceRegs = emulContext->deviceStreamRegs[deviceId];
|
||||
|
||||
StreamReg* newStreamRegs = malloc(sizeof(StreamReg) * deviceRegs->regCount);
|
||||
NULL_GUARD(newStreamRegs);
|
||||
size_t newStreamRegsId = 0;
|
||||
|
||||
for(size_t i = 0; i < deviceRegs->regCount; i++)
|
||||
{
|
||||
StreamReg* reg = &deviceRegs->regs[i];
|
||||
if(reg->clientContext->clientId != ctx->clientId)
|
||||
{
|
||||
newStreamRegs[newStreamRegsId] = *reg;
|
||||
newStreamRegsId++;
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("Removing reg %d mode stream for %lu/%lu+%lu: [%u]\n", reg->mode, deviceId, reg->startAddr, reg->segLen, reg->regId);
|
||||
}
|
||||
}
|
||||
free(deviceRegs->regs);
|
||||
deviceRegs->regCount = newStreamRegsId;
|
||||
deviceRegs->regs = newStreamRegs;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void handleStreamRegMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext, uint8_t X)
|
||||
{
|
||||
uint64_t devId = decodeBytesToU64(msg->payload);
|
||||
uint64_t segId = decodeBytesToU64(msg->payload + 8);
|
||||
uint64_t startAddr = decodeBytesToU64(msg->payload + 16);
|
||||
uint64_t segLen = decodeBytesToU64(msg->payload + 24);
|
||||
|
||||
if(devId >= emulContext->devicesCount)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
DeviceSegStreamReg* deviceRegs = emulContext->deviceStreamRegs[devId];
|
||||
if(deviceRegs->regCount + 1 >= deviceRegs->allocatedSize)
|
||||
{
|
||||
size_t newAllocatedSize = deviceRegs->allocatedSize * 2;
|
||||
if(newAllocatedSize <= deviceRegs->regCount + 1)
|
||||
{
|
||||
newAllocatedSize = deviceRegs->regCount + 1;
|
||||
}
|
||||
StreamReg* newRegs = realloc(deviceRegs->regs, sizeof(StreamReg) * newAllocatedSize);
|
||||
NULL_GUARD(newRegs);
|
||||
deviceRegs->allocatedSize = newAllocatedSize;
|
||||
deviceRegs->regs = newRegs;
|
||||
}
|
||||
|
||||
//TODO: lookup config for global segment addr conversion
|
||||
uint64_t globalStartAddr = startAddr + segId * 128;
|
||||
|
||||
StreamReg* reg = &deviceRegs->regs[deviceRegs->regCount];
|
||||
|
||||
reg->clientContext = ctx;
|
||||
reg->segLen = segLen;
|
||||
reg->startAddr = globalStartAddr;
|
||||
reg->regId = ctx->streamRegIterator;
|
||||
reg->mode = X;
|
||||
|
||||
ctx->streamRegIterator++;
|
||||
deviceRegs->regCount++;
|
||||
|
||||
printf("Done registering %d mode stream for %lu/%lu+%lu: [%u]\n", X, devId, globalStartAddr, segLen, reg->regId);
|
||||
|
||||
size_t len = 0;
|
||||
uint8_t* notifMsg = createDoneRegMessage(msg->nonce, X, devId, segId, startAddr, segLen, reg->regId, &len);
|
||||
dispatchOutgoingMessage(emulContext->outBufs, ctx, notifMsg, len);
|
||||
|
||||
}
|
||||
|
||||
void handleIncomingStreamMessage(BaseMessage* msg, ClientContext* ctx, EmulContext* emulContext)
|
||||
{
|
||||
uint8_t X = msg->payloadHeader >> 3;
|
||||
uint8_t streamType = msg->payloadHeader & 0b111;
|
||||
|
||||
if(streamType == STREAM_TYPE_REG_REQUEST)
|
||||
{
|
||||
handleStreamRegMessage(msg, ctx, emulContext, X);
|
||||
}
|
||||
else if(streamType == STREAM_TYPE_REG_DISCARD)
|
||||
{
|
||||
uint32_t regId = decodeBytesToU32(msg->payload);
|
||||
unregisterClientStream(emulContext, ctx, regId);
|
||||
}
|
||||
}
|
||||
156
src/proto/handlers/ws.c
Normal file
156
src/proto/handlers/ws.c
Normal file
@@ -0,0 +1,156 @@
|
||||
#include "proto/handlers/ws.h"
|
||||
#include "proto/handlers.h"
|
||||
|
||||
#include "panic.h"
|
||||
#include "proto/enums.h"
|
||||
#include "my_mutex.h"
|
||||
#include "proto/handlers/auth.h"
|
||||
#include "proto/msg.h"
|
||||
|
||||
void onWsOpen(ws_cli_conn_t client)
|
||||
{
|
||||
char errbuf[1024];
|
||||
|
||||
ptQueue* incomeQueue = ptQueueCreate(errbuf);
|
||||
NULL_GUARD(incomeQueue, "Unable to create income queue: %s\n", errbuf);
|
||||
|
||||
// ptQueue* outcomeQueue = ptQueueCreate(errbuf);
|
||||
|
||||
// NULL_GUARD(outcomeQueue, "Unable to create outcome queue: %s\n", errbuf);
|
||||
|
||||
ClientContext* cctx = malloc(sizeof(ClientContext));
|
||||
if(cctx == NULL)
|
||||
{
|
||||
ptQueueFree(incomeQueue);
|
||||
// ptQueueFree(outcomeQueue);
|
||||
panic("Unable to allocate client context\n");
|
||||
}
|
||||
cctx->clientId = client;
|
||||
cctx->isAuthed = 0;
|
||||
cctx->streamRegIterator = 0;
|
||||
cctx->incomeQ = incomeQueue;
|
||||
// cctx->outcomeQ = outcomeQueue;
|
||||
cctx->connectedAt = (uint64_t)time(NULL);
|
||||
cctx->orphanedAt = 0;
|
||||
cctx->orphanedDeadTimeout = 60;
|
||||
cctx->fallbackOutcomeQPadding = 0;
|
||||
cctx->fallbackOutcomeQ = malloc(sizeof(SizedPtr));
|
||||
NULL_GUARD(cctx->fallbackOutcomeQ);
|
||||
cctx->fallbackOutcomeQ->ptr = (void*) calloc(1024, sizeof(OutgoingMessage));
|
||||
NULL_GUARD(cctx->fallbackOutcomeQ->ptr);
|
||||
cctx->fallbackOutcomeQ->allocatedSize = 1024;
|
||||
cctx->fallbackOutcomeQ->size = 0;
|
||||
|
||||
ClientRegistrationEvent* ev = malloc(sizeof(ClientContext));
|
||||
if(ev == NULL)
|
||||
{
|
||||
ptQueueFree(incomeQueue);
|
||||
// ptQueueFree(outcomeQueue);
|
||||
free(cctx->fallbackOutcomeQ->ptr);
|
||||
free(cctx->fallbackOutcomeQ);
|
||||
free(cctx);
|
||||
|
||||
panic("Unable to allocate register event");
|
||||
}
|
||||
ev->regType = REG_EVTYPE_CONNECT;
|
||||
ev->ctx = cctx;
|
||||
|
||||
ws_set_connection_context(client, cctx);
|
||||
|
||||
ServerContext* ctx = ws_get_server_context(client);
|
||||
|
||||
with_lock(&ctx->registerMutex)
|
||||
{
|
||||
cctx->seatId = ctx->seatCounter;
|
||||
ctx->seatCounter++;
|
||||
|
||||
int exitCode = ptQueuePush(ctx->regQueue, ev, errbuf);
|
||||
if(exitCode)
|
||||
{
|
||||
ptQueueFree(incomeQueue);
|
||||
// ptQueueFree(outcomeQueue);
|
||||
free(cctx->fallbackOutcomeQ->ptr);
|
||||
free(cctx->fallbackOutcomeQ);
|
||||
free(cctx);
|
||||
|
||||
panic("Unable to push to reg queue: %s\n", errbuf);
|
||||
}
|
||||
}
|
||||
|
||||
char *cli;
|
||||
cli = ws_getaddress(client);
|
||||
printf("Connection %lu opened, addr: %s\n", client, cli);
|
||||
}
|
||||
|
||||
|
||||
void onWsClose(ws_cli_conn_t client)
|
||||
{
|
||||
pthread_t tid = pthread_self();
|
||||
char *cli;
|
||||
cli = ws_getaddress(client);
|
||||
printf("Connection %lu closed, addr: %s, thread id: %lu\n", client, cli, tid);
|
||||
|
||||
|
||||
char errbuf[1024];
|
||||
|
||||
ClientContext* cctx = ws_get_connection_context(client);
|
||||
if(cctx == NULL)
|
||||
{
|
||||
cctx = NULL;
|
||||
printf("Unable to get client context for client %lu\n", client);
|
||||
}
|
||||
|
||||
|
||||
ServerContext* ctx = ws_get_server_context(client);
|
||||
if(ctx == NULL)
|
||||
{
|
||||
printf("Unable to get server context for client %lu\n", client);
|
||||
return;
|
||||
}
|
||||
|
||||
ClientRegistrationEvent* ev = malloc(sizeof(ClientContext));
|
||||
NULL_GUARD(ev);
|
||||
ev->regType = REG_EVTYPE_CLOSE;
|
||||
ev->ctx = cctx;
|
||||
|
||||
|
||||
with_lock(&ctx->registerMutex)
|
||||
{
|
||||
int exitCode = ptQueuePush(ctx->regQueue, ev, errbuf);
|
||||
if(exitCode)
|
||||
{
|
||||
panic("Unable to push to reg queue: %s\n", errbuf);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
void onWsMessage(ws_cli_conn_t client, const unsigned char *msg, uint64_t size, int type)
|
||||
{
|
||||
ClientContext* cctx = ws_get_connection_context(client);
|
||||
if(cctx == NULL)
|
||||
{
|
||||
panic("Unable to get client context\n");
|
||||
}
|
||||
|
||||
if (cctx->isAuthed == 0)
|
||||
{
|
||||
handle_auth(cctx, client, msg, size, type);
|
||||
}
|
||||
else
|
||||
{
|
||||
char errbuf[1024];
|
||||
BaseMessage* baseMsg = parseMessage(msg, size);
|
||||
printf("msg\n");
|
||||
int isErr = ptQueuePush(cctx->incomeQ, baseMsg, errbuf);
|
||||
if(isErr)
|
||||
{
|
||||
panic("Unable to dispatch client message: %s\n", errbuf);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
107
src/proto/msg.c
Normal file
107
src/proto/msg.c
Normal file
@@ -0,0 +1,107 @@
|
||||
#include "proto/msg.h"
|
||||
#include "panic.h"
|
||||
|
||||
#include "proto/enums.h"
|
||||
#include "proto/pack.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
BaseMessage* parseMessage(const uint8_t* bytes, size_t size)
|
||||
{
|
||||
const uint8_t headerSize = 9;
|
||||
BaseMessage* msg = malloc(sizeof(BaseMessage));
|
||||
NULL_GUARD(msg);
|
||||
uint64_t nonce = decodeBytesToU64(bytes);
|
||||
|
||||
|
||||
msg->nonce = nonce;
|
||||
|
||||
msg->packetType = bytes[8] >> 4;
|
||||
msg->payloadHeader = bytes[8] & 0b1111;
|
||||
|
||||
msg->payloadLen = size - headerSize;
|
||||
|
||||
uint8_t* payload = malloc(sizeof(uint8_t) * (msg->payloadLen));
|
||||
NULL_GUARD(payload);
|
||||
|
||||
memcpy(payload, bytes + headerSize, msg->payloadLen);
|
||||
msg->payload = payload;
|
||||
|
||||
return msg;
|
||||
}
|
||||
|
||||
|
||||
void fillHead(uint64_t nonce, uint8_t packetType, uint8_t payloadHeader, uint8_t* outmsg)
|
||||
{
|
||||
encodeUintToBytes(nonce, outmsg);
|
||||
outmsg[8] = (uint8_t)((packetType & 0b1111) << 4);
|
||||
outmsg[8] |= payloadHeader & 0b1111;
|
||||
}
|
||||
|
||||
uint8_t* createControlNotifyMessage(uint64_t nonce, uint64_t clockCounter, uint8_t newEmulState, size_t* lenOut)
|
||||
{
|
||||
*lenOut = 9 + 10;
|
||||
uint8_t* outmsg = malloc(sizeof(uint8_t) * (*lenOut));
|
||||
NULL_GUARD(outmsg, "Unable to allocate message");
|
||||
|
||||
fillHead(nonce, PACKET_TYPE_CTRL, CTRL_TYPE_NOTIF_STATE, outmsg);
|
||||
|
||||
outmsg[9] = NOTIF_TYPE_EXEC;
|
||||
encodeUintToBytes(clockCounter, outmsg + 10);
|
||||
outmsg[18] = newEmulState;
|
||||
return outmsg;
|
||||
}
|
||||
|
||||
|
||||
uint8_t* createDoneRegMessage(uint64_t nonce, uint8_t X, uint64_t devId, uint64_t segId, uint64_t startAddr, uint64_t segLength, uint32_t regId, size_t* lenOut)
|
||||
{
|
||||
*lenOut = 36 + 9;
|
||||
uint8_t* outmsg = malloc(sizeof(uint8_t) * (*lenOut));
|
||||
NULL_GUARD(outmsg);
|
||||
|
||||
|
||||
fillHead(nonce, PACKET_TYPE_STREAM, (uint8_t)(X << 3) | STREAM_TYPE_REG_CONFIRM, outmsg);
|
||||
|
||||
encodeUintToBytes(devId, outmsg + 9);
|
||||
encodeUintToBytes(segId, outmsg + 9 + 8);
|
||||
encodeUintToBytes(startAddr, outmsg + 9 + 8 + 8);
|
||||
encodeUintToBytes(segLength, outmsg + 9 + 8 + 8 + 8);
|
||||
encodeUintToBytes(regId, outmsg + 9 + 8 + 8 + 8 + 8);
|
||||
return outmsg;
|
||||
}
|
||||
|
||||
|
||||
uint8_t* createStreamSegmentPush(uint8_t mode, uint32_t regId, uint64_t clockCounter, uint8_t* payload, size_t payloadLen, size_t* lenOut)
|
||||
{
|
||||
*lenOut = 9 + 4 + 8 + payloadLen;
|
||||
|
||||
uint8_t* outmsg = malloc(sizeof(uint8_t) * (*lenOut));
|
||||
NULL_GUARD(outmsg);
|
||||
|
||||
uint64_t nonce = (uint64_t)~0;
|
||||
encodeUintToBytes(nonce, outmsg);
|
||||
outmsg[8] = (uint8_t)((PACKET_TYPE_STREAM << 4) | (mode << 3) | STREAM_TYPE_SEND);
|
||||
|
||||
fillHead(nonce, PACKET_TYPE_STREAM, (uint8_t)((mode << 3) | STREAM_TYPE_SEND), outmsg);
|
||||
|
||||
encodeUintToBytes(regId, outmsg + 9);
|
||||
encodeUintToBytes(clockCounter, outmsg + 9 + 4);
|
||||
memcpy(outmsg + 9 + 4 + 8, payload, payloadLen);
|
||||
|
||||
return outmsg;
|
||||
}
|
||||
|
||||
|
||||
uint8_t* createClientSetup(uint64_t nonce, ClientContext* ctx, size_t* lenOut)
|
||||
{
|
||||
*lenOut = 9 + 8 + 8;
|
||||
uint8_t* outmsg = malloc(sizeof(uint8_t) * (*lenOut));
|
||||
NULL_GUARD(outmsg);
|
||||
|
||||
fillHead(nonce, PACKET_TYPE_CTRL, CTRL_TYPE_SETUP_CONNECTION, outmsg);
|
||||
uint8_t* payload = outmsg + 9;
|
||||
encodeUintToBytes((uint64_t)ctx->fallbackOutcomeQ->allocatedSize, payload);
|
||||
encodeUintToBytes((uint64_t)ctx->orphanedDeadTimeout, payload + 8);
|
||||
|
||||
return outmsg;
|
||||
}
|
||||
57
src/proto/pack.c
Normal file
57
src/proto/pack.c
Normal file
@@ -0,0 +1,57 @@
|
||||
#include "proto/pack.h"
|
||||
|
||||
uint64_t decodeBytesToU64(const uint8_t* bytes)
|
||||
{
|
||||
uint64_t ret = 0;
|
||||
for(uint8_t i = 0; i < 8; i++)
|
||||
{
|
||||
ret |= (uint64_t)((uint64_t)(bytes[i]) << ((7 - i) * 8));
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
uint32_t decodeBytesToU32(const uint8_t* bytes)
|
||||
{
|
||||
uint32_t ret = 0;
|
||||
for(uint8_t i = 0; i < 4; i++)
|
||||
{
|
||||
ret |= (uint32_t)((uint32_t)(bytes[i]) << ((3 - i) * 4));
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
uint16_t decodeBytesToU16(const uint8_t* bytes)
|
||||
{
|
||||
return ((uint16_t)(bytes[0] << 8)) | ((uint16_t)bytes[1]);
|
||||
}
|
||||
|
||||
void encodeUint8ToBytes(uint8_t num, uint8_t* tgt)
|
||||
{
|
||||
tgt[0] = num;
|
||||
}
|
||||
|
||||
|
||||
void encodeUint16ToBytes(uint16_t num, uint8_t* tgt)
|
||||
{
|
||||
for(uint8_t i = 0; i < 2; i++)
|
||||
{
|
||||
tgt[i] = (uint8_t)(((uint32_t)num) >> (2 * (1 - i)));
|
||||
}
|
||||
}
|
||||
|
||||
void encodeUint32ToBytes(uint32_t num, uint8_t* tgt)
|
||||
{
|
||||
for(uint8_t i = 0; i < 4; i++)
|
||||
{
|
||||
tgt[i] = (uint8_t)(((uint32_t)num) >> (4 * (3 - i)));
|
||||
}
|
||||
}
|
||||
|
||||
void encodeUint64ToBytes(uint64_t num, uint8_t* tgt)
|
||||
{
|
||||
for(uint8_t i = 0; i < 8; i++)
|
||||
{
|
||||
tgt[i] = (uint8_t)(((uint64_t)num) >> (8 * (7 - i)));
|
||||
}
|
||||
}
|
||||
|
||||
57
src/state.c
Normal file
57
src/state.c
Normal file
@@ -0,0 +1,57 @@
|
||||
#include <stdint.h>
|
||||
|
||||
#include "state.h"
|
||||
|
||||
|
||||
uint8_t switchNewEmulState(const uint8_t currentState, const uint8_t controlOp)
|
||||
{
|
||||
|
||||
switch (currentState)
|
||||
{
|
||||
case EMUL_STATE_STILL:
|
||||
{
|
||||
if(controlOp == EMUL_STATE_OP_START)
|
||||
{
|
||||
return EMUL_STATE_EXEC;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case EMUL_STATE_EXEC:
|
||||
{
|
||||
if(controlOp == EMUL_STATE_OP_PAUSE)
|
||||
{
|
||||
return EMUL_STATE_PAUSE;
|
||||
}
|
||||
else if(controlOp == EMUL_STATE_OP_STOP)
|
||||
{
|
||||
return EMUL_STATE_STOP;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case EMUL_STATE_PAUSE:
|
||||
{
|
||||
if(controlOp == EMUL_STATE_OP_RESUME)
|
||||
{
|
||||
return EMUL_STATE_EXEC;
|
||||
}
|
||||
else if (controlOp == EMUL_STATE_OP_RESET)
|
||||
{
|
||||
return EMUL_STATE_STILL;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case EMUL_STATE_STOP:
|
||||
{
|
||||
if(controlOp == EMUL_STATE_OP_RESET)
|
||||
{
|
||||
return EMUL_STATE_STILL;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return currentState;
|
||||
}
|
||||
Reference in New Issue
Block a user