Committed by
Thomas Vachuska
[ONOS-3833]Implementation for FiveTuple class to store 5 tuple packet info
Change-Id: I15f2b41187c1577677428a332b44fa1c5351cb05
Showing
2 changed files
with
285 additions
and
0 deletions
| 1 | +/* | ||
| 2 | + * Copyright 2016 Open Networking Laboratory | ||
| 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 | +package org.onosproject.vtnrsc; | ||
| 17 | + | ||
| 18 | +import static com.google.common.base.MoreObjects.toStringHelper; | ||
| 19 | +import static com.google.common.base.Preconditions.checkArgument; | ||
| 20 | + | ||
| 21 | +import java.util.Objects; | ||
| 22 | + | ||
| 23 | +import org.onlab.packet.IPv4; | ||
| 24 | +import org.onlab.packet.IpAddress; | ||
| 25 | +import org.onosproject.net.PortNumber; | ||
| 26 | + | ||
| 27 | +/** | ||
| 28 | + * Representation for five tuple information from packet. | ||
| 29 | + */ | ||
| 30 | +public final class DefaultFiveTuple implements FiveTuple { | ||
| 31 | + | ||
| 32 | + private final IpAddress ipSrc; | ||
| 33 | + private final IpAddress ipDst; | ||
| 34 | + private final PortNumber portSrc; | ||
| 35 | + private final PortNumber portDst; | ||
| 36 | + private final byte protocol; | ||
| 37 | + private final TenantId tenantId; | ||
| 38 | + | ||
| 39 | + /** | ||
| 40 | + * Constructor for packet five tuple information. | ||
| 41 | + * | ||
| 42 | + * @param protocol protocol of the packet | ||
| 43 | + * @param ipSrc source ip address of the packet | ||
| 44 | + * @param ipDst destination ip address of the packet | ||
| 45 | + * @param portSrc source port of the packet | ||
| 46 | + * @param portDst destination port of the packet | ||
| 47 | + */ | ||
| 48 | + private DefaultFiveTuple(byte protocol, IpAddress ipSrc, IpAddress ipDst, PortNumber portSrc, PortNumber portDst, | ||
| 49 | + TenantId tenantId) { | ||
| 50 | + | ||
| 51 | + this.protocol = protocol; | ||
| 52 | + this.ipSrc = ipSrc; | ||
| 53 | + this.ipDst = ipDst; | ||
| 54 | + this.portSrc = portSrc; | ||
| 55 | + this.portDst = portDst; | ||
| 56 | + this.tenantId = tenantId; | ||
| 57 | + } | ||
| 58 | + | ||
| 59 | + @Override | ||
| 60 | + public byte protocol() { | ||
| 61 | + return protocol; | ||
| 62 | + } | ||
| 63 | + | ||
| 64 | + @Override | ||
| 65 | + public IpAddress ipSrc() { | ||
| 66 | + return ipSrc; | ||
| 67 | + } | ||
| 68 | + | ||
| 69 | + @Override | ||
| 70 | + public IpAddress ipDst() { | ||
| 71 | + return ipDst; | ||
| 72 | + } | ||
| 73 | + | ||
| 74 | + @Override | ||
| 75 | + public PortNumber portSrc() { | ||
| 76 | + return portSrc; | ||
| 77 | + } | ||
| 78 | + | ||
| 79 | + @Override | ||
| 80 | + public PortNumber portDst() { | ||
| 81 | + return portDst; | ||
| 82 | + } | ||
| 83 | + | ||
| 84 | + @Override | ||
| 85 | + public TenantId tenantId() { | ||
| 86 | + return tenantId; | ||
| 87 | + } | ||
| 88 | + | ||
| 89 | + @Override | ||
| 90 | + public boolean equals(Object obj) { | ||
| 91 | + if (this == obj) { | ||
| 92 | + return true; | ||
| 93 | + } | ||
| 94 | + if (obj instanceof DefaultFiveTuple) { | ||
| 95 | + final DefaultFiveTuple other = (DefaultFiveTuple) obj; | ||
| 96 | + return Objects.equals(this.protocol, other.protocol) && | ||
| 97 | + Objects.equals(this.ipSrc, other.ipSrc) && | ||
| 98 | + Objects.equals(this.ipDst, other.ipDst) && | ||
| 99 | + Objects.equals(this.portSrc, other.portSrc) && | ||
| 100 | + Objects.equals(this.portDst, other.portDst); | ||
| 101 | + } | ||
| 102 | + return false; | ||
| 103 | + } | ||
| 104 | + | ||
| 105 | + @Override | ||
| 106 | + public int hashCode() { | ||
| 107 | + return Objects.hash(this.protocol, this.ipSrc, this.ipDst, this.portSrc, this.portDst); | ||
| 108 | + } | ||
| 109 | + | ||
| 110 | + @Override | ||
| 111 | + public String toString() { | ||
| 112 | + return toStringHelper(this) | ||
| 113 | + .omitNullValues() | ||
| 114 | + .add("protocol", protocol) | ||
| 115 | + .add("ipSrc", ipSrc) | ||
| 116 | + .add("ipDst", ipDst) | ||
| 117 | + .add("portSrc", portSrc) | ||
| 118 | + .add("portDst", portDst) | ||
| 119 | + .toString(); | ||
| 120 | + } | ||
| 121 | + | ||
| 122 | + /** | ||
| 123 | + * To create an instance of the builder. | ||
| 124 | + * | ||
| 125 | + * @return instance of builder | ||
| 126 | + */ | ||
| 127 | + public static Builder builder() { | ||
| 128 | + return new Builder(); | ||
| 129 | + } | ||
| 130 | + | ||
| 131 | + /** | ||
| 132 | + * Builder class for Five tuple info. | ||
| 133 | + */ | ||
| 134 | + public static final class Builder implements FiveTuple.Builder { | ||
| 135 | + | ||
| 136 | + private IpAddress ipSrc; | ||
| 137 | + private IpAddress ipDst; | ||
| 138 | + private PortNumber portSrc; | ||
| 139 | + private PortNumber portDst; | ||
| 140 | + private byte protocol; | ||
| 141 | + private TenantId tenantId; | ||
| 142 | + | ||
| 143 | + @Override | ||
| 144 | + public Builder setIpSrc(IpAddress ipSrc) { | ||
| 145 | + this.ipSrc = ipSrc; | ||
| 146 | + return this; | ||
| 147 | + } | ||
| 148 | + | ||
| 149 | + @Override | ||
| 150 | + public Builder setIpDst(IpAddress ipDst) { | ||
| 151 | + this.ipDst = ipDst; | ||
| 152 | + return this; | ||
| 153 | + } | ||
| 154 | + | ||
| 155 | + @Override | ||
| 156 | + public Builder setPortSrc(PortNumber portSrc) { | ||
| 157 | + this.portSrc = portSrc; | ||
| 158 | + return this; | ||
| 159 | + } | ||
| 160 | + | ||
| 161 | + @Override | ||
| 162 | + public Builder setPortDst(PortNumber portDst) { | ||
| 163 | + this.portDst = portDst; | ||
| 164 | + return this; | ||
| 165 | + } | ||
| 166 | + | ||
| 167 | + @Override | ||
| 168 | + public Builder setProtocol(byte protocol) { | ||
| 169 | + this.protocol = protocol; | ||
| 170 | + return this; | ||
| 171 | + } | ||
| 172 | + | ||
| 173 | + @Override | ||
| 174 | + public org.onosproject.vtnrsc.FiveTuple.Builder setTenantId(TenantId tenantId) { | ||
| 175 | + this.tenantId = tenantId; | ||
| 176 | + return this; | ||
| 177 | + } | ||
| 178 | + | ||
| 179 | + @Override | ||
| 180 | + public FiveTuple build() { | ||
| 181 | + checkArgument(protocol == IPv4.PROTOCOL_TCP || protocol == IPv4.PROTOCOL_UDP || | ||
| 182 | + protocol == IPv4.PROTOCOL_ICMP, "Unsupported value for protocol while creating five tuple"); | ||
| 183 | + | ||
| 184 | + return new DefaultFiveTuple(protocol, ipSrc, ipDst, portSrc, portDst, tenantId); | ||
| 185 | + } | ||
| 186 | + } | ||
| 187 | +} |
| 1 | +/* | ||
| 2 | + * Copyright 2016 Open Networking Laboratory | ||
| 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 | +package org.onosproject.vtnrsc; | ||
| 17 | + | ||
| 18 | +import static org.hamcrest.MatcherAssert.assertThat; | ||
| 19 | +import static org.hamcrest.Matchers.is; | ||
| 20 | +import static org.hamcrest.Matchers.notNullValue; | ||
| 21 | +import static org.onlab.junit.ImmutableClassChecker.assertThatClassIsImmutable; | ||
| 22 | + | ||
| 23 | +import org.junit.Test; | ||
| 24 | +import org.onlab.packet.IPv4; | ||
| 25 | +import org.onlab.packet.IpAddress; | ||
| 26 | +import org.onosproject.net.PortNumber; | ||
| 27 | + | ||
| 28 | +import com.google.common.testing.EqualsTester; | ||
| 29 | + | ||
| 30 | +/** | ||
| 31 | + * Unit tests for FiveTuple class. | ||
| 32 | + */ | ||
| 33 | +public class DefaultFiveTupleTest { | ||
| 34 | + | ||
| 35 | + | ||
| 36 | + final FiveTuple fiveTuple1 = DefaultFiveTuple.builder().setIpSrc(IpAddress.valueOf("1.1.1.1")) | ||
| 37 | + .setIpDst(IpAddress.valueOf("2.2.2.2")) | ||
| 38 | + .setPortSrc(PortNumber.portNumber(500)) | ||
| 39 | + .setPortDst(PortNumber.portNumber(1000)) | ||
| 40 | + .setProtocol(IPv4.PROTOCOL_TCP) | ||
| 41 | + .setTenantId(TenantId.tenantId("aaa")) | ||
| 42 | + .build(); | ||
| 43 | + | ||
| 44 | + final FiveTuple sameAsFiveTuple1 = DefaultFiveTuple.builder().setIpSrc(IpAddress.valueOf("1.1.1.1")) | ||
| 45 | + .setIpDst(IpAddress.valueOf("2.2.2.2")) | ||
| 46 | + .setPortSrc(PortNumber.portNumber(500)) | ||
| 47 | + .setPortDst(PortNumber.portNumber(1000)) | ||
| 48 | + .setProtocol(IPv4.PROTOCOL_TCP) | ||
| 49 | + .setTenantId(TenantId.tenantId("aaa")) | ||
| 50 | + .build(); | ||
| 51 | + | ||
| 52 | + final FiveTuple fiveTuple2 = DefaultFiveTuple.builder().setIpSrc(IpAddress.valueOf("3.3.3.3")) | ||
| 53 | + .setIpDst(IpAddress.valueOf("4.4.4.4")) | ||
| 54 | + .setPortSrc(PortNumber.portNumber(1500)) | ||
| 55 | + .setPortDst(PortNumber.portNumber(2000)) | ||
| 56 | + .setProtocol(IPv4.PROTOCOL_UDP) | ||
| 57 | + .setTenantId(TenantId.tenantId("bbb")) | ||
| 58 | + .build(); | ||
| 59 | + | ||
| 60 | + /** | ||
| 61 | + * Checks that the FiveTuple class is immutable. | ||
| 62 | + */ | ||
| 63 | + @Test | ||
| 64 | + public void testImmutability() { | ||
| 65 | + assertThatClassIsImmutable(DefaultFiveTuple.class); | ||
| 66 | + } | ||
| 67 | + | ||
| 68 | + /** | ||
| 69 | + * Checks the operation of equals() methods. | ||
| 70 | + */ | ||
| 71 | + @Test | ||
| 72 | + public void testEquals() { | ||
| 73 | + new EqualsTester().addEqualityGroup(fiveTuple1, sameAsFiveTuple1).addEqualityGroup(fiveTuple2) | ||
| 74 | + .testEquals(); | ||
| 75 | + } | ||
| 76 | + | ||
| 77 | + /** | ||
| 78 | + * Checks the construction of a FiveTuple object. | ||
| 79 | + */ | ||
| 80 | + @Test | ||
| 81 | + public void testConstruction() { | ||
| 82 | + final FiveTuple fiveTuple1 = DefaultFiveTuple.builder().setIpSrc(IpAddress.valueOf("1.1.1.1")) | ||
| 83 | + .setIpDst(IpAddress.valueOf("2.2.2.2")) | ||
| 84 | + .setPortSrc(PortNumber.portNumber(500)) | ||
| 85 | + .setPortDst(PortNumber.portNumber(1000)) | ||
| 86 | + .setProtocol(IPv4.PROTOCOL_TCP) | ||
| 87 | + .setTenantId(TenantId.tenantId("aaa")) | ||
| 88 | + .build(); | ||
| 89 | + | ||
| 90 | + assertThat(fiveTuple1, is(notNullValue())); | ||
| 91 | + assertThat(fiveTuple1.protocol(), is(IPv4.PROTOCOL_TCP)); | ||
| 92 | + assertThat(fiveTuple1.ipSrc(), is(IpAddress.valueOf("1.1.1.1"))); | ||
| 93 | + assertThat(fiveTuple1.ipDst(), is(IpAddress.valueOf("2.2.2.2"))); | ||
| 94 | + assertThat(fiveTuple1.portSrc(), is(PortNumber.portNumber(500))); | ||
| 95 | + assertThat(fiveTuple1.portDst(), is(PortNumber.portNumber(1000))); | ||
| 96 | + assertThat(fiveTuple1.tenantId(), is(TenantId.tenantId("aaa"))); | ||
| 97 | + } | ||
| 98 | +} |
-
Please register or login to post a comment