void wi_test_set_creation(void) {
wi_set_t *set;
set = wi_set();
WI_TEST_ASSERT_NOT_NULL(set, "");
WI_TEST_ASSERT_EQUALS(wi_set_count(set), 0U, "");
set = wi_set_with_data(WI_STR("foo"), WI_STR("bar"), NULL);
WI_TEST_ASSERT_NOT_NULL(set, "");
WI_TEST_ASSERT_EQUALS(wi_set_count(set), 2U, "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("foo")), "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("bar")), "");
set = wi_autorelease(wi_set_init_with_data(wi_set_alloc(), WI_STR("foo"), WI_STR("bar"), NULL));
WI_TEST_ASSERT_NOT_NULL(set, "");
WI_TEST_ASSERT_EQUALS(wi_set_count(set), 2U, "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("foo")), "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("bar")), "");
set = wi_autorelease(wi_set_init_with_array(wi_set_alloc(), wi_array_with_data(WI_STR("foo"), WI_STR("bar"), NULL)));
WI_TEST_ASSERT_NOT_NULL(set, "");
WI_TEST_ASSERT_EQUALS(wi_set_count(set), 2U, "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("foo")), "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("bar")), "");
}
void wi_test_set_mutation(void) {
wi_mutable_set_t *set1, *set2;
set1 = wi_set_init_with_capacity_and_callbacks(wi_mutable_set_alloc(), 0, true, wi_set_null_callbacks);
WI_TEST_ASSERT_EQUALS(wi_set_count(set1), 0U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("foo")), 0U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("bar")), 0U, "");
wi_mutable_set_add_data(set1, WI_STR("foo"));
wi_mutable_set_add_data(set1, WI_STR("bar"));
WI_TEST_ASSERT_EQUALS(wi_set_count(set1), 2U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("foo")), 1U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("bar")), 1U, "");
wi_mutable_set_add_data(set1, WI_STR("foo"));
wi_mutable_set_add_data(set1, WI_STR("bar"));
WI_TEST_ASSERT_EQUALS(wi_set_count(set1), 2U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("foo")), 2U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("bar")), 2U, "");
wi_mutable_set_remove_data(set1, WI_STR("foo"));
wi_mutable_set_remove_data(set1, WI_STR("bar"));
WI_TEST_ASSERT_EQUALS(wi_set_count(set1), 2U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("foo")), 1U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("bar")), 1U, "");
wi_mutable_set_remove_data(set1, WI_STR("foo"));
wi_mutable_set_remove_data(set1, WI_STR("bar"));
WI_TEST_ASSERT_EQUALS(wi_set_count(set1), 0U, "");
wi_mutable_set_add_data(set1, WI_STR("foo"));
wi_mutable_set_add_data(set1, WI_STR("bar"));
set2 = wi_mutable_copy(set1);
wi_mutable_set_add_data(set2, WI_STR("baz"));
wi_mutable_set_set_set(set1, set2);
WI_TEST_ASSERT_EQUAL_INSTANCES(set1, set2, "");
wi_mutable_set_remove_all_data(set1);
WI_TEST_ASSERT_EQUALS(wi_set_count(set1), 0U, "");
wi_mutable_set_add_data_from_array(set1, wi_array_with_data(WI_STR("foo"), WI_STR("bar"), NULL));
WI_TEST_ASSERT_EQUALS(wi_set_count(set1), 2U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("foo")), 1U, "");
WI_TEST_ASSERT_EQUALS(wi_set_count_for_data(set1, WI_STR("bar")), 1U, "");
}
void wi_test_set_instances(void) {
wi_set_t *set;
wi_array_t *array;
set = wi_set_with_data(WI_STR("foo"), WI_STR("bar"), NULL);
WI_TEST_ASSERT_NOT_NULL(set, "");
WI_TEST_ASSERT_EQUALS(wi_set_count(set), 2U, "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("foo")), "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, WI_STR("bar")), "");
array = wi_set_all_data(set);
WI_TEST_ASSERT_EQUALS(wi_array_count(array), 2U, "");
WI_TEST_ASSERT_TRUE(wi_array_contains_data(array, WI_STR("foo")), "");
WI_TEST_ASSERT_TRUE(wi_array_contains_data(array, WI_STR("bar")), "");
}
void wi_test_set_scalars(void) {
wi_mutable_set_t *set;
wi_array_t *array;
set = wi_set_init_with_capacity_and_callbacks(wi_mutable_set_alloc(), 0, false, wi_set_null_callbacks);
wi_mutable_set_add_data(set, "foo");
wi_mutable_set_add_data(set, "bar");
WI_TEST_ASSERT_NOT_NULL(set, "");
WI_TEST_ASSERT_EQUALS(wi_set_count(set), 2U, "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, "foo"), "");
WI_TEST_ASSERT_TRUE(wi_set_contains_data(set, "bar"), "");
array = wi_set_all_data(set);
WI_TEST_ASSERT_EQUALS(wi_array_count(array), 2U, "");
WI_TEST_ASSERT_TRUE(wi_array_contains_data(array, "foo"), "");
WI_TEST_ASSERT_TRUE(wi_array_contains_data(array, "bar"), "");
}
static void wd_transfers_update_queue(void) {
wi_mutable_set_t *users;
wi_mutable_array_t *sorted_users, *transfers_queue, *failed_transfers;
wd_transfer_t *transfer;
wd_user_t *user;
wi_uinteger_t position;
wi_uinteger_t i, count;
wi_uinteger_t total_downloads, total_uploads, user_downloads, user_uploads, active_downloads, active_uploads;
wi_boolean_t queue;
wi_array_rdlock(wd_transfers);
total_downloads = wd_settings.totaldownloads;
user_downloads = wd_settings.clientdownloads;
total_uploads = wd_settings.totaluploads;
user_uploads = wd_settings.clientuploads;
active_downloads = 0;
active_uploads = 0;
failed_transfers = wi_array_init(wi_mutable_array_alloc());
users = wi_set_init(wi_mutable_set_alloc());
count = wi_array_count(wd_transfers);
for(i = 0; i < count; i++) {
transfer = WI_ARRAY(wd_transfers, i);
if(wd_transfer_state(transfer) == WD_TRANSFER_QUEUED) {
wi_mutable_array_add_data(wd_user_transfers_queue(transfer->user), transfer);
wi_mutable_set_add_data(users, transfer->user);
}
wd_user_clear_downloads(transfer->user);
wd_user_clear_uploads(transfer->user);
}
for(i = 0; i < count; i++) {
transfer = WI_ARRAY(wd_transfers, i);
if(wd_transfer_state(transfer) == WD_TRANSFER_RUNNING) {
if(transfer->type == WD_TRANSFER_DOWNLOAD) {
active_downloads++;
wd_user_increase_downloads(transfer->user);
} else {
active_uploads++;
wd_user_increase_uploads(transfer->user);
}
}
}
count = wi_set_count(users);
if(count > 0) {
sorted_users = wi_autorelease(wi_mutable_copy(wi_set_all_data(users)));
wi_mutable_array_sort(sorted_users, wd_transfers_compare_user);
position = 1;
while(count > 0) {
for(i = 0; i < count; i++) {
user = WI_ARRAY(sorted_users, i);
transfers_queue = wd_user_transfers_queue(user);
transfer = WI_ARRAY(transfers_queue, 0);
if(transfer->type == WD_TRANSFER_DOWNLOAD) {
queue = ((total_downloads > 0 && active_downloads >= total_downloads) ||
(user_downloads > 0 && wd_user_downloads(transfer->user) >= user_downloads));
} else {
queue = ((total_uploads > 0 && active_uploads >= total_uploads) ||
(user_uploads > 0 && wd_user_uploads(transfer->user) >= user_uploads));
}
if(queue) {
if(transfer->queue != position) {
transfer->queue = position;
wd_user_lock_socket(transfer->user);
wd_sreply(wd_user_socket(transfer->user), 401, WI_STR("%#@%c%u"),
transfer->path, WD_FIELD_SEPARATOR,
transfer->queue);
wd_user_unlock_socket(transfer->user);
}
position++;
} else {
transfer->queue = 0;
transfer->waiting_time = wi_time_interval();
wd_transfer_set_state(transfer, WD_TRANSFER_WAITING);
if(wd_transfer_open(transfer)) {
wd_transfer_create_timer(transfer);
wd_user_lock_socket(transfer->user);
wd_sreply(wd_user_socket(transfer->user), 400, WI_STR("%#@%c%llu%c%#@"),
transfer->path, WD_FIELD_SEPARATOR,
transfer->offset, WD_FIELD_SEPARATOR,
transfer->hash);
wd_user_unlock_socket(transfer->user);
} else {
wd_user_lock_socket(transfer->user);
wd_sreply(wd_user_socket(transfer->user), 500, WI_STR("Command Failed"));
wd_user_unlock_socket(transfer->user);
wi_mutable_array_add_data(failed_transfers, transfer);
}
}
wi_mutable_array_remove_data_at_index(transfers_queue, 0);
if(wi_array_count(transfers_queue) == 0) {
wi_mutable_array_remove_data_at_index(sorted_users, i);
i--;
count--;
}
}
}
}
wi_mutable_array_remove_data_in_array(wd_transfers, failed_transfers);
wi_array_unlock(wd_transfers);
wi_release(users);
wi_release(failed_transfers);
}
