1/***************************************************************************
2 *                                  _   _ ____  _
3 *  Project                     ___| | | |  _ \| |
4 *                             / __| | | | |_) | |
5 *                            | (__| |_| |  _ <| |___
6 *                             \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 1998 - 2013, Daniel Stenberg, <daniel@haxx.se>, et al.
9 *
10 * This software is licensed as described in the file COPYING, which
11 * you should have received as part of this distribution. The terms
12 * are also available at http://curl.haxx.se/docs/copyright.html.
13 *
14 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15 * copies of the Software, and permit persons to whom the Software is
16 * furnished to do so, under the terms of the COPYING file.
17 *
18 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19 * KIND, either express or implied.
20 *
21 ***************************************************************************/
22#include "tool_setup.h"
23
24#ifdef HAVE_FSETXATTR
25#  include <sys/xattr.h> /* header from libc, not from libattr */
26#  define USE_XATTR
27#elif defined(__FreeBSD_version) && (__FreeBSD_version > 500000)
28#  include <sys/types.h>
29#  include <sys/extattr.h>
30#  define USE_XATTR
31#endif
32
33#include "tool_xattr.h"
34
35#include "memdebug.h" /* keep this as LAST include */
36
37#ifdef USE_XATTR
38
39/* mapping table of curl metadata to extended attribute names */
40static const struct xattr_mapping {
41  const char *attr; /* name of the xattr */
42  CURLINFO info;
43} mappings[] = {
44  /* mappings proposed by
45   * http://freedesktop.org/wiki/CommonExtendedAttributes
46   */
47  { "user.xdg.origin.url", CURLINFO_EFFECTIVE_URL },
48  { "user.mime_type",      CURLINFO_CONTENT_TYPE },
49  { NULL,                  CURLINFO_NONE } /* last element, abort loop here */
50};
51
52/* store metadata from the curl request alongside the downloaded
53 * file using extended attributes
54 */
55int fwrite_xattr(CURL *curl, int fd)
56{
57  int i = 0;
58  int err = 0;
59  /* loop through all xattr-curlinfo pairs and abort on a set error */
60  while(err == 0 && mappings[i].attr != NULL) {
61    char *value = NULL;
62    CURLcode rc = curl_easy_getinfo(curl, mappings[i].info, &value);
63    if(rc == CURLE_OK && value) {
64#ifdef HAVE_FSETXATTR_6
65      err = fsetxattr(fd, mappings[i].attr, value, strlen(value), 0, 0);
66#elif defined(HAVE_FSETXATTR_5)
67      err = fsetxattr(fd, mappings[i].attr, value, strlen(value), 0);
68#elif defined(__FreeBSD_version)
69      err = extattr_set_fd(fd, EXTATTR_NAMESPACE_USER, mappings[i].attr, value,
70                           strlen(value));
71      /* FreeBSD's extattr_set_fd returns the length of the extended attribute
72       */
73      err = err < 0 ? err : 0;
74#endif
75    }
76    i++;
77  }
78  return err;
79}
80#else
81int fwrite_xattr(CURL *curl, int fd)
82{
83  (void)curl;
84  (void)fd;
85  return 0;
86}
87#endif
88