Merge pull request #2763 from masagrator/patch-1

Fix off by 1 bug + allocation error + missing initialize in PGL

libraries/libstratosphere/source/pgl/srv/pgl_srv_shell_host_utils.cpp

@@ -77,7 +77,7 @@ namespace ams::pgl::srv {
             private:
                 char m_content_path[fs::EntryNameLengthMax]   = {};
                 ExtensionType m_extension_type                = ExtensionType::None;
-                char m_mount_name[fs::MountNameLengthMax]   = {};
+                char m_mount_name[fs::MountNameLengthMax + 1] = {};
                 bool m_is_mounted                             = false;
                 ncm::AutoBuffer m_content_meta_buffer;
                 ncm::ProgramId m_program_id                   = ncm::InvalidProgramId;

stratosphere/pgl/source/pgl_main.cpp

@@ -37,6 +37,7 @@ namespace ams {
             R_ABORT_UNLESS(pmshellInitialize());
             R_ABORT_UNLESS(ldrShellInitialize());
             R_ABORT_UNLESS(lrInitialize());
+            lr::Initialize();
 
             /* Verify that we can sanely execute. */
             ams::CheckApiVersion();
@@ -58,3 +59,36 @@ namespace ams {
     }
 
 }
+
+/* Override operator new. */
+void *operator new(size_t size) {
+    return ams::pgl::srv::Allocate(size);
+}
+
+void *operator new(size_t size, const std::nothrow_t &) {
+    return ams::pgl::srv::Allocate(size);
+}
+
+void operator delete(void *p) {
+    return ams::pgl::srv::Deallocate(p, 0);
+}
+
+void operator delete(void *p, size_t size) {
+    return ams::pgl::srv::Deallocate(p, size);
+}
+
+void *operator new[](size_t size) {
+    return ams::pgl::srv::Allocate(size);
+}
+
+void *operator new[](size_t size, const std::nothrow_t &) {
+    return ams::pgl::srv::Allocate(size);
+}
+
+void operator delete[](void *p) {
+    return ams::pgl::srv::Deallocate(p, 0);
+}
+
+void operator delete[](void *p, size_t size) {
+    return ams::pgl::srv::Deallocate(p, size);
+}