]> wimlib.net Git - wimlib/blobdiff - src/symlink.c
rpfix extract on UNIX
[wimlib] / src / symlink.c
index 1e0109fde59a7b91f7fb085ed0550fc3e22e68c5..dc365e8cf5787a9b335e3ac77bce214f6ac853d3 100644 (file)
@@ -48,7 +48,7 @@ static ssize_t
 get_symlink_name(const void *resource, size_t resource_len, char *buf,
                 size_t buf_len, u32 reparse_tag)
 {
-       const u8 *p = resource;
+       const void *p = resource;
        u16 substitute_name_offset;
        u16 substitute_name_len;
        u16 print_name_offset;
@@ -57,9 +57,7 @@ get_symlink_name(const void *resource, size_t resource_len, char *buf,
        size_t link_target_len;
        ssize_t ret;
        unsigned header_size;
-       char *translated_target;
-       bool is_absolute;
-       u32 flags;
+       bool translate_slashes;
 
        if (resource_len < 12)
                return -EIO;
@@ -67,63 +65,81 @@ get_symlink_name(const void *resource, size_t resource_len, char *buf,
        p = get_u16(p, &substitute_name_len);
        p = get_u16(p, &print_name_offset);
        p = get_u16(p, &print_name_len);
-       get_u32(p, &flags);
 
        wimlib_assert(reparse_tag == WIM_IO_REPARSE_TAG_SYMLINK ||
                      reparse_tag == WIM_IO_REPARSE_TAG_MOUNT_POINT);
 
-       if (reparse_tag == WIM_IO_REPARSE_TAG_MOUNT_POINT) {
+       if (reparse_tag == WIM_IO_REPARSE_TAG_MOUNT_POINT)
                header_size = 8;
-       } else {
-               is_absolute = (flags & 1) ? false : true;
+       else {
                header_size = 12;
                p += 4;
        }
        if (header_size + substitute_name_offset + substitute_name_len > resource_len)
                return -EIO;
 
-       ret = utf16le_to_mbs((const utf16lechar*)(p + substitute_name_offset),
-                           substitute_name_len,
-                           &link_target, &link_target_len);
+       ret = utf16le_to_tstr((const utf16lechar*)(p + substitute_name_offset),
+                             substitute_name_len,
+                             &link_target, &link_target_len);
        if (ret)
                return -errno;
 
-       wimlib_assert(ret == 0);
-
-       if (!link_target)
-               return -EIO;
-
        if (link_target_len + 1 > buf_len) {
                ret = -ENAMETOOLONG;
                goto out;
        }
 
-       translated_target = link_target;
-       if (reparse_tag == WIM_IO_REPARSE_TAG_MOUNT_POINT || is_absolute) {
-               if (link_target_len < 7
-                     || memcmp(translated_target, "\\??\\", 4) != 0
-                     || translated_target[4] == '\0'
-                     || translated_target[5] != ':'
-                     || translated_target[6] != '\\') {
-                       ret = -EIO;
-                       goto out;
-               }
-               translated_target += 4;
-               link_target_len -= 4;
-               /* There's a drive letter, so just leave the backslashes since
-                * it won't go anyhwere on UNIX anyway...
-                *
-                * XXX
-                * NTFS-3g tries to re-map these links to actually point to
-                * something, so maybe we could do something like that here
-                * XXX*/
+       DEBUG("Interpeting substitute name \"%s\" (ReparseTag=0x%x)",
+             link_target, reparse_tag);
+       translate_slashes = true;
+       if (link_target_len >= 7 &&
+           link_target[0] == '\\' &&
+           link_target[1] == '?' &&
+           link_target[2] == '?' &&
+           link_target[3] == '\\' &&
+           link_target[4] != '\0' &&
+           link_target[5] == ':' &&
+           link_target[6] == '\\')
+       {
+               /* "Full" symlink or junction (\??\x:\ prefixed path) */
+               link_target += 6;
+               link_target_len -= 6;
+       } else if (reparse_tag == WIM_IO_REPARSE_TAG_MOUNT_POINT &&
+                  link_target_len >= 12 &&
+                  memcmp(link_target, "\\\\?\\Volume{", 11) == 0 &&
+                  link_target[link_target_len - 1] == '\\')
+       {
+               /* Volume junction.  Can't really do anything with it. */
+               translate_slashes = false;
+       } else if (reparse_tag == WIM_IO_REPARSE_TAG_SYMLINK &&
+                  link_target_len >= 3 &&
+                  link_target[0] != '\0' &&
+                  link_target[1] == ':' &&
+                  link_target[2] == '/')
+       {
+               /* "Absolute" symlink, with drive letter */
+               link_target += 2;
+               link_target_len -= 2;
+       } else if (reparse_tag == WIM_IO_REPARSE_TAG_SYMLINK &&
+                  link_target_len >= 1)
+       {
+               if (link_target[0] == '/')
+                       /* "Absolute" symlink, without drive letter */
+                       ;
+               else
+                       /* "Relative" symlink, without drive letter */
+                       ;
        } else {
-               for (size_t i = 0; i < link_target_len; i++)
-                       if (translated_target[i] == '\\')
-                               translated_target[i] = '/';
+               ERROR("Invalid reparse point: \"%s\"", link_target);
+               ret = -EIO;
+               goto out;
        }
 
-       memcpy(buf, translated_target, link_target_len + 1);
+       if (translate_slashes)
+               for (size_t i = 0; i < link_target_len; i++)
+                       if (link_target[i] == '\\')
+                               link_target[i] = '/';
+       memcpy(buf, link_target, link_target_len + 1);
        ret = link_target_len;
 out:
        FREE(link_target);
@@ -150,7 +166,7 @@ make_symlink_reparse_data_buf(const char *symlink_target,
        size_t len = 12 + name_utf16le_nbytes * 2;
        void *buf = MALLOC(len);
        if (buf) {
-               u8 *p = buf;
+               void *p = buf;
                p = put_u16(p, name_utf16le_nbytes); /* Substitute name offset */
                p = put_u16(p, name_utf16le_nbytes); /* Substitute name length */
                p = put_u16(p, 0); /* Print name offset */
@@ -176,7 +192,7 @@ make_symlink_reparse_data_buf(const char *symlink_target,
  */
 ssize_t
 inode_readlink(const struct wim_inode *inode, char *buf, size_t buf_len,
-              const WIMStruct *w, int read_resource_flags)
+              const WIMStruct *w, bool threadsafe)
 {
        const struct wim_lookup_table_entry *lte;
        int ret;
@@ -187,12 +203,12 @@ inode_readlink(const struct wim_inode *inode, char *buf, size_t buf_len,
        if (!lte)
                return -EIO;
 
-       if (wim_resource_size(lte) > 10000)
+       if (wim_resource_size(lte) > REPARSE_POINT_MAX_SIZE)
                return -EIO;
 
        u8 res_buf[wim_resource_size(lte)];
-       ret = read_full_wim_resource(lte, res_buf, read_resource_flags);
-       if (ret != 0)
+       ret = read_full_resource_into_buf(lte, res_buf, threadsafe);
+       if (ret)
                return -EIO;
        return get_symlink_name(res_buf, wim_resource_size(lte), buf,
                                buf_len, inode->i_reparse_tag);
@@ -224,7 +240,7 @@ inode_set_symlink(struct wim_inode *inode,
 
        ret = make_symlink_reparse_data_buf(target, &symlink_buf_len,
                                            &symlink_buf);
-       if (ret != 0)
+       if (ret)
                return ret;
 
        DEBUG("Made symlink reparse data buf (len = %zu, name len = %zu)",
@@ -248,7 +264,6 @@ inode_set_symlink(struct wim_inode *inode,
                lte->resource_location            = RESOURCE_IN_ATTACHED_BUFFER;
                lte->attached_buffer              = symlink_buf;
                lte->resource_entry.original_size = symlink_buf_len;
-               lte->resource_entry.size          = symlink_buf_len;
                copy_hash(lte->hash, symlink_buf_hash);
        }
 
@@ -269,4 +284,4 @@ out_free_symlink_buf:
        return ret;
 }
 
-#endif /* !defined(__WIN32__) && !defined(WITH_FUSE) */
+#endif /* !defined(__WIN32__) */