Supports UPD and TCP ports for Segment Routing policy add CLI (srpolicy-add).
Change-Id: Ib55701e429bc1080211bcc09adb0dc2b21ee4461
diff --git a/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/cli/PolicyAddCommand.java b/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/cli/PolicyAddCommand.java
index adcb936..2931307 100644
--- a/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/cli/PolicyAddCommand.java
+++ b/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/cli/PolicyAddCommand.java
@@ -46,17 +46,32 @@
required = false, multiValued = false)
String srcIp;
- @Argument(index = 3, name = "dst IP",
+ @Argument(index = 3, name = "src port",
+ description = "src port",
+ required = false, multiValued = false)
+ short srcPort;
+
+ @Argument(index = 4, name = "dst IP",
description = "dst IP",
required = false, multiValued = false)
String dstIp;
- @Argument(index = 4, name = "policy type",
+ @Argument(index = 5, name = "dst port",
+ description = "dst port",
+ required = false, multiValued = false)
+ short dstPort;
+
+ @Argument(index = 6, name = "proto",
+ description = "proto",
+ required = false, multiValued = false)
+ String proto;
+
+ @Argument(index = 7, name = "policy type",
description = "policy type",
required = true, multiValued = false)
String policyType;
- @Argument(index = 5, name = "tunnel ID",
+ @Argument(index = 8, name = "tunnel ID",
description = "tunnel ID",
required = false, multiValued = false)
String tunnelId;
@@ -77,6 +92,15 @@
if (dstIp != null) {
tpb.setDstIp(dstIp);
}
+ if (srcPort != 0) {
+ tpb.setSrcPort(srcPort);
+ }
+ if (dstPort != 0) {
+ tpb.setDstPort(dstPort);
+ }
+ if (!proto.equals("ip")) {
+ tpb.setIpProto(proto);
+ }
if (Policy.Type.valueOf(policyType) == Policy.Type.TUNNEL_FLOW) {
if (tunnelId == null) {
// TODO: handle errors