private static void initialize() {
    if (ApplicationManager.getApplication().isUnitTestMode()) {
      return;
    }

    Ole32 ole32 = Ole32.INSTANCE;
    ole32.CoInitializeEx(Pointer.NULL, 0);

    Guid.GUID CLSID_TaskbarList =
        Ole32Util.getGUIDFromString("{56FDF344-FD6D-11d0-958A-006097C9A090}");
    Guid.GUID IID_ITaskbarList3 =
        Ole32Util.getGUIDFromString("{EA1AFB91-9E28-4B86-90E9-9E9F8A5EEFAF}");
    PointerByReference p = new PointerByReference();
    ole32.CoCreateInstance(
        CLSID_TaskbarList, Pointer.NULL, ObjBase.CLSCTX_ALL, IID_ITaskbarList3, p);

    myInterfacePointer = p.getValue();
    Pointer vTablePointer = myInterfacePointer.getPointer(0);
    Pointer[] vTable = new Pointer[TaskBarList_Methods];
    vTablePointer.read(0, vTable, 0, vTable.length);

    mySetProgressValue =
        Function.getFunction(vTable[TaskBarList_SetProgressValue], Function.ALT_CONVENTION);
    mySetProgressState =
        Function.getFunction(vTable[TaskBarList_SetProgressState], Function.ALT_CONVENTION);
    mySetOverlayIcon =
        Function.getFunction(vTable[TaskBarList_SetOverlayIcon], Function.ALT_CONVENTION);
  }
Beispiel #2
0
  public void memoryMove(long src, long dest, long len) {
    final Pointer srcPointer = new Pointer(src);
    final Pointer destPointer = new Pointer(dest);

    if (memmove != null)
      memmove.invoke(Pointer.class, new Object[] {destPointer, srcPointer, new NativeLong(len)});
    else if (bcopy != null)
      bcopy.invokeVoid(new Object[] {srcPointer, destPointer, new NativeLong(len)});
    else {
      if (src > dest) for (long n = 0; n < len; n++) destPointer.setByte(n, srcPointer.getByte(n));
      else for (long n = len - 1; n >= 0; n--) destPointer.setByte(n, srcPointer.getByte(n));
    }
  }
  static void setProgress(IdeFrame frame, double value, boolean isOk) {
    if (!isEnabled()) {
      return;
    }

    WinDef.HWND handle = getHandle(frame);
    mySetProgressState.invokeInt(
        new Object[] {myInterfacePointer, handle, isOk ? TBPF_NORMAL : TBPF_ERROR});
    mySetProgressValue.invokeInt(
        new Object[] {
          myInterfacePointer, handle, new WinDef.ULONGLONG((long) (value * 100)), TOTAL_PROGRESS
        });
  }
  static void hideProgress(IdeFrame frame) {
    if (!isEnabled()) {
      return;
    }

    mySetProgressState.invokeInt(
        new Object[] {myInterfacePointer, getHandle(frame), TBPF_NOPROGRESS});
  }
Beispiel #5
0
    public Object invoke(Object proxy, Method method, Object[] inArgs) throws Throwable {

      // Intercept Object methods
      if (OBJECT_TOSTRING.equals(method)) {
        return "Proxy interface to " + nativeLibrary;
      } else if (OBJECT_HASHCODE.equals(method)) {
        return new Integer(hashCode());
      } else if (OBJECT_EQUALS.equals(method)) {
        Object o = inArgs[0];
        if (o != null && Proxy.isProxyClass(o.getClass())) {
          return Function.valueOf(Proxy.getInvocationHandler(o) == this);
        }
        return Boolean.FALSE;
      }

      FunctionInfo f = null;
      synchronized (functions) {
        f = (FunctionInfo) functions.get(method);
        if (f == null) {
          f = new FunctionInfo();
          f.isVarArgs = Function.isVarArgs(method);
          if (invocationMapper != null) {
            f.handler = invocationMapper.getInvocationHandler(nativeLibrary, method);
          }
          if (f.handler == null) {
            // Find the function to invoke
            String methodName = functionMapper.getFunctionName(nativeLibrary, method);
            if (methodName == null) {
              // Just in case the function mapper screwed up
              methodName = method.getName();
            }
            f.function = nativeLibrary.getFunction(methodName, method);
            f.options = new HashMap(this.options);
            f.options.put(Function.OPTION_INVOKING_METHOD, method);
          }
          functions.put(method, f);
        }
      }
      if (f.isVarArgs) {
        inArgs = Function.concatenateVarArgs(inArgs);
      }
      if (f.handler != null) {
        return f.handler.invoke(proxy, method, inArgs);
      }
      return f.function.invoke(method.getReturnType(), inArgs, f.options);
    }
 public void testFunctionHoldsLibraryReference() throws Exception {
   NativeLibrary lib = NativeLibrary.getInstance("testlib");
   WeakReference ref = new WeakReference(lib);
   Function f = lib.getFunction("callCount");
   lib = null;
   System.gc();
   long start = System.currentTimeMillis();
   while (ref.get() != null && System.currentTimeMillis() - start < 2000) {
     Thread.sleep(10);
   }
   assertNotNull("Library GC'd when it should not be", ref.get());
   f.invokeInt(new Object[0]);
   f = null;
   System.gc();
   while (ref.get() != null && System.currentTimeMillis() - start < 5000) {
     Thread.sleep(10);
   }
   assertNull("Library not GC'd", ref.get());
 }
  static void setOverlayIcon(IdeFrame frame, Object icon, boolean dispose) {
    if (!isEnabled()) {
      return;
    }

    if (icon == null) {
      icon = Pointer.NULL;
    }
    mySetOverlayIcon.invokeInt(
        new Object[] {myInterfacePointer, getHandle(frame), icon, Pointer.NULL});
    if (dispose) {
      User32.INSTANCE.DestroyIcon((WinDef.HICON) icon);
    }
  }
