#12933 Resolve target file from Windows shortcut (.lnk) and open editor

plugins/org.jkiss.dbeaver.core/META-INF/MANIFEST.MF

@@ -22,6 +22,7 @@ Export-Package: org.jkiss.dbeaver,
  org.jkiss.dbeaver.ui.preferences,
  org.jkiss.dbeaver.ui.resources,
  org.jkiss.dbeaver.ui.resources.bookmarks,
+ org.jkiss.dbeaver.ui.resources.shortcuts,
  org.jkiss.dbeaver.ui.views,
  org.jkiss.dbeaver.ui.views.process,
  org.jkiss.dbeaver.ui.views.qm

plugins/org.jkiss.dbeaver.core/plugin.xml

@@ -29,6 +29,14 @@
             name="%content-type.org.jkiss.dbeaver.bookmark.name"
             priority="normal"
             describer="org.jkiss.dbeaver.ui.resources.bookmarks.BookmarkContentTypeDescriber"/>
+
+        <content-type file-extensions="lnk" id="org.jkiss.dbeaver.shortcut" name="Windows Shortcut" priority="normal">
+            <describer class="org.eclipse.core.runtime.content.BinarySignatureDescriber">
+                <!-- http://msdn.microsoft.com/en-us/library/dd871305%28PROT.10%29.aspx -->
+                <parameter name="signature" value="4C,00,00,00,01,14,02,00,00,00,00,00,C0,00,00,00,00,00,00,46"/>
+                <parameter name="offset" value="0"/>
+            </describer>
+        </content-type>
     </extension>
 
     <extension point="org.eclipse.core.expressions.propertyTesters">
@@ -1131,6 +1139,9 @@
             <root folder="Bookmarks"/>
             <contentTypeBinding contentTypeId="org.jkiss.dbeaver.bookmark"/>
         </handler>
+        <handler id="shortcuts" class="org.jkiss.dbeaver.ui.resources.shortcuts.ShortcutsHandlerImpl">
+            <contentTypeBinding contentTypeId="org.jkiss.dbeaver.shortcut"/>
+        </handler>
     </extension>
 
     <extension point="org.jkiss.dbeaver.ui.propertyConfigurator">

plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/ui/resources/shortcuts/ShortcutsHandlerImpl.java

+/*
+ * DBeaver - Universal Database Manager
+ * Copyright (C) 2010-2021 DBeaver Corp and others
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.jkiss.dbeaver.ui.resources.shortcuts;
+
+import org.eclipse.core.resources.IFile;
+import org.eclipse.core.resources.IResource;
+import org.eclipse.core.runtime.CoreException;
+import org.jkiss.code.NotNull;
+import org.jkiss.dbeaver.DBException;
+import org.jkiss.dbeaver.ui.UIUtils;
+import org.jkiss.dbeaver.ui.editors.EditorUtils;
+import org.jkiss.dbeaver.ui.resources.AbstractResourceHandler;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.ByteBuffer;
+import java.nio.ByteOrder;
+import java.nio.file.Files;
+
+/**
+ * Windows Shell Link (.lnk) handler
+ */
+public class ShortcutsHandlerImpl extends AbstractResourceHandler {
+    @Override
+    public void openResource(@NotNull IResource resource) throws CoreException, DBException {
+        try {
+            final File path = resource.getLocation().toFile();
+            final File resolved = resolve(path);
+
+            EditorUtils.openExternalFileEditor(resolved, UIUtils.getActiveWorkbenchWindow());
+        } catch (IOException e) {
+            throw new DBException("Error resolving shell link path", e);
+        }
+    }
+
+    @Override
+    public int getFeatures(IResource resource) {
+        if (resource instanceof IFile) {
+            return FEATURE_OPEN | FEATURE_DELETE | FEATURE_RENAME;
+        }
+        return super.getFeatures(resource);
+    }
+
+    /**
+     * Resolves absolute path from given Shell Link file.
+     * <p>
+     * See <a href="http://msdn.microsoft.com/en-us/library/dd871305%28PROT.10%29.aspx">Shell Link (.LNK) Binary File Format</a>
+     */
+    @NotNull
+    private static File resolve(@NotNull File shellLinkFile) throws IOException {
+        final ByteBuffer buf = ByteBuffer
+            .wrap(Files.readAllBytes(shellLinkFile.toPath()))
+            .order(ByteOrder.LITTLE_ENDIAN);
+
+        if ((buf.getInt(20) & 1) > 0) {
+            // Check the flags, if LinkTargetIDList is present, then seek to sizeof(ShellLinkHeader) + sizeof(LinkTargetIDList)
+            buf.position(buf.getInt(0) + buf.getShort(76) + 2);
+        } else {
+            // Otherwise, seek past linkFlags
+            buf.position(24);
+        }
+
+        // There's no better way to determine whether path is stored in unicode format
+        final boolean unicode = buf.getInt(buf.position() + 4) >= 36;
+        final StringBuilder path = new StringBuilder();
+
+        // Seek past fields we're not interested in (if `unicode` there are three additional fields)
+        buf.position(buf.position() + buf.getInt(buf.position() + (unicode ? 28 : 16)));
+
+        while (true) {
+            final char ch = unicode ? buf.getChar() : (char) buf.get();
+            if (ch == 0) {
+                // Hit string terminator, stop
+                break;
+            }
+            path.append(ch);
+            if (path.length() > 260) {
+                // Maximum path length on Windows is 260 characters. Let's pretend no one uses NTFS' `\\?\` prefix
+                break;
+            }
+        }
+
+        return new File(path.toString());
+    }
+}