`<?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); } `