Пример #1
0
int create_tcp_packet(struct sk_buff **skb, l3_protocol l3_proto, bool syn, bool rst, bool fin)
{
    struct tcphdr *hdr_tcp;
    struct tuple tuple;
    int error;

    switch (l3_proto) {
    case L3PROTO_IPV4:
        error = init_tuple4(&tuple, "8.7.6.5", 8765, "5.6.7.8", 5678, L4PROTO_TCP);
        if (error)
            return error;
        error = create_skb4_tcp(&tuple, skb, 100, 32);
        if (error)
            return error;
        break;
    case L3PROTO_IPV6:
        error = init_tuple6(&tuple, "1::2", 1212, "3::4", 3434, L4PROTO_TCP);
        if (error)
            return error;
        error = create_skb6_tcp(&tuple, skb, 100, 32);
        if (error)
            return error;
        break;
    }

    hdr_tcp = tcp_hdr(*skb);
    hdr_tcp->syn = syn;
    hdr_tcp->rst = rst;
    hdr_tcp->fin = fin;

    return 0;
}
Пример #2
0
/**
 * "ars" means add, remove, send
 */
static bool test_pkt_queue_ars(void)
{
	struct session_entry *session;
	struct sk_buff *skb;
	struct tuple tuple4;
	struct tcphdr *hdr_tcp;
	bool success = true;

	/* Prepare */
	if (is_error(init_ipv4_tuple(&tuple4, "5.6.7.8", 5678, "192.168.2.1", 8765, L4PROTO_TCP)))
		return false;
	session = session_create_str_tcp("1::2", 1212, "3::4", 3434, "192.168.2.1", 8765, "5.6.7.8", 5678,
			V4_INIT); /* The session entry that is supposed to be created in "tcp_close_state_handle". */
	if (!session)
		return false;

	if (is_error(create_skb4_tcp(&tuple4, &skb, 100, 32))) {
		session_return(session);
		return false;
	}
	hdr_tcp = tcp_hdr(skb);
	hdr_tcp->syn = true;
	hdr_tcp->rst = false;
	hdr_tcp->fin = false;

	success &= assert_equals_int(0, pktqueue_add(session, skb), "pktqueue_add 1");
	success &= assert_equals_int(0, pktqueue_remove(session), "pktqueue_remove 1");
	success &= assert_equals_int(-ENOENT, pktqueue_send(session), "pktqueue_send 1");
	success &= assert_equals_int(0, icmp64_pop(), "pktqueue not sent an icmp error");


	session_return(session);
	/* kfree_skb(skb); "skb" kfreed when pktqueue_remove is executed */
	return success;
}