4141import com .vmware .vim25 .HostListSummaryQuickStats ;
4242import com .vmware .vim25 .HostNetworkInfo ;
4343import com .vmware .vim25 .HostNetworkPolicy ;
44+ import com .vmware .vim25 .HostNetworkSecurityPolicy ;
4445import com .vmware .vim25 .HostNetworkTrafficShapingPolicy ;
4546import com .vmware .vim25 .HostPortGroup ;
4647import com .vmware .vim25 .HostPortGroupSpec ;
@@ -365,7 +366,7 @@ public boolean hasPortGroup(HostVirtualSwitch vSwitch, String portGroupName) thr
365366 return false ;
366367 }
367368
368- public void createPortGroup (HostVirtualSwitch vSwitch , String portGroupName , Integer vlanId , HostNetworkTrafficShapingPolicy shapingPolicy ) throws Exception {
369+ public void createPortGroup (HostVirtualSwitch vSwitch , String portGroupName , Integer vlanId , HostNetworkSecurityPolicy secPolicy , HostNetworkTrafficShapingPolicy shapingPolicy ) throws Exception {
369370 assert (portGroupName != null );
370371 HostNetworkSystemMO hostNetMo = getHostNetworkSystemMO ();
371372 assert (hostNetMo != null );
@@ -376,13 +377,15 @@ public void createPortGroup(HostVirtualSwitch vSwitch, String portGroupName, Int
376377 if (vlanId != null )
377378 spec .setVlanId (vlanId .intValue ());
378379 HostNetworkPolicy policy = new HostNetworkPolicy ();
380+ if (secPolicy != null )
381+ policy .setSecurity (secPolicy );
379382 policy .setShapingPolicy (shapingPolicy );
380383 spec .setPolicy (policy );
381384 spec .setVswitchName (vSwitch .getName ());
382385 hostNetMo .addPortGroup (spec );
383386 }
384387
385- public void updatePortGroup (HostVirtualSwitch vSwitch , String portGroupName , Integer vlanId , HostNetworkTrafficShapingPolicy shapingPolicy ) throws Exception {
388+ public void updatePortGroup (HostVirtualSwitch vSwitch , String portGroupName , Integer vlanId , HostNetworkSecurityPolicy secPolicy , HostNetworkTrafficShapingPolicy shapingPolicy ) throws Exception {
386389 assert (portGroupName != null );
387390 HostNetworkSystemMO hostNetMo = getHostNetworkSystemMO ();
388391 assert (hostNetMo != null );
@@ -393,6 +396,8 @@ public void updatePortGroup(HostVirtualSwitch vSwitch, String portGroupName, Int
393396 if (vlanId != null )
394397 spec .setVlanId (vlanId .intValue ());
395398 HostNetworkPolicy policy = new HostNetworkPolicy ();
399+ if (secPolicy != null )
400+ policy .setSecurity (secPolicy );
396401 policy .setShapingPolicy (shapingPolicy );
397402 spec .setPolicy (policy );
398403 spec .setVswitchName (vSwitch .getName ());
0 commit comments