Merge branch 'main' into emilia-t
commit
7c05911921
@ -0,0 +1,104 @@
|
||||
using OnlineMsgServer.Core;
|
||||
using WebSocketSharp.Server;
|
||||
|
||||
namespace OnlineMsgServer.Common
|
||||
{
|
||||
class RenameMessage : Message
|
||||
{
|
||||
public RenameMessage()
|
||||
{
|
||||
Type = "rename";
|
||||
Key = "";
|
||||
}
|
||||
|
||||
public override Task Handler(string wsid, WebSocketSessionManager Sessions)
|
||||
{
|
||||
return Task.Run(() =>
|
||||
{
|
||||
try
|
||||
{
|
||||
if (!UserService.IsAuthenticated(wsid))
|
||||
{
|
||||
Log.Security("rename_denied_unauthenticated", $"wsid={wsid}");
|
||||
return;
|
||||
}
|
||||
|
||||
string key = Key?.Trim() ?? "";
|
||||
if (!SignedMessagePayload.TryParse(Data, out SignedMessagePayload payload, out string parseError))
|
||||
{
|
||||
Log.Security("rename_payload_invalid", $"wsid={wsid} reason={parseError}");
|
||||
SendEncryptedResult(Sessions, wsid, "rename_error", "rename payload invalid");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!SecurityValidator.VerifySignedMessage(wsid, Type, key, payload, out string securityReason))
|
||||
{
|
||||
Log.Security("rename_security_failed", $"wsid={wsid} reason={securityReason}");
|
||||
SendEncryptedResult(Sessions, wsid, "rename_error", "rename signature invalid");
|
||||
return;
|
||||
}
|
||||
|
||||
string nextName = payload.Payload.Trim();
|
||||
if (string.IsNullOrWhiteSpace(nextName))
|
||||
{
|
||||
Log.Security("rename_invalid_name", $"wsid={wsid} reason=blank");
|
||||
SendEncryptedResult(Sessions, wsid, "rename_error", "display name cannot be empty");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!UserService.IsPeerNodeSession(wsid) && PeerNetworkService.IsPeerUserName(nextName))
|
||||
{
|
||||
Log.Security("rename_invalid_name", $"wsid={wsid} reason=peer_prefix");
|
||||
SendEncryptedResult(Sessions, wsid, "rename_error", "display name uses reserved prefix");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!UserService.TryUpdateUserName(wsid, nextName, out string appliedName))
|
||||
{
|
||||
Log.Security("rename_update_failed", $"wsid={wsid}");
|
||||
SendEncryptedResult(Sessions, wsid, "rename_error", "display name update failed");
|
||||
return;
|
||||
}
|
||||
|
||||
Log.Security("rename_success", $"wsid={wsid} user={appliedName}");
|
||||
SendEncryptedResult(Sessions, wsid, "rename_ok", appliedName);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Log.Security("rename_error", $"wsid={wsid} error={ex.Message}");
|
||||
SendEncryptedResult(Sessions, wsid, "rename_error", "display name update failed");
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private static void SendEncryptedResult(
|
||||
WebSocketSessionManager sessions,
|
||||
string wsid,
|
||||
string type,
|
||||
string data
|
||||
)
|
||||
{
|
||||
string? publicKey = UserService.GetUserPublicKeyByID(wsid);
|
||||
if (string.IsNullOrWhiteSpace(publicKey))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
Message result = new()
|
||||
{
|
||||
Type = type,
|
||||
Data = data
|
||||
};
|
||||
string encrypted = RsaService.EncryptForClient(publicKey, result.ToJsonString());
|
||||
|
||||
foreach (IWebSocketSession session in sessions.Sessions)
|
||||
{
|
||||
if (session.ID == wsid)
|
||||
{
|
||||
session.Context.WebSocket.Send(encrypted);
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue