void test_profile_sync_get_size(void)
{
KAA_TRACE_IN(logger);
kaa_error_t error_code = KAA_ERR_NONE;
kaa_profile_t *profile = kaa_profile_basic_endpoint_profile_test_create();
profile->profile_body = kaa_string_copy_create("dummy");
size_t serialized_profile_size = profile->get_size(profile);
char *serialized_profile = (char *) KAA_MALLOC(serialized_profile_size * sizeof(char));
avro_writer_t writer = avro_writer_memory(serialized_profile, serialized_profile_size);
profile->serialize(writer, profile);
size_t expected_size = KAA_EXTENSION_HEADER_SIZE
+ sizeof(uint32_t) // profile size
+ kaa_aligned_size_get(serialized_profile_size);
size_t profile_sync_size = 0;
error_code = kaa_profile_manager_update_profile(profile_manager, profile);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
status->is_registered = true;
error_code = kaa_profile_request_get_size(profile_manager, &profile_sync_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
ASSERT_EQUAL(expected_size, profile_sync_size);
status->is_registered = false;
expected_size += sizeof(uint32_t)
+ TEST_PUB_KEY_SIZE;
error_code = kaa_profile_request_get_size(profile_manager, &profile_sync_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
ASSERT_EQUAL(expected_size, profile_sync_size);
const char *access_token = "access token";
error_code = kaa_profile_manager_set_endpoint_access_token(profile_manager, access_token);
expected_size += sizeof(uint32_t)
+ strlen(access_token);
error_code = kaa_profile_request_get_size(profile_manager, &profile_sync_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
ASSERT_EQUAL(expected_size, profile_sync_size);
avro_writer_free(writer);
KAA_FREE(serialized_profile);
profile->destroy(profile);
KAA_TRACE_OUT(logger);
}
kaa_error_t kaa_extension_profile_request_serialize(void *context, uint32_t request_id,
uint8_t *buffer, size_t *size, bool *need_resync)
{
(void)request_id;
// TODO(KAA-982): Use asserts
if (!context || !size || !need_resync) {
return KAA_ERR_BADPARAM;
}
*need_resync = resync_is_required(context);
if (!*need_resync) {
*size = 0;
return KAA_ERR_NONE;
}
size_t size_needed;
kaa_error_t error = kaa_profile_request_get_size(context, &size_needed);
if (error) {
return error;
}
if (!buffer || *size < size_needed) {
*size = size_needed;
return KAA_ERR_BUFFER_IS_NOT_ENOUGH;
}
*size = size_needed;
kaa_platform_message_writer_t writer = KAA_MESSAGE_WRITER(buffer, *size);
error = kaa_profile_request_serialize(context, &writer);
if (error) {
return error;
}
*size = writer.current - buffer;
return KAA_ERR_NONE;
}
kaa_error_t kaa_extension_profile_request_get_size(void *context,
size_t *expected_size)
{
return kaa_profile_request_get_size(context, expected_size);
}
void test_profile_sync_serialize(void)
{
KAA_TRACE_IN(logger);
kaa_error_t error_code;
kaa_platform_message_writer_t *manual_writer;
kaa_platform_message_writer_t *auto_writer;
const char *access_token = "access token";
const size_t access_token_size = strlen(access_token);
kaa_profile_t *profile = kaa_profile_basic_endpoint_profile_test_create();
profile->profile_body = kaa_string_copy_create("dummy");
size_t serialized_profile_size = profile->get_size(profile);
char *serialized_profile = (char *) KAA_MALLOC(serialized_profile_size * sizeof(char));
avro_writer_t avro_writer = avro_writer_memory(serialized_profile, serialized_profile_size);
profile->serialize(avro_writer, profile);
error_code = kaa_profile_manager_update_profile(profile_manager, profile);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
status->is_registered = false;
error_code = kaa_profile_manager_set_endpoint_access_token(profile_manager, access_token);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
size_t profile_sync_size;
error_code = kaa_profile_request_get_size(profile_manager, &profile_sync_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
char buffer[profile_sync_size];
error_code = kaa_platform_message_writer_create(&manual_writer, buffer, profile_sync_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
uint32_t network_order_32;
error_code = kaa_platform_message_write_extension_header(manual_writer
, KAA_PROFILE_EXTENSION_TYPE
, 0
, profile_sync_size - KAA_EXTENSION_HEADER_SIZE);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
bool need_resync = true;
ASSERT_EQUAL(kaa_profile_need_profile_resync(profile_manager, &need_resync), KAA_ERR_NONE);
network_order_32 = KAA_HTONL(0);
if (need_resync)
network_order_32 = KAA_HTONL(serialized_profile_size);
error_code = kaa_platform_message_write(manual_writer, &network_order_32, sizeof(uint32_t));
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
if (need_resync) {
error_code = kaa_platform_message_write_aligned(manual_writer, serialized_profile, serialized_profile_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
}
network_order_32 = KAA_HTONS(TEST_PUB_KEY_SIZE) << 16 | PUB_KEY_VALUE;
error_code = kaa_platform_message_write(manual_writer, &network_order_32, sizeof(uint32_t));
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
error_code = kaa_platform_message_write_aligned(manual_writer, test_ep_key, TEST_PUB_KEY_SIZE);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
network_order_32 = KAA_HTONS(access_token_size) << 16 | ACCESS_TOKEN_VALUE;
error_code = kaa_platform_message_write(manual_writer, &network_order_32, sizeof(uint32_t));
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
error_code = kaa_platform_message_write_aligned(manual_writer, access_token, access_token_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
char buffer2[profile_sync_size];
error_code = kaa_platform_message_writer_create(&auto_writer, buffer2, profile_sync_size);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
error_code = kaa_profile_request_serialize(profile_manager, auto_writer);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
error_code = (memcmp(buffer, buffer2, profile_sync_size) == 0 ? KAA_ERR_NONE : KAA_ERR_BADDATA);
ASSERT_EQUAL(error_code, KAA_ERR_NONE);
KAA_FREE(serialized_profile);
avro_writer_free(avro_writer);
profile->destroy(profile);
kaa_platform_message_writer_destroy(auto_writer);
kaa_platform_message_writer_destroy(manual_writer);
KAA_TRACE_OUT(logger);
}
static kaa_error_t kaa_client_sync_get_size(kaa_platform_protocol_t *self
, const kaa_service_t services[]
, size_t services_count
, size_t *expected_size)
{
KAA_RETURN_IF_NIL4(self, services, services_count, expected_size, KAA_ERR_BADPARAM)
*expected_size = KAA_PROTOCOL_MESSAGE_HEADER_SIZE;
size_t extension_size = 0;
kaa_error_t err_code = kaa_meta_data_request_get_size(&extension_size);
if (err_code) {
KAA_LOG_ERROR(self->logger, err_code, "Failed to query size for meta extension");
return err_code;
}
*expected_size += extension_size;
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated meta extension size %u", extension_size);
while (!err_code && services_count--) {
switch (services[services_count]) {
case KAA_SERVICE_BOOTSTRAP: {
err_code = kaa_channel_manager_bootstrap_request_get_size(self->kaa_context->channel_manager
, &extension_size);
if (!err_code)
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated bootstrap extension size %u", extension_size);
break;
}
case KAA_SERVICE_PROFILE: {
bool need_resync = false;
err_code = kaa_profile_need_profile_resync(self->kaa_context->profile_manager
, &need_resync);
if (err_code) {
KAA_LOG_ERROR(self->logger, err_code, "Failed to read 'need_resync' flag");
}
if (!err_code && need_resync) {
err_code = kaa_profile_request_get_size(self->kaa_context->profile_manager
, &extension_size);
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated profile extension size %u", extension_size);
}
break;
}
case KAA_SERVICE_USER: {
err_code = kaa_user_request_get_size(self->kaa_context->user_manager
, &extension_size);
if (!err_code)
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated user extension size %u", extension_size);
break;
}
#ifndef KAA_DISABLE_FEATURE_EVENTS
case KAA_SERVICE_EVENT: {
err_code = kaa_event_request_get_size(self->kaa_context->event_manager
, &extension_size);
if (!err_code)
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated event extension size %u", extension_size);
break;
}
#endif
#ifndef KAA_DISABLE_FEATURE_LOGGING
case KAA_SERVICE_LOGGING: {
err_code = kaa_logging_request_get_size(self->kaa_context->log_collector
, &extension_size);
if (!err_code)
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated logging extension size %u", extension_size);
break;
}
#endif
#ifndef KAA_DISABLE_FEATURE_CONFIGURATION
case KAA_SERVICE_CONFIGURATION: {
err_code = kaa_configuration_manager_get_size(self->kaa_context->configuration_manager
, &extension_size);
if (!err_code)
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated configuration extension size %u", extension_size);
break;
}
#endif
#ifndef KAA_DISABLE_FEATURE_NOTIFICATION
case KAA_SERVICE_NOTIFICATION: {
err_code = kaa_notification_manager_get_size(self->kaa_context->notification_manager
, &extension_size);
if (!err_code)
KAA_LOG_TRACE(self->logger, KAA_ERR_NONE, "Calculated notification extension size %u", extension_size);
break;
}
#endif
default:
extension_size = 0;
break;
}
*expected_size += extension_size;
}
if (err_code) {
KAA_LOG_ERROR(self->logger, err_code, "Failed to query extension size in %u service"
, services[services_count]);
}
return err_code;
}
