psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2018-present Open Networking Foundation |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | * |
| 16 | */ |
| 17 | |
| 18 | package org.onosproject.dhcprelay.rest; |
| 19 | |
| 20 | import org.onlab.packet.IpAddress; |
| 21 | import org.onlab.packet.IpPrefix; |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 22 | import org.onlab.util.Tools; |
| 23 | import com.fasterxml.jackson.databind.ObjectMapper; |
| 24 | import com.fasterxml.jackson.databind.node.ArrayNode; |
| 25 | import com.fasterxml.jackson.databind.node.ObjectNode; |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 26 | import org.onosproject.dhcprelay.api.DhcpRelayService; |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 27 | import org.onosproject.dhcprelay.api.DhcpServerInfo; |
| 28 | import org.onosproject.dhcprelay.cli.DhcpRelayCommand; |
| 29 | import org.onosproject.dhcprelay.store.DhcpRecord; |
| 30 | import org.onosproject.dhcprelay.store.DhcpRelayCounters; |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 31 | import org.onosproject.dhcprelay.store.DhcpRelayCountersStore; |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 32 | import org.onosproject.rest.AbstractWebResource; |
| 33 | import org.onosproject.routeservice.Route; |
| 34 | import org.onosproject.routeservice.RouteStore; |
| 35 | import org.onosproject.routing.fpm.api.FpmRecord; |
| 36 | import org.slf4j.Logger; |
| 37 | |
| 38 | import javax.ws.rs.DELETE; |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 39 | import javax.ws.rs.GET; |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 40 | import javax.ws.rs.Path; |
| 41 | import javax.ws.rs.PathParam; |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 42 | import javax.ws.rs.Produces; |
| 43 | import javax.ws.rs.core.MediaType; |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 44 | import javax.ws.rs.core.Response; |
| 45 | |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 46 | import java.util.Collection; |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 47 | import java.util.List; |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 48 | import java.util.Map; |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 49 | import java.util.Optional; |
| 50 | |
| 51 | import static org.slf4j.LoggerFactory.getLogger; |
| 52 | |
| 53 | /** |
| 54 | * DHCP Relay agent REST API. |
| 55 | */ |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 56 | @Path("dhcp-relay") |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 57 | public class DhcpRelayWebResource extends AbstractWebResource { |
| 58 | private static final Logger LOG = getLogger(DhcpRelayWebResource.class); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 59 | private final ObjectMapper mapper = new ObjectMapper(); |
| 60 | private final DhcpRelayService dhcpDelayService = get(DhcpRelayService.class); |
| 61 | private static final String NA = "N/A"; |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 62 | private static final String HEADER_A_COUNTERS = "DHCP-Relay-Aggregate-Counters"; |
| 63 | private static final String GCOUNT_KEY = "global"; |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 64 | List<DhcpServerInfo> defaultDhcpServerInfoList = dhcpDelayService.getDefaultDhcpServerInfoList(); |
| 65 | List<DhcpServerInfo> indirectDhcpServerInfoList = dhcpDelayService.getIndirectDhcpServerInfoList(); |
| 66 | Collection<DhcpRecord> records = dhcpDelayService.getDhcpRecords(); |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 67 | DhcpRelayCountersStore counterStore = get(DhcpRelayCountersStore.class); |
| 68 | Optional<DhcpRelayCounters> perClassCounters = counterStore.getCounters(GCOUNT_KEY); |
| 69 | private static final String DIRECTLY = "[D]"; |
| 70 | private static final String EMPTY = ""; |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 71 | |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 72 | |
| 73 | /** |
| 74 | * Deletes the fpm route from fpm record. |
| 75 | * Corresponding route from the route store |
| 76 | * |
| 77 | * @param prefix IpPrefix |
pier | c9539e6 | 2018-12-12 19:00:29 +0100 | [diff] [blame] | 78 | * @return 204 NO CONTENT, 404; 401 |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 79 | */ |
| 80 | @DELETE |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 81 | @Path("fpm/{prefix}") |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 82 | public Response dhcpFpmDelete(@PathParam("prefix") String prefix) { |
| 83 | DhcpRelayService dhcpRelayService = get(DhcpRelayService.class); |
| 84 | RouteStore routeStore = get(RouteStore.class); |
| 85 | |
| 86 | try { |
| 87 | // removes fpm route from fpm record |
| 88 | Optional<FpmRecord> fpmRecord = dhcpRelayService.removeFpmRecord(IpPrefix.valueOf(prefix)); |
| 89 | if (fpmRecord.isPresent()) { |
| 90 | IpAddress nextHop = fpmRecord.get().nextHop(); |
| 91 | Route route = new Route(Route.Source.DHCP, IpPrefix.valueOf(prefix), nextHop); |
| 92 | // removes DHCP route from route store |
| 93 | routeStore.removeRoute(route); |
| 94 | } else { |
| 95 | LOG.warn("fpmRecord is not present"); |
| 96 | } |
| 97 | } catch (IllegalArgumentException ex) { |
| 98 | throw new IllegalArgumentException(ex); |
| 99 | } |
| 100 | |
| 101 | return Response.noContent().build(); |
| 102 | } |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 103 | |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 104 | /** |
| 105 | * Returns the response object with list of dhcp servers without counters. |
| 106 | * |
| 107 | * @return 200 OK with component properties of given component and variable |
| 108 | */ |
| 109 | @GET |
| 110 | @Produces(MediaType.APPLICATION_JSON) |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 111 | public Response getDhcpRelayServers() { |
| 112 | ObjectNode node = getDhcpRelayServersJsonOutput(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 113 | return Response.status(200).entity(node).build(); |
| 114 | } |
| 115 | |
| 116 | /** |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 117 | * Returns dhcp counters details. |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 118 | * |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 119 | * @return 200 OK with component properties of given component and variable |
| 120 | */ |
| 121 | @GET |
| 122 | @Produces(MediaType.APPLICATION_JSON) |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 123 | @Path("counters") |
| 124 | public Response getDhcpRelayCounters() { |
| 125 | ObjectNode node = getDhcpRelayCountersJsonOutput(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 126 | return Response.status(200).entity(node).build(); |
| 127 | } |
| 128 | |
| 129 | /** |
| 130 | * To reset the dhcp relay counters. |
| 131 | * |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 132 | * @return 200 OK with component properties of given component and variable |
| 133 | */ |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 134 | @DELETE |
| 135 | @Path("counters") |
| 136 | public Response resetDhcpRelayCounters() { |
| 137 | resetDhcpRelayCountersInternal(); |
| 138 | return Response.status(200).build(); |
| 139 | } |
| 140 | |
| 141 | /** |
| 142 | * Returns results with aggregate of counters. |
| 143 | * |
| 144 | * @return 200 OK with component properties of given component and variable |
| 145 | */ |
| 146 | @GET |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 147 | @Produces(MediaType.APPLICATION_JSON) |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 148 | @Path("aggregate-counters") |
| 149 | public Response getDhcpRelayAggCounters() { |
| 150 | ObjectNode node = getDhcpRelayAggCountersJsonOutput(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 151 | return Response.status(200).entity(node).build(); |
| 152 | } |
| 153 | |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 154 | /** |
| 155 | * Reset dhcp relay aggregate counters. |
| 156 | * |
| 157 | * @return 200 OK with component properties of given component and variable |
| 158 | */ |
| 159 | @DELETE |
| 160 | @Produces(MediaType.APPLICATION_JSON) |
| 161 | @Path("aggregate-counters") |
| 162 | public Response resetDhcpRelayAggCounters() { |
| 163 | resetDhcpRelayAggCountersInternal(); |
| 164 | return Response.status(200).build(); |
| 165 | } |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 166 | |
| 167 | /** |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 168 | * To create json output of Dhcp Relay servers. |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 169 | * |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 170 | * @return node type ObjectNode. |
| 171 | */ |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 172 | private ObjectNode getDhcpRelayServersJsonOutput() { |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 173 | ObjectNode node = mapper.createObjectNode(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 174 | |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 175 | if (!defaultDhcpServerInfoList.isEmpty()) { |
| 176 | ArrayNode defaultDhcpServers = getDefaultDhcpServers(); |
| 177 | node.put("Default-DHCP-Servers", defaultDhcpServers); |
| 178 | } |
| 179 | |
| 180 | if (!indirectDhcpServerInfoList.isEmpty()) { |
| 181 | ArrayNode indirectDhcpServers = getIndirectDhcpServers(); |
| 182 | node.put("Indirect-DHCP-Servers", indirectDhcpServers); |
| 183 | } |
| 184 | if (!records.isEmpty()) { |
| 185 | ArrayNode dhcpRelayRecords = dhcpRelayRecords(); |
| 186 | node.put("DHCP-Relay-Records([D]:Directly-Connected)", dhcpRelayRecords); |
| 187 | } |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 188 | |
| 189 | return node; |
| 190 | |
| 191 | } |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 192 | |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 193 | /** |
| 194 | * To get the liset of dhcp servers. |
| 195 | * |
| 196 | * @param dhcpServerInfoList type List |
| 197 | * @return servers type ArrayNode. |
| 198 | */ |
| 199 | private ArrayNode listServers(List<DhcpServerInfo> dhcpServerInfoList) { |
| 200 | ArrayNode servers = mapper.createArrayNode(); |
| 201 | dhcpServerInfoList.forEach(dhcpServerInfo -> { |
| 202 | ObjectNode serverNode = mapper.createObjectNode(); |
| 203 | String connectPoint = dhcpServerInfo.getDhcpServerConnectPoint() |
| 204 | .map(cp -> cp.toString()).orElse(NA); |
| 205 | String serverMac = dhcpServerInfo.getDhcpConnectMac() |
| 206 | .map(mac -> mac.toString()).orElse(NA); |
| 207 | String gatewayAddress; |
| 208 | String serverIp; |
| 209 | |
| 210 | switch (dhcpServerInfo.getVersion()) { |
| 211 | case DHCP_V4: |
| 212 | gatewayAddress = dhcpServerInfo.getDhcpGatewayIp4() |
| 213 | .map(gw -> gw.toString()).orElse(null); |
| 214 | serverIp = dhcpServerInfo.getDhcpServerIp4() |
| 215 | .map(ip -> ip.toString()).orElse(NA); |
| 216 | break; |
| 217 | case DHCP_V6: |
| 218 | gatewayAddress = dhcpServerInfo.getDhcpGatewayIp6() |
| 219 | .map(gw -> gw.toString()).orElse(null); |
| 220 | serverIp = dhcpServerInfo.getDhcpServerIp6() |
| 221 | .map(ip -> ip.toString()).orElse(NA); |
| 222 | break; |
| 223 | default: |
| 224 | return; |
| 225 | } |
| 226 | |
| 227 | serverNode.put("connectPoint", connectPoint); |
| 228 | if (gatewayAddress != null) { |
| 229 | serverNode.put("server", serverIp.concat(" via ").concat(gatewayAddress)); |
| 230 | } else { |
| 231 | serverNode.put("server", serverIp); |
| 232 | } |
| 233 | serverNode.put("mac", serverMac); |
| 234 | servers.add(serverNode); |
| 235 | }); |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 236 | return servers; |
| 237 | } |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 238 | |
| 239 | /** |
| 240 | * To get the list of dhcp relay records. |
| 241 | * |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 242 | * @return dhcpRelayRecords type ArrayNode. |
| 243 | */ |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 244 | private ArrayNode dhcpRelayRecords() { |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 245 | DhcpRelayCommand dhcpRelayCommand = new DhcpRelayCommand(); |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 246 | ObjectNode node = mapper.createObjectNode(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 247 | ArrayNode dhcpRelayRecords = mapper.createArrayNode(); |
| 248 | records.forEach(record -> { |
| 249 | ObjectNode dhcpRecord = mapper.createObjectNode(); |
| 250 | dhcpRecord.put("id", record.macAddress() + "/" + record.vlanId()); |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 251 | dhcpRecord.put("locations", record.locations().toString() |
| 252 | .concat(record.directlyConnected() ? DIRECTLY : EMPTY)); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 253 | dhcpRecord.put("last-seen", Tools.timeAgo(record.lastSeen())); |
| 254 | dhcpRecord.put("IPv4", dhcpRelayCommand.ip4State(record)); |
| 255 | dhcpRecord.put("IPv6", dhcpRelayCommand.ip6State(record)); |
| 256 | dhcpRelayRecords.add(dhcpRecord); |
| 257 | }); |
| 258 | return dhcpRelayRecords; |
| 259 | } |
| 260 | |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 261 | /** |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 262 | * To get the details of dhcp relay counters. |
| 263 | * |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 264 | * @return counterArray type ArrayNode. |
| 265 | */ |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 266 | private ObjectNode getDhcpRelayCountersJsonOutput() { |
| 267 | ObjectNode node = mapper.createObjectNode(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 268 | ObjectNode counters = mapper.createObjectNode(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 269 | ArrayNode counterArray = mapper.createArrayNode(); |
| 270 | records.forEach(record -> { |
| 271 | DhcpRelayCounters v6Counters = record.getV6Counters(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 272 | Map<String, Integer> countersMap = v6Counters.getCounters(); |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 273 | ObjectNode counterPackets = mapper.createObjectNode(); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 274 | countersMap.forEach((name, value) -> { |
| 275 | counterPackets.put(name, value); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 276 | }); |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 277 | counters.put(record.macAddress() + "/" + record.vlanId().toString(), counterPackets); |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 278 | }); |
Anjali K K | f15ec4a | 2018-11-21 08:39:01 -0500 | [diff] [blame^] | 279 | counterArray.add(counters); |
| 280 | node.put("v6-DHCP-Relay-Counter", counterArray); |
| 281 | return node; |
| 282 | } |
| 283 | |
| 284 | /** |
| 285 | * To get the list of default dhcp servers. |
| 286 | * |
| 287 | * @return node type ObjectNode. |
| 288 | */ |
| 289 | private ArrayNode getDefaultDhcpServers() { |
| 290 | ObjectNode node = mapper.createObjectNode(); |
| 291 | ArrayNode defaultDhcpServers = listServers(defaultDhcpServerInfoList); |
| 292 | return defaultDhcpServers; |
| 293 | } |
| 294 | |
| 295 | /** |
| 296 | * To get the list of indirect dhcp servers. |
| 297 | * |
| 298 | * @return node type ObjectNode. |
| 299 | */ |
| 300 | private ArrayNode getIndirectDhcpServers() { |
| 301 | ArrayNode indirectDhcpServers = listServers(indirectDhcpServerInfoList); |
| 302 | return indirectDhcpServers; |
| 303 | } |
| 304 | |
| 305 | /** |
| 306 | * To reset dhcp relay counters. |
| 307 | * |
| 308 | * @return counterArray type ArrayNode. |
| 309 | */ |
| 310 | private void resetDhcpRelayCountersInternal() { |
| 311 | records.forEach(record -> { |
| 312 | DhcpRelayCounters v6Counters = record.getV6Counters(); |
| 313 | v6Counters.resetCounters(); |
| 314 | }); |
| 315 | } |
| 316 | |
| 317 | /** |
| 318 | * To get dhcp relay aggregate counters. |
| 319 | * |
| 320 | * @return counterPackets type ObjectNode. |
| 321 | */ |
| 322 | private ObjectNode getDhcpRelayAggCountersJsonOutput() { |
| 323 | ObjectNode counterPackets = mapper.createObjectNode(); |
| 324 | ObjectNode dhcpRelayAggCounterNode = mapper.createObjectNode(); |
| 325 | if (perClassCounters.isPresent()) { |
| 326 | Map<String, Integer> counters = perClassCounters.get().getCounters(); |
| 327 | if (counters.size() > 0) { |
| 328 | counters.forEach((name, value) -> { |
| 329 | counterPackets.put(name, value); |
| 330 | }); |
| 331 | } |
| 332 | } |
| 333 | dhcpRelayAggCounterNode.put(HEADER_A_COUNTERS, counterPackets); |
| 334 | return dhcpRelayAggCounterNode; |
| 335 | } |
| 336 | |
| 337 | /** |
| 338 | * To reset aggregate counters. |
| 339 | * |
| 340 | * @return counterPackets type ObjectNode. |
| 341 | */ |
| 342 | private void resetDhcpRelayAggCountersInternal() { |
| 343 | if (perClassCounters.isPresent()) { |
| 344 | counterStore.resetCounters(GCOUNT_KEY); |
| 345 | } |
anjalikk | 072e791 | 2018-11-02 08:25:20 -0400 | [diff] [blame] | 346 | } |
psneha | b710909 | 2018-10-10 08:26:02 -0400 | [diff] [blame] | 347 | |
| 348 | } |