]> git.proxmox.com Git - mirror_edk2.git/blobdiff - UnixPkg/Sec/UgaX11.c
Fix issue with fixing tabs.
[mirror_edk2.git] / UnixPkg / Sec / UgaX11.c
index b096dc660dd6b012e5891a1bbfc027dbc1dc40d5..5250fb3c61d7f6ea7b5e203190b69a16b8ad321d 100644 (file)
@@ -1,21 +1,36 @@
+/*++
+
+Copyright (c) 2004 - 2009, Intel Corporation. All rights reserved.<BR>
+Portions copyright (c) 2008 - 2010, Apple Inc. All rights reserved.<BR>
+This program and the accompanying materials                          
+are licensed and made available under the terms and conditions of the BSD License         
+which accompanies this distribution.  The full text of the license may be found at        
+http://opensource.org/licenses/bsd-license.php                                            
+                                                                                          
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             
+
+--*/
+#include <Common/UnixInclude.h>
+
 #include <sys/ipc.h>
 #include <sys/shm.h>
-#include <sys/dir.h>
-#include <stdio.h>
-#include <stdlib.h>
-
-#include "PiPei.h"
-#include "Protocol/UnixThunk.h"
-#include "Protocol/SimpleTextIn.h"
-#include "Protocol/UgaDraw.h"
-#include "Protocol/UnixUgaIo.h"
+
+#include <PiPei.h>
+#include <Protocol/SimplePointer.h>
+#include <Protocol/SimpleTextIn.h>
+#include <Protocol/SimpleTextInEx.h>
+#include <Protocol/UgaDraw.h>
 #include <X11/Xlib.h>
 #include <X11/Xutil.h>
 #include <X11/Xos.h>
 #include <X11/extensions/XShm.h>
 #include <X11/keysym.h>
+#include <X11/cursorfont.h>
+
+#include <Protocol/UnixThunk.h>
+#include <Protocol/UnixUgaIo.h>
 
-#include <Guid/PeiPeCoffLoader.h>
 #include <Ppi/StatusCode.h>
 
 #include <Library/PeCoffLib.h>
 #include <Library/PcdLib.h>
 #include <Library/DebugLib.h>
 
+#include "Gasket.h"
+#include "SecMain.h"
+
+
 extern void msSleep (unsigned long Milliseconds);
 
 /* XQueryPointer  */
 
-struct uga_drv_shift_mask
-{
+struct uga_drv_shift_mask {
   unsigned char shift;
   unsigned char size;
   unsigned char csize;
 };
 
 #define NBR_KEYS 32
-typedef struct
-{
+typedef struct {
   EFI_UNIX_UGA_IO_PROTOCOL UgaIo;
 
   Display *display;
-  int screen;                  /* values for window_size in main */
+  int screen;      /* values for window_size in main */
   Window win;
   GC gc;
   Visual *visual;
@@ -63,13 +80,23 @@ typedef struct
   unsigned int key_rd;
   unsigned int key_wr;
   unsigned int key_count;
-  EFI_INPUT_KEY keys[NBR_KEYS];
+  EFI_KEY_DATA keys[NBR_KEYS];
+
+  EFI_KEY_STATE KeyState;
+  
+  UGA_REGISTER_KEY_NOTIFY_CALLBACK RegisterdKeyCallback;
+  VOID                             *RegisterdKeyCallbackContext;
+  
+  int                        previous_x;
+  int                        previous_y;
+  EFI_SIMPLE_POINTER_STATE   pointer_state;
+  int                        pointer_state_changed;
 } UGA_IO_PRIVATE;
 
-static void
+void
 HandleEvents(UGA_IO_PRIVATE *drv);
 
-static void
+void
 fill_shift_mask (struct uga_drv_shift_mask *sm, unsigned long mask)
 {
   sm->shift = 0;
@@ -87,12 +114,12 @@ fill_shift_mask (struct uga_drv_shift_mask *sm, unsigned long mask)
   sm->csize = 8 - sm->size;
 }
 
-static int
+int
 TryCreateShmImage(UGA_IO_PRIVATE *drv)
 {
   drv->image = XShmCreateImage (drv->display, drv->visual,
-                               drv->depth, ZPixmap, NULL, &drv->xshm_info,
-                               drv->width, drv->height);
+                 drv->depth, ZPixmap, NULL, &drv->xshm_info,
+                 drv->width, drv->height);
   if (drv->image == NULL)
     return 0;
 
@@ -109,37 +136,40 @@ TryCreateShmImage(UGA_IO_PRIVATE *drv)
   }
 
   drv->xshm_info.shmid = shmget
-       (IPC_PRIVATE, drv->image->bytes_per_line * drv->image->height,
-        IPC_CREAT | 0777);
-  if (drv->xshm_info.shmid < 0)
-    {
-      XDestroyImage(drv->image);
-      return 0;
-    }
+                          (IPC_PRIVATE, drv->image->bytes_per_line * drv->image->height,
+                          IPC_CREAT | 0777);
+  if (drv->xshm_info.shmid < 0) {
+    XDestroyImage(drv->image);
+    return 0;
+  }
       
   drv->image_data = shmat (drv->xshm_info.shmid, NULL, 0);
-  if(!drv->image_data)
-    {
-      shmctl (drv->xshm_info.shmid, IPC_RMID, NULL);
-      XDestroyImage(drv->image);
-      return 0;
-    }
+  if(!drv->image_data) {
+    shmctl (drv->xshm_info.shmid, IPC_RMID, NULL);
+    XDestroyImage(drv->image);
+    return 0;
+  }
+  
+#ifndef __APPLE__  
+  //
+  // This closes shared memory in real time on OS X. Only closes after folks quit using
+  // it on Linux. 
+  //
   /* Can this fail ?  */
   shmctl (drv->xshm_info.shmid, IPC_RMID, NULL);
+#endif
 
   drv->xshm_info.shmaddr = (char*)drv->image_data;
   drv->image->data = (char*)drv->image_data;
-         
-  if (!XShmAttach (drv->display, &drv->xshm_info))
-    {
-      shmdt (drv->image_data);
-      XDestroyImage(drv->image);
-      return 0;
-    }
+
+  if (!XShmAttach (drv->display, &drv->xshm_info)) {
+    shmdt (drv->image_data);
+    XDestroyImage(drv->image);
+    return 0;
+  }
   return 1;
 }
 
