From 056d0b43831857c7258fa734d0ecac05858fa449 Mon Sep 17 00:00:00 2001
From: Sadie Powell
Date: Sun, 27 Mar 2022 07:57:50 +0100
Subject: Pack the mode set more efficiently and add support for numeric modes.

---
 src/mode.cpp | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

(limited to 'src/mode.cpp')

diff --git a/src/mode.cpp b/src/mode.cpp
index 17a46da3c..a005f81fb 100644
--- a/src/mode.cpp
+++ b/src/mode.cpp
@@ -600,7 +600,7 @@ void ModeParser::AddMode(ModeHandler* mh)
 		}
 	}
 
-	ModeHandler*& slot = modehandlers[mh->GetModeType()][mh->GetModeChar()-65];
+	ModeHandler*& slot = modehandlers[mh->GetModeType()][ModeParser::GetModeIndex(mh->GetModeChar())];
 	if (slot)
 	{
 		throw ModuleException(mh->creator, InspIRCd::Format("Mode letter for %s already used by %s from %s: %c",
@@ -647,7 +647,7 @@ bool ModeParser::DelMode(ModeHandler* mh)
 	if ((mhmapit == mhmap.end()) || (mhmapit->second != mh))
 		return false;
 
-	ModeHandler*& slot = modehandlers[mh->GetModeType()][mh->GetModeChar()-65];
+	ModeHandler*& slot = modehandlers[mh->GetModeType()][ModeParser::GetModeIndex(mh->GetModeChar())];
 	if (slot != mh)
 		return false;
 
@@ -710,7 +710,7 @@ ModeHandler* ModeParser::FindMode(unsigned char modeletter, ModeType mt)
 	if (!ModeParser::IsModeChar(modeletter))
 		return NULL;
 
-	return modehandlers[mt][modeletter-65];
+	return modehandlers[mt][ModeParser::GetModeIndex(modeletter)];
 }
 
 PrefixMode* ModeParser::FindPrefixMode(unsigned char modeletter)
@@ -785,7 +785,25 @@ void PrefixMode::RemoveMode(Channel* chan, Modes::ChangeList& changelist)
 
 bool ModeParser::IsModeChar(char chr)
 {
-	return ((chr >= 'A' && chr <= 'Z') || (chr >= 'a' && chr <= 'z'));
+	return ((chr >= '0' && chr <= '9') || (chr >= 'A' && chr <= 'Z') || (chr >= 'a' && chr <= 'z'));
+}
+
+size_t ModeParser::GetModeIndex(char chr)
+{
+	// Bitset layout:
+	//   0123456789                 = 10 [0-9]
+	//   ABCDEFGHIJKLMNOPQRSTUVWXYZ = 26 [10-35]
+	//   abcdefghijklmnopqrstuvwxyz = 26 [36-61]
+	if (chr >= '0' && chr <= '9')
+		return chr - '0';
+
+	if (chr >= 'A' && chr <= 'Z')
+		return chr - 'A' + 10; // [0-9] = 10
+
+	if (chr >= 'a' && chr <= 'z')
+		return chr - 'a' + 36; // [0-9]+[A-Z] = 10+26 = 36
+
+	return ModeParser::MODEID_MAX;
 }
 
 ModeParser::ModeParser()
-- 
cgit v1.3.1-10-gc9f91