`<?php // 文件路径 define('ABSPATH', dirname(FILE)); // 主进程数, 一般为CPU的1至4倍 define('WORKER_NUM', 2); // 允许最大连接数, 不可大于系统ulimit -n的值 define('MAX_REQUEST', 1000); // 线程数 define('MAX_PROCESS', 10); // 自动查找间隔, 单位为毫秒 define('AUTO_FIND_TIME', 10000); // 发送find_node间隔, 单位秒 define('NEXT_FIND_NODE_TIME', 0.5);

// 载入类文件 require_once ABSPATH . '/inc/Node.class.php'; require_once ABSPATH . '/inc/Bencode.class.php'; require_once ABSPATH .'/inc/Base.class.php'; require_once ABSPATH .'/inc/Socket.class.php';

// 保存swoole_server对象 $serv = NULL; // 设置自身node id $nid = Base::get_node_id(); //echo $nid; // 初始化路由器 $table = array(); // 最后请求时间 $last_find = time(); // 保存线程列表 $threads = array(); //$threads = []; // 长期在线node $bootstrap_nodes = array( // array('dht.transmissionbt.com', 6881), // array('router.bittorrent.com', 6881), // array('router.utorrent.com', 6881) array('160.120.47.92',6829) );

write(date('Y-m-d H:i:s', time()) . " - 服务启动...\n");

//$serv = new swoole_server('0.0.0.0', 6882, SWOOLE_PROCESS, SWOOLE_SOCK_UDP); $socket = socket_create( AF_INET, SOCK_DGRAM, SOL_UDP ); function myswool_server(){ global $nid, $table; global $socket; if ( $socket === false ) { echo "socket_create() failed:reason:" . socket_strerror( socket_last_error() ) . "\n"; } $ok = socket_bind( $socket, '0.0.0.0', 6882 ); if ( $ok === false ) { echo "socket_bind() failed:reason:" . socket_strerror( socket_last_error( $socket ) ); } // auto_find_node(); while ( true ) { auto_find_node(); $from = ""; $port = 0; $len = socket_recvfrom( $socket, $buf,1024, 0, $from, $port ); // echo ' socket_recvfrom len:'.$len.' \n '; $len = recv( $buf,$from, $port); get_peers(array($from, $port), $nid); // echo ' recv len:'.$len.' \r\n '; usleep( 1000 ); // auto_find_node(); } } function recv( $data,$from, $port){ // 检查数据长度 if(strlen($data) == 0) return false;

// 对数据进行解码

$msg = Base::decode($data);

// 获取对端链接信息, udp链接需要加上$from_id参数

// $fdinfo = $serv->connection_info($fd, $from_id); $fdinfo['remote_ip']=$from; $fdinfo['remote_port']=$port; // print_r($msg); // echo $from.' ['.$port.'] '.$msg['y']." \n "; // 对接收到的数据进行类型判断 if($msg['y'] == 'r'){ // 如果是回复, 且包含nodes信息 if(array_key_exists('nodes', $msg['r'])) // 对nodes进行操作 response_action($msg, array($fdinfo['remote_ip'], $fdinfo['remote_port'])); if(array_key_exists('token', $msg['r'])) on_get_peers($msg, $fdinfo); // print_r($msg); }elseif($msg['y'] == 'q'){ echo 'request_action:'.$from.' ['.$port.'] '.$msg['y'].' '; // 如果是请求, 则执行请求判断 request_action($msg, array($fdinfo['remote_ip'], $fdinfo['remote_port'])); }else{ return false; } } myswool_server(); exit; $serv = new Socket(); $serv->set(array( 'worker_num' => WORKER_NUM, 'daemonize' => true, 'max_request' => MAX_REQUEST, 'dispatch_mode' => 2, 'log_file' => ABSPATH . 'error.log' )); $serv->on('WorkerStart', function($serv, $worker_id){ // 添加一个定时器, 使服务器定时寻找节点 $serv->addtimer(AUTO_FIND_TIME); auto_find_node(); }); $serv->on('Receive', function($serv, $fd, $from_id, $data){ // 检查数据长度 if(strlen($data) == 0) return false;

// 对数据进行解码

$msg = Base::decode($data);

// 获取对端链接信息, udp链接需要加上$from_id参数

$fdinfo = $serv->connection_info($fd, $from_id);

// 对接收到的数据进行类型判断

if($msg['y'] == 'r'){

// 如果是回复, 且包含nodes信息

if(array_key_exists('nodes', $msg['r']))

// 对nodes进行操作

response_action($msg, array($fdinfo['remote_ip'], $fdinfo['remote_port']));

}elseif($msg['y'] == 'q'){

// 如果是请求, 则执行请求判断

request_action($msg, array($fdinfo['remote_ip'], $fdinfo['remote_port']));

}else{

return false;

}

}); $serv->on('Timer', function($interval){ for($i=0; $istart(); $threads[$pid] = $process; swoole_process::wait(); } }); $serv->start();

/**

自动查找节点方法, 将在DHT网络中自动搜寻节点信息

@return void */ function auto_find_node(){ global $table; // print_r($table); // echo count($table)."\n"; // 如果路由表中没有数据则先加入DHT网络 if(count($table) == 0) return join_dht(); return; // 循环处理路由表 while(count($table)){ // 从路由表中删除第一个node并返回被删除的node $node = array_shift($table); print_r($node); // 发送查找find_node到node中 find_node(array($node->ip, $node->port), $node->nid); sleep(0.005); } }

/**

加入dht网络

@return void */ function join_dht(){ global $table, $bootstrap_nodes; // print_r($bootstrap_nodes); // 循环操作 foreach($bootstrap_nodes as $node){ // 将node域名解析为IP地址, 并发送find_node请求 find_node(array(gethostbyname($node[0]), $node[1])); } }

/**

发送find_node请求

@param array $address 对端链接信息

@param string $id node id

@return void */ function find_node($address, $id = null){ global $nid, $table;

// 若未指定id则使用自身node id if(is_null($id)) $mid = $nid; else // 否则伪造一个相邻id $mid = Base::get_neighbor($id, $nid);

// 定义发送数据 $msg = array( 't' => Base::entropy(2), 'y' => 'q', 'q' => 'find_node', 'a' => array( 'id' => $nid, 'target' => $mid ) ); // print_r($msg); // 发送请求数据到对端 send_response($msg, $address); }

function get_peers($address, $id = null){ global $nid, $table;

// 定义发送数据

$msg = array(

't' => Base::entropy(2),

'y' => 'q',

'q' => 'get_peers',

'a' => array(

'id' => $nid,

'info_hash' => "mnopqrstuvwxyz123456"

)

);

// print_r($msg); // 发送请求数据到对端 send_response($msg, $address); } /**

处理对端发来的请求

@param array $msg 接收到的请求数据

@param array $address 对端链接信息

@return void */ function request_action($msg, $address){ switch($msg['q']){ case 'ping': on_ping($msg, $address); break; case 'find_node': on_find_node($msg, $address); break; case 'get_peers': // 处理get_peers请求 on_get_peers($msg, $address); break; case 'announce_peer': // 处理announce_peer请求 on_announce_peer($msg, $address); break; default: return false; } }

/**

处理接收到的find_node回复

@param array $msg 接收到的数据

@param array $address 对端链接信息

@return void */ function response_action($msg, $address){ // 先检查接收到的信息是否正确 if(!isset($msg['r']['nodes']) || !isset($msg['r']['nodes'][1])) return false; // print_r( $msg);

// 对nodes数据进行解码 $nodes = Base::decode_nodes($msg['r']['nodes']);

// print_r( $nodes); // 对nodes循环处理 foreach($nodes as $node){ // 将node加入到路由表中 append($node); } }

/**

处理ping请求

@param array $msg 接收到的ping请求数据

@param array $address 对端链接信息

@return void */ function on_ping($msg, $address){ global $nid; echo "on_ping\n"; // 获取对端node id $id = $msg['a']['id']; // 生成回复数据 $msg = array( 't' => $msg['t'], 'y' => 'r', 'r' => array( 'id' => $nid ) );

// 将node加入路由表 append(new Node($id, $address[0], $address[1])); // 发送回复数据 send_response($msg, $address); }

/**

处理find_node请求

@param array $msg 接收到的find_node请求数据

@param array $address 对端链接信息

@return void */ function on_find_node($msg, $address){ global $nid;

// 获取node列表 $nodes = get_nodes(16); // 获取对端node id $id = $msg['a']['id']; // 生成回复数据 $msg = array( 't' => $msg['t'], 'y' => 'r', 'r' => array( 'id' => $nid, 'nodes' => Base::encode_nodes($nodes) ) );

// 将node加入路由表 append(new Node($id, $address[0], $address[1])); // 发送回复数据 send_response($msg, $address); }

/**

处理get_peers请求

@param array $msg 接收到的get_peers请求数据

@param array $address 对端链接信息

@return void */ function on_get_peers($msg, $address){ global $nid; echo "on_get_peers\n";

// 获取info_hash信息 $infohash = $msg['a']['info_hash']; // 获取node id $id = $msg['a']['id'];

// 生成回复数据 $msg = array( 't' => $msg['t'], 'y' => 'r', 'r' => array( 'id' => $nid, 'nodes' => Base::encode_nodes(get_nodes()), 'token' => substr($infohash, 0, 2) ) );

// 将node加入路由表 append(new Node($id, $address[0], $address[1])); // 向对端发送回复数据 send_response($msg, $address); }

/**

处理announce_peer请求

@param array $msg 接收到的announce_peer请求数据

@param array $address 对端链接信息

@return void */ function on_announce_peer($msg, $address){ global $nid; echo "on_announce_peer\n";

// 获取infohash $infohash = $msg['a']['info_hash']; // 获取token $token = $msg['a']['token']; // 获取node id $id = $msg['a']['id'];

// 验证token是否正确 if(substr($infohash, 0, 2) == $token){ /$txt = array( 'action' => 'announce_peer', 'msg' => array( 'ip' => $address[0], 'port1' => $address[1], 'port2' => $msg['a']['port'], 'infohash' => $infohash ) ); var_dump($txt);/ write(date('Y-m-d H:i:s', time()) . " 获取到info_hash: " . strtoupper(bin2hex($infohash)) . "\n"); }

// 生成回复数据 $msg = array( 't' => $msg['t'], 'y' => 'r', 'r' => array( 'id' => $nid ) ); /* $msg = array( 't' => $msg['t'], 'y' => "q", 'q' => "find_node", 'a' => array('id' => $nid, 'target' => Base::get_node_id()) );//一样可以 */ // 发送请求回复 send_response($msg, $address); }

/**

向对端发送数据

@param array $msg 要发送的数据 Array ( [t] => - [y] => q [q] => find_node [a] => Array ( [id] => Ii蝶 峗落| 唃猎 [target] => Ii蝶 峗落| 唃猎 )

)

@param array $address 对端链接信息

@return void */ function send_response($msg, $address){ global $serv,$socket; // print_r($msg); // echo $msg['a']['target']."\n"; // $str=$msg['a']['target']; // $str=Base::encode($msg); // echo bin2hex($str)."\n"; socket_sendto($socket,Base::encode($msg),strlen(Base::encode($msg)),0,$address[0], $address[1]); // $serv->sendto($address[0], $address[1], Base::encode($msg)); }

/**

添加node到路由表

@param Node $node node模型

@return boolean 是否添加成功 */ function append($node){ global $nid, $table;

// 检查node id是否正确 if(!isset($node->nid[19])) return false;

// 检查是否为自身node id if($node->nid == $nid) return false;

// 检查node是否已存在 if(in_array($node, $table)) return false;

// 如果路由表中的项达到200时, 删除第一项 if(count($table) >= 200) array_shift($table); else{ echo count($table).']'.$node->ip.':'.$node->port."\n"; // get_peers(array($node->ip, $node->port), $node->nid); // find_node(array($node->ip, $node->port), $node->nid); } // print_r($node);

return array_push($table, $node); }

function get_nodes($len = 8){ global $table;

if(count($table) <= $len)

return $table;

$nodes = array();

for($i=0; $i

$nodes[] = $table[mt_rand(0, count($table) - 1)];

}

return $nodes;

}

/**

将数据写入文件

@param string $msg 要写入文件的数据

@return void */ function write($msg){ $fp = fopen('./infohash.log', 'ab'); fwrite($fp, $msg); fclose($fp); } `