added checks and crossfiles
This commit is contained in:
Félix
parent
53e4fae669
commit
b7883cd0aa
123
src/extension.js
123
src/extension.js
|
|
@ -1,37 +1,41 @@
|
|||
const vscode = require('vscode');
|
||||
const axios = require('axios');
|
||||
const cheerio = require('cheerio');
|
||||
const list = require('./mnemonics.js');
|
||||
|
||||
function activate(context) {
|
||||
// Existing hover provider
|
||||
// Hover provider for mnemonics
|
||||
const hoverProvider = vscode.languages.registerHoverProvider('x86asm', {
|
||||
async provideHover(document, position, token) {
|
||||
const range = document.getWordRangeAtPosition(position, /\b[a-zA-Z0-9_]+\b/);
|
||||
const word = document.getText(range);
|
||||
|
||||
if (word) {
|
||||
const instructionInfo = await fetchInstructionInfo(word.toLowerCase());
|
||||
if (instructionInfo) {
|
||||
return new vscode.Hover(instructionInfo);
|
||||
} else if (isLabel(document, word)) {
|
||||
return new vscode.Hover(`Label: ${word}`);
|
||||
}
|
||||
// Check if the word is a valid mnemonic
|
||||
if (list.x86x64Mnemonics.includes(word.toUpperCase())) {
|
||||
const instructionInfo = await fetchInstructionInfo(word.toLowerCase());
|
||||
if (instructionInfo) {
|
||||
return new vscode.Hover(instructionInfo);
|
||||
}
|
||||
} else {
|
||||
return new vscode.Hover(`Unrecognized mnemonic: ${word}`);
|
||||
}
|
||||
}
|
||||
|
||||
return undefined;
|
||||
}
|
||||
});
|
||||
|
||||
// New definition provider (Ctrl+Click to jump)
|
||||
// Definition provider (Ctrl+Click to jump)
|
||||
const definitionProvider = vscode.languages.registerDefinitionProvider('x86asm', {
|
||||
provideDefinition(document, position, token) {
|
||||
async provideDefinition(document, position, token) {
|
||||
const range = document.getWordRangeAtPosition(position);
|
||||
if (!range) {
|
||||
return;
|
||||
}
|
||||
|
||||
const word = document.getText(range);
|
||||
return findDefinition(document, word);
|
||||
return await findDefinition(word);
|
||||
}
|
||||
});
|
||||
|
||||
|
|
@ -76,79 +80,19 @@ async function fetchInstructionInfo(instruction) {
|
|||
}
|
||||
}
|
||||
|
||||
function findDefinition(document, word) {
|
||||
const text = document.getText();
|
||||
const lines = text.split('\n');
|
||||
async function findDefinition(word) {
|
||||
// Get all x86 assembly files in the workspace
|
||||
const files = await vscode.workspace.findFiles('**/*.s', '**/node_modules/**');
|
||||
|
||||
// Look for .def directive
|
||||
const defRegex = new RegExp(`\\.def\\s+${word}\\s*;`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (defRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
for (const file of files) {
|
||||
const document = await vscode.workspace.openTextDocument(file);
|
||||
const text = document.getText();
|
||||
const lines = text.split('\n');
|
||||
|
||||
// Look for label
|
||||
const labelRegex = new RegExp(`^${word}:`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (labelRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
// Look for extern declarations
|
||||
const externRegex = new RegExp(`^extern\\s+${word}\\s*`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (externRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
// Look for global declarations
|
||||
const globalRegex = new RegExp(`^global\\s+${word}\\s*`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (globalRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
// Look for struct definitions
|
||||
const structRegex = new RegExp(`^\\s*struc\\s+${word}\\s*`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (structRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
// Look for section declarations
|
||||
const sectionRegex = new RegExp(`^section\\s+${word}\\s*`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (sectionRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
// Look for macro definitions
|
||||
const macroRegex = new RegExp(`^\\s*%define\\s+${word}\\s*`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (macroRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
// Look for equates
|
||||
const equateRegex = new RegExp(`\\bequ\\s+${word}\\s*`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (equateRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
// Look for data declarations (db, dw, dd)
|
||||
const dataRegex = new RegExp(`\\b${word}\\s+`, 'i');
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (dataRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
// Check for definitions in each file
|
||||
const location = searchForDefinition(lines, word, document);
|
||||
if (location) {
|
||||
return location;
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -156,11 +100,18 @@ function findDefinition(document, word) {
|
|||
return undefined;
|
||||
}
|
||||
|
||||
// Helper function to check if a word is a label in the document
|
||||
function isLabel(document, word) {
|
||||
const text = document.getText();
|
||||
function searchForDefinition(lines, word, document) {
|
||||
const externRegex = new RegExp(`^extern\\s+${word}\\s*`, 'i');
|
||||
const globalRegex = new RegExp(`^global\\s+${word}\\s*`, 'i');
|
||||
const labelRegex = new RegExp(`^${word}:`, 'i');
|
||||
return labelRegex.test(text);
|
||||
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
if (externRegex.test(lines[i]) || globalRegex.test(lines[i]) || labelRegex.test(lines[i])) {
|
||||
return new vscode.Location(document.uri, new vscode.Position(i, 0));
|
||||
}
|
||||
}
|
||||
|
||||
return undefined; // If no match is found
|
||||
}
|
||||
|
||||
function deactivate() {}
|
||||
|
|
@ -168,4 +119,4 @@ function deactivate() {}
|
|||
module.exports = {
|
||||
activate,
|
||||
deactivate
|
||||
};
|
||||
};
|
||||
|
|
|
|||
55
src/mnemonics.js
Normal file
55
src/mnemonics.js
Normal file
|
|
@ -0,0 +1,55 @@
|
|||
const x86x64Mnemonics = [
|
||||
// Data Transfer
|
||||
"MOV", "LEA", "PUSH", "POP", "XCHG", "MOVSX", "MOVZX", "CVTSI2SD", "CVTSD2SI",
|
||||
"MOVD", "MOVQ", "MOVSS", "MOVSD", "MOVAPD", "MOVDQA", "MOVDQU",
|
||||
"MOVNTI", "MOVNTQ", "MOVNTDQ", "MOVNTDQA", "MOVNTQ", "MOVNTDQ",
|
||||
// Arithmetic and Logical
|
||||
"ADD", "SUB", "MUL", "DIV", "INC", "DEC", "AND", "OR", "XOR", "NOT", "SHL", "SHR",
|
||||
"SAL", "SAR", "SHRD", "SHLD", "IMUL", "IDIV", "IDIVQ", "MULSD", "DIVSD",
|
||||
"ADDPD", "SUBPD", "MULPD", "DIVPD", "ANDPD", "ORPD", "XORPD",
|
||||
"ADDPS", "SUBPS", "MULPS", "DIVPS", "ANDPS", "ORPS", "XORPS",
|
||||
// Control Flow
|
||||
"JMP", "Jcc", "CALL", "RET", "LOOP", "JNZ", "JZ", "JE", "JNE", "JL", "JLE",
|
||||
"JG", "JGE", "JB", "JBE", "JA", "JAE", "LOOPNE", "LOOPE",
|
||||
"LOOPNZ", "LOOPZ", "CMOVcc", "SETcc",
|
||||
// Other
|
||||
"CMP", "TEST", "NEG", "NOT", "CLD", "STD", "CWD", "CDQ", "CQO",
|
||||
"LEA", "LES", "LDS", "LSS", "LFS", "LGS",
|
||||
"LES", "LDS", "LSS", "LFS", "LGS",
|
||||
"REP", "REPE", "REPNE", "REPZ", "REPNZ",
|
||||
"BSF", "BSR", "BT", "BTC", "BTR", "BTS", "LEAVE",
|
||||
// Floating-Point
|
||||
"FADD", "FSUB", "FMUL", "FDIV", "FCHS", "FABS", "FCOM", "FCOMP", "FCOMPP",
|
||||
"FADD", "FSUB", "FMUL", "FDIV", "FCHS", "FABS", "FCOM", "FCOMP", "FCOMPP",
|
||||
"FST", "FSTPN", "FSTP", "FLD", "FLD1", "FLDPI", "FLDL2T", "FLDL2E",
|
||||
"FLDCW", "FSTCW",
|
||||
"FCOMI", "FCOMIP", "FUCOMI", "FUCOMIP",
|
||||
// SIMD
|
||||
"MOVDQA", "MOVDQU", "PSHUFD", "PSHUFLW", "PSHUFLH", "PSLLD", "PSLLQ", "PSLLW",
|
||||
"PSRAD", "PSRAQ", "PSRAW", "PADDW", "PADDQ", "PADDUSB", "PADDUSW",
|
||||
"PSUBW", "PSUBQ", "PSUBUSB", "PSUBUSW",
|
||||
"PMADDWD", "PMULHW", "PMULLW",
|
||||
"PMAXSW", "PMAXUW", "PMINSW", "PMINUW",
|
||||
// AVX Instructions
|
||||
"VMOVD", "VMOVQ", "VMOVSS", "VMOVSD", "VMOVAPD", "VMOVDQA", "VMOVDQU",
|
||||
"VADDPD", "VADDPS", "VADDL", "VADDL2", "VADDL3", "VADDL4",
|
||||
"VSUBPD", "VSUBPS", "VSUBL", "VSUBL2", "VSUBL3", "VSUBL4",
|
||||
"VMULPD", "VMULPS", "VMULL", "VMULL2", "VMULL3", "VMULL4",
|
||||
"VDIVPD", "VDIVPS", "VDIVL", "VDIVL2", "VDIVL3", "VDIVL4",
|
||||
"VANDPD", "VANDPS", "VANDL", "VANDL2", "VANDL3", "VANDL4",
|
||||
"VORPD", "VORPS", "VORL", "VORL2", "VORL3", "VORL4",
|
||||
"VXORPD", "VXORPS", "VXORL", "VXORL2", "VXORL3", "VXORL4",
|
||||
// AVX-512 Instructions
|
||||
"VPMOVD", "VPMOVQ", "VPMOVSS", "VPMOVSD", "VPMOVAPD", "VPMOVDQA", "VPMOVDQU",
|
||||
"VPADDPD", "VPADDPS", "VPADDL", "VPADDL2", "VPADDL3", "VPADDL4",
|
||||
"VPSUBPD", "VPSUBPS", "VPSUBL", "VPSUBL2", "VPSUBL3", "VPSUBL4",
|
||||
"VPMULPD", "VPMULPS", "VPMULL", "VPMULL2", "VPMULL3", "VPMULL4",
|
||||
"VPDIVPD", "VPDIVPS", "VPDIVL", "VPDIVL2", "VPDIVL3", "VPDIVL4",
|
||||
"VPANDPD", "VPANDPS", "VPANDL", "VPANDL2", "VPANDL3", "VPANDL4",
|
||||
"VPORPD", "VPORPS", "VPORL", "VPORL2", "VPORL3", "VPORL4",
|
||||
"VPXORPD", "VPXORPS", "VPXORL", "VPXORL2", "VPXORL3", "VPXORL4",
|
||||
];
|
||||
|
||||
module.exports = {
|
||||
x86x64Mnemonics
|
||||
};
|
||||
Loading…
Reference in a new issue