129 lines
2.5 KiB
C
129 lines
2.5 KiB
C
#include <socks/object.h>
|
|
|
|
struct set {
|
|
queue_t s_list;
|
|
char s_name[OBJECT_NAME_MAX];
|
|
};
|
|
|
|
static kern_status_t set_query_name(object_t *obj, char out[OBJECT_NAME_MAX])
|
|
{
|
|
struct set *set = object_data(obj);
|
|
strncpy(out, set->s_name, OBJECT_NAME_MAX - 1);
|
|
out[OBJECT_NAME_MAX - 1] = 0;
|
|
|
|
return KERN_OK;
|
|
}
|
|
|
|
static kern_status_t set_get_child_at(object_t *obj, size_t at, object_t **out)
|
|
{
|
|
struct set *set = object_data(obj);
|
|
size_t i = 0;
|
|
queue_foreach(object_t, child, &set->s_list, ob_list) {
|
|
if (i == at) {
|
|
*out = object_ref(child);
|
|
return KERN_OK;
|
|
}
|
|
|
|
i++;
|
|
}
|
|
|
|
return KERN_NO_ENTRY;
|
|
}
|
|
|
|
static kern_status_t set_get_child_named(object_t *obj, const char *name, object_t **out)
|
|
{
|
|
struct set *set = object_data(obj);
|
|
char child_name[OBJECT_NAME_MAX];
|
|
|
|
queue_foreach(object_t, child, &set->s_list, ob_list) {
|
|
kern_status_t status = object_query_name(child, child_name);
|
|
if (status != KERN_OK) {
|
|
continue;
|
|
}
|
|
|
|
if (!strcmp(child_name, name)) {
|
|
*out = object_ref(child);
|
|
return KERN_OK;
|
|
}
|
|
}
|
|
|
|
return KERN_NO_ENTRY;
|
|
}
|
|
|
|
static object_type_t set_type = {
|
|
.ob_name = "set",
|
|
.ob_size = sizeof(struct set),
|
|
.ob_ops = {
|
|
.query_name = set_query_name,
|
|
.get_named = set_get_child_named,
|
|
.get_at = set_get_child_at,
|
|
},
|
|
};
|
|
|
|
void init_set_objects(void)
|
|
{
|
|
object_type_register(&set_type);
|
|
}
|
|
|
|
object_t *set_create(const char *name)
|
|
{
|
|
object_t *set_obj = object_create(&set_type);
|
|
if (!set_obj) {
|
|
return NULL;
|
|
}
|
|
|
|
struct set *set = object_data(set_obj);
|
|
set->s_list = QUEUE_INIT;
|
|
strncpy(set->s_name, name, sizeof set->s_name - 1);
|
|
set->s_name[sizeof set->s_name - 1] = 0;
|
|
|
|
return set_obj;
|
|
}
|
|
|
|
kern_status_t set_add_object(object_t *set_obj, object_t *obj)
|
|
{
|
|
if (!object_is_set(set_obj)) {
|
|
return KERN_INVALID_ARGUMENT;
|
|
}
|
|
|
|
struct set *set = object_data(set_obj);
|
|
|
|
char child_name[OBJECT_NAME_MAX];
|
|
char obj_name[OBJECT_NAME_MAX];
|
|
|
|
kern_status_t status = object_query_name(obj, obj_name);
|
|
if (status != KERN_OK) {
|
|
return status;
|
|
}
|
|
|
|
queue_foreach (object_t, child, &set->s_list, ob_list) {
|
|
object_query_name(child, child_name);
|
|
|
|
if (!strcmp(child_name, obj_name)) {
|
|
return KERN_NAME_EXISTS;
|
|
}
|
|
}
|
|
|
|
object_ref(obj);
|
|
queue_push_back(&set->s_list, &obj->ob_list);
|
|
return KERN_OK;
|
|
}
|
|
|
|
kern_status_t set_remove_object(object_t *set_obj, object_t *obj)
|
|
{
|
|
if (!object_is_set(set_obj)) {
|
|
return KERN_INVALID_ARGUMENT;
|
|
}
|
|
|
|
struct set *set = object_data(set_obj);
|
|
queue_delete(&set->s_list, &obj->ob_list);
|
|
object_deref(obj);
|
|
|
|
return KERN_OK;
|
|
}
|
|
|
|
bool object_is_set(object_t *obj)
|
|
{
|
|
return obj->ob_type == &set_type;
|
|
}
|