如何用java实现一个p2p种子搜索(3)-dht协议实现

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。