jcifs.dcerpc.DcerpcBinding Maven / Gradle / Ivy

/* jcifs msrpc client library in Java
 * Copyright (C) 2006  "Michael B. Allen" 
 *                     "Eric Glass" 
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

package jcifs.dcerpc;


import java.util.HashMap;
import java.util.Locale;
import java.util.Map;
import java.util.Map.Entry;

import jcifs.dcerpc.msrpc.lsarpc;
import jcifs.dcerpc.msrpc.netdfs;
import jcifs.dcerpc.msrpc.samr;
import jcifs.dcerpc.msrpc.srvsvc;


/**
 *
 */
public class DcerpcBinding {

    private static final Map INTERFACES = new HashMap<>();


    static {
        INTERFACES.put("srvsvc", srvsvc.getSyntax());
        INTERFACES.put("lsarpc", lsarpc.getSyntax());
        INTERFACES.put("samr", samr.getSyntax());
        INTERFACES.put("netdfs", netdfs.getSyntax());
        INTERFACES.put("netlogon", "12345678-1234-abcd-ef00-01234567cffb:1.0");
        INTERFACES.put("wkssvc", "6BFFD098-A112-3610-9833-46C3F87E345A:1.0");
        INTERFACES.put("samr", "12345778-1234-ABCD-EF00-0123456789AC:1.0");
    }


    /**
     * Add an interface to the registry
     * 
     * @param name
     * @param syntax
     */
    public static void addInterface ( String name, String syntax ) {
        INTERFACES.put(name, syntax);
    }

    private String proto;
    private Map options = null;
    private String server;
    private String endpoint = null;
    private UUID uuid = null;
    private int major;
    private int minor;


    DcerpcBinding ( String proto, String server ) {
        this.proto = proto;
        this.server = server;
    }


    /**
     * @return the proto
     */
    public String getProto () {
        return this.proto;
    }


    /**
     * @return the options
     */
    public Map getOptions () {
        return this.options;
    }


    /**
     * @return the server
     */
    public String getServer () {
        return this.server;
    }


    /**
     * @return the endpoint
     */
    public String getEndpoint () {
        return this.endpoint;
    }


    /**
     * @return the uuid
     */
    UUID getUuid () {
        return this.uuid;
    }


    /**
     * @return the major
     */
    int getMajor () {
        return this.major;
    }


    /**
     * @return the minor
     */
    int getMinor () {
        return this.minor;
    }


    void setOption ( String key, Object val ) throws DcerpcException {
        if ( key.equals("endpoint") ) {
            this.endpoint = val.toString();
            String lep = this.endpoint.toLowerCase(Locale.ENGLISH);
            if ( lep.startsWith("\\pipe\\") ) {
                String iface = INTERFACES.get(lep.substring(6));
                if ( iface != null ) {
                    int c, p;
                    c = iface.indexOf(':');
                    p = iface.indexOf('.', c + 1);
                    this.uuid = new UUID(iface.substring(0, c));
                    this.major = Integer.parseInt(iface.substring(c + 1, p));
                    this.minor = Integer.parseInt(iface.substring(p + 1));
                    return;
                }
            }
            throw new DcerpcException("Bad endpoint: " + this.endpoint);
        }
        if ( this.options == null )
            this.options = new HashMap<>();
        this.options.put(key, val);
    }


    Object getOption ( String key ) {
        if ( key.equals("endpoint") )
            return this.endpoint;
        if ( this.options != null )
            return this.options.get(key);
        return null;
    }


    @Override
    public String toString () {
        String ret = this.proto + ":" + this.server + "[" + this.endpoint;
        if ( this.options != null ) {
            for ( Entry entry : this.options.entrySet() ) {
                ret += "," + entry.getKey() + "=" + entry.getValue();
            }
        }
        ret += "]";
        return ret;
    }
}