From 2b619b00851a0cceb3c616209af0a6e78d351071 Mon Sep 17 00:00:00 2001
From: probonopd <probonopd@users.noreply.github.com>
Date: Sat, 16 Nov 2024 23:09:47 +0000
Subject: [PATCH] /SD/wpa_supplicant.conf

---
 src/net/ftpworker.cpp | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/net/ftpworker.cpp b/src/net/ftpworker.cpp
index fb57254..0dcaa04 100644
--- a/src/net/ftpworker.cpp
+++ b/src/net/ftpworker.cpp
@@ -27,6 +27,7 @@
 #include <circle/net/netsubsystem.h>
 #include <circle/sched/scheduler.h>
 #include <circle/timer.h>
+#include <circle/startup.h>
 #include <fatfs/ff.h>
 
 #include <cstdio>
@@ -618,7 +619,7 @@ bool CFTPWorker::Retrieve(const char* pArgs)
 	CString Path = RealPath(pArgs);
 
 	// If the filename is "wpa_supplicant.conf", don't allow it to be retrieved
-	if (strcmp(Path, "/wpa_supplicant.conf") == 0)
+	if (strcmp(Path, "/SD/wpa_supplicant.conf") == 0)
 	{
 		SendStatus(TFTPStatus::FileActionNotTaken, "File action not taken.");
 		return false;
@@ -993,7 +994,7 @@ bool CFTPWorker::ListFileNames(const char* pArgs)
 			const TDirectoryListEntry& Entry = pDirEntries[i];
 			if (Entry.Type == TDirectoryListEntryType::Directory)
 				continue;
-			if (strcmp(Entry.Name, "wpa_supplicant.conf") == 0)
+			if (strcmp(Entry.Name, "/SD/wpa_supplicant.conf") == 0)
 			{
 				continue;
 			}
@@ -1055,6 +1056,9 @@ bool CFTPWorker::Bye(const char* pArgs)
 	SendStatus(TFTPStatus::ClosingControl, "Goodbye.");
 	delete m_pControlSocket;
 	m_pControlSocket = nullptr;
+	
+	// Reboot the system if the user disconnects in order to apply any changes made
+	reboot ();
 	return true;
 }