static int lua_downstring(lua_State *L){
int argc = lua_gettop(L);
#ifndef SKIP_ERROR_HANDLING
if (argc != 1) return luaL_error(L, "wrong number of arguments");
#endif
const char* url = luaL_checkstring(L,1);
httpcContext context;
Result ret = httpcOpenContext(&context, (char*)url , 0);
#ifndef SKIP_ERROR_HANDLING
if(ret==0){
#endif
httpcBeginRequest(&context);
/*httpcReqStatus loading;
httpcGetRequestState(&context, &loading);
while (loading == 0x5){
httpcGetRequestState(&context, &loading);
}*/
u32 statuscode=0;
u32 contentsize=0;
httpcGetResponseStatusCode(&context, &statuscode, 0);
char text[128];
sprintf(text,"%i",statuscode);
if (statuscode != 200) luaL_error(L, text);
httpcGetDownloadSizeState(&context, NULL, &contentsize);
unsigned char *buffer = (unsigned char*)malloc(contentsize+1);
httpcDownloadData(&context, buffer, contentsize, NULL);
buffer[contentsize] = 0;
lua_pushlstring(L,(const char*)buffer,contentsize);
free(buffer);
#ifndef SKIP_ERROR_HANDLING
}else luaL_error(L, "error opening url");
#endif
httpcCloseContext(&context);
return 1;
}
Result http_download(httpcContext *context, u8** out_buf, u32* out_size)
{
Result ret=0;
u32 statuscode=0;
u32 contentsize=0;
u8 *buf;
ret = httpcBeginRequest(context);
if(ret!=0)return ret;
ret = httpcGetResponseStatusCode(context, &statuscode, 0);
if(ret!=0)return ret;
if(statuscode!=200)return -2;
ret=httpcGetDownloadSizeState(context, NULL, &contentsize);
if(ret!=0)return ret;
buf = (u8*)malloc(contentsize);
if(buf==NULL)return -1;
memset(buf, 0, contentsize);
ret = httpcDownloadData(context, buf, contentsize, NULL);
if(ret!=0)
{
free(buf);
return ret;
}
if(out_size)*out_size = contentsize;
if(out_buf)*out_buf = buf;
else free(buf);
return 0;
}
/***
Download and return the data of the context.
@function :downloadData
@treturn[1] string data
@treturn[2] nil in case of error
@treturn[2] integer error code
*/
static int httpc_downloadData(lua_State *L) {
httpcContext *context = lua_touserdata(L, 1);
u32 status = 0;
Result ret = httpcGetResponseStatusCode(context, &status, 0);
if (ret != 0) {
lua_pushnil(L);
lua_pushinteger(L, ret);
return 2;
}
u32 size = 0;
httpcGetDownloadSizeState(context, NULL, &size);
u8 *buff = (u8*)malloc(size);
ret = httpcDownloadData(context, buff, size, NULL);
if (ret != 0) {
free(buff);
lua_pushnil(L);
lua_pushinteger(L, ret);
return 2;
}
lua_pushstring(L, (char*)buff);
free(buff);
//lua_pushinteger(L, size); // only for test purposes.
return 1;
}
const char* HttpService::get(char *url) {
u32 statuscode = 0;
u32 contentsize = 0;
u8 *buf;
httpcContext context;
Result ret = 0;
ret = httpcOpenContext(&context, url, 1);
ret = httpcBeginRequest(&context);
if(ret!=0)return NULL;
ret = httpcGetResponseStatusCode(&context, &statuscode, 0);
if(ret!=0)return NULL;
if(statuscode!=200)return NULL;
ret=httpcGetDownloadSizeState(&context, NULL, &contentsize);
if(ret!=0)return NULL;
buf = (u8*)malloc(contentsize + 1);
if(buf==NULL)return NULL;
memset(buf, 0, contentsize + 1);
ret = httpcDownloadData(&context, buf, contentsize, NULL);
if(ret!=0)
{
free(buf);
return NULL;
}
httpcCloseContext(&context);
return (const char*)buf;
}
Result DownloadFile_Internal(const char *url, void *out, bool bProgress,
void (*write)(void* out, unsigned char* buffer, u32 readSize))
{
httpcContext context;
u32 fileSize = 0;
u32 procSize = 0;
Result ret = 0;
Result dlret = HTTPC_RESULTCODE_DOWNLOADPENDING;
u32 status;
u32 bufSize = 0x100000;
u32 readSize = 0;
httpcOpenContext(&context, HTTPC_METHOD_GET, (char*)url, 1);
ret = httpcBeginRequest(&context);
if (ret != 0) goto _out;
ret = httpcGetResponseStatusCode(&context, &status, 0);
if (ret != 0) goto _out;
if (status != 200)
{
ret = status;
goto _out;
}
ret = httpcGetDownloadSizeState(&context, NULL, &fileSize);
if (ret != 0) goto _out;
{
unsigned char *buffer = (unsigned char *)linearAlloc(bufSize);
if (buffer == NULL)
{
printf("Error allocating download buffer\n");
ret = -1;
goto _out;
}
while (dlret == (s32)HTTPC_RESULTCODE_DOWNLOADPENDING)
{
memset(buffer, 0, bufSize);
dlret = httpcDownloadData(&context, buffer, bufSize, &readSize);
write(out, buffer, readSize);
procSize += readSize;
if (bProgress)
{
PrintProgress(fileSize, procSize);
}
}
linearFree(buffer);
}
_out:
httpcCloseContext(&context);
return ret;
}
Result http_download(httpcContext *context)
{
ret=0;
//u8* framebuf_top;
u32 statuscode=0;
//u32 size=0;
u32 contentsize=0;
u8 *buf;
ret = httpcBeginRequest(context);
if(ret!=0)return ret;
ret = httpcGetResponseStatusCode(context, &statuscode, 0);
if(ret!=0)return ret;
//printf("http status code: %i\n", statuscode);
if(statuscode!=200){
printf("status code not 200, it was %i", statuscode);
gfxFlushBuffers();
return -2;
}
ret=httpcGetDownloadSizeState(context, NULL, &contentsize);
if(ret!=0)return ret;
unsigned char *buffer = (unsigned char*)malloc(contentsize+1);
consoleSelect(&topScreen);
printf("HTTP status code: %i\n", statuscode);
printf("%i bytes\n", contentsize);
gfxFlushBuffers();
buf = (u8*)malloc(contentsize);
if(buf==NULL)return -1;
memset(buf, 0, contentsize);
ret = httpcDownloadData(context, buffer, contentsize, NULL);
if(ret!=0)
{
free(buf);
return ret;
}
consoleSelect(&bottomScreen);
printf("%s", buffer);
free(buf);
return 0;
}
/***
Return the status code returned by the request.
@function :getStatusCode
@treturn[1] integer the status code
@treturn[2] nil in case of error
@treturn[2] integer error code
*/
static int httpc_getStatusCode(lua_State *L) {
httpcContext *context = lua_touserdata(L, 1);
u32 statusCode = 0;
Result ret = httpcGetResponseStatusCode(context, &statusCode, 0);
if (ret != 0) {
lua_pushnil(L);
lua_pushinteger(L, ret);
return 2;
}
lua_pushinteger(L, statusCode);
return 1;
}
static int lua_download(lua_State *L){
int argc = lua_gettop(L);
#ifndef SKIP_ERROR_HANDLING
if (argc != 2) return luaL_error(L, "wrong number of arguments");
#endif
const char* url = luaL_checkstring(L,1);
const char* file = luaL_checkstring(L,2);
httpcContext context;
Result ret = httpcOpenContext(&context, (char*)url , 0);
#ifndef SKIP_ERROR_HANDLING
if(ret==0){
#endif
httpcBeginRequest(&context);
/*httpcReqStatus loading;
httpcGetRequestState(&context, &loading);
while (loading == 0x5){
httpcGetRequestState(&context, &loading);
}*/
u32 statuscode=0;
u32 contentsize=0;
httpcGetResponseStatusCode(&context, &statuscode, 0);
#ifndef SKIP_ERROR_HANDLING
if (statuscode != 200) luaL_error(L, "download request error");
#endif
httpcGetDownloadSizeState(&context, NULL, &contentsize);
u8* buf = (u8*)malloc(contentsize);
memset(buf, 0, contentsize);
httpcDownloadData(&context, buf, contentsize, NULL);
Handle fileHandle;
u32 bytesWritten;
FS_Archive sdmcArchive=(FS_Archive){ARCHIVE_SDMC, (FS_Path){PATH_EMPTY, 1, (u8*)""}};
FS_Path filePath=fsMakePath(PATH_ASCII, file);
FSUSER_OpenFileDirectly( &fileHandle, sdmcArchive, filePath, FS_OPEN_CREATE|FS_OPEN_WRITE, 0x00000000);
FSFILE_Write(fileHandle, &bytesWritten, 0, buf, contentsize,0x10001);
FSFILE_Close(fileHandle);
svcCloseHandle(fileHandle);
free(buf);
#ifndef SKIP_ERROR_HANDLING
}else luaL_error(L, "error opening url");
#endif
httpcCloseContext(&context);
return 0;
}
static Result action_install_cdn_open_src(void* data, u32 index, u32* handle) {
install_cdn_data* installData = (install_cdn_data*) data;
Result res = 0;
httpcContext* context = (httpcContext*) calloc(1, sizeof(httpcContext));
if(context != NULL) {
char url[256];
if(index == 0) {
snprintf(url, 256, "http://ccs.cdn.c.shop.nintendowifi.net/ccs/download/%016llX/tmd", installData->ticket->titleId);
} else {
snprintf(url, 256, "http://ccs.cdn.c.shop.nintendowifi.net/ccs/download/%016llX/%08lX", installData->ticket->titleId, installData->contentIds[index - 1]);
}
if(R_SUCCEEDED(res = httpcOpenContext(context, HTTPC_METHOD_GET, url, 1))) {
httpcSetSSLOpt(context, SSLCOPT_DisableVerify);
if(R_SUCCEEDED(res = httpcBeginRequest(context)) && R_SUCCEEDED(res = httpcGetResponseStatusCode(context, &installData->responseCode, 0))) {
if(installData->responseCode == 200) {
*handle = (u32) context;
} else {
res = R_FBI_HTTP_RESPONSE_CODE;
}
}
if(R_FAILED(res)) {
httpcCloseContext(context);
}
}
if(R_FAILED(res)) {
free(context);
}
} else {
res = R_FBI_OUT_OF_MEMORY;
}
return res;
}
Result download_file(httpcContext *context, void** buffer, size_t* size, char* user_agent)
{
Result ret;
ret = httpcAddRequestHeaderField(context, "User-Agent", user_agent);
if(R_FAILED(ret)) return ret;
ret = httpcBeginRequest(context);
if(R_FAILED(ret)) return ret;
u32 status_code = 0;
ret = httpcGetResponseStatusCode(context, &status_code);
if(R_FAILED(ret)) return ret;
if(status_code != 200) return -1;
u32 sz = 0;
ret = httpcGetDownloadSizeState(context, NULL, &sz);
if(R_FAILED(ret)) return ret;
void* buf = malloc(sz);
if(!buf) return -2;
memset(buf, 0, sz);
ret = httpcDownloadData(context, buf, sz, NULL);
if(R_FAILED(ret))
{
free(buf);
return ret;
}
if(size) *size = sz;
if(buffer) *buffer = buf;
else free(buf);
return 0;
}
Result downloadPage(Handle httpcHandle, char* url, const short* filename)
{
Result ret;
httpcContext context;
u32 statuscode, size;
ret = httpcOpenContext(httpcHandle, &context, url);
if(!ret)
{
ret = httpcBeginRequest(&context);
if(ret) goto exit;
ret = httpcGetResponseStatusCode(&context, &statuscode);
if(ret || statuscode != 200) goto exit;
ret = httpcGetDownloadSizeState(&context, 0, &size);
if(ret) goto exit;
ret = downloadPageToSDCard(&context, filename, size);
exit: httpcCloseContext(httpcHandle, &context);
}
return ret;
}
int httpGet(const char* url, u8** buf, u32* size) {
httpcContext context;
CHECK(httpcOpenContext(&context, HTTPC_METHOD_GET, (char*)url, 0), "Could not open HTTP context");
// Add User Agent field (required by Github API calls)
CHECK(httpcAddRequestHeaderField(&context, (char*)"User-Agent", (char*)"ARN-UPDATER"), "Could not set User Agent");
CHECK(httpcBeginRequest(&context), "Could not begin request");
// Add root CA required for Github and AWS URLs
CHECK(httpcAddTrustedRootCA(&context, digicert_cer, digicert_cer_len), "Could not add Digicert root CA");
CHECK(httpcAddTrustedRootCA(&context, cybertrust_cer, cybertrust_cer_len), "Could not add Cybertrust root CA");
u32 statuscode = 0;
CHECK(httpcGetResponseStatusCode(&context, &statuscode, 0), "Could not get status code");
if (statuscode != 200) {
// Handle 3xx codes
if (statuscode >= 300 && statuscode < 400) {
char newUrl[1024];
CHECK(httpcGetResponseHeader(&context, (char*)"Location", newUrl, 1024), "Could not get Location header for 3xx reply");
CHECK(httpcCloseContext(&context), "Could not close HTTP context");
return httpGet(newUrl, buf, size);
}
throw formatErrMessage("Non-200 status code", statuscode);
}
CHECK(httpcGetDownloadSizeState(&context, NULL, size), "Could not get file size");
*buf = (u8*)std::malloc(*size);
if (*buf == NULL) throw formatErrMessage("Could not allocate enough memory", *size);
std::memset(*buf, 0, *size);
CHECK(httpcDownloadData(&context, *buf, *size, NULL), "Could not download data");
CHECK(httpcCloseContext(&context), "Could not close HTTP context");
return 1;
}
Result http_downloadsave(httpcContext *context, char *filename)//This error handling needs updated with proper text printing once ctrulib itself supports that.
{
ret = 0;
//u8* framebuf_top;
u32 statuscode=0;
//u32 size=0;
u32 contentsize=0;
u8 *buf;
ret = httpcBeginRequest(context);
if(ret!=0)return ret;
ret = httpcGetResponseStatusCode(context, &statuscode, 0);
if(ret!=0)return ret;
//printf("http status code: %i\n", statuscode);
if(statuscode!=200){
printf("status code not 200, it was %i\n", statuscode);
gfxFlushBuffers();
return -2;
}
ret=httpcGetDownloadSizeState(context, NULL, &contentsize);
if(ret!=0)return ret;
unsigned char *buffer = (unsigned char*)malloc(contentsize+1);
consoleSelect(&topScreen);
printf("HTTP status code: %i\n", statuscode);
printf("%i bytes\n", contentsize);
gfxFlushBuffers();
buf = (u8*)malloc(contentsize);
if(buf==NULL)return -1;
memset(buf, 0, contentsize);
ret = httpcDownloadData(context, buffer, contentsize, NULL);
if(ret!=0)
{
free(buf);
return ret;
}
/*size = contentsize;
if(size>(240*400*3*2))size = 240*400*3*2;
framebuf_top = gfxGetFramebuffer(GFX_TOP, GFX_LEFT, NULL, NULL);
memcpy(framebuf_top, buf, size);
gfxFlushBuffers();
gfxSwapBuffers();
framebuf_top = gfxGetFramebuffer(GFX_TOP, GFX_LEFT, NULL, NULL);
memcpy(framebuf_top, buf, size);
gfxFlushBuffers();
gfxSwapBuffers();
gspWaitForVBlank();*/
printf("Got file\n");
//char filename[32];
FILE *dlfile;
char *fnameformat;
//snprintf(filename, sizeof(char) * 32, "koopadl%i.txt", ++dlCounter);
printf("Saving to %s\n", filename);
dlfile = fopen(filename, "w");
fwrite(buffer, 1, contentsize, dlfile);
fclose(dlfile);
consoleSelect(&topScreen);
printf("Saved to %s\n", filename);
//printf("%s", buffer);
free(buf);
return 0;
}
static int lua_downstring(lua_State *L){
int argc = lua_gettop(L);
#ifndef SKIP_ERROR_HANDLING
if (argc < 1 || argc > 4) return luaL_error(L, "wrong number of arguments");
#endif
const char* url = luaL_checkstring(L,1);
const char* headers = (argc >= 2) ? luaL_checkstring(L,2) : NULL;
u8 method = (argc >= 3) ? luaL_checkinteger(L,3) : 0;
const char* postdata = (argc >= 4) ? luaL_checkstring(L,4) : NULL;
httpcContext context;
HTTPC_RequestMethod useMethod = HTTPC_METHOD_GET;
if(method <= 3 && method >= 1) useMethod = (HTTPC_RequestMethod)method;
u32 statuscode=0;
do {
if (statuscode >= 301 && statuscode <= 308) {
char newurl[4096];
httpcGetResponseHeader(&context, (char*)"Location", &newurl[0], 4096);
url = &newurl[0];
httpcCloseContext(&context);
}
Result ret = httpcOpenContext(&context, useMethod, (char*)url , 0);
// Lets just disable SSL verification instead of loading default certs.
httpcSetSSLOpt(&context, SSLCOPT_DisableVerify);
if(headers != NULL){
char *tokenheader = (char*)malloc(strlen(headers)+1);
strcpy(tokenheader, headers);
char *toker = tokenheader;
char *headername = NULL;
char *headervalue = NULL;
do {
headername = strtok(toker, ":");
if (headername == NULL) break;
headervalue = strtok(NULL, "\n");
if (headervalue == NULL) break;
if (headervalue[0] == ' ') headervalue++;
httpcAddRequestHeaderField(&context, headername, headervalue);
toker = NULL;
} while (headername != NULL && headervalue != NULL);
free(tokenheader);
}
if (useMethod == HTTPC_METHOD_POST && postdata != NULL) {
httpcAddPostDataRaw(&context, (u32*)postdata, strlen(postdata));
}
#ifndef SKIP_ERROR_HANDLING
if(ret==0){
#endif
httpcBeginRequest(&context);
long int contentsize=0; // Crash on the += if u32. WTF?
u32 readSize=0;
httpcGetResponseStatusCode(&context, &statuscode, 0);
if (statuscode == 200) {
unsigned char *buffer = (unsigned char*)malloc(0x1000);
do {
ret = httpcDownloadData(&context, buffer+contentsize, 0x1000, &readSize);
contentsize += readSize;
if (ret == (s32)HTTPC_RESULTCODE_DOWNLOADPENDING)
buffer = (unsigned char*)realloc(buffer, contentsize + 0x1000);
} while (ret == (s32)HTTPC_RESULTCODE_DOWNLOADPENDING);
buffer = (unsigned char*)realloc(buffer, contentsize + 1);
buffer[contentsize] = 0;
lua_pushlstring(L,(const char*)buffer,contentsize);
free(buffer);
}
#ifndef SKIP_ERROR_HANDLING
}
#endif
} while ((statuscode >= 301 && statuscode <= 303) || (statuscode >= 307 && statuscode <= 308));
#ifndef SKIP_ERROR_HANDLING
if ((statuscode < 200 && statuscode > 226) && statuscode != 304) luaL_error(L, "error opening url");
#endif
httpcCloseContext(&context);
return 1;
}
Result http_download_payload(char *url, u32 *payloadsize)
{
Result ret=0;
u32 statuscode=0;
u32 contentsize=0;
httpcContext context;
ret = httpcOpenContext(&context, url, 1);
if(ret!=0)return ret;
ret = httpcAddRequestHeaderField(&context, "User-Agent", "hblauncher_loader/"VERSION);
if(ret!=0)
{
httpcCloseContext(&context);
return ret;
}
ret = httpcBeginRequest(&context);
if(ret!=0)
{
httpcCloseContext(&context);
return ret;
}
ret = httpcGetResponseStatusCode(&context, &statuscode, 0);
if(ret!=0)
{
httpcCloseContext(&context);
return ret;
}
if(statuscode!=200)
{
printf("Error: server returned HTTP statuscode %u.\n", (unsigned int)statuscode);
httpcCloseContext(&context);
return -2;
}
ret=httpcGetDownloadSizeState(&context, NULL, &contentsize);
if(ret!=0)
{
httpcCloseContext(&context);
return ret;
}
if(contentsize==0 || contentsize>PAYLOAD_TEXTMAXSIZE)
{
printf("Invalid HTTP content-size: 0x%08x.\n", (unsigned int)contentsize);
ret = -3;
httpcCloseContext(&context);
return ret;
}
ret = httpcDownloadData(&context, filebuffer, contentsize, NULL);
if(ret!=0)
{
httpcCloseContext(&context);
return ret;
}
httpcCloseContext(&context);
*payloadsize = contentsize;
return 0;
}
static int lua_sendmail(lua_State *L){ //BETA func
int argc = lua_gettop(L);
#ifndef SKIP_ERROR_HANDLING
if (argc != 3) return luaL_error(L, "wrong number of arguments");
#endif
char* to = (char*)luaL_checkstring(L,1);
char* subj = (char*)luaL_checkstring(L,2);
char* mex = (char*)luaL_checkstring(L,3);
int req_size = 70;
int nss = SpaceCounter(subj);
int nsm = SpaceCounter(mex);
req_size = req_size + nss * 2 + nsm * 2 + strlen(subj) + strlen(mex) + strlen(to);
char* url = (char*)malloc(req_size);
strcpy(url,"http://rinnegatamante.netsons.org/tmp_mail_lpp_beta.php?t=");
strcat(url,to);
strcat(url,"&s=");
char* subj_p = subj;
char* url_p = &url[strlen(url)];
while (*subj_p){
if (subj_p[0] == 0x20){
url_p[0] = '%';
url_p++;
url_p[0] = '2';
url_p++;
url_p[0] = '0';
}else url_p[0] = subj_p[0];
url_p++;
subj_p++;
}
strcat(url,"&b=");
char* mex_p = mex;
url_p = &url[strlen(url)];
while (*mex_p){
if (mex_p[0] == 0x20){
url_p[0] = '%';
url_p++;
url_p[0] = '2';
url_p++;
url_p[0] = '0';
}else url_p[0] = mex_p[0];
url_p++;
mex_p++;
}
url_p[0] = 0;
httpcContext context;
Result ret = httpcOpenContext(&context, (char*)url , 0);
#ifndef SKIP_ERROR_HANDLING
if(ret==0){
#endif
httpcBeginRequest(&context);
HTTPC_RequestStatus loading;
httpcGetRequestState(&context, &loading);
while (loading == HTTPC_STATUS_REQUEST_IN_PROGRESS){
httpcGetRequestState(&context, &loading);
}
u32 statuscode=0;
u32 contentsize=0;
httpcGetResponseStatusCode(&context, &statuscode, 0);
if (statuscode != 200) luaL_error(L, "request error");
httpcGetDownloadSizeState(&context, NULL, &contentsize);
u8 response;
httpcDownloadData(&context, &response, contentsize, NULL);
lua_pushboolean(L,response);
free(url);
#ifndef SKIP_ERROR_HANDLING
}else luaL_error(L, "error opening url");
#endif
httpcCloseContext(&context);
return 1;
}
Result download_config(char *url, u8 *cert, u32 certsize, u8 *filebuffer, u32 dlsize, u32 *out_statuscode)
{
Result ret=0;
u32 statuscode=0;
httpcContext context;
ret = httpcOpenContext(&context, HTTPC_METHOD_GET, url, 1);
if(R_FAILED(ret))
{
printf("httpcOpenContext returned 0x%08x.\n", (unsigned int)ret);
return ret;
}
ret = httpcAddRequestHeaderField(&context, "User-Agent", "ctr-httpwn/"VERSION);
if(R_FAILED(ret))
{
printf("httpcAddRequestHeaderField returned 0x%08x.\n", (unsigned int)ret);
httpcCloseContext(&context);
return ret;
}
ret = httpcAddTrustedRootCA(&context, cert, certsize);
if(R_FAILED(ret))
{
printf("httpcAddTrustedRootCA returned 0x%08x.\n", (unsigned int)ret);
httpcCloseContext(&context);
return ret;
}
ret = httpcBeginRequest(&context);
if(R_FAILED(ret))
{
printf("httpcBeginRequest returned 0x%08x.\n", (unsigned int)ret);
httpcCloseContext(&context);
return ret;
}
ret = httpcGetResponseStatusCode(&context, &statuscode, 0);
if(R_FAILED(ret))
{
printf("httpcGetResponseStatusCode returned 0x%08x.\n", (unsigned int)ret);
httpcCloseContext(&context);
return ret;
}
if(out_statuscode)*out_statuscode = statuscode;
if(statuscode==200 || statuscode==500)
{
ret = httpcDownloadData(&context, filebuffer, dlsize, NULL);
if(ret!=0 && statuscode==200)
{
printf("httpcDownloadData returned 0x%08x.\n", (unsigned int)ret);
httpcCloseContext(&context);
return ret;
}
}
ret = httpcCloseContext(&context);
if(R_FAILED(ret))
{
printf("httpcCloseContext returned 0x%08x.\n", (unsigned int)ret);
return ret;
}
if(statuscode!=200)
{
printf("Invalid statuscode: %u.\n", (unsigned int)statuscode);
return -5;
}
return 0;
}
Result DownloadTitle(std::string titleId, std::string encTitleKey, std::string titleName, std::string region)
{
// Convert the titleid to a u64 for later use
char* nTitleId = parse_string(titleId);
u64 uTitleId = u8_to_u64((u8*)nTitleId, BIG_ENDIAN);
free (nTitleId);
// Wait for wifi to be available
u32 wifi = 0;
Result ret = 0;
Result res = 0;
while(R_SUCCEEDED(ret = ACU_GetWifiStatus(&wifi)) && wifi == 0)
{
hidScanInput();
if (hidKeysDown() & KEY_B)
{
ret = -1;
break;
}
}
if (R_FAILED(ret))
{
printf("Unable to access internet.\n");
return ret;
}
std::string outputDir = "/CIAngel";
if (titleName.length() == 0)
{
titleName = titleId;
}
// Include region in filename
if (region.length() > 0)
{
titleName = titleName + " (" + region + ")";
}
std::string mode_text;
if(config.GetMode() == CConfig::Mode::DOWNLOAD_CIA)
{
mode_text = "create";
}
else if(config.GetMode() == CConfig::Mode::INSTALL_CIA)
{
mode_text = "install";
}
printf("Starting - %s\n", titleName.c_str());
// If in install mode, download/install the SEED entry
if (config.GetMode() == CConfig::Mode::INSTALL_CIA)
{
// Download and install the SEEDDB entry if install mode
// Code based on code from FBI: https://github.com/Steveice10/FBI/blob/master/source/core/util.c#L254
// Copyright (C) 2015 Steveice10
u8 seed[16];
static const char* regionStrings[] = {"JP", "US", "GB", "GB", "HK", "KR", "TW"};
u8 region = CFG_REGION_USA;
CFGU_GetSystemLanguage(®ion);
if(region <= CFG_REGION_TWN) {
char url[128];
snprintf(url, 128, SEED_URL "0x%016llX/ext_key?country=%s", uTitleId, regionStrings[region]);
httpcContext context;
if(R_SUCCEEDED(res = httpcOpenContext(&context, HTTPC_METHOD_GET, url, 1))) {
httpcSetSSLOpt(&context, SSLCOPT_DisableVerify);
u32 responseCode = 0;
if(R_SUCCEEDED(res = httpcBeginRequest(&context)) && R_SUCCEEDED(res = httpcGetResponseStatusCode(&context, &responseCode, 0))) {
if(responseCode == 200) {
u32 pos = 0;
u32 bytesRead = 0;
while(pos < sizeof(seed) && (R_SUCCEEDED(res = httpcDownloadData(&context, &seed[pos], sizeof(seed) - pos, &bytesRead)) || (u32)res == HTTPC_RESULTCODE_DOWNLOADPENDING)) {
pos += bytesRead;
}
} else {
res = -1;
}
}
httpcCloseContext(&context);
}
if (R_SUCCEEDED(res))
{
res = InstallSeed(uTitleId, seed);
if (R_FAILED(res))
{
printf("Error installing SEEDDB entry: 0x%lx\n", res);
}
}
}
}
// Make sure the CIA doesn't already exist
std::string cp = outputDir + "/" + titleName + ".cia";
if (config.GetMode() == CConfig::Mode::DOWNLOAD_CIA && FileExists(cp.c_str()))
{
printf("%s already exists.\n", cp.c_str());
return 0;
}
std::ofstream ofs;
FILE *oh = fopen((outputDir + "/tmp/tmd").c_str(), "wb");
if (!oh)
{
printf("Error opening %s/tmp/tmd\n", outputDir.c_str());
return -1;
}
res = DownloadFile((NUS_URL + titleId + "/tmd").c_str(), oh, false);
fclose(oh);
if (res != 0)
{
printf("Could not download TMD. Internet/Title ID is OK?\n");
return res;
}
// Read version
std::ifstream tmdfs;
tmdfs.open(outputDir + "/tmp/tmd", std::ofstream::out | std::ofstream::in | std::ofstream::binary);
char titleVersion[2];
tmdfs.seekg(top+0x9C, std::ios::beg);
tmdfs.read(titleVersion, 0x2);
tmdfs.close();
CreateTicket(titleId, encTitleKey, titleVersion, outputDir + "/tmp/ticket");
printf("Now %s the CIA...\n", mode_text.c_str());
res = ProcessCIA(outputDir + "/tmp", titleName);
if (res != 0)
{
printf("Could not %s the CIA.\n", mode_text.c_str());
return res;
}
if (config.GetMode() == CConfig::Mode::DOWNLOAD_CIA)
{
rename((outputDir + "/tmp/" + titleName + ".cia").c_str(), (outputDir + "/" + titleName + ".cia").c_str());
}
printf(" DONE!\n");
return res;
}
Result http_download(httpcContext *context)//This error handling needs updated with proper text printing once ctrulib itself supports that.
{
Result ret=0;
u8* framebuf_top, *framebuf_bottom;
u32 statuscode=0;
u32 size=0, contentsize=0;
u8 *buf;
framebuf_bottom = gfxGetFramebuffer(GFX_BOTTOM, GFX_LEFT, NULL, NULL);
memset(framebuf_bottom, 0x40, 240*320*3);
gfxFlushBuffers();
gfxSwapBuffers();
framebuf_bottom = gfxGetFramebuffer(GFX_BOTTOM, GFX_LEFT, NULL, NULL);
memset(framebuf_bottom, 0x40, 240*320*3);
gfxFlushBuffers();
gfxSwapBuffers();
gspWaitForVBlank();
ret = httpcBeginRequest(context);
if(ret!=0)return ret;
ret = httpcGetResponseStatusCode(context, &statuscode, 0);
if(ret!=0)return ret;
if(statuscode!=200)return -2;
ret=httpcGetDownloadSizeState(context, NULL, &contentsize);
if(ret!=0)return ret;
buf = (u8*)malloc(contentsize);
if(buf==NULL)return -1;
memset(buf, 0, contentsize);
framebuf_bottom = gfxGetFramebuffer(GFX_BOTTOM, GFX_LEFT, NULL, NULL);
memset(framebuf_bottom, 0xc0, 240*320*3);
gfxFlushBuffers();
gfxSwapBuffers();
framebuf_bottom = gfxGetFramebuffer(GFX_BOTTOM, GFX_LEFT, NULL, NULL);
memset(framebuf_bottom, 0xc0, 240*320*3);
gfxFlushBuffers();
gfxSwapBuffers();
gspWaitForVBlank();
framebuf_top = gfxGetFramebuffer(GFX_TOP, GFX_LEFT, NULL, NULL);
framebuf_bottom = gfxGetFramebuffer(GFX_BOTTOM, GFX_LEFT, NULL, NULL);
ret = httpcDownloadData(context, buf, contentsize, NULL);
if(ret!=0)
{
free(buf);
return ret;
}
size = contentsize;
if(size>(240*400*3))size = 240*400*3;
memset(framebuf_bottom, 0xff, 240*320*3);
memcpy(framebuf_top, buf, size);
gfxFlushBuffers();
gfxSwapBuffers();
framebuf_top = gfxGetFramebuffer(GFX_TOP, GFX_LEFT, NULL, NULL);
framebuf_bottom = gfxGetFramebuffer(GFX_BOTTOM, GFX_LEFT, NULL, NULL);
memset(framebuf_bottom, 0xff, 240*320*3);
memcpy(framebuf_top, buf, size);
gfxFlushBuffers();
gfxSwapBuffers();
gspWaitForVBlank();
free(buf);
return 0;
}
static int lua_download(lua_State *L){
int argc = lua_gettop(L);
#ifndef SKIP_ERROR_HANDLING
if (argc < 2 || argc > 5) return luaL_error(L, "wrong number of arguments");
#endif
const char* url = luaL_checkstring(L,1);
const char* file = luaL_checkstring(L,2);
const char* headers = (argc >= 3) ? luaL_checkstring(L,3) : NULL;
u8 method = (argc >= 4) ? luaL_checkinteger(L,4) : 0;
const char* postdata = (argc >= 5) ? luaL_checkstring(L,5) : NULL;
httpcContext context;
u32 statuscode=0;
HTTPC_RequestMethod useMethod = HTTPC_METHOD_GET;
if(method <= 3 && method >= 1) useMethod = (HTTPC_RequestMethod)method;
do {
if (statuscode >= 301 && statuscode <= 308) {
char newurl[4096];
httpcGetResponseHeader(&context, (char*)"Location", &newurl[0], 4096);
url = &newurl[0];
httpcCloseContext(&context);
}
Result ret = httpcOpenContext(&context, useMethod, (char*)url, 0);
// Just disable SSL verification instead of loading default certs.
httpcSetSSLOpt(&context, SSLCOPT_DisableVerify);
if(headers != NULL){
char *tokenheader = (char*)malloc(strlen(headers)+1);
strcpy(tokenheader, headers);
char *toker = tokenheader;
char *headername = NULL;
char *headervalue = NULL;
do {
headername = strtok(toker, ":");
if (headername == NULL) break;
headervalue = strtok(NULL, "\n");
if (headervalue == NULL) break;
if (headervalue[0] == ' ') headervalue++;
httpcAddRequestHeaderField(&context, headername, headervalue);
toker = NULL;
} while (headername != NULL && headervalue != NULL);
free(tokenheader);
}
if (useMethod == HTTPC_METHOD_POST && postdata != NULL) {
httpcAddPostDataRaw(&context, (u32*)postdata, strlen(postdata));
}
#ifndef SKIP_ERROR_HANDLING
if(ret==0){
#endif
httpcBeginRequest(&context);
u32 contentsize=0;
httpcGetResponseStatusCode(&context, &statuscode, 0);
if (statuscode == 200){
u32 readSize = 0;
long int bytesWritten = 0;
u8* buf = (u8*)malloc(0x1000);
memset(buf, 0, 0x1000);
Handle fileHandle;
FS_Archive sdmcArchive=(FS_Archive){ARCHIVE_SDMC, (FS_Path){PATH_EMPTY, 1, (u8*)""}};
FS_Path filePath=fsMakePath(PATH_ASCII, file);
FSUSER_OpenFileDirectly( &fileHandle, sdmcArchive, filePath, FS_OPEN_CREATE|FS_OPEN_WRITE, 0x00000000);
do {
ret = httpcDownloadData(&context, buf, 0x1000, &readSize);
FSFILE_Write(fileHandle, NULL, bytesWritten, buf, readSize, 0x10001);
bytesWritten += readSize;
} while (ret == (s32)HTTPC_RESULTCODE_DOWNLOADPENDING);
FSFILE_Close(fileHandle);
svcCloseHandle(fileHandle);
free(buf);
}
#ifndef SKIP_ERROR_HANDLING
}
#endif
} while ((statuscode >= 301 && statuscode <= 303) || (statuscode >= 307 && statuscode <= 308));
#ifndef SKIP_ERROR_HANDLING
if ((statuscode < 200 && statuscode > 226) && statuscode != 304) luaL_error(L, "error opening url");
#endif
httpcCloseContext(&context);
lua_pushinteger(L, statuscode);
return 1;
}
Result InitializeClockOffset() {
/* Compares the system clock with current UTC time from timeapi.com */
/* ESSENTIAL for correct OTP generation, unless system clock is UTC */
/* Returns difference, in seconds. */
/* Add the result of this function to the 3DS time to get UTC. */
if (IsValidTimeOffset(g_SystemClockUtcOffset)) return true;
Result ret = 0;
unsigned long statusCode = 0;
unsigned long contentSize = 0;
unsigned char *buffer = NULL;
httpcContext context; // NOTE: Uninitialized memory
httpcInit(0);
char * url = "http://old-labs.muffinti.me/atoolyoucanputonthewall.php";
/* put this line into a blank PHP file:
<?php date_default_timezone_set("UTC"); echo time(); exit; ?>
*/
/* URL returning current time in UTC */
if (ret) {
ret = httpcOpenContext(&context, HTTPC_METHOD_GET, url, 1);
}
if (ret) {
ret = httpcBeginRequest(&context);
}
if (ret) {
ret = httpcGetResponseStatusCode(&context, &statusCode);
if (ret && statusCode != 200) {
printf("WARNING: HTTP status code returned was %d\n", statusCode);
}
}
if (ret) {
ret = httpcGetDownloadSizeState(&context, NULL, &contentSize);
}
if (ret) {
if(contentSize+1 < contentSize) {
ret = R_TINYTOT_OVERFLOW; // overflow -- do not allow
}
}
if (ret) {
buffer = (unsigned char*)malloc(contentSize+1);
if(buffer == NULL) {
ret = R_TINYTOT_OUTOFMEMORY;
}
}
if (ret) {
memset(buffer, 0, contentSize+1); // zero that last byte also
ret = httpcDownloadData(&context, buffer, contentSize, NULL);
}
if (ret) {
time_t utcTime = (time_t)atol(buffer);
time_t systemTime = time(NULL);
signed long timeDifference = systemTime - utcTime; /* time(NULL) + timeDifference = UTC */
g_SystemClockUtcOffset = timeDifference;
}
if (NULL != buffer) {
free(buffer);
}
return ret;
}
static Result action_url_install_open_src(void* data, u32 index, u32* handle) {
url_install_data* installData = (url_install_data*) data;
Result res = 0;
httpcContext* context = (httpcContext*) calloc(1, sizeof(httpcContext));
if(context != NULL) {
if(R_SUCCEEDED(res = httpcOpenContext(context, HTTPC_METHOD_GET, installData->urls[index], 1))) {
char userAgent[128];
snprintf(userAgent, sizeof(userAgent), "Mozilla/5.0 (Nintendo 3DS; Mobile; rv:10.0) Gecko/20100101 FBI/%d.%d.%d", VERSION_MAJOR, VERSION_MINOR, VERSION_MICRO);
if(R_SUCCEEDED(res = httpcSetSSLOpt(context, SSLCOPT_DisableVerify)) && R_SUCCEEDED(res = httpcAddRequestHeaderField(context, "User-Agent", userAgent)) && R_SUCCEEDED(res = httpcBeginRequest(context)) && R_SUCCEEDED(res = httpcGetResponseStatusCode(context, &installData->responseCode, 0))) {
if(installData->responseCode == 200) {
*handle = (u32) context;
} else if(installData->responseCode == 301 || installData->responseCode == 302 || installData->responseCode == 303) {
memset(installData->urls[index], '\0', URL_MAX);
if(R_SUCCEEDED(res = httpcGetResponseHeader(context, "Location", installData->urls[index], URL_MAX))) {
httpcCloseContext(context);
free(context);
return action_url_install_open_src(data, index, handle);
}
} else {
res = R_FBI_HTTP_RESPONSE_CODE;
}
}
if(R_FAILED(res)) {
httpcCloseContext(context);
}
}
if(R_FAILED(res)) {
free(context);
}
} else {
res = R_FBI_OUT_OF_MEMORY;
}
return res;
}
