6 #include "inspircd_config.h"
7 #include "configreader.h"
18 /* $ModDesc: Provides zlib link support for servers */
19 /* $LinkerFlags: -lz */
23 enum izip_status { IZIP_WAITFIRST, IZIP_OPEN, IZIP_CLOSED };
25 const unsigned int CHUNK = 16384;
27 /** Represents an ZIP user's extra data
29 class izip_session : public classbase
32 z_stream c_stream; /* compression stream */
33 z_stream d_stream; /* decompress stream */
38 class ModuleZLib : public Module
40 izip_session sessions[MAX_DESCRIPTORS];
44 ModuleZLib(InspIRCd* Me)
47 ServerInstance->PublishInterface("InspSocketHook", this);
54 virtual Version GetVersion()
56 return Version(1, 1, 0, 0, VF_VENDOR, API_VERSION);
59 void Implements(char* List)
61 List[I_OnRawSocketConnect] = List[I_OnRawSocketAccept] = List[I_OnRawSocketClose] = List[I_OnRawSocketRead] = List[I_OnRawSocketWrite] = 1;
62 List[I_OnRequest] = 1;
65 virtual char* OnRequest(Request* request)
67 ISHRequest* ISR = (ISHRequest*)request;
68 if (strcmp("IS_NAME", request->GetId()) == 0)
72 else if (strcmp("IS_HOOK", request->GetId()) == 0)
74 return ServerInstance->Config->AddIOHook((Module*)this, (InspSocket*)ISR->Sock) ? (char*)"OK" : NULL;
76 else if (strcmp("IS_UNHOOK", request->GetId()) == 0)
78 return ServerInstance->Config->DelIOHook((InspSocket*)ISR->Sock) ? (char*)"OK" : NULL;
80 else if (strcmp("IS_HSDONE", request->GetId()) == 0)
84 else if (strcmp("IS_ATTACH", request->GetId()) == 0)
92 virtual void OnRawSocketAccept(int fd, const std::string &ip, int localport)
94 izip_session* session = &sessions[fd];
96 /* allocate deflate state */
98 session->status = IZIP_WAITFIRST;
100 session->c_stream.zalloc = (alloc_func)0;
101 session->c_stream.zfree = (free_func)0;
102 session->c_stream.opaque = (voidpf)0;
104 if (deflateInit(&session->c_stream, Z_DEFAULT_COMPRESSION) != Z_OK)
107 session->d_stream.zalloc = (alloc_func)0;
108 session->d_stream.zfree = (free_func)0;
109 session->d_stream.opaque = (voidpf)0;
111 if (deflateInit(&session->d_stream, Z_DEFAULT_COMPRESSION) != Z_OK)
115 virtual void OnRawSocketConnect(int fd)
117 OnRawSocketAccept(fd, "", 0);
120 virtual void OnRawSocketClose(int fd)
122 CloseSession(&sessions[fd]);
125 virtual int OnRawSocketRead(int fd, char* buffer, unsigned int count, int &readresult)
127 izip_session* session = &sessions[fd];
129 if (session->status == IZIP_CLOSED)
132 readresult = read(fd, buffer, count);
136 unsigned char uncompr[count];
138 if(session->status != IZIP_WAITFIRST)
140 session->d_stream.next_in = (Bytef*)buffer;
141 session->d_stream.avail_in = 0;
142 session->d_stream.next_out = uncompr;
143 if (inflateInit(&session->d_stream) != Z_OK)
145 session->status = IZIP_OPEN;
148 while ((session->d_stream.total_out < count) && (session->d_stream.total_in < (unsigned int)readresult))
150 session->d_stream.avail_in = session->d_stream.avail_out = 1; /* force small buffers */
151 if (inflate(&session->d_stream, Z_NO_FLUSH) == Z_STREAM_END)
155 readresult = session->d_stream.total_out;
157 return (readresult > 0);
160 virtual int OnRawSocketWrite(int fd, const char* buffer, int count)
165 unsigned char compr[count*2];
167 izip_session* session = &sessions[fd];
169 if(session->status != IZIP_WAITFIRST)
171 deflateInit(&session->c_stream, Z_DEFAULT_COMPRESSION);
172 session->status = IZIP_OPEN;
175 if(session->status != IZIP_OPEN)
178 session->c_stream.next_in = (Bytef*)buffer;
179 session->c_stream.next_out = compr;
181 while ((session->c_stream.total_in < (unsigned int)count) && (session->c_stream.total_out < (unsigned int)count*2))
183 session->c_stream.avail_in = session->c_stream.avail_out = 1; /* force small buffers */
184 if (deflate(&session->c_stream, Z_NO_FLUSH) != Z_OK)
187 /* Finish the stream, still forcing small buffers: */
190 session->c_stream.avail_out = 1;
191 if (deflate(&session->c_stream, Z_FINISH) == Z_STREAM_END)
195 return write(fd, compr, session->c_stream.total_out);
198 void CloseSession(izip_session* session)
200 if(session->status == IZIP_OPEN)
202 deflateEnd(&session->c_stream);
203 inflateEnd(&session->d_stream);
205 session->status = IZIP_CLOSED;
210 class ModuleZLibFactory : public ModuleFactory
221 virtual Module * CreateModule(InspIRCd* Me)
223 return new ModuleZLib(Me);
228 extern "C" void * init_module( void )
230 return new ModuleZLibFactory;