src/lib/mdns/minimal/records/Srv.h

   1 /*
   2  *
   3  *    Copyright (c) 2020 Project CHIP Authors
   4  *
   5  *    Licensed under the Apache License, Version 2.0 (the "License");
   6  *    you may not use this file except in compliance with the License.
   7  *    You may obtain a copy of the License at
   8  *
   9  *        http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  *    Unless required by applicable law or agreed to in writing, software
  12  *    distributed under the License is distributed on an "AS IS" BASIS,
  13  *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  *    See the License for the specific language governing permissions and
  15  *    limitations under the License.
  16  */
  17
  18 #pragma once
  19
  20 #include <mdns/minimal/records/ResourceRecord.h>
  21
  22 namespace mdns {
  23 namespace Minimal {
  24
  25 class SrvResourceRecord : public ResourceRecord
  26 {
  27 public:
  28     SrvResourceRecord(const FullQName & qName, const FullQName & serverName, uint16_t port) :
  29         ResourceRecord(QType::SRV, qName), mServerName(serverName), mPort(port)
  30     {}
  31     SrvResourceRecord & operator=(const SrvResourceRecord & other) = default;
  32
  33     FullQName GetServerName() const { return mServerName; }
  34     uint16_t GetPort() const { return mPort; }
  35     uint16_t GetPriority() const { return mPriority; }
  36     uint16_t GetWeight() const { return mWeight; }
  37
  38     void SetPriority(uint16_t value) { mPriority = value; }
  39     void SetWeight(uint16_t value) { mWeight = value; }
  40
  41 protected:
  42     bool WriteData(chip::Encoding::BigEndian::BufferWriter & out) const override
  43     {
  44         out.Put16(mPriority);
  45         out.Put16(mWeight);
  46         out.Put16(mPort);
  47         mServerName.Output(out);
  48
  49         return out.Fit();
  50     }
  51
  52 private:
  53     FullQName mServerName;
  54     uint16_t mPort;
  55     uint16_t mPriority = 0;
  56     uint16_t mWeight   = 0;
  57 };
  58
  59 } // namespace Minimal
  60 } // namespace mdns