mirror of
https://github.com/rclone/rclone.git
synced 2024-11-30 04:23:48 +08:00
f8d56bebaf
Restructuring of config code in v1.55 resulted in config file being loaded early at process startup. If configuration file is encrypted this means user will need to supply the password, even when running commands that does not use config. This also lead to an issue where mount with --deamon failed to decrypt the config file when it had to prompt user for passord. Fixes #5236 Fixes #5228
206 lines
7.2 KiB
Go
206 lines
7.2 KiB
Go
package dlna
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"html"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/anacrolix/dms/soap"
|
|
|
|
"github.com/rclone/rclone/fs/config/configfile"
|
|
"github.com/rclone/rclone/vfs"
|
|
|
|
_ "github.com/rclone/rclone/backend/local"
|
|
"github.com/rclone/rclone/cmd/serve/dlna/dlnaflags"
|
|
"github.com/rclone/rclone/fs"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var (
|
|
dlnaServer *server
|
|
baseURL string
|
|
)
|
|
|
|
const (
|
|
testBindAddress = "localhost:0"
|
|
)
|
|
|
|
func startServer(t *testing.T, f fs.Fs) {
|
|
opt := dlnaflags.DefaultOpt
|
|
opt.ListenAddr = testBindAddress
|
|
dlnaServer = newServer(f, &opt)
|
|
assert.NoError(t, dlnaServer.Serve())
|
|
baseURL = "http://" + dlnaServer.HTTPConn.Addr().String()
|
|
}
|
|
|
|
func TestInit(t *testing.T) {
|
|
configfile.Install()
|
|
|
|
f, err := fs.NewFs(context.Background(), "testdata/files")
|
|
l, _ := f.List(context.Background(), "")
|
|
fmt.Println(l)
|
|
require.NoError(t, err)
|
|
|
|
startServer(t, f)
|
|
}
|
|
|
|
// Make sure that it serves rootDesc.xml (SCPD in uPnP parlance).
|
|
func TestRootSCPD(t *testing.T) {
|
|
req, err := http.NewRequest("GET", baseURL+rootDescPath, nil)
|
|
require.NoError(t, err)
|
|
resp, err := http.DefaultClient.Do(req)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
// Make sure that the SCPD contains a CDS service.
|
|
require.Contains(t, string(body),
|
|
"<serviceType>urn:schemas-upnp-org:service:ContentDirectory:1</serviceType>")
|
|
// Make sure that the SCPD contains a CM service.
|
|
require.Contains(t, string(body),
|
|
"<serviceType>urn:schemas-upnp-org:service:ConnectionManager:1</serviceType>")
|
|
// Ensure that the SCPD url is configured.
|
|
require.Regexp(t, "<SCPDURL>/.*</SCPDURL>", string(body))
|
|
}
|
|
|
|
// Make sure that it serves content from the remote.
|
|
func TestServeContent(t *testing.T) {
|
|
req, err := http.NewRequest("GET", baseURL+resPath+"video.mp4", nil)
|
|
require.NoError(t, err)
|
|
resp, err := http.DefaultClient.Do(req)
|
|
require.NoError(t, err)
|
|
defer fs.CheckClose(resp.Body, &err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
actualContents, err := ioutil.ReadAll(resp.Body)
|
|
assert.NoError(t, err)
|
|
|
|
// Now compare the contents with the golden file.
|
|
node, err := dlnaServer.vfs.Stat("/video.mp4")
|
|
assert.NoError(t, err)
|
|
goldenFile := node.(*vfs.File)
|
|
goldenReader, err := goldenFile.Open(os.O_RDONLY)
|
|
assert.NoError(t, err)
|
|
defer fs.CheckClose(goldenReader, &err)
|
|
goldenContents, err := ioutil.ReadAll(goldenReader)
|
|
assert.NoError(t, err)
|
|
|
|
require.Equal(t, goldenContents, actualContents)
|
|
}
|
|
|
|
// Check that ContentDirectory#Browse returns appropriate metadata on the root container.
|
|
func TestContentDirectoryBrowseMetadata(t *testing.T) {
|
|
// Sample from: https://github.com/rclone/rclone/issues/3253#issuecomment-524317469
|
|
req, err := http.NewRequest("POST", baseURL+serviceControlURL, strings.NewReader(`
|
|
<?xml version="1.0" encoding="utf-8"?>
|
|
<s:Envelope xmlns:s="http://schemas.xmlsoap.org/soap/envelope/"
|
|
s:encodingStyle="http://schemas.xmlsoap.org/soap/encoding/">
|
|
<s:Body>
|
|
<u:Browse xmlns:u="urn:schemas-upnp-org:service:ContentDirectory:1">
|
|
<ObjectID>0</ObjectID>
|
|
<BrowseFlag>BrowseMetadata</BrowseFlag>
|
|
<Filter>*</Filter>
|
|
<StartingIndex>0</StartingIndex>
|
|
<RequestedCount>0</RequestedCount>
|
|
<SortCriteria></SortCriteria>
|
|
</u:Browse>
|
|
</s:Body>
|
|
</s:Envelope>`))
|
|
require.NoError(t, err)
|
|
req.Header.Set("SOAPACTION", `"urn:schemas-upnp-org:service:ContentDirectory:1#Browse"`)
|
|
resp, err := http.DefaultClient.Do(req)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
// expect a <container> element
|
|
require.Contains(t, string(body), html.EscapeString("<container "))
|
|
require.NotContains(t, string(body), html.EscapeString("<item "))
|
|
// if there is a childCount, it better not be zero
|
|
require.NotContains(t, string(body), html.EscapeString(" childCount=\"0\""))
|
|
// should have a dc:date element
|
|
require.Contains(t, string(body), html.EscapeString("<dc:date>"))
|
|
}
|
|
|
|
// Check that the X_MS_MediaReceiverRegistrar is faked out properly.
|
|
func TestMediaReceiverRegistrarService(t *testing.T) {
|
|
env := soap.Envelope{
|
|
Body: soap.Body{
|
|
Action: []byte("RegisterDevice"),
|
|
},
|
|
}
|
|
req, err := http.NewRequest("POST", baseURL+serviceControlURL, bytes.NewReader(mustMarshalXML(env)))
|
|
require.NoError(t, err)
|
|
req.Header.Set("SOAPACTION", `"urn:microsoft.com:service:X_MS_MediaReceiverRegistrar:1#RegisterDevice"`)
|
|
resp, err := http.DefaultClient.Do(req)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
require.Contains(t, string(body), "<RegistrationRespMsg>")
|
|
}
|
|
|
|
// Check that ContentDirectory#Browse returns the expected items.
|
|
func TestContentDirectoryBrowseDirectChildren(t *testing.T) {
|
|
// First the root...
|
|
req, err := http.NewRequest("POST", baseURL+serviceControlURL, strings.NewReader(`
|
|
<?xml version="1.0" encoding="utf-8"?>
|
|
<s:Envelope xmlns:s="http://schemas.xmlsoap.org/soap/envelope/"
|
|
s:encodingStyle="http://schemas.xmlsoap.org/soap/encoding/">
|
|
<s:Body>
|
|
<u:Browse xmlns:u="urn:schemas-upnp-org:service:ContentDirectory:1">
|
|
<ObjectID>0</ObjectID>
|
|
<BrowseFlag>BrowseDirectChildren</BrowseFlag>
|
|
<Filter>*</Filter>
|
|
<StartingIndex>0</StartingIndex>
|
|
<RequestedCount>0</RequestedCount>
|
|
<SortCriteria></SortCriteria>
|
|
</u:Browse>
|
|
</s:Body>
|
|
</s:Envelope>`))
|
|
require.NoError(t, err)
|
|
req.Header.Set("SOAPACTION", `"urn:schemas-upnp-org:service:ContentDirectory:1#Browse"`)
|
|
resp, err := http.DefaultClient.Do(req)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
// expect video.mp4, video.srt, video.en.srt URLs to be in the DIDL
|
|
require.Contains(t, string(body), "/r/video.mp4")
|
|
require.Contains(t, string(body), "/r/video.srt")
|
|
require.Contains(t, string(body), "/r/video.en.srt")
|
|
|
|
// Then a subdirectory
|
|
req, err = http.NewRequest("POST", baseURL+serviceControlURL, strings.NewReader(`
|
|
<?xml version="1.0" encoding="utf-8"?>
|
|
<s:Envelope xmlns:s="http://schemas.xmlsoap.org/soap/envelope/"
|
|
s:encodingStyle="http://schemas.xmlsoap.org/soap/encoding/">
|
|
<s:Body>
|
|
<u:Browse xmlns:u="urn:schemas-upnp-org:service:ContentDirectory:1">
|
|
<ObjectID>%2Fsubdir</ObjectID>
|
|
<BrowseFlag>BrowseDirectChildren</BrowseFlag>
|
|
<Filter>*</Filter>
|
|
<StartingIndex>0</StartingIndex>
|
|
<RequestedCount>0</RequestedCount>
|
|
<SortCriteria></SortCriteria>
|
|
</u:Browse>
|
|
</s:Body>
|
|
</s:Envelope>`))
|
|
require.NoError(t, err)
|
|
req.Header.Set("SOAPACTION", `"urn:schemas-upnp-org:service:ContentDirectory:1#Browse"`)
|
|
resp, err = http.DefaultClient.Do(req)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
body, err = ioutil.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
// expect video.mp4, video.srt, URLs to be in the DIDL
|
|
require.Contains(t, string(body), "/r/subdir/video.mp4")
|
|
require.Contains(t, string(body), "/r/subdir/video.srt")
|
|
}
|