From 741e1fc82f678aa3366cb707eeb6bfffe043c6e5 Mon Sep 17 00:00:00 2001
From: Mike McCormack <mike@codeweavers.com>
Date: Tue, 16 Sep 2003 20:26:52 +0000
Subject: [PATCH] Implemented ImageGetCertificateData.

---
 dlls/imagehlp/integrity.c | 140 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 133 insertions(+), 7 deletions(-)

diff --git a/dlls/imagehlp/integrity.c b/dlls/imagehlp/integrity.c
index 8a30ffa59a8..93be8e4ea91 100644
--- a/dlls/imagehlp/integrity.c
+++ b/dlls/imagehlp/integrity.c
@@ -2,6 +2,7 @@
  *	IMAGEHLP library
  *
  *	Copyright 1998	Patrik Stridvall
+ *	Copyright 2003	Mike McCormack
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
@@ -23,11 +24,104 @@
 #include "windef.h"
 #include "winbase.h"
 #include "winerror.h"
+#include "winreg.h"
+#include "winternl.h"
+#include "winnt.h"
+#include "ntstatus.h"
 #include "imagehlp.h"
 #include "wine/debug.h"
 
 WINE_DEFAULT_DEBUG_CHANNEL(imagehlp);
 
+/*
+ * These functions are partially documented at:
+ *   http://www.cs.auckland.ac.nz/~pgut001/pubs/authenticode.txt
+ */
+
+/***********************************************************************
+ * IMAGEHLP_GetSecurityDirOffset (INTERNAL)
+ *
+ * Read a file's PE header, and return the offset and size of the 
+ *  security directory.
+ */
+static BOOL IMAGEHLP_GetSecurityDirOffset( HANDLE handle, DWORD num,
+                                           DWORD *pdwOfs, DWORD *pdwSize )
+{
+    IMAGE_DOS_HEADER dos_hdr;
+    IMAGE_NT_HEADERS nt_hdr;
+    DWORD size, count, offset, len;
+    BOOL r;
+    IMAGE_DATA_DIRECTORY *sd;
+
+    TRACE("handle %p\n", handle );
+
+    /* read the DOS header */
+    count = SetFilePointer( handle, 0, NULL, FILE_BEGIN );
+    if( count == INVALID_SET_FILE_POINTER )
+        return FALSE;
+    count = 0;
+    r = ReadFile( handle, &dos_hdr, sizeof dos_hdr, &count, NULL );
+    if( !r )
+        return FALSE;
+    if( count != sizeof dos_hdr )
+        return FALSE;
+
+    /* read the PE header */
+    count = SetFilePointer( handle, dos_hdr.e_lfanew, NULL, FILE_BEGIN );
+    if( count == INVALID_SET_FILE_POINTER )
+        return FALSE;
+    count = 0;
+    r = ReadFile( handle, &nt_hdr, sizeof nt_hdr, &count, NULL );
+    if( !r )
+        return FALSE;
+    if( count != sizeof nt_hdr )
+        return FALSE;
+
+    sd = &nt_hdr.OptionalHeader.
+                    DataDirectory[IMAGE_FILE_SECURITY_DIRECTORY];
+
+    TRACE("len = %lx addr = %lx\n", sd->Size, sd->VirtualAddress);
+
+    offset = 0;
+    size = sd->Size;
+
+    /* take the n'th certificate */
+    while( 1 )
+    {
+        /* read the length of the current certificate */
+        count = SetFilePointer( handle, sd->VirtualAddress + offset,
+                                 NULL, FILE_BEGIN );
+        if( count == INVALID_SET_FILE_POINTER )
+            return FALSE;
+        r = ReadFile( handle, &len, sizeof len, &count, NULL );
+        if( !r )
+            return FALSE;
+        if( count != sizeof len )
+            return FALSE;
+
+        /* check the certificate is not too big or too small */
+        if( len < sizeof len )
+            return FALSE;
+        if( len > (size-offset) )
+            return FALSE;
+        if( !num-- )
+            break;
+
+        /* calculate the offset of the next certificate */
+        offset += len;
+        if( offset >= size )
+            return FALSE;
+    }
+
+    *pdwOfs = sd->VirtualAddress + offset;
+    *pdwSize = len;
+
+    TRACE("len = %lx addr = %lx\n", len, sd->VirtualAddress + offset);
+
+    return TRUE;
+}
+
+
 /***********************************************************************
  *		ImageAddCertificate (IMAGEHLP.@)
  */
@@ -58,16 +152,48 @@ BOOL WINAPI ImageEnumerateCertificates(
 
 /***********************************************************************
  *		ImageGetCertificateData (IMAGEHLP.@)
+ *
+ *  FIXME: not sure that I'm dealing with the Index the right way
  */
 BOOL WINAPI ImageGetCertificateData(
-  HANDLE FileHandle, DWORD CertificateIndex,
-  PWIN_CERTIFICATE Certificate, PDWORD RequiredLength)
+                HANDLE handle, DWORD Index,
+                PWIN_CERTIFICATE Certificate, PDWORD RequiredLength)
 {
-  FIXME("(%p, %ld, %p, %p): stub\n",
-    FileHandle, CertificateIndex, Certificate, RequiredLength
-  );
-  SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
-  return FALSE;
+    DWORD r, offset, ofs, size, count;
+
+    TRACE("%p %ld %p %p\n", handle, Index, Certificate, RequiredLength);
+
+    if( !IMAGEHLP_GetSecurityDirOffset( handle, Index, &ofs, &size ) )
+        return FALSE;
+
+    if( !Certificate )
+    {
+        *RequiredLength = size;
+        return TRUE;
+    }
+
+    if( *RequiredLength < size )
+    {
+        *RequiredLength = size;
+        SetLastError( ERROR_INSUFFICIENT_BUFFER );
+        return FALSE;
+    }
+
+    *RequiredLength = size;
+
+    offset = SetFilePointer( handle, ofs, NULL, FILE_BEGIN );
+    if( offset == INVALID_SET_FILE_POINTER )
+        return FALSE;
+
+    r = ReadFile( handle, Certificate, size, &count, NULL );
+    if( !r )
+        return FALSE;
+    if( count != size )
+        return FALSE;
+
+    TRACE("OK\n");
+
+    return TRUE;
 }
 
 /***********************************************************************