ClassTableIterator *ClassTableIteratorNew(const ClassTable *table,
const char *ns,
bool is_hard, bool is_soft)
{
ClassTableIterator *iter = xmalloc(sizeof(*iter));
iter->ns = ns ? xstrdup(ns) : NULL;
iter->iter = MapIteratorInit(table->classes->impl);
iter->is_soft = is_soft;
iter->is_hard = is_hard;
return iter;
}
bool MapContainsSameKeys(const Map *map1, const Map *map2)
{
assert(map1 != NULL);
assert(map2 != NULL);
MapIterator i = MapIteratorInit((Map *)map1);
MapKeyValue *item;
size_t count = 0;
while ((item = MapIteratorNext(&i)))
{
count++;
if (!MapHasKey(map2, item->key))
{
return false;
}
}
return (count == MapSize(map2));
}
size_t MapSize(const Map *map)
{
assert(map != NULL);
if (IsArrayMap(map))
{
return map->arraymap->size;
}
else
{
MapIterator i = MapIteratorInit((Map*)map);
size_t size = 0;
while (MapIteratorNext(&i))
{
size++;
}
return size;
}
}
SetIterator SetIteratorInit(Set *set)
{
return MapIteratorInit(set);
}
bool ScanLastSeenQuality(LastSeenQualityCallback callback, void *ctx)
{
StringMap *lastseen_db = LoadDatabaseToStringMap(dbid_lastseen);
if (!lastseen_db)
{
return false;
}
MapIterator it = MapIteratorInit(lastseen_db->impl);
MapKeyValue *item;
Seq *hostkeys = SeqNew(100, free);
while ((item = MapIteratorNext(&it)) != NULL)
{
char *key = item->key;
/* Only look for "keyhost" entries */
if (key[0] != 'k')
{
continue;
}
SeqAppend(hostkeys, xstrdup(key + 1));
}
for (int i = 0; i < SeqLength(hostkeys); ++i)
{
const char *hostkey = SeqAt(hostkeys, i);
char keyhost_key[CF_BUFSIZE];
snprintf(keyhost_key, CF_BUFSIZE, "k%s", hostkey);
char *address = NULL;
address = (char*)StringMapGet(lastseen_db, keyhost_key);
if (!address)
{
Log(LOG_LEVEL_ERR, "Failed to read address for key '%s'.", hostkey);
continue;
}
char incoming_key[CF_BUFSIZE];
snprintf(incoming_key, CF_BUFSIZE, "qi%s", hostkey);
KeyHostSeen *incoming = NULL;
incoming = (KeyHostSeen*)StringMapGet(lastseen_db, incoming_key);
if (incoming)
{
if (!(*callback)(hostkey, address, true, incoming, ctx))
{
break;
}
}
char outgoing_key[CF_BUFSIZE];
snprintf(outgoing_key, CF_BUFSIZE, "qo%s", hostkey);
KeyHostSeen *outgoing = NULL;
outgoing = (KeyHostSeen*)StringMapGet(lastseen_db, outgoing_key);
if (outgoing)
{
if (!(*callback)(hostkey, address, false, outgoing, ctx))
{
break;
}
}
}
StringMapDestroy(lastseen_db);
SeqDestroy(hostkeys);
return true;
}
SetIterator SetIteratorInit(Set *set)
{
assert(set != NULL);
return MapIteratorInit(set);
}