跳到主要內容

JCIFS Hello World

Introduction

JCIFS是java-based存取網路芳鄰通訊協定的API。

之所以接觸這個API,是因為好奇Windows管理工具psexec,能將Local程式放到Remote並執行的做法。後來看到類似工具paexec程式碼,與經過自己的測試,發現原理是將程式複製到遠端的管理共享資料夾,再透過WMI呼叫外部程式。

接下來將說明我有使用的API。

How to?

首先至link下載library。接下來就是寫寫寫。

Create File

大部分的操作都可以透過SmbFile類別完成。而與遠端的機器認證,是透過NTLM通訊協定。

	private static NtlmPasswordAuthentication createAuth(String aUser, String aPasswd){
		StringBuffer sb = new StringBuffer(aUser); 
		sb.append(':').append(aPasswd);
		return new NtlmPasswordAuthentication(sb.toString());
	}
 
	public static SmbFile createSmbFile(String aUser, String aPasswd, String aTarget) throws IOException {
		NtlmPasswordAuthentication auth = createAuth(aUser, aPasswd);
		return new SmbFile(aTarget, auth);
	}

如果有網域,可以使用

new NtlmPasswordAuthentication(domain, user, passwd);

在建立完SmbFile物件後,剩下其實就和操作一般檔案無異。

	public static void createFile(String aUser, String aPasswd, String aTarget, String aContent) throws IOException {
		SmbFile sFile = createSmbFile(aUser, aPasswd, aTarget);
		SmbFileOutputStream sfos = null;
		try {
			sfos = new SmbFileOutputStream(sFile);
			sfos.write(aContent.getBytes());
		} finally {
			close(sfos);
		}
	}

最後是關閉串流的處理。

	private static void close(Closeable aCloseable){
		if( aCloseable == null )
			return;
 
		try {
			aCloseable.close();
		} catch (IOException e) {
			//  log exception
		}
	}

Delete File

刪除檔案:

	public static void deleteFile(String aUser, String aPasswd, String aTarget) throws IOException {
		SmbFile sFile = createSmbFile(aUser, aPasswd, aTarget);
		sFile.delete();
	}

Exist

確認檔案是否存在:

	public static boolean exists(String aUser, String aPasswd, String aTarget) throws IOException {
		SmbFile sFile = createSmbFile(aUser, aPasswd, aTarget);
		return sFile.exists();
	}

Copy File

複製本地檔案到遠端:

	public static void copyFileTo(String aUser, String aPasswd, String aSource, String aTarget) throws IOException {
		SmbFile sFile = createSmbFile(aUser, aPasswd, aTarget);
		SmbFileOutputStream sfos = null;
		FileInputStream fis = null;
		try {
			sfos = new SmbFileOutputStream(sFile);
			fis = new FileInputStream(new File(aSource));
 
			byte[] buf = new byte[1024];
			int len;
			while(( len = fis.read(buf) )> 0 ){
				sfos.write(buf, 0, len);
			}
		} finally {
			close(sfos);
			close(fis);
		}
	}

複製遠端檔案到本地:

public static void copyFileFrom(String aUser, String aPasswd, String aSource, String aTarget) throws IOException {
		SmbFile sFile = createSmbFile(aUser, aPasswd, aSource);
		SmbFileInputStream sfis = null;
		FileOutputStream fos = null;
		try {
			sfis = new SmbFileInputStream(sFile);
			fos = new FileOutputStream(new File(aTarget));
 
			byte[] buf = new byte[1024];
			int len;
			while(( len = sfis.read(buf) )> 0 ){
				fos.write(buf, 0, len);
			}
		} finally {
			close(sfis);
			close(fos);
		}
	}

Test

簡單的寫了對檔案建立與複製的單元測試,這是對某台Win7機器的管理共享做存取。

public class SmbFileUtilTest {
 
	private String targetFolder = "smb://192.168.1.25/admin$/";
	private String user = "administrator";
	private String passwd = "123456";
 
	private String remoteTmpFile = null;
	private String localTmpFile = null;
 
	@After
	public void tearDown() throws IOException{
		if( remoteTmpFile != null ){
			SmbFileUtil.deleteFile(user, passwd, remoteTmpFile);
			assertFalse(SmbFileUtil.exists(user, passwd, remoteTmpFile));
		}
		if( localTmpFile != null ){
			new File(localTmpFile).delete();
		}
	}
 
	@Test
	public void copyFile() throws IOException{
		remoteTmpFile = targetFolder+"temp.txt";
		localTmpFile = "temp.tmp";
		SmbFileUtil.copyFileTo(user, passwd, "libs/jcifs-1.3.18.jar", remoteTmpFile);
		assertTrue(SmbFileUtil.exists(user, passwd, remoteTmpFile));
 
		SmbFileUtil.copyFileFrom(user, passwd, remoteTmpFile,  localTmpFile);
		File localFile = new File(localTmpFile);
		assertTrue(localFile.exists());
 
		assertEquals(new File("libs/jcifs-1.3.18.jar").length(), localFile.length());
	}
 
	@Test
	public void createFile() throws IOException{
		remoteTmpFile = targetFolder+"temp.txt";
		SmbFileUtil.createFile(user, passwd , remoteTmpFile, "test");
		assertTrue(SmbFileUtil.exists(user, passwd, remoteTmpFile));
	}
}

Summary

要將檔案複製到遠端,也可以透過

  1. 在本地建立網芳或Samba Server,讓遠端由本地複製檔案。
  2. 使用echo在遠端建立wget的VB Script,從本地的http server去wget檔案。

