shutdown

immisterio · Jan 1, 2023 · 5953a00 · 5953a00
1 parent 6266ba8
commit 5953a00
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/Engine/Middlewares/Accs.cs b/Engine/Middlewares/Accs.cs
@@ -61,12 +61,6 @@ public Task Invoke(HttpContext httpContext)
                 return _next(httpContext);
             #endregion
 
-            #region Методы работающие без авторизации
-            if (httpContext.Request.Path.Value.StartsWith("/shutdown"))
-                return httpContext.Response.WriteAsync("");
-            #endregion
-
-
             if (Startup.settings.AuthorizationRequired)
             {
                 #region Авторизация по домену

diff --git a/Engine/Middlewares/TorAPI.cs b/Engine/Middlewares/TorAPI.cs
@@ -87,6 +87,9 @@ async public Task InvokeAsync(HttpContext httpContext)
 
             var userData = httpContext.Features.Get<UserData>();
 
+            if (!userData.shutdown && httpContext.Request.Path.Value.StartsWith("/shutdown"))
+                return;
+
             string dbKeyOrLogin = userData.login;
             if (userData.IsShared)
                 dbKeyOrLogin = $"{userData.login}:{httpContext.Connection.RemoteIpAddress}";
@@ -280,6 +283,12 @@ async public Task InvokeAsync(HttpContext httpContext)
                 await CopyProxyHttpResponse(httpContext, response, info);
             }
             #endregion
+
+            if (httpContext.Request.Path.Value.StartsWith("/shutdown")) 
+            {
+                db.TryRemove(dbKeyOrLogin, out _);
+                info.Dispose();
+            }
         }
 
 

diff --git a/Models/UserData.cs b/Models/UserData.cs
@@ -21,6 +21,9 @@ public class UserData
         [JsonIgnore]
         public bool IsShared { get; set; }
 
+        [JsonIgnore]
+        public bool shutdown { get; set; }
+
         [JsonIgnore]
         public byte maxiptoIsLockHostOrUser { get; set; }
     }