1/*
2 * Copyright © 2011 Kristian Høgsberg
3 * Copyright © 2011 Benjamin Franzke
4 *
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
11 *
12 * The above copyright notice and this permission notice (including the next
13 * paragraph) shall be included in all copies or substantial portions of the
14 * Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19 * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
20 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
21 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
23 * DEALINGS IN THE SOFTWARE.
24 *
25 * Authors:
26 *    Kristian Høgsberg <krh@bitplanet.net>
27 *    Benjamin Franzke <benjaminfranzke@googlemail.com>
28 */
29
30#include <stdio.h>
31#include <stdlib.h>
32#include <string.h>
33#include <stddef.h>
34#include <unistd.h>
35
36#include <wayland-server.h>
37#include "wayland-drm.h"
38#include "wayland-drm-server-protocol.h"
39
40#define MIN(x,y) (((x)<(y))?(x):(y))
41
42static void
43destroy_buffer(struct wl_resource *resource)
44{
45	struct wl_drm_buffer *buffer = wl_resource_get_user_data(resource);
46	struct wl_drm *drm = buffer->drm;
47
48	drm->callbacks.release_buffer(drm->user_data, buffer);
49	free(buffer);
50}
51
52static void
53buffer_destroy(struct wl_client *client, struct wl_resource *resource)
54{
55	wl_resource_destroy(resource);
56}
57
58static void
59create_buffer(struct wl_client *client, struct wl_resource *resource,
60              uint32_t id, uint32_t name, int fd,
61              int32_t width, int32_t height,
62              uint32_t format,
63              int32_t offset0, int32_t stride0,
64              int32_t offset1, int32_t stride1,
65              int32_t offset2, int32_t stride2)
66{
67	struct wl_drm *drm = wl_resource_get_user_data(resource);
68	struct wl_drm_buffer *buffer;
69
70	buffer = calloc(1, sizeof *buffer);
71	if (buffer == NULL) {
72		wl_resource_post_no_memory(resource);
73		return;
74	}
75
76	buffer->drm = drm;
77	buffer->width = width;
78	buffer->height = height;
79	buffer->format = format;
80	buffer->offset[0] = offset0;
81	buffer->stride[0] = stride0;
82	buffer->offset[1] = offset1;
83	buffer->stride[1] = stride1;
84	buffer->offset[2] = offset2;
85	buffer->stride[2] = stride2;
86
87        drm->callbacks.reference_buffer(drm->user_data, name, fd, buffer);
88	if (buffer->driver_buffer == NULL) {
89		wl_resource_post_error(resource,
90				       WL_DRM_ERROR_INVALID_NAME,
91				       "invalid name");
92		return;
93	}
94
95	buffer->resource =
96		wl_resource_create(client, &wl_buffer_interface, 1, id);
97	if (!buffer->resource) {
98		wl_resource_post_no_memory(resource);
99		free(buffer);
100		return;
101	}
102
103	wl_resource_set_implementation(buffer->resource,
104				       (void (**)(void)) &drm->buffer_interface,
105				       buffer, destroy_buffer);
106}
107
108static void
109drm_create_buffer(struct wl_client *client, struct wl_resource *resource,
110		  uint32_t id, uint32_t name, int32_t width, int32_t height,
111		  uint32_t stride, uint32_t format)
112{
113        switch (format) {
114        case WL_DRM_FORMAT_ABGR2101010:
115        case WL_DRM_FORMAT_XBGR2101010:
116        case WL_DRM_FORMAT_ARGB2101010:
117        case WL_DRM_FORMAT_XRGB2101010:
118        case WL_DRM_FORMAT_ARGB8888:
119        case WL_DRM_FORMAT_XRGB8888:
120        case WL_DRM_FORMAT_YUYV:
121        case WL_DRM_FORMAT_RGB565:
122                break;
123        default:
124                wl_resource_post_error(resource,
125                                       WL_DRM_ERROR_INVALID_FORMAT,
126                                       "invalid format");
127           return;
128        }
129
130        create_buffer(client, resource, id,
131                      name, -1, width, height, format, 0, stride, 0, 0, 0, 0);
132}
133
134static void
135drm_create_planar_buffer(struct wl_client *client,
136                         struct wl_resource *resource,
137                         uint32_t id, uint32_t name,
138                         int32_t width, int32_t height, uint32_t format,
139                         int32_t offset0, int32_t stride0,
140                         int32_t offset1, int32_t stride1,
141                         int32_t offset2, int32_t stride2)
142{
143        switch (format) {
144	case WL_DRM_FORMAT_YUV410:
145	case WL_DRM_FORMAT_YUV411:
146	case WL_DRM_FORMAT_YUV420:
147	case WL_DRM_FORMAT_YUV422:
148	case WL_DRM_FORMAT_YUV444:
149	case WL_DRM_FORMAT_NV12:
150        case WL_DRM_FORMAT_NV16:
151                break;
152        default:
153                wl_resource_post_error(resource,
154                                       WL_DRM_ERROR_INVALID_FORMAT,
155                                       "invalid format");
156           return;
157        }
158
159        create_buffer(client, resource, id, name, -1, width, height, format,
160                      offset0, stride0, offset1, stride1, offset2, stride2);
161}
162
163static void
164drm_create_prime_buffer(struct wl_client *client,
165                        struct wl_resource *resource,
166                        uint32_t id, int fd,
167                        int32_t width, int32_t height, uint32_t format,
168                        int32_t offset0, int32_t stride0,
169                        int32_t offset1, int32_t stride1,
170                        int32_t offset2, int32_t stride2)
171{
172        create_buffer(client, resource, id, 0, fd, width, height, format,
173                      offset0, stride0, offset1, stride1, offset2, stride2);
174        close(fd);
175}
176
177static void
178drm_authenticate(struct wl_client *client,
179		 struct wl_resource *resource, uint32_t id)
180{
181	struct wl_drm *drm = wl_resource_get_user_data(resource);
182
183	if (drm->callbacks.authenticate(drm->user_data, id) < 0)
184		wl_resource_post_error(resource,
185				       WL_DRM_ERROR_AUTHENTICATE_FAIL,
186				       "authenicate failed");
187	else
188		wl_resource_post_event(resource, WL_DRM_AUTHENTICATED);
189}
190
191static const struct wl_drm_interface drm_interface = {
192	drm_authenticate,
193	drm_create_buffer,
194        drm_create_planar_buffer,
195        drm_create_prime_buffer
196};
197
198static void
199bind_drm(struct wl_client *client, void *data, uint32_t version, uint32_t id)
200{
201	struct wl_drm *drm = data;
202	struct wl_resource *resource;
203        uint32_t capabilities;
204
205	resource = wl_resource_create(client, &wl_drm_interface,
206				      MIN(version, 2), id);
207	if (!resource) {
208		wl_client_post_no_memory(client);
209		return;
210	}
211
212	wl_resource_set_implementation(resource, &drm_interface, data, NULL);
213
214	wl_resource_post_event(resource, WL_DRM_DEVICE, drm->device_name);
215
216	if (drm->callbacks.is_format_supported(drm->user_data,
217					       WL_DRM_FORMAT_ARGB2101010)) {
218		wl_resource_post_event(resource, WL_DRM_FORMAT,
219				       WL_DRM_FORMAT_ARGB2101010);
220	}
221
222	if (drm->callbacks.is_format_supported(drm->user_data,
223					       WL_DRM_FORMAT_XRGB2101010)) {
224		wl_resource_post_event(resource, WL_DRM_FORMAT,
225				       WL_DRM_FORMAT_XRGB2101010);
226	}
227
228	if (drm->callbacks.is_format_supported(drm->user_data,
229					       WL_DRM_FORMAT_ABGR2101010)) {
230		wl_resource_post_event(resource, WL_DRM_FORMAT,
231				       WL_DRM_FORMAT_ABGR2101010);
232	}
233
234	if (drm->callbacks.is_format_supported(drm->user_data,
235					       WL_DRM_FORMAT_XBGR2101010)) {
236		wl_resource_post_event(resource, WL_DRM_FORMAT,
237				       WL_DRM_FORMAT_XBGR2101010);
238	}
239
240	wl_resource_post_event(resource, WL_DRM_FORMAT,
241			       WL_DRM_FORMAT_ARGB8888);
242	wl_resource_post_event(resource, WL_DRM_FORMAT,
243			       WL_DRM_FORMAT_XRGB8888);
244        wl_resource_post_event(resource, WL_DRM_FORMAT,
245                               WL_DRM_FORMAT_RGB565);
246        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_YUV410);
247        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_YUV411);
248        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_YUV420);
249        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_YUV422);
250        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_YUV444);
251        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_NV12);
252        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_NV16);
253        wl_resource_post_event(resource, WL_DRM_FORMAT, WL_DRM_FORMAT_YUYV);
254
255        capabilities = 0;
256        if (drm->flags & WAYLAND_DRM_PRIME)
257           capabilities |= WL_DRM_CAPABILITY_PRIME;
258
259        if (version >= 2)
260           wl_resource_post_event(resource, WL_DRM_CAPABILITIES, capabilities);
261}
262
263struct wl_drm *
264wayland_drm_init(struct wl_display *display, char *device_name,
265                 const struct wayland_drm_callbacks *callbacks, void *user_data,
266                 uint32_t flags)
267{
268	struct wl_drm *drm;
269
270	drm = malloc(sizeof *drm);
271	if (!drm)
272		return NULL;
273
274	drm->display = display;
275	drm->device_name = strdup(device_name);
276	drm->callbacks = *callbacks;
277	drm->user_data = user_data;
278        drm->flags = flags;
279
280        drm->buffer_interface.destroy = buffer_destroy;
281
282	drm->wl_drm_global =
283		wl_global_create(display, &wl_drm_interface, 2,
284				 drm, bind_drm);
285
286	return drm;
287}
288
289void
290wayland_drm_uninit(struct wl_drm *drm)
291{
292	free(drm->device_name);
293
294	wl_global_destroy(drm->wl_drm_global);
295
296	free(drm);
297}
298