#pragma once

#include <sys/socket.h>

#include <sys/un.h>

#include <netinet/in.h>

#include <pthread.h>

#include "hashmap.h"

#include "prioq.h"

#include "list.h"

#include "util.h"

#include "refcnt.h"

#include "sd-bus.h"

#include "bus-error.h"

#include "bus-match.h"

#include "bus-kernel.h"

struct reply_callback {

sd_bus_message_handler_t callback;

void *userdata;

usec_t timeout;

uint64_t serial;

unsigned prioq_idx;

};

struct filter_callback {

sd_bus_message_handler_t callback;

void *userdata;

unsigned last_iteration;

LIST_FIELDS(struct filter_callback, callbacks);

};

struct node {

char *path;

struct node *parent;

LIST_HEAD(struct node, child);

LIST_FIELDS(struct node, siblings);

LIST_HEAD(struct node_callback, callbacks);

LIST_HEAD(struct node_vtable, vtables);

LIST_HEAD(struct node_enumerator, enumerators);

bool object_manager;

};

struct node_callback {

struct node *node;

bool is_fallback;

sd_bus_message_handler_t callback;

void *userdata;

unsigned last_iteration;

LIST_FIELDS(struct node_callback, callbacks);

};

struct node_enumerator {

struct node *node;

sd_bus_node_enumerator_t callback;

void *userdata;

unsigned last_iteration;

LIST_FIELDS(struct node_enumerator, enumerators);

};

struct node_vtable {

struct node *node;

char *interface;

bool is_fallback;

const sd_bus_vtable *vtable;

void *userdata;

sd_bus_object_find_t find;

unsigned last_iteration;

LIST_FIELDS(struct node_vtable, vtables);

};

struct vtable_member {

const char *path;

const char *interface;

const char *member;

struct node_vtable *parent;

unsigned last_iteration;

const sd_bus_vtable *vtable;

};

enum bus_state {

BUS_UNSET,

BUS_OPENING,

BUS_AUTHENTICATING,

BUS_HELLO,

BUS_RUNNING,

BUS_CLOSING,

BUS_CLOSED

};

static inline bool BUS_IS_OPEN(enum bus_state state) {

return state > BUS_UNSET && state < BUS_CLOSING;

}

enum bus_auth {

_BUS_AUTH_INVALID,

BUS_AUTH_EXTERNAL,

BUS_AUTH_ANONYMOUS

};

struct sd_bus {

/* We use atomic ref counting here since sd_bus_message

RefCount n_ref;

enum bus_state state;

int input_fd, output_fd;

int message_version;

bool is_kernel:1;

bool can_fds:1;

bool bus_client:1;

bool ucred_valid:1;

bool is_server:1;

bool anonymous_auth:1;

bool prefer_readv:1;

bool prefer_writev:1;

bool match_callbacks_modified:1;

bool filter_callbacks_modified:1;

bool nodes_modified:1;

int use_memfd;

void *rbuffer;

size_t rbuffer_size;

sd_bus_message **rqueue;

unsigned rqueue_size;

sd_bus_message **wqueue;

unsigned wqueue_size;

size_t windex;

uint64_t serial;

char *unique_name;

struct bus_match_node match_callbacks;

Prioq *reply_callbacks_prioq;

Hashmap *reply_callbacks;

LIST_HEAD(struct filter_callback, filter_callbacks);

Hashmap *nodes;

Hashmap *vtable_methods;

Hashmap *vtable_properties;

union {

struct sockaddr sa;

struct sockaddr_un un;

struct sockaddr_in in;

struct sockaddr_in6 in6;

} sockaddr;

socklen_t sockaddr_size;

char *kernel;

char *machine;

sd_id128_t server_id;

char *address;

unsigned address_index;

int last_connect_error;

enum bus_auth auth;

size_t auth_rbegin;

struct iovec auth_iovec[3];

unsigned auth_index;

char *auth_buffer;

usec_t auth_timeout;

struct ucred ucred;

char label[NAME_MAX];

int *fds;

unsigned n_fds;

char *exec_path;

char **exec_argv;

uint64_t hello_serial;

unsigned iteration_counter;

void *kdbus_buffer;

/* We do locking around the memfd cache, since we want to

pthread_mutex_t memfd_cache_mutex;

struct memfd_cache memfd_cache[MEMFD_CACHE_MAX];

unsigned n_memfd_cache;

pid_t original_pid;

uint64_t hello_flags;

uint64_t match_cookie;

sd_event_source *input_io_event_source;

sd_event_source *output_io_event_source;

sd_event_source *time_event_source;

sd_event_source *quit_event_source;

sd_event *event;

sd_bus_message *current;

sd_bus **default_bus_ptr;

pid_t tid;

};

#define BUS_DEFAULT_TIMEOUT ((usec_t) (25 * USEC_PER_SEC))

#define BUS_WQUEUE_MAX 1024

#define BUS_RQUEUE_MAX 64*1024

#define BUS_MESSAGE_SIZE_MAX (64*1024*1024)

#define BUS_AUTH_SIZE_MAX (64*1024)

#define BUS_CONTAINER_DEPTH 128

#define BUS_ARRAY_MAX_SIZE 67108864

#define BUS_FDS_MAX 1024

#define BUS_EXEC_ARGV_MAX 256

bool interface_name_is_valid(const char *p);

bool service_name_is_valid(const char *p);

bool sender_name_is_valid(const char *p);

bool member_name_is_valid(const char *p);

bool object_path_is_valid(const char *p);

char *object_path_startswith(const char *a, const char *b);

bool namespace_complex_pattern(const char *pattern, const char *value);

bool path_complex_pattern(const char *pattern, const char *value);

bool namespace_simple_pattern(const char *pattern, const char *value);

bool path_simple_pattern(const char *pattern, const char *value);

int bus_message_type_from_string(const char *s, uint8_t *u);

const char *bus_message_type_to_string(uint8_t u);

#define error_name_is_valid interface_name_is_valid

int bus_ensure_running(sd_bus *bus);

int bus_start_running(sd_bus *bus);

int bus_next_address(sd_bus *bus);

bool bus_pid_changed(sd_bus *bus);

char *bus_address_escape(const char *v);

#define OBJECT_PATH_FOREACH_PREFIX(prefix, path) \

for (char *_slash = ({ strcpy((prefix), (path)); streq((prefix), "/") ? NULL : strrchr((prefix), '/'); }) ; \

_slash && !(_slash[(_slash) == (prefix)] = 0); \

_slash = streq((prefix), "/") ? NULL : strrchr((prefix), '/'))

#define BUS_DONT_DESTROY(bus) \

_cleanup_bus_unref_ _unused_ sd_bus *_dont_destroy_##bus = sd_bus_ref(bus)