Synnefo » snf-ganeti » ganeti-local

import Text.Printf (printf)

      "OP_NETWORK_ADD" ->

        OpCodes.OpNetworkAdd <$> genNameNE <*> arbitrary <*> genIp4Net <*>

          getMaybe genIp4Addr <*> pure Nothing <*> pure Nothing <*>

          getMaybe genMacPrefix <*> getMaybe (listOf genIp4Addr) <*>

          (genTags >>= mapM mkNonEmpty)

      "OP_NETWORK_REMOVE" ->

        OpCodes.OpNetworkRemove <$> genNameNE <*> arbitrary

      "OP_NETWORK_SET_PARAMS" ->

        OpCodes.OpNetworkSetParams <$> genNameNE <*> arbitrary <*>

          getMaybe genIp4Addr <*> pure Nothing <*> pure Nothing <*>

          getMaybe genMacPrefix <*> getMaybe (listOf genIp4Addr) <*>

          getMaybe (listOf genIp4Addr)

      "OP_NETWORK_CONNECT" ->

        OpCodes.OpNetworkConnect <$> genNameNE <*> genNameNE <*>

          arbitrary <*> genNameNE <*> arbitrary

      "OP_NETWORK_DISCONNECT" ->

        OpCodes.OpNetworkDisconnect <$> genNameNE <*> genNameNE <*> arbitrary

      "OP_NETWORK_QUERY" ->

        OpCodes.OpNetworkQuery <$> genFieldsNE <*> genNamesNE

-- | Generate an arbitrary IPv4 address in textual form.

genIp4Addr :: Gen NonEmptyString

genIp4Addr = do

  a <- choose (1::Int, 255)

  b <- choose (0::Int, 255)

  c <- choose (0::Int, 255)

  d <- choose (0::Int, 255)

  mkNonEmpty $ intercalate "." (map show [a, b, c, d])

-- | Generate an arbitrary IPv4 network address in textual form.

genIp4Net :: Gen NonEmptyString

genIp4Net = do

  netmask <- choose (8::Int, 30)

  ip <- genIp4Addr

  mkNonEmpty $ fromNonEmpty ip ++ "/" ++ show netmask

-- | Generate a 3-byte MAC prefix.

genMacPrefix :: Gen NonEmptyString

genMacPrefix = do

  octets <- vectorOf 3 $ choose (0::Int, 255)

  mkNonEmpty . intercalate ":" $ map (printf "%02x") octets