Add hello world for r2ai with Google Bard

r2ai/bard.r2.js

@@ -0,0 +1,99 @@
+(function() {
+ 	// global settings
+	const settings = {
+		usePdc: true
+	}
+	function queryEsil(question) {
+		if (question.indexOf('expression') !== -1) {
+			const res = r2.cmd("aoeq");
+			bard (question + res);
+		} else if (question.indexOf('ranslate') !== -1) {
+			const res = r2.cmd("piq 1 @e:scr.color=0");
+			bard (question + ": " + res);
+		} else {
+			const res = r2.cmd("pdf @e:asm.bytes=0@e:asm.esil=true@e:scr.color=0");
+			const message = question + ":\n```\n" + res + '\n```\n';
+			bard (message);
+		}
+	}
+	function queryProgram(question) {
+		const res = r2.cmd("afs @@@F");
+		const quote = (x) => `"${x}"`;
+		if (res.length > 0) {
+			let message = "";
+			const fun = res[0];
+			message += 'Considering a program with the following functions:\n```';
+			message += res + '\n```\n';
+			message += question;
+			console.log(message);
+			bard(message);
+		} else {
+			console.error ("No function found");
+		}
+	}
+	function queryFunction(question) {
+		const res = r2.cmdj("afij");
+		const quote = (x) => `"${x}"`;
+		if (res.length > 0) {
+			let message = "";
+			const fun = res[0];
+			message += `The function have this signature '${fun.signature}'.\n`;
+			const pdsf = r2.cmd("pdsf@e:scr.color=0");
+			const imports = [];
+			const strings = [];
+			for (const line of pdsf.split(/\n/g)) {
+				const words = line.split(/ /g);
+				for (const word of words) {
+					if (word.startsWith("sym.imp.")) {
+						imports.push (word.slice(8));
+					}
+					if (word.startsWith("str.")) {
+						strings.push (word.slice(4));
+					}
+				}
+			}
+			if (imports.length > 0) {
+				message += " It is calling the following external functions: " + imports.join(', ') + ".\n";
+			}
+			if (strings.length > 0) {
+				message += " And uses these strings: " + strings.map(quote).join(', ') + ".\n";
+			}
+			if (settings.usePdc || (imports.length === 0 && strings.length === 0)) {
+				message += ' The pseudo code in base64 looks like:\n```c\n' + r2.cmd("pdc@e:scr.color=0") + '```';
+			}
+			message += question;
+			bard(message);
+		} else {
+			console.error ("No function found");
+		}
+	}
+	if (false) {
+		const
+		// message = "\nExplain the following ESIL expression: ";
+		// message = "\nOptimize and give me a decompilation in python of the given function in ESIL";
+		message = "\nTranslate the following instruction to ESIL";
+		queryEsil (message);
+	}
+	if (false) {
+		const
+		message = "\nGive me a frida script to hook the write function and print the arguments passed.";
+		queryProgram (message);
+	}
+	if (true) {
+		const
+		// message = "\nCan you give this function a better name?";
+		// message = "\nCan you provide a pseudocode in python?";
+		message = "\nPlease, explain what this function is doing.";
+		// message = "\nCan you optimize and decompile this function for me?";
+		queryFunction (message);
+	}
+	function bard(message) {
+		console.log(message);
+		r2.cmd("p6ds "+ b64(message) + " > q.txt");
+
+		// r2.call('!x="$(cat q.txt)"; bard-cli "$x"');
+		r2.call('!x="$(cat q.txt)"; bard-cli "$x"');
+		// console.log("\"!bard-cli `echo '" + b64(message) + "' | base64 -D`\"");
+		// r2.call("\"!bard-cli `echo '" + b64(message) + "' | base64 -D`\"");
+	}
+})();