47 lines
1.1 KiB
C++
47 lines
1.1 KiB
C++
|
|
#define CHUNK 16384
|
|
#include <module_storage.hpp>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include "cookiebot_conf.h"
|
|
#include <iostream>
|
|
#if USE_GZIP == ON
|
|
#include <zlib.h>
|
|
#endif
|
|
auto getModuleStorage(std::string filename) -> std::string {
|
|
std::string::size_type idx;
|
|
idx = filename.rfind('.');
|
|
|
|
if(idx != std::string::npos)
|
|
{
|
|
std::string extension = filename.substr(idx+1);
|
|
#if USE_GZIP == ON
|
|
if(extension=="gz") {
|
|
FILE *in,*out;
|
|
in=fopen(filename.c_str(),"rb");
|
|
fseek(in, 0, SEEK_END);
|
|
fseek(in, -4, SEEK_CUR);
|
|
unsigned int len=0;
|
|
fread((void*)&len, 4, 1, in);
|
|
fclose(in);
|
|
out=fopen(filename.substr(0,idx).c_str(), "wb");
|
|
struct gzFile_s *file = gzopen(filename.c_str(),"rb");
|
|
gzrewind(file);
|
|
uint8_t *buf=new uint8_t[len];
|
|
gzread(file, (void*)buf, len);
|
|
std::cout << buf << std::endl;
|
|
fwrite((void*)buf, len, 1, out);
|
|
gzclose(file);
|
|
fclose(out);
|
|
delete[] buf;
|
|
return filename.substr(0,idx);
|
|
}
|
|
#endif
|
|
return filename;
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|