ConfigParser把key转小写的bug,feed_child兼容性问题,添加css效果

Author: needle-wang

css.css

@@ -0,0 +1,19 @@
+/* 2019年 05月 17日 星期五 00:11:44 CST */
+
+checkbutton:checked {
+  background-color: #f6d488;
+  border-radius: 6px;
+  transition: 250ms ease-in-out;
+}
+
+button {
+  border-radius: 3px;
+  border-width: 1px;
+  border-style: solid;
+}
+
+button:active {
+  transition: 250ms ease-in-out;
+  border-style: inset;
+}
+

handlers.py

@@ -38,7 +38,7 @@ def build_all(self, button):
     # print(_final_line)
     if _final_line is not None:
       self.m._cmd_entry.set_text(_final_line.strip())
-      self.m._cmd_entry.grab_focus()
+      # self.m._cmd_entry.grab_focus()
 
   def run_cmdline(self, button):
     '''
@@ -50,9 +50,13 @@ def run_cmdline(self, button):
     if IS_POSIX:
       self.w.main_notebook.next_page()
       _cmdline_str = '%s %s\n' % (sqlmap_path, _sqlmap_opts)
-      # print(_cmdline_str)
+      # print(_cmdline_str, len(_cmdline_str.encode('utf8')))
       # self.m._page2_cmdline_str_label.set_text("running: " + _cmdline_str)
-      self.m._page2_terminal.feed_child(_cmdline_str, len(_cmdline_str))
+      if Vte.MAJOR_VERSION >= 0 and Vte.MINOR_VERSION > 52:
+        self.m._page2_terminal.feed_child_binary(_cmdline_str.encode('utf8'))
+      else:
+        # 旧版本的api, len()还要用encode后的长度
+        self.m._page2_terminal.feed_child(_cmdline_str, len(_cmdline_str.encode('utf8')))
       self.m._page2_terminal.grab_focus()
 
   def respawn_terminal(self, button):
@@ -757,9 +761,19 @@ def _escape_quote(self, widget_text):
 
 
 def main():
+  from widgets import d
   from sqlmap_gtk import Window
 
   win = Window()
+
+  css_provider = g.CssProvider.new()
+  css_provider.load_from_path('css.css')
+  g.StyleContext.add_provider_for_screen(
+    d.Screen.get_default(),
+    css_provider,
+    g.STYLE_PROVIDER_PRIORITY_APPLICATION
+  )
+
   win.connect('destroy', g.main_quit)
   win.show_all()
   g.main()

model.py

@@ -296,7 +296,7 @@ def __init__(self):
     self._page1_general_charset_entry = et()
     self._page1_general_encoding_ckbtn = cb('字符编码(用于数据获取)')
     self._page1_general_encoding_entry = et()
-    self._page1_general_session_file_ckbtn = cb('指定会话文件')
+    self._page1_general_session_file_ckbtn = cb('载入会话文件')
     self._page1_general_session_file_entry = FileEntry()
     self._page1_general_session_file_chooser = btn.new_with_label('打开')
     self._page1_general_output_dir_ckbtn = cb('输出的保存目录')

opts_gtk.py

@@ -138,7 +138,7 @@ def _build_page1_setting_tech(self, m):
     _row7.pack_end(m._tech_area_second_url_entry, True, True, 5)
 
     _row8 = Box()
-    _row8.pack_start(m._tech_area_second_req_ckbtn, True, True, 5)
+    _row8.pack_start(m._tech_area_second_req_ckbtn, False, True, 5)
 
     _row9 = Box()
     m._tech_area_second_req_chooser.connect(
@@ -295,7 +295,7 @@ def _build_page1_setting_inject(self, m):
     # set_active(True)为选中状态
     m._inject_area_skip_static_ckbtn.set_active(True)
 
-    _row2.pack_start(m._inject_area_skip_static_ckbtn, True, True, 5)
+    _row2.pack_start(m._inject_area_skip_static_ckbtn, False, True, 5)
 
     _row3 = Box()
     _row3.pack_start(m._inject_area_prefix_ckbtn, False, True, 5)
@@ -1175,18 +1175,26 @@ def _build_page1_other_general(self, m):
 
 
 def main():
+  from widgets import d
   from model import Model
   from handlers import Handler
 
   win = g.Window(title = 'sqlmap-options')
-  m = Model()
 
+  css_provider = g.CssProvider.new()
+  css_provider.load_from_path('css.css')
+  g.StyleContext.add_provider_for_screen(
+    d.Screen.get_default(),
+    css_provider,
+    g.STYLE_PROVIDER_PRIORITY_APPLICATION
+  )
+
+  m = Model()
   n = Notebook(m, Handler(win, m))
   win.add(n)
 
   win.connect('destroy', g.main_quit)
   win.show_all()
-
   g.main()
 
 

session.py

@@ -14,9 +14,14 @@ def __init__(self, m):
     '''
     m: model.Model
     '''
-    self._cfg = ConfigParser()
     self.m = m
 
+    self._cfg = ConfigParser()
+    # https://stackoverflow.com/questions/19359556/configparser-reads-capital-keys-and-make-them-lower-case
+    # 所有选项的key, 都会传给optionxform(), 该方法会将key转成小写!
+    # 将optionxform替换成str, 表示不做转换
+    self._cfg.optionxform = str
+
   def save_to_tmp(self):
     self._save_to_tmp_target()
     self._save_to_tmp_ckbtn()

sqlmap_gtk.py

@@ -529,10 +529,18 @@ def _build_page6(self):
 
 def main():
   win = Window()
+
+  css_provider = g.CssProvider.new()
+  css_provider.load_from_path('css.css')
+  g.StyleContext.add_provider_for_screen(
+    d.Screen.get_default(),
+    css_provider,
+    g.STYLE_PROVIDER_PRIORITY_APPLICATION
+  )
+
   win.connect('destroy', lambda x: win.on_window_destroy())
   # win.maximize()
   win.show_all()
-
   g.main()
 
 

tooltips.py

@@ -624,10 +624,19 @@ def _set_tooltip(self, tooltip, *widgets):
 
 
 def main():
-  from widgets import g
-  from sqlmap import Window
+  from widgets import d, g
+  from sqlmap_gtk import Window
 
   win = Window()
+
+  css_provider = g.CssProvider.new()
+  css_provider.load_from_path('css.css')
+  g.StyleContext.add_provider_for_screen(
+    d.Screen.get_default(),
+    css_provider,
+    g.STYLE_PROVIDER_PRIORITY_APPLICATION
+  )
+
   win.connect('destroy', g.main_quit)
   win.show_all()
   g.main()