+++ /dev/null
-// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
-// vim: ts=8 sw=2 smarttab
-/*
- * Ceph - scalable distributed file system
- *
- * Copyright (C) 2004-2006 Sage Weil <sage@newdream.net>
- *
- * This is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License version 2.1, as published by the Free Software
- * Foundation. See file COPYING.
- *
- */
-
-#ifndef CEPH_MSG_TYPES_H
-#define CEPH_MSG_TYPES_H
-
-#include <netinet/in.h>
-
-#include "include/ceph_features.h"
-#include "include/types.h"
-#include "include/blobhash.h"
-#include "include/encoding.h"
-
-namespace ceph {
- class Formatter;
-}
-
-extern ostream& operator<<(ostream& out, const sockaddr_storage &ss);
-extern ostream& operator<<(ostream& out, const sockaddr *sa);
-
-typedef uint8_t entity_type_t;
-
-class entity_name_t {
-public:
- entity_type_t _type;
- int64_t _num;
-
-public:
- static const int TYPE_MON = CEPH_ENTITY_TYPE_MON;
- static const int TYPE_MDS = CEPH_ENTITY_TYPE_MDS;
- static const int TYPE_OSD = CEPH_ENTITY_TYPE_OSD;
- static const int TYPE_CLIENT = CEPH_ENTITY_TYPE_CLIENT;
- static const int TYPE_MGR = CEPH_ENTITY_TYPE_MGR;
-
- static const int64_t NEW = -1;
-
- // cons
- entity_name_t() : _type(0), _num(0) { }
- entity_name_t(int t, int64_t n) : _type(t), _num(n) { }
- explicit entity_name_t(const ceph_entity_name &n) :
- _type(n.type), _num(n.num) { }
-
- // static cons
- static entity_name_t MON(int64_t i=NEW) { return entity_name_t(TYPE_MON, i); }
- static entity_name_t MDS(int64_t i=NEW) { return entity_name_t(TYPE_MDS, i); }
- static entity_name_t OSD(int64_t i=NEW) { return entity_name_t(TYPE_OSD, i); }
- static entity_name_t CLIENT(int64_t i=NEW) { return entity_name_t(TYPE_CLIENT, i); }
- static entity_name_t MGR(int64_t i=NEW) { return entity_name_t(TYPE_MGR, i); }
-
- int64_t num() const { return _num; }
- int type() const { return _type; }
- const char *type_str() const {
- return ceph_entity_type_name(type());
- }
-
- bool is_new() const { return num() < 0; }
-
- bool is_client() const { return type() == TYPE_CLIENT; }
- bool is_mds() const { return type() == TYPE_MDS; }
- bool is_osd() const { return type() == TYPE_OSD; }
- bool is_mon() const { return type() == TYPE_MON; }
- bool is_mgr() const { return type() == TYPE_MGR; }
-
- operator ceph_entity_name() const {
- ceph_entity_name n = { _type, init_le64(_num) };
- return n;
- }
-
- bool parse(const string& s) {
- const char *start = s.c_str();
- char *end;
- bool got = parse(start, &end);
- return got && end == start + s.length();
- }
- bool parse(const char *start, char **end) {
- if (strstr(start, "mon.") == start) {
- _type = TYPE_MON;
- start += 4;
- } else if (strstr(start, "osd.") == start) {
- _type = TYPE_OSD;
- start += 4;
- } else if (strstr(start, "mds.") == start) {
- _type = TYPE_MDS;
- start += 4;
- } else if (strstr(start, "client.") == start) {
- _type = TYPE_CLIENT;
- start += 7;
- } else if (strstr(start, "mgr.") == start) {
- _type = TYPE_MGR;
- start += 4;
- } else {
- return false;
- }
- if (isspace(*start))
- return false;
- _num = strtoll(start, end, 10);
- if (*end == NULL || *end == start)
- return false;
- return true;
- }
-
- DENC(entity_name_t, v, p) {
- denc(v._type, p);
- denc(v._num, p);
- }
- void dump(Formatter *f) const;
-
- static void generate_test_instances(list<entity_name_t*>& o);
-};
-WRITE_CLASS_DENC(entity_name_t)
-
-inline bool operator== (const entity_name_t& l, const entity_name_t& r) {
- return (l.type() == r.type()) && (l.num() == r.num()); }
-inline bool operator!= (const entity_name_t& l, const entity_name_t& r) {
- return (l.type() != r.type()) || (l.num() != r.num()); }
-inline bool operator< (const entity_name_t& l, const entity_name_t& r) {
- return (l.type() < r.type()) || (l.type() == r.type() && l.num() < r.num()); }
-
-inline std::ostream& operator<<(std::ostream& out, const entity_name_t& addr) {
- //if (addr.is_namer()) return out << "namer";
- if (addr.is_new() || addr.num() < 0)
- return out << addr.type_str() << ".?";
- else
- return out << addr.type_str() << '.' << addr.num();
-}
-inline std::ostream& operator<<(std::ostream& out, const ceph_entity_name& addr) {
- return out << *(const entity_name_t*)&addr;
-}
-
-namespace std {
- template<> struct hash< entity_name_t >
- {
- size_t operator()( const entity_name_t &m ) const
- {
- return rjhash32(m.type() ^ m.num());
- }
- };
-} // namespace std
-
-/*
- * an entity's network address.
- * includes a random value that prevents it from being reused.
- * thus identifies a particular process instance.
- * ipv4 for now.
- */
-
-#if defined(__linux__) || defined(DARWIN) || defined(__FreeBSD__)
-/*
- * encode sockaddr.ss_family as network byte order
- */
-static inline void encode(const sockaddr_storage& a, bufferlist& bl) {
- struct sockaddr_storage ss = a;
-#if defined(DARWIN) || defined(__FreeBSD__)
- unsigned short *ss_family = reinterpret_cast<unsigned short*>(&ss);
- *ss_family = htons(a.ss_family);
-#else
- ss.ss_family = htons(ss.ss_family);
-#endif
- ::encode_raw(ss, bl);
-}
-static inline void decode(sockaddr_storage& a, bufferlist::iterator& bl) {
- ::decode_raw(a, bl);
-#if defined(DARWIN) || defined(__FreeBSD__)
- unsigned short *ss_family = reinterpret_cast<unsigned short *>(&a);
- a.ss_family = ntohs(*ss_family);
- a.ss_len = 0;
-#else
- a.ss_family = ntohs(a.ss_family);
-#endif
-}
-#endif
-
-// define a wire format for sockaddr that matches Linux's.
-struct ceph_sockaddr_storage {
- __le16 ss_family;
- __u8 __ss_padding[128 - sizeof(__le16)];
-
- void encode(bufferlist& bl) const {
- struct ceph_sockaddr_storage ss = *this;
- ss.ss_family = htons(ss.ss_family);
- ::encode_raw(ss, bl);
- }
-
- void decode(bufferlist::iterator& bl) {
- struct ceph_sockaddr_storage ss;
- ::decode_raw(ss, bl);
- ss.ss_family = ntohs(ss.ss_family);
- *this = ss;
- }
-} __attribute__ ((__packed__));
-WRITE_CLASS_ENCODER(ceph_sockaddr_storage)
-
-struct entity_addr_t {
- typedef enum {
- TYPE_NONE = 0,
- TYPE_LEGACY = 1, ///< legacy msgr1 protocol (ceph jewel and older)
- TYPE_MSGR2 = 2, ///< msgr2 protocol (new in ceph kraken)
- } type_t;
- static const type_t TYPE_DEFAULT = TYPE_LEGACY;
- static const char *get_type_name(int t) {
- switch (t) {
- case TYPE_NONE: return "none";
- case TYPE_LEGACY: return "legacy";
- case TYPE_MSGR2: return "msgr2";
- default: return "???";
- }
- };
-
- __u32 type;
- __u32 nonce;
- union {
- sockaddr sa;
- sockaddr_in sin;
- sockaddr_in6 sin6;
- } u;
-
- entity_addr_t() : type(0), nonce(0) {
- memset(&u, 0, sizeof(u));
- }
- entity_addr_t(__u32 _type, __u32 _nonce) : type(_type), nonce(_nonce) {
- memset(&u, 0, sizeof(u));
- }
- explicit entity_addr_t(const ceph_entity_addr &o) {
- type = o.type;
- nonce = o.nonce;
- memcpy(&u, &o.in_addr, sizeof(u));
-#if !defined(__FreeBSD__)
- u.sa.sa_family = ntohs(u.sa.sa_family);
-#endif
- }
-
- uint32_t get_type() const { return type; }
- void set_type(uint32_t t) { type = t; }
-
- __u32 get_nonce() const { return nonce; }
- void set_nonce(__u32 n) { nonce = n; }
-
- int get_family() const {
- return u.sa.sa_family;
- }
- void set_family(int f) {
- u.sa.sa_family = f;
- }
-
- sockaddr_in &in4_addr() {
- return u.sin;
- }
- const sockaddr_in &in4_addr() const{
- return u.sin;
- }
- sockaddr_in6 &in6_addr(){
- return u.sin6;
- }
- const sockaddr_in6 &in6_addr() const{
- return u.sin6;
- }
- const sockaddr *get_sockaddr() const {
- return &u.sa;
- }
- size_t get_sockaddr_len() const {
- switch (u.sa.sa_family) {
- case AF_INET:
- return sizeof(u.sin);
- case AF_INET6:
- return sizeof(u.sin6);
- }
- return sizeof(u);
- }
- bool set_sockaddr(const struct sockaddr *sa)
- {
- switch (sa->sa_family) {
- case AF_INET:
- memcpy(&u.sin, sa, sizeof(u.sin));
- break;
- case AF_INET6:
- memcpy(&u.sin6, sa, sizeof(u.sin6));
- break;
- default:
- return false;
- }
- return true;
- }
-
- sockaddr_storage get_sockaddr_storage() const {
- sockaddr_storage ss;
- memcpy(&ss, &u, sizeof(u));
- memset((char*)&ss + sizeof(u), 0, sizeof(ss) - sizeof(u));
- return ss;
- }
-
- void set_in4_quad(int pos, int val) {
- u.sin.sin_family = AF_INET;
- unsigned char *ipq = (unsigned char*)&u.sin.sin_addr.s_addr;
- ipq[pos] = val;
- }
- void set_port(int port) {
- switch (u.sa.sa_family) {
- case AF_INET:
- u.sin.sin_port = htons(port);
- break;
- case AF_INET6:
- u.sin6.sin6_port = htons(port);
- break;
- default:
- ceph_abort();
- }
- }
- int get_port() const {
- switch (u.sa.sa_family) {
- case AF_INET:
- return ntohs(u.sin.sin_port);
- break;
- case AF_INET6:
- return ntohs(u.sin6.sin6_port);
- break;
- }
- return 0;
- }
-
- operator ceph_entity_addr() const {
- ceph_entity_addr a;
- a.type = 0;
- a.nonce = nonce;
- a.in_addr = get_sockaddr_storage();
-#if !defined(__FreeBSD__)
- a.in_addr.ss_family = htons(a.in_addr.ss_family);
-#endif
- return a;
- }
-
- bool probably_equals(const entity_addr_t &o) const {
- if (get_port() != o.get_port())
- return false;
- if (get_nonce() != o.get_nonce())
- return false;
- if (is_blank_ip() || o.is_blank_ip())
- return true;
- if (memcmp(&u, &o.u, sizeof(u)) == 0)
- return true;
- return false;
- }
-
- bool is_same_host(const entity_addr_t &o) const {
- if (u.sa.sa_family != o.u.sa.sa_family)
- return false;
- if (u.sa.sa_family == AF_INET)
- return u.sin.sin_addr.s_addr == o.u.sin.sin_addr.s_addr;
- if (u.sa.sa_family == AF_INET6)
- return memcmp(u.sin6.sin6_addr.s6_addr,
- o.u.sin6.sin6_addr.s6_addr,
- sizeof(u.sin6.sin6_addr.s6_addr)) == 0;
- return false;
- }
-
- bool is_blank_ip() const {
- switch (u.sa.sa_family) {
- case AF_INET:
- return u.sin.sin_addr.s_addr == INADDR_ANY;
- case AF_INET6:
- return memcmp(&u.sin6.sin6_addr, &in6addr_any, sizeof(in6addr_any)) == 0;
- default:
- return true;
- }
- }
-
- bool is_ip() const {
- switch (u.sa.sa_family) {
- case AF_INET:
- case AF_INET6:
- return true;
- default:
- return false;
- }
- }
-
- bool parse(const char *s, const char **end = 0);
-
- void decode_legacy_addr_after_marker(bufferlist::iterator& bl)
- {
- __u8 marker;
- __u16 rest;
- ::decode(marker, bl);
- ::decode(rest, bl);
- type = TYPE_LEGACY;
- ::decode(nonce, bl);
- sockaddr_storage ss;
-#if defined(__linux__) || defined(DARWIN) || defined(__FreeBSD__)
- ::decode(ss, bl);
-#else
- ceph_sockaddr_storage wireaddr;
- ::memset(&wireaddr, '\0', sizeof(wireaddr));
- ::decode(wireaddr, bl);
- unsigned copysize = MIN(sizeof(wireaddr), sizeof(ss));
- ::memcpy(&ss, &wireaddr, copysize);
-#endif
- set_sockaddr((sockaddr*)&ss);
- }
-
- // Right now, these only deal with sockaddr_storage that have only family and content.
- // Apparently on BSD there is also an ss_len that we need to handle; this requires
- // broader study
-
- void encode(bufferlist& bl, uint64_t features) const {
- if ((features & CEPH_FEATURE_MSG_ADDR2) == 0) {
- ::encode((__u32)0, bl);
- ::encode(nonce, bl);
- sockaddr_storage ss = get_sockaddr_storage();
-#if defined(__linux__) || defined(DARWIN) || defined(__FreeBSD__)
- ::encode(ss, bl);
-#else
- ceph_sockaddr_storage wireaddr;
- ::memset(&wireaddr, '\0', sizeof(wireaddr));
- unsigned copysize = MIN(sizeof(wireaddr), sizeof(ss));
- // ceph_sockaddr_storage is in host byte order
- ::memcpy(&wireaddr, &ss, copysize);
- ::encode(wireaddr, bl);
-#endif
- return;
- }
- ::encode((__u8)1, bl);
- ENCODE_START(1, 1, bl);
- ::encode(type, bl);
- ::encode(nonce, bl);
- __u32 elen = get_sockaddr_len();
- ::encode(elen, bl);
- if (elen) {
- bl.append((char*)get_sockaddr(), elen);
- }
- ENCODE_FINISH(bl);
- }
- void decode(bufferlist::iterator& bl) {
- __u8 marker;
- ::decode(marker, bl);
- if (marker == 0) {
- decode_legacy_addr_after_marker(bl);
- return;
- }
- if (marker != 1)
- throw buffer::malformed_input("entity_addr_t marker != 1");
- DECODE_START(1, bl);
- ::decode(type, bl);
- ::decode(nonce, bl);
- __u32 elen;
- ::decode(elen, bl);
- if (elen) {
- bl.copy(elen, (char*)get_sockaddr());
- }
- DECODE_FINISH(bl);
- }
-
- void dump(Formatter *f) const;
-
- static void generate_test_instances(list<entity_addr_t*>& o);
-};
-WRITE_CLASS_ENCODER_FEATURES(entity_addr_t)
-
-ostream& operator<<(ostream& out, const entity_addr_t &addr);
-
-inline bool operator==(const entity_addr_t& a, const entity_addr_t& b) { return memcmp(&a, &b, sizeof(a)) == 0; }
-inline bool operator!=(const entity_addr_t& a, const entity_addr_t& b) { return memcmp(&a, &b, sizeof(a)) != 0; }
-inline bool operator<(const entity_addr_t& a, const entity_addr_t& b) { return memcmp(&a, &b, sizeof(a)) < 0; }
-inline bool operator<=(const entity_addr_t& a, const entity_addr_t& b) { return memcmp(&a, &b, sizeof(a)) <= 0; }
-inline bool operator>(const entity_addr_t& a, const entity_addr_t& b) { return memcmp(&a, &b, sizeof(a)) > 0; }
-inline bool operator>=(const entity_addr_t& a, const entity_addr_t& b) { return memcmp(&a, &b, sizeof(a)) >= 0; }
-
-namespace std {
- template<> struct hash< entity_addr_t >
- {
- size_t operator()( const entity_addr_t& x ) const
- {
- static blobhash H;
- return H((const char*)&x, sizeof(x));
- }
- };
-} // namespace std
-
-struct entity_addrvec_t {
- vector<entity_addr_t> v;
-
- unsigned size() const { return v.size(); }
- bool empty() const { return v.empty(); }
-
- void encode(bufferlist& bl, uint64_t features) const;
- void decode(bufferlist::iterator& bl);
- void dump(Formatter *f) const;
- static void generate_test_instances(list<entity_addrvec_t*>& ls);
-};
-WRITE_CLASS_ENCODER_FEATURES(entity_addrvec_t);
-
-/*
- * a particular entity instance
- */
-struct entity_inst_t {
- entity_name_t name;
- entity_addr_t addr;
- entity_inst_t() {}
- entity_inst_t(entity_name_t n, const entity_addr_t& a) : name(n), addr(a) {}
- // cppcheck-suppress noExplicitConstructor
- entity_inst_t(const ceph_entity_inst& i) : name(i.name), addr(i.addr) { }
- entity_inst_t(const ceph_entity_name& n, const ceph_entity_addr &a) : name(n), addr(a) {}
- operator ceph_entity_inst() {
- ceph_entity_inst i = {name, addr};
- return i;
- }
-
- void encode(bufferlist& bl, uint64_t features) const {
- ::encode(name, bl);
- ::encode(addr, bl, features);
- }
- void decode(bufferlist::iterator& bl) {
- ::decode(name, bl);
- ::decode(addr, bl);
- }
-
- void dump(Formatter *f) const;
- static void generate_test_instances(list<entity_inst_t*>& o);
-};
-WRITE_CLASS_ENCODER_FEATURES(entity_inst_t)
-
-
-inline bool operator==(const entity_inst_t& a, const entity_inst_t& b) {
- return a.name == b.name && a.addr == b.addr;
-}
-inline bool operator!=(const entity_inst_t& a, const entity_inst_t& b) {
- return a.name != b.name || a.addr != b.addr;
-}
-inline bool operator<(const entity_inst_t& a, const entity_inst_t& b) {
- return a.name < b.name || (a.name == b.name && a.addr < b.addr);
-}
-inline bool operator<=(const entity_inst_t& a, const entity_inst_t& b) {
- return a.name < b.name || (a.name == b.name && a.addr <= b.addr);
-}
-inline bool operator>(const entity_inst_t& a, const entity_inst_t& b) { return b < a; }
-inline bool operator>=(const entity_inst_t& a, const entity_inst_t& b) { return b <= a; }
-
-namespace std {
- template<> struct hash< entity_inst_t >
- {
- size_t operator()( const entity_inst_t& x ) const
- {
- static hash< entity_name_t > H;
- static hash< entity_addr_t > I;
- return H(x.name) ^ I(x.addr);
- }
- };
-} // namespace std
-
-
-inline ostream& operator<<(ostream& out, const entity_inst_t &i)
-{
- return out << i.name << " " << i.addr;
-}
-inline ostream& operator<<(ostream& out, const ceph_entity_inst &i)
-{
- entity_inst_t n = i;
- return out << n;
-}
-
-#endif