static struct tlv_packet *core_set_session_guid(struct tlv_handler_ctx *ctx)
{
size_t guid_len = 0;
struct mettle *m = ctx->arg;
struct tlv_dispatcher *td = mettle_get_tlv_dispatcher(m);
char *guid = tlv_packet_get_raw(ctx->req, TLV_TYPE_SESSION_GUID, &guid_len);
if (!guid || guid_len != SESSION_GUID_LEN)
return tlv_packet_response_result(ctx, TLV_RESULT_FAILURE);
tlv_dispatcher_set_session_guid(td, guid);
return tlv_packet_response_result(ctx, TLV_RESULT_SUCCESS);
}
static struct tlv_packet *core_set_uuid(struct tlv_handler_ctx *ctx)
{
size_t uuid_len = 0;
struct mettle *m = ctx->arg;
struct tlv_dispatcher *td = mettle_get_tlv_dispatcher(m);
char *uuid = tlv_packet_get_raw(ctx->req, TLV_TYPE_UUID, &uuid_len);
if (uuid && uuid_len) {
tlv_dispatcher_set_uuid(td, uuid, uuid_len);
}
return tlv_packet_response_result(ctx, TLV_RESULT_SUCCESS);
}
static struct tlv_packet *core_loadlib(struct tlv_handler_ctx *ctx)
{
uint32_t flags;
size_t extension_len;
struct mettle *m = ctx->arg;
struct tlv_dispatcher *td = mettle_get_tlv_dispatcher(m);
struct tlv_packet *p = tlv_packet_response(ctx);
int fd = -1;
int tlv_result = TLV_RESULT_FAILURE;
const char *library_path = tlv_packet_get_str(ctx->req, TLV_TYPE_LIBRARY_PATH);
const char *target_path = tlv_packet_get_str(ctx->req, TLV_TYPE_TARGET_PATH);
const unsigned char *extension = tlv_packet_get_raw(ctx->req, TLV_TYPE_DATA, &extension_len);
tlv_packet_get_u32(ctx->req, TLV_TYPE_FLAGS, &flags);
if (!extension || extension_len == 0) {
log_error("No extension received");
goto done;
}
if (!library_path) {
log_error("No extension name specified");
goto done;
}
/* Check if this is a binary image or executable. */
const char bin_magic_number[] = BIN_MAGIC_NUMBER;
if (strncmp((const char *)&extension[extension_len - sizeof(bin_magic_number)], bin_magic_number, sizeof(bin_magic_number)) == 0) {
log_info("Loading extension '%s' from binary image", library_path);
// Make a copy of this image, don't use the TLV buffer...
unsigned char *extension_copy = malloc(extension_len);
if (extension_copy == NULL) {
log_error("Failed to allocate memory for '%s' binary image", library_path);
goto done;
}
// TODO: free this mem when the extension no longer is running
memcpy(extension_copy, extension, extension_len);
if (extension_start_binary_image(m, library_path, extension, extension_len, NULL))
{
log_error("Failed to start extension from binary image '%s'", library_path);
goto done;
}
} else {
/* This is an executable that needs to be saved to the filesystem. */
log_info("Loading extension '%s' from executable file", library_path);
fd = open(target_path, O_CREAT|O_WRONLY, 0755);
if(fd == -1) {
log_error("Failed to open '%s': %s", target_path, strerror(errno));
goto done;
}
int ret_val = write(fd, extension, extension_len);
if (ret_val == -1) {
log_error("Failed to write '%s': %s", target_path, strerror(errno));
goto done;
}
if (ret_val != extension_len) {
log_error("Failed to write the entire extension '%s' to disk", target_path);
goto done;
}
close(fd);
fd = -1;
if (extension_start_executable(m, target_path, NULL))
{
log_error("Failed to start extension from file '%s'", target_path);
goto done;
}
}
tlv_result = TLV_RESULT_SUCCESS;
done:
if (fd != -1) {
close(fd);
}
p = tlv_packet_add_result(p, tlv_result);
return p;
}
