summaryrefslogtreecommitdiff
path: root/httptnt/resourceStreamer.ecpp
blob: 3ab17bd0f3f59117570e34a64245874025dd7834 (plain)
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
<%args>
objectID;
int resourceID = 0;
</%args>
<%pre>
#include <string>
#include <stdint.h>
#include <limits>
#include <boost/shared_ptr.hpp>
#include "../include/tools.h"
#include "../include/media/mediaManager.h"
#include "../include/server.h"

#include <tnt/httpreply.h>
#include <tnt/httprequest.h>
#include <tnt/http.h>

using namespace upnp;
</%pre>
<#
int doRequest(tnt::HttpReply reply, tnt::HttpRequest request, std::string objectID, int resourceID) {
#>
<%cpp>
  if(objectID.empty()){
    reply.out() << "Object ID missing";
    return HTTP_BAD_REQUEST;
  }

  cMediaServer* server = cMediaServer::GetInstance();
  cMediaManager& manager = server->GetManager();
  boost::shared_ptr<cResourceStreamer> streamer(manager.GetResourceStreamer(objectID, resourceID));

  if(!streamer.get()){
    reply.out() << "Object ID not found or invalid";
    return HTTP_NOT_FOUND;
  }

  int code = HTTP_OK;

  reply.setContentType(streamer->GetContentType());
  reply.setHeader("friendlyName.dlna.org  ", server->GetServerDescription().friendlyName);
  reply.setHeader("contentFeatures.dlna.org  ", streamer->GetContentFeatures());
  reply.setHeader("transferMode.dlna.org  ", streamer->GetTransferMode(reply.getHeader("transferMode.dlna.org")));

  if(streamer->GetContentLength() > 0){
    reply.setContentLengthHeader(streamer->GetContentLength());
    if(streamer->Seekable()) reply.setHeader("Accept-Ranges", "bytes");
  }

  if(!request.isMethodHEAD()){

    if(!streamer->Open()){
      code = HTTP_INTERNAL_SERVER_ERROR;
      goto ret;
    }

    int offset = 0, length = std::numeric_limits<long>::max();
    bool hasRange = false;

    if(streamer->Seekable() && streamer->GetContentLength() > 0){
      if(request.hasHeader("Range")){
        std::string rangeRequest = request.getHeader("Range");
        if(rangeRequest.find("bytes=",0) == 0){
          int minus = rangeRequest.find_first_of('-',6);
          offset  = atoi(rangeRequest.substr(6, minus - 6).c_str());
          length = atoi(rangeRequest.substr(minus + 1).c_str()) - offset;

          if(length == 0) length = std::numeric_limits<long>::max();

          hasRange = (offset == 0 && length == std::numeric_limits<long>::max()) ? false : true;
        }
      }

      if(!streamer->Seek(offset, SEEK_SET)){
        code = HTTP_INTERNAL_SERVER_ERROR;
        goto ret;
      }
    }

    char buffer[KB(16)];
    ssize_t bytesRead = 0;
    size_t totalBytesRead = 0;
    size_t toRead = std::min<size_t>(length, sizeof(buffer));
    while (0 < (bytesRead = streamer->Read(buffer, toRead)) && length > 0) {
      reply.out().write(buffer, bytesRead);
      if (!reply.out()) {
        code = HTTP_GONE;
        goto ret;
      }
      length -= bytesRead;
      totalBytesRead += bytesRead;
      toRead = std::min<size_t>(length, sizeof(buffer));
    }

    reply.setContentLengthHeader(totalBytesRead);

    if ((bytesRead < 0 || length <= 0) && hasRange) {
      std::stringstream ss;
      ss << "bytes " << offset << "-" << offset + totalBytesRead << "/" << streamer->GetContentLength();
      reply.setHeader("Content-Range", ss.str());
      code = HTTP_PARTIAL_CONTENT;
      goto ret;
    }

  }

  ret:
  streamer->Close();
  //reply.out() << std::flush;
  return code;
</%cpp>
<#
}
#>