Result http_getactual_payloadurl(char *requrl, char *outurl, u32 outurl_maxsize)
{
Result ret=0;
httpcContext context;
ret = httpcOpenContext(&context, requrl, 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 = httpcGetResponseHeader(&context, "Location", outurl, outurl_maxsize);
httpcCloseContext(&context);
return 0;
}
/***
Get a header field from a response.
@function :getResponseHeader
@tparam string name name of the header field to get
@tparam[opt=2048] number maximum size of the value to get
@treturn string field value
*/
static int httpc_getResponseHeader(lua_State *L) {
httpcContext *context = lua_touserdata(L, 1);
char *name = (char*)luaL_checkstring(L, 2);
u32 maxSize = luaL_checkinteger(L, 3);
char* value = 0;
httpcGetResponseHeader(context, name, value, maxSize);
lua_pushstring(L, value);
return 1;
}
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;
}
Result get_redirect(char *url, char *out, size_t out_size, char *user_agent)
{
Result ret;
httpcContext context;
ret = httpcOpenContext(&context, HTTPC_METHOD_GET, url, 0);
if(R_FAILED(ret)) return ret;
ret = httpcAddRequestHeaderField(&context, "User-Agent", user_agent);
if(R_SUCCEEDED(ret)) ret = httpcBeginRequest(&context);
if(R_FAILED(ret))
{
httpcCloseContext(&context);
return ret;
}
ret = httpcGetResponseHeader(&context, "Location", out, out_size);
httpcCloseContext(&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;
}
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;
}
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;
}
