dht协议实现

上一篇完成了路由表的实现,建立了路由表后，我们还要对路由表进行初始化，因为一开始路由表为空，所以我们需要借助一些知名的dht网络中的节点，对这些节点进行find_node，然后一步步初始化路由表。这里我们选dht.transmissionbt.com来进行初始化。在dht网络都是使用upd协议进行数据的传输，所以我们需要开启一个upd端口，这里使用netty来做通信框架。这里我们主要来看怎么发送find_node

/**

 * find_node

 * @param node 自己的id

 * @param target 目标id

 * @param address 请求地址

 * @param num channel号

 */

public void findNode(String node,String target,InetSocketAddress address,int num){

    if(!channels.get(num).isWritable()){

        return;

    }

    FindNodeRequest findNodeRequest=new FindNodeRequest(node,target);

    channels.get(num).writeAndFlush(new DatagramPacket(Unpooled.copiedBuffer(bencode.encode(DHTUtil.beanToMap(findNodeRequest))), address));

}

FindNodeRequest 主要有哪些字段

@Data

@AllArgsConstructor()

public class BaseRequest {

    private String t;//messageid 2 byte

    private String y;//"q" for query, "r" for response, or "e" for error

    private String q;//method ping/find_node/get_peers/announce_peer

    //private String v;//client identifier registered in BEP 20 这里不管这个

}

public class FindNodeRequest extends BaseRequest{

    private FindNodeRequestContent a=new FindNodeRequestContent();

    @Data

    @Accessors(chain = true)

    private static class FindNodeRequestContent{

        private String id;

        private String target;

    }

    public FindNodeRequest(String id,String target){

        super(DHTUtil.generateMessageId(),

                MessageTypeEnmu.QUERY.getKey(),

                MethodEnmu.FIND_NODE.getKey());

        a.id=id;

        a.target=target;

    }

    public static void main(String[] args) {

        FindNodeRequest f=new FindNodeRequest("df","23");

    }

}

find_node返回结果后，会对其进行bencoded解码最后解析成ProcessDto对象，接下来我们来看怎么处理返回结果

@Override

public void activeProcess(ProcessDto processDto) {

    Map<String, Object> rMap = DHTUtil.getParamMap(processDto.getRawMap(), "r", "FIND_NODE,找不到r参数.map:" + processDto.getRawMap());

    List<Node> nodeList = DHTUtil.getNodeListByRMap(rMap);

    //为空退出

    if (CollectionUtils.isEmpty(nodeList)) return;

    //去重

    Node[] nodes = nodeList.stream().distinct().toArray(Node[]::new);

    //将nodes加入发送队列继续find_node

    for (Node node : nodes) {

        FileUtil.wirteNode("nodes1=="+node.getIp()+","+node.getPort()+"\r\n");

        findNodeTask.put(node.toAddress());

    }

    byte[] id = DHTUtil.getParamString(rMap, "id", "FIND_NODE,找不到id参数.map:" + processDto.getRawMap()).getBytes(CharsetUtil.ISO_8859_1);

    //将发送消息的节点加入路由表

    routingTables.get(processDto.getNum()).put(new Node(id, processDto.getSender(), NodeRankEnum.FIND_NODE_RECEIVE.getKey()));

}

建立路由表的进行find_node的时候，对方也会保存我们的信息到他们的路由表所以我们只需要实现ping用来回应自己一直在线。最后获取infohash最主要的是

get_peers和announce_peer分别对应GetPeersRequest和AnnoucePeersRequest，发送方式也处理方式也类似，可以参考一下源码https://github.com/mistletoe9527/dht-spider

好了，现在已经可以通过dht网络获取到infohash了，但是只有infohash还不够，因为还需要获取种子的metadata，这样才能知道这个infohash对应的种子所对应的文件。只有建立起文件和infohash的对应关系，我们才能提供搜索，下一篇会介绍怎么从dht网络中获取种子的infohash。