Initial revision
git-svn-id: svn://svn.code.sf.net/p/tigervnc/code/trunk@2 3789f03b-4d11-0410-bbf8-ca57d06f2519
diff --git a/x0vncserver/Image.cxx b/x0vncserver/Image.cxx
new file mode 100644
index 0000000..56a197e
--- /dev/null
+++ b/x0vncserver/Image.cxx
@@ -0,0 +1,149 @@
+/* Copyright (C) 2002-2003 RealVNC Ltd. All Rights Reserved.
+ *
+ * This is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this software; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
+ * USA.
+ */
+//
+// Image.cxx
+//
+
+
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/ipc.h>
+#include <sys/shm.h>
+#include <X11/Xlib.h>
+#include <X11/Xutil.h>
+#include <X11/extensions/XShm.h>
+#include "Image.h"
+#include <list>
+
+class ImageCleanup {
+public:
+ std::list<Image*> images;
+
+ ~ImageCleanup()
+ {
+ fprintf(stderr,"~ImageCleanup called\n");
+
+ while (!images.empty()) {
+ delete images.front();
+ }
+ }
+};
+
+ImageCleanup imageCleanup;
+
+static bool caughtShmError = false;
+
+static int ShmCreationXErrorHandler(Display *dpy, XErrorEvent *error)
+{
+ caughtShmError = true;
+ return 0;
+}
+
+Image::Image(Display* d, int width, int height)
+ : xim(0), dpy(d), shminfo(0), usingShm(false)
+{
+ if (createShmImage(width, height)) return;
+
+ xim = XCreateImage(dpy, DefaultVisual(dpy, DefaultScreen(dpy)),
+ DefaultDepth(dpy,DefaultScreen(dpy)), ZPixmap,
+ 0, 0, width, height, BitmapPad(dpy), 0);
+
+ xim->data = (char*)malloc(xim->bytes_per_line * xim->height);
+ if (!xim->data) {
+ fprintf(stderr,"malloc failed\n");
+ exit(1);
+ }
+}
+
+Image::~Image()
+{
+ fprintf(stderr,"~Image called - usingShm %d\n",usingShm);
+ if (usingShm) {
+ usingShm = false;
+ shmdt(shminfo->shmaddr);
+ shmctl(shminfo->shmid, IPC_RMID, 0);
+ imageCleanup.images.remove(this);
+ }
+ delete shminfo;
+ if (xim) XDestroyImage(xim);
+}
+
+void Image::get(Window w)
+{
+ if (usingShm) {
+ XShmGetImage(dpy, w, xim, 0, 0, AllPlanes);
+ } else {
+ XGetSubImage(dpy, w, 0, 0, xim->width, xim->height,
+ AllPlanes, ZPixmap, xim, 0, 0);
+ }
+}
+
+bool Image::createShmImage(int width, int height)
+{
+ if (XShmQueryExtension(dpy)) {
+ shminfo = new XShmSegmentInfo;
+
+ xim = XShmCreateImage(dpy, DefaultVisual(dpy, DefaultScreen(dpy)),
+ DefaultDepth(dpy,DefaultScreen(dpy)), ZPixmap,
+ 0, shminfo, width, height);
+
+ if (xim) {
+ shminfo->shmid = shmget(IPC_PRIVATE,
+ xim->bytes_per_line * xim->height,
+ IPC_CREAT|0777);
+
+ if (shminfo->shmid != -1) {
+ shminfo->shmaddr = xim->data = (char*)shmat(shminfo->shmid, 0, 0);
+
+ if (shminfo->shmaddr != (char *)-1) {
+
+ shminfo->readOnly = False;
+
+ XErrorHandler oldHdlr = XSetErrorHandler(ShmCreationXErrorHandler);
+ XShmAttach(dpy, shminfo);
+ XSync(dpy, False);
+ XSetErrorHandler(oldHdlr);
+
+ if (!caughtShmError) {
+ fprintf(stderr,"Using shared memory XImage\n");
+ usingShm = true;
+ imageCleanup.images.push_back(this);
+ return true;
+ }
+
+ shmdt(shminfo->shmaddr);
+ } else {
+ fprintf(stderr,"shmat failed\n");
+ perror("shmat");
+ }
+
+ shmctl(shminfo->shmid, IPC_RMID, 0);
+ } else {
+ fprintf(stderr,"shmget failed\n");
+ perror("shmget");
+ }
+
+ XDestroyImage(xim);
+ xim = 0;
+ } else {
+ fprintf(stderr,"XShmCreateImage failed\n");
+ }
+ }
+
+ return false;
+}