/* This routine is so old I can't even remember writing it. But I do
* remember that it was an .h file because I didn't know putting code in a
* header was bad magic yet. anyway, this routine opens a connection to a
* system and returns the fd.
* - steve
*
* Cleaned up some of the code to use memory routines which are now the
* default. Also, the routine first checks to see if the address is in
* dotted-decimal form before it does a name lookup.
* - rjkaes
*/
int
opensock(char *ip_addr, uint16_t port)
{
int sock_fd;
struct sockaddr_in port_info;
struct sockaddr_in bind_addr;
int ret;
WSockInit(); //Initialize WSock -- load winsock32.dll version 2.2
assert(ip_addr != NULL);
assert(port > 0);
memset((struct sockaddr *) &port_info, 0, sizeof(port_info));
port_info.sin_family = AF_INET;
/* Lookup and return the address if possible */
ret = lookup_domain(&port_info.sin_addr, ip_addr);
if (ret < 0) {
log_message(LOG_ERR,
"opensock: Could not lookup address \"%s\".",
ip_addr);
return -1;
}
port_info.sin_port = htons(port);
if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
log_message(LOG_ERR, "opensock: socket() error \"%s\".",
strerror(errno));
return -1;
}
/* Bind to the specified address */
if (config.bind_address) {
memset(&bind_addr, 0, sizeof(bind_addr));
bind_addr.sin_family = AF_INET;
bind_addr.sin_addr.s_addr = inet_addr(config.bind_address);
ret = bind(sock_fd, (struct sockaddr *)&bind_addr, sizeof(bind_addr));
if (ret < 0) {
log_message(LOG_ERR, "Could not bind local address \"%\" because of %s",
config.bind_address,
strerror(errno));
return -1;
}
}
if (connect(sock_fd, (struct sockaddr *) &port_info, sizeof(port_info)) < 0) {
log_message(LOG_ERR, "opensock: connect() error \"%s\".",
strerror(errno));
return -1;
}
return sock_fd;
}
MsgMng::MsgMng(ULONG nicAddr, int portNo, Cfg *_cfg)
{
status = FALSE;
packetNo = (ULONG)Time();
udp_sd = tcp_sd = INVALID_SOCKET;
hAsyncWnd = 0;
local.addr = nicAddr;
local.portNo = Thtons(portNo);
cfg = _cfg;
if (!WSockInit(cfg ? TRUE : FALSE)) return;
WCHAR wbuf[128];
DWORD size = sizeof(wbuf);
if (!::GetComputerNameW(wbuf, &size)) {
GetLastErrorMsg("GetComputerName()");
return;
}
WtoU8(wbuf, local.hostName, sizeof(local.hostName));
WtoA(wbuf, localA.hostName, sizeof(localA.hostName));
if (nicAddr == INADDR_ANY)
{
char host[MAX_BUF];
if (::Tgethostname(host, sizeof(host)) == -1)
strcpy(host, local.hostName);
hostent *ent = ::Tgethostbyname(host);
if (ent)
local.addr = *(ULONG *)ent->h_addr_list[0];
}
size = sizeof(wbuf);
if (!::GetUserNameW(wbuf, &size)) {
GetLastErrorMsg("GetUserName()");
return;
}
WtoU8(wbuf, local.userName, sizeof(local.userName));
WtoA(wbuf, localA.userName, sizeof(localA.userName));
orgLocal = local;
status = TRUE;
}
/*
* Start listening to a socket. Create a socket with the selected port.
* The size of the socket address will be returned to the caller through
* the pointer, while the socket is returned as a default return.
* - rjkaes
*/
int
listen_sock(uint16_t port, socklen_t* addrlen)
{
int listenfd;
const int on = 1;
struct sockaddr_in addr;
assert(port > 0);
assert(addrlen != NULL);
WSockInit(); //Initialize WSock -- load winsock32.dll version 2.2
listenfd = socket(AF_INET, SOCK_STREAM, 0);
setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
if (config.ipAddr) {
addr.sin_addr.s_addr = inet_addr(config.ipAddr);
} else {
addr.sin_addr.s_addr = inet_addr("0.0.0.0");
}
if (bind(listenfd, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
log_message(LOG_ERR, "Unable to bind listening socket because of %s",
strerror(errno));
return -1;
}
if (listen(listenfd, MAXLISTEN) < 0) {
log_message(LOG_ERR, "Unable to start listening socket because of %s",
strerror(errno));
return -1;
}
*addrlen = sizeof(addr);
return listenfd;
}
BOOL CWinHTTrackApp::InitInstance()
{
/* See <https://msdn.microsoft.com/library/ff919712> */
#if (defined(_WIN32) && (!defined(_DEBUG)))
{
/* See KB 2389418
"If this parameter is an empty string (""), the call removes the
current directory from the default DLL search order" */
BOOL (WINAPI*const k32_SetDllDirectoryA)(LPCSTR) =
(BOOL (WINAPI *)(LPCSTR))
GetProcAddress(GetModuleHandle("kernel32.dll"), "SetDllDirectoryA");
if (k32_SetDllDirectoryA != NULL && !k32_SetDllDirectoryA("")) {
/* Do no choke on NT or 98SE with KernelEx NT API (James Blough) */
const DWORD dwVersion = GetVersion();
const DWORD dwMajorVersion = (DWORD)(LOBYTE(LOWORD(dwVersion)));
const DWORD dwMinorVersion = (DWORD)(HIBYTE(LOWORD(dwVersion)));
if (dwMajorVersion >= 5) {
assertf(!"SetDllDirectory failed");
}
}
}
#endif
/* No error messageboxes */
SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOALIGNMENTFAULTEXCEPT | SEM_NOGPFAULTERRORBOX);
/* Inits */
CrashReportInit();
hts_set_error_callback(httrackErrorCallback);
hts_init();
WhttMutex = CreateMutex(NULL,FALSE,NULL);
// Change the registry key under which our settings are stored.
// TODO: You should modify this string to be something appropriate
// such as the name of your company or organization.
SetRegistryKey("WinHTTrack Website Copier");
LANG_INIT(); // petite init langue
/* INDISPENSABLE pour le drag&drop! */
InitCommonControls();
if (!AfxOleInit())
{
AfxMessageBox(LANG(LANG_F1));
return FALSE;
}
AfxEnableControlContainer();
// Pointeur sur CShellApp
CShellApp_app=&app;
this_app=this;
_Cinprogress_inst=NULL;
LibRasUse=0;
// Standard initialization
// If you are not using these features and wish to reduce the size
// of your final executable, you should remove from the following
// the specific initialization routines you do not need.
httrack_icon=AfxGetApp()->LoadIcon(IDR_MAINFRAME);
LoadStdProfileSettings(); // Load standard INI file options (including MRU)
// Register the application's document templates. Document templates
// serve as the connection between documents, frame windows and views.
// DOC //
CMultiDocTemplate* pDocTemplate;
pDocTemplate = new CMultiDocTemplate(
IDR_MAINFRAME,
RUNTIME_CLASS(CWinHTTrackDoc),
RUNTIME_CLASS(CSplitterFrame), // main SDI frame window
RUNTIME_CLASS(CView));
AddDocTemplate(pDocTemplate);
/*
CMDIFrameWnd* pMainFrame = new CMDIFrameWnd;
if (!pMainFrame->LoadFrame(IDR_MAINFRAME))
return FALSE;
*/
// create main window
CMainFrame* pMainFrame = new CMainFrame;
if (!pMainFrame->LoadFrame(IDR_MAINFRAME))
return FALSE;
m_pMainWnd = pMainFrame;
int nCmdShow = m_nCmdShow;
// Also in this example, there is only one menubar shared between
// all the views. The automatic menu enabling support of MFC
// will disable the menu items that don't apply based on the
// currently active view. The one MenuBar is used for all
// document types, including when there are no open documents.
// enable file manager drag/drop and DDE Execute open
pMainFrame->DragAcceptFiles();
// Now finally show the main menu
//pMainFrame->ShowWindow(m_nCmdShow);
//pMainFrame->UpdateWindow();
m_pMainWnd = pMainFrame;
// command line arguments are ignored, create a new (empty) document
//OnFileNew();
// DOC //
// Parse command line for standard shell commands, DDE, file open
CCommandLineInfo cmdInfo;
ParseCommandLine(cmdInfo);
TCHAR ModulePath[MAX_PATH + 1];
ModulePath[0] = '\0';
::GetModuleFileName(NULL, ModulePath, sizeof(ModulePath)/sizeof(TCHAR) - 1);
hts_rootdir(ModulePath);
// Restore position
((CMainFrame*)m_pMainWnd)->InitialShowWindow(nCmdShow);
pMainFrame->UpdateWindow();
// Dispatch commands specified on the command line
if (!ProcessShellCommand(cmdInfo))
return FALSE;
// Init Winsock
WSockInit();
// The one and only window has been initialized, so show and update it.
//m_pMainWnd->ShowWindow(SW_SHOW);
//m_pMainWnd->UpdateWindow();
/*CWinApp* app=AfxGetApp();
POSITION pos;
pos=app->GetFirstDocTemplatePosition();
CDocTemplate* templ = app->GetNextDocTemplate(pos);
pos=templ->GetFirstDocPosition();
CDocument* doc = templ->GetNextDoc(pos);
CRuntimeClass* pRuntimeClass = RUNTIME_CLASS( CTest );
CObject* pObject = pRuntimeClass->CreateObject();
ASSERT( pObject->IsKindOf( RUNTIME_CLASS( CTest ) ) );
doc->AddView((CView*) pObject);
*/
{
// enable file manager drag/drop and DDE Execute open
EnableShellOpen();
RegisterShellFileTypes();
CWinApp* pApp = AfxGetApp();
// register "New File" handler
if (pApp->GetProfileInt("Interface","SetupRun",0) != 1
|| pApp->GetProfileInt("Interface","SetupHasRegistered",0) == 1) {
HKEY phkResult;
DWORD creResult;
if (RegCreateKeyEx(HKEY_CLASSES_ROOT,".whtt",0,NULL,REG_OPTION_NON_VOLATILE,KEY_ALL_ACCESS,NULL,&phkResult,&creResult)==ERROR_SUCCESS) {
RegCloseKey(phkResult);
if (RegCreateKeyEx(HKEY_CLASSES_ROOT,".whtt\\ShellNew",0,NULL,REG_OPTION_NON_VOLATILE,KEY_ALL_ACCESS,NULL,&phkResult,&creResult)==ERROR_SUCCESS) {
char voidbuff='\0';
RegSetValueEx(phkResult,"NullFile",0,REG_SZ,(LPBYTE)&voidbuff,1);
RegCloseKey(phkResult);
}
}
}
// Infos la 1ere fois!
if (pApp->GetProfileInt("Interface","FirstRun",0) != 3) {
pApp->WriteProfileInt("Interface","FirstRun",3);
Cabout about;
about.DoModal();
// Default proxy? Check is the current IP looks local or not.
BOOL isPublic = FALSE;
char hostname[256];
if (gethostname(hostname, sizeof(hostname) - 1) == 0) {
struct addrinfo* res = NULL;
struct addrinfo hints;
memset(&hints, 0, sizeof(hints));
hints.ai_family = PF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
hints.ai_protocol = IPPROTO_TCP;
if (getaddrinfo(hostname, NULL, &hints, &res) == 0) {
if (res->ai_addr != NULL && res->ai_addrlen > 0) {
if (res->ai_family == AF_INET) {
sockaddr_in *const si = (sockaddr_in*) res->ai_addr;
const unsigned char *const ipv4 = (unsigned char*) &si->sin_addr;
isPublic = ! (
ipv4[0] == 10 /* 10/8 */
|| (ipv4[0] == 192 && ipv4[1] == 168) /* 192.168/16 */
|| (ipv4[0] == 172 && ipv4[1] >= 16 && ipv4[1] <= 31) /* 172.16/12 */
);
} else if (res->ai_family == AF_INET6) { /* assume no more proxy */
isPublic = TRUE;
}
}
}
if (res) {
freeaddrinfo(res);
}
}
if (!isPublic && maintab) {
maintab->DefineDefaultProxy();
if (maintab->DoModal()!=IDCANCEL) {
// Default proxy values
CString strSection = "OptionsValues";
MyWriteProfileString("",strSection, "Proxy",maintab->m_option10.m_proxy);
MyWriteProfileString("",strSection, "Port",maintab->m_option10.m_port);
}
maintab->UnDefineDefaultProxy();
}
}
}
#ifdef HTTRACK_AFF_WARNING
#ifndef _DEBUG
AfxMessageBox("--WARNING--\r\n"HTTRACK_AFF_WARNING);
#endif
#endif
return TRUE;
}
BOOL MsgMng::WSockReset(void)
{
WSockTerm();
return WSockInit(TRUE);
}
