Files
2026-06-22 16:18:34 +02:00

643 lines
25 KiB
C#

#if UNITY_EDITOR || DEVELOPMENT_BUILD
#define DEVELOPMENT
#endif
using System;
using FishNet.CodeGenerating;
using FishNet.Connection;
using FishNet.Documenting;
using FishNet.Managing;
using FishNet.Managing.Transporting;
using FishNet.Object.Delegating;
using FishNet.Serializing;
using FishNet.Transporting;
using GameKit.Dependencies.Utilities;
using System.Collections.Generic;
using System.Text;
using FishNet.Managing.Statistic;
using FishNet.Serializing.Helping;
using UnityEngine;
namespace FishNet.Object
{
public abstract partial class NetworkBehaviour : MonoBehaviour
{
#region Types.
private struct BufferedRpc
{
/// <summary>
/// Writer containing the full RPC.
/// </summary>
public PooledWriter Writer;
/// <summary>
/// Which order to send the data in relation to other packets.
/// </summary>
public DataOrderType OrderType;
/// <summary>
/// True if owner should be excluded.
/// </summary>
public bool ExcludeOwner;
public BufferedRpc(PooledWriter writer, DataOrderType orderType, bool excludeOwner)
{
Writer = writer;
OrderType = orderType;
ExcludeOwner = excludeOwner;
}
}
#endregion
#region Private.
#if UNITY_EDITOR
/// <summary>
/// Used to fetch RPC names for debug.
/// </summary>
private Dictionary<uint, string> _rpcNames;
#endif
/// <summary>
/// Registered ServerRpc methods.
/// </summary>
private readonly Dictionary<uint, ServerRpcDelegate> _serverRpcDelegates = new();
/// <summary>
/// Registered ObserversRpc methods.
/// </summary>
private readonly Dictionary<uint, ClientRpcDelegate> _observersRpcDelegates = new();
/// <summary>
/// Registered TargetRpc methods.
/// </summary>
private readonly Dictionary<uint, ClientRpcDelegate> _targetRpcDelegates = new();
/// <summary>
/// Number of total RPC methods for scripts in the same inheritance tree for this instance.
/// </summary>
private uint _rpcMethodCount;
/// <summary>
/// Size of every rpcHash for this networkBehaviour.
/// </summary>
private byte _rpcHashSize = 1;
/// <summary>
/// RPCs buffered for new clients.
/// </summary>
private readonly Dictionary<uint, BufferedRpc> _bufferedRpcs = new();
/// <summary>
/// Connections to exclude from RPCs, such as ExcludeOwner or ExcludeServer.
/// </summary>
private readonly HashSet<NetworkConnection> _networkConnectionCache = new();
/// <summary>
/// Used for debug output.
/// </summary>
private static StringBuilder _stringBuilder = new();
#endregion
#region Const.
/// <summary>
/// This is an estimated value of what the maximum possible size of a RPC could be.
/// Realistically this value is much smaller but this value is used as a buffer.
/// </summary>
private const int MAXIMUM_RPC_HEADER_SIZE = 10;
#if DEVELOPMENT
/// <summary>
/// Bytes used to write length for validating Rpc length.
/// </summary>
private const int VALIDATE_RPC_LENGTH_BYTES = 4;
#endif
#endregion
/// <summary>
/// Called when buffered RPCs should be sent.
/// </summary>
internal void SendBufferedRpcs(NetworkConnection conn)
{
TransportManager tm = _networkObjectCache.NetworkManager.TransportManager;
foreach (BufferedRpc bRpc in _bufferedRpcs.Values)
{
if (bRpc.ExcludeOwner && conn == Owner)
continue;
tm.SendToClient((byte)Channel.Reliable, bRpc.Writer.GetArraySegment(), conn, bRpc.OrderType);
}
}
/// <summary>
/// Registers a RPC method.
/// </summary>
/// <param name = "hash"></param>
/// <param name = "del"></param>
[APIExclude]
[MakePublic]
internal void RegisterServerRpc(uint hash, ServerRpcDelegate del)
{
AddRpcName(PacketId.ServerRpc, hash, del.Method.Name);
if (_serverRpcDelegates.TryAdd(hash, del))
IncreaseRpcMethodCount();
else
NetworkManager.LogError($"ServerRpc key {hash} has already been added for {GetType().FullName} on {gameObject.name}");
}
/// <summary>
/// Registers a RPC method.
/// </summary>
/// <param name = "hash"></param>
/// <param name = "del"></param>
[APIExclude]
[MakePublic]
internal void RegisterObserversRpc(uint hash, ClientRpcDelegate del)
{
AddRpcName(PacketId.ObserversRpc, hash, del.Method.Name);
if (_observersRpcDelegates.TryAdd(hash, del))
IncreaseRpcMethodCount();
else
NetworkManager.LogError($"ObserversRpc key {hash} has already been added for {GetType().FullName} on {gameObject.name}");
}
/// <summary>
/// Registers a RPC method.
/// </summary>
/// <param name = "hash"></param>
/// <param name = "del"></param>
[APIExclude]
[MakePublic]
internal void RegisterTargetRpc(uint hash, ClientRpcDelegate del)
{
AddRpcName(PacketId.TargetRpc, hash, del.Method.Name);
if (_targetRpcDelegates.TryAdd(hash, del))
IncreaseRpcMethodCount();
else
NetworkManager.LogError($"TargetRpc key {hash} has already been added for {GetType().FullName} on {gameObject.name}");
}
/// <summary>
/// Adds a RpcName for hash.
/// </summary>
private void AddRpcName(PacketId packetId, uint hash, string methodName)
{
#if UNITY_EDITOR
/* Maximum Rpc hash will be ushort.maxValue, and packetId will be
* well below that. Multiple packetId by ushort.MaxValue and add on
* hash. This is an inexpensive and quick way to put all hashes in one
* collection. */
uint value = (uint)((ushort)packetId * ushort.MaxValue) + hash;
if (_rpcNames == null)
_rpcNames = new();
_stringBuilder.Clear();
/* This parsing to get the name originally is kind of dirty
* but its only done by editor and won't cause issues
* if not found. */
const string indicator = "___";
int indicatorIndex = methodName.IndexOf(indicator, StringComparison.CurrentCultureIgnoreCase);
if (indicatorIndex < 0)
return;
// Trim start of first indicator.
methodName = methodName.Substring(indicatorIndex + indicator.Length);
indicatorIndex = methodName.IndexOf(indicator, StringComparison.CurrentCultureIgnoreCase);
if (indicatorIndex < 1)
return;
// Trim end of last indicator.
methodName = methodName.Substring(0, indicatorIndex);
_rpcNames[value] = methodName;
#endif
}
/// <summary>
/// Gets a RpcName for hash.
/// </summary>
private string GetRpcName(PacketId packetId, uint hash)
{
string result;
#if UNITY_EDITOR
if (_rpcNames == null)
return string.Empty;
// Set SetRpcName for why this is done.
uint value = (uint)((ushort)packetId * ushort.MaxValue) + hash;
_rpcNames.TryGetValueIL2CPP(value, out result);
#else
result = string.Empty;
#endif
return result;
}
/// <summary>
/// Increases rpcMethodCount and rpcHashSize.
/// </summary>
private void IncreaseRpcMethodCount()
{
_rpcMethodCount++;
if (_rpcMethodCount <= byte.MaxValue)
_rpcHashSize = 1;
else
_rpcHashSize = 2;
}
/// <summary>
/// Clears all buffered RPCs for this NetworkBehaviour.
/// </summary>
public void ClearBuffedRpcs()
{
foreach (BufferedRpc bRpc in _bufferedRpcs.Values)
bRpc.Writer.Store();
_bufferedRpcs.Clear();
}
/// <summary>
/// Reads a RPC hash.
/// </summary>
/// <param name = "reader"></param>
/// <returns></returns>
private uint ReadRpcHash(PooledReader reader)
{
if (_rpcHashSize == 1)
return reader.ReadUInt8Unpacked();
else
return reader.ReadUInt16();
}
/// <summary>
/// Called when a ServerRpc is received.
/// </summary>
internal void ReadServerRpc(int readerPositionAfterDebug, bool fromRpcLink, uint hash, PooledReader reader, NetworkConnection sendingClient, Channel channel)
{
if (!fromRpcLink)
hash = ReadRpcHash(reader);
if (sendingClient == null)
{
_networkObjectCache.NetworkManager.LogError($"NetworkConnection is null. ServerRpc {hash} on object {gameObject.name} [id {ObjectId}] will not complete. Remainder of packet may become corrupt.");
return;
}
if (_serverRpcDelegates.TryGetValueIL2CPP(hash, out ServerRpcDelegate data))
data.Invoke(reader, channel, sendingClient);
else
_networkObjectCache.NetworkManager.LogError($"ServerRpc not found for hash {hash} on object {gameObject.name} [id {ObjectId}]. Remainder of packet may become corrupt.");
#if !UNITY_SERVER
if (_networkTrafficStatistics != null)
_networkTrafficStatistics.AddInboundPacketIdData(PacketId.ServerRpc, GetRpcName(PacketId.ServerRpc, hash), reader.Position - readerPositionAfterDebug + TransportManager.PACKETID_LENGTH, gameObject, asServer: true);
#endif
}
/// <summary>
/// Called when an ObserversRpc is received.
/// </summary>
internal void ReadObserversRpc(int readerPositionAfterDebug, bool fromRpcLink, uint hash, PooledReader reader, Channel channel)
{
if (!fromRpcLink)
hash = ReadRpcHash(reader);
if (_observersRpcDelegates.TryGetValueIL2CPP(hash, out ClientRpcDelegate del))
del.Invoke(reader, channel);
else
_networkObjectCache.NetworkManager.LogError($"ObserversRpc not found for hash {hash} on object {gameObject.name} [id {ObjectId}] . Remainder of packet may become corrupt.");
#if !UNITY_SERVER
if (_networkTrafficStatistics != null)
_networkTrafficStatistics.AddInboundPacketIdData(PacketId.ObserversRpc, GetRpcName(PacketId.ObserversRpc, hash), reader.Position - readerPositionAfterDebug + TransportManager.PACKETID_LENGTH, gameObject, asServer: false);
#endif
}
/// <summary>
/// Called when an TargetRpc is received.
/// </summary>
internal void ReadTargetRpc(int readerPositionAfterDebug, bool fromRpcLink, uint hash, PooledReader reader, Channel channel)
{
if (!fromRpcLink)
hash = ReadRpcHash(reader);
if (_targetRpcDelegates.TryGetValueIL2CPP(hash, out ClientRpcDelegate del))
del.Invoke(reader, channel);
else
_networkObjectCache.NetworkManager.LogError($"TargetRpc not found for hash [{hash}] on gameObject [{gameObject.name}] ObjectId [{ObjectId}] NetworkBehaviour [{this.GetType().Name}]. The remainder of the packet may become corrupt.");
#if !UNITY_SERVER
if (_networkTrafficStatistics != null)
_networkTrafficStatistics.AddInboundPacketIdData(PacketId.TargetRpc, GetRpcName(PacketId.TargetRpc, hash), reader.Position - readerPositionAfterDebug + TransportManager.PACKETID_LENGTH, gameObject, asServer: false);
#endif
}
/// <summary>
/// Sends a RPC to server.
/// </summary>
/// <param name = "hash"></param>
/// <param name = "methodWriter"></param>
/// <param name = "channel"></param>
[MakePublic]
internal void SendServerRpc(uint hash, PooledWriter methodWriter, Channel channel, DataOrderType orderType)
{
if (!IsSpawnedWithWarning())
return;
channel = _transportManagerCache.GetReliableChannelIfOverMTU(methodWriter.Length + MAXIMUM_RPC_HEADER_SIZE, channel);
PooledWriter writer = CreateRpc(hash, methodWriter, PacketId.ServerRpc, channel);
#if DEVELOPMENT && !UNITY_SERVER
if (_networkTrafficStatistics != null)
_networkTrafficStatistics.AddOutboundPacketIdData(PacketId.ServerRpc, GetRpcName(PacketId.ServerRpc, hash), writer.Length, gameObject, asServer: false);
#endif
_networkObjectCache.NetworkManager.TransportManager.SendToServer((byte)channel, writer.GetArraySegment(), orderType);
writer.StoreLength();
}
/// <summary>
/// Sends a RPC to observers.
/// </summary>
/// <param name = "hash"></param>
/// <param name = "methodWriter"></param>
/// <param name = "channel"></param>
[APIExclude]
[MakePublic]
internal void SendObserversRpc(uint hash, PooledWriter methodWriter, Channel channel, DataOrderType orderType, bool bufferLast, bool excludeServer, bool excludeOwner)
{
if (!IsSpawnedWithWarning())
return;
channel = _transportManagerCache.GetReliableChannelIfOverMTU(methodWriter.Length + MAXIMUM_RPC_HEADER_SIZE, channel);
PooledWriter writer = lCreateRpc(channel);
SetNetworkConnectionCache(excludeServer, excludeOwner);
_networkObjectCache.NetworkManager.TransportManager.SendToClients((byte)channel, writer.GetArraySegment(), _networkObjectCache.Observers, _networkConnectionCache, orderType);
/* If buffered then dispose of any already buffered
* writers and replace with new one. Writers should
* automatically dispose when references are lost
* anyway but better safe than sorry. */
if (bufferLast)
{
if (_bufferedRpcs.TryGetValueIL2CPP(hash, out BufferedRpc result))
result.Writer.StoreLength();
/* If sent on unreliable the RPC has to be rebuilt for
* reliable headers since buffered RPCs always send reliably
* to new connections. */
if (channel == Channel.Unreliable)
{
writer.StoreLength();
writer = lCreateRpc(Channel.Reliable);
}
_bufferedRpcs[hash] = new(writer, orderType, excludeOwner);
}
// If not buffered then dispose immediately.
else
{
writer.StoreLength();
}
PooledWriter lCreateRpc(Channel c)
{
#if DEVELOPMENT
if (!NetworkManager.DebugManager.DisableObserversRpcLinks && _rpcLinks.TryGetValueIL2CPP(hash, out RpcLinkType link))
#else
if (_rpcLinks.TryGetValueIL2CPP(hash, out RpcLinkType link))
#endif
writer = CreateLinkedRpc(link, methodWriter, c);
else
writer = CreateRpc(hash, methodWriter, PacketId.ObserversRpc, c);
#if DEVELOPMENT && !UNITY_SERVER
if (_networkTrafficStatistics != null)
{
int written = writer.Length * _networkObjectCache.Observers.Count;
_networkTrafficStatistics.AddOutboundPacketIdData(PacketId.ObserversRpc, GetRpcName(PacketId.ObserversRpc, hash), written, gameObject, asServer: true);
}
#endif
return writer;
}
}
/// <summary>
/// Sends a RPC to target.
/// </summary>
[MakePublic]
internal void SendTargetRpc(uint hash, PooledWriter methodWriter, Channel channel, DataOrderType orderType, NetworkConnection target, bool excludeServer, bool validateTarget = true)
{
if (!IsSpawnedWithWarning())
return;
channel = _transportManagerCache.GetReliableChannelIfOverMTU(methodWriter.Length + MAXIMUM_RPC_HEADER_SIZE, channel);
if (validateTarget)
{
if (target == null)
{
_networkObjectCache.NetworkManager.LogWarning($"Action cannot be completed as no Target is specified.");
return;
}
else
{
// If target is not an observer.
if (!_networkObjectCache.Observers.Contains(target))
{
_networkObjectCache.NetworkManager.LogWarning($"Action cannot be completed as Target is not an observer for object {gameObject.name} [id {ObjectId}].");
return;
}
}
}
// Excluding server.
if (excludeServer && target.IsLocalClient)
return;
PooledWriter writer;
#if DEVELOPMENT
if (!NetworkManager.DebugManager.DisableTargetRpcLinks && _rpcLinks.TryGetValueIL2CPP(hash, out RpcLinkType link))
#else
if (_rpcLinks.TryGetValueIL2CPP(hash, out RpcLinkType link))
#endif
writer = CreateLinkedRpc(link, methodWriter, channel);
else
writer = CreateRpc(hash, methodWriter, PacketId.TargetRpc, channel);
#if DEVELOPMENT && !UNITY_SERVER
if (_networkTrafficStatistics != null)
_networkTrafficStatistics.AddOutboundPacketIdData(PacketId.TargetRpc, GetRpcName(PacketId.TargetRpc, hash), writer.Length, gameObject, asServer: true);
#endif
_networkObjectCache.NetworkManager.TransportManager.SendToClient((byte)channel, writer.GetArraySegment(), target, orderType);
writer.Store();
}
/// <summary>
/// Adds excluded connections to ExcludedRpcConnections.
/// </summary>
private void SetNetworkConnectionCache(bool addClientHost, bool addOwner)
{
_networkConnectionCache.Clear();
if (addClientHost && IsClientStarted)
_networkConnectionCache.Add(LocalConnection);
if (addOwner && Owner.IsValid)
_networkConnectionCache.Add(Owner);
}
/// <summary>
/// Returns if spawned and throws a warning if not.
/// </summary>
/// <returns></returns>
private bool IsSpawnedWithWarning()
{
bool result = IsSpawned;
if (!result)
_networkObjectCache.NetworkManager.LogWarning($"Action cannot be completed as object {gameObject.name} [Id {ObjectId}] is not spawned.");
return result;
}
/// <summary>
/// Writes a full RPC and returns the writer.
/// </summary>
private PooledWriter CreateRpc(uint hash, PooledWriter methodWriter, PacketId packetId, Channel channel)
{
int rpcHeaderBufferLength = GetEstimatedRpcHeaderLength();
int methodWriterLength = methodWriter.Length;
// Writer containing full packet.
PooledWriter writer = WriterPool.Retrieve(rpcHeaderBufferLength + methodWriterLength);
writer.WritePacketIdUnpacked(packetId);
#if DEVELOPMENT
int written = WriteDebugForValidateRpc(writer, packetId, hash);
#endif
writer.WriteNetworkBehaviour(this);
// Only write length if reliable.
if (channel == Channel.Reliable)
writer.WriteInt32(methodWriterLength + _rpcHashSize);
// Hash and data.
WriteRpcHash(hash, writer);
writer.WriteArraySegment(methodWriter.GetArraySegment());
#if DEVELOPMENT
WriteDebugLengthForValidateRpc(writer, written);
#endif
return writer;
}
#if DEVELOPMENT
/// <summary>
/// Gets the method name for a Rpc using packetId and Rpc hash.
/// </summary>
private string GetRpcMethodName(PacketId packetId, uint hash)
{
try
{
if (packetId == PacketId.ObserversRpc)
return _observersRpcDelegates[hash].Method.Name;
else if (packetId == PacketId.TargetRpc)
return _targetRpcDelegates[hash].Method.Name;
else if (packetId == PacketId.ServerRpc)
return _serverRpcDelegates[hash].Method.Name;
else if (packetId == PacketId.Replicate)
return _replicateRpcDelegates[hash].Method.Name;
else if (packetId == PacketId.Reconcile)
return _reconcileRpcDelegates[hash].Method.Name;
else
_networkObjectCache.NetworkManager.LogError($"Unhandled packetId of {packetId} for hash {hash}.");
}
// This should not ever happen.
catch
{
_networkObjectCache.NetworkManager.LogError($"Rpc method name not found for packetId {packetId}, hash {hash}.");
}
return "Error";
}
#endif
/// <summary>
/// Writes rpcHash to writer.
/// </summary>
/// <param name = "hash"></param>
/// <param name = "writer"></param>
private void WriteRpcHash(uint hash, PooledWriter writer)
{
if (_rpcHashSize == 1)
writer.WriteUInt8Unpacked((byte)hash);
else
writer.WriteUInt16((byte)hash);
}
#if DEVELOPMENT
private int WriteDebugForValidateRpc(Writer writer, PacketId packetId, uint hash)
{
if (!_networkObjectCache.NetworkManager.DebugManager.ValidateRpcLengths)
return -1;
writer.Skip(VALIDATE_RPC_LENGTH_BYTES);
int positionStart = writer.Position;
string txt = $"NetworkObject Details: {_networkObjectCache.ToString()}. NetworkBehaviour Details: Name [{GetType().Name}]. Rpc Details: Name [{GetRpcMethodName(packetId, hash)}] PacketId [{packetId}] Hash [{hash}]";
writer.WriteString(txt);
return positionStart;
}
private void WriteDebugLengthForValidateRpc(Writer writer, int positionStart)
{
if (!_networkObjectCache.NetworkManager.DebugManager.ValidateRpcLengths)
return;
// Write length.
int writtenLength = writer.Position - positionStart;
writer.InsertInt32Unpacked(writtenLength, positionStart - VALIDATE_RPC_LENGTH_BYTES);
}
/// <summary>
/// Parses written data used to validate a Rpc packet.
/// </summary>
internal static void ReadDebugForValidatedRpc(NetworkManager manager, PooledReader reader, out int readerRemainingAfterLength, out string rpcInformation, out uint expectedReadAmount)
{
rpcInformation = null;
expectedReadAmount = 0;
readerRemainingAfterLength = 0;
if (!manager.DebugManager.ValidateRpcLengths)
return;
expectedReadAmount = (uint)reader.ReadInt32Unpacked();
readerRemainingAfterLength = reader.Remaining;
rpcInformation = reader.ReadStringAllocated();
}
/// <summary>
/// Prints an error if an Rpc packet did not validate correctly.
/// </summary>
/// <returns>True if an error occurred.</returns>
internal static bool TryPrintDebugForValidatedRpc(bool fromRpcLink, NetworkManager manager, PooledReader reader, int startReaderRemaining, string rpcInformation, uint expectedReadAmount, Channel channel)
{
if (!manager.DebugManager.ValidateRpcLengths)
return false;
int readAmount = startReaderRemaining - reader.Remaining;
if (readAmount != expectedReadAmount)
{
string src = fromRpcLink ? "RpcLink" : "Rpc";
string msg = $"A {src} read an incorrect amount of data on channel {channel}. Read length was {readAmount}, expected length is {expectedReadAmount}. {rpcInformation}." + $" {manager.PacketIdHistory.GetReceivedPacketIds(packetsFromServer: reader.Source == Reader.DataSource.Server)}.";
manager.LogError(msg);
return true;
}
return false;
}
#endif
}
}