summaryrefslogtreecommitdiff
path: root/plugins/provider/fileProvider/fileProvider.cpp
blob: 15abc10b37a1d3b0c9975dd47d785cfe1e55ef34 (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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
/*
 * fileProvider.cpp
 *
 *  Created on: 15.11.2012
 *      Author: savop
 */

#include <plugin.h>
#include <fstream>

using namespace std;

namespace upnp {

class FileProvider : public cUPnPResourceProvider {
private:

  StringMap directoryMap;
  FILE* fileFD;

  bool Load(const string& filename)
  {
    bool result = false;
    if (access(filename.c_str(), F_OK) == 0) {
      isyslog("loading %s", filename.c_str());
      ifstream file;
      file.open(filename.c_str(), ifstream::in);
      string line; int pos;
      while(getline(file, line)){
        if(line.length() > 0 && line[0] != '#'){
          if((pos = line.find_first_of(':')) != string::npos){
            directoryMap[line.substr(0,pos)] = line.substr(pos+1);
          }
        }
      }
    }
    return result;
  }

  string GetFile(const string& uri){
    string mountPoint = uri.substr(6, uri.find_first_of('/',6) - 7);

    string file;
    if(!mountPoint.empty() && !directoryMap[mountPoint].empty()){
      file = directoryMap[mountPoint] + uri.substr(uri.find_first_of('/',6));
    }

    return file;
  }

  bool GetFileStat(const string& uri, struct stat& fileStat){
    struct stat s;
    if(stat(GetFile(uri).c_str(), &s) == 0){
      fileStat = s;
      return true;
    }

    return false;
  }

public:

  virtual string ProvidesSchema() { return "file"; }

  virtual string GetRootContainer() {
    return ProvidesSchema() + "://";
  }

  virtual bool IsContainer(const string& uri) {
    struct stat fileStat;
    if(GetFileStat(uri, fileStat) && S_ISDIR(fileStat.st_mode)) return true;

    return false;
  }

  virtual bool IsLink(const string& uri, string& target) {
    struct stat fileStat;
    if(GetFileStat(uri, fileStat) && S_ISLNK(fileStat.st_mode)) return true;

    return false;
  }

  virtual long GetContainerUpdateId(const string& uri) {
    struct stat fileStat;
    if(GetFileStat(uri, fileStat)){
      return std::max<time_t>(fileStat.st_ctim.tv_sec, fileStat.st_mtim.tv_sec);
    }

    return 0;
  }

  virtual StringList GetContainerEntries(const string& uri) {
    StringList list;

    DIR* dirHandle;
    struct dirent* dirEntry;

    if((dirHandle = opendir(GetFile(uri).c_str())) == NULL){
      return list;
    }

    string filename;
    while ((dirEntry = readdir(dirHandle)) != NULL) {
      filename = dirEntry->d_name;
      if(filename.compare(".") || filename.compare("..")){
        list.push_back(filename);
      }
    }
    closedir(dirHandle);

    return list;
  }

  virtual bool Seekable() const {
    return true;
  }

  virtual bool Open(const string& uri) {
    if(fileFD)
      Close();

    string file = GetFile(uri);

    fileFD = fopen(file.c_str(), "r");

    return (fileFD) ? true : false;
  }

  virtual size_t Read(char* buf, size_t bufLen) {
    if(!fileFD) return 0;

    return fread(buf, 1, bufLen, fileFD);
  }

  virtual bool Seek(size_t offset, int origin) {
    if(!fileFD) return 0;

    return fseek(fileFD, offset, origin) == 0;
  }

  virtual void Close() {
    if(fileFD){
      fclose(fileFD);
      fileFD = NULL;
    }
  }

};

UPNP_REGISTER_RESOURCE_PROVIDER(FileProvider);

}