2017-10-20 50 views
1

커널 모듈에 tcp 패키지를 만들고 dev_queue_xmit (skb) 함수를 사용하여 IP 주소로 다른 호스트로 보내려합니다. 그러나 Mac 주소를 직접 수동으로 채우고 싶지 않습니다. 다음은 패키지 생성을위한 코드입니다. 일부 커널 함수는 eth_rebuild_header (skb)를 호출하여 Mac 헤더를 다시 작성합니다. 그러나 제 경우에는 전화를 건 후에 컴퓨터가 멈 춥니 다. Arp 캐시가 해당 IP 주소에 대한 항목을 포함하지 않으면 arp_find가 OS를 크래시 할 것이라고 Google 결과가 응답합니다. 그러나 쉘 명령 "arp -v"에 의해 인쇄 된 것처럼 ARP 캐시에 존재합니다.커널 모듈 개발에서 IP 주소의 Mac 주소를 확인하는 방법은 무엇입니까?

static int build_and_xmit_tcp(char * eth, u_char * smac, u_char * dmac, 
      u_long sip, u_long dip, 
      u_short sport, u_short dport, 
      u_char * pkt, int pkt_len, 
      int syn, int ack, int fin, 
      __be32 seq, __be32 seq_ack) 
{ 
    struct sk_buff * skb = NULL; 
    struct net_device * dev = NULL; 
    struct ethhdr * ethdr = NULL; 
    struct iphdr * iph = NULL; 
    struct tcphdr * tcph = NULL; 
    u_char * pdata = NULL; 
    if(NULL == smac || NULL == dmac) 
     goto out; 

    if(NULL == (dev= dev_get_by_name(&init_net, eth))) 
     goto out; 
    skb = alloc_skb(pkt_len + sizeof(struct iphdr) + sizeof(struct tcphdr) + LL_RESERVED_SPACE(dev), GFP_ATOMIC); 

    if(NULL == skb) 
     goto out; 
    skb_reserve(skb, LL_RESERVED_SPACE(dev)); 

    skb->dev = dev; 
    skb->pkt_type = PACKET_OTHERHOST; 
    skb->protocol = __constant_htons(ETH_P_IP); 
    skb->ip_summed = CHECKSUM_NONE; 
    skb->priority = 0; 
    skb_set_network_header(skb, 0); 
    skb_put(skb, sizeof(struct iphdr)); 
skb_set_transport_header(skb, sizeof(struct iphdr)); 
    skb_put(skb, sizeof(struct tcphdr)); 

    pdata = skb_put(skb, pkt_len); 
    { 
    if(NULL != pkt) 
     memcpy(pdata, pkt, pkt_len); 
    } 

    tcph = tcp_hdr(skb); 
    memset(tcph, 0, sizeof(struct tcphdr)); 
    tcph->source = sport; 
    tcph->dest = dport; 
    tcph->doff=5; 
    tcph->seq = htonl(seq); 
    tcph-> ack_seq= htonl(seq_ack); 
    tcph->psh = pkt_len>0? 1:0; 
    tcph ->fin = fin; 
    tcph->ack = ack; 
    tcph->syn=syn; 
    tcph->window=__constant_htons (65535); 
    skb->csum = 0; 
    tcph->check = 0; 
    iph = ip_hdr(skb); 
    iph->version = 4; 
    iph->ihl = sizeof(struct iphdr)>>2; 
    iph->frag_off = 0; 
    iph->protocol = IPPROTO_TCP; 
    iph->tos = 0; 
    iph->daddr = dip; 
    iph->saddr = sip; 
    iph->ttl = 0x40; 
    iph->tot_len = __constant_htons(skb->len); 
    iph->check = 0; 
    iph->check = ip_fast_csum((unsigned char *)iph,iph->ihl); 

    skb->csum = skb_checksum(skb, iph->ihl*4, skb->len - iph->ihl * 4, 0); 
    tcph->check = csum_tcpudp_magic(sip, dip, skb->len - iph->ihl * 4, IPPROTO_TCP, skb->csum); 
    skb_push(skb, 14); 
    skb_set_mac_header(skb, 0); 
    ethdr = (struct ethhdr *)eth_hdr(skb); 
// memcpy(ethdr->h_dest, dmac, ETH_ALEN); 
// memcpy(ethdr->h_source, smac, ETH_ALEN); 
    ethdr->h_proto = __constant_htons(ETH_P_IP); 

// arp_send(ARPOP_REQUEST, ETH_P_ARP, target, dev, saddr, 
//  dst_ha, dev->dev_addr, NULL); 
    eth_rebuild_header(skb);  // kernel hang.... 
    if(0 > dev_queue_xmit(skb)) { 
     dev_put (dev); 
     kfree_skb (skb); 
    } 
    return(NF_ACCEPT); 

out: 
     dev_put (dev); 
     kfree_skb (skb); 
    return(NF_ACCEPT); 
} 


