gio/gdocumentportal.c - third_party/glib - Git at Google

 /* GIO - GLib Input, Output and Streaming Library
  *
  * Copyright 2016 Endless Mobile, Inc.
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General
  * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
  */

 #include "config.h"

 #include <sys/stat.h>
 #include <fcntl.h>
 #include <errno.h>
 #include <string.h>

 #include "gdocumentportal.h"
 #include "xdp-dbus.h"
 #include "gstdio.h"

 #ifdef G_OS_UNIX
 #include "gunixfdlist.h"
 #endif

 #ifndef O_PATH
 #define O_PATH 0
 #endif
 #ifndef O_CLOEXEC
 #define O_CLOEXEC 0
 #else
 #define HAVE_O_CLOEXEC 1
 #endif

 static GXdpDocuments *documents;
 static char *documents_mountpoint;

 static gboolean
 init_document_portal (void)
 {
   static gsize documents_inited = 0;

   if (g_once_init_enter (&documents_inited))
     {
       GError *error = NULL;
       GDBusConnection *connection = g_bus_get_sync (G_BUS_TYPE_SESSION, NULL, &error);

       if (connection != NULL)
         {
           documents = gxdp_documents_proxy_new_sync (connection, 0,
                                                      "org.freedesktop.portal.Documents",
                                                      "/org/freedesktop/portal/documents",
                                                      NULL, &error);
           if (documents != NULL)
             {
               gxdp_documents_call_get_mount_point_sync (documents,
                                                         &documents_mountpoint,
                                                         NULL, &error);

               if (error != NULL)
                 {
                   g_warning ("Cannot get document portal mount point: %s", error->message);
                   g_error_free (error);
                 }
             }
           else
             {
               g_warning ("Cannot create document portal proxy: %s", error->message);
               g_error_free (error);
             }

           g_object_unref (connection);
         }
       else
         {
           g_warning ("Cannot connect to session bus when initializing document portal: %s",
                      error->message);
           g_error_free (error);
         }

       g_once_init_leave (&documents_inited, 1);
     }

   return (documents != NULL && documents_mountpoint != NULL);
 }

 char *
 g_document_portal_add_document (GFile   *file,
                                 GError **error)
 {
   char *doc_path, *basename;
   char *doc_id = NULL;
   char *doc_uri = NULL;
   char *path = NULL;
   GUnixFDList *fd_list = NULL;
   int fd, fd_in, errsv;
   gboolean ret;

   if (!init_document_portal ())
     {
       g_set_error (error, G_IO_ERROR, G_IO_ERROR_NOT_INITIALIZED,
                    "Document portal is not available");
       goto out;
     }

   path = g_file_get_path (file);
   fd = g_open (path, O_PATH | O_CLOEXEC);
   errsv = errno;

   if (fd == -1)
     {
       g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errsv),
                    "Failed to open %s", path);
       goto out;
     }

 #ifndef HAVE_O_CLOEXEC
   fcntl (fd, F_SETFD, FD_CLOEXEC);
 #endif

   fd_list = g_unix_fd_list_new ();
   fd_in = g_unix_fd_list_append (fd_list, fd, error);
   g_close (fd, NULL);

   if (fd_in == -1)
     goto out;

   ret = gxdp_documents_call_add_sync (documents,
                                       g_variant_new_handle (fd_in),
                                       TRUE,
                                       TRUE,
                                       fd_list,
                                       &doc_id,
                                       NULL,
                                       NULL,
                                       error);

   if (!ret)
     goto out;

   basename = g_path_get_basename (path);
   doc_path = g_build_filename (documents_mountpoint, doc_id, basename, NULL);
   g_free (basename);

   doc_uri = g_filename_to_uri (doc_path, NULL, NULL);
   g_free (doc_path);

  out:
   if (fd_list)
     g_object_unref (fd_list);
   g_free (path);
   g_free (doc_id);

   return doc_uri;
 }

 /* Flags accepted by org.freedesktop.portal.Documents.AddFull */
 enum {
   XDP_ADD_FLAGS_REUSE_EXISTING             =  (1 << 0),
   XDP_ADD_FLAGS_PERSISTENT                 =  (1 << 1),
   XDP_ADD_FLAGS_AS_NEEDED_BY_APP           =  (1 << 2),
   XDP_ADD_FLAGS_FLAGS_ALL                  = ((1 << 3) - 1)
 };

 GList *
 g_document_portal_add_documents (GList       *uris,
                                  const char  *app_id,
                                  GError     **error)
 {
   int length;
   GList *ruris = NULL;
   gboolean *as_is;
   GVariantBuilder builder;
   GUnixFDList *fd_list = NULL;
   GList *l;
   gsize i, j;
   const char *permissions[] = { "read", "write", NULL };
   char **doc_ids = NULL;
   GVariant *extra_out = NULL;

   if (!init_document_portal ())
     {
       g_set_error (error, G_IO_ERROR, G_IO_ERROR_NOT_INITIALIZED,
                    "Document portal is not available");
       return NULL;
     }

   length = g_list_length (uris);
   as_is = g_new0 (gboolean, length);

   g_variant_builder_init (&builder, G_VARIANT_TYPE ("ah"));

   fd_list = g_unix_fd_list_new ();
   for (l = uris, i = 0; l; l = l->next, i++)
     {
       const char *uri = l->data;
       int idx = -1;
       char *path = NULL;

       path = g_filename_from_uri (uri, NULL, NULL);
       if (path != NULL)
         {
           int fd;

           fd = g_open (path, O_CLOEXEC | O_PATH);
           if (fd >= 0)
             {
 #ifndef HAVE_O_CLOEXEC
               fcntl (fd, F_SETFD, FD_CLOEXEC);
 #endif
               idx = g_unix_fd_list_append (fd_list, fd, NULL);
               close (fd);
             }
         }

       g_free (path);

       if (idx != -1)
         g_variant_builder_add (&builder, "h", idx);
       else
         as_is[i] = TRUE;
     }

   if (g_unix_fd_list_get_length (fd_list) > 0)
     {
       if (!gxdp_documents_call_add_full_sync (documents,
                                               g_variant_builder_end (&builder),
                                               XDP_ADD_FLAGS_AS_NEEDED_BY_APP,
                                               app_id,
                                               permissions,
                                               fd_list,
                                               &doc_ids,
                                               &extra_out,
                                               NULL,
                                               NULL,
                                               error))
         goto out;

       for (l = uris, i = 0, j = 0; l; l = l->next, i++)
         {
           const char *uri = l->data;
           char *ruri;

           if (as_is[i]) /* use as-is, not a file uri */
             {
               ruri = g_strdup (uri);
             }
           else if (strcmp (doc_ids[j], "") == 0) /* not rewritten */
             {
               ruri = g_strdup (uri);
               j++;
             }
           else
             {
               char *basename = g_path_get_basename (uri + strlen ("file:"));
               char *doc_path = g_build_filename (documents_mountpoint, doc_ids[j], basename, NULL);
               ruri = g_strconcat ("file:", doc_path, NULL);
               g_free (basename);
               g_free (doc_path);
               j++;
             }

           ruris = g_list_prepend (ruris, ruri);
         }

       ruris = g_list_reverse (ruris);
     }
   else
     {
       ruris = g_list_copy_deep (uris, (GCopyFunc)g_strdup, NULL);
     }

 out:
   g_clear_object (&fd_list);
   g_clear_pointer (&extra_out, g_variant_unref);
   g_clear_pointer (&doc_ids, g_strfreev);
   g_free (as_is);

   return ruris;
 }
	/* GIO - GLib Input, Output and Streaming Library
	*
	* Copyright 2016 Endless Mobile, Inc.
	*
	* This library is free software; you can redistribute it and/or
	* modify it under the terms of the GNU Lesser General Public
	* License as published by the Free Software Foundation; either
	* version 2.1 of the License, or (at your option) any later version.
	*
	* This library is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General
	* Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
	*/

	#include "config.h"

	#include <sys/stat.h>
	#include <fcntl.h>
	#include <errno.h>
	#include <string.h>

	#include "gdocumentportal.h"
	#include "xdp-dbus.h"
	#include "gstdio.h"

	#ifdef G_OS_UNIX
	#include "gunixfdlist.h"
	#endif

	#ifndef O_PATH
	#define O_PATH 0
	#endif
	#ifndef O_CLOEXEC
	#define O_CLOEXEC 0
	#else
	#define HAVE_O_CLOEXEC 1
	#endif

	static GXdpDocuments *documents;
	static char *documents_mountpoint;

	static gboolean
	init_document_portal (void)
	{
	static gsize documents_inited = 0;

	if (g_once_init_enter (&documents_inited))
	{
	GError *error = NULL;
	GDBusConnection *connection = g_bus_get_sync (G_BUS_TYPE_SESSION, NULL, &error);

	if (connection != NULL)
	{
	documents = gxdp_documents_proxy_new_sync (connection, 0,
	"org.freedesktop.portal.Documents",
	"/org/freedesktop/portal/documents",
	NULL, &error);
	if (documents != NULL)
	{
	gxdp_documents_call_get_mount_point_sync (documents,
	&documents_mountpoint,
	NULL, &error);

	if (error != NULL)
	{
	g_warning ("Cannot get document portal mount point: %s", error->message);
	g_error_free (error);
	}
	}
	else
	{
	g_warning ("Cannot create document portal proxy: %s", error->message);
	g_error_free (error);
	}

	g_object_unref (connection);
	}
	else
	{
	g_warning ("Cannot connect to session bus when initializing document portal: %s",
	error->message);
	g_error_free (error);
	}

	g_once_init_leave (&documents_inited, 1);
	}

	return (documents != NULL && documents_mountpoint != NULL);
	}

	char *
	g_document_portal_add_document (GFile *file,
	GError **error)
	{
	char doc_path, basename;
	char *doc_id = NULL;
	char *doc_uri = NULL;
	char *path = NULL;
	GUnixFDList *fd_list = NULL;
	int fd, fd_in, errsv;
	gboolean ret;

	if (!init_document_portal ())
	{
	g_set_error (error, G_IO_ERROR, G_IO_ERROR_NOT_INITIALIZED,
	"Document portal is not available");
	goto out;
	}

	path = g_file_get_path (file);
	fd = g_open (path, O_PATH \| O_CLOEXEC);
	errsv = errno;

	if (fd == -1)
	{
	g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errsv),
	"Failed to open %s", path);
	goto out;
	}

	#ifndef HAVE_O_CLOEXEC
	fcntl (fd, F_SETFD, FD_CLOEXEC);
	#endif

	fd_list = g_unix_fd_list_new ();
	fd_in = g_unix_fd_list_append (fd_list, fd, error);
	g_close (fd, NULL);

	if (fd_in == -1)
	goto out;

	ret = gxdp_documents_call_add_sync (documents,
	g_variant_new_handle (fd_in),
	TRUE,
	TRUE,
	fd_list,
	&doc_id,
	NULL,
	NULL,
	error);

	if (!ret)
	goto out;

	basename = g_path_get_basename (path);
	doc_path = g_build_filename (documents_mountpoint, doc_id, basename, NULL);
	g_free (basename);

	doc_uri = g_filename_to_uri (doc_path, NULL, NULL);
	g_free (doc_path);

	out:
	if (fd_list)
	g_object_unref (fd_list);
	g_free (path);
	g_free (doc_id);

	return doc_uri;
	}

	/* Flags accepted by org.freedesktop.portal.Documents.AddFull */
	enum {
	XDP_ADD_FLAGS_REUSE_EXISTING = (1 << 0),
	XDP_ADD_FLAGS_PERSISTENT = (1 << 1),
	XDP_ADD_FLAGS_AS_NEEDED_BY_APP = (1 << 2),
	XDP_ADD_FLAGS_FLAGS_ALL = ((1 << 3) - 1)
	};

	GList *
	g_document_portal_add_documents (GList *uris,
	const char *app_id,
	GError **error)
	{
	int length;
	GList *ruris = NULL;
	gboolean *as_is;
	GVariantBuilder builder;
	GUnixFDList *fd_list = NULL;
	GList *l;
	gsize i, j;
	const char *permissions[] = { "read", "write", NULL };
	char **doc_ids = NULL;
	GVariant *extra_out = NULL;

	if (!init_document_portal ())
	{
	g_set_error (error, G_IO_ERROR, G_IO_ERROR_NOT_INITIALIZED,
	"Document portal is not available");
	return NULL;
	}

	length = g_list_length (uris);
	as_is = g_new0 (gboolean, length);

	g_variant_builder_init (&builder, G_VARIANT_TYPE ("ah"));

	fd_list = g_unix_fd_list_new ();
	for (l = uris, i = 0; l; l = l->next, i++)
	{
	const char *uri = l->data;
	int idx = -1;
	char *path = NULL;

	path = g_filename_from_uri (uri, NULL, NULL);
	if (path != NULL)
	{
	int fd;

	fd = g_open (path, O_CLOEXEC \| O_PATH);
	if (fd >= 0)
	{
	#ifndef HAVE_O_CLOEXEC
	fcntl (fd, F_SETFD, FD_CLOEXEC);
	#endif
	idx = g_unix_fd_list_append (fd_list, fd, NULL);
	close (fd);
	}
	}

	g_free (path);

	if (idx != -1)
	g_variant_builder_add (&builder, "h", idx);
	else
	as_is[i] = TRUE;
	}

	if (g_unix_fd_list_get_length (fd_list) > 0)
	{
	if (!gxdp_documents_call_add_full_sync (documents,
	g_variant_builder_end (&builder),
	XDP_ADD_FLAGS_AS_NEEDED_BY_APP,
	app_id,
	permissions,
	fd_list,
	&doc_ids,
	&extra_out,
	NULL,
	NULL,
	error))
	goto out;

	for (l = uris, i = 0, j = 0; l; l = l->next, i++)
	{
	const char *uri = l->data;
	char *ruri;

	if (as_is[i]) /* use as-is, not a file uri */
	{
	ruri = g_strdup (uri);
	}
	else if (strcmp (doc_ids[j], "") == 0) /* not rewritten */
	{
	ruri = g_strdup (uri);
	j++;
	}
	else
	{
	char *basename = g_path_get_basename (uri + strlen ("file:"));
	char *doc_path = g_build_filename (documents_mountpoint, doc_ids[j], basename, NULL);
	ruri = g_strconcat ("file:", doc_path, NULL);
	g_free (basename);
	g_free (doc_path);
	j++;
	}

	ruris = g_list_prepend (ruris, ruri);
	}

	ruris = g_list_reverse (ruris);
	}
	else
	{
	ruris = g_list_copy_deep (uris, (GCopyFunc)g_strdup, NULL);
	}

	out:
	g_clear_object (&fd_list);
	g_clear_pointer (&extra_out, g_variant_unref);
	g_clear_pointer (&doc_ids, g_strfreev);
	g_free (as_is);

	return ruris;
	}