Add wl_client_get_credentials() to get unix credentials for client

diff --git a/src/wayland-server.c b/src/wayland-server.c
index 8239291..fd14c7d 100644 (file)
--- a/src/wayland-server.c
+++ b/src/wayland-server.c
@@ -63,6 +63,7 @@ struct wl_client {
        uint32_t mask;
        struct wl_list link;
        struct wl_map objects;
+       struct ucred ucred;
        int error;
 };
 
@@ -286,6 +287,7 @@ WL_EXPORT struct wl_client *
 wl_client_create(struct wl_display *display, int fd)
 {
        struct wl_client *client;
+       socklen_t len;
 
        client = malloc(sizeof *client);
        if (client == NULL)
@@ -296,6 +298,14 @@ wl_client_create(struct wl_display *display, int fd)
        client->source = wl_event_loop_add_fd(display->loop, fd,
                                              WL_EVENT_READABLE,
                                              wl_client_connection_data, client);
+
+       len = sizeof client->ucred;
+       if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED,
+                      &client->ucred, &len) < 0) {
+               free(client);
+               return NULL;
+       }
+
        client->connection =
                wl_connection_create(fd, wl_client_connection_update, client);
        if (client->connection == NULL) {
@@ -319,6 +329,18 @@ wl_client_create(struct wl_display *display, int fd)
 }
 
 WL_EXPORT void
+wl_client_get_credentials(struct wl_client *client,
+                         pid_t *pid, uid_t *uid, gid_t *gid)
+{
+       if (pid)
+               *pid = client->ucred.pid;
+       if (uid)
+               *uid = client->ucred.uid;
+       if (gid)
+               *gid = client->ucred.gid;
+}
+
+WL_EXPORT void
 wl_client_add_resource(struct wl_client *client,
                       struct wl_resource *resource)
 {

diff --git a/src/wayland-server.h b/src/wayland-server.h
index 616df3b..67706b5 100644 (file)
--- a/src/wayland-server.h
+++ b/src/wayland-server.h
@@ -99,6 +99,8 @@ void wl_display_remove_global(struct wl_display *display,
 struct wl_client *wl_client_create(struct wl_display *display, int fd);
 void wl_client_destroy(struct wl_client *client);
 void wl_client_flush(struct wl_client *client);
+void wl_client_get_credentials(struct wl_client *client,
+                              pid_t *pid, uid_t *uid, gid_t *gid);
 
 struct wl_resource *
 wl_client_add_object(struct wl_client *client,