From f1b34d716f5c7a3bced8fb651972007de10081b7 Mon Sep 17 00:00:00 2001 From: Markus Valentin Date: Thu, 9 Mar 2023 10:57:14 +0100 Subject: [PATCH] login-common: Add shutting_down and resource_constraint fields to client struct --- src/login-common/client-common.c | 7 +++++++ src/login-common/client-common.h | 2 ++ 2 files changed, 9 insertions(+) diff --git a/src/login-common/client-common.c b/src/login-common/client-common.c index 03a5ad207a..028929bd51 100644 --- a/src/login-common/client-common.c +++ b/src/login-common/client-common.c @@ -1272,6 +1272,13 @@ void client_notify_disconnect(struct client *client, if (!client->notified_disconnect) { if (client->v.notify_disconnect != NULL) client->v.notify_disconnect(client, reason, text); + if (reason == CLIENT_DISCONNECT_SYSTEM_SHUTDOWN) { + client->shutting_down = TRUE; + } + if (reason == CLIENT_DISCONNECT_RESOURCE_CONSTRAINT) { + client->shutting_down = TRUE; + client->resource_constraint = TRUE; + } client->notified_disconnect = TRUE; } } diff --git a/src/login-common/client-common.h b/src/login-common/client-common.h index 48db31cc5d..39fdd3ee24 100644 --- a/src/login-common/client-common.h +++ b/src/login-common/client-common.h @@ -285,6 +285,8 @@ struct client { bool notified_auth_ready:1; bool notified_disconnect:1; bool fd_proxying:1; + bool shutting_down:1; + bool resource_constraint:1; /* ... */ }; -- 2.47.3