-static
 EFI_STATUS
 UgaClose (EFI_UNIX_UGA_IO_PROTOCOL *UgaIo)
 {
@@ -152,18 +182,23 @@ UgaClose (EFI_UNIX_UGA_IO_PROTOCOL *UgaIo)
       XDestroyImage(drv->image);
 
       if (drv->use_shm)
-       shmdt (drv->image_data);
+        shmdt (drv->image_data);
 
       drv->image_data = NULL;
       drv->image = NULL;
     }
   XDestroyWindow(drv->display, drv->win);
   XCloseDisplay(drv->display);
+  
+#ifdef __APPLE__
+  // Free up the shared memory
+  shmctl (drv->xshm_info.shmid, IPC_RMID, NULL);
+#endif
+  
   free(drv);
   return EFI_SUCCESS;
 }
 
-static
 EFI_STATUS
 UgaSize(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo, UINT32 Width, UINT32 Height)
 {
@@ -173,10 +208,11 @@ UgaSize(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo, UINT32 Width, UINT32 Height)
   /* Destroy current buffer if created.  */
   if (drv->image != NULL)
     {
+      /* Before destroy buffer, need to make sure the buffer available for access. */
       XDestroyImage(drv->image);
 
       if (drv->use_shm)
-       shmdt (drv->image_data);
+        shmdt (drv->image_data);
 
       drv->image_data = NULL;
       drv->image = NULL;
@@ -187,25 +223,22 @@ UgaSize(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo, UINT32 Width, UINT32 Height)
   XResizeWindow (drv->display, drv->win, Width, Height);
 
   /* Allocate image.  */
-  if (XShmQueryExtension(drv->display) && TryCreateShmImage(drv))
-    {
-      drv->use_shm = 1;
-    }  
-  else 
-    {
-      drv->use_shm = 0;
-      if (drv->depth > 16)
-       drv->pixel_shift = 2;
-      else if (drv->depth > 8)
-       drv->pixel_shift = 1;
-      else
-       drv->pixel_shift = 0;
+  if (XShmQueryExtension(drv->display) && TryCreateShmImage(drv)) {
+    drv->use_shm = 1;
+  } else {
+    drv->use_shm = 0;
+    if (drv->depth > 16)
+      drv->pixel_shift = 2;
+    else if (drv->depth > 8)
+      drv->pixel_shift = 1;
+    else
+      drv->pixel_shift = 0;
       
       drv->image_data = malloc((drv->width * drv->height) << drv->pixel_shift);
       drv->image = XCreateImage (drv->display, drv->visual, drv->depth,
-                                ZPixmap, 0, (char *)drv->image_data,
-                                drv->width, drv->height,
-                                8 << drv->pixel_shift, 0);
+                                  ZPixmap, 0, (char *)drv->image_data,
+                                  drv->width, drv->height,
+                                  8 << drv->pixel_shift, 0);
     }
   drv->line_bytes = drv->image->bytes_per_line;
   fill_shift_mask (&drv->r, drv->image->red_mask);
@@ -223,86 +256,214 @@ UgaSize(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo, UINT32 Width, UINT32 Height)
   return EFI_SUCCESS;
 }
 
-static void
+void
 handleKeyEvent(UGA_IO_PRIVATE *drv, XEvent *ev)
 {
   KeySym keysym;
   char str[4];
-  EFI_INPUT_KEY Key;
+  EFI_KEY_DATA KeyData;
   int res;
 
   if (drv->key_count == NBR_KEYS)
     return;
 
   res = XLookupString(&ev->xkey, str, sizeof(str), &keysym, NULL);
-  Key.ScanCode = 0;\r
-  Key.UnicodeChar = 0;
-  switch (keysym) {\r
-  case XK_Home:       Key.ScanCode = SCAN_HOME;       break;\r
-  case XK_End:        Key.ScanCode = SCAN_END;        break;\r
-  case XK_Left:       Key.ScanCode = SCAN_LEFT;       break;\r
-  case XK_Right:      Key.ScanCode = SCAN_RIGHT;      break;\r
-  case XK_Up:         Key.ScanCode = SCAN_UP;         break;\r
-  case XK_Down:       Key.ScanCode = SCAN_DOWN;       break;\r
-  case XK_Delete:     Key.ScanCode = SCAN_DELETE;     break;\r
-  case XK_Insert:     Key.ScanCode = SCAN_INSERT;     break;\r
-  case XK_Page_Up:    Key.ScanCode = SCAN_PAGE_UP;    break;\r
-  case XK_Page_Down:  Key.ScanCode = SCAN_PAGE_DOWN;  break;\r
-  case XK_Escape:     Key.ScanCode = SCAN_ESC;        break;\r
-\r
-  case XK_F1:   Key.ScanCode = SCAN_F1;   break;\r
-  case XK_F2:   Key.ScanCode = SCAN_F2;   break;\r
-  case XK_F3:   Key.ScanCode = SCAN_F3;   break;\r
-  case XK_F4:   Key.ScanCode = SCAN_F4;   break;\r
-  case XK_F5:   Key.ScanCode = SCAN_F5;   break;\r
-  case XK_F6:   Key.ScanCode = SCAN_F6;   break;\r
-  case XK_F7:   Key.ScanCode = SCAN_F7;   break;\r
-  case XK_F8:   Key.ScanCode = SCAN_F8;   break;\r
-  case XK_F9:   Key.ScanCode = SCAN_F9;   break;\r
+  KeyData.Key.ScanCode = 0;
+  KeyData.Key.UnicodeChar = 0;
+  KeyData.KeyState.KeyShiftState = 0;
+
+  //
+  // KeyRelease is not supported (on Mac) so we can not easily implement Ex functions.
+  // If a modifier key is hit by its self we get a keysym. If a modfifier and key is hit
+  // we get the state bit set and keysym is the modified key. 
+  //
+  // We use lack of state bits being set to clear ToggleState and KeyShiftState. We can 
+  // also use the stat bits to set ToggleState and KeyShiftState. 
+  // Skipping EFI_SCROLL_LOCK_ACTIVE & EFI_NUM_LOCK_ACTIVE since they are not on Macs  
+  //
+  if ((ev->xkey.state & LockMask) == 0) {
+    drv->KeyState.KeyToggleState &= ~EFI_CAPS_LOCK_ACTIVE;
+  } else {
+    drv->KeyState.KeyToggleState |= EFI_CAPS_LOCK_ACTIVE;
+  }
+  
+  if ((ev->xkey.state & ControlMask) == 0) {
+    drv->KeyState.KeyShiftState &= ~(EFI_RIGHT_CONTROL_PRESSED | EFI_LEFT_CONTROL_PRESSED);
+  } else if ((drv->KeyState.KeyShiftState & EFI_RIGHT_CONTROL_PRESSED) == 0) {
+    drv->KeyState.KeyShiftState |= EFI_LEFT_CONTROL_PRESSED;
+  }
+  
+  if ((ev->xkey.state & ShiftMask) == 0) {
+    drv->KeyState.KeyShiftState &= ~(EFI_RIGHT_SHIFT_PRESSED | EFI_LEFT_SHIFT_PRESSED);
+  } else if ((drv->KeyState.KeyShiftState & EFI_RIGHT_SHIFT_PRESSED) == 0) {
+    drv->KeyState.KeyShiftState |= EFI_LEFT_SHIFT_PRESSED;
+  }
+  
+  if ((ev->xkey.state & Mod2Mask) == 0) {
+    drv->KeyState.KeyShiftState &= ~(EFI_RIGHT_LOGO_PRESSED | EFI_LEFT_LOGO_PRESSED);
+  } else if ((drv->KeyState.KeyShiftState & EFI_RIGHT_LOGO_PRESSED) == 0) {
+    drv->KeyState.KeyShiftState |= EFI_LEFT_LOGO_PRESSED;
+  }
+
+  if ((ev->xkey.state & 0x2000) == 0) {
+    drv->KeyState.KeyShiftState &= ~(EFI_LEFT_ALT_PRESSED);
+  } else {
+    drv->KeyState.KeyShiftState |= EFI_LEFT_ALT_PRESSED;
+  }
+  
+  
+  switch (keysym) {
+  case XK_Control_R:
+    drv->KeyState.KeyShiftState |= EFI_RIGHT_CONTROL_PRESSED;
+    break;
+  case XK_Control_L:
+    drv->KeyState.KeyShiftState |= EFI_LEFT_CONTROL_PRESSED;
+    break;
+
+  case XK_Shift_R:
+    drv->KeyState.KeyShiftState |= EFI_RIGHT_SHIFT_PRESSED;
+    break;
+  case XK_Shift_L:
+    drv->KeyState.KeyShiftState |= EFI_LEFT_SHIFT_PRESSED;
+    break;
+  
+  case XK_Mode_switch:
+    drv->KeyState.KeyShiftState |= EFI_LEFT_ALT_PRESSED;
+    break;
+
+  case XK_Meta_R:
+    drv->KeyState.KeyShiftState |= EFI_RIGHT_LOGO_PRESSED;
+    break;
+  case XK_Meta_L:
+    drv->KeyState.KeyShiftState |= EFI_LEFT_LOGO_PRESSED;
+    break;
+
+  case XK_Home:       KeyData.Key.ScanCode = SCAN_HOME;       break;
+  case XK_End:        KeyData.Key.ScanCode = SCAN_END;        break;
+  case XK_Left:       KeyData.Key.ScanCode = SCAN_LEFT;       break;
+  case XK_Right:      KeyData.Key.ScanCode = SCAN_RIGHT;      break;
+  case XK_Up:         KeyData.Key.ScanCode = SCAN_UP;         break;
+  case XK_Down:       KeyData.Key.ScanCode = SCAN_DOWN;       break;
+  case XK_Delete:     KeyData.Key.ScanCode = SCAN_DELETE;     break;
+  case XK_Insert:     KeyData.Key.ScanCode = SCAN_INSERT;     break;
+  case XK_Page_Up:    KeyData.Key.ScanCode = SCAN_PAGE_UP;    break;
+  case XK_Page_Down:  KeyData.Key.ScanCode = SCAN_PAGE_DOWN;  break;
+  case XK_Escape:     KeyData.Key.ScanCode = SCAN_ESC;        break;
+
+  case XK_F1:   KeyData.Key.ScanCode = SCAN_F1;   break;
+  case XK_F2:   KeyData.Key.ScanCode = SCAN_F2;   break;
+  case XK_F3:   KeyData.Key.ScanCode = SCAN_F3;   break;
+  case XK_F4:   KeyData.Key.ScanCode = SCAN_F4;   break;
+  case XK_F5:   KeyData.Key.ScanCode = SCAN_F5;   break;
+  case XK_F6:   KeyData.Key.ScanCode = SCAN_F6;   break;
+  case XK_F7:   KeyData.Key.ScanCode = SCAN_F7;   break;
+  case XK_F8:   KeyData.Key.ScanCode = SCAN_F8;   break;
+  case XK_F9:   KeyData.Key.ScanCode = SCAN_F9;   break;
 
   default:
     if (res == 1) {
-      Key.UnicodeChar = str[0];
+      KeyData.Key.UnicodeChar = str[0];
     } else {
       return;
     }
-  }\r
-\r
-  drv->keys[drv->key_wr] = Key;
+  }
+
+  // The global state is our state
+  KeyData.KeyState.KeyShiftState = drv->KeyState.KeyShiftState;
+  KeyData.KeyState.KeyToggleState = drv->KeyState.KeyToggleState;
+
+  CopyMem (&drv->keys[drv->key_wr], &KeyData, sizeof (EFI_KEY_DATA));
   drv->key_wr = (drv->key_wr + 1) % NBR_KEYS;
   drv->key_count++;
+  
+  
+#if defined(__APPLE__) || defined(MDE_CPU_X64)
+  ReverseGasketUint64Uint64 (drv->RegisterdKeyCallback ,drv->RegisterdKeyCallbackContext, &KeyData);
+#else
+  drv->RegisterdKeyCallback (drv->RegisterdKeyCallbackContext, &KeyData);
+#endif
+
+  
+}
+
+
+void
+handleMouseMoved(UGA_IO_PRIVATE *drv, XEvent *ev)
+{
+  if ( ev->xmotion.x != drv->previous_x )
+  {
+    drv->pointer_state.RelativeMovementX += ( ev->xmotion.x - drv->previous_x );
+  drv->previous_x = ev->xmotion.x;
+  drv->pointer_state_changed = 1;
+  }
+
+  if ( ev->xmotion.y != drv->previous_y )
+  {
+    drv->pointer_state.RelativeMovementY += ( ev->xmotion.y - drv->previous_y );
+    drv->previous_y = ev->xmotion.y;
+  drv->pointer_state_changed = 1;
+  }
+
+  drv->pointer_state.RelativeMovementZ = 0;
+}
+
+void
+handleMouseDown(UGA_IO_PRIVATE *drv, XEvent *ev, BOOLEAN Pressed)
+{
+  if ( ev->xbutton.button == Button1 )
+  {
+    drv->pointer_state_changed = ( drv->pointer_state.LeftButton != Pressed );
+  drv->pointer_state.LeftButton = Pressed;
+  }
+  if ( ev->xbutton.button == Button2 )
+  {
+    drv->pointer_state_changed = ( drv->pointer_state.RightButton != Pressed );
+  drv->pointer_state.RightButton = Pressed;
+  }
 }
 
-static void
+void
 Redraw(UGA_IO_PRIVATE *drv, UINTN X, UINTN Y, UINTN Width, UINTN Height)
 {
   if (drv->use_shm)
     XShmPutImage (drv->display, drv->win, drv->gc, drv->image,
-                 X, Y, X, Y, Width, Height, False);
+                   X, Y, X, Y, Width, Height, False);
   else
     XPutImage (drv->display, drv->win, drv->gc, drv->image,
-                 X, Y, X, Y, Width, Height);
+                X, Y, X, Y, Width, Height);
+  XFlush(drv->display);
 }
 
