/*
 * This file is part of PowerDNS or dnsdist.
 * Copyright -- PowerDNS.COM B.V. and its contributors
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * In addition, for the avoidance of any doubt, permission is granted to
 * link this program with OpenSSL and to (re)distribute the binaries
 * produced as the result of such linking.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
#pragma once
#include <stdio.h>
#include <thread>
#include <vector>

#include "dnsseckeeper.hh"
#include "dns.hh"

void writeLStringToSocket(int fd, const string& msg);
bool readLStringFromSocket(int fd, string& msg);

/** input: DNSZoneRecords ordered in qname,qtype (we emit a signature chunk on a break)
 *  output: "chunks" of those very same DNSZoneRecords, interleaved with signatures
 */

class ChunkedSigningPipe
{
public:
  typedef vector<DNSZoneRecord> rrset_t; 
  typedef rrset_t chunk_t; // for now
  
  ChunkedSigningPipe(const ChunkedSigningPipe&) = delete;
  void operator=(const ChunkedSigningPipe&) = delete;
  ChunkedSigningPipe(DNSName  signerName, bool mustSign, unsigned int numWorkers, unsigned int maxChunkRecords);
  ~ChunkedSigningPipe();
  bool submit(const DNSZoneRecord& rr);
  chunk_t getChunk(bool final=false);
  unsigned int getReady() const;

  std::atomic<unsigned long> d_signed;
  unsigned int d_queued;
  unsigned int d_outstanding;

private:
  void flushToSign();	
  void dedupRRSet();
  void sendRRSetToWorker(); // dispatch RRSET to worker
  void addSignedToChunks(std::unique_ptr<chunk_t>& signedChunk);
  pair<vector<int>, vector<int> > waitForRW(bool rd, bool wr, int seconds);

  static void* helperWorker(ChunkedSigningPipe* csp, int fd);
  void worker(int fd);

  unsigned int d_numworkers;
  unsigned int d_submitted;

  std::unique_ptr<rrset_t> d_rrsetToSign;
  std::deque< std::vector<DNSZoneRecord> > d_chunks;
  DNSName d_signer;
  
  chunk_t::size_type d_maxchunkrecords;
  
  std::vector<int> d_sockets;
  std::set<int> d_eof;
  std::map<int,int> d_outstandings;

  vector<std::thread> d_threads;
  bool d_mustSign;
  bool d_final;
};
