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
|
(* virt-v2v
* Copyright (C) 2009-2020 Red Hat Inc.
*
* 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.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*)
(* Network, bridge and MAC address mapping. *)
open Printf
open Tools_utils
open Common_gettext.Gettext
open Types
type t = {
(* Map specific NIC with MAC address to a network or bridge. *)
mutable macs : (vnet_type * string) StringMap.t;
(* If specific NIC mapping fails, for networks we use this to map
* a named network, or use the default network if no named
* network exists.
*)
mutable network_map : string StringMap.t;
mutable default_network : string option;
(* If that fails, same as above but for bridges. *)
mutable bridge_map : string StringMap.t;
mutable default_bridge : string option;
}
let map t nic =
try
let mac = match nic.s_mac with None -> raise Not_found | Some mac -> mac in
let mac = String.lowercase_ascii mac in
let vnet_type, vnet = StringMap.find mac t.macs in
{ nic with s_vnet_type = vnet_type; s_vnet = vnet }
with Not_found ->
match nic.s_vnet_type with
| Network ->
(try
let vnet = StringMap.find nic.s_vnet t.network_map in
{ nic with s_vnet = vnet }
with Not_found ->
match t.default_network with
| None -> nic (* no mapping done *)
| Some default_network ->
{ nic with s_vnet = default_network }
)
| Bridge ->
(try
let vnet = StringMap.find nic.s_vnet t.bridge_map in
{ nic with s_vnet = vnet }
with Not_found ->
match t.default_bridge with
| None -> nic (* no mapping done *)
| Some default_bridge ->
{ nic with s_vnet = default_bridge }
)
let create () = {
macs = StringMap.empty;
network_map = StringMap.empty;
default_network = None;
bridge_map = StringMap.empty;
default_bridge = None
}
let add_mac t mac vnet_type vnet =
let mac = String.lowercase_ascii mac in
if StringMap.mem mac t.macs then
error (f_"duplicate --mac parameter. Duplicate mappings specified \
for MAC address %s.") mac;
t.macs <- StringMap.add mac (vnet_type, vnet) t.macs
let add_network t i o =
if StringMap.mem i t.network_map then
error (f_"duplicate -n/--network parameter. Duplicate mappings \
specified for network %s.") i;
t.network_map <- StringMap.add i o t.network_map
let add_default_network t o =
if t.default_network <> None then
error (f_"duplicate -n/--network parameter. Only one \
default mapping is allowed.");
t.default_network <- Some o
let add_bridge t i o =
if StringMap.mem i t.bridge_map then
error (f_"duplicate -b/--bridge parameter. Duplicate mappings \
specified for bridge %s.") i;
t.bridge_map <- StringMap.add i o t.bridge_map
let add_default_bridge t o =
if t.default_bridge <> None then
error (f_"duplicate -b/--bridge parameter. Only one default mapping \
is allowed.");
t.default_bridge <- Some o
|
