forked from Telodendria/Telodendria
Compare commits
3 commits
6f045083a2
...
55652eaa15
Author | SHA1 | Date | |
---|---|---|---|
|
55652eaa15 | ||
|
3d700de17d | ||
|
c25db688ba |
4 changed files with 164 additions and 68 deletions
113
src/Filter.c
113
src/Filter.c
|
@ -24,12 +24,121 @@
|
||||||
*/
|
*/
|
||||||
#include <Filter.h>
|
#include <Filter.h>
|
||||||
|
|
||||||
|
#include <Cytoplasm/Memory.h>
|
||||||
|
#include <Cytoplasm/Json.h>
|
||||||
|
#include <Cytoplasm/Str.h>
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
#include <Schema/Filter.h>
|
#include <Schema/Filter.h>
|
||||||
|
|
||||||
|
|
||||||
|
bool
|
||||||
|
IsRoomFiltered(Filter *filter, char *roomID)
|
||||||
|
{
|
||||||
|
size_t i, count;
|
||||||
|
if (!filter || !roomID)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
count = ArraySize(filter->room.not_rooms);
|
||||||
|
for (i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
char *notRoom = ArrayGet(filter->room.not_rooms, i);
|
||||||
|
if (StrEquals(roomID, notRoom))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
count = ArraySize(filter->room.rooms);
|
||||||
|
if (count)
|
||||||
|
{
|
||||||
|
for (i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
char *room = ArrayGet(filter->room.rooms, i);
|
||||||
|
if (StrEquals(roomID, room))
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
HashMap *
|
HashMap *
|
||||||
FilterApply(Filter * filter, HashMap * event)
|
FilterApply(Filter * filter, HashMap * event)
|
||||||
{
|
{
|
||||||
(void) filter;
|
HashMap *copy;
|
||||||
(void) event;
|
char *sender, *room;
|
||||||
|
if (!event)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
if (!filter)
|
||||||
|
{
|
||||||
|
/* Do NOT filter anything out */
|
||||||
|
return JsonDuplicate(event);
|
||||||
|
}
|
||||||
|
|
||||||
|
sender = JsonValueAsString(HashMapGet(event, "sender"));
|
||||||
|
room = JsonValueAsString(HashMapGet(event, "room_id"));
|
||||||
|
|
||||||
|
if (IsRoomFiltered(filter, room))
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
copy = JsonDuplicate(event);
|
||||||
|
|
||||||
|
(void) sender;
|
||||||
|
return copy;
|
||||||
|
}
|
||||||
|
Filter *
|
||||||
|
FilterDecode(Db *db, char *user, char *filterStr)
|
||||||
|
{
|
||||||
|
Filter *ret;
|
||||||
|
DbRef *filterRef;
|
||||||
|
HashMap *filterObj;
|
||||||
|
if (!db || !user || !filterStr)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (*filterStr != '{')
|
||||||
|
{
|
||||||
|
char *err; /* TODO: use error status somewhere... */
|
||||||
|
|
||||||
|
filterRef = DbLock(db, 3, "filters", user, filterStr);
|
||||||
|
filterObj = DbJson(filterRef);
|
||||||
|
ret = Malloc(sizeof(*ret));
|
||||||
|
memset(ret, 0, sizeof(*ret));
|
||||||
|
|
||||||
|
if (!FilterFromJson(filterObj, ret, &err))
|
||||||
|
{
|
||||||
|
DbUnlock(db, filterRef);
|
||||||
|
Free(ret);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
DbUnlock(db, filterRef);
|
||||||
|
(void) err;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* TODO: Decode JSON here... */
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
FilterDestroy(Filter *filter)
|
||||||
|
{
|
||||||
|
if (!filter)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
FilterFree(filter);
|
||||||
|
Free(filter);
|
||||||
|
}
|
||||||
|
|
|
@ -34,6 +34,7 @@
|
||||||
#include <Cytoplasm/Util.h>
|
#include <Cytoplasm/Util.h>
|
||||||
#include <Cytoplasm/Str.h>
|
#include <Cytoplasm/Str.h>
|
||||||
|
|
||||||
|
#include <Filter.h>
|
||||||
#include <State.h>
|
#include <State.h>
|
||||||
#include <User.h>
|
#include <User.h>
|
||||||
#include <Room.h>
|
#include <Room.h>
|
||||||
|
@ -59,42 +60,6 @@ StripStateEventSync(HashMap *pdu)
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Verify whenever a room is filtered out by a filter given in. */
|
|
||||||
static bool
|
|
||||||
IsRoomFiltered(Filter *filter, char *roomID)
|
|
||||||
{
|
|
||||||
size_t i, count;
|
|
||||||
if (!filter || !roomID)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
count = ArraySize(filter->room.not_rooms);
|
|
||||||
for (i = 0; i < count; i++)
|
|
||||||
{
|
|
||||||
char *notRoom = ArrayGet(filter->room.not_rooms, i);
|
|
||||||
if (StrEquals(roomID, notRoom))
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
count = ArraySize(filter->room.rooms);
|
|
||||||
if (count)
|
|
||||||
{
|
|
||||||
for (i = 0; i < count; i++)
|
|
||||||
{
|
|
||||||
char *room = ArrayGet(filter->room.rooms, i);
|
|
||||||
if (StrEquals(roomID, room))
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
ROUTE_IMPL(RouteSync, path, argp)
|
ROUTE_IMPL(RouteSync, path, argp)
|
||||||
{
|
{
|
||||||
RouteArgs *args = argp;
|
RouteArgs *args = argp;
|
||||||
|
@ -103,7 +68,7 @@ ROUTE_IMPL(RouteSync, path, argp)
|
||||||
HashMap *params = NULL;
|
HashMap *params = NULL;
|
||||||
HashMap *response = NULL;
|
HashMap *response = NULL;
|
||||||
SyncResponse sync = { 0 };
|
SyncResponse sync = { 0 };
|
||||||
Filter filterData = { 0 };
|
Filter *filterData = NULL;
|
||||||
|
|
||||||
Array *invites;
|
Array *invites;
|
||||||
Array *joins;
|
Array *joins;
|
||||||
|
@ -153,25 +118,15 @@ ROUTE_IMPL(RouteSync, path, argp)
|
||||||
filter = HashMapGet(params, "filter");
|
filter = HashMapGet(params, "filter");
|
||||||
timeoutDuration = timeout ? atoi(timeout) : 0;
|
timeoutDuration = timeout ? atoi(timeout) : 0;
|
||||||
|
|
||||||
if (filter && *filter != '{')
|
if (filter)
|
||||||
{
|
{
|
||||||
/* The filter must be located in the database */
|
char *userName = UserGetName(user);
|
||||||
DbRef *filterRef = DbLock(db, 3,
|
if (!(filterData = FilterDecode(db, userName, filter)))
|
||||||
"filters", UserGetName(user), filter
|
|
||||||
);
|
|
||||||
HashMap *filterJson = DbJson(filterRef);
|
|
||||||
|
|
||||||
if (!FilterFromJson(filterJson, &filterData, &err))
|
|
||||||
{
|
{
|
||||||
|
err = "Couldn't decode filter given in";
|
||||||
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
|
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
|
||||||
response = MatrixErrorCreate(M_BAD_JSON, err);
|
response = MatrixErrorCreate(M_BAD_JSON, err);
|
||||||
|
|
||||||
DbUnlock(db, filterRef);
|
|
||||||
goto finish;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* We now grabbed a filter JSON */
|
|
||||||
DbUnlock(db, filterRef);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!prevBatch)
|
if (!prevBatch)
|
||||||
|
@ -188,8 +143,9 @@ ROUTE_IMPL(RouteSync, path, argp)
|
||||||
UserUnlock(user);
|
UserUnlock(user);
|
||||||
|
|
||||||
/* TODO: Using pthreads' condition variables with a timeout would be
|
/* TODO: Using pthreads' condition variables with a timeout would be
|
||||||
* the best way to proceed (as in Parsee's ParseeAwaitStanza function)
|
* the best way to proceed (as in Parsee's ParseeAwaitStanza function).
|
||||||
*/
|
* Also, how would Cytoplasm proceed if the all of HTTP threads are used
|
||||||
|
* for syncing? That sounds like a really bad bottleneck... */
|
||||||
while (passed < timeoutDuration)
|
while (passed < timeoutDuration)
|
||||||
{
|
{
|
||||||
if (UserGetNotification(name))
|
if (UserGetNotification(name))
|
||||||
|
@ -217,7 +173,7 @@ ROUTE_IMPL(RouteSync, path, argp)
|
||||||
InvitedRooms invited = { 0 };
|
InvitedRooms invited = { 0 };
|
||||||
HashMap *invitedObject;
|
HashMap *invitedObject;
|
||||||
|
|
||||||
if (IsRoomFiltered(&filterData, roomId))
|
if (IsRoomFiltered(filterData, roomId))
|
||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
@ -248,7 +204,7 @@ ROUTE_IMPL(RouteSync, path, argp)
|
||||||
size_t j;
|
size_t j;
|
||||||
Room *r;
|
Room *r;
|
||||||
|
|
||||||
if (IsRoomFiltered(&filterData, roomId))
|
if (IsRoomFiltered(filterData, roomId))
|
||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
@ -262,20 +218,26 @@ ROUTE_IMPL(RouteSync, path, argp)
|
||||||
for (j = 0; j < ArraySize(el); j++)
|
for (j = 0; j < ArraySize(el); j++)
|
||||||
{
|
{
|
||||||
char *event = ArrayGet(el, j);
|
char *event = ArrayGet(el, j);
|
||||||
HashMap *e = RoomEventFetch(r, event);
|
HashMap *eventObj = RoomEventFetch(r, event);
|
||||||
ClientEventWithoutRoomID rc = ClientfyEventSync(e);
|
HashMap *filteredObj = FilterApply(filterData, eventObj);
|
||||||
|
|
||||||
|
if (filteredObj)
|
||||||
|
{
|
||||||
|
ClientEventWithoutRoomID rc = ClientfyEventSync(filteredObj);
|
||||||
ClientEventWithoutRoomID *c = Malloc(sizeof(*c));
|
ClientEventWithoutRoomID *c = Malloc(sizeof(*c));
|
||||||
memcpy(c, &rc, sizeof(*c));
|
memcpy(c, &rc, sizeof(*c));
|
||||||
|
|
||||||
JsonFree(e);
|
if (!firstEvent)
|
||||||
|
|
||||||
if (j == 0)
|
|
||||||
{
|
{
|
||||||
firstEvent = c->event_id;
|
firstEvent = c->event_id;
|
||||||
}
|
}
|
||||||
|
|
||||||
ArrayAdd(joined.timeline.events, c);
|
ArrayAdd(joined.timeline.events, c);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
JsonFree(eventObj);
|
||||||
|
JsonFree(filteredObj);
|
||||||
|
}
|
||||||
joined.timeline.prev_batch = UserNewMessageToken(
|
joined.timeline.prev_batch = UserNewMessageToken(
|
||||||
user, roomId, firstEvent
|
user, roomId, firstEvent
|
||||||
);
|
);
|
||||||
|
@ -316,7 +278,7 @@ ROUTE_IMPL(RouteSync, path, argp)
|
||||||
response = SyncResponseToJson(&sync);
|
response = SyncResponseToJson(&sync);
|
||||||
SyncResponseFree(&sync);
|
SyncResponseFree(&sync);
|
||||||
finish:
|
finish:
|
||||||
FilterFree(&filterData);
|
FilterDestroy(filterData);
|
||||||
UserUnlock(user);
|
UserUnlock(user);
|
||||||
(void) path;
|
(void) path;
|
||||||
return response;
|
return response;
|
||||||
|
|
|
@ -379,7 +379,13 @@ StateCurrent(Room *room)
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
bool StateIterate(HashMap *state, char **type, char **key, void **event)
|
|
||||||
|
/* TODO: USING A ',' DELIMITED HASHMAP IS A _BAD_ IDEA!
|
||||||
|
* I'll need to change these functions, and some things in the room
|
||||||
|
* implementation, thankfully _that_ was abstracted away, so I don't have
|
||||||
|
* to think about it too much... */
|
||||||
|
bool
|
||||||
|
StateIterate(HashMap *state, char **type, char **key, void **event)
|
||||||
{
|
{
|
||||||
char *tuple;
|
char *tuple;
|
||||||
bool ret;
|
bool ret;
|
||||||
|
@ -400,6 +406,7 @@ bool StateIterate(HashMap *state, char **type, char **key, void **event)
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
char *
|
char *
|
||||||
StateGet(HashMap *state, char *type, char *key)
|
StateGet(HashMap *state, char *type, char *key)
|
||||||
{
|
{
|
||||||
|
|
|
@ -28,6 +28,7 @@
|
||||||
#include <Schema/Filter.h>
|
#include <Schema/Filter.h>
|
||||||
|
|
||||||
#include <Cytoplasm/HashMap.h>
|
#include <Cytoplasm/HashMap.h>
|
||||||
|
#include <Cytoplasm/Db.h>
|
||||||
|
|
||||||
/***
|
/***
|
||||||
* @Nm Filter
|
* @Nm Filter
|
||||||
|
@ -47,4 +48,21 @@
|
||||||
extern HashMap *
|
extern HashMap *
|
||||||
FilterApply(Filter *, HashMap *);
|
FilterApply(Filter *, HashMap *);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Verifies if a room would be filtered out by a filter given in.
|
||||||
|
*/
|
||||||
|
extern bool IsRoomFiltered(Filter *, char *);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Decodes a filter from a JSON stream to decode or an ID that was
|
||||||
|
* registered using the filter API.
|
||||||
|
*/
|
||||||
|
extern Filter * FilterDecode(Db *, char *, char *);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Frees all memory created by
|
||||||
|
* .Fn FilterDecode .
|
||||||
|
*/
|
||||||
|
extern void FilterDestroy(Filter *);
|
||||||
|
|
||||||
#endif /* TELODENDRIA_FILTER_H */
|
#endif /* TELODENDRIA_FILTER_H */
|
||||||
|
|
Loading…
Reference in a new issue