1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
#include "mock-server.h"
struct wl_display* display = NULL;
void* alloc_zeroed(size_t size)
{
void* data = malloc(size);
memset(data, 0, size);
return data;
}
static const char* get_display_name()
{
const char* result = getenv("WAYLAND_DISPLAY");
if (!result)
{
FATAL("WAYLAND_DISPLAY not set");
}
return result;
}
typedef struct
{
const struct wl_message* message;
RequestOverrideFunction function;
struct wl_list link;
} RequestOverride;
struct wl_list request_overrides;
void install_request_override(const struct wl_interface* interface, const char* name, RequestOverrideFunction function)
{
for (int i = 0; i < interface->method_count; i++)
{
if (strcmp(name, interface->methods[i].name) == 0)
{
RequestOverride* override = ALLOC_STRUCT(RequestOverride);
override->message = &interface->methods[i];
override->function = function;
wl_list_insert(&request_overrides, &override->link);
return;
}
}
FATAL_FMT("Interface %s does not have a request named %s", interface->name, name);
}
static int default_dispatcher(const void* data, void* resource, uint32_t opcode, const struct wl_message* message, union wl_argument* args)
{
// First, check if there is an override
RequestOverride* override;
wl_list_for_each(override, &request_overrides, link)
{
if (override->message == message)
{
override->function(resource, message, args);
return 0;
}
}
// If there are any new-id type arguments, resources need to be created for them
// See https://wayland.freedesktop.org/docs/html/apb.html#Client-structwl__message
int arg = 0;
for (const char* c = message->signature; *c; c++)
{
if (*c == 'n' && args[arg].n != 0)
{
struct wl_resource* new_resource = wl_resource_create(
wl_resource_get_client(resource),
message->types[arg],
wl_resource_get_version(resource),
args[arg].n);
wl_resource_set_dispatcher(new_resource, default_dispatcher, NULL, NULL, NULL);
}
if (*c >= 'a' && *c <= 'z')
arg++;
}
if (strcmp(message->name, "destroy") == 0)
{
wl_resource_destroy(resource);
}
return 0;
}
void use_default_impl(struct wl_resource* resource)
{
wl_resource_set_dispatcher(resource, default_dispatcher, NULL, NULL, NULL);
}
static void default_global_bind(struct wl_client* client, void* data, uint32_t version, uint32_t id)
{
struct wl_interface* interface = data;
struct wl_resource* resource = wl_resource_create(client, interface, version, id);
use_default_impl(resource);
};
void default_global_create(struct wl_display* display, const struct wl_interface* interface, int version)
{
wl_global_create(display, interface, version, (void*)interface, default_global_bind);
}
char type_code_at_index(const struct wl_message* message, int index)
{
int i = 0;
for (const char* c = message->signature; *c; c++)
{
if (*c >= 'a' && *c <= 'z')
{
if (i == index)
return *c;
else
i++;
}
}
FATAL_FMT(".%s does not have an argument %d", message->name, index);
}
static void client_disconnect(struct wl_listener *listener, void *data)
{
wl_display_terminate(display);
}
static struct wl_listener client_disconnect_listener = {
.notify = client_disconnect,
};
static void client_connect(struct wl_listener *listener, void *data)
{
struct wl_client* client = (struct wl_client*)data;
wl_client_add_destroy_listener(client, &client_disconnect_listener);
}
static struct wl_listener client_connect_listener = {
.notify = client_connect,
};
int main(int argc, const char** argv)
{
wl_list_init(&request_overrides);
display = wl_display_create();
if (wl_display_add_socket(display, get_display_name()) != 0)
{
FATAL_FMT("server failed to connect to Wayland display %s", get_display_name());
}
wl_display_add_client_created_listener(display, &client_connect_listener);
init();
wl_display_run(display);
wl_display_destroy(display);
return 0;
}
|
