123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138 |
- /***************************************************************************
- * _ _ ____ _
- * Project ___| | | | _ \| |
- * / __| | | | |_) | |
- * | (__| |_| | _ <| |___
- * \___|\___/|_| \_\_____|
- *
- * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
- *
- * This software is licensed as described in the file COPYING, which
- * you should have received as part of this distribution. The terms
- * are also available at https://curl.se/docs/copyright.html.
- *
- * You may opt to use, copy, modify, merge, publish, distribute and/or sell
- * copies of the Software, and permit persons to whom the Software is
- * furnished to do so, under the terms of the COPYING file.
- *
- * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
- * KIND, either express or implied.
- *
- * SPDX-License-Identifier: curl
- *
- ***************************************************************************/
- #include "tool_setup.h"
- #include "tool_xattr.h"
- #include "memdebug.h" /* keep this as LAST include */
- #ifdef USE_XATTR
- /* mapping table of curl metadata to extended attribute names */
- static const struct xattr_mapping {
- const char *attr; /* name of the xattr */
- CURLINFO info;
- } mappings[] = {
- /* mappings proposed by
- * https://freedesktop.org/wiki/CommonExtendedAttributes/
- */
- { "user.xdg.referrer.url", CURLINFO_REFERER },
- { "user.mime_type", CURLINFO_CONTENT_TYPE },
- { NULL, CURLINFO_NONE } /* last element, abort here */
- };
- /* returns a new URL that needs to be freed */
- /* @unittest: 1621 */
- #ifdef UNITTESTS
- char *stripcredentials(const char *url);
- #else
- static
- #endif
- char *stripcredentials(const char *url)
- {
- CURLU *u;
- CURLUcode uc;
- char *nurl;
- u = curl_url();
- if(u) {
- uc = curl_url_set(u, CURLUPART_URL, url, 0);
- if(uc)
- goto error;
- uc = curl_url_set(u, CURLUPART_USER, NULL, 0);
- if(uc)
- goto error;
- uc = curl_url_set(u, CURLUPART_PASSWORD, NULL, 0);
- if(uc)
- goto error;
- uc = curl_url_get(u, CURLUPART_URL, &nurl, 0);
- if(uc)
- goto error;
- curl_url_cleanup(u);
- return nurl;
- }
- error:
- curl_url_cleanup(u);
- return NULL;
- }
- static int xattr(int fd,
- const char *attr, /* name of the xattr */
- const char *value)
- {
- int err = 0;
- if(value) {
- #ifdef DEBUGBUILD
- (void)fd;
- if(getenv("CURL_FAKE_XATTR")) {
- printf("%s => %s\n", attr, value);
- }
- return 0;
- #else
- #ifdef HAVE_FSETXATTR_6
- err = fsetxattr(fd, attr, value, strlen(value), 0, 0);
- #elif defined(HAVE_FSETXATTR_5)
- err = fsetxattr(fd, attr, value, strlen(value), 0);
- #elif defined(__FreeBSD_version) || defined(__MidnightBSD_version)
- {
- ssize_t rc = extattr_set_fd(fd, EXTATTR_NAMESPACE_USER,
- attr, value, strlen(value));
- /* FreeBSD's extattr_set_fd returns the length of the extended
- attribute */
- err = (rc < 0 ? -1 : 0);
- }
- #endif
- #endif
- }
- return err;
- }
- /* store metadata from the curl request alongside the downloaded
- * file using extended attributes
- */
- int fwrite_xattr(CURL *curl, const char *url, int fd)
- {
- int i = 0;
- int err = 0;
- /* loop through all xattr-curlinfo pairs and abort on a set error */
- while(!err && mappings[i].attr) {
- char *value = NULL;
- CURLcode result = curl_easy_getinfo(curl, mappings[i].info, &value);
- if(!result && value)
- err = xattr(fd, mappings[i].attr, value);
- i++;
- }
- if(!err) {
- char *nurl = stripcredentials(url);
- if(!nurl)
- return 1;
- err = xattr(fd, "user.xdg.origin.url", nurl);
- curl_free(nurl);
- }
- return err;
- }
- #endif
|