[TOMATO] 利用HTMLUnit网页自动化工具管理linksys路由器

可可火山

需求：
家里4M网络4人共享，6，7台电脑同时上网，网络流量和QoS管理的需求下，买了LinkSYS WRT54G，开始用DDWRT，但是它的QoS我的硬件太差所以选择了TOMATO。
TOMATO的工作很好，晚上大家上网游戏基本OK（有人开P2P软件的话还是有点卡的），白天大家都上班只有1，2台机器开着下载电影。这个时候QoS关掉比较好，开着速度被限制了（或者说TOMATO自带的QoS还不够灵活）。
问题：
路由器的QoS功能需要能定时打开关闭，我知道QoS是由路由器的tc控制的，但是我找不到命令管理界面对应的命令（有知道的请告知，谢谢）

解决：
后来我想既然是网页的管理界面，那就用一些网页自动化测试工具就应该可以实现吧，做了些常识，有些工具JavaScript出错，或者工作模式比较麻烦，最后用htmlunit实现了要的功能。

大概流程：
让htmlunit模拟浏览器，
打开QoS设置URL：http://192.168.1.1/qos-settings.asp
用户登录：
根据要求勾选Enable QoS 复选框。
点击Save按钮。

一开始我的Save一直没反应，因为这个路由器的界面很多是JavaScript写的，有些AJAX的概念。像“Save“按钮需要等待一下，具体详见HTMLUnit FAQ。

下面是我的代码：

1. 
   
2. //CopyLeft：可可火山 softwarezjut@gmail.com
   
3. import java.io.IOException;
   
4. import java.net.MalformedURLException;
   
5. 
   
6. import com.gargoylesoftware.htmlunit.DefaultCredentialsProvider;
   
7. import com.gargoylesoftware.htmlunit.FailingHttpStatusCodeException;
   
8. import com.gargoylesoftware.htmlunit.NicelyResynchronizingAjaxController;
   
9. import com.gargoylesoftware.htmlunit.Page;
   
10. import com.gargoylesoftware.htmlunit.WebClient;
    
11. 
    
12. import com.gargoylesoftware.htmlunit.html.HtmlButtonInput;
    
13. import com.gargoylesoftware.htmlunit.html.HtmlCheckBoxInput;
    
14. import com.gargoylesoftware.htmlunit.html.HtmlForm;
    
15. import com.gargoylesoftware.htmlunit.html.HtmlPage;
    
16. 
    
17. public class ToggleQoS {
    
18. 
    
19.     public static void main(String[] args) {
    
20. 
    
21.         WebClient webClient = new WebClient();
    
22.         HtmlPage page;
    
23.         boolean bqostoon = false;
    
24.         if (args.length > 0 && args[0].equalsIgnoreCase("on")) {
    
25.             bqostoon = true;
    
26.             System.out.println("I am going to turn on qos");
    
27.         } else {
    
28.             System.out.println("I am going to turn off qos");
    
29.         }
    
30. 
    
31.         try {
    
32.             DefaultCredentialsProvider login = new DefaultCredentialsProvider();
    
33.             login.addCredentials("admin", "admin");
    
34.             webClient.setCredentialsProvider(login);
    
35.             page = webClient.getPage("http://192.168.1.1/qos-settings.asp");
    
36. 
    
37.             HtmlForm form = page.getForms().get(0);
    
38. 
    
39.             HtmlCheckBoxInput qoscheckbox = form.getInputByName("f_qos_enable");
    
40.             boolean bqosnow = qoscheckbox.isChecked();
    
41. 
    
42.             if (bqosnow == bqostoon) {
    
43.                 System.out.println("QoS already " + (bqosnow ? "on" : "off"));
    
44.                 System.exit(0);
    
45.             }
    
46.             //Page qoscheckpage = null;
    
47.             if (bqostoon) {
    
48.                 System.out.println("now enable it");
    
49.                 qoscheckbox.setChecked(bqostoon);
    
50.             } else {
    
51.                 System.out.println("now disable it");
    
52.                 qoscheckbox.setChecked(bqostoon);
    
53.             }         
    
54. 
    
55.             //http://htmlunit.sourceforge.net/faq.html#AJAXDoesNotWork
    
56.             webClient.setAjaxController(new NicelyResynchronizingAjaxController());
    
57. 
    
58.             HtmlButtonInput saveBtn = (HtmlButtonInput)((page).getElementById("save-button"));
    
59.             if (saveBtn != null) {
    
60.                 System.out.println("click save");
    
61.                 saveBtn.click();
    
62.             }
    
63. 
    
64. 
    
65.         } catch (FailingHttpStatusCodeException e) {
    
66.             e.printStackTrace();
    
67.         } catch (MalformedURLException e) {
    
68.             e.printStackTrace();
    
69.         } catch (IOException e) {
    
70.             e.printStackTrace();
    
71.         }
    
72. 
    
73.     }
    
74. 
    
75. }
    

复制代码

QoS管理页面：

zhoutao0712

又看到你了，火山

0vk0

是用htmlunit实现了？

可可火山

回复 3# 0vk0


是的，htmlunit主要是用来web自动化测试的。同类型软件或产品很多。尝试了下这个支持javascript较好。

可可火山

回复 2# zhoutao0712

兄弟是哪位那？ zhoutao？

可可火山

update:20120130
我的LinkSys WRT54G v1.0被我折腾成砖头了，用修复线N次尝试未果，最后买了ZD改的WR-500U，硬件性能非常不错。

同时对于上面的问题，最近也找到更高效的方法。
这些路由器由nvram来管理配置，找到对应的 key就可以了。

比如我的WR-500U
Tomato DualWAN
版本 1.28.0542 MIPSR1 K26 USB WR-500U专用版

对应有3处QoS

• 新增功能-->智能QoS
• 新增功能-->QoS限制
• QoS 基本设置 //tomato传统QoS
  

对应的nvram分别是

• SQ_enable=1
• new_qoslimit_enable=1
• qos_enable=0
  

比如我要关闭 新增功能-->智能QoS的话
root@WR500U:/opt# nvram set SQ_enable=0
root@WR500U:/opt# nvram commit
Commit... done.

就可以了，非常方便和高效。