Merge "Contrail: Fix controlplane/dataplane network asignments & enable optional dpdk"

1 files changed, 14 insertions, 0 deletions

diff --git a/lib/puppet/parser/functions/netmask_to_cidr.rb b/lib/puppet/parser/functions/netmask_to_cidr.rb
new file mode 100644
index 0000000..68ee4cf
--- /dev/null
+++ b/lib/puppet/parser/functions/netmask_to_cidr.rb
@@ -0,0 +1,14 @@
+# Custom function to transform netmask from IP notation to
+# CIDR format. Input is an IP address, output a CIDR:
+# 255.255.255.0 = 24
+# The CIDR formated netmask is needed for some
+# Contrail configuration files
+require 'ipaddr'
+module Puppet::Parser::Functions
+  newfunction(:netmask_to_cidr, :type => :rvalue) do |args|
+    if args[0].class != String
+      raise Puppet::ParseError, "Syntax error: #{args[0]} must be a String"
+    end
+    IPAddr.new(args[0]).to_i.to_s(2).count("1")
+  end
+end