Jacek Caban : conhost: Implement IOCTL_CONDRV_READ_OUTPUT.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: f4661f1b38c479bd08baee93d9ddb1f9e93c173d
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=f4661f1b38c479bd08baee93d9ddb1f9e93c173d

Author: Jacek Caban <jacek at codeweavers.com>
Date:   Fri Aug 28 14:27:05 2020 +0200

conhost: Implement IOCTL_CONDRV_READ_OUTPUT.

Signed-off-by: Jacek Caban <jacek at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 programs/conhost/conhost.c | 82 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)

diff --git a/programs/conhost/conhost.c b/programs/conhost/conhost.c
index 4fbfc19a08..d17a8f4e68 100644
--- a/programs/conhost/conhost.c
+++ b/programs/conhost/conhost.c
@@ -545,6 +545,84 @@ static NTSTATUS write_output( struct screen_buffer *screen_buffer, const struct
     return STATUS_SUCCESS;
 }
 
+static NTSTATUS read_output( struct screen_buffer *screen_buffer, const struct condrv_output_params *params,
+                             size_t *out_size )
+{
+    enum char_info_mode mode;
+    unsigned int x, y, width;
+    unsigned int i, count;
+
+    x = params->x;
+    y = params->y;
+    mode  = params->mode;
+    width = params->width;
+    TRACE( "(%u %u) mode %u width %u\n", x, y, mode, width );
+
+    switch(mode)
+    {
+    case CHAR_INFO_MODE_TEXT:
+        {
+            WCHAR *data;
+            char_info_t *src;
+            if (x >= screen_buffer->width || y >= screen_buffer->height)
+            {
+                *out_size = 0;
+                return STATUS_SUCCESS;
+            }
+            src = screen_buffer->data + y * screen_buffer->width + x;
+            count = min( screen_buffer->data + screen_buffer->height * screen_buffer->width - src,
+                         *out_size / sizeof(*data) );
+            *out_size = count * sizeof(*data);
+            if (!(data = alloc_ioctl_buffer( *out_size ))) return STATUS_NO_MEMORY;
+            for (i = 0; i < count; i++) data[i] = src[i].ch;
+        }
+        break;
+    case CHAR_INFO_MODE_ATTR:
+        {
+            unsigned short *data;
+            char_info_t *src;
+            if (x >= screen_buffer->width || y >= screen_buffer->height)
+            {
+                *out_size = 0;
+                return STATUS_SUCCESS;
+            }
+            src = screen_buffer->data + y * screen_buffer->width + x;
+            count = min( screen_buffer->data + screen_buffer->height * screen_buffer->width - src,
+                         *out_size / sizeof(*data) );
+            *out_size = count * sizeof(*data);
+            if (!(data = alloc_ioctl_buffer( *out_size ))) return STATUS_NO_MEMORY;
+            for (i = 0; i < count; i++) data[i] = src[i].attr;
+        }
+        break;
+    case CHAR_INFO_MODE_TEXTATTR:
+        {
+            SMALL_RECT *region;
+            char_info_t *data;
+            if (!width || *out_size < sizeof(*region) || x >= screen_buffer->width || y >= screen_buffer->height)
+                return STATUS_INVALID_PARAMETER;
+            count = min( (*out_size - sizeof(*region)) / (width * sizeof(*data)), screen_buffer->height - y );
+            width = min( width, screen_buffer->width - x );
+            *out_size = sizeof(*region) + width * count * sizeof(*data);
+            if (!(region = alloc_ioctl_buffer( *out_size ))) return STATUS_NO_MEMORY;
+            region->Left   = x;
+            region->Top    = y;
+            region->Right  = x + width - 1;
+            region->Bottom = y + count - 1;
+            data = (char_info_t *)(region + 1);
+            for (i = 0; i < count; i++)
+            {
+                memcpy( &data[i * width], &screen_buffer->data[(y + i) * screen_buffer->width + x],
+                        width * sizeof(*data) );
+            }
+        }
+        break;
+    default:
+        return STATUS_INVALID_PARAMETER;
+    }
+
+    return STATUS_SUCCESS;
+}
+
 static NTSTATUS set_console_title( struct console *console, const WCHAR *in_title, size_t size )
 {
     WCHAR *title = NULL;
@@ -599,6 +677,10 @@ static NTSTATUS screen_buffer_ioctl( struct screen_buffer *screen_buffer, unsign
             return STATUS_INVALID_PARAMETER;
         return write_output( screen_buffer, in_data, in_size, out_size );
 
+    case IOCTL_CONDRV_READ_OUTPUT:
+        if (in_size != sizeof(struct condrv_output_params)) return STATUS_INVALID_PARAMETER;
+        return read_output( screen_buffer, in_data, out_size );
+
     case IOCTL_CONDRV_GET_OUTPUT_INFO:
         if (in_size || *out_size < sizeof(struct condrv_output_info)) return STATUS_INVALID_PARAMETER;
         return get_output_info( screen_buffer, out_size );