Changeset 1108 in kBuild for trunk/src/kmk/kmkbuiltin

Timestamp:

Sep 23, 2007 8:33:10 AM (18 years ago)

Author:

bird

Message:

Added symlink link support (vista).

File:

• trunk/src/kmk/kmkbuiltin/mscfakes.c (modified) (5 diffs)

trunk/src/kmk/kmkbuiltin/mscfakes.c

@@ -31 +31 @@
 #include <io.h>
 #include <fcntl.h>
+#include <sys/stat.h>
 #include "err.h"
 #include "mscfakes.h"
+
+#define timeval windows_timeval
+#include <Windows.h>
+#undef timeval
 
 
@@ -55 +60 @@
     if (rc)
     {
-        int len = strlen(path);
+        size_t len = strlen(path);
         if (len > 0 && (path[len - 1] == '/' || path[len - 1] == '\\'))
         {
@@ -73 +78 @@
     if (rc)
     {
-        int len = strlen(path);
+        size_t len = strlen(path);
         if (len > 0 && (path[len - 1] == '/' || path[len - 1] == '\\'))
         {
@@ -126 +131 @@
 
 
+/** Unix to DOS. */
+static char *fix_slashes(char *psz)
+{
+    char *pszRet = psz;
+    for (; *psz; psz++)
+        if (*psz == '/')
+            *psz = '\\';
+    return pszRet;
+}
+
+
+/** Calcs the SYMBOLIC_LINK_FLAG_DIRECTORY flag for CreatesymbolcLink.  */
+static DWORD is_directory(const char *pszPath, const char *pszRelativeTo)
+{
+    size_t cchPath = strlen(pszPath);
+    struct stat st;
+    if (cchPath > 0 && pszPath[cchPath - 1] == '\\' || pszPath[cchPath - 1] == '/')
+        return 1; /* SYMBOLIC_LINK_FLAG_DIRECTORY */
+
+    if (stat(pszPath, &st))
+    {
+        size_t cchRelativeTo = strlen(pszRelativeTo);
+        char *psz = malloc(cchPath + cchRelativeTo + 4);
+        memcpy(psz, pszRelativeTo, cchRelativeTo);
+        memcpy(psz + cchRelativeTo, "\\", 1);
+        memcpy(psz + cchRelativeTo + 1, pszPath, cchPath + 1);
+        if (stat(pszPath, &st))
+            st.st_mode = _S_IFREG;
+        free(psz);
+    }
+
+    return (st.st_mode & _S_IFMT) == _S_IFDIR ? 1 : 0;
+}
+
+
 int symlink(const char *pszDst, const char *pszLink)
 {
+    static BOOL (WINAPI *s_pfnCreateSymbolicLinkA)(LPCSTR, LPCSTR, DWORD) = 0;
+    static BOOL s_fTried = FALSE;
+
+    if (!s_fTried)
+    {
+        HMODULE hmod = LoadLibrary("KERNEL32.DLL");
+        if (hmod)
+            *(FARPROC *)&s_pfnCreateSymbolicLinkA = GetProcAddress(hmod, "CreateSymbolicLinkA");
+        s_fTried = TRUE;
+    }
+
+    if (s_pfnCreateSymbolicLinkA)
+    {
+        char *pszDstCopy = fix_slashes(strdup(pszDst));
+        char *pszLinkCopy = fix_slashes(strdup(pszLink));
+        BOOL fRc = s_pfnCreateSymbolicLinkA(pszLinkCopy, pszDstCopy,
+                                            is_directory(pszDstCopy, pszLinkCopy));
+        DWORD err = GetLastError();
+        free(pszDstCopy);
+        free(pszLinkCopy);
+        if (fRc)
+            return 0;
+        switch (err)
+        {
+            case ERROR_NOT_SUPPORTED:       errno = ENOSYS; break;
+            case ERROR_ALREADY_EXISTS:
+            case ERROR_FILE_EXISTS:         errno = EEXIST; break;
+            case ERROR_DIRECTORY:           errno = ENOTDIR; break;
+            case ERROR_ACCESS_DENIED:
+            case ERROR_PRIVILEGE_NOT_HELD:  errno = EPERM; break;
+            default:                        errno = EINVAL; break;
+        }
+        return -1;
+    }
+
     errno = ENOSYS;
     err(1, "symlink() is not implemented on windows!");
@@ -160 +235 @@
     for (i = 0; i < count; i++)
     {
-        int cb = write(fd, vector[i].iov_base, vector[i].iov_len);
+        int cb = (int)write(fd, vector[i].iov_base, (int)vector[i].iov_len);
         if (cb < 0)
             return -1;