Revert "fix: System shared memory boundary check (#8335) (#8337)"

This reverts commit dfd3148.

Author: mc-nv

qa/L0_shared_memory/shared_memory_test.py

@@ -47,7 +47,6 @@
 
 class SystemSharedMemoryTestBase(tu.TestResultCollector):
     DEFAULT_SHM_BYTE_SIZE = 64
-    SYS_PAGE_SIZE = os.sysconf("SC_PAGE_SIZE")
 
     def setUp(self):
         self._setup_client()
@@ -306,10 +305,11 @@ def test_large_shm_register_offset(self):
 
             # Test for large offset
             error_msg = []
+            page_size = os.sysconf("SC_PAGE_SIZE")
             # Create a large shm size (page_size * 1024 is large enough to reproduce a segfault).
             # Register offset at 1 page before the end of the shm region to give enough space for the input/output data.
-            create_byte_size = self.SYS_PAGE_SIZE * 1024
-            register_offset = self.SYS_PAGE_SIZE * 1023
+            create_byte_size = page_size * 1024
+            register_offset = page_size * 1023
             self._configure_server(
                 create_byte_size=create_byte_size,
                 register_offset=register_offset,
@@ -372,12 +372,7 @@ def test_unregisterall(self):
     def test_infer_offset_out_of_bound(self):
         # Shared memory offset outside output region - Throws error
         error_msg = []
-        create_byte_size = self.SYS_PAGE_SIZE + self.DEFAULT_SHM_BYTE_SIZE
-        register_offset = self.SYS_PAGE_SIZE
-        self._configure_server(
-            create_byte_size=create_byte_size,
-            register_offset=register_offset,
-        )
+        self._configure_server()
         if self.protocol == "http":
             # -32 when placed in an int64 signed type, to get a negative offset
             # by overflowing
@@ -407,13 +402,8 @@ def test_infer_offset_out_of_bound(self):
     def test_infer_byte_size_out_of_bound(self):
         # Shared memory byte_size outside output region - Throws error
         error_msg = []
-        create_byte_size = self.SYS_PAGE_SIZE + self.DEFAULT_SHM_BYTE_SIZE
-        register_offset = self.SYS_PAGE_SIZE
-        self._configure_server(
-            create_byte_size=create_byte_size,
-            register_offset=register_offset,
-        )
-        offset = 1
+        self._configure_server()
+        offset = 60
         byte_size = self.DEFAULT_SHM_BYTE_SIZE
 
         iu.shm_basic_infer(

src/shared_memory_manager.cc

@@ -475,11 +475,14 @@ SharedMemoryManager::GetMemoryInfo(
   }
 
   // validate offset
-  size_t shm_region_size = 0;
+  size_t shm_region_end = 0;
+  if (it->second->kind_ == TRITONSERVER_MEMORY_CPU) {
+    shm_region_end = it->second->offset_;
+  }
   if (it->second->byte_size_ > 0) {
-    shm_region_size += it->second->byte_size_;
+    shm_region_end += it->second->byte_size_ - 1;
   }
-  if (offset >= shm_region_size) {
+  if (offset > shm_region_end) {
     return TRITONSERVER_ErrorNew(
         TRITONSERVER_ERROR_INVALID_ARG,
         std::string("Invalid offset for shared memory region: '" + name + "'")
@@ -499,8 +502,8 @@ SharedMemoryManager::GetMemoryInfo(
   }
 
   // validate byte_size + offset is within memory bounds
-  size_t total_req_shm = offset + byte_size;
-  if (total_req_shm > shm_region_size) {
+  size_t total_req_shm = offset + byte_size - 1;
+  if (total_req_shm > shm_region_end) {
     return TRITONSERVER_ErrorNew(
         TRITONSERVER_ERROR_INVALID_ARG,
         std::string(