summaryrefslogtreecommitdiffstats
path: root/vpp-api/java/japi/org/openvpp/vppjapi/vppConn.java
blob: 3e8c12a93f9fb8595d089114ee541a465898b570 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
/*
 * Copyright (c) 2015 Cisco and/or its affiliates.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.openvpp.vppjapi;

import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;
import java.nio.file.attribute.PosixFilePermission;
import java.nio.file.attribute.PosixFilePermissions;
import java.util.Set;

import org.openvpp.vppjapi.vppVersion;
import org.openvpp.vppjapi.vppInterfaceDetails;
import org.openvpp.vppjapi.vppInterfaceCounters;
import org.openvpp.vppjapi.vppBridgeDomainDetails;
import org.openvpp.vppjapi.vppIPv4Address;
import org.openvpp.vppjapi.vppIPv6Address;
import org.openvpp.vppjapi.vppVxlanTunnelDetails;

public class vppConn implements AutoCloseable {
    private static final String LIBNAME = "libvppjni.so.0.0.0";

    static {
        try {
            loadLibrary();
        } catch (Exception e) {
            System.out.printf("Can't find vpp jni library: %s\n", LIBNAME);
            throw new ExceptionInInitializerError(e);
        }
    }

    private static void loadStream(final InputStream is) throws IOException {
        final Set<PosixFilePermission> perms = PosixFilePermissions.fromString("rwxr-x---");
        final Path p = Files.createTempFile(LIBNAME, null, PosixFilePermissions.asFileAttribute(perms));
        try {
            Files.copy(is, p, StandardCopyOption.REPLACE_EXISTING);

            try {
                Runtime.getRuntime().load(p.toString());
            } catch (UnsatisfiedLinkError e) {
                throw new IOException(String.format("Failed to load library %s", p), e);
            }
        } finally {
            try {
                Files.deleteIfExists(p);
            } catch (IOException e) {
            }
        }
    }

    private static void loadLibrary() throws IOException {
      try (final InputStream is = vppConn.class.getResourceAsStream('/' + LIBNAME)) {
          if (is == null) {
              throw new IOException(String.format("Failed to open library resource %s",
                                                LIBNAME));
          }
          loadStream(is);
        }
    }

    private static vppConn currentConnection = null;
    private final String clientName;
    private volatile boolean disconnected = false;

    // Hidden on purpose to prevent external instantiation
    vppConn(final String clientName) throws IOException {
        this.clientName = clientName;

        synchronized (vppConn.class) {
            if (currentConnection != null) {
                throw new IOException("Already connected as " + currentConnection.clientName);
            }

            final int ret = clientConnect(clientName);
            if (ret != 0) {
                throw new IOException("Connection returned error " + ret);
            }

            currentConnection = this;
        }
    }

    @Override
    public synchronized final void close() {
        if (!disconnected) {
            disconnected = true;

            synchronized (vppConn.class) {
                clientDisconnect();
                currentConnection = null;
            }
        }
    }

    /**
     * Check if this instance is connected.
     *
     * @throws IllegalStateException if this instance was disconnected.
     */
    protected final void checkConnected() {
        if (disconnected) {
            throw new IllegalStateException("Disconnected client " + clientName);
        }
    }

    public final int getRetval(int context, int release) {
        checkConnected();
        return getRetval0(context, release);
    }

    public final String getInterfaceList (String nameFilter) {
        checkConnected();
        return getInterfaceList0(nameFilter);
    }

    public final int swIfIndexFromName (String interfaceName) {
        checkConnected();
        return swIfIndexFromName0(interfaceName);
    }

    public final String interfaceNameFromSwIfIndex (int swIfIndex) {
        checkConnected();
        return interfaceNameFromSwIfIndex0(swIfIndex);
    }

    public final void clearInterfaceTable () {
        checkConnected();
        clearInterfaceTable0();
    }

    public final vppInterfaceDetails[] swInterfaceDump (byte nameFilterValid, byte [] nameFilter) {
        checkConnected();
        return swInterfaceDump0(nameFilterValid, nameFilter);
    }

    public final int bridgeDomainIdFromName(String bridgeDomain) {
        checkConnected();
        return bridgeDomainIdFromName0(bridgeDomain);
    }

    public final int findOrAddBridgeDomainId(String bridgeDomain) {
        checkConnected();
        return findOrAddBridgeDomainId0(bridgeDomain);
    }

    public final vppVersion getVppVersion() {
        checkConnected();
        return getVppVersion0();
    }

    public final vppInterfaceCounters getInterfaceCounters(int swIfIndex) {
        checkConnected();
        return getInterfaceCounters0(swIfIndex);
    }

    public final int[] bridgeDomainDump(int bdId) {
        checkConnected();
        return bridgeDomainDump0(bdId);
    }

    public final vppBridgeDomainDetails getBridgeDomainDetails(int bdId) {
        checkConnected();
        return getBridgeDomainDetails0(bdId);
    }

    public final vppL2Fib[] l2FibTableDump(int bdId) {
        checkConnected();
        return l2FibTableDump0(bdId);
    }

    public final int bridgeDomainIdFromInterfaceName(String interfaceName) {
        checkConnected();
        return bridgeDomainIdFromInterfaceName0(interfaceName);
    }

    public final vppIPv4Address[] ipv4AddressDump(String interfaceName) {
        checkConnected();
        return ipv4AddressDump0(interfaceName);
    }

    public final vppIPv6Address[] ipv6AddressDump(String interfaceName) {
        checkConnected();
        return ipv6AddressDump0(interfaceName);
    }

    public final vppVxlanTunnelDetails[] vxlanTunnelDump(int swIfIndex) {
        checkConnected();
        return vxlanTunnelDump0(swIfIndex);
    }

    public final int setInterfaceDescription(String ifName, String ifDesc) {
        checkConnected();
        return setInterfaceDescription0(ifName, ifDesc);
    }

    public final String getInterfaceDescription(String ifName) {
        checkConnected();
        return getInterfaceDescription0(ifName);
    }

    private static native int clientConnect(String clientName);
    private static native void clientDisconnect();
    private static native int getRetval0(int context, int release);
    private static native String getInterfaceList0(String nameFilter);
    private static native int swIfIndexFromName0(String interfaceName);
    private static native String interfaceNameFromSwIfIndex0(int swIfIndex);
    private static native void clearInterfaceTable0();
    private static native vppInterfaceDetails[] swInterfaceDump0(byte nameFilterValid, byte [] nameFilter);
    private static native int bridgeDomainIdFromName0(String bridgeDomain);
    private static native int findOrAddBridgeDomainId0(String bridgeDomain);
    private static native vppVersion getVppVersion0();
    private static native vppInterfaceCounters getInterfaceCounters0(int swIfIndex);
    private static native int[] bridgeDomainDump0(int bdId);
    private static native vppBridgeDomainDetails getBridgeDomainDetails0(int bdId);
    private static native vppL2Fib[] l2FibTableDump0(int bdId);
    private static native int bridgeDomainIdFromInterfaceName0(String interfaceName);
    private static native vppIPv4Address[] ipv4AddressDump0(String interfaceName);
    private static native vppIPv6Address[] ipv6AddressDump0(String interfaceName);
    private static native vppVxlanTunnelDetails[] vxlanTunnelDump0(int swIfIndex);
    private static native int setInterfaceDescription0(String ifName, String ifDesc);
    private static native String getInterfaceDescription0(String ifName);
}
> Exception("First execute the script!") if self._ret_code == 0: raise AssertionError( "VAT Script execution passed, but failure was expected: {cmd}" .format(cmd=self._script_name)) def script_should_have_passed(self): """Read return code from last executed script and raise exception if the script failed.""" if self._ret_code is None: raise Exception("First execute the script!") if self._ret_code != 0: raise AssertionError( "VAT Script execution failed, but success was expected: {cmd}" .format(cmd=self._script_name)) def get_script_stdout(self): """Returns value of stdout from last executed script.""" return self._stdout def get_script_stderr(self): """Returns value of stderr from last executed script.""" return self._stderr @staticmethod def cmd_from_template(node, vat_template_file, json_param=True, **vat_args): """Execute VAT script on specified node. This method supports script templates with parameters. :param node: Node in topology on witch the script is executed. :param vat_template_file: Template file of VAT script. :param vat_args: Arguments to the template file. :returns: List of JSON objects returned by VAT. """ with VatTerminal(node, json_param=json_param) as vat: return vat.vat_terminal_exec_cmd_from_template(vat_template_file, **vat_args) class VatTerminal(object): """VAT interactive terminal. :param node: Node to open VAT terminal on. :param json_param: Defines if outputs from VAT are in JSON format. Default is True. :type node: dict :type json_param: bool """ __VAT_PROMPT = ("vat# ", ) __LINUX_PROMPT = (":~# ", ":~$ ", "~]$ ", "~]# ") def __init__(self, node, json_param=True): json_text = ' json' if json_param else '' self.json = json_param self._node = node self._ssh = SSH() self._ssh.connect(self._node) try: self._tty = self._ssh.interactive_terminal_open() except Exception: raise RuntimeError("Cannot open interactive terminal on node {0}". format(self._node)) for _ in range(3): try: self._ssh.interactive_terminal_exec_command( self._tty, 'sudo -S {0}{1}'.format(Constants.VAT_BIN_NAME, json_text), self.__VAT_PROMPT) except Exception: continue else: break else: vpp_pid = get_vpp_pid(self._node) if vpp_pid: if isinstance(vpp_pid, int): logger.trace("VPP running on node {0}". format(self._node['host'])) else: logger.error("More instances of VPP running on node {0}.". format(self._node['host'])) else: logger.error("VPP not running on node {0}.". format(self._node['host'])) raise RuntimeError("Failed to open VAT console on node {0}". format(self._node['host'])) self._exec_failure = False self.vat_stdout = None def __enter__(self): return self def __exit__(self, exc_type, exc_val, exc_tb): self.vat_terminal_close() def vat_terminal_exec_cmd(self, cmd): """Execute command on the opened VAT terminal. :param cmd: Command to be executed. :returns: Command output in python representation of JSON format or None if not in JSON mode. """ PapiHistory.add_to_papi_history(self._node, cmd, papi=False) logger.debug("Executing command in VAT terminal: {0}".format(cmd)) try: out = self._ssh.interactive_terminal_exec_command(self._tty, cmd, self.__VAT_PROMPT) self.vat_stdout = out except Exception: self._exec_failure = True vpp_pid = get_vpp_pid(self._node) if vpp_pid: if isinstance(vpp_pid, int): raise RuntimeError("VPP running on node {0} but VAT command" " {1} execution failed.". format(self._node['host'], cmd)) else: raise RuntimeError("More instances of VPP running on node " "{0}. VAT command {1} execution failed.". format(self._node['host'], cmd)) raise RuntimeError("VPP not running on node {0}. VAT command " "{1} execution failed.". format(self._node['host'], cmd)) logger.debug("VAT output: {0}".format(out)) if self.json: obj_start = out.find('{') obj_end = out.rfind('}') array_start = out.find('[') array_end = out.rfind(']') if obj_start == -1 and array_start == -1: raise RuntimeError("VAT command {0}: no JSON data.".format(cmd)) if obj_start < array_start or array_start == -1: start = obj_start end = obj_end + 1 else: start = array_start end = array_end + 1 out = out[start:end] json_out = json.loads(out) return json_out else: return None def vat_terminal_close(self): """Close VAT terminal.""" # interactive terminal is dead, we only need to close session if not self._exec_failure: try: self._ssh.interactive_terminal_exec_command(self._tty, 'quit', self.__LINUX_PROMPT) except Exception: vpp_pid = get_vpp_pid(self._node) if vpp_pid: if isinstance(vpp_pid, int): logger.trace("VPP running on node {0}.". format(self._node['host'])) else: logger.error("More instances of VPP running on node " "{0}.".format(self._node['host'])) else: logger.error("VPP not running on node {0}.". format(self._node['host'])) raise RuntimeError("Failed to close VAT console on node {0}". format(self._node['host'])) try: self._ssh.interactive_terminal_close(self._tty) except: raise RuntimeError("Cannot close interactive terminal on node {0}". format(self._node['host'])) def vat_terminal_exec_cmd_from_template(self, vat_template_file, **args): """Execute VAT script from a file. :param vat_template_file: Template file name of a VAT script. :param args: Dictionary of parameters for VAT script. :returns: List of JSON objects returned by VAT. """ file_path = '{}/{}'.format(Constants.RESOURCES_TPL_VAT, vat_template_file) with open(file_path, 'r') as template_file: cmd_template = template_file.readlines() ret = [] for line_tmpl in cmd_template: vat_cmd = line_tmpl.format(**args) ret.append(self.vat_terminal_exec_cmd(vat_cmd.replace('\n', ''))) return ret