From 16ec1a0e4a020eafde41b183d240757751caa417 Mon Sep 17 00:00:00 2001
From: Brian Hinz <bphinz@users.sourceforge.net>
Date: Thu, 1 Dec 2011 00:50:38 +0000
Subject: [PATCH] Tight decoder fixes to address issues described in bug
 #3444605

git-svn-id: svn://svn.code.sf.net/p/tigervnc/code/trunk@4820 3789f03b-4d11-0410-bbf8-ca57d06f2519
---
 java/com/tigervnc/rfb/CMsgReader.java   |  4 ++--
 java/com/tigervnc/rfb/TightDecoder.java | 31 +++++++++++++++++++------
 2 files changed, 26 insertions(+), 9 deletions(-)

diff --git a/java/com/tigervnc/rfb/CMsgReader.java b/java/com/tigervnc/rfb/CMsgReader.java
index 41230560..c0e9bf6f 100644
--- a/java/com/tigervnc/rfb/CMsgReader.java
+++ b/java/com/tigervnc/rfb/CMsgReader.java
@@ -138,8 +138,8 @@ abstract public class CMsgReader {
 
   public int[] getImageBuf(int required, int requested, int nPixels) 
   {
-    int requiredBytes = required * (handler.cp.pf().bpp / 8);
-    int requestedBytes = requested * (handler.cp.pf().bpp / 8);
+    int requiredBytes = required;
+    int requestedBytes = requested;
     int size = requestedBytes;
     if (size > imageBufIdealSize) size = imageBufIdealSize;
 
diff --git a/java/com/tigervnc/rfb/TightDecoder.java b/java/com/tigervnc/rfb/TightDecoder.java
index 6f87ffdc..c029d690 100644
--- a/java/com/tigervnc/rfb/TightDecoder.java
+++ b/java/com/tigervnc/rfb/TightDecoder.java
@@ -25,6 +25,8 @@ import java.util.ArrayList;
 import java.awt.image.BufferedImage;
 import java.io.ByteArrayInputStream;
 import javax.imageio.ImageIO;
+import com.sun.image.codec.jpeg.*;
+import java.io.InputStream;
 
 public class TightDecoder extends Decoder {
 
@@ -157,7 +159,6 @@ public class TightDecoder extends Decoder {
 
     int stride = r.width();
     int[] buf = reader.getImageBuf(r.area());
-    
 
     if (palSize == 0) {
       // Truecolor data.
@@ -170,12 +171,26 @@ public class TightDecoder extends Decoder {
       } else {
         // Copy
         int h = r.height();
+        int ptr = 0;
+        int srcPtr = 0;
         int w = r.width();
         if (cutZeros) {
-          serverpf.bufferFromRGB(buf, 0, netbuf, 0, w*h);
+          serverpf.bufferFromRGB(buf, ptr, netbuf, srcPtr, w*h);
         } else {
-          for (int i = 0; i < dataSize; i++)
-            buf[i] = netbuf[i] & 0xff;
+          int pixelSize = (bpp >= 24) ? 3 : bpp/8;
+          while (h > 0) {
+            for (int i = 0; i < w; i++) {
+              if (bpp == 8) {
+                buf[ptr+i] = netbuf[srcPtr+i] & 0xff;
+              } else {
+                for (int j = pixelSize-1; j >= 0; j--)
+                  buf[ptr+i] |= ((netbuf[srcPtr+i+j] & 0xff) << j*8);
+              }
+            }
+            ptr += stride;
+            srcPtr += w * pixelSize;
+            h--;
+          }
         }
       }
     } else {
@@ -235,16 +250,18 @@ public class TightDecoder extends Decoder {
     // Create an Image object from the JPEG data.
     int imageType = BufferedImage.TYPE_4BYTE_ABGR_PRE;
         
+    int w = r.width();
+    int h = r.height();
     BufferedImage jpeg = 
-      new BufferedImage(r.width(), r.height(), imageType);
+      new BufferedImage(w, h, imageType);
     jpeg.setAccelerationPriority(1);
     try {
       jpeg = ImageIO.read(new ByteArrayInputStream(netbuf));
     } catch (java.io.IOException e) {
       e.printStackTrace();
     }
-    int[] buf = reader.getImageBuf(r.area());
-    jpeg.getRGB(0, 0, r.width(), r.height(), buf, 0, r.width());
+    int[] buf = reader.getImageBuf(w*h);
+    jpeg.getRGB(0, 0, w, h, buf, 0, w);
     jpeg = null;
     handler.imageRect(r, buf);
   }
-- 
2.39.5