-static void
+void
 HandleEvent(UGA_IO_PRIVATE *drv, XEvent *ev)
 {
   switch (ev->type)
     {
     case Expose:
       Redraw(drv, ev->xexpose.x, ev->xexpose.y,
-            ev->xexpose.width, ev->xexpose.height);
+        ev->xexpose.width, ev->xexpose.height);
       break;
     case GraphicsExpose:
       Redraw(drv, ev->xgraphicsexpose.x, ev->xgraphicsexpose.y,
-            ev->xgraphicsexpose.width, ev->xgraphicsexpose.height);
+        ev->xgraphicsexpose.width, ev->xgraphicsexpose.height);
       break;
     case KeyPress:
       handleKeyEvent(drv, ev);
       break;
+    case KeyRelease:
+      break;
     case MappingNotify:
       XRefreshKeyboardMapping(&ev->xmapping);
       break;
+    case MotionNotify:
+      handleMouseMoved(drv, ev);
+      break;
+    case ButtonPress:
+      handleMouseDown(drv, ev, TRUE);
+    break;
+    case ButtonRelease:
+      handleMouseDown(drv, ev, FALSE);
+    break;
 #if 0
     case DestroyNotify:
       XCloseDisplay (drv->display);
@@ -315,19 +476,18 @@ HandleEvent(UGA_IO_PRIVATE *drv, XEvent *ev)
     }
 }
 
