2 changed files with 0 additions and 191 deletions
@ -1,125 +0,0 @@ |
|||||
#define _GNU_SOURCE |
|
||||
#include <stdio.h> |
|
||||
#include <stdlib.h> |
|
||||
#include <arpa/inet.h> |
|
||||
#include <string.h> |
|
||||
#include <stdint.h> |
|
||||
#include <stdbool.h> |
|
||||
#include <time.h> |
|
||||
|
|
||||
|
|
||||
#define FLOW_COLS 8 |
|
||||
#define uint unsigned int |
|
||||
|
|
||||
|
|
||||
uint32_t ip2int(const char* ip) |
|
||||
{ |
|
||||
uint32_t res = 0; |
|
||||
inet_pton(AF_INET, ip, &res); |
|
||||
return htonl(res); |
|
||||
} |
|
||||
|
|
||||
|
|
||||
uint str_split(char* str, const char* delimiter, char** pInChunks) |
|
||||
{ |
|
||||
char* dat = strtok(str, " "); |
|
||||
register uint n=0; |
|
||||
while(dat) |
|
||||
{ |
|
||||
pInChunks[n++] = dat; |
|
||||
dat = strtok(NULL, " "); |
|
||||
} |
|
||||
return n; |
|
||||
} |
|
||||
|
|
||||
|
|
||||
void curtime(char* pInStrTime, const uint maxlen) |
|
||||
{ |
|
||||
time_t rawtime; |
|
||||
time( &rawtime ); |
|
||||
strftime(pInStrTime, maxlen, "flowstat_%d%m%Y", localtime( &rawtime )); |
|
||||
} |
|
||||
|
|
||||
|
|
||||
void convert(char* query, char* pInRes) |
|
||||
{ |
|
||||
char* chunks[FLOW_COLS] = {NULL}; |
|
||||
|
|
||||
int chunk_count = str_split(query, " ", chunks); |
|
||||
|
|
||||
if(chunk_count < 7) |
|
||||
{ |
|
||||
printf("Too short input line\n"); |
|
||||
exit(1); |
|
||||
} |
|
||||
|
|
||||
uint32_t src_ip = ip2int(chunks[0]); |
|
||||
uint32_t dst_ip = ip2int(chunks[1]); |
|
||||
uint proto = atoi(chunks[2]); |
|
||||
uint16_t src_port = ip2int(chunks[3]); |
|
||||
uint16_t dst_port = ip2int(chunks[4]); |
|
||||
uint octets = atoi(chunks[5]); |
|
||||
uint packets = atoi(chunks[6]); |
|
||||
|
|
||||
sprintf(pInRes, ",(%u,%u,%u,%u,%u,%u,%u)\0", |
|
||||
src_ip, dst_ip, proto, src_port, dst_port, octets, packets); |
|
||||
} |
|
||||
|
|
||||
|
|
||||
int main() |
|
||||
{ |
|
||||
char buf_result_convert[0xff] = {0}; |
|
||||
FILE* f = stdin; |
|
||||
char* input_line = malloc(0xff); |
|
||||
size_t input_line_len = 0; |
|
||||
ssize_t read_len = 0; |
|
||||
char table_name[19] = {0}; |
|
||||
|
|
||||
curtime(table_name, 19); |
|
||||
|
|
||||
printf("CREATE TABLE IF NOT EXISTS %s (\n", table_name); |
|
||||
printf("`id` int(10) AUTO_INCREMENT NOT NULL,\n"); |
|
||||
printf("`src_ip` INT(10) UNSIGNED NOT NULL,\n"); |
|
||||
printf("`dst_ip` INT(10) UNSIGNED NOT NULL,\n"); |
|
||||
printf("`proto` smallint(2) unsigned NOT NULL DEFAULT 0,\n"); |
|
||||
printf("`src_port` smallint(5) unsigned NOT NULL DEFAULT 0,\n"); |
|
||||
printf("`dst_port` smallint(5) unsigned NOT NULL DEFAULT 0,\n"); |
|
||||
printf("`octets` INT unsigned NOT NULL DEFAULT 0,\n"); |
|
||||
printf("`packets` INT unsigned NOT NULL DEFAULT 0,\n"); |
|
||||
printf("PRIMARY KEY (`id`)\n"); |
|
||||
printf(") ENGINE=MyISAM DEFAULT CHARSET=utf8;\n"); |
|
||||
|
|
||||
char ins_sql[0xff] = {0}; |
|
||||
sprintf(ins_sql, "INSERT INTO %s(`src_ip`, `dst_ip`, `proto`, `src_port`, `dst_port`, `octets`, `packets`) VALUES", table_name); |
|
||||
|
|
||||
// always none |
|
||||
read_len = getline(&input_line, &input_line_len, f); |
|
||||
|
|
||||
while(true) |
|
||||
{ |
|
||||
register uint n=0xfff; |
|
||||
read_len = getline(&input_line, &input_line_len, f); |
|
||||
if(read_len <= 0) |
|
||||
break; |
|
||||
convert(input_line, buf_result_convert); |
|
||||
|
|
||||
printf("%s\n", ins_sql); |
|
||||
|
|
||||
// without first comma |
|
||||
printf("%s\n", buf_result_convert+1); |
|
||||
|
|
||||
while(n>0) |
|
||||
{ |
|
||||
read_len = getline(&input_line, &input_line_len, f); |
|
||||
if(read_len <= 0) |
|
||||
break; |
|
||||
convert(input_line, buf_result_convert); |
|
||||
printf("%s\n", buf_result_convert); |
|
||||
n--; |
|
||||
} |
|
||||
putc(';', stdout); |
|
||||
} |
|
||||
|
|
||||
free(input_line); |
|
||||
return 0; |
|
||||
} |
|
||||
@ -1,66 +0,0 @@ |
|||||
#!/bin/env python3 |
|
||||
import sys |
|
||||
import socket |
|
||||
import struct |
|
||||
from re import sub |
|
||||
from django.utils import timezone |
|
||||
|
|
||||
|
|
||||
def ip2int(strip): |
|
||||
return struct.unpack("!I", socket.inet_aton(strip))[0] |
|
||||
|
|
||||
|
|
||||
def convert(query): |
|
||||
dat = sub(r'\s+', ' ', query.strip('\n')).split(' ') |
|
||||
|
|
||||
if len(dat) == 1: |
|
||||
return |
|
||||
|
|
||||
src_ip = ip2int(dat[0]) |
|
||||
dst_ip = ip2int(dat[1]) |
|
||||
proto = int(dat[2]) |
|
||||
src_port = int(dat[3]) |
|
||||
dst_port = int(dat[4]) |
|
||||
octets = int(dat[5]) |
|
||||
packets = int(dat[6]) |
|
||||
|
|
||||
sql = ",(%d,%d,%d,%d,%d,%d,%d)" % ( |
|
||||
src_ip, dst_ip, proto, src_port, dst_port, octets, packets |
|
||||
) |
|
||||
return sql |
|
||||
|
|
||||
|
|
||||
if __name__ == '__main__': |
|
||||
f = sys.stdin |
|
||||
table_name = "flowstat_%s" % timezone.now().strftime("%d%m%Y") |
|
||||
print(("CREATE TABLE IF NOT EXISTS %s (" % table_name)) |
|
||||
print("`id` int(10) AUTO_INCREMENT NOT NULL,") |
|
||||
print("`src_ip` INT(10) UNSIGNED NOT NULL,") |
|
||||
print("`dst_ip` INT(10) UNSIGNED NOT NULL,") |
|
||||
print("`proto` smallint(2) unsigned NOT NULL DEFAULT 0,") |
|
||||
print("`src_port` smallint(5) unsigned NOT NULL DEFAULT 0,") |
|
||||
print("`dst_port` smallint(5) unsigned NOT NULL DEFAULT 0,") |
|
||||
print("`octets` INT unsigned NOT NULL DEFAULT 0,") |
|
||||
print("`packets` INT unsigned NOT NULL DEFAULT 0,") |
|
||||
print("PRIMARY KEY (`id`)") |
|
||||
print(") ENGINE=MyISAM DEFAULT CHARSET=utf8;") |
|
||||
ins_sql = r"INSERT INTO %s(`src_ip`, `dst_ip`, `proto`, `src_port`, `dst_port`, `octets`, `packets`) VALUES" % table_name |
|
||||
|
|
||||
# always none |
|
||||
f.readline() |
|
||||
|
|
||||
while True: |
|
||||
n = 0xfff |
|
||||
rs = convert(f.readline()) |
|
||||
if not rs: exit() |
|
||||
# without first comma |
|
||||
print(ins_sql) |
|
||||
print((rs[1:])) |
|
||||
while n > 0: |
|
||||
rs = convert(f.readline()) |
|
||||
if not rs: exit() |
|
||||
print(rs) |
|
||||
n -= 1 |
|
||||
print(';') |
|
||||
|
|
||||
f.close() |
|
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue