about summary refs log tree commit diff
path: root/third_party/nix/src/libstore/download.hh
blob: 8e07403bcccd3fb1b591ecfd1ea6990cde4e523f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#pragma once

#include <future>
#include <string>
#include "globals.hh"
#include "hash.hh"
#include "types.hh"

namespace nix {

struct DownloadSettings : Config {
  Setting<bool> enableHttp2{this, true, "http2",
                            "Whether to enable HTTP/2 support."};

  Setting<std::string> userAgentSuffix{
      this, "", "user-agent-suffix",
      "String appended to the user agent in HTTP requests."};

  Setting<size_t> httpConnections{this,
                                  25,
                                  "http-connections",
                                  "Number of parallel HTTP connections.",
                                  {"binary-caches-parallel-connections"}};

  Setting<unsigned long> connectTimeout{
      this, 0, "connect-timeout",
      "Timeout for connecting to servers during downloads. 0 means use curl's "
      "builtin default."};

  Setting<unsigned long> stalledDownloadTimeout{
      this, 300, "stalled-download-timeout",
      "Timeout (in seconds) for receiving data from servers during download. "
      "Nix cancels idle downloads after this timeout's duration."};

  Setting<unsigned int> tries{
      this, 5, "download-attempts",
      "How often Nix will attempt to download a file before giving up."};
};

extern DownloadSettings downloadSettings;

struct DownloadRequest {
  std::string uri;
  std::string expectedETag;
  bool verifyTLS = true;
  bool head = false;
  size_t tries = downloadSettings.tries;
  unsigned int baseRetryTimeMs = 250;
  ActivityId parentAct;
  bool decompress = true;
  std::shared_ptr<std::string> data;
  std::string mimeType;
  std::function<void(char*, size_t)> dataCallback;

  DownloadRequest(const std::string& uri)
      : uri(uri), parentAct(getCurActivity()) {}

  std::string verb() { return data ? "upload" : "download"; }
};

struct DownloadResult {
  bool cached = false;
  std::string etag;
  std::string effectiveUri;
  std::shared_ptr<std::string> data;
  uint64_t bodySize = 0;
};

struct CachedDownloadRequest {
  std::string uri;
  bool unpack = false;
  std::string name;
  Hash expectedHash;
  unsigned int ttl = settings.tarballTtl;

  CachedDownloadRequest(const std::string& uri) : uri(uri) {}
};

struct CachedDownloadResult {
  // Note: 'storePath' may be different from 'path' when using a
  // chroot store.
  Path storePath;
  Path path;
  std::optional<std::string> etag;
  std::string effectiveUri;
};

class Store;

struct Downloader {
  virtual ~Downloader() {}

  /* Enqueue a download request, returning a future to the result of
     the download. The future may throw a DownloadError
     exception. */
  virtual void enqueueDownload(const DownloadRequest& request,
                               Callback<DownloadResult> callback) = 0;

  std::future<DownloadResult> enqueueDownload(const DownloadRequest& request);

  /* Synchronously download a file. */
  DownloadResult download(const DownloadRequest& request);

  /* Download a file, writing its data to a sink. The sink will be
     invoked on the thread of the caller. */
  void download(DownloadRequest&& request, Sink& sink);

  /* Check if the specified file is already in ~/.cache/nix/tarballs
     and is more recent than ‘tarball-ttl’ seconds. Otherwise,
     use the recorded ETag to verify if the server has a more
     recent version, and if so, download it to the Nix store. */
  CachedDownloadResult downloadCached(ref<Store> store,
                                      const CachedDownloadRequest& request);

  enum Error { NotFound, Forbidden, Misc, Transient, Interrupted };
};

/* Return a shared Downloader object. Using this object is preferred
   because it enables connection reuse and HTTP/2 multiplexing. */
ref<Downloader> getDownloader();

/* Return a new Downloader object. */
ref<Downloader> makeDownloader();

class DownloadError : public Error {
 public:
  Downloader::Error error;
  DownloadError(Downloader::Error error, const FormatOrString& fs)
      : Error(fs), error(error) {}
};

bool isUri(const string& s);

}  // namespace nix