thieumao

Thêm vào file XML

Oct 1st, 2013
120
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 3.40 KB | None | 0 0
  1.     private void btNhapActionPerformed(java.awt.event.ActionEvent evt) {                                      
  2.         // TODO add your handling code here:
  3.         if (txtFile.getText().isEmpty()) {
  4.             JOptionPane.showMessageDialog(null, "Error!");
  5.         } else {
  6.             try {
  7.                 // Lay ma Hash
  8.                 strMD5 = ToMD5(file.getAbsolutePath());
  9.                 strSHA1 = ToSHA1(file.getAbsolutePath());
  10.                 strSHA256 = ToSHA256(file.getAbsolutePath());
  11.                 strSHA384 = ToSHA384(file.getAbsolutePath());
  12.                 strSHA512 = ToSHA512(file.getAbsolutePath());
  13.  
  14.                 // Doc file
  15.                 SAXBuilder builder = new SAXBuilder();
  16.                 File xmlFile = new File("MalwareData.xml");
  17.                 Document doc = (Document) builder.build(xmlFile);
  18.                 Element rootNode = doc.getRootElement();
  19.  
  20.                 // update them file
  21.                 Element objects = rootNode.getChild("objects");
  22.                 List<Element> test = objects.getChildren();
  23.                 //System.out.println(test.size() + 1);
  24.                 String IDnext = String.valueOf(test.size() + 1);
  25.                 Element eFile = new Element("file").setAttribute("ID", IDnext);
  26.                 eFile.addContent(new Element("md5").setText(strMD5));
  27.                 eFile.addContent(new Element("sha1").setText(strSHA1));
  28.                 eFile.addContent(new Element("sha256").setText(strSHA256));
  29.                 eFile.addContent(new Element("sha384").setText(strSHA384));
  30.                 eFile.addContent(new Element("sha512").setText(strSHA512));
  31.                 eFile.addContent(new Element("filename").setText(file.getName()));
  32.                 eFile.addContent(new Element("size").setText(String.valueOf(file.length()) + " byte"));
  33.                 int dot = file.getName().lastIndexOf(".");
  34.                 String strType = "NULL";
  35.                 if (dot >= 0) {
  36.                     strType = file.getName().substring(dot + 1).toUpperCase();
  37.                 }
  38.                 eFile.addContent(new Element("MINEType").setText(strType));
  39.                 String strMalwareName = txtSign.getText();
  40.                 if (txtSign.getText().isEmpty()) {
  41.                     strMalwareName = "NULL";
  42.                 }
  43.                 eFile.addContent(new Element("MalwareName").setText(strMalwareName));
  44.                 String strSign = txtSign.getText();
  45.                 if (txtSign.getText().isEmpty()) {
  46.                     strSign = "NULL";
  47.                 }
  48.                 eFile.addContent(new Element("Sign").setText(strSign));
  49.  
  50.                 objects.addContent(eFile);
  51.  
  52.  
  53.                 // cap nhat thoi gian
  54.                 DateFormat dateFormat = new SimpleDateFormat("HH:mm:ss dd/MM/yyyy");
  55.                 String timeCurrent = dateFormat.format(new Date());
  56.                 Element timestamp = rootNode.getChild("timestamp");
  57.                 timestamp.setText(timeCurrent);
  58.  
  59.                 // ghi file
  60.                 XMLOutputter xmlOutput = new XMLOutputter();
  61.                 xmlOutput.setFormat(Format.getPrettyFormat());
  62.                 xmlOutput.output(doc, new FileWriter("MalwareData.xml"));
  63.                 // Thong bao thanh cong
  64.                 System.out.println("File updated!");
  65.                 JOptionPane.showMessageDialog(null, "File updated!");
  66.             } catch (Exception ex) {
  67.             }
  68.         }
  69.     }
Advertisement
Add Comment
Please, Sign In to add comment