From d2461e190821e70fb15f1f9f4372f23f182761fb Mon Sep 17 00:00:00 2001
From: Grider <65988663+himekifee@users.noreply.github.com>
Date: Sun, 5 Feb 2023 23:41:35 +0000
Subject: [PATCH] Fix wayland capture on nvidia for kms (#884)

---
 src/platform/linux/kmsgrab.cpp | 6 ++++++
 src/platform/linux/wlgrab.cpp  | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/platform/linux/kmsgrab.cpp b/src/platform/linux/kmsgrab.cpp
index 240ce696..6ca7220c 100644
--- a/src/platform/linux/kmsgrab.cpp
+++ b/src/platform/linux/kmsgrab.cpp
@@ -755,6 +755,12 @@ public:
     auto &rgb = *rgb_opt;
 
     gl::ctx.BindTexture(GL_TEXTURE_2D, rgb->tex[0]);
+
+    int w, h;
+    gl::ctx.GetTexLevelParameteriv(GL_TEXTURE_2D, 0, GL_TEXTURE_WIDTH, &w);
+    gl::ctx.GetTexLevelParameteriv(GL_TEXTURE_2D, 0, GL_TEXTURE_HEIGHT, &h);
+    BOOST_LOG(debug) << "width and height: w "sv << w << " h "sv << h;
+
     gl::ctx.GetTextureSubImage(rgb->tex[0], 0, img_offset_x, img_offset_y, 0, width, height, 1, GL_BGRA, GL_UNSIGNED_BYTE, img_out_base->height * img_out_base->row_pitch, img_out_base->data);
 
     if(cursor_opt && cursor) {
diff --git a/src/platform/linux/wlgrab.cpp b/src/platform/linux/wlgrab.cpp
index ea4adc8d..00a34c28 100644
--- a/src/platform/linux/wlgrab.cpp
+++ b/src/platform/linux/wlgrab.cpp
@@ -167,7 +167,7 @@ public:
 
     int w, h;
     gl::ctx.GetTexLevelParameteriv(GL_TEXTURE_2D, 0, GL_TEXTURE_WIDTH, &w);
-    gl::ctx.GetTexLevelParameteriv(GL_TEXTURE_2D, 0, GL_TEXTURE_WIDTH, &h);
+    gl::ctx.GetTexLevelParameteriv(GL_TEXTURE_2D, 0, GL_TEXTURE_HEIGHT, &h);
     BOOST_LOG(debug) << "width and height: w "sv << w << " h "sv << h;
 
     gl::ctx.GetTextureSubImage((*rgb_opt)->tex[0], 0, 0, 0, 0, width, height, 1, GL_BGRA, GL_UNSIGNED_BYTE, img_out_base->height * img_out_base->row_pitch, img_out_base->data);