---

SmbTool.java

Go to the documentation of this file.

package net.threebit.utils.sosc;

/*
        Copyright 2003 Shawn Deleurme
        Copyright 2003 Kevin O'Donnell

        This program is free software; you can redistribute it and/or modify
        it under the terms of the GNU General Public License as published by
        the Free Software Foundation; either version 2 of the License, or
        (at your option) any later version.

        This program is distributed in the hope that it will be useful,
        but WITHOUT ANY WARRANTY; without even the implied warranty of
        MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
        GNU General Public License for more details.

        You should have received a copy of the GNU General Public License
        along with this program; if not, write to the Free Software
        Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

import com.oreilly.servlet.*;
import java.io.*;
import java.sql.*;
import java.util.*;

public class SmbTool {

        static String smbClient = null;
        static {
                // Find out where smbClient is.
                String[] p = new String[] {
                        "/bin/smbclient",
                        "/sbin/smbclient",
                        "/usr/bin/smbclient",
                        "/usr/sbin/smbclient",
                        "/usr/local/bin/smbclient",
                        "/usr/local/sbin/smbclient",
                        "/usr/local/samba/bin/smbclient",
                        "/usr/local/samba/sbin/smbclient"
                };
                for (int x = 0; x < p.length; x++) {
                        if (new File(p[x]).exists()) {
                                smbClient = p[x];
                                break;
                        }
                }
                if (smbClient == null) {
                        String paths = "";
                        for (int x = 0; x < p.length; x++) { paths = p[x] + ";"; }
                        throw new RuntimeException("Could not find 'smbClient' {"+paths+")");
                }
        }

        private String userName = null;
        private String password = null;

        public SmbTool (String userName, String password) {
                this.userName = userName;
                this.password = password;
        }

        public static String getHostAndShare (String uncPath) throws Exception {
                if (!uncPath.startsWith("//")) { throw new Exception("UNC paths must begin with '//': '" + uncPath + "'"); }
                String[] path = uncPath.split("/");
                if (path.length < 4) { throw new Exception("Ugly UNC Path: '" + uncPath + "'"); }
                return "//" + path[2] + "/" + path[3];
        }

        public static String getRemotePath (String uncPath) throws Exception {
                String remotePath = "";
                if (!uncPath.startsWith("//")) { throw new Exception("UNC paths must begin with '//': '" + uncPath + "'"); }
                String[] path = uncPath.split("/");
                for (int x = 4; x < path.length-1; x++) { remotePath = remotePath + path[x] + "\\"; }
                remotePath = remotePath + path[path.length-1];
                return remotePath;
        }

        public StringBuffer get (String uncPath) throws Exception {
                String sharePath = getHostAndShare(uncPath);
                String remotePath = getRemotePath(uncPath);

                File tempFile = File.createTempFile("SmbToolGet","filedata");
                String[] cmd = new String[] {
                        smbClient,
                        sharePath,
                        password,
                        "-U",userName,
                        "-c",
                        "get " + remotePath + " " + tempFile.getAbsolutePath()
                };

                ProcTool worker = new ProcTool(cmd);
                worker.waitFor();

                // Suck the file up
                StringBuffer sb = new StringBuffer();
                BufferedReader br = new BufferedReader( new FileReader( tempFile ));
                String line;
                while ((line = br.readLine()) != null) {
                        sb.append(line + "\n");
                }

                return sb;
        }

        public ProcTool put (StringBuffer source, String uncPath) throws Exception {
                // Create a temporary file
                File tempFile = File.createTempFile("SmbToolPutbyStringBuffer","filedata");
                PrintWriter pw = new PrintWriter( new FileWriter(tempFile));
                pw.print(source.toString());
                pw.close();

                return put(tempFile.getAbsolutePath(), uncPath);
        }

        public ProcTool put (String source, String uncPath) throws Exception {
                String sharePath = getHostAndShare(uncPath);
                String remotePath = getRemotePath(uncPath);

                String[] cmd = new String[] {
                        smbClient,
                        sharePath,
                        password,
                        "-U",userName,
                        "-c",
                        "put " + source + " " + remotePath
                };
                return(new ProcTool(cmd));

        }

}

---