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
|
/* +------------------------------------+
* | Inspire Internet Relay Chat Daemon |
* +------------------------------------+
*
* InspIRCd: (C) 2002-2007 InspIRCd Development Team
* See: http://www.inspircd.org/wiki/index.php/Credits
*
* This program is free but copyrighted software; see
* the file COPYING for details.
*
* ---------------------------------------------------
*/
#include "inspircd.h"
#include "httpclient.h"
/* $ModDesc: The base module for remote includes */
class ModuleRemoteInclude : public Module
{
std::map<std::string, std::stringstream*> assoc;
public:
ModuleRemoteInclude(InspIRCd* Me)
: Module(Me)
{
ServerInstance->Modules->Attach(I_OnDownloadFile, this);
ServerInstance->Modules->Attach(I_OnRequest, this);
}
virtual ~ModuleRemoteInclude()
{
}
virtual Version GetVersion()
{
// this method instantiates a class of type Version, and returns
// the modules version information using it.
return Version(1,1,0,1,VF_VENDOR,API_VERSION);
}
char* OnRequest(Request* req)
{
if (!strcmp(req->GetId(), HTTP_CLIENT_RESPONSE))
{
HTTPClientResponse* resp = (HTTPClientResponse*)req;
ServerInstance->Log(DEBUG, "Got http file for %s", resp->GetURL().c_str());
std::map<std::string, std::stringstream*>::iterator n = assoc.find(resp->GetURL());
if (n == assoc.end())
ServerInstance->Config->Complete(resp->GetURL(), true);
*(n->second) << resp->GetData();
ServerInstance->Log(DEBUG, "Got data: %s", resp->GetData().c_str());
ServerInstance->Log(DEBUG, "Flag file complete without error");
ServerInstance->Config->Complete(resp->GetURL(), false);
/* Erase from our association map, but dont delete the pointer.
* the core will want to access this pointer for the file data.
*/
assoc.erase(n);
}
else if (!strcmp(req->GetId(), HTTP_CLIENT_ERROR))
{
HTTPClientError* resp = (HTTPClientError*)req;
ServerInstance->Log(DEBUG, "Got http error when accessing %s", resp->GetURL().c_str());
ServerInstance->Config->Complete(resp->GetURL(), true);
std::map<std::string, std::stringstream*>::iterator n = assoc.find(resp->GetURL());
if (n != assoc.end())
assoc.erase(n);
}
return NULL;
}
int OnDownloadFile(const std::string &name, std::istream* &filedata)
{
if (name.substr(0, 7) == "http://")
{
Module* target = ServerInstance->Modules->Find("m_http_client.so");
if (target)
{
ServerInstance->Log(DEBUG,"Claiming schema http://, making fetch request");
HTTPClientRequest req(ServerInstance, this, target, name);
req.Send();
assoc[name] = new std::stringstream();
delete filedata;
filedata = assoc[name];
return true;
}
}
return false;
}
};
MODULE_INIT(ModuleRemoteInclude)
|