static int __init myhook_init(void) 
{ 
    printk("=========insert module......\n"); 
    build_and_xmit_tcp(ETH_O, GWMAC_O, DMAC, in_aton(GWIP_O), in_aton(DIP), htons(8888), htons(DPORT), 
        "", 0, 
        1, 0, 0, 0, 0); 
} 

static void __exit myhook_fini(void) 
{ 
    printk("=========rmmod ......\n"); 
} 

module_init(myhook_init); 
module_exit(myhook_fini); 

Google은 나에게 다른 해결책을 제시합니다. 그들은 arp_ioctl이 Mac 주소를 해석 할 수 있다고 말합니다. 그러나 사실 arp_ioctl은 vmlinuz로 정적으로 컴파일되어 다른 모듈에 심볼로 내 보내지 않습니다.

int arp_get(char *ifname, char *ipStr) 
{ 
    struct arpreq req; 
    struct sockaddr_in *sin; 
    int ret = 0; 
    int sock_fd = 0; 
    struct net_device * dev = NULL; 
    printk("arp ---- \n"); 
    if(NULL == (dev= dev_get_by_name(&init_net, ifname))){ 
     dev_put (dev); 
     printk("error dev get \n"); 
     return -1; 
    } 
    struct net *net_arp = dev_net(dev); 
    memset(&req, 0, sizeof(struct arpreq)); 

    sin = (struct sockaddr_in *)&req.arp_pa; 
    sin->sin_family = AF_INET; 
    sin->sin_addr.s_addr = in_aton(ipStr); 


    strncpy(req.arp_dev, ifname, 15); 
    ret = arp_ioctl(net_arp, SIOCGARP, &req);  // can't be called 
    unsigned char *hw = (unsigned char *)req.arp_ha.sa_data; 
    printk("%#x-%#x-%#x-%#x-%#x-%#x\n", hw[0], hw[1], hw[2], hw[3], hw[4], hw[5]); 
    return 0; 
} 

어쩌면 소켓 구조를 만들고 소켓을 기반으로 몇 가지 상위 함수를 사용해보십시오. 하지만 ...

커널 버전을 수행하는 방법 : 2.6.32 OS 버전 : 우분투 9.10 GCC 버전 : 4.41

+0

어떤 OS입니까? 어느 언어? – mikep

+0

@mikep "커널 버전 : 2.6.32 os 버전 : 우분투 9.10 gcc 버전 : 4.41" –

+0

다른 방법을 시도하려면 "ip_route_input"함수를 호출하여 "dst_entry"라우트 구조를 가져 오지만 오류 코드 -113이 반환됩니다. – river

답변

0
int ip_xmit(struct sk_buff *skb) 
{ 
    struct iphdr *iph = ip_hdr(skb); 
    struct tcphdr *tcph = tcp_hdr(skb); 
    printk("dst is %d\n", skb->_skb_dst); 
    int err; 
// err = ip_route_input(skb, iph->daddr, iph->saddr, 0, skb->dev); 

    struct rtalbe *rt; 

    { 
     struct flowi fl = { .oif = 0, 
        .nl_u = { .ip4_u = 
          { .daddr = iph->daddr, 
         .saddr = iph->saddr, 
         .tos = 0 } }, 
        .proto = IPPROTO_TCP, 
        .flags = 0, 
        .uli_u = { .ports = 
          { .sport = tcph->source, 
         .dport = tcph->dest } } }; 

     if (err = ip_route_output_key(&init_net, &rt, &fl)) 
      return err; 
      printk("err is %d\n", err); 
    } 
    skb_dst_set(skb, rt); 
    if(0 > ip_local_out(skb)) { 
     printk("dev error\n"); 
     kfree_skb (skb); 
     return -1; 
    } 
    return 0; 
} 

기능 ip_route_output_key는 경로 대상을 얻을 수 있고, 밖으로 SKB을 제공 ip_local_out을 .