Validate that structured cache bucket identifiers are alphanumeric

1 files changed, 23 insertions, 2 deletions

diff --git a/zenserver-test/zenserver-test.cpp b/zenserver-test/zenserver-test.cpp
index 6600af6de..114c3e86b 100644
--- a/zenserver-test/zenserver-test.cpp
+++ b/zenserver-test/zenserver-test.cpp
@@ -1097,6 +1097,8 @@ TEST_CASE("z$.basic")
 	const int  kIterationCount = 100;
 	const auto BaseUri		   = "http://localhost:{}/z$"_format(PortNumber);
 
+	auto HashKey = [](int i) -> zen::IoHash { return zen::IoHash::HashMemory(&i, sizeof i); };
+
 	{
 		ZenServerInstance Instance1(TestEnv);
 		Instance1.SetTestDir(TestDir);
@@ -1114,7 +1116,7 @@ TEST_CASE("z$.basic")
 			zen::BinaryWriter	 Writer{MemOut};
 			Cbo.Save(Writer);
 
-			zen::IoHash Key = zen::IoHash::HashMemory(&i, sizeof i);
+			zen::IoHash Key = HashKey(i);
 
 			cpr::Response Result = cpr::Put(cpr::Url{"{}/{}/{}"_format(BaseUri, "test", Key)},
 											cpr::Body{(const char*)MemOut.Data(), MemOut.Size()},
@@ -1133,6 +1135,25 @@ TEST_CASE("z$.basic")
 
 			CHECK(Result.status_code == 200);
 		}
+
+		// Ensure bad bucket identifiers are rejected
+
+		{
+			zen::CbObjectWriter Cbo;
+			Cbo << "index" << 42;
+
+			zen::MemoryOutStream MemOut;
+			zen::BinaryWriter	 Writer{MemOut};
+			Cbo.Save(Writer);
+
+			zen::IoHash Key = HashKey(442);
+
+			cpr::Response Result = cpr::Put(cpr::Url{"{}/{}/{}"_format(BaseUri, "te!st", Key)},
+											cpr::Body{(const char*)MemOut.Data(), MemOut.Size()},
+											cpr::Header{{"Content-Type", "application/x-ue-cb"}});
+
+			CHECK(Result.status_code == 400);
+		}
 	}
 
 	// Verify that the data persists between process runs (the previous server has exited at this point)
@@ -1147,7 +1168,7 @@ TEST_CASE("z$.basic")
 
 		for (int i = 0; i < kIterationCount; ++i)
 		{
-			zen::IoHash Key = zen::IoHash::HashMemory(&i, sizeof i);
+			zen::IoHash Key = HashKey(i);
 
 			cpr::Response Result = cpr::Get(cpr::Url{"{}/{}/{}"_format(BaseUri, "test", Key)});