コード例 #1
0
ファイル: forward_moudle.cpp プロジェクト: mrshelly/avrouter
	void forward_moudle::process_packet(google::protobuf::Message* msg, connection_ptr connection, connection_manager&)
	{
		proto::avpacket* pkt = dynamic_cast<proto::avpacket*>(msg);
		if (pkt->dest().domain() != m_thisdomain)
		{
			// TODO 暂时不实现非本域的转发.
		}

		// 根据发送人更新 routing_table
		if (m_routing_table.find(pkt->src().username()) == std::end(m_routing_table))
		{
			connection->store_module_private("username", pkt->src().username());
			m_routing_table.insert(std::make_pair(pkt->src().username(), connection));
		}

		// 根据用户名找到连接.
		auto forward_target = m_routing_table.find(pkt->dest().username());
		connection_ptr conn = forward_target->second.lock();
		if (forward_target != m_routing_table.end() && conn)
		{
			// 找到, 转发过去.
			// TTL 减1.
			if (pkt->time_to_live() > 1)
			{
				pkt->set_time_to_live(pkt->time_to_live() - 1);
				conn->write_msg(encode(*pkt));
			}
			else
			{
				// TODO 发送 ttl = 0 消息.
			}
		}
		else
		{
			// 没找到,回一个 aGMP 消息报告 no route to host.
		}
		// TODO 根据目的地址转发消息.
	}