server_test.go 35 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256
  1. // Copyright 2017 The go-ethereum Authors
  2. // This file is part of the go-ethereum library.
  3. //
  4. // The go-ethereum library is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Lesser General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // The go-ethereum library is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Lesser General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Lesser General Public License
  15. // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
  16. package http
  17. import (
  18. "archive/tar"
  19. "bytes"
  20. "context"
  21. "crypto/rand"
  22. "encoding/json"
  23. "errors"
  24. "flag"
  25. "fmt"
  26. "io"
  27. "io/ioutil"
  28. "math/big"
  29. "mime/multipart"
  30. "net/http"
  31. "net/url"
  32. "os"
  33. "path"
  34. "strconv"
  35. "strings"
  36. "testing"
  37. "time"
  38. "github.com/ethereum/go-ethereum/swarm/storage/feeds/lookup"
  39. "github.com/ethereum/go-ethereum/common"
  40. "github.com/ethereum/go-ethereum/core/types"
  41. "github.com/ethereum/go-ethereum/crypto"
  42. "github.com/ethereum/go-ethereum/log"
  43. "github.com/ethereum/go-ethereum/swarm/api"
  44. swarm "github.com/ethereum/go-ethereum/swarm/api/client"
  45. "github.com/ethereum/go-ethereum/swarm/multihash"
  46. "github.com/ethereum/go-ethereum/swarm/storage"
  47. "github.com/ethereum/go-ethereum/swarm/storage/feeds"
  48. "github.com/ethereum/go-ethereum/swarm/testutil"
  49. )
  50. func init() {
  51. loglevel := flag.Int("loglevel", 2, "loglevel")
  52. flag.Parse()
  53. log.Root().SetHandler(log.CallerFileHandler(log.LvlFilterHandler(log.Lvl(*loglevel), log.StreamHandler(os.Stderr, log.TerminalFormat(true)))))
  54. }
  55. func serverFunc(api *api.API) testutil.TestServer {
  56. return NewServer(api, "")
  57. }
  58. func newTestSigner() (*feeds.GenericSigner, error) {
  59. privKey, err := crypto.HexToECDSA("deadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeef")
  60. if err != nil {
  61. return nil, err
  62. }
  63. return feeds.NewGenericSigner(privKey), nil
  64. }
  65. // test the transparent resolving of multihash-containing feed updates with bzz:// scheme
  66. //
  67. // first upload data, and store the multihash to the resulting manifest in a feed update
  68. // retrieving the update with the multihash should return the manifest pointing directly to the data
  69. // and raw retrieve of that hash should return the data
  70. func TestBzzFeedMultihash(t *testing.T) {
  71. signer, _ := newTestSigner()
  72. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  73. defer srv.Close()
  74. // add the data our multihash aliased manifest will point to
  75. databytes := "bar"
  76. testBzzUrl := fmt.Sprintf("%s/bzz:/", srv.URL)
  77. resp, err := http.Post(testBzzUrl, "text/plain", bytes.NewReader([]byte(databytes)))
  78. if err != nil {
  79. t.Fatal(err)
  80. }
  81. defer resp.Body.Close()
  82. if resp.StatusCode != http.StatusOK {
  83. t.Fatalf("err %s", resp.Status)
  84. }
  85. b, err := ioutil.ReadAll(resp.Body)
  86. if err != nil {
  87. t.Fatal(err)
  88. }
  89. s := common.FromHex(string(b))
  90. mh := multihash.ToMultihash(s)
  91. log.Info("added data", "manifest", string(b), "data", common.ToHex(mh))
  92. topic, _ := feeds.NewTopic("foo.eth", nil)
  93. updateRequest := feeds.NewFirstRequest(topic)
  94. updateRequest.SetData(mh)
  95. if err := updateRequest.Sign(signer); err != nil {
  96. t.Fatal(err)
  97. }
  98. log.Info("added data", "manifest", string(b), "data", common.ToHex(mh))
  99. testUrl, err := url.Parse(fmt.Sprintf("%s/bzz-feed:/", srv.URL))
  100. if err != nil {
  101. t.Fatal(err)
  102. }
  103. query := testUrl.Query()
  104. body := updateRequest.AppendValues(query) // this adds all query parameters and returns the data to be posted
  105. query.Set("manifest", "1") // indicate we want a manifest back
  106. testUrl.RawQuery = query.Encode()
  107. // create the multihash update
  108. resp, err = http.Post(testUrl.String(), "application/octet-stream", bytes.NewReader(body))
  109. if err != nil {
  110. t.Fatal(err)
  111. }
  112. defer resp.Body.Close()
  113. if resp.StatusCode != http.StatusOK {
  114. t.Fatalf("err %s", resp.Status)
  115. }
  116. b, err = ioutil.ReadAll(resp.Body)
  117. if err != nil {
  118. t.Fatal(err)
  119. }
  120. rsrcResp := &storage.Address{}
  121. err = json.Unmarshal(b, rsrcResp)
  122. if err != nil {
  123. t.Fatalf("data %s could not be unmarshaled: %v", b, err)
  124. }
  125. correctManifestAddrHex := "bb056a5264c295c2b0f613c8409b9c87ce9d71576ace02458160df4cc894210b"
  126. if rsrcResp.Hex() != correctManifestAddrHex {
  127. t.Fatalf("Response feed manifest address mismatch, expected '%s', got '%s'", correctManifestAddrHex, rsrcResp.Hex())
  128. }
  129. // get bzz manifest transparent feed update resolve
  130. testBzzUrl = fmt.Sprintf("%s/bzz:/%s", srv.URL, rsrcResp)
  131. resp, err = http.Get(testBzzUrl)
  132. if err != nil {
  133. t.Fatal(err)
  134. }
  135. defer resp.Body.Close()
  136. if resp.StatusCode != http.StatusOK {
  137. t.Fatalf("err %s", resp.Status)
  138. }
  139. b, err = ioutil.ReadAll(resp.Body)
  140. if err != nil {
  141. t.Fatal(err)
  142. }
  143. if !bytes.Equal(b, []byte(databytes)) {
  144. t.Fatalf("retrieved data mismatch, expected %x, got %x", databytes, b)
  145. }
  146. }
  147. // Test Swarm Feeds using the raw update methods
  148. func TestBzzFeed(t *testing.T) {
  149. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  150. signer, _ := newTestSigner()
  151. defer srv.Close()
  152. // data of update 1
  153. update1Data := make([]byte, 666)
  154. update1Timestamp := srv.CurrentTime
  155. _, err := rand.Read(update1Data)
  156. if err != nil {
  157. t.Fatal(err)
  158. }
  159. //data for update 2
  160. update2Data := []byte("foo")
  161. topic, _ := feeds.NewTopic("foo.eth", nil)
  162. updateRequest := feeds.NewFirstRequest(topic)
  163. if err != nil {
  164. t.Fatal(err)
  165. }
  166. updateRequest.SetData(update1Data)
  167. if err := updateRequest.Sign(signer); err != nil {
  168. t.Fatal(err)
  169. }
  170. // creates feed and sets update 1
  171. testUrl, err := url.Parse(fmt.Sprintf("%s/bzz-feed:/", srv.URL))
  172. if err != nil {
  173. t.Fatal(err)
  174. }
  175. urlQuery := testUrl.Query()
  176. body := updateRequest.AppendValues(urlQuery) // this adds all query parameters
  177. urlQuery.Set("manifest", "1") // indicate we want a manifest back
  178. testUrl.RawQuery = urlQuery.Encode()
  179. resp, err := http.Post(testUrl.String(), "application/octet-stream", bytes.NewReader(body))
  180. if err != nil {
  181. t.Fatal(err)
  182. }
  183. defer resp.Body.Close()
  184. if resp.StatusCode != http.StatusOK {
  185. t.Fatalf("err %s", resp.Status)
  186. }
  187. b, err := ioutil.ReadAll(resp.Body)
  188. if err != nil {
  189. t.Fatal(err)
  190. }
  191. rsrcResp := &storage.Address{}
  192. err = json.Unmarshal(b, rsrcResp)
  193. if err != nil {
  194. t.Fatalf("data %s could not be unmarshaled: %v", b, err)
  195. }
  196. correctManifestAddrHex := "bb056a5264c295c2b0f613c8409b9c87ce9d71576ace02458160df4cc894210b"
  197. if rsrcResp.Hex() != correctManifestAddrHex {
  198. t.Fatalf("Response feed manifest mismatch, expected '%s', got '%s'", correctManifestAddrHex, rsrcResp.Hex())
  199. }
  200. // get the manifest
  201. testRawUrl := fmt.Sprintf("%s/bzz-raw:/%s", srv.URL, rsrcResp)
  202. resp, err = http.Get(testRawUrl)
  203. if err != nil {
  204. t.Fatal(err)
  205. }
  206. defer resp.Body.Close()
  207. if resp.StatusCode != http.StatusOK {
  208. t.Fatalf("err %s", resp.Status)
  209. }
  210. b, err = ioutil.ReadAll(resp.Body)
  211. if err != nil {
  212. t.Fatal(err)
  213. }
  214. manifest := &api.Manifest{}
  215. err = json.Unmarshal(b, manifest)
  216. if err != nil {
  217. t.Fatal(err)
  218. }
  219. if len(manifest.Entries) != 1 {
  220. t.Fatalf("Manifest has %d entries", len(manifest.Entries))
  221. }
  222. correctFeedHex := "0x666f6f2e65746800000000000000000000000000000000000000000000000000c96aaa54e2d44c299564da76e1cd3184a2386b8d"
  223. if manifest.Entries[0].Feed.Hex() != correctFeedHex {
  224. t.Fatalf("Expected manifest Feed '%s', got '%s'", correctFeedHex, manifest.Entries[0].Feed.Hex())
  225. }
  226. // get bzz manifest transparent feed update resolve
  227. testBzzUrl := fmt.Sprintf("%s/bzz:/%s", srv.URL, rsrcResp)
  228. resp, err = http.Get(testBzzUrl)
  229. if err != nil {
  230. t.Fatal(err)
  231. }
  232. defer resp.Body.Close()
  233. if resp.StatusCode == http.StatusOK {
  234. t.Fatal("Expected error status since feed update does not contain multihash. Received 200 OK")
  235. }
  236. b, err = ioutil.ReadAll(resp.Body)
  237. if err != nil {
  238. t.Fatal(err)
  239. }
  240. // get non-existent name, should fail
  241. testBzzResUrl := fmt.Sprintf("%s/bzz-feed:/bar", srv.URL)
  242. resp, err = http.Get(testBzzResUrl)
  243. if err != nil {
  244. t.Fatal(err)
  245. }
  246. if resp.StatusCode != http.StatusNotFound {
  247. t.Fatalf("Expected get non-existent feed manifest to fail with StatusNotFound (404), got %d", resp.StatusCode)
  248. }
  249. resp.Body.Close()
  250. // get latest update through bzz-feed directly
  251. log.Info("get update latest = 1.1", "addr", correctManifestAddrHex)
  252. testBzzResUrl = fmt.Sprintf("%s/bzz-feed:/%s", srv.URL, correctManifestAddrHex)
  253. resp, err = http.Get(testBzzResUrl)
  254. if err != nil {
  255. t.Fatal(err)
  256. }
  257. defer resp.Body.Close()
  258. if resp.StatusCode != http.StatusOK {
  259. t.Fatalf("err %s", resp.Status)
  260. }
  261. b, err = ioutil.ReadAll(resp.Body)
  262. if err != nil {
  263. t.Fatal(err)
  264. }
  265. if !bytes.Equal(update1Data, b) {
  266. t.Fatalf("Expected body '%x', got '%x'", update1Data, b)
  267. }
  268. // update 2
  269. // Move the clock ahead 1 second
  270. srv.CurrentTime++
  271. log.Info("update 2")
  272. // 1.- get metadata about this Feed
  273. testBzzResUrl = fmt.Sprintf("%s/bzz-feed:/%s/", srv.URL, correctManifestAddrHex)
  274. resp, err = http.Get(testBzzResUrl + "?meta=1")
  275. if err != nil {
  276. t.Fatal(err)
  277. }
  278. defer resp.Body.Close()
  279. if resp.StatusCode != http.StatusOK {
  280. t.Fatalf("Get feed metadata returned %s", resp.Status)
  281. }
  282. b, err = ioutil.ReadAll(resp.Body)
  283. if err != nil {
  284. t.Fatal(err)
  285. }
  286. updateRequest = &feeds.Request{}
  287. if err = updateRequest.UnmarshalJSON(b); err != nil {
  288. t.Fatalf("Error decoding feed metadata: %s", err)
  289. }
  290. updateRequest.SetData(update2Data)
  291. if err = updateRequest.Sign(signer); err != nil {
  292. t.Fatal(err)
  293. }
  294. testUrl, err = url.Parse(fmt.Sprintf("%s/bzz-feed:/", srv.URL))
  295. if err != nil {
  296. t.Fatal(err)
  297. }
  298. urlQuery = testUrl.Query()
  299. body = updateRequest.AppendValues(urlQuery) // this adds all query parameters
  300. testUrl.RawQuery = urlQuery.Encode()
  301. resp, err = http.Post(testUrl.String(), "application/octet-stream", bytes.NewReader(body))
  302. if err != nil {
  303. t.Fatal(err)
  304. }
  305. defer resp.Body.Close()
  306. if resp.StatusCode != http.StatusOK {
  307. t.Fatalf("Update returned %s", resp.Status)
  308. }
  309. // get latest update through bzz-feed directly
  310. log.Info("get update 1.2")
  311. testBzzResUrl = fmt.Sprintf("%s/bzz-feed:/%s", srv.URL, correctManifestAddrHex)
  312. resp, err = http.Get(testBzzResUrl)
  313. if err != nil {
  314. t.Fatal(err)
  315. }
  316. defer resp.Body.Close()
  317. if resp.StatusCode != http.StatusOK {
  318. t.Fatalf("err %s", resp.Status)
  319. }
  320. b, err = ioutil.ReadAll(resp.Body)
  321. if err != nil {
  322. t.Fatal(err)
  323. }
  324. if !bytes.Equal(update2Data, b) {
  325. t.Fatalf("Expected body '%x', got '%x'", update2Data, b)
  326. }
  327. // test manifest-less queries
  328. log.Info("get first update in update1Timestamp via direct query")
  329. query := feeds.NewQuery(&updateRequest.Feed, update1Timestamp, lookup.NoClue)
  330. urlq, err := url.Parse(fmt.Sprintf("%s/bzz-feed:/", srv.URL))
  331. if err != nil {
  332. t.Fatal(err)
  333. }
  334. values := urlq.Query()
  335. query.AppendValues(values) // this adds feed query parameters
  336. urlq.RawQuery = values.Encode()
  337. resp, err = http.Get(urlq.String())
  338. if err != nil {
  339. t.Fatal(err)
  340. }
  341. defer resp.Body.Close()
  342. if resp.StatusCode != http.StatusOK {
  343. t.Fatalf("err %s", resp.Status)
  344. }
  345. b, err = ioutil.ReadAll(resp.Body)
  346. if err != nil {
  347. t.Fatal(err)
  348. }
  349. if !bytes.Equal(update1Data, b) {
  350. t.Fatalf("Expected body '%x', got '%x'", update1Data, b)
  351. }
  352. }
  353. func TestBzzGetPath(t *testing.T) {
  354. testBzzGetPath(false, t)
  355. testBzzGetPath(true, t)
  356. }
  357. func testBzzGetPath(encrypted bool, t *testing.T) {
  358. var err error
  359. testmanifest := []string{
  360. `{"entries":[{"path":"b","hash":"011b4d03dd8c01f1049143cf9c4c817e4b167f1d1b83e5c6f0f10d89ba1e7bce","contentType":"","status":0},{"path":"c","hash":"011b4d03dd8c01f1049143cf9c4c817e4b167f1d1b83e5c6f0f10d89ba1e7bce","contentType":"","status":0}]}`,
  361. `{"entries":[{"path":"a","hash":"011b4d03dd8c01f1049143cf9c4c817e4b167f1d1b83e5c6f0f10d89ba1e7bce","contentType":"","status":0},{"path":"b/","hash":"<key0>","contentType":"application/bzz-manifest+json","status":0}]}`,
  362. `{"entries":[{"path":"a/","hash":"<key1>","contentType":"application/bzz-manifest+json","status":0}]}`,
  363. }
  364. testrequests := make(map[string]int)
  365. testrequests["/"] = 2
  366. testrequests["/a/"] = 1
  367. testrequests["/a/b/"] = 0
  368. testrequests["/x"] = 0
  369. testrequests[""] = 0
  370. expectedfailrequests := []string{"", "/x"}
  371. reader := [3]*bytes.Reader{}
  372. addr := [3]storage.Address{}
  373. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  374. defer srv.Close()
  375. for i, mf := range testmanifest {
  376. reader[i] = bytes.NewReader([]byte(mf))
  377. var wait func(context.Context) error
  378. ctx := context.TODO()
  379. addr[i], wait, err = srv.FileStore.Store(ctx, reader[i], int64(len(mf)), encrypted)
  380. if err != nil {
  381. t.Fatal(err)
  382. }
  383. for j := i + 1; j < len(testmanifest); j++ {
  384. testmanifest[j] = strings.Replace(testmanifest[j], fmt.Sprintf("<key%v>", i), addr[i].Hex(), -1)
  385. }
  386. err = wait(ctx)
  387. if err != nil {
  388. t.Fatal(err)
  389. }
  390. }
  391. rootRef := addr[2].Hex()
  392. _, err = http.Get(srv.URL + "/bzz-raw:/" + rootRef + "/a")
  393. if err != nil {
  394. t.Fatalf("Failed to connect to proxy: %v", err)
  395. }
  396. for k, v := range testrequests {
  397. var resp *http.Response
  398. var respbody []byte
  399. url := srv.URL + "/bzz-raw:/"
  400. if k != "" {
  401. url += rootRef + "/" + k[1:] + "?content_type=text/plain"
  402. }
  403. resp, err = http.Get(url)
  404. if err != nil {
  405. t.Fatalf("Request failed: %v", err)
  406. }
  407. defer resp.Body.Close()
  408. respbody, err = ioutil.ReadAll(resp.Body)
  409. if string(respbody) != testmanifest[v] {
  410. isexpectedfailrequest := false
  411. for _, r := range expectedfailrequests {
  412. if k == r {
  413. isexpectedfailrequest = true
  414. }
  415. }
  416. if !isexpectedfailrequest {
  417. t.Fatalf("Response body does not match, expected: %v, got %v", testmanifest[v], string(respbody))
  418. }
  419. }
  420. }
  421. for k, v := range testrequests {
  422. var resp *http.Response
  423. var respbody []byte
  424. url := srv.URL + "/bzz-hash:/"
  425. if k != "" {
  426. url += rootRef + "/" + k[1:]
  427. }
  428. resp, err = http.Get(url)
  429. if err != nil {
  430. t.Fatalf("Request failed: %v", err)
  431. }
  432. defer resp.Body.Close()
  433. respbody, err = ioutil.ReadAll(resp.Body)
  434. if err != nil {
  435. t.Fatalf("Read request body: %v", err)
  436. }
  437. if string(respbody) != addr[v].Hex() {
  438. isexpectedfailrequest := false
  439. for _, r := range expectedfailrequests {
  440. if k == r {
  441. isexpectedfailrequest = true
  442. }
  443. }
  444. if !isexpectedfailrequest {
  445. t.Fatalf("Response body does not match, expected: %v, got %v", addr[v], string(respbody))
  446. }
  447. }
  448. }
  449. ref := addr[2].Hex()
  450. for _, c := range []struct {
  451. path string
  452. json string
  453. pageFragments []string
  454. }{
  455. {
  456. path: "/",
  457. json: `{"common_prefixes":["a/"]}`,
  458. pageFragments: []string{
  459. fmt.Sprintf("Swarm index of bzz:/%s/", ref),
  460. `<a class="normal-link" href="a/">a/</a>`,
  461. },
  462. },
  463. {
  464. path: "/a/",
  465. json: `{"common_prefixes":["a/b/"],"entries":[{"hash":"011b4d03dd8c01f1049143cf9c4c817e4b167f1d1b83e5c6f0f10d89ba1e7bce","path":"a/a","mod_time":"0001-01-01T00:00:00Z"}]}`,
  466. pageFragments: []string{
  467. fmt.Sprintf("Swarm index of bzz:/%s/a/", ref),
  468. `<a class="normal-link" href="b/">b/</a>`,
  469. fmt.Sprintf(`<a class="normal-link" href="/bzz:/%s/a/a">a</a>`, ref),
  470. },
  471. },
  472. {
  473. path: "/a/b/",
  474. json: `{"entries":[{"hash":"011b4d03dd8c01f1049143cf9c4c817e4b167f1d1b83e5c6f0f10d89ba1e7bce","path":"a/b/b","mod_time":"0001-01-01T00:00:00Z"},{"hash":"011b4d03dd8c01f1049143cf9c4c817e4b167f1d1b83e5c6f0f10d89ba1e7bce","path":"a/b/c","mod_time":"0001-01-01T00:00:00Z"}]}`,
  475. pageFragments: []string{
  476. fmt.Sprintf("Swarm index of bzz:/%s/a/b/", ref),
  477. fmt.Sprintf(`<a class="normal-link" href="/bzz:/%s/a/b/b">b</a>`, ref),
  478. fmt.Sprintf(`<a class="normal-link" href="/bzz:/%s/a/b/c">c</a>`, ref),
  479. },
  480. },
  481. {
  482. path: "/x",
  483. },
  484. {
  485. path: "",
  486. },
  487. } {
  488. k := c.path
  489. url := srv.URL + "/bzz-list:/"
  490. if k != "" {
  491. url += rootRef + "/" + k[1:]
  492. }
  493. t.Run("json list "+c.path, func(t *testing.T) {
  494. resp, err := http.Get(url)
  495. if err != nil {
  496. t.Fatalf("HTTP request: %v", err)
  497. }
  498. defer resp.Body.Close()
  499. respbody, err := ioutil.ReadAll(resp.Body)
  500. if err != nil {
  501. t.Fatalf("Read response body: %v", err)
  502. }
  503. body := strings.TrimSpace(string(respbody))
  504. if body != c.json {
  505. isexpectedfailrequest := false
  506. for _, r := range expectedfailrequests {
  507. if k == r {
  508. isexpectedfailrequest = true
  509. }
  510. }
  511. if !isexpectedfailrequest {
  512. t.Errorf("Response list body %q does not match, expected: %v, got %v", k, c.json, body)
  513. }
  514. }
  515. })
  516. t.Run("html list "+c.path, func(t *testing.T) {
  517. req, err := http.NewRequest(http.MethodGet, url, nil)
  518. if err != nil {
  519. t.Fatalf("New request: %v", err)
  520. }
  521. req.Header.Set("Accept", "text/html")
  522. resp, err := http.DefaultClient.Do(req)
  523. if err != nil {
  524. t.Fatalf("HTTP request: %v", err)
  525. }
  526. defer resp.Body.Close()
  527. b, err := ioutil.ReadAll(resp.Body)
  528. if err != nil {
  529. t.Fatalf("Read response body: %v", err)
  530. }
  531. body := string(b)
  532. for _, f := range c.pageFragments {
  533. if !strings.Contains(body, f) {
  534. isexpectedfailrequest := false
  535. for _, r := range expectedfailrequests {
  536. if k == r {
  537. isexpectedfailrequest = true
  538. }
  539. }
  540. if !isexpectedfailrequest {
  541. t.Errorf("Response list body %q does not contain %q: body %q", k, f, body)
  542. }
  543. }
  544. }
  545. })
  546. }
  547. nonhashtests := []string{
  548. srv.URL + "/bzz:/name",
  549. srv.URL + "/bzz-immutable:/nonhash",
  550. srv.URL + "/bzz-raw:/nonhash",
  551. srv.URL + "/bzz-list:/nonhash",
  552. srv.URL + "/bzz-hash:/nonhash",
  553. }
  554. nonhashresponses := []string{
  555. `cannot resolve name: no DNS to resolve name: "name"`,
  556. `cannot resolve nonhash: no DNS to resolve name: "nonhash"`,
  557. `cannot resolve nonhash: no DNS to resolve name: "nonhash"`,
  558. `cannot resolve nonhash: no DNS to resolve name: "nonhash"`,
  559. `cannot resolve nonhash: no DNS to resolve name: "nonhash"`,
  560. }
  561. for i, url := range nonhashtests {
  562. var resp *http.Response
  563. var respbody []byte
  564. resp, err = http.Get(url)
  565. if err != nil {
  566. t.Fatalf("Request failed: %v", err)
  567. }
  568. defer resp.Body.Close()
  569. respbody, err = ioutil.ReadAll(resp.Body)
  570. if err != nil {
  571. t.Fatalf("ReadAll failed: %v", err)
  572. }
  573. if !strings.Contains(string(respbody), nonhashresponses[i]) {
  574. t.Fatalf("Non-Hash response body does not match, expected: %v, got: %v", nonhashresponses[i], string(respbody))
  575. }
  576. }
  577. }
  578. func TestBzzTar(t *testing.T) {
  579. testBzzTar(false, t)
  580. testBzzTar(true, t)
  581. }
  582. func testBzzTar(encrypted bool, t *testing.T) {
  583. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  584. defer srv.Close()
  585. fileNames := []string{"tmp1.txt", "tmp2.lock", "tmp3.rtf"}
  586. fileContents := []string{"tmp1textfilevalue", "tmp2lockfilelocked", "tmp3isjustaplaintextfile"}
  587. buf := &bytes.Buffer{}
  588. tw := tar.NewWriter(buf)
  589. defer tw.Close()
  590. for i, v := range fileNames {
  591. size := int64(len(fileContents[i]))
  592. hdr := &tar.Header{
  593. Name: v,
  594. Mode: 0644,
  595. Size: size,
  596. ModTime: time.Now(),
  597. Xattrs: map[string]string{
  598. "user.swarm.content-type": "text/plain",
  599. },
  600. }
  601. if err := tw.WriteHeader(hdr); err != nil {
  602. t.Fatal(err)
  603. }
  604. // copy the file into the tar stream
  605. n, err := io.Copy(tw, bytes.NewBufferString(fileContents[i]))
  606. if err != nil {
  607. t.Fatal(err)
  608. } else if n != size {
  609. t.Fatal("size mismatch")
  610. }
  611. }
  612. //post tar stream
  613. url := srv.URL + "/bzz:/"
  614. if encrypted {
  615. url = url + "encrypt"
  616. }
  617. req, err := http.NewRequest("POST", url, buf)
  618. if err != nil {
  619. t.Fatal(err)
  620. }
  621. req.Header.Add("Content-Type", "application/x-tar")
  622. client := &http.Client{}
  623. resp2, err := client.Do(req)
  624. if err != nil {
  625. t.Fatal(err)
  626. }
  627. if resp2.StatusCode != http.StatusOK {
  628. t.Fatalf("err %s", resp2.Status)
  629. }
  630. swarmHash, err := ioutil.ReadAll(resp2.Body)
  631. resp2.Body.Close()
  632. if err != nil {
  633. t.Fatal(err)
  634. }
  635. // now do a GET to get a tarball back
  636. req, err = http.NewRequest("GET", fmt.Sprintf(srv.URL+"/bzz:/%s", string(swarmHash)), nil)
  637. if err != nil {
  638. t.Fatal(err)
  639. }
  640. req.Header.Add("Accept", "application/x-tar")
  641. resp2, err = client.Do(req)
  642. if err != nil {
  643. t.Fatal(err)
  644. }
  645. defer resp2.Body.Close()
  646. if h := resp2.Header.Get("Content-Type"); h != "application/x-tar" {
  647. t.Fatalf("Content-Type header expected: application/x-tar, got: %s", h)
  648. }
  649. expectedFileName := string(swarmHash) + ".tar"
  650. expectedContentDisposition := fmt.Sprintf("inline; filename=\"%s\"", expectedFileName)
  651. if h := resp2.Header.Get("Content-Disposition"); h != expectedContentDisposition {
  652. t.Fatalf("Content-Disposition header expected: %s, got: %s", expectedContentDisposition, h)
  653. }
  654. file, err := ioutil.TempFile("", "swarm-downloaded-tarball")
  655. if err != nil {
  656. t.Fatal(err)
  657. }
  658. defer os.Remove(file.Name())
  659. _, err = io.Copy(file, resp2.Body)
  660. if err != nil {
  661. t.Fatalf("error getting tarball: %v", err)
  662. }
  663. file.Sync()
  664. file.Close()
  665. tarFileHandle, err := os.Open(file.Name())
  666. if err != nil {
  667. t.Fatal(err)
  668. }
  669. tr := tar.NewReader(tarFileHandle)
  670. for {
  671. hdr, err := tr.Next()
  672. if err == io.EOF {
  673. break
  674. } else if err != nil {
  675. t.Fatalf("error reading tar stream: %s", err)
  676. }
  677. bb := make([]byte, hdr.Size)
  678. _, err = tr.Read(bb)
  679. if err != nil && err != io.EOF {
  680. t.Fatal(err)
  681. }
  682. passed := false
  683. for i, v := range fileNames {
  684. if v == hdr.Name {
  685. if string(bb) == fileContents[i] {
  686. passed = true
  687. break
  688. }
  689. }
  690. }
  691. if !passed {
  692. t.Fatalf("file %s did not pass content assertion", hdr.Name)
  693. }
  694. }
  695. }
  696. // TestBzzRootRedirect tests that getting the root path of a manifest without
  697. // a trailing slash gets redirected to include the trailing slash so that
  698. // relative URLs work as expected.
  699. func TestBzzRootRedirect(t *testing.T) {
  700. testBzzRootRedirect(false, t)
  701. }
  702. func TestBzzRootRedirectEncrypted(t *testing.T) {
  703. testBzzRootRedirect(true, t)
  704. }
  705. func testBzzRootRedirect(toEncrypt bool, t *testing.T) {
  706. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  707. defer srv.Close()
  708. // create a manifest with some data at the root path
  709. client := swarm.NewClient(srv.URL)
  710. data := []byte("data")
  711. file := &swarm.File{
  712. ReadCloser: ioutil.NopCloser(bytes.NewReader(data)),
  713. ManifestEntry: api.ManifestEntry{
  714. Path: "",
  715. ContentType: "text/plain",
  716. Size: int64(len(data)),
  717. },
  718. }
  719. hash, err := client.Upload(file, "", toEncrypt)
  720. if err != nil {
  721. t.Fatal(err)
  722. }
  723. // define a CheckRedirect hook which ensures there is only a single
  724. // redirect to the correct URL
  725. redirected := false
  726. httpClient := http.Client{
  727. CheckRedirect: func(req *http.Request, via []*http.Request) error {
  728. if redirected {
  729. return errors.New("too many redirects")
  730. }
  731. redirected = true
  732. expectedPath := "/bzz:/" + hash + "/"
  733. if req.URL.Path != expectedPath {
  734. return fmt.Errorf("expected redirect to %q, got %q", expectedPath, req.URL.Path)
  735. }
  736. return nil
  737. },
  738. }
  739. // perform the GET request and assert the response
  740. res, err := httpClient.Get(srv.URL + "/bzz:/" + hash)
  741. if err != nil {
  742. t.Fatal(err)
  743. }
  744. defer res.Body.Close()
  745. if !redirected {
  746. t.Fatal("expected GET /bzz:/<hash> to redirect to /bzz:/<hash>/ but it didn't")
  747. }
  748. gotData, err := ioutil.ReadAll(res.Body)
  749. if err != nil {
  750. t.Fatal(err)
  751. }
  752. if !bytes.Equal(gotData, data) {
  753. t.Fatalf("expected response to equal %q, got %q", data, gotData)
  754. }
  755. }
  756. func TestMethodsNotAllowed(t *testing.T) {
  757. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  758. defer srv.Close()
  759. databytes := "bar"
  760. for _, c := range []struct {
  761. url string
  762. code int
  763. }{
  764. {
  765. url: fmt.Sprintf("%s/bzz-list:/", srv.URL),
  766. code: http.StatusMethodNotAllowed,
  767. }, {
  768. url: fmt.Sprintf("%s/bzz-hash:/", srv.URL),
  769. code: http.StatusMethodNotAllowed,
  770. },
  771. {
  772. url: fmt.Sprintf("%s/bzz-immutable:/", srv.URL),
  773. code: http.StatusMethodNotAllowed,
  774. },
  775. } {
  776. res, _ := http.Post(c.url, "text/plain", bytes.NewReader([]byte(databytes)))
  777. if res.StatusCode != c.code {
  778. t.Fatalf("should have failed. requested url: %s, expected code %d, got %d", c.url, c.code, res.StatusCode)
  779. }
  780. }
  781. }
  782. func httpDo(httpMethod string, url string, reqBody io.Reader, headers map[string]string, verbose bool, t *testing.T) (*http.Response, string) {
  783. // Build the Request
  784. req, err := http.NewRequest(httpMethod, url, reqBody)
  785. if err != nil {
  786. t.Fatal(err)
  787. }
  788. for key, value := range headers {
  789. req.Header.Set(key, value)
  790. }
  791. if verbose {
  792. t.Log(req.Method, req.URL, req.Header, req.Body)
  793. }
  794. // Send Request out
  795. httpClient := &http.Client{}
  796. res, err := httpClient.Do(req)
  797. if err != nil {
  798. t.Fatal(err)
  799. }
  800. // Read the HTTP Body
  801. buffer, err := ioutil.ReadAll(res.Body)
  802. if err != nil {
  803. t.Fatal(err)
  804. }
  805. defer res.Body.Close()
  806. body := string(buffer)
  807. return res, body
  808. }
  809. func TestGet(t *testing.T) {
  810. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  811. defer srv.Close()
  812. for _, testCase := range []struct {
  813. uri string
  814. method string
  815. headers map[string]string
  816. expectedStatusCode int
  817. assertResponseBody string
  818. verbose bool
  819. }{
  820. {
  821. uri: fmt.Sprintf("%s/", srv.URL),
  822. method: "GET",
  823. headers: map[string]string{"Accept": "text/html"},
  824. expectedStatusCode: http.StatusOK,
  825. assertResponseBody: "Swarm: Serverless Hosting Incentivised Peer-To-Peer Storage And Content Distribution",
  826. verbose: false,
  827. },
  828. {
  829. uri: fmt.Sprintf("%s/", srv.URL),
  830. method: "GET",
  831. headers: map[string]string{"Accept": "application/json"},
  832. expectedStatusCode: http.StatusOK,
  833. assertResponseBody: "Swarm: Please request a valid ENS or swarm hash with the appropriate bzz scheme",
  834. verbose: false,
  835. },
  836. {
  837. uri: fmt.Sprintf("%s/robots.txt", srv.URL),
  838. method: "GET",
  839. headers: map[string]string{"Accept": "text/html"},
  840. expectedStatusCode: http.StatusOK,
  841. assertResponseBody: "User-agent: *\nDisallow: /",
  842. verbose: false,
  843. },
  844. {
  845. uri: fmt.Sprintf("%s/nonexistent_path", srv.URL),
  846. method: "GET",
  847. headers: map[string]string{},
  848. expectedStatusCode: http.StatusNotFound,
  849. verbose: false,
  850. },
  851. {
  852. uri: fmt.Sprintf("%s/bzz:asdf/", srv.URL),
  853. method: "GET",
  854. headers: map[string]string{},
  855. expectedStatusCode: http.StatusNotFound,
  856. verbose: false,
  857. },
  858. {
  859. uri: fmt.Sprintf("%s/tbz2/", srv.URL),
  860. method: "GET",
  861. headers: map[string]string{},
  862. expectedStatusCode: http.StatusNotFound,
  863. verbose: false,
  864. },
  865. {
  866. uri: fmt.Sprintf("%s/bzz-rack:/", srv.URL),
  867. method: "GET",
  868. headers: map[string]string{},
  869. expectedStatusCode: http.StatusNotFound,
  870. verbose: false,
  871. },
  872. {
  873. uri: fmt.Sprintf("%s/bzz-ls", srv.URL),
  874. method: "GET",
  875. headers: map[string]string{},
  876. expectedStatusCode: http.StatusNotFound,
  877. verbose: false,
  878. }} {
  879. t.Run("GET "+testCase.uri, func(t *testing.T) {
  880. res, body := httpDo(testCase.method, testCase.uri, nil, testCase.headers, testCase.verbose, t)
  881. if res.StatusCode != testCase.expectedStatusCode {
  882. t.Fatalf("expected status code %d but got %d", testCase.expectedStatusCode, res.StatusCode)
  883. }
  884. if testCase.assertResponseBody != "" && !strings.Contains(body, testCase.assertResponseBody) {
  885. t.Fatalf("expected response to be: %s but got: %s", testCase.assertResponseBody, body)
  886. }
  887. })
  888. }
  889. }
  890. func TestModify(t *testing.T) {
  891. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  892. defer srv.Close()
  893. swarmClient := swarm.NewClient(srv.URL)
  894. data := []byte("data")
  895. file := &swarm.File{
  896. ReadCloser: ioutil.NopCloser(bytes.NewReader(data)),
  897. ManifestEntry: api.ManifestEntry{
  898. Path: "",
  899. ContentType: "text/plain",
  900. Size: int64(len(data)),
  901. },
  902. }
  903. hash, err := swarmClient.Upload(file, "", false)
  904. if err != nil {
  905. t.Fatal(err)
  906. }
  907. for _, testCase := range []struct {
  908. uri string
  909. method string
  910. headers map[string]string
  911. requestBody []byte
  912. expectedStatusCode int
  913. assertResponseBody string
  914. assertResponseHeaders map[string]string
  915. verbose bool
  916. }{
  917. {
  918. uri: fmt.Sprintf("%s/bzz:/%s", srv.URL, hash),
  919. method: "DELETE",
  920. headers: map[string]string{},
  921. expectedStatusCode: http.StatusOK,
  922. assertResponseBody: "8b634aea26eec353ac0ecbec20c94f44d6f8d11f38d4578a4c207a84c74ef731",
  923. verbose: false,
  924. },
  925. {
  926. uri: fmt.Sprintf("%s/bzz:/%s", srv.URL, hash),
  927. method: "PUT",
  928. headers: map[string]string{},
  929. expectedStatusCode: http.StatusMethodNotAllowed,
  930. verbose: false,
  931. },
  932. {
  933. uri: fmt.Sprintf("%s/bzz-raw:/%s", srv.URL, hash),
  934. method: "PUT",
  935. headers: map[string]string{},
  936. expectedStatusCode: http.StatusMethodNotAllowed,
  937. verbose: false,
  938. },
  939. {
  940. uri: fmt.Sprintf("%s/bzz:/%s", srv.URL, hash),
  941. method: "PATCH",
  942. headers: map[string]string{},
  943. expectedStatusCode: http.StatusMethodNotAllowed,
  944. verbose: false,
  945. },
  946. {
  947. uri: fmt.Sprintf("%s/bzz-raw:/", srv.URL),
  948. method: "POST",
  949. headers: map[string]string{},
  950. requestBody: []byte("POSTdata"),
  951. expectedStatusCode: http.StatusOK,
  952. assertResponseHeaders: map[string]string{"Content-Length": "64"},
  953. verbose: false,
  954. },
  955. {
  956. uri: fmt.Sprintf("%s/bzz-raw:/encrypt", srv.URL),
  957. method: "POST",
  958. headers: map[string]string{},
  959. requestBody: []byte("POSTdata"),
  960. expectedStatusCode: http.StatusOK,
  961. assertResponseHeaders: map[string]string{"Content-Length": "128"},
  962. verbose: false,
  963. },
  964. } {
  965. t.Run(testCase.method+" "+testCase.uri, func(t *testing.T) {
  966. reqBody := bytes.NewReader(testCase.requestBody)
  967. res, body := httpDo(testCase.method, testCase.uri, reqBody, testCase.headers, testCase.verbose, t)
  968. if res.StatusCode != testCase.expectedStatusCode {
  969. t.Fatalf("expected status code %d but got %d, %s", testCase.expectedStatusCode, res.StatusCode, body)
  970. }
  971. if testCase.assertResponseBody != "" && !strings.Contains(body, testCase.assertResponseBody) {
  972. t.Log(body)
  973. t.Fatalf("expected response %s but got %s", testCase.assertResponseBody, body)
  974. }
  975. for key, value := range testCase.assertResponseHeaders {
  976. if res.Header.Get(key) != value {
  977. t.Logf("expected %s=%s in HTTP response header but got %s", key, value, res.Header.Get(key))
  978. }
  979. }
  980. })
  981. }
  982. }
  983. func TestMultiPartUpload(t *testing.T) {
  984. // POST /bzz:/ Content-Type: multipart/form-data
  985. verbose := false
  986. // Setup Swarm
  987. srv := testutil.NewTestSwarmServer(t, serverFunc, nil)
  988. defer srv.Close()
  989. url := fmt.Sprintf("%s/bzz:/", srv.URL)
  990. buf := new(bytes.Buffer)
  991. form := multipart.NewWriter(buf)
  992. form.WriteField("name", "John Doe")
  993. file1, _ := form.CreateFormFile("cv", "cv.txt")
  994. file1.Write([]byte("John Doe's Credentials"))
  995. file2, _ := form.CreateFormFile("profile_picture", "profile.jpg")
  996. file2.Write([]byte("imaginethisisjpegdata"))
  997. form.Close()
  998. headers := map[string]string{
  999. "Content-Type": form.FormDataContentType(),
  1000. "Content-Length": strconv.Itoa(buf.Len()),
  1001. }
  1002. res, body := httpDo("POST", url, buf, headers, verbose, t)
  1003. if res.StatusCode != http.StatusOK {
  1004. t.Fatalf("expected POST multipart/form-data to return 200, but it returned %d", res.StatusCode)
  1005. }
  1006. if len(body) != 64 {
  1007. t.Fatalf("expected POST multipart/form-data to return a 64 char manifest but the answer was %d chars long", len(body))
  1008. }
  1009. }
  1010. // TestBzzGetFileWithResolver tests fetching a file using a mocked ENS resolver
  1011. func TestBzzGetFileWithResolver(t *testing.T) {
  1012. resolver := newTestResolveValidator("")
  1013. srv := testutil.NewTestSwarmServer(t, serverFunc, resolver)
  1014. defer srv.Close()
  1015. fileNames := []string{"dir1/tmp1.txt", "dir2/tmp2.lock", "dir3/tmp3.rtf"}
  1016. fileContents := []string{"tmp1textfilevalue", "tmp2lockfilelocked", "tmp3isjustaplaintextfile"}
  1017. buf := &bytes.Buffer{}
  1018. tw := tar.NewWriter(buf)
  1019. for i, v := range fileNames {
  1020. size := len(fileContents[i])
  1021. hdr := &tar.Header{
  1022. Name: v,
  1023. Mode: 0644,
  1024. Size: int64(size),
  1025. ModTime: time.Now(),
  1026. Xattrs: map[string]string{
  1027. "user.swarm.content-type": "text/plain",
  1028. },
  1029. }
  1030. if err := tw.WriteHeader(hdr); err != nil {
  1031. t.Fatal(err)
  1032. }
  1033. // copy the file into the tar stream
  1034. n, err := io.WriteString(tw, fileContents[i])
  1035. if err != nil {
  1036. t.Fatal(err)
  1037. } else if n != size {
  1038. t.Fatal("size mismatch")
  1039. }
  1040. }
  1041. if err := tw.Close(); err != nil {
  1042. t.Fatal(err)
  1043. }
  1044. //post tar stream
  1045. url := srv.URL + "/bzz:/"
  1046. req, err := http.NewRequest("POST", url, buf)
  1047. if err != nil {
  1048. t.Fatal(err)
  1049. }
  1050. req.Header.Add("Content-Type", "application/x-tar")
  1051. client := &http.Client{}
  1052. serverResponse, err := client.Do(req)
  1053. if err != nil {
  1054. t.Fatal(err)
  1055. }
  1056. if serverResponse.StatusCode != http.StatusOK {
  1057. t.Fatalf("err %s", serverResponse.Status)
  1058. }
  1059. swarmHash, err := ioutil.ReadAll(serverResponse.Body)
  1060. serverResponse.Body.Close()
  1061. if err != nil {
  1062. t.Fatal(err)
  1063. }
  1064. // set the resolved hash to be the swarm hash of what we've just uploaded
  1065. hash := common.HexToHash(string(swarmHash))
  1066. resolver.hash = &hash
  1067. for _, v := range []struct {
  1068. addr string
  1069. path string
  1070. expectedStatusCode int
  1071. expectedContentType string
  1072. expectedFileName string
  1073. }{
  1074. {
  1075. addr: string(swarmHash),
  1076. path: fileNames[0],
  1077. expectedStatusCode: http.StatusOK,
  1078. expectedContentType: "text/plain",
  1079. expectedFileName: path.Base(fileNames[0]),
  1080. },
  1081. {
  1082. addr: "somebogusensname",
  1083. path: fileNames[0],
  1084. expectedStatusCode: http.StatusOK,
  1085. expectedContentType: "text/plain",
  1086. expectedFileName: path.Base(fileNames[0]),
  1087. },
  1088. } {
  1089. req, err := http.NewRequest("GET", fmt.Sprintf(srv.URL+"/bzz:/%s/%s", v.addr, v.path), nil)
  1090. if err != nil {
  1091. t.Fatal(err)
  1092. }
  1093. serverResponse, err := client.Do(req)
  1094. if err != nil {
  1095. t.Fatal(err)
  1096. }
  1097. defer serverResponse.Body.Close()
  1098. if serverResponse.StatusCode != v.expectedStatusCode {
  1099. t.Fatalf("expected %d, got %d", v.expectedStatusCode, serverResponse.StatusCode)
  1100. }
  1101. if h := serverResponse.Header.Get("Content-Type"); h != v.expectedContentType {
  1102. t.Fatalf("Content-Type header expected: %s, got %s", v.expectedContentType, h)
  1103. }
  1104. expectedContentDisposition := fmt.Sprintf("inline; filename=\"%s\"", v.expectedFileName)
  1105. if h := serverResponse.Header.Get("Content-Disposition"); h != expectedContentDisposition {
  1106. t.Fatalf("Content-Disposition header expected: %s, got: %s", expectedContentDisposition, h)
  1107. }
  1108. }
  1109. }
  1110. // testResolver implements the Resolver interface and either returns the given
  1111. // hash if it is set, or returns a "name not found" error
  1112. type testResolveValidator struct {
  1113. hash *common.Hash
  1114. }
  1115. func newTestResolveValidator(addr string) *testResolveValidator {
  1116. r := &testResolveValidator{}
  1117. if addr != "" {
  1118. hash := common.HexToHash(addr)
  1119. r.hash = &hash
  1120. }
  1121. return r
  1122. }
  1123. func (t *testResolveValidator) Resolve(addr string) (common.Hash, error) {
  1124. if t.hash == nil {
  1125. return common.Hash{}, fmt.Errorf("DNS name not found: %q", addr)
  1126. }
  1127. return *t.hash, nil
  1128. }
  1129. func (t *testResolveValidator) Owner(node [32]byte) (addr common.Address, err error) {
  1130. return
  1131. }
  1132. func (t *testResolveValidator) HeaderByNumber(context.Context, *big.Int) (header *types.Header, err error) {
  1133. return
  1134. }