홍길동 / onos

Go to a project
Toggle navigation Toggle navigation pinning
Jonathan Hart
Committed by Gerrit Code Review
81b09142 1 parent 4752cfa9

Support source-specific IGMP joins. 

Change-Id: I422f54f908998460ceff994f9b3bfbf3d2f81a56
Inline Side-by-side
Showing 1 changed file with 38 additions and 13 deletions
......@@ -320,30 +320,55 @@ public class IgmpSnoop {
IGMPMembership membership = (IGMPMembership) group;
// TODO allow pulling source from IGMP packet
IpAddress source = ssmTranslateTable.get(group.getGaddr());
if (source == null) {
log.warn("No source found in SSM translate table for {}", group.getGaddr());
return;
}
McastRoute route = new McastRoute(source,
group.getGaddr(),
McastRoute.Type.IGMP);
IpAddress groupAddress = membership.getGaddr();
if (membership.getRecordType() == IGMPMembership.MODE_IS_INCLUDE ||
membership.getRecordType() == IGMPMembership.CHANGE_TO_INCLUDE_MODE) {
multicastService.removeSink(route, location);
// TODO remove route if all sinks are gone
if (membership.getSources().isEmpty()) {
McastRoute route = ssmTranslateRoute(groupAddress);
if (route != null) {
removeRoute(route, location);
}
} else {
membership.getSources().stream()
.map(source -> new McastRoute(source, groupAddress, McastRoute.Type.IGMP))
.forEach(route -> addRoute(route, location));
}
} else if (membership.getRecordType() == IGMPMembership.MODE_IS_EXCLUDE ||
membership.getRecordType() == IGMPMembership.CHANGE_TO_EXCLUDE_MODE) {
if (membership.getSources().isEmpty()) {
McastRoute route = ssmTranslateRoute(groupAddress);
if (route != null) {
addRoute(route, location);
}
} else {
membership.getSources().stream()
.map(source -> new McastRoute(source, groupAddress, McastRoute.Type.IGMP))
.forEach(route -> removeRoute(route, location));
}
}
});
}
private McastRoute ssmTranslateRoute(IpAddress group) {
IpAddress source = ssmTranslateTable.get(group);
if (source == null) {
log.warn("No SSM translate source found for group {}", group);
return null;
}
return new McastRoute(source, group, McastRoute.Type.IGMP);
}
private void addRoute(McastRoute route, ConnectPoint location) {
multicastService.add(route);
multicastService.addSink(route, location);
}
});
private void removeRoute(McastRoute route, ConnectPoint location) {
multicastService.removeSink(route, location);
// TODO remove route if all sinks are gone
}
private ByteBuffer buildQueryPacket() {
......