c1719a0adf
Fixes <https://issues.guix.gnu.org/54723>. Reported by Guillaume Le Vaillant <glv@posteo.net>. Regression introduced in f743f2046be2c5a338ab871ae8666d8f6de7440b. With commit f743f2046be2c5a338ab871ae8666d8f6de7440b, responses to pipelined GETs would end up being written concurrently by many threads. Thus the body of those responses could be interleaved and garbled. * guix/scripts/publish.scm: Revert f743f2046be2c5a338ab871ae8666d8f6de7440b. * tests/publish.scm ("/*.narinfo pipeline"): New test.
786 lines
32 KiB
Scheme
786 lines
32 KiB
Scheme
;;; GNU Guix --- Functional package management for GNU
|
||
;;; Copyright © 2015 David Thompson <davet@gnu.org>
|
||
;;; Copyright © 2020 by Amar M. Singh <nly@disroot.org>
|
||
;;; Copyright © 2016-2022 Ludovic Courtès <ludo@gnu.org>
|
||
;;;
|
||
;;; This file is part of GNU Guix.
|
||
;;;
|
||
;;; GNU Guix is free software; you can redistribute it and/or modify it
|
||
;;; under the terms of the GNU General Public License as published by
|
||
;;; the Free Software Foundation; either version 3 of the License, or (at
|
||
;;; your option) any later version.
|
||
;;;
|
||
;;; GNU Guix 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 General Public License for more details.
|
||
;;;
|
||
;;; You should have received a copy of the GNU General Public License
|
||
;;; along with GNU Guix. If not, see <http://www.gnu.org/licenses/>.
|
||
|
||
;; Avoid interference.
|
||
(unsetenv "http_proxy")
|
||
|
||
(define-module (test-publish)
|
||
#:use-module (guix scripts publish)
|
||
#:use-module (guix tests)
|
||
#:use-module (guix config)
|
||
#:use-module (guix utils)
|
||
#:use-module (gcrypt hash)
|
||
#:use-module (guix store)
|
||
#:use-module (guix derivations)
|
||
#:use-module (guix gexp)
|
||
#:use-module (guix base32)
|
||
#:use-module (guix base64)
|
||
#:use-module ((guix records) #:select (recutils->alist))
|
||
#:use-module ((guix serialization) #:select (restore-file))
|
||
#:use-module (gcrypt pk-crypto)
|
||
#:use-module ((guix pki) #:select (%public-key-file %private-key-file))
|
||
#:use-module (zlib)
|
||
#:use-module (lzlib)
|
||
#:autoload (zstd) (call-with-zstd-input-port)
|
||
#:use-module (web uri)
|
||
#:use-module (web client)
|
||
#:use-module (web request)
|
||
#:use-module (web response)
|
||
#:use-module ((guix http-client) #:select (http-multiple-get))
|
||
#:use-module (rnrs bytevectors)
|
||
#:use-module (ice-9 binary-ports)
|
||
#:use-module (srfi srfi-1)
|
||
#:use-module (srfi srfi-26)
|
||
#:use-module (srfi srfi-64)
|
||
#:use-module (srfi srfi-71)
|
||
#:use-module (ice-9 threads)
|
||
#:use-module (ice-9 format)
|
||
#:use-module (ice-9 match)
|
||
#:use-module (ice-9 rdelim))
|
||
|
||
(define %store
|
||
(open-connection-for-tests))
|
||
|
||
(define (zstd-supported?)
|
||
(resolve-module '(zstd) #t #f #:ensure #f))
|
||
|
||
(define %reference (add-text-to-store %store "ref" "foo"))
|
||
|
||
(define %item (add-text-to-store %store "item" "bar" (list %reference)))
|
||
|
||
(define (http-get-body uri)
|
||
(call-with-values (lambda () (http-get uri))
|
||
(lambda (response body) body)))
|
||
|
||
(define (http-get-port uri)
|
||
(let ((socket (open-socket-for-uri uri)))
|
||
;; Make sure to use an unbuffered port so that we can then peek at the
|
||
;; underlying file descriptor via 'call-with-gzip-input-port'.
|
||
(setvbuf socket 'none)
|
||
(call-with-values
|
||
(lambda ()
|
||
(http-get uri #:port socket #:streaming? #t))
|
||
(lambda (response port)
|
||
;; Don't (setvbuf port 'none) because of <http://bugs.gnu.org/19610>
|
||
;; (PORT might be a custom binary input port).
|
||
port))))
|
||
|
||
(define (publish-uri route)
|
||
(string-append "http://localhost:6789" route))
|
||
|
||
(define-syntax-rule (with-separate-output-ports exp ...)
|
||
;; Since ports aren't thread-safe in Guile 2.0, duplicate the output and
|
||
;; error ports to make sure the two threads don't end up stepping on each
|
||
;; other's toes.
|
||
(with-output-to-port (duplicate-port (current-output-port) "w")
|
||
(lambda ()
|
||
(with-error-to-port (duplicate-port (current-error-port) "w")
|
||
(lambda ()
|
||
exp ...)))))
|
||
|
||
;; Run a local publishing server in a separate thread.
|
||
(with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6789" "-C0")))) ;attempt to avoid port collision
|
||
|
||
(define (wait-until-ready port)
|
||
;; Wait until the server is accepting connections.
|
||
(let ((conn (socket PF_INET SOCK_STREAM 0)))
|
||
(let loop ()
|
||
(unless (false-if-exception
|
||
(connect conn AF_INET (inet-pton AF_INET "127.0.0.1") port))
|
||
(loop)))))
|
||
|
||
(define (wait-for-file file)
|
||
;; Wait until FILE shows up.
|
||
(let loop ((i 20))
|
||
(cond ((file-exists? file)
|
||
#t)
|
||
((zero? i)
|
||
(error "file didn't show up" file))
|
||
(else
|
||
(pk 'wait-for-file file)
|
||
(sleep 1)
|
||
(loop (- i 1))))))
|
||
|
||
(define %gzip-magic-bytes
|
||
;; Magic bytes of gzip file.
|
||
#vu8(#x1f #x8b))
|
||
|
||
;; Wait until the two servers are ready.
|
||
(wait-until-ready 6789)
|
||
|
||
;; Initialize the public/private key SRFI-39 parameters.
|
||
(%public-key (read-file-sexp %public-key-file))
|
||
(%private-key (read-file-sexp %private-key-file))
|
||
|
||
|
||
(test-begin "publish")
|
||
|
||
(test-equal "/nix-cache-info"
|
||
(format #f "StoreDir: ~a\nWantMassQuery: 0\nPriority: 100\n"
|
||
%store-directory)
|
||
(http-get-body (publish-uri "/nix-cache-info")))
|
||
|
||
(test-equal "/*.narinfo"
|
||
(let* ((info (query-path-info %store %item))
|
||
(unsigned-info
|
||
(format #f
|
||
"StorePath: ~a
|
||
NarHash: sha256:~a
|
||
NarSize: ~d
|
||
References: ~a~%"
|
||
%item
|
||
(bytevector->nix-base32-string
|
||
(path-info-hash info))
|
||
(path-info-nar-size info)
|
||
(basename (first (path-info-references info)))))
|
||
(signature (base64-encode
|
||
(string->utf8
|
||
(canonical-sexp->string
|
||
(signed-string unsigned-info))))))
|
||
(format #f "~aSignature: 1;~a;~a
|
||
URL: nar/~a
|
||
Compression: none
|
||
FileSize: ~a\n"
|
||
unsigned-info (gethostname) signature
|
||
(basename %item)
|
||
(path-info-nar-size info)))
|
||
(utf8->string
|
||
(http-get-body
|
||
(publish-uri
|
||
(string-append "/" (store-path-hash-part %item) ".narinfo")))))
|
||
|
||
(test-equal "/*.narinfo pipeline"
|
||
(make-list 500 200)
|
||
;; Make sure clients can pipeline requests and correct responses, in the
|
||
;; right order. See <https://issues.guix.gnu.org/54723>.
|
||
(let* ((uri (string->uri (publish-uri
|
||
(string-append "/"
|
||
(store-path-hash-part %item)
|
||
".narinfo"))))
|
||
(_ expected (http-get uri #:streaming? #f #:decode-body? #f)))
|
||
(http-multiple-get (string->uri (publish-uri ""))
|
||
(lambda (request response port result)
|
||
(and (bytevector=? expected
|
||
(get-bytevector-n port
|
||
(response-content-length
|
||
response)))
|
||
(cons (response-code response) result)))
|
||
'()
|
||
(make-list 500 (build-request uri))
|
||
#:batch-size 77)))
|
||
|
||
(test-equal "/*.narinfo with properly encoded '+' sign"
|
||
;; See <http://bugs.gnu.org/21888>.
|
||
(let* ((item (add-text-to-store %store "fake-gtk+" "Congrats!"))
|
||
(info (query-path-info %store item))
|
||
(unsigned-info
|
||
(format #f
|
||
"StorePath: ~a
|
||
NarHash: sha256:~a
|
||
NarSize: ~d
|
||
References: ~%"
|
||
item
|
||
(bytevector->nix-base32-string
|
||
(path-info-hash info))
|
||
(path-info-nar-size info)))
|
||
(signature (base64-encode
|
||
(string->utf8
|
||
(canonical-sexp->string
|
||
(signed-string unsigned-info))))))
|
||
(format #f "~aSignature: 1;~a;~a
|
||
URL: nar/~a
|
||
Compression: none
|
||
FileSize: ~a~%"
|
||
unsigned-info (gethostname) signature
|
||
(uri-encode (basename item))
|
||
(path-info-nar-size info)))
|
||
|
||
(let ((item (add-text-to-store %store "fake-gtk+" "Congrats!")))
|
||
(utf8->string
|
||
(http-get-body
|
||
(publish-uri
|
||
(string-append "/" (store-path-hash-part item) ".narinfo"))))))
|
||
|
||
(test-equal "/nar/*"
|
||
"bar"
|
||
(call-with-temporary-output-file
|
||
(lambda (temp port)
|
||
(let ((nar (utf8->string
|
||
(http-get-body
|
||
(publish-uri
|
||
(string-append "/nar/" (basename %item)))))))
|
||
(call-with-input-string nar (cut restore-file <> temp)))
|
||
(call-with-input-file temp read-string))))
|
||
|
||
(test-equal "/nar/gzip/*"
|
||
"bar"
|
||
(call-with-temporary-output-file
|
||
(lambda (temp port)
|
||
(let ((nar (http-get-port
|
||
(publish-uri
|
||
(string-append "/nar/gzip/" (basename %item))))))
|
||
(call-with-gzip-input-port nar
|
||
(cut restore-file <> temp)))
|
||
(call-with-input-file temp read-string))))
|
||
|
||
(test-equal "/nar/gzip/* is really gzip"
|
||
%gzip-magic-bytes
|
||
;; Since 'gzdopen' (aka. 'call-with-gzip-input-port') transparently reads
|
||
;; uncompressed gzip, the test above doesn't check whether it's actually
|
||
;; gzip. This is what this test does. See <https://bugs.gnu.org/30184>.
|
||
(let ((nar (http-get-port
|
||
(publish-uri
|
||
(string-append "/nar/gzip/" (basename %item))))))
|
||
(get-bytevector-n nar (bytevector-length %gzip-magic-bytes))))
|
||
|
||
(test-equal "/nar/lzip/*"
|
||
"bar"
|
||
(call-with-temporary-output-file
|
||
(lambda (temp port)
|
||
(let ((nar (http-get-port
|
||
(publish-uri
|
||
(string-append "/nar/lzip/" (basename %item))))))
|
||
(call-with-lzip-input-port nar
|
||
(cut restore-file <> temp)))
|
||
(call-with-input-file temp read-string))))
|
||
|
||
(unless (zstd-supported?) (test-skip 1))
|
||
(test-equal "/nar/zstd/*"
|
||
"bar"
|
||
(call-with-temporary-output-file
|
||
(lambda (temp port)
|
||
(let ((nar (http-get-port
|
||
(publish-uri
|
||
(string-append "/nar/zstd/" (basename %item))))))
|
||
(call-with-zstd-input-port nar
|
||
(cut restore-file <> temp)))
|
||
(call-with-input-file temp read-string))))
|
||
|
||
(test-equal "/*.narinfo with compression"
|
||
`(("StorePath" . ,%item)
|
||
("URL" . ,(string-append "nar/gzip/" (basename %item)))
|
||
("Compression" . "gzip"))
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6799" "-C5"))))))
|
||
(wait-until-ready 6799)
|
||
(let* ((url (string-append "http://localhost:6799/"
|
||
(store-path-hash-part %item) ".narinfo"))
|
||
(body (http-get-port url)))
|
||
(filter (lambda (item)
|
||
(match item
|
||
(("Compression" . _) #t)
|
||
(("StorePath" . _) #t)
|
||
(("URL" . _) #t)
|
||
(_ #f)))
|
||
(recutils->alist body)))))
|
||
|
||
(test-equal "/*.narinfo with lzip compression"
|
||
`(("StorePath" . ,%item)
|
||
("URL" . ,(string-append "nar/lzip/" (basename %item)))
|
||
("Compression" . "lzip"))
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6790" "-Clzip"))))))
|
||
(wait-until-ready 6790)
|
||
(let* ((url (string-append "http://localhost:6790/"
|
||
(store-path-hash-part %item) ".narinfo"))
|
||
(body (http-get-port url)))
|
||
(filter (lambda (item)
|
||
(match item
|
||
(("Compression" . _) #t)
|
||
(("StorePath" . _) #t)
|
||
(("URL" . _) #t)
|
||
(_ #f)))
|
||
(recutils->alist body)))))
|
||
|
||
(test-equal "/*.narinfo for a compressed file"
|
||
'("none" "nar") ;compression-less nar
|
||
;; Assume 'guix publish -C' is already running on port 6799.
|
||
(let* ((item (add-text-to-store %store "fake.tar.gz"
|
||
"This is a fake compressed file."))
|
||
(url (string-append "http://localhost:6799/"
|
||
(store-path-hash-part item) ".narinfo"))
|
||
(body (http-get-port url))
|
||
(info (recutils->alist body)))
|
||
(list (assoc-ref info "Compression")
|
||
(dirname (assoc-ref info "URL")))))
|
||
|
||
(test-equal "/*.narinfo with lzip + gzip"
|
||
`((("StorePath" . ,%item)
|
||
("URL" . ,(string-append "nar/gzip/" (basename %item)))
|
||
("Compression" . "gzip")
|
||
("URL" . ,(string-append "nar/lzip/" (basename %item)))
|
||
("Compression" . "lzip"))
|
||
200
|
||
200)
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6793" "-Cgzip:2" "-Clzip:2"))))))
|
||
(wait-until-ready 6793)
|
||
(let* ((base "http://localhost:6793/")
|
||
(part (store-path-hash-part %item))
|
||
(url (string-append base part ".narinfo"))
|
||
(body (http-get-port url)))
|
||
(list (filter (match-lambda
|
||
(("StorePath" . _) #t)
|
||
(("URL" . _) #t)
|
||
(("Compression" . _) #t)
|
||
(_ #f))
|
||
(recutils->alist body))
|
||
(response-code
|
||
(http-get (string-append base "nar/gzip/"
|
||
(basename %item))))
|
||
(response-code
|
||
(http-get (string-append base "nar/lzip/"
|
||
(basename %item))))))))))
|
||
|
||
(test-equal "custom nar path"
|
||
;; Serve nars at /foo/bar/chbouib instead of /nar.
|
||
(list `(("StorePath" . ,%item)
|
||
("URL" . ,(string-append "foo/bar/chbouib/" (basename %item)))
|
||
("Compression" . "none"))
|
||
200
|
||
404)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6798" "-C0"
|
||
"--nar-path=///foo/bar//chbouib/"))))))
|
||
(wait-until-ready 6798)
|
||
(let* ((base "http://localhost:6798/")
|
||
(part (store-path-hash-part %item))
|
||
(url (string-append base part ".narinfo"))
|
||
(nar-url (string-append base "foo/bar/chbouib/"
|
||
(basename %item)))
|
||
(body (http-get-port url)))
|
||
(list (filter (lambda (item)
|
||
(match item
|
||
(("Compression" . _) #t)
|
||
(("StorePath" . _) #t)
|
||
(("URL" . _) #t)
|
||
(_ #f)))
|
||
(recutils->alist body))
|
||
(response-code (http-get nar-url))
|
||
(response-code
|
||
(http-get (string-append base "nar/" (basename %item))))))))
|
||
|
||
(test-equal "/nar/ with properly encoded '+' sign"
|
||
"Congrats!"
|
||
(let ((item (add-text-to-store %store "fake-gtk+" "Congrats!")))
|
||
(call-with-temporary-output-file
|
||
(lambda (temp port)
|
||
(let ((nar (utf8->string
|
||
(http-get-body
|
||
(publish-uri
|
||
(string-append "/nar/" (uri-encode (basename item))))))))
|
||
(call-with-input-string nar (cut restore-file <> temp)))
|
||
(call-with-input-file temp read-string)))))
|
||
|
||
(test-equal "/nar/invalid"
|
||
404
|
||
(begin
|
||
(call-with-output-file (string-append (%store-prefix) "/invalid")
|
||
(lambda (port)
|
||
(display "This file is not a valid store item." port)))
|
||
(response-code (http-get (publish-uri (string-append "/nar/invalid"))))))
|
||
|
||
(test-equal "/file/NAME/sha256/HASH"
|
||
"Hello, Guix world!"
|
||
(let* ((data "Hello, Guix world!")
|
||
(hash (call-with-input-string data port-sha256))
|
||
(drv (run-with-store %store
|
||
(gexp->derivation "the-file.txt"
|
||
#~(call-with-output-file #$output
|
||
(lambda (port)
|
||
(display #$data port)))
|
||
#:hash-algo 'sha256
|
||
#:hash hash)))
|
||
(out (build-derivations %store (list drv))))
|
||
(utf8->string
|
||
(http-get-body
|
||
(publish-uri
|
||
(string-append "/file/the-file.txt/sha256/"
|
||
(bytevector->nix-base32-string hash)))))))
|
||
|
||
(test-equal "/file/NAME/sha256/INVALID-NIX-BASE32-STRING"
|
||
404
|
||
(let ((uri (publish-uri
|
||
"/file/the-file.txt/sha256/not-a-nix-base32-string")))
|
||
(response-code (http-get uri))))
|
||
|
||
(test-equal "/file/NAME/sha256/INVALID-HASH"
|
||
404
|
||
(let ((uri (publish-uri
|
||
(string-append "/file/the-file.txt/sha256/"
|
||
(bytevector->nix-base32-string
|
||
(call-with-input-string "" port-sha256))))))
|
||
(response-code (http-get uri))))
|
||
|
||
(test-equal "with cache"
|
||
(list #t
|
||
`(("StorePath" . ,%item)
|
||
("URL" . ,(string-append "nar/gzip/" (basename %item)))
|
||
("Compression" . "gzip"))
|
||
200 ;nar/gzip/…
|
||
#t ;Content-Length
|
||
#t ;FileSize
|
||
404) ;nar/…
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6797" "-C2"
|
||
(string-append "--cache=" cache)
|
||
"--cache-bypass-threshold=0"))))))
|
||
(wait-until-ready 6797)
|
||
(let* ((base "http://localhost:6797/")
|
||
(part (store-path-hash-part %item))
|
||
(url (string-append base part ".narinfo"))
|
||
(nar-url (string-append base "nar/gzip/" (basename %item)))
|
||
(cached (string-append cache "/gzip/" (basename %item)
|
||
".narinfo"))
|
||
(nar (string-append cache "/gzip/"
|
||
(basename %item) ".nar"))
|
||
(response (http-get url)))
|
||
(and (= 404 (response-code response))
|
||
|
||
;; We should get an explicitly short TTL for 404 in this case
|
||
;; because it's going to become 200 shortly.
|
||
(match (assq-ref (response-headers response) 'cache-control)
|
||
((('max-age . ttl))
|
||
(< ttl 3600)))
|
||
|
||
(wait-for-file cached)
|
||
|
||
;; Both the narinfo and nar should be world-readable.
|
||
(= #o444 (logand #o444 (stat:perms (lstat cached))))
|
||
(= #o444 (logand #o444 (stat:perms (lstat nar))))
|
||
|
||
(let* ((body (http-get-port url))
|
||
(compressed (http-get nar-url))
|
||
(uncompressed (http-get (string-append base "nar/"
|
||
(basename %item))))
|
||
(narinfo (recutils->alist body)))
|
||
(list (file-exists? nar)
|
||
(filter (lambda (item)
|
||
(match item
|
||
(("Compression" . _) #t)
|
||
(("StorePath" . _) #t)
|
||
(("URL" . _) #t)
|
||
(_ #f)))
|
||
narinfo)
|
||
(response-code compressed)
|
||
(= (response-content-length compressed)
|
||
(stat:size (stat nar)))
|
||
(= (string->number
|
||
(assoc-ref narinfo "FileSize"))
|
||
(stat:size (stat nar)))
|
||
(response-code uncompressed)))))))))
|
||
|
||
(test-equal "with cache, lzip + gzip"
|
||
'(200 200 404)
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6794" "-Cgzip:2" "-Clzip:2"
|
||
(string-append "--cache=" cache)
|
||
"--cache-bypass-threshold=0"))))))
|
||
(wait-until-ready 6794)
|
||
(let* ((base "http://localhost:6794/")
|
||
(part (store-path-hash-part %item))
|
||
(url (string-append base part ".narinfo"))
|
||
(nar-url (cute string-append "nar/" <> "/"
|
||
(basename %item)))
|
||
(cached (cute string-append cache "/" <> "/"
|
||
(basename %item) ".narinfo"))
|
||
(nar (cute string-append cache "/" <> "/"
|
||
(basename %item) ".nar"))
|
||
(response (http-get url)))
|
||
(wait-for-file (cached "gzip"))
|
||
(let* ((body (http-get-port url))
|
||
(narinfo (recutils->alist body))
|
||
(uncompressed (string-append base "nar/"
|
||
(basename %item))))
|
||
(and (file-exists? (nar "gzip"))
|
||
(file-exists? (nar "lzip"))
|
||
(match (pk 'narinfo/gzip+lzip narinfo)
|
||
((("StorePath" . path)
|
||
_ ...
|
||
("Signature" . _)
|
||
("URL" . gzip-url)
|
||
("Compression" . "gzip")
|
||
("FileSize" . (= string->number gzip-size))
|
||
("URL" . lzip-url)
|
||
("Compression" . "lzip")
|
||
("FileSize" . (= string->number lzip-size)))
|
||
(and (string=? gzip-url (nar-url "gzip"))
|
||
(string=? lzip-url (nar-url "lzip"))
|
||
(= gzip-size
|
||
(stat:size (stat (nar "gzip"))))
|
||
(= lzip-size
|
||
(stat:size (stat (nar "lzip")))))))
|
||
(list (response-code
|
||
(http-get (string-append base (nar-url "gzip"))))
|
||
(response-code
|
||
(http-get (string-append base (nar-url "lzip"))))
|
||
(response-code
|
||
(http-get uncompressed))))))))))
|
||
|
||
(let ((item (add-text-to-store %store "fake-compressed-thing.tar.gz"
|
||
(random-text))))
|
||
(test-equal "with cache, uncompressed"
|
||
(list #t
|
||
(* 42 3600) ;TTL on narinfo
|
||
`(("StorePath" . ,item)
|
||
("URL" . ,(string-append "nar/" (basename item)))
|
||
("Compression" . "none"))
|
||
200 ;nar/…
|
||
(* 42 3600) ;TTL on nar/…
|
||
(path-info-nar-size
|
||
(query-path-info %store item)) ;FileSize
|
||
404) ;nar/gzip/…
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6796" "-C2" "--ttl=42h"
|
||
(string-append "--cache=" cache)
|
||
"--cache-bypass-threshold=0"))))))
|
||
(wait-until-ready 6796)
|
||
(let* ((base "http://localhost:6796/")
|
||
(part (store-path-hash-part item))
|
||
(url (string-append base part ".narinfo"))
|
||
(cached (string-append cache "/none/"
|
||
(basename item) ".narinfo"))
|
||
(nar (string-append cache "/none/"
|
||
(basename item) ".nar"))
|
||
(response (http-get url)))
|
||
(and (= 404 (response-code response))
|
||
|
||
(wait-for-file cached)
|
||
(let* ((response (http-get url))
|
||
(body (http-get-port url))
|
||
(compressed (http-get (string-append base "nar/gzip/"
|
||
(basename item))))
|
||
(uncompressed (http-get (string-append base "nar/"
|
||
(basename item))))
|
||
(narinfo (recutils->alist body)))
|
||
(list (file-exists? nar)
|
||
(match (assq-ref (response-headers response)
|
||
'cache-control)
|
||
((('max-age . ttl)) ttl)
|
||
(_ #f))
|
||
|
||
(filter (lambda (item)
|
||
(match item
|
||
(("Compression" . _) #t)
|
||
(("StorePath" . _) #t)
|
||
(("URL" . _) #t)
|
||
(_ #f)))
|
||
narinfo)
|
||
(response-code uncompressed)
|
||
(match (assq-ref (response-headers uncompressed)
|
||
'cache-control)
|
||
((('max-age . ttl)) ttl)
|
||
(_ #f))
|
||
|
||
(string->number
|
||
(assoc-ref narinfo "FileSize"))
|
||
(response-code compressed))))))))))
|
||
|
||
(test-equal "with cache, vanishing item" ;<https://bugs.gnu.org/33897>
|
||
200
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6795"
|
||
(string-append "--cache=" cache)))))))
|
||
(wait-until-ready 6795)
|
||
|
||
;; Make sure that, even if ITEM disappears, we're still able to fetch
|
||
;; it.
|
||
(let* ((base "http://localhost:6795/")
|
||
(item (add-text-to-store %store "random" (random-text)))
|
||
(part (store-path-hash-part item))
|
||
(url (string-append base part ".narinfo"))
|
||
(cached (string-append cache "/gzip/"
|
||
(basename item)
|
||
".narinfo"))
|
||
(response (http-get url)))
|
||
(and (= 200 (response-code response)) ;we're below the threshold
|
||
(wait-for-file cached)
|
||
(begin
|
||
(delete-paths %store (list item))
|
||
(response-code (pk 'response (http-get url))))))))))
|
||
|
||
(test-equal "with cache, cache bypass"
|
||
200
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6788" "-C" "gzip"
|
||
(string-append "--cache=" cache)))))))
|
||
(wait-until-ready 6788)
|
||
|
||
(let* ((base "http://localhost:6788/")
|
||
(item (add-text-to-store %store "random" (random-text)))
|
||
(part (store-path-hash-part item))
|
||
(narinfo (string-append base part ".narinfo"))
|
||
(nar (string-append base "nar/gzip/" (basename item)))
|
||
(cached (string-append cache "/gzip/" (basename item)
|
||
".narinfo")))
|
||
;; We're below the default cache bypass threshold, so NAR and NARINFO
|
||
;; should immediately return 200. The NARINFO request should trigger
|
||
;; caching, and the next request to NAR should return 200 as well.
|
||
(and (let ((response (pk 'r1 (http-get nar))))
|
||
(and (= 200 (response-code response))
|
||
(not (response-content-length response)))) ;not known
|
||
(= 200 (response-code (http-get narinfo)))
|
||
(begin
|
||
(wait-for-file cached)
|
||
(let ((response (pk 'r2 (http-get nar))))
|
||
(and (> (response-content-length response)
|
||
(stat:size (stat item)))
|
||
(response-code response))))))))))
|
||
|
||
(test-equal "with cache, cache bypass, unmapped hash part"
|
||
200
|
||
|
||
;; This test reproduces the bug described in <https://bugs.gnu.org/44442>:
|
||
;; the daemon connection would be closed as a side effect of a nar request
|
||
;; for a non-existing file name.
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6787" "-C" "gzip"
|
||
(string-append "--cache=" cache)))))))
|
||
(wait-until-ready 6787)
|
||
|
||
(let* ((base "http://localhost:6787/")
|
||
(item (add-text-to-store %store "random" (random-text)))
|
||
(part (store-path-hash-part item))
|
||
(narinfo (string-append base part ".narinfo"))
|
||
(nar (string-append base "nar/gzip/" (basename item)))
|
||
(cached (string-append cache "/gzip/" (basename item)
|
||
".narinfo")))
|
||
;; The first response used to be 500 and to terminate the daemon
|
||
;; connection as a side effect.
|
||
(and (= (response-code
|
||
(http-get (string-append base "nar/gzip/"
|
||
(make-string 32 #\e)
|
||
"-does-not-exist")))
|
||
404)
|
||
(= 200 (response-code (http-get nar)))
|
||
(= 200 (response-code (http-get narinfo)))
|
||
(begin
|
||
(wait-for-file cached)
|
||
(response-code (http-get nar)))))))))
|
||
|
||
(test-equal "/log/NAME"
|
||
`(200 #t text/plain (gzip))
|
||
(let ((drv (run-with-store %store
|
||
(gexp->derivation "with-log"
|
||
#~(call-with-output-file #$output
|
||
(lambda (port)
|
||
(display "Hello, build log!"
|
||
(current-error-port))
|
||
(display #$(random-text) port)))))))
|
||
(build-derivations %store (list drv))
|
||
(let* ((response (http-get
|
||
(publish-uri (string-append "/log/"
|
||
(basename (derivation->output-path drv))))
|
||
#:decode-body? #f))
|
||
(base (basename (derivation-file-name drv)))
|
||
(log (string-append (dirname %state-directory)
|
||
"/log/guix/drvs/" (string-take base 2)
|
||
"/" (string-drop base 2) ".gz")))
|
||
(list (response-code response)
|
||
(= (response-content-length response) (stat:size (stat log)))
|
||
(first (response-content-type response))
|
||
(response-content-encoding response)))))
|
||
|
||
(test-equal "negative TTL"
|
||
`(404 42)
|
||
|
||
(call-with-temporary-directory
|
||
(lambda (cache)
|
||
(let ((thread (with-separate-output-ports
|
||
(call-with-new-thread
|
||
(lambda ()
|
||
(guix-publish "--port=6786" "-C0"
|
||
"--negative-ttl=42s"))))))
|
||
(wait-until-ready 6786)
|
||
|
||
(let* ((base "http://localhost:6786/")
|
||
(url (string-append base (make-string 32 #\z)
|
||
".narinfo"))
|
||
(response (http-get url)))
|
||
(list (response-code response)
|
||
(match (assq-ref (response-headers response) 'cache-control)
|
||
((('max-age . ttl)) ttl)
|
||
(_ #f))))))))
|
||
|
||
(test-equal "no negative TTL"
|
||
`(404 #f)
|
||
(let* ((uri (publish-uri
|
||
(string-append "/" (make-string 32 #\z)
|
||
".narinfo")))
|
||
(response (http-get uri)))
|
||
(list (response-code response)
|
||
(assq-ref (response-headers response) 'cache-control))))
|
||
|
||
(test-equal "/log/NAME not found"
|
||
404
|
||
(let ((uri (publish-uri "/log/does-not-exist")))
|
||
(response-code (http-get uri))))
|
||
|
||
(test-equal "/signing-key.pub"
|
||
200
|
||
(response-code (http-get (publish-uri "/signing-key.pub"))))
|
||
|
||
(test-equal "non-GET query"
|
||
'(200 404)
|
||
(let ((path (string-append "/" (store-path-hash-part %item)
|
||
".narinfo")))
|
||
(map response-code
|
||
(list (http-get (publish-uri path))
|
||
(http-post (publish-uri path))))))
|
||
|
||
(test-end "publish")
|