feat: 提供了修改功能

.gitignore

@@ -22,3 +22,4 @@ dist-ssr
 *.njsproj
 *.sln
 *.sw?
+test_files

src-tauri/src/app_error.rs

@@ -21,6 +21,10 @@ pub enum AppError {
     /// 初始化打开文件失败
     #[error("初始化打开文件失败!: {0}")]
     InitOpenFileFailed(String),
+
+    /// 无法修改文件
+    #[error("无法修改文件!: {0}")]
+    CannotModifyFile(String),
 }
 
 impl serde::Serialize for AppError {

src-tauri/src/app_state.rs

@@ -17,14 +17,37 @@ impl AppState {
     /// 初始化打开文件
     pub fn init_open_file(&mut self, file_path: &str) -> Result<(), AppError> {
         self.file_path = Some(file_path.to_string());
-        self.mmap = Some(file::mmap_mut_file(file_path)?);
+        self.mmap = Some(file::mmap_file(file_path)?);
         // 读取PE格式来判断是否是64位
         let mmap: &Mmap = self.mmap.as_ref().unwrap();
         self.is_64_bit = Some(mmap.is_64_bit()?);
         Ok(())
     }
-    /// 获取文件内存映射
+
+    /// 获取文件内存映射的引用
     pub fn get_mmap_ref(&self) -> Result<&Mmap, AppError> {
         self.mmap.as_ref().ok_or(AppError::NoFileOpened)
     }
+
+    /// 设置文件内存映射
+    pub fn set_mmap(&mut self, mmap: Mmap) {
+        self.mmap = Some(mmap);
+    }
+
+    /// 在指定偏移处写入数据，此举会同步修改文件内容
+    pub fn write_data(&mut self, offset: usize, data: &[u8]) -> Result<(), AppError> {
+        // 1. 重新以可读可写打开文件
+        let file_path = self.file_path.as_ref().ok_or(AppError::NoFileOpened)?;
+        let mut rw_mmap = file::mmap_file_rw(file_path)?;
+        // 2. 向内存映射中写入数据
+        rw_mmap[offset..(offset + data.len())].copy_from_slice(data);
+        rw_mmap.flush()?;
+
+        // 3. 重新打开文件
+        self.mmap = Some(file::mmap_file(file_path)?);
+        self.is_64_bit = Some(self.get_mmap_ref()?.is_64_bit()?);
+
+        Ok(())
+    }
+
 }

src-tauri/src/commands.rs

@@ -30,7 +30,7 @@ pub fn set_complete(app: AppHandle) -> Result<(), AppError> {
 }
 
 // 命令，打开文件
-#[tauri::command]
+#[tauri::command(async)]
 pub fn command_open_file(
     file_path: &str,
     app_state: State<'_, Mutex<AppState>>,
@@ -43,7 +43,7 @@ pub fn command_open_file(
     Ok(())
 }
 
-#[tauri::command]
+#[tauri::command(async)]
 pub fn command_get_file_pe_node_tree_data(
     app_state: State<'_, Mutex<AppState>>,
 ) -> Result<Vec<PeNodeTreeData>, AppError> {
@@ -100,7 +100,7 @@ pub fn command_get_file_pe_node_tree_data(
 }
 
 // 命令，获取DOS头数据
-#[tauri::command]
+#[tauri::command(async)]
 pub fn command_get_pe_data_dos_header(
     app_state: State<'_, Mutex<AppState>>,
 ) -> Result<Value, AppError> {
@@ -114,7 +114,7 @@ pub fn command_get_pe_data_dos_header(
 }
 
 // 命令，获取NT头数据
-#[tauri::command]
+#[tauri::command(async)]
 pub fn command_get_pe_data_nt_header(
     app_state: State<'_, Mutex<AppState>>,
 ) -> Result<Value, AppError> {
@@ -129,7 +129,7 @@ pub fn command_get_pe_data_nt_header(
 }
 
 // 命令，获取文件头数据
-#[tauri::command]
+#[tauri::command(async)]
 pub fn command_get_pe_data_file_header(app_state: State<'_, Mutex<AppState>>) -> Result<Value, AppError> {
     let app_state = app_state.lock().unwrap();
     let mmap = app_state.get_mmap_ref()?;
@@ -142,7 +142,7 @@ pub fn command_get_pe_data_file_header(app_state: State<'_, Mutex<AppState>>) ->
 }
 
 // 命令，获取可选头数据
-#[tauri::command]
+#[tauri::command(async)]
 pub fn command_get_pe_data_optional_header(app_state: State<'_, Mutex<AppState>>) -> Result<Value, AppError> {
     let app_state = app_state.lock().unwrap();
     let mmap = app_state.get_mmap_ref()?;
@@ -157,7 +157,7 @@ pub fn command_get_pe_data_optional_header(app_state: State<'_, Mutex<AppState>>
 }
 
 // 命令，获取节区头数据
-#[tauri::command]
+#[tauri::command(async)]
 pub fn command_get_pe_data_section_headers(app_state: State<'_, Mutex<AppState>>) -> Result<Value, AppError> {
     let app_state = app_state.lock().unwrap();
     let mmap = app_state.get_mmap_ref()?;
@@ -169,3 +169,15 @@ pub fn command_get_pe_data_section_headers(app_state: State<'_, Mutex<AppState>>
         "base_offset": section_headers_offset,
     }))
 }
+
+// 命令，修改偏移处数据
+#[tauri::command(async)]
+pub fn command_write_data(
+    app_state: State<'_, Mutex<AppState>>,
+    offset: usize,
+    data: Vec<u8>,
+) -> Result<(), AppError> {
+    let mut app_state = app_state.lock().unwrap();
+    app_state.write_data(offset, &data)?;
+    Ok(())
+}

src-tauri/src/lib.rs

@@ -15,6 +15,7 @@ pub fn run() {
         .plugin(tauri_plugin_dialog::init())
         .plugin(tauri_plugin_shell::init())
         .invoke_handler(tauri::generate_handler![
+            commands::set_complete,
             commands::command_open_file,
             commands::command_get_file_pe_node_tree_data,
             commands::command_get_pe_data_dos_header,
@@ -22,7 +23,7 @@ pub fn run() {
             commands::command_get_pe_data_file_header,
             commands::command_get_pe_data_optional_header,
             commands::command_get_pe_data_section_headers,
-            commands::set_complete,
+            commands::command_write_data,
         ])
         .setup(|app| {
             app.manage(Mutex::new(app_state::AppState::default()));

src-tauri/src/services/file.rs

@@ -1,7 +1,7 @@
 use memmap2::*;
 
 /// 以只读的方式创建文件映射
-pub fn mmap_mut_file(file_path: &str) -> Result<Mmap, std::io::Error> {
+pub fn mmap_file(file_path: &str) -> Result<Mmap, std::io::Error> {
     let file = std::fs::OpenOptions::new().read(true).open(file_path)?;
     unsafe {
         MmapOptions::new()
@@ -9,3 +9,13 @@ pub fn mmap_mut_file(file_path: &str) -> Result<Mmap, std::io::Error> {
             .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e.to_string()))
     }
 }
+
+/// 以读写方式创建文件映射
+pub fn mmap_file_rw(file_path: &str) -> Result<MmapMut, std::io::Error> {
+    let file = std::fs::OpenOptions::new().read(true).write(true).open(file_path)?;
+    unsafe {
+        MmapOptions::new()
+            .map_mut(&file)
+            .map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e.to_string()))
+    }
+}

src/components/DosHeader/DosHeader.tsx

@@ -1,16 +1,73 @@
 import { useEffect, useState } from "react";
 import NodeTableComponent, {
   DataTemplateInterface,
+  DataType,
 } from "../NodeTableComponent/NodeTableComponent";
-import { invoke } from '@tauri-apps/api/core';
+import { invoke } from "@tauri-apps/api/core";
+import { message } from "antd";
+interface DosHeaderResponseData {
+  base_offset: number;
+  fields: {
+    [key: string]: number | number[];
+  };
+}
+
+const checkEditValue = (item: DataType) => {
+  const { value, data_type, size } = item;
+  if (data_type === "hex") {
+    let numberValue = value as number;
+    if (typeof value === "string") {
+      numberValue = parseInt(value, 10);
+    }
+    // 如果是2位 最大为 0xFFFF 10进制为 65535
+    let maxBit = Math.pow(2, size * 8) - 1;
+    console.log("maxBit", maxBit);
+    if (numberValue > maxBit) {
+      return false;
+    }
+    return true;
+  }
+};
+
+const formatEditValue = (item: DataType) => {
+  const { value, data_type, size } = item;
+  let bufs = [];
+
+  if (data_type === "hex") {
+    let numberValue = value as number;
+    if (typeof value === "string") {
+      numberValue = parseInt(value, 10);
+    }
+    // 转小尾
+    for (let i = 0; i < size; i++) {
+      bufs.push(numberValue & 0xff);
+      numberValue = numberValue >> 8;
+    }
+    console.log("bufs", bufs);
+    return bufs;
+  }
+};
 
 export default function DosHeader() {
-
+  const [data, setData] = useState<DosHeaderResponseData>();
-  const [data, setData] = useState<any[]>([]);
+  const [messageApi, contextHolder] = message.useMessage();
+  const success = () => {
+    messageApi.open({
+      type: "success",
+      content: "修改成功!",
+    });
+  };
+  const error = () => {
+    messageApi.open({
+      type: "error",
+      content: "修改失败!",
+    });
+  };
 
   const dataTemplate: DataTemplateInterface = {
     e_magic: {
       name: "e_magic",
+      description: "标识文件的魔数，DOS 可执行文件的值通常为 'MZ'。",
       value: 0,
       size: 2,
       offset: 0,
@@ -18,6 +75,7 @@ export default function DosHeader() {
     },
     e_cblp: {
       name: "e_cblp",
+      description: "文件最后一页的字节数，指示文件不完整页的大小。",
       value: 0,
       size: 2,
       offset: 2,
@@ -25,6 +83,7 @@ export default function DosHeader() {
     },
     e_cp: {
       name: "e_cp",
+      description: "文件的总页数（每页 512 字节）。",
       value: 0,
       size: 2,
       offset: 4,
@@ -32,6 +91,7 @@ export default function DosHeader() {
     },
     e_crlc: {
       name: "e_crlc",
+      description: "重定位表中的条目数。",
       value: 0,
       size: 2,
       offset: 6,
@@ -39,6 +99,7 @@ export default function DosHeader() {
     },
     e_cparhdr: {
       name: "e_cparhdr",
+      description: "文件头部的段数，单位为 16 字节段。",
       value: 0,
       size: 2,
       offset: 8,
@@ -46,6 +107,7 @@ export default function DosHeader() {
     },
     e_minalloc: {
       name: "e_minalloc",
+      description: "程序需要的最小额外段数。",
       value: 0,
       size: 2,
       offset: 10,
@@ -53,6 +115,7 @@ export default function DosHeader() {
     },
     e_maxalloc: {
       name: "e_maxalloc",
+      description: "程序可以分配的最大额外段数。",
       value: 0,
       size: 2,
       offset: 12,
@@ -60,6 +123,7 @@ export default function DosHeader() {
     },
     e_ss: {
       name: "e_ss",
+      description: "初始的 SS 寄存器值，段偏移地址。",
       value: 0,
       size: 2,
       offset: 14,
@@ -67,6 +131,7 @@ export default function DosHeader() {
     },
     e_sp: {
       name: "e_sp",
+      description: "初始的 SP 寄存器值，栈顶指针。",
       value: 0,
       size: 2,
       offset: 16,
@@ -74,6 +139,7 @@ export default function DosHeader() {
     },
     e_csum: {
       name: "e_csum",
+      description: "校验和（一般为 0）。",
       value: 0,
       size: 2,
       offset: 18,
@@ -81,6 +147,7 @@ export default function DosHeader() {
     },
     e_ip: {
       name: "e_ip",
+      description: "初始的 IP 寄存器值，代码段的入口偏移地址。",
       value: 0,
       size: 2,
       offset: 20,
@@ -88,6 +155,7 @@ export default function DosHeader() {
     },
     e_cs: {
       name: "e_cs",
+      description: "初始的 CS 寄存器值，代码段的段基址。",
       value: 0,
       size: 2,
       offset: 22,
@@ -95,6 +163,7 @@ export default function DosHeader() {
     },
     e_lfarlc: {
       name: "e_lfarlc",
+      description: "重定位表的文件偏移。",
       value: 0,
       size: 2,
       offset: 24,
@@ -102,6 +171,7 @@ export default function DosHeader() {
     },
     e_ovno: {
       name: "e_ovno",
+      description: "覆盖号（通常为 0）。",
       value: 0,
       size: 2,
       offset: 26,
@@ -109,15 +179,16 @@ export default function DosHeader() {
     },
     e_res: {
       name: "e_res",
+      description: "保留字段，通常为 0。",
       value: null,
       size: 8,
       offset: 28,
-      // field_type: "[WORD; 4]",
       array_element_size: 2,
       data_type: "array",
     },
     e_oemid: {
       name: "e_oemid",
+      description: "OEM 标识符。",
       value: 0,
       size: 2,
       offset: 36,
@@ -125,6 +196,7 @@ export default function DosHeader() {
     },
     e_oeminfo: {
       name: "e_oeminfo",
+      description: "OEM 信息。",
       value: 0,
       size: 2,
       offset: 38,
@@ -132,6 +204,7 @@ export default function DosHeader() {
     },
     e_res2: {
       name: "e_res2",
+      description: "保留字段，通常为 0。",
       value: 0,
       size: 20,
       offset: 40,
@@ -140,6 +213,7 @@ export default function DosHeader() {
     },
     e_lfanew: {
       name: "e_lfanew",
+      description: "PE 文件头的偏移地址。",
       value: 0,
       size: 4,
       offset: 60,
@@ -154,20 +228,20 @@ export default function DosHeader() {
       title: "字段名",
       dataIndex: "name",
       key: "name",
-      width: 120
+      width: 120,
     },
     {
       title: "偏移",
       dataIndex: "offset",
       key: "offset",
       hexSwitch: true,
-      width: 80
+      width: 80,
     },
     {
       title: "大小",
       dataIndex: "size",
       key: "size",
-      width: 80
+      width: 80,
     },
     {
       title: "值",
@@ -175,28 +249,56 @@ export default function DosHeader() {
       key: "value",
       hexSwitch: true,
       editable: true,
-      minWidth: 200
+      minWidth: 200,
     },
     {
       title: "描述信息",
       dataIndex: "description",
       key: "description",
-      width: 250
+      width: 250,
     },
   ];
 
-  useEffect(()=>{
+  const handleSave = (item: DataType) => {
-    invoke(command).then(res => {
+    console.log("handleSave", item);
-      console.log("asdfsadf",res);
+    if (!checkEditValue(item)) {
-      setData(res as any);
+      return;
+    }
+    const bufs = formatEditValue(item);
+    console.log("change value", bufs);
+    invoke("command_write_data", {
+      offset: item.offset,
+      data: bufs,
     })
-  }, [])
+      .then((res) => {
+        // 提示
+        success();
+        getData();
+      })
+      .catch(() => {
+        error();
+      });
+  };
+
+  const getData = () => {
+    invoke(command).then((res) => {
+      setData(res as any);
+    });
+  };
+
+  useEffect(() => {
+    getData();
+  }, []);
 
   return (
-    <NodeTableComponent
+    <>
-      dataTemplate={dataTemplate}
+      {contextHolder}
-      defaultData={data}
+      <NodeTableComponent
-      columns={columns as any}
+        dataTemplate={dataTemplate}
-    ></NodeTableComponent>
+        defaultData={data}
+        columns={columns as any}
+        onEdit={handleSave}
+      ></NodeTableComponent>
+    </>
   );
 }

src/components/NTHeader/NTHeader.tsx

@@ -14,6 +14,7 @@ export default function NTHeader() {
       size: 4,
       value: 0,
       data_type: "hex",
+      description: "PE文件的标志，可执行文件的标志是0x00004550",
     },
   };
   const command = "command_get_pe_data_nt_header";

src/components/NodeTableComponent/NodeTableComponent.tsx

@@ -54,7 +54,8 @@ interface NodeTableComponentProps {
   dataTemplate: DataTemplateInterface; // 数据模板
   // command: string; // 请求数据的命令
   columns: NodeTableColumnInterface<DataType>[];
-  defaultData?: DataType[];
+  defaultData?: ResponsData;
+  onEdit?: (record: DataType) => void;
 }
 
 interface EditableRowProps {
@@ -77,7 +78,7 @@ interface EditableCellProps {
   editable: boolean;
   dataIndex: keyof DataType;
   record: DataType;
-  handleSave: (record: DataType) => void;
+  handleSave?: (record: DataType) => void;
 }
 
 const EditableCell: React.FC<React.PropsWithChildren<EditableCellProps>> = ({
@@ -108,7 +109,7 @@ const EditableCell: React.FC<React.PropsWithChildren<EditableCellProps>> = ({
     try {
       const values = await form.validateFields();
       toggleEdit();
-      handleSave({ ...record, ...values });
+      handleSave?.({ ...record, ...values });
     } catch (errInfo) {
       console.log("Save failed:", errInfo);
     }
@@ -151,10 +152,6 @@ export default function NodeTableComponent(props: NodeTableComponentProps) {
     setHexSwitchStatus(newHexSwitchStatus);
   };
 
-  const handleSave = (_row: DataType) => {
-    // TODO: 保存逻辑
-  };
-
   const hexSwitchRender = (text: string, dataIndex: string) => {
     if (text == null || text == undefined || typeof text !== "number") {
       return text;
@@ -180,7 +177,7 @@ export default function NodeTableComponent(props: NodeTableComponentProps) {
           editable: col.editable,
           dataIndex: col.dataIndex,
           title: col.title,
-          handleSave,
+          handleSave: props.onEdit,
         });
       }
       if (col.hexSwitch) {

src/components/SectionHeaders/SectionHeaders.tsx

@@ -133,7 +133,6 @@ export default function SectionHeaders() {
       setData(formatDataHandler(res));
     });
   }, []);
-  console.log("yhyy", data);
   return (
     <Flex className={styles.content}>
       {data.map((item, index) => {