不管是以上兩種方法,或是透過管理共享,都是有可能因為使用者權限與安全性設定而失敗。只能見招拆招了。

Reference

留言

這個網誌中的熱門文章

解決RobotFramework從3.1.2升級到3.2.2之後,Choose File突然會整個Hand住的問題

考慮到自動測試環境的維護,我們很久以前就使用java去執行robot framework。前陣子開始處理從3.1.2升級到3.2.2的事情,主要先把明確的runtime語法錯誤與deprecate item處理好,這部分內容可以參考: link 。 直到最近才發現,透過SeleniumLibrary執行Choose File去上傳檔案的動作,會導致測試案例timeout。本篇文章主要分享心路歷程與解決方法,我也送了一條issue給robot framework: link 。 我的環境如下: RobotFramework: 3.2.2 Selenium: 3.141.0 SeleniumLibrary: 3.3.1 Remote Selenium Version: selenium-server-standalone-3.141.59 首先並非所有Choose File的動作都會hang住,有些測試案例是可以執行的,但是上傳一個作業系統ISO檔案一定會發生問題。後來我透過wireshark去比對新舊版本的上傳動作,因為我使用 Remote Selenium ,所以Selenium會先把檔案透過REST API發送到Remote Selenium Server上。從下圖我們可以發現,在3.2.2的最後一個TCP封包,比3.1.2大概少了500個bytes。 於是就開始了我trace code之路。包含SeleniumLibrary產生要送給Remote Selenium Server的request內容,還有HTTP Content-Length的計算,我都確認過沒有問題。 最後發現問題是出在socket API的使用上,就是下圖的這支code: 最後發現可能因為開始使用nio的方式送資料,但沒處理到尚未送完的資料內容,而導致發生問題。加一個loop去做計算就可以解決了。 最後我有把解法提供給robot framework官方,在他們出新的版本之前,我是將改完的_socket.py放在我們自己的Lib底下,好讓我們測試可以正常進行。(shutil.py應該也是為了解某個bug而產生的樣子..)

Show NIC selection when setting the network command with the device option

 Problem  在answer file中設定網卡名稱後,安裝時會停在以下畫面: 所使用的command參數如下: network --onboot = yes --bootproto =dhcp --ipv6 =auto --device =eth1 Diagnostic Result 這樣的參數,以前試驗過是可以安裝完成的。因此在發生這個問題後,我檢查了它的debug console: 從console得知,eth1可能是沒有連接網路線或者是網路太慢而導致的問題。後來和Ivy再三確認,有問題的是有接網路線的網卡,且問題是發生在activate階段: Solution 我想既然有retry應該就有次數或者timeout限制,因此發現在Anaconda的說明文件中( link ),有提到dhcptimeout這個boot參數。看了一些人的使用範例,應該是可以直接串在isolinux.cfg中,如下: default linux ksdevice = link ip =dhcp ks =cdrom: / ks.cfg dhcptimeout = 90 然而我在RHEL/CentOS 6.7與6.8試驗後都無效。 因此我就拿了顯示的錯誤字串,問問Google大師,想找一下Anaconda source code來看一下。最後找到別人根據Anaconda code修改的版本: link ,關鍵在於setupIfaceStruct函式中的setupIfaceStruct與readNetConfig: setupIfaceStruct: 會在dhcp時設定dhcptimeout。 readNetConfig: 在writeEnabledNetInfo將timeout寫入dhclient config中;在wait_for_iface_activation內會根據timeout做retry。 再來從log與code可以得知,它讀取的檔案是answer file而不是boot command line。因此我接下來的測試,就是在answer file的network command上加入dhcptimeout: network --onboot = yes --bootproto =dhcp --ipv6 =auto --device =eth1 --...

Robot Framework - Evaluate該怎麼用?

Evaluate該怎麼用? 前言 Builtin的RobotFramework Library提供了Evaluate Keyword。它所提供的功能是「執行Python描述句」。但實際上到底有什麼用途呢?原本我僅僅拿來將string轉為int的功用,經過一些查詢與試驗,我將心得整理給大家。 Builtin Builtin的function可以參考Library Doc for Evaluate。我以有使用過的function做說明。 數字轉換 Python提供了int、long、float與complex等function讓你可以將字串轉為數字,也可以透過它們做四則運算。首先以字串轉數字為例,我將8設於${num_str}中,再透過Evaluate+int轉為數字。這裡必須注意的是: 「int()中放變數必須以單引號'括起」。否則,假如你設定的數字為08,在轉換int時會出現Syntax Error。 ${num_str} | Set Variable | 8 ${num} | Evaluate | int('${num_str}') 其中int與long的第二個參數為base,這是根據你的input所決定: Comment | num = 9 ${num} | Evaluate | int('11', 8) Comment | num = 11 ${num} | Evaluate | int('11', 10) Comment | num = 17 ${num} | Evaluate | int('11', 16) 其它還有像bin、oct、hex,可以將整數轉為2、8、16進位。 運算 四則運算: 直接將運算子加上即可: ${num} | Evaluate | int('${hour}')*60 + int('${min}') 指數: 可以用pow。以下面兩個例子來說,第一個是2的3次方為8,第二個是2的3次方再mod 7為1。需注意的是: 「傳入值必須是數字不可為字串」。 ${num} | Evaluate | pow(2,3) ${num} | Evaluate | pow(2,3,7) 取最大最小值: 使用max/min,可以選擇丟一個array的方式...