1
0
mirror of https://github.com/profanity-im/profanity.git synced 2024-12-04 14:46:46 -05:00

Fixed jid handling to allow @ and / in resource

This commit is contained in:
James Booth 2013-01-26 16:13:11 +00:00
parent 527e739ac6
commit d239a1278b
2 changed files with 100 additions and 61 deletions

113
src/jid.c
View File

@ -36,70 +36,59 @@ jid_create(const gchar * const str)
if (str == NULL) { if (str == NULL) {
return NULL; return NULL;
} }
if (strlen(str) == 0) {
return NULL;
}
gchar *trimmed = g_strdup(str); gchar *trimmed = g_strdup(str);
if (strlen(trimmed) == 0) {
return NULL;
}
if (g_str_has_prefix(trimmed, "/") || g_str_has_prefix(trimmed, "@")) { if (g_str_has_prefix(trimmed, "/") || g_str_has_prefix(trimmed, "@")) {
g_free(trimmed); g_free(trimmed);
return NULL; return NULL;
} else if (g_str_has_suffix(trimmed, "/") || g_str_has_suffix(trimmed, "@")) { }
g_free(trimmed);
if (!g_utf8_validate(trimmed, -1, NULL)) {
return NULL; return NULL;
} }
gchar *slashp = g_strrstr(trimmed, "/"); result = malloc(sizeof(struct jid_t));
result->localpart = NULL;
result->domainpart = NULL;
result->resourcepart = NULL;
result->barejid = NULL;
result->fulljid = NULL;
// has resourcepart gchar *atp = g_utf8_strchr(trimmed, -1, '@');
if (slashp != NULL) { gchar *slashp = g_utf8_strchr(trimmed, -1, '/');
result = malloc(sizeof(struct jid_t)); gchar *domain_start = trimmed;
result->str = strdup(trimmed);
result->resourcepart = g_strdup(slashp + 1);
result->barejid = g_strndup(trimmed, strlen(trimmed) - strlen(result->resourcepart) - 1);
result->fulljid = g_strdup(trimmed);
gchar *atp = g_strrstr(result->barejid, "@");
// has domain and local parts if (atp != NULL) {
if (atp != NULL) { result->localpart = g_utf8_substring(trimmed, 0, g_utf8_pointer_to_offset(trimmed, atp));
result->domainpart = g_strndup(atp + 1, strlen(trimmed)); domain_start = atp + 1;
result->localpart = g_strndup(trimmed, strlen(trimmed) - (strlen(result->resourcepart) + strlen(result->domainpart) + 2));
// only domain part
} else {
result->domainpart = strdup(result->barejid);
result->localpart = NULL;
}
g_free(trimmed);
return result;
// no resourcepart
} else {
result = malloc(sizeof(struct jid_t));
result->str = strdup(trimmed);
result->resourcepart = NULL;
result->barejid = g_strdup(trimmed);
result->fulljid = NULL;
gchar *atp = g_strrstr(trimmed, "@");
// has local and domain parts
if (atp != NULL) {
result->domainpart = g_strdup(atp + 1);
result->localpart = g_strndup(trimmed, strlen(trimmed) - strlen(result->domainpart) - 1);
// only domain part
} else {
result->domainpart = g_strdup(trimmed);
result->localpart = NULL;
}
g_free(trimmed);
return result;
} }
if (slashp != NULL) {
result->resourcepart = g_strdup(slashp + 1);
result->domainpart = g_utf8_substring(domain_start, 0, g_utf8_pointer_to_offset(domain_start, slashp));
result->barejid = g_utf8_substring(trimmed, 0, g_utf8_pointer_to_offset(trimmed, slashp));
result->fulljid = g_strdup(trimmed);
} else {
result->domainpart = g_strdup(domain_start);
result->barejid = g_strdup(trimmed);
}
if (result->domainpart == NULL) {
free(trimmed);
return NULL;
}
result->str = g_strdup(trimmed);
free(trimmed);
return result;
} }
Jid * Jid *
@ -142,21 +131,23 @@ jid_is_valid_room_form(Jid *jid)
gboolean gboolean
parse_room_jid(const char * const full_room_jid, char **room, char **nick) parse_room_jid(const char * const full_room_jid, char **room, char **nick)
{ {
gboolean result = FALSE; Jid *jid = jid_create(full_room_jid);
char **tokens = g_strsplit(full_room_jid, "/", 0);
if (tokens == NULL) if (jid == NULL) {
return FALSE; return FALSE;
if (tokens[0] != NULL && tokens[1] != NULL) {
*room = strdup(tokens[0]);
*nick = strdup(tokens[1]);
result = TRUE;
} }
g_strfreev(tokens); if (jid->resourcepart == NULL) {
jid_destroy(jid);
return FALSE;
}
return result; *room = strdup(jid->barejid);
*nick = strdup(jid->resourcepart);
jid_destroy(jid);
return TRUE;
} }
/* /*

View File

@ -119,6 +119,50 @@ void create_room_jid_returns_nick(void)
assert_string_equals("myname", result->resourcepart); assert_string_equals("myname", result->resourcepart);
} }
void create_with_slash_in_resource(void)
{
Jid *result = jid_create("room@conference.domain.org/my/nick");
assert_string_equals("room", result->localpart);
assert_string_equals("conference.domain.org", result->domainpart);
assert_string_equals("my/nick", result->resourcepart);
assert_string_equals("room@conference.domain.org", result->barejid);
assert_string_equals("room@conference.domain.org/my/nick", result->fulljid);
}
void create_with_at_in_resource(void)
{
Jid *result = jid_create("room@conference.domain.org/my@nick");
assert_string_equals("room", result->localpart);
assert_string_equals("conference.domain.org", result->domainpart);
assert_string_equals("my@nick", result->resourcepart);
assert_string_equals("room@conference.domain.org", result->barejid);
assert_string_equals("room@conference.domain.org/my@nick", result->fulljid);
}
void create_with_at_and_slash_in_resource(void)
{
Jid *result = jid_create("room@conference.domain.org/my@nick/something");
assert_string_equals("room", result->localpart);
assert_string_equals("conference.domain.org", result->domainpart);
assert_string_equals("my@nick/something", result->resourcepart);
assert_string_equals("room@conference.domain.org", result->barejid);
assert_string_equals("room@conference.domain.org/my@nick/something", result->fulljid);
}
void create_full_with_trailing_slash(void)
{
Jid *result = jid_create("room@conference.domain.org/nick/");
assert_string_equals("room", result->localpart);
assert_string_equals("conference.domain.org", result->domainpart);
assert_string_equals("nick/", result->resourcepart);
assert_string_equals("room@conference.domain.org", result->barejid);
assert_string_equals("room@conference.domain.org/nick/", result->fulljid);
}
void register_jid_tests(void) void register_jid_tests(void)
{ {
TEST_MODULE("jid tests"); TEST_MODULE("jid tests");
@ -141,4 +185,8 @@ void register_jid_tests(void)
TEST(create_jid_from_bare_returns_domainpart); TEST(create_jid_from_bare_returns_domainpart);
TEST(create_room_jid_returns_room); TEST(create_room_jid_returns_room);
TEST(create_room_jid_returns_nick); TEST(create_room_jid_returns_nick);
TEST(create_with_slash_in_resource);
TEST(create_with_at_in_resource);
TEST(create_with_at_and_slash_in_resource);
TEST(create_full_with_trailing_slash);
} }