token: Added function to remove duplicates
Added a function to remove duplicates from lists. This function is used in a new provided function to append lists removing duplicates. Signed-off-by: Anderson Toshiyuki Sasaki <ansasaki@redhat.com> Reviewed-by: Jakub Jelen <jjelen@redhat.com>
Этот коммит содержится в:
Anderson Toshiyuki Sasaki
родитель
e5a64a3d6b
Коммит
548753b338
@ -38,7 +38,11 @@ void ssh_tokens_free(struct ssh_tokens_st *tokens);
|
||||
char *ssh_find_matching(const char *available_d,
|
||||
const char *preferred_d);
|
||||
|
||||
|
||||
char *ssh_find_all_matching(const char *available_d,
|
||||
const char *preferred_d);
|
||||
|
||||
char *ssh_remove_duplicates(const char *list);
|
||||
|
||||
char *ssh_append_without_duplicates(const char *list,
|
||||
const char *appended_list);
|
||||
#endif /* TOKEN_H_ */
|
||||
|
||||
150
src/token.c
150
src/token.c
@ -26,6 +26,7 @@
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdbool.h>
|
||||
|
||||
#include "libssh/priv.h"
|
||||
#include "libssh/token.h"
|
||||
@ -260,3 +261,152 @@ out:
|
||||
ssh_tokens_free(p_tok);
|
||||
return ret;
|
||||
}
|
||||
|
||||
/**
|
||||
* @internal
|
||||
*
|
||||
* @brief Given a string containing a list of elements, remove all duplicates
|
||||
* and return in a newly allocated string.
|
||||
*
|
||||
* @param[in] list The list to be freed of duplicates
|
||||
*
|
||||
* @return A newly allocated copy of the string free of duplicates; NULL in
|
||||
* case of error.
|
||||
*/
|
||||
char *ssh_remove_duplicates(const char *list)
|
||||
{
|
||||
struct ssh_tokens_st *tok = NULL;
|
||||
|
||||
size_t i, j, num_tokens, max_len;
|
||||
char *ret = NULL;
|
||||
bool *should_copy = NULL, need_comma = false;
|
||||
|
||||
if (list == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* The maximum number of tokens is the size of the list */
|
||||
max_len = strlen(list);
|
||||
if (max_len == 0) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* Add space for ending '\0' */
|
||||
max_len++;
|
||||
|
||||
tok = ssh_tokenize(list, ',');
|
||||
if ((tok == NULL) || (tok->tokens == NULL) || (tok->tokens[0] == NULL)) {
|
||||
goto out;
|
||||
}
|
||||
|
||||
should_copy = calloc(1, max_len);
|
||||
if (should_copy == NULL) {
|
||||
goto out;
|
||||
}
|
||||
|
||||
if (strlen(tok->tokens[0]) > 0) {
|
||||
should_copy[0] = true;
|
||||
}
|
||||
|
||||
for (i = 1; tok->tokens[i]; i++) {
|
||||
for (j = 0; j < i; j++) {
|
||||
if (strcmp(tok->tokens[i], tok->tokens[j]) == 0) {
|
||||
/* Found a duplicate; do not copy */
|
||||
should_copy[i] = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/* No matching token before */
|
||||
if (j == i) {
|
||||
/* Only copy if it is not an empty string */
|
||||
if (strlen(tok->tokens[i]) > 0) {
|
||||
should_copy[i] = true;
|
||||
} else {
|
||||
should_copy[i] = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
num_tokens = i;
|
||||
|
||||
ret = calloc(1, max_len);
|
||||
if (ret == NULL) {
|
||||
goto out;
|
||||
}
|
||||
|
||||
for (i = 0; i < num_tokens; i++) {
|
||||
if (should_copy[i]) {
|
||||
if (need_comma) {
|
||||
strncat(ret, ",", (max_len - strlen(ret) - 1));
|
||||
}
|
||||
strncat(ret, tok->tokens[i], (max_len - strlen(ret) - 1));
|
||||
need_comma = true;
|
||||
}
|
||||
}
|
||||
|
||||
/* If no comma is needed, nothing was copied */
|
||||
if (!need_comma) {
|
||||
SAFE_FREE(ret);
|
||||
}
|
||||
|
||||
out:
|
||||
SAFE_FREE(should_copy);
|
||||
ssh_tokens_free(tok);
|
||||
return ret;
|
||||
}
|
||||
|
||||
/**
|
||||
* @internal
|
||||
*
|
||||
* @brief Given two strings containing lists of tokens, return a newly
|
||||
* allocated string containing all the elements of the first list appended with
|
||||
* all the elements of the second list, without duplicates. The order of the
|
||||
* elements will be preserved.
|
||||
*
|
||||
* @param[in] list The first list
|
||||
* @param[in] appended_list The list to be appended
|
||||
*
|
||||
* @return A newly allocated copy list containing all the elements of the
|
||||
* kept_list appended with the elements of the appended_list without duplicates;
|
||||
* NULL in case of error.
|
||||
*/
|
||||
char *ssh_append_without_duplicates(const char *list,
|
||||
const char *appended_list)
|
||||
{
|
||||
size_t concat_len = 0;
|
||||
char *ret = NULL, *concat = NULL;
|
||||
|
||||
if (list != NULL) {
|
||||
concat_len = strlen(list);
|
||||
}
|
||||
|
||||
if (appended_list != NULL) {
|
||||
concat_len += strlen(appended_list);
|
||||
}
|
||||
|
||||
if (concat_len == 0) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* Add room for ending '\0' and for middle ',' */
|
||||
concat_len += 2;
|
||||
concat = calloc(1, concat_len);
|
||||
if (concat == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (list != NULL) {
|
||||
strcpy(concat, list);
|
||||
strncat(concat, ",", concat_len - strlen(concat) - 1);
|
||||
}
|
||||
if (appended_list != NULL) {
|
||||
strncat(concat, appended_list, concat_len - strlen(concat) - 1);
|
||||
}
|
||||
|
||||
ret = ssh_remove_duplicates(concat);
|
||||
|
||||
SAFE_FREE(concat);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
@ -146,6 +146,126 @@ static void torture_tokens_sanity(UNUSED_PARAM(void **state))
|
||||
tokenize_compare_expected(",", single_colon, 1);
|
||||
}
|
||||
|
||||
static void torture_remove_duplicate(UNUSED_PARAM(void **state))
|
||||
{
|
||||
|
||||
const char *simple[] = {"a,a,b,b,c,c",
|
||||
"a,b,c,a,b,c",
|
||||
"a,b,c,c,b,a",
|
||||
"a,a,,b,b,,c,c",
|
||||
",a,a,b,b,c,c",
|
||||
"a,a,b,b,c,c,"};
|
||||
const char *empty[] = {"",
|
||||
",,,,,,,,,",
|
||||
NULL};
|
||||
char *ret = NULL;
|
||||
int i;
|
||||
|
||||
for (i = 0; i < 6; i++) {
|
||||
ret = ssh_remove_duplicates(simple[i]);
|
||||
assert_non_null(ret);
|
||||
assert_string_equal("a,b,c", ret);
|
||||
printf("simple[%d] resulted in '%s'\n", i, ret);
|
||||
SAFE_FREE(ret);
|
||||
}
|
||||
|
||||
for (i = 0; i < 3; i++) {
|
||||
ret = ssh_remove_duplicates(empty[i]);
|
||||
if (ret != NULL) {
|
||||
printf("empty[%d] resulted in '%s'\n", i, ret);
|
||||
}
|
||||
assert_null(ret);
|
||||
}
|
||||
|
||||
ret = ssh_remove_duplicates("a");
|
||||
assert_non_null(ret);
|
||||
assert_string_equal("a", ret);
|
||||
SAFE_FREE(ret);
|
||||
}
|
||||
|
||||
static void torture_append_without_duplicate(UNUSED_PARAM(void **state))
|
||||
{
|
||||
const char *s1[] = {"a,a,b,b,c,c",
|
||||
"a,b,c,a,b,c",
|
||||
"a,b,c,c,b,a",
|
||||
"a,a,,b,b,,c,c",
|
||||
",a,a,b,b,c,c",
|
||||
"a,a,b,b,c,c,"};
|
||||
const char *s2[] = {"a,a,b,b,c,c,d,d",
|
||||
"a,b,c,d,a,b,c,d",
|
||||
"a,b,c,d,d,c,b,a",
|
||||
"a,a,,b,b,,c,c,,d,d",
|
||||
",a,a,b,b,c,c,d,d",
|
||||
"a,a,b,b,c,c,d,d,",
|
||||
"d"};
|
||||
const char *empty[] = {"",
|
||||
",,,,,,,,,",
|
||||
NULL,
|
||||
NULL};
|
||||
char *ret = NULL;
|
||||
int i, j;
|
||||
|
||||
ret = ssh_append_without_duplicates("a", "a");
|
||||
assert_non_null(ret);
|
||||
assert_string_equal("a", ret);
|
||||
SAFE_FREE(ret);
|
||||
|
||||
ret = ssh_append_without_duplicates("a", "b");
|
||||
assert_non_null(ret);
|
||||
assert_string_equal("a,b", ret);
|
||||
SAFE_FREE(ret);
|
||||
|
||||
ret = ssh_append_without_duplicates("a", NULL);
|
||||
assert_non_null(ret);
|
||||
assert_string_equal("a", ret);
|
||||
SAFE_FREE(ret);
|
||||
|
||||
ret = ssh_append_without_duplicates(NULL, "b");
|
||||
assert_non_null(ret);
|
||||
assert_string_equal("b", ret);
|
||||
SAFE_FREE(ret);
|
||||
|
||||
for (i = 0; i < 6; i++) {
|
||||
for (j = 0; j < 7; j++) {
|
||||
ret = ssh_append_without_duplicates(s1[i], s2[j]);
|
||||
assert_non_null(ret);
|
||||
printf("s1[%d] + s2[%d] resulted in '%s'\n", i, j, ret);
|
||||
assert_string_equal("a,b,c,d", ret);
|
||||
SAFE_FREE(ret);
|
||||
}
|
||||
}
|
||||
|
||||
for (i = 0; i < 6; i++) {
|
||||
for (j = 0; j < 3; j++) {
|
||||
ret = ssh_append_without_duplicates(s1[i], empty[j]);
|
||||
assert_non_null(ret);
|
||||
printf("s1[%d] + empty[%d] resulted in '%s'\n", i, j, ret);
|
||||
assert_string_equal("a,b,c", ret);
|
||||
SAFE_FREE(ret);
|
||||
}
|
||||
}
|
||||
|
||||
for (i = 0; i < 3; i++) {
|
||||
for (j = 0; j < 6; j++) {
|
||||
ret = ssh_append_without_duplicates(empty[i], s1[j]);
|
||||
assert_non_null(ret);
|
||||
printf("empty[%d] + s1[%d] resulted in '%s'\n", i, j, ret);
|
||||
assert_string_equal("a,b,c", ret);
|
||||
SAFE_FREE(ret);
|
||||
}
|
||||
}
|
||||
for (i = 0; i < 4; i++) {
|
||||
for (j = 0; j < 4; j++) {
|
||||
ret = ssh_append_without_duplicates(empty[i], empty[j]);
|
||||
if (ret != NULL) {
|
||||
printf("empty[%d] + empty[%d] resulted in '%s'\n", i, j, ret);
|
||||
}
|
||||
assert_null(ret);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int torture_run_tests(void)
|
||||
{
|
||||
int rc;
|
||||
@ -153,6 +273,8 @@ int torture_run_tests(void)
|
||||
cmocka_unit_test(torture_tokens_sanity),
|
||||
cmocka_unit_test(torture_find_matching),
|
||||
cmocka_unit_test(torture_find_all_matching),
|
||||
cmocka_unit_test(torture_remove_duplicate),
|
||||
cmocka_unit_test(torture_append_without_duplicate),
|
||||
};
|
||||
|
||||
ssh_init();
|
||||
|
||||
Загрузка…
Ссылка в новой задаче
Block a user