Gepetto

Gepetto

Gepetto เป็นปลั๊กอิน Python ซึ่งใช้โมเดลภาษาขนาดใหญ่ต่าง ๆ เพื่อให้ความหมายกับฟังก์ชั่นที่สลายตัวโดย IDA Pro (≥ 7.4) มันสามารถใช้ประโยชน์จากพวกเขาเพื่ออธิบายว่าฟังก์ชั่นทำอะไรและเปลี่ยนชื่อตัวแปรโดยอัตโนมัติ นี่คือตัวอย่างง่ายๆของผลลัพธ์ที่สามารถให้ได้ในไม่กี่วินาที:

การตั้งค่า

เพียงแค่วางสคริปต์นี้ ( gepetto.py รวมถึง gepetto/ โฟลเดอร์) ลงในโฟลเดอร์ IDA Plugins ของคุณ ( $IDAUSR/plugins ) โดยค่าเริ่มต้นบน Windows นี่ควรเป็น %AppData%Hex-RaysIDA Proplugins (คุณอาจต้องสร้างมัน)

คุณจะต้องเพิ่มแพ็คเกจที่จำเป็นในการติดตั้ง Python ของ IDA เพื่อให้สคริปต์ทำงานได้ ค้นหาว่าล่าม IDA ที่ใช้โดยการตรวจสอบคีย์รีจิสทรีต่อไปนี้: ComputerHKEY_CURRENT_USERSoftwareHex-RaysIDA (ค่าเริ่มต้นบน windows: %LOCALAPPDATA%ProgramsPythonPython39 ) ในที่สุดด้วยล่ามที่สอดคล้องกันเพียงแค่เรียกใช้:

 [/path/to/python] -m pip install -r requirements.txt

คุณจะต้องแก้ไขไฟล์การกำหนดค่า (พบเป็น gepetto/config.ini ) และเพิ่มปุ่ม API ของคุณเอง สำหรับ OpenAI สามารถพบได้ในหน้านี้ โปรดทราบว่าการสืบค้น API มักจะไม่ฟรี (แม้ว่าจะไม่แพงมาก) และคุณจะต้องตั้งค่าวิธีการชำระเงินกับผู้ให้บริการที่เกี่ยวข้อง

รุ่นที่รองรับ

• Openai
  • GPT-3.5-turbo-0125
  • GPT-4-turbo
  • GPT-4O (แนะนำสำหรับผู้เริ่มต้น)
• โอลลา
  • โมเดลท้องถิ่นใด ๆ ที่เปิดเผยผ่าน Ollama (จะไม่ปรากฏหาก Ollama ไม่ทำงาน)
• การทำ
  • LLAMA-3.1-70B-Versatile
  • LLAMA-3.2-90B-TEXT-PREVIEW
  • Mixtral-8x7b-32768
• ด้วยกัน
  • Mistralai/Mixtral-8x22b-Instruct-V0.1 (ไม่รองรับตัวแปรการเปลี่ยนชื่อ)
• Novita AI
  • Meta-llama/llama-3.1-70B-Instruct
  • Meta-llama/llama-3.1-405B-Instruct

การเพิ่มการสนับสนุนสำหรับโมเดลเพิ่มเติมไม่ควรยากเกินไปหากผู้ให้บริการใดก็ตามที่คุณกำลังพิจารณาเปิดเผย API ที่คล้ายกับ Openai ดูโฟลเดอร์ gepetto/models สำหรับแรงบันดาลใจหรือเปิดปัญหาหากคุณไม่สามารถคิดออกได้

การใช้งาน

เมื่อติดตั้งปลั๊กอินอย่างถูกต้องคุณควรจะสามารถเรียกใช้มันได้จากเมนูบริบทของหน้าต่าง Pseudocode ของ IDA ดังที่แสดงในภาพหน้าจอด้านล่าง:

สลับระหว่างรุ่นที่รองรับโดย gepetto จากเมนูแก้ไข> gepetto:

Gepetto ยังมีอินเทอร์เฟซ CLI ที่คุณสามารถใช้เพื่อถามคำถามกับ LLM โดยตรงจาก IDA ตรวจสอบให้แน่ใจว่าได้เลือก Gepetto ในแถบอินพุต:

คีย์

มีสายลัดต่อไปนี้:

• ขอให้โมเดลอธิบายฟังก์ชั่น: Ctrl + Alt + G
• ขอชื่อที่ดีกว่าสำหรับตัวแปรของฟังก์ชัน: Ctrl + Alt + R

การทดสอบเบื้องต้นแสดงให้เห็นว่าการขอชื่อที่ดีขึ้นทำงานได้ดีขึ้นถ้าคุณขอคำอธิบายของฟังก์ชั่นก่อน - ฉันคิดว่าเพราะโมเดลนั้นใช้ความคิดเห็นของตัวเองเพื่อให้คำแนะนำที่แม่นยำยิ่งขึ้น มีองค์ประกอบของการสุ่มเพื่อตอบกลับของ AI หากด้วยเหตุผลบางอย่างการตอบกลับครั้งแรกที่คุณได้รับไม่เหมาะกับคุณคุณสามารถเรียกใช้คำสั่งได้อีกครั้ง

ข้อ จำกัด

• ปลั๊กอินต้องการการเข้าถึง decompiler hexrays เพื่อใช้งาน
• LLMs ที่รองรับทั้งหมดเป็นวัตถุประสงค์ทั่วไปและอาจได้รับสิ่งผิดปกติ! เป็นสิ่งสำคัญในการส่งคืนผลลัพธ์เสมอ!

การแปล

คุณสามารถเปลี่ยนภาษาของ Gepetto ได้โดยการแก้ไขสถานที่ในการกำหนดค่า ตัวอย่างเช่นในการใช้ปลั๊กอินเป็นภาษาฝรั่งเศสคุณจะเพิ่ม:

 [Gepetto]
LANGUAGE = " fr_FR "

สถานที่ที่เลือกจะต้องตรงกับชื่อโฟลเดอร์ใน gepetto/locales หากภาษาที่ต้องการไม่สามารถใช้ได้คุณสามารถมีส่วนร่วมในโครงการโดยเพิ่มด้วยตัวคุณเอง! สร้างโฟลเดอร์ใหม่สำหรับสถานที่ที่ต้องการ (เช่น: gepetto/locales/de_DE/LC_MESSAGES/ ) และเปิดคำขอดึงใหม่ด้วยไฟล์ .po ที่อัปเดตซึ่งคุณสามารถสร้างได้โดยการคัดลอกและแก้ไข msgstr gepetto/locales/gepetto.pot ทั้งหมด

กิตติกรรมประกาศ

• Openai สำหรับการสร้างโมเดลที่น่าทึ่งเหล่านี้อย่างชัดเจน
• Hex Rays ผู้ผลิตของ Ida สำหรับการสนับสนุน Lightning Fast ของพวกเขา
• Kaspersky สำหรับการระดมทุนในขั้นต้นโครงการนี้
• Harfanglab ผู้สนับสนุนปัจจุบันทำให้งานนี้เป็นไปได้
• @vanhauser-thc สำหรับการสนับสนุนแนวคิดของแบบจำลองและผู้ให้บริการเพิ่มเติมเพื่อสนับสนุนผ่านทางส้อมของเขา
• ทุกคนที่มีส่วนร่วมในการแปล: @seifreed, @kot-igor, @ruzgarkanar, @orangetw