Beispiel #8
0
  static {
    Function memmoveFc;
    try {
      memmoveFc = Function.getFunction(Platform.C_LIBRARY_NAME, "memmove");
    } catch (UnsatisfiedLinkError linkError) {
      memmoveFc = null;
    }

    Function bcopyFc;
    try {
      bcopyFc = Function.getFunction(Platform.C_LIBRARY_NAME, "bcopy");
    } catch (UnsatisfiedLinkError linkError) {
      bcopyFc = null;
    }

    memmove = memmoveFc;
    bcopy = bcopyFc;
    OLogManager.instance()
        .debug(
            CLibrary.class,
            "Following c library functions were found memmove : %s , bcopy : %s.",
            memmoveFc != null ? "yes" : "no",
            bcopyFc != null ? "yes" : "no");
  }
  public static int sched_getcpu() {
    final CLibrary lib = CLibrary.INSTANCE;
    try {
      final int ret = lib.sched_getcpu();
      if (ret < 0) {
        throw new IllegalStateException("sched_getcpu() failed; errno=" + Native.getLastError());
      }
      return ret;
    } catch (LastErrorException e) {
      throw new IllegalStateException("sched_getcpu() failed; errno=" + e.getErrorCode(), e);
    } catch (UnsatisfiedLinkError ule) {
      try {
        final IntByReference cpu = new IntByReference();
        final IntByReference node = new IntByReference();
        final int ret = lib.syscall(318, cpu, node, null);
        if (ret != 0) {
          throw new IllegalStateException("getcpu() failed; errno=" + Native.getLastError());
        }
        return cpu.getValue();
      } catch (LastErrorException lee) {
        if (lee.getErrorCode() == 38 && Platform.is64Bit()) { // unknown call
          final Pointer getcpuAddr = new Pointer((-10L << 20) + 1024L * 2L);
          final Function getcpu = Function.getFunction(getcpuAddr, Function.C_CONVENTION);
          final IntByReference cpu = new IntByReference();
          if (getcpu.invokeInt(new Object[] {cpu, null, null}) < 0) {
            throw new IllegalStateException("getcpu() failed; errno=" + Native.getLastError());

          } else {
            return cpu.getValue();
          }
        } else {
          throw new IllegalStateException("getcpu() failed; errno=" + lee.getErrorCode(), lee);
        }
      }
    }
  }
Beispiel #10
0
 public void testInterceptLastError() {
   if (!Platform.isWindows()) {
     return;
   }
   NativeLibrary kernel32 =
       (NativeLibrary) NativeLibrary.getInstance("kernel32", W32APIOptions.DEFAULT_OPTIONS);
   Function get = kernel32.getFunction("GetLastError");
   Function set = kernel32.getFunction("SetLastError");
   assertEquals("SetLastError should not be customized", Function.class, set.getClass());
   assertTrue(
       "GetLastError should be a Function", Function.class.isAssignableFrom(get.getClass()));
   assertTrue("GetLastError should be a customized Function", get.getClass() != Function.class);
   final int EXPECTED = 42;
   set.invokeVoid(new Object[] {new Integer(EXPECTED)});
   assertEquals("Wrong error", EXPECTED, get.invokeInt(null));
 }