BinYars-SideWidget/binyarsdiraboutwidget.py at main · xorhex/BinYars-SideWidget · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
from PySide6.QtCore import Qt

from PySide6.QtWidgets import (
    QHBoxLayout,
    QLabel,
    QWidget,
    QPushButton,
    QSizePolicy,
    QStyle,
    QSpacerItem,
    QMessageBox,
)

from binaryninja.interaction import get_directory_name_input
from binaryninja import Settings

from binaryninja.log import Logger
from .binyarscanner import BinYarScanner
from .logo import BINYARS_LOGO_BASE64
from .constants import PLUGIN_SETTINGS_DIR

logger = Logger(session_id=0, logger_name=__name__)


class YaraRulesDirWidget(QWidget):
    def __init__(self, current_yara_rules_dir: str | None, parent=None):
        super().__init__(parent)

        scanner = BinYarScanner()
        self.yara_info = scanner.get_yara_version()
        # Horizontal layout
        layout = QHBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        layout.setSpacing(5)

        # QLabel Field Name
        self.yara_rules_label = QLabel("YARA-X Rules Directory: ")
        layout.addWidget(self.yara_rules_label)

        # QLabel For Value
        self.yara_rules_dir = QLabel()
        layout.addWidget(self.yara_rules_dir)

        # Horizontal spacer to push the button to the right
        spacer = QSpacerItem(
            40,
            20,
            QSizePolicy.Policy.Expanding,  # width expands
            QSizePolicy.Policy.Minimum,  # height stays minimal
        )
        layout.addItem(spacer)

        # Refresh QPushButton
        self.refresh_button = QPushButton("Refresh")
        layout.addWidget(self.refresh_button)

        self.update_label(current_yara_rules_dir)

        # Info button
        self.info_button = QPushButton()
        self.info_button.setIcon(
            self.style().standardIcon(QStyle.StandardPixmap.SP_MessageBoxInformation)
        )
        self.info_button.setToolTip("About BinYars")
        layout.addWidget(self.info_button)

        # Connect info button to popup
        self.info_button.clicked.connect(self.show_info_popup)

    def update_label(self, text: str, prompt: bool = False):
        """Helper method to update the QLabel text."""

        if text.strip() == "":
            if prompt:
                if text := self.yar_dir_prompt():
                    self.yara_rules_dir.setText(text)
                    self.refresh_button.setText("Refresh")
                    self.refresh_button.setIcon(
                        self.style().standardIcon(
                            QStyle.StandardPixmap.SP_BrowserReload
                        )
                    )
                else:
                    self.yara_rules_dir.setText("NEED TO SET")
                    self.refresh_button.setText("Set Dir")
                    self.refresh_button.setIcon(
                        self.style().standardIcon(
                            QStyle.StandardPixmap.SP_MessageBoxWarning
                        )
                    )
            else:
                self.yara_rules_dir.setText("NEED TO SET")
                self.refresh_button.setText("Set Dir")
                self.refresh_button.setIcon(
                    self.style().standardIcon(
                        QStyle.StandardPixmap.SP_MessageBoxWarning
                    )
                )
        else:
            self.yara_rules_dir.setText(text)
            self.refresh_button.setText("Refresh")
            self.refresh_button.setIcon(
                self.style().standardIcon(QStyle.StandardPixmap.SP_BrowserReload)
            )

    def yar_dir_prompt(self):
        if result := get_directory_name_input("Select the yara rules folder to use:"):
            Settings().set_string(PLUGIN_SETTINGS_DIR, result)
        return result

    def show_info_popup(self):
        """Show a rich-text popup with information about BinYars."""
        logo_data_uri = f"data:image/png;base64,{BINYARS_LOGO_BASE64.strip()}"
        info_text = f"""
        <html>
        <head>
            <style>
                body {{font - family: 'Segoe UI', sans-serif; }}
                h2 {{color: #2b6cb0; margin-bottom: 6px; }}
                p {{margin - top: 4px; margin-bottom: 8px; }}
                ul {{margin - left: 15px; }}
            </style>
        </head>
        <body>
            <table>
                <tr>
                    <td><img src='{logo_data_uri}' width='32' height='32'></td>
                    <td><h2>BinYars</h2></td>
                </tr>
            </table>

            <p><b>BinYars</b> is built leveraging <b>Binary Ninja</b> and <b>YARA-X</b>
            to enable powerful binary pattern analysis and rule-driven workflows.</p>

            <p><b>Developed By: </b><a href='https://blog.xorhex.com'>xorhex</a><p>

            <p>Plus a big <b>THANK YOU</b> to <b><a href='https://cxiao.net/'>cxiao</a></b> who provided quality feedback, making the plugin even better!</p>

            <p>Check for updates and learn more at:<br>
            <a href='https://github.com/xorhex/BinYars'>https://github.com/xorhex/BinYars</a></p>

            <p>Compiled with <a href='https://virustotal.github.io/yara-x/'>YARA-X</a> version: {self.yara_info["yara-x"]}</p>

            <p style='margin-top:12px; font-size: small; color: #666;'>
            © 2025 xorhex. MIT License
            </p>
        </body>
        </html>
        """

        msg = QMessageBox(self)
        msg.setWindowTitle("About BinYars")
        msg.setTextFormat(Qt.TextFormat.RichText)
        msg.setText(info_text)
        msg.setStandardButtons(QMessageBox.StandardButton.Ok)
        msg.setIcon(QMessageBox.Icon.Information)
        msg.exec()