ceph/src/test/test_addrs.cc

   1 // -*- mode:C; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3 /*
   4  * Ceph - scalable distributed file system
   5  *
   6  * Copyright (C) 2011 New Dream Network
   7  *
   8  * This is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU General Public
  10  * License version 2, as published by the Free Software
  11  * Foundation.  See file COPYING.
  12  *
  13  */
  14
  15 #include "include/types.h"
  16 #include "msg/msg_types.h"
  17 #include "gtest/gtest.h"
  18
  19 #include <sstream>
  20
  21 // input, parsed+printed addr output, leftover
  22 // if the parse fails, output + leftover should both be blank.
  23 const char *addr_checks[][3] = {
  24   { "127.0.0.1", "127.0.0.1:0/0", "" },
  25   { "127.0.0.1 foo", "127.0.0.1:0/0", " foo" },
  26   { "127.0.0.1:1234 foo", "127.0.0.1:1234/0", " foo" },
  27   { "127.0.0.1:1234/5678 foo", "127.0.0.1:1234/5678", " foo" },
  28   { "1.2.3:4 a", "", "" },
  29   { "2607:f298:4:2243::5522", "[2607:f298:4:2243::5522]:0/0", "" },
  30   { "[2607:f298:4:2243::5522]", "[2607:f298:4:2243::5522]:0/0", "" },
  31   { "2607:f298:4:2243::5522a", "", "" },
  32   { "[2607:f298:4:2243::5522]a", "[2607:f298:4:2243::5522]:0/0", "a" },
  33   { "[2607:f298:4:2243::5522]:1234a", "[2607:f298:4:2243::5522]:1234/0", "a" },
  34   { "2001:0db8:85a3:0000:0000:8a2e:0370:7334", "[2001:db8:85a3::8a2e:370:7334]:0/0", "" },
  35   { "2001:2db8:85a3:4334:4324:8a2e:1370:7334", "[2001:2db8:85a3:4334:4324:8a2e:1370:7334]:0/0", "" },
  36   { "::", "[::]:0/0", "" },
  37   { "::zz", "[::]:0/0", "zz" },
  38   { ":: 12:34", "[::]:0/0", " 12:34" },
  39   { "-", "-", "" },
  40   { "-asdf", "-", "asdf" },
  41   { "legacy:1.2.3.4", "1.2.3.4:0/0", "" },
  42   { "legacy:1.2.3.4:12", "1.2.3.4:12/0", "" },
  43   { "legacy:1.2.3.4:12/34", "1.2.3.4:12/34", "" },
  44   { "msgr2:1.2.3.4", "msgr2:1.2.3.4:0/0", "" },
  45   { "msgr2:1.2.3.4:12", "msgr2:1.2.3.4:12/0", "" },
  46   { "msgr2:1.2.3.4:12/34", "msgr2:1.2.3.4:12/34", "" },
  47   { NULL, NULL, NULL },
  48 };
  49
  50 TEST(Msgr, TestAddrParsing)
  51 {
  52   for (unsigned i = 0; addr_checks[i][0]; ++i) {
  53     entity_addr_t a;
  54     const char *end = "";
  55     bool ok = a.parse(addr_checks[i][0], &end);
  56     string out;
  57     if (ok) {
  58       stringstream ss;
  59       ss << a;
  60       getline(ss, out);
  61     }
  62     string left = end;
  63
  64     cout << "'" << addr_checks[i][0] << "' -> '" << out << "' + '" << left << "'" << std::endl;
  65
  66     ASSERT_EQ(out, addr_checks[i][1]);
  67     ASSERT_EQ(left, addr_checks[i][2]);
  68   }
  69 }
  70
  71 const char *addr_checks2[][3] = {
  72   { "127.0.0.1", "127.0.0.1:0/0", "" },
  73   { "127.0.0.1 foo", "127.0.0.1:0/0", " foo" },
  74   { "127.0.0.1:1234 foo", "127.0.0.1:1234/0", " foo" },
  75   { "127.0.0.1:1234/5678 foo", "127.0.0.1:1234/5678", " foo" },
  76   { "2607:f298:4:2243::5522", "[2607:f298:4:2243::5522]:0/0", "" },
  77   { "[2607:f298:4:2243::5522]", "[2607:f298:4:2243::5522]:0/0", "" },
  78   { "[2607:f298:4:2243::5522]a", "[2607:f298:4:2243::5522]:0/0", "a" },
  79   { "[2607:f298:4:2243::5522]:1234a", "[2607:f298:4:2243::5522]:1234/0", "a" },
  80   { "2001:0db8:85a3:0000:0000:8a2e:0370:7334", "[2001:db8:85a3::8a2e:370:7334]:0/0", "" },
  81   { "2001:2db8:85a3:4334:4324:8a2e:1370:7334", "[2001:2db8:85a3:4334:4324:8a2e:1370:7334]:0/0", "" },
  82   { "legacy:1.2.3.4", "1.2.3.4:0/0", "" },
  83   { "legacy:1.2.3.4:12", "1.2.3.4:12/0", "" },
  84   { "legacy:1.2.3.4:12/34", "1.2.3.4:12/34", "" },
  85   { NULL, NULL, NULL },
  86 };
  87
  88 TEST(Msgr, TestAddrEncodeAddrvecDecode)
  89 {
  90   for (unsigned i = 0; addr_checks2[i][0]; ++i) {
  91     entity_addr_t addr;
  92     entity_addrvec_t addrvec;
  93     bufferlist::iterator bli;
  94     const char *end = "";
  95     bool ok = addr.parse(addr_checks2[i][0], &end);
  96     ASSERT_TRUE(ok);
  97     bufferlist bl;
  98     addr.encode(bl, 0);
  99     bli = bl.begin();
 100     addrvec.decode(bli);
 101     ASSERT_EQ(addr, addrvec.v[0]);
 102   }
 103 }
 104
 105 TEST(Msgr, TestAddrvec0EncodeAddrDecode)
 106 {
 107   for (unsigned i = 0; addr_checks2[i][0]; ++i) {
 108     entity_addr_t addr;
 109     entity_addrvec_t addrvec;
 110     bufferlist bl;
 111     bufferlist::iterator bli;
 112     const char *end = "";
 113     bool ok = addr.parse(addr_checks2[i][0], &end);
 114     ASSERT_TRUE(ok);
 115     addrvec.v.push_back(addr);
 116     addrvec.encode(bl, 0);
 117     bli = bl.begin();
 118     entity_addr_t a;
 119     a.decode(bli);
 120     ASSERT_EQ(addr, a);
 121   }
 122 }
 123
 124 TEST(Msgr, TestEmptyAddrvecEncodeAddrDecode)
 125 {
 126   entity_addrvec_t addrvec;
 127   entity_addr_t addr;
 128   bufferlist bl;
 129   bufferlist::iterator bli;
 130   addrvec.encode(bl, 0);
 131   bli = bl.begin();
 132   addr.decode(bli);
 133   ASSERT_EQ(addr, entity_addr_t(1, 0));
 134 }
 135
 136 const char *addrvec_checks[][4] = {
 137   { "legacy:1.2.3.4", "msgr2:1.2.3.4", "legacy:1.2.3.4", "msgr2:1.2.3.4" },
 138   { "msgr2:1.2.3.5", "legacy:1.2.3.5", "legacy:1.2.3.5", "msgr2:1.2.3.5" },
 139   { "msgr2:1.2.3.6", "msgr2:1.2.3.6", "legacy:1.2.3.6", "msgr2:1.2.3.6" },
 140   { "msgr2:1.2.3.7", "legacy:1.2.3.7", "legacy:1.2.3.7", "msgr2:1.2.3.7" },
 141   { NULL, NULL, NULL, NULL },
 142 };
 143
 144 /*
 145  * multiple addrs where one is legacy and others are not
 146  * legacy addr is in position 0
 147  */
 148 TEST(Msgr, TestAddrvecEncodeAddrDecode0)
 149 {
 150   entity_addr_t addr;
 151   entity_addrvec_t addrvec;
 152   bufferlist bl;
 153   bufferlist::iterator bli;
 154
 155   for (unsigned i = 0; addrvec_checks[i][0]; ++i) {
 156     const char *end = "";
 157     bool ok = addr.parse(addrvec_checks[i][0], &end);
 158     ASSERT_TRUE(ok);
 159     addrvec.v.push_back(addr);
 160   }
 161
 162   addrvec.encode(bl, 0);
 163   bli = bl.begin();
 164
 165   addr.decode(bli);
 166
 167   ASSERT_EQ(addr, addrvec.v[0]);
 168 }
 169
 170 /*
 171  * multiple addrs where one is legacy and others are not
 172  * legacy addr is not in position 0
 173  */
 174 TEST(Msgr, TestAddrvecEncodeAddrDecode1)
 175 {
 176   entity_addr_t addr, a;
 177   entity_addrvec_t addrvec;
 178   bufferlist bl;
 179   bufferlist::iterator bli;
 180   bool flag = true;
 181
 182   for (unsigned i = 0; addrvec_checks[i][1]; ++i) {
 183     const char *end = "";
 184     bool ok = addr.parse(addrvec_checks[i][1], &end);
 185     ASSERT_TRUE(ok);
 186     if (addr.type == entity_addr_t::TYPE_LEGACY && flag) {
 187       a = addr;
 188       flag = !flag;
 189     }
 190     addrvec.v.push_back(addr);
 191   }
 192
 193   addrvec.encode(bl, 0);
 194   bli = bl.begin();
 195
 196   addr.decode(bli);
 197
 198   ASSERT_EQ(addr, a);
 199 }
 200
 201 /* multiple legacy addrs */
 202 TEST(Msgr, TestAddrvecEncodeAddrDecode2)
 203 {
 204   entity_addr_t addr;
 205   entity_addrvec_t addrvec;
 206   bufferlist bl;
 207   bufferlist::iterator bli;
 208
 209   for (unsigned i = 0; addrvec_checks[i][2]; ++i) {
 210     const char *end = "";
 211     bool ok = addr.parse(addrvec_checks[i][2], &end);
 212     ASSERT_TRUE(ok);
 213     addrvec.v.push_back(addr);
 214   }
 215
 216   addrvec.encode(bl, 0);
 217   bli = bl.begin();
 218
 219   addr.decode(bli);
 220
 221   ASSERT_EQ(addr, addrvec.v[0]);
 222 }
 223
 224 /* all non-legacy addrs */
 225 TEST(Msgr, TestAddrvecEncodeAddrDecode3)
 226 {
 227   entity_addr_t addr;
 228   entity_addrvec_t addrvec;
 229   bufferlist bl;
 230   bufferlist::iterator bli;
 231
 232   for (unsigned i = 0; addrvec_checks[i][3]; ++i) {
 233     const char *end = "";
 234     bool ok = addr.parse(addrvec_checks[i][3], &end);
 235     ASSERT_TRUE(ok);
 236     addrvec.v.push_back(addr);
 237   }
 238
 239   addrvec.encode(bl, 0);
 240   bli = bl.begin();
 241
 242   addr.decode(bli);
 243
 244   ASSERT_NE(addr, addrvec.v[0]); // it's not the first addr(which is non-legacy)
 245   ASSERT_NE(addr, entity_addr_t(1, 0)); // it's not a blank addr either
 246 }