static void New(char *args)
{
/*
* Create a new save file.
*/
strcpy(AutoSaveFileName, GetTmpFileName());
ForceMode = FALSE;
FreePosition(CurrentPosition);
CurrentPosition = InitialPosition();
if(State != STATE_ANALYZING) {
State = STATE_WAITING;
}
}
int main(int argc, char *argv[])
{
#if HAVE_SETBUF
setbuf(stdin, NULL);
#endif
OpenLogFile("Amy.log");
InitMoves();
InitAll();
HashInit();
/*
* Process rc file first, then command line options. This way command
* line options can override rc file settings.
*/
ProcessRCFile();
ProcessOptions(argc, argv);
ShowVersion();
AllocateHT();
InitEGTB(EGTBPath);
RecogInit();
DoBookLearning();
Print(0, "\n");
strcpy(AutoSaveFileName, GetTmpFileName());
/* Ensure true random behavior. */
InitRandom(GetTime());
StateMachine();
return 0;
}
// Extract HTTP headers via temporary file with -D switch.
// HTTP status code is extracted from curl output (-w switches).
// Redirects are handled recursively. TODO(AlexZ): avoid infinite redirects loop.
bool HttpClient::RunHttpRequest()
{
ScopedRemoveFile headers_deleter(GetTmpFileName());
ScopedRemoveFile body_deleter;
ScopedRemoveFile received_file_deleter;
string cmd = "curl -s -w '%{http_code}' -X " + m_httpMethod + " -D '" + headers_deleter.m_fileName + "' ";
for (auto const & header : m_headers)
{
cmd += "-H '" + header.first + ": " + header.second + "' ";
}
if (!m_cookies.empty())
cmd += "-b '" + m_cookies + "' ";
if (!m_bodyData.empty())
{
body_deleter.m_fileName = GetTmpFileName();
// POST body through tmp file to avoid breaking command line.
if (!WriteToFile(body_deleter.m_fileName, m_bodyData))
return false;
// TODO(AlexZ): Correctly clean up this internal var to avoid client confusion.
m_inputFile = body_deleter.m_fileName;
}
// Content-Length is added automatically by curl.
if (!m_inputFile.empty())
cmd += "--data-binary '@" + m_inputFile + "' ";
// Use temporary file to receive data from server.
// If user has specified file name to save data, it is not temporary and is not deleted automatically.
string rfile = m_outputFile;
if (rfile.empty())
{
rfile = GetTmpFileName();
received_file_deleter.m_fileName = rfile;
}
cmd += "-o " + rfile + strings::to_string(" ") + "'" + m_urlRequested + "'";
LOG(LDEBUG, ("Executing", cmd));
try
{
m_errorCode = stoi(RunCurl(cmd));
}
catch (RootException const & ex)
{
LOG(LERROR, (ex.Msg()));
return false;
}
m_headers.clear();
Headers const headers = ParseHeaders(ReadFileAsString(headers_deleter.m_fileName));
string serverCookies;
string headerKey;
for (auto const & header : headers)
{
if (header.first == "Set-Cookie")
{
serverCookies += header.second + ", ";
}
else
{
if (header.first == "Location")
m_urlReceived = header.second;
if (m_loadHeaders)
{
headerKey = header.first;
strings::AsciiToLower(headerKey);
m_headers.emplace(headerKey, header.second);
}
}
}
m_headers.emplace("Set-Cookie", NormalizeServerCookies(move(serverCookies)));
if (m_urlReceived.empty())
{
m_urlReceived = m_urlRequested;
// Load body contents in final request only (skip redirects).
// Sometimes server can reply with empty body, and it's ok.
if (m_outputFile.empty())
m_serverResponse = ReadFileAsString(rfile);
}
else
{
// Handle HTTP redirect.
// TODO(AlexZ): Should we check HTTP redirect code here?
LOG(LDEBUG, ("HTTP redirect", m_errorCode, "to", m_urlReceived));
HttpClient redirect(m_urlReceived);
redirect.SetCookies(CombinedCookies());
if (!redirect.RunHttpRequest())
{
m_errorCode = -1;
return false;
}
m_errorCode = redirect.ErrorCode();
m_urlReceived = redirect.UrlReceived();
m_headers = move(redirect.m_headers);
m_serverResponse = move(redirect.m_serverResponse);
}
return true;
}