-static void
+void
 HandleEvents(UGA_IO_PRIVATE *drv)
 {
   while (XPending(drv->display) != 0)
     {
       XEvent ev;
-         
+
       XNextEvent (drv->display, &ev);
       HandleEvent(drv, &ev);
     }
 }
 
-static
 unsigned long
 UgaPixelToColor (UGA_IO_PRIVATE *drv, EFI_UGA_PIXEL pixel)
 {
@@ -336,7 +496,6 @@ UgaPixelToColor (UGA_IO_PRIVATE *drv, EFI_UGA_PIXEL pixel)
     | ((pixel.Blue >> drv->b.csize) << drv->b.shift);
 }
 
-static
 EFI_UGA_PIXEL
 UgaColorToPixel (UGA_IO_PRIVATE *drv, unsigned long val)
 {
@@ -351,58 +510,102 @@ UgaColorToPixel (UGA_IO_PRIVATE *drv, unsigned long val)
   return res;
 }
 
-static
-EFI_STATUS
-UgaCheckKey(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo)
+STATIC EFI_STATUS
+CheckKeyInternal( UGA_IO_PRIVATE *drv, BOOLEAN delay )
 {
-  UGA_IO_PRIVATE *drv = (UGA_IO_PRIVATE *)UgaIo;
   HandleEvents(drv);
   if (drv->key_count != 0)
     return EFI_SUCCESS;
-  else {
-    /* EFI is certainly polling.  Be CPU-friendly.  */
+  if ( delay )
+    /* EFI is polling.  Be CPU-friendly.  */
     msSleep (20);
     return EFI_NOT_READY;
   }
+
+EFI_STATUS
+UgaCheckKey(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo)
+{
+  UGA_IO_PRIVATE  *drv = (UGA_IO_PRIVATE *)UgaIo;
+  return( CheckKeyInternal( drv, TRUE ) );
 }
 
-static
 EFI_STATUS
-UgaGetKey(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo, EFI_INPUT_KEY *key)
+EFIAPI
+UgaGetKey (
+  IN  EFI_UNIX_UGA_IO_PROTOCOL  *UgaIo, 
+  IN  EFI_KEY_DATA              *KeyData
+  )
 {
   UGA_IO_PRIVATE *drv = (UGA_IO_PRIVATE *)UgaIo;
   EFI_STATUS status;
 
-  status = UgaCheckKey(UgaIo);
+  status = CheckKeyInternal(drv, FALSE);
   if (status != EFI_SUCCESS)
     return status;
 
-  *key = drv->keys[drv->key_rd];
+  CopyMem (KeyData, &drv->keys[drv->key_rd], sizeof (EFI_KEY_DATA));
   drv->key_rd = (drv->key_rd + 1) % NBR_KEYS;
   drv->key_count--;
   return EFI_SUCCESS;
 }
 
-EFI_STATUS\r
-UgaBlt(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo,
-       IN  EFI_UGA_PIXEL                           *BltBuffer OPTIONAL,\r
-       IN  EFI_UGA_BLT_OPERATION                   BltOperation,\r
-       IN  UINTN                                   SourceX,\r
-       IN  UINTN                                   SourceY,\r
-       IN  UINTN                                   DestinationX,\r
-       IN  UINTN                                   DestinationY,\r
-       IN  UINTN                                   Width,\r
-       IN  UINTN                                   Height,\r
-       IN  UINTN                                   Delta OPTIONAL\r
-  )\r
+
+EFI_STATUS
+EFIAPI
+UgaKeySetState (
+  IN EFI_UNIX_UGA_IO_PROTOCOL   *UgaIo, 
+  IN EFI_KEY_TOGGLE_STATE       *KeyToggleState
+  )
+{
+  UGA_IO_PRIVATE  *drv = (UGA_IO_PRIVATE *)UgaIo;
+//  XKeyEvent       event;
+  
+  if (*KeyToggleState & EFI_CAPS_LOCK_ACTIVE) {
+    if ((drv->KeyState.KeyToggleState & EFI_CAPS_LOCK_ACTIVE) == 0) {
+      //
+      // We could create an XKeyEvent and send a XK_Caps_Lock to
+      // the UGA/GOP Window
+      //
+    }
+  }
+    
+  drv->KeyState.KeyToggleState = *KeyToggleState;
+  return EFI_SUCCESS;
+}
+
+
+EFI_STATUS
+EFIAPI
+UgaRegisterKeyNotify (
+  IN EFI_UNIX_UGA_IO_PROTOCOL           *UgaIo, 
+  IN UGA_REGISTER_KEY_NOTIFY_CALLBACK   CallBack,
+  IN VOID                               *Context
+  )
+{
+  UGA_IO_PRIVATE  *drv = (UGA_IO_PRIVATE *)UgaIo;
+
+  drv->RegisterdKeyCallback = CallBack;
+  drv->RegisterdKeyCallbackContext = Context;
+
+  return EFI_SUCCESS;
+}
+
+
+EFI_STATUS
+UgaBlt(
+  IN EFI_UNIX_UGA_IO_PROTOCOL                 *UgaIo,
+  IN  EFI_UGA_PIXEL                           *BltBuffer OPTIONAL,
+  IN  EFI_UGA_BLT_OPERATION                   BltOperation,
+  IN  UGA_BLT_ARGS                            *Args
+  )
 {
   UGA_IO_PRIVATE *Private = (UGA_IO_PRIVATE *)UgaIo;
-  UINTN             DstY;\r
-  UINTN             SrcY;\r
+  UINTN             DstY;
+  UINTN             SrcY;
   UINTN             DstX;
   UINTN             SrcX;
   UINTN             Index;
-  EFI_UGA_PIXEL     *Blt;\r
+  EFI_UGA_PIXEL     *Blt;
   UINT8             *Dst;
   UINT8             *Src;
   UINTN             Nbr;
@@ -412,120 +615,121 @@ UgaBlt(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo,
   //  Check bounds
   //
   if (BltOperation == EfiUgaVideoToBltBuffer
-      || BltOperation == EfiUgaVideoToVideo) {\r
-    //\r
+      || BltOperation == EfiUgaVideoToVideo) {
+    //
     // Source is Video.
-    //\r
-    if (SourceY + Height > Private->height) {\r
-      return EFI_INVALID_PARAMETER;\r
-    }\r
-\r
-    if (SourceX + Width > Private->width) {\r
-      return EFI_INVALID_PARAMETER;\r
-    }\r
+    //
+    if (Args->SourceY + Args->Height > Private->height) {
+      return EFI_INVALID_PARAMETER;
+    }
+
+    if (Args->SourceX + Args->Width > Private->width) {
+      return EFI_INVALID_PARAMETER;
+    }
   }
 
   if (BltOperation == EfiUgaBltBufferToVideo
       || BltOperation == EfiUgaVideoToVideo
-      || BltOperation == EfiUgaVideoFill) {\r
-    //\r
-    // Destination is Video\r
-    //\r
-    if (DestinationY + Height > Private->height) {\r
-      return EFI_INVALID_PARAMETER;\r
-    }\r
-\r
-    if (DestinationX + Width > Private->width) {\r
-      return EFI_INVALID_PARAMETER;\r
-    }\r
+      || BltOperation == EfiUgaVideoFill) {
+    //
+    // Destination is Video
+    //
+    if (Args->DestinationY + Args->Height > Private->height) {
+      return EFI_INVALID_PARAMETER;
+    }
+
+    if (Args->DestinationX + Args->Width > Private->width) {
+      return EFI_INVALID_PARAMETER;
+    }
   }
 
-  switch (BltOperation) {\r
+  switch (BltOperation) {
   case EfiUgaVideoToBltBuffer:
-    Blt = BltBuffer;
-    Delta -= Width * sizeof (EFI_UGA_PIXEL);
-    for (SrcY = SourceY; SrcY < (Height + SourceY); SrcY++) {\r
-      for (SrcX = SourceX; SrcX < (Width + SourceX); SrcX++) {
-       *Blt++ = UgaColorToPixel(Private,
-                                XGetPixel(Private->image, SrcX, SrcY));
+    Blt = (EFI_UGA_PIXEL *)((UINT8 *)BltBuffer + (Args->DestinationY * Args->Delta) + Args->DestinationX * sizeof (EFI_UGA_PIXEL));
+    Args->Delta -= Args->Width * sizeof (EFI_UGA_PIXEL);
+    for (SrcY = Args->SourceY; SrcY < (Args->Height + Args->SourceY); SrcY++) {
+      for (SrcX = Args->SourceX; SrcX < (Args->Width + Args->SourceX); SrcX++) {
+        *Blt++ = UgaColorToPixel(Private,
+                                  XGetPixel(Private->image, SrcX, SrcY));
       }
-      Blt = (EFI_UGA_PIXEL *) ((UINT8 *) Blt + Delta);
-    }\r
+      Blt = (EFI_UGA_PIXEL *) ((UINT8 *) Blt + Args->Delta);
+    }
     break;
-  case EfiUgaBltBufferToVideo:\r
-    Blt = BltBuffer;
-    Delta -= Width * sizeof (EFI_UGA_PIXEL);
-    for (DstY = DestinationY; DstY < (Height + DestinationY); DstY++) {\r
-      for (DstX = DestinationX; DstX < (Width + DestinationX); DstX++) {\r
-       XPutPixel(Private->image, DstX, DstY, UgaPixelToColor(Private, *Blt));
-       Blt++;
+  case EfiUgaBltBufferToVideo:
+    Blt = (EFI_UGA_PIXEL *)((UINT8 *)BltBuffer + (Args->SourceY * Args->Delta) + Args->SourceX * sizeof (EFI_UGA_PIXEL));
+    Args->Delta -= Args->Width * sizeof (EFI_UGA_PIXEL);
+    for (DstY = Args->DestinationY; DstY < (Args->Height + Args->DestinationY); DstY++) {
+      for (DstX = Args->DestinationX; DstX < (Args->Width + Args->DestinationX); DstX++) {
+        XPutPixel(Private->image, DstX, DstY, UgaPixelToColor(Private, *Blt));
+        Blt++;
       }
-      Blt = (EFI_UGA_PIXEL *) ((UINT8 *) Blt + Delta);
+      Blt = (EFI_UGA_PIXEL *) ((UINT8 *) Blt + Args->Delta);
     }
     break;
-  case EfiUgaVideoToVideo:\r
-    Dst = Private->image_data + (DestinationX << Private->pixel_shift)
-      + DestinationY * Private->line_bytes;
-    Src = Private->image_data + (SourceX << Private->pixel_shift)
-      + SourceY * Private->line_bytes;
-    Nbr = Width << Private->pixel_shift;
-    if (DestinationY < SourceY) {\r
-      for (Index = 0; Index < Height; Index++) {
-       memcpy (Dst, Src, Nbr);
-       Dst += Private->line_bytes;
-       Src += Private->line_bytes;
+  case EfiUgaVideoToVideo:
+    Dst = Private->image_data + (Args->DestinationX << Private->pixel_shift)
+      + Args->DestinationY * Private->line_bytes;
+    Src = Private->image_data + (Args->SourceX << Private->pixel_shift)
+      + Args->SourceY * Private->line_bytes;
+    Nbr = Args->Width << Private->pixel_shift;
+    if (Args->DestinationY < Args->SourceY) {
+      for (Index = 0; Index < Args->Height; Index++) {
+        memcpy (Dst, Src, Nbr);
+        Dst += Private->line_bytes;
+        Src += Private->line_bytes;
       }
     }
     else {
-      Dst += (Height - 1) * Private->line_bytes;
-      Src += (Height - 1) * Private->line_bytes;
-      for (Index = 0; Index < Height; Index++) {
-       //
-       // Source and Destination Y may be equal, therefore Dst and Src may
-       // overlap.
-       //
-       memmove (Dst, Src, Nbr);
-       Dst -= Private->line_bytes;
-       Src -= Private->line_bytes;
+      Dst += (Args->Height - 1) * Private->line_bytes;
+      Src += (Args->Height - 1) * Private->line_bytes;
+      for (Index = 0; Index < Args->Height; Index++) {
+      //
+      // Source and Destination Y may be equal, therefore Dst and Src may
+      // overlap.
+      //
+      memmove (Dst, Src, Nbr);
+      Dst -= Private->line_bytes;
+      Src -= Private->line_bytes;
       }
     }
     break;
   case EfiUgaVideoFill:
     Color = UgaPixelToColor(Private, *BltBuffer);
-    for (DstY = DestinationY; DstY < (Height + DestinationY); DstY++) {\r
-      for (DstX = DestinationX; DstX < (Width + DestinationX); DstX++) {
-       XPutPixel(Private->image, DstX, DstY, Color);
+    for (DstY = Args->DestinationY; DstY < (Args->Height + Args->DestinationY); DstY++) {
+      for (DstX = Args->DestinationX; DstX < (Args->Width + Args->DestinationX); DstX++) {
+        XPutPixel(Private->image, DstX, DstY, Color);
       }
-    }\r
+    }
     break;
   default:
-      return EFI_INVALID_PARAMETER;\r
+      return EFI_INVALID_PARAMETER;
   }
 
   //
   //  Refresh screen.
   //
-  switch (BltOperation) {\r
-  case EfiUgaVideoToVideo:\r
+  switch (BltOperation) {
+  case EfiUgaVideoToVideo:
     XCopyArea(Private->display, Private->win, Private->win, Private->gc,
-             SourceX, SourceY, Width, Height, DestinationX, DestinationY);
+               Args->SourceX, Args->SourceY, Args->Width, Args->Height, Args->DestinationX, Args->DestinationY);
     while (1) {
       XEvent ev;
-         
+
       XNextEvent (Private->display, &ev);
       HandleEvent(Private, &ev);
       if (ev.type == NoExpose || ev.type == GraphicsExpose)
-       break;
+        break;
     }
     break;
   case EfiUgaVideoFill:
     Color = UgaPixelToColor(Private, *BltBuffer);
     XSetForeground(Private->display, Private->gc, Color);
     XFillRectangle(Private->display, Private->win, Private->gc,
-                  DestinationX, DestinationY, Width, Height);
+                    Args->DestinationX, Args->DestinationY, Args->Width, Args->Height);
+    XFlush(Private->display);
     break;
-  case EfiUgaBltBufferToVideo:\r
-    Redraw(Private, DestinationX, DestinationY, Width, Height);
+  case EfiUgaBltBufferToVideo:
+    Redraw(Private, Args->DestinationX, Args->DestinationY, Args->Width, Args->Height);
     break;
   default:
     break;
@@ -533,6 +737,44 @@ UgaBlt(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo,
   return EFI_SUCCESS;
 }
 
+STATIC EFI_STATUS
+CheckPointerInternal( UGA_IO_PRIVATE *drv, BOOLEAN delay )
+{
+  HandleEvents(drv);
+  if (drv->pointer_state_changed != 0)
+    return EFI_SUCCESS;
+  if ( delay )
+    /* EFI is polling.  Be CPU-friendly.  */
+    msSleep (20);
+  return EFI_NOT_READY;
+}
+
+EFI_STATUS
+UgaCheckPointer(EFI_UNIX_UGA_IO_PROTOCOL *UgaIo)
+{
+  UGA_IO_PRIVATE  *drv = (UGA_IO_PRIVATE *)UgaIo;
+  return( CheckPointerInternal( drv, TRUE ) );
+}
+
+EFI_STATUS
+UgaGetPointerState (EFI_UNIX_UGA_IO_PROTOCOL *UgaIo, EFI_SIMPLE_POINTER_STATE *state)
+{
+  UGA_IO_PRIVATE *drv = (UGA_IO_PRIVATE *)UgaIo;
+  EFI_STATUS status;
+
+  status = CheckPointerInternal( drv, FALSE );
+  if (status != EFI_SUCCESS)
+    return status;
+  
+  memcpy( state, &drv->pointer_state, sizeof( EFI_SIMPLE_POINTER_STATE ) );
+
+  drv->pointer_state.RelativeMovementX = 0;
+  drv->pointer_state.RelativeMovementY = 0;
+  drv->pointer_state.RelativeMovementZ = 0;
+  drv->pointer_state_changed = 0;
+  return EFI_SUCCESS;
+}
+
 EFI_STATUS
 UgaCreate (EFI_UNIX_UGA_IO_PROTOCOL **Uga, CONST CHAR16 *Title)
 {
@@ -541,37 +783,64 @@ UgaCreate (EFI_UNIX_UGA_IO_PROTOCOL **Uga, CONST CHAR16 *Title)
   char *display_name = NULL;
   int title_len;
 
-  drv = (UGA_IO_PRIVATE *)
-    calloc (1, sizeof (UGA_IO_PRIVATE));
+  drv = (UGA_IO_PRIVATE *)calloc (1, sizeof (UGA_IO_PRIVATE));
   if (drv == NULL)
     return EFI_OUT_OF_RESOURCES;
 
+#if defined(__APPLE__) || defined(MDE_CPU_X64)
+//
+//
+//
+  drv->UgaIo.UgaClose    = GasketUgaClose; 
+  drv->UgaIo.UgaSize     = GasketUgaSize;
+  drv->UgaIo.UgaCheckKey = GasketUgaCheckKey;
+  drv->UgaIo.UgaGetKey   = GasketUgaGetKey;
+  drv->UgaIo.UgaKeySetState         = GasketUgaKeySetState;
+  drv->UgaIo.UgaRegisterKeyNotify   = GasketUgaRegisterKeyNotify;
+  drv->UgaIo.UgaBlt      = GasketUgaBlt;
+  drv->UgaIo.UgaCheckPointer        = GasketUgaCheckPointer;
+  drv->UgaIo.UgaGetPointerState     = GasketUgaGetPointerState;
+#else
   drv->UgaIo.UgaClose = UgaClose;
   drv->UgaIo.UgaSize = UgaSize;
   drv->UgaIo.UgaCheckKey = UgaCheckKey;
   drv->UgaIo.UgaGetKey = UgaGetKey;
+  drv->UgaIo.UgaKeySetState         = UgaKeySetState;
+  drv->UgaIo.UgaRegisterKeyNotify   = UgaRegisterKeyNotify;
   drv->UgaIo.UgaBlt = UgaBlt;
+  drv->UgaIo.UgaCheckPointer        = UgaCheckPointer;
+  drv->UgaIo.UgaGetPointerState     = UgaGetPointerState;
+#endif
+  
+  
 
   drv->key_count = 0;
   drv->key_rd = 0;
   drv->key_wr = 0;
+  drv->KeyState.KeyShiftState  = EFI_SHIFT_STATE_VALID;
+  drv->KeyState.KeyToggleState = EFI_TOGGLE_STATE_VALID;
+  drv->RegisterdKeyCallback = NULL;
+  drv->RegisterdKeyCallbackContext = NULL;
+  
+  
   drv->display = XOpenDisplay (display_name);
   if (drv->display == NULL)
     {
       fprintf (stderr, "uga: cannot connect to X server %s\n",
-              XDisplayName (display_name));
+                XDisplayName (display_name));
       free (drv);
       return EFI_DEVICE_ERROR;
     }
   drv->screen = DefaultScreen (drv->display);
   drv->visual = DefaultVisual (drv->display, drv->screen);
   drv->win = XCreateSimpleWindow
-       (drv->display, RootWindow (drv->display, drv->screen),
-        0, 0, 4, 4, border_width,
-        BlackPixel (drv->display, drv->screen),
-        WhitePixel (drv->display, drv->screen));
+               (drv->display, RootWindow (drv->display, drv->screen),
+                0, 0, 4, 4, border_width,
+                WhitePixel (drv->display, drv->screen),
+                BlackPixel (drv->display, drv->screen));
 
   drv->depth = DefaultDepth (drv->display, drv->screen);
+  XDefineCursor (drv->display, drv->win, XCreateFontCursor (drv->display, XC_pirate)); 
 
   /* Compute title len and convert to Ascii.  */
   for (title_len = 0; Title[title_len] != 0; title_len++)
@@ -587,7 +856,7 @@ UgaCreate (EFI_UNIX_UGA_IO_PROTOCOL **Uga, CONST CHAR16 *Title)
   }
 
   XSelectInput (drv->display, drv->win,
-               ExposureMask | KeyPressMask);
+                 ExposureMask | KeyPressMask | PointerMotionMask | ButtonPressMask | ButtonReleaseMask );
   drv->gc = DefaultGC (drv->display, drv->screen);
 
   *Uga = (EFI_UNIX_UGA_IO_PROTOCOL *)drv;