1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2  *
3  *  DataDifferential Utility Library
4  *
5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
6  *  All rights reserved.
7  *
8  *  Redistribution and use in source and binary forms, with or without
9  *  modification, are permitted provided that the following conditions are
10  *  met:
11  *
12  *      * Redistributions of source code must retain the above copyright
13  *  notice, this list of conditions and the following disclaimer.
14  *
15  *      * Redistributions in binary form must reproduce the above
16  *  copyright notice, this list of conditions and the following disclaimer
17  *  in the documentation and/or other materials provided with the
18  *  distribution.
19  *
20  *      * The names of its contributors may not be used to endorse or
21  *  promote products derived from this software without specific prior
22  *  written permission.
23  *
24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35  *
36  */
37 
38 #pragma once
39 
40 #include <cassert>
41 #include <cerrno>
42 #include <cstddef>
43 #include <cstdio>
44 #include <netinet/in.h>
45 #include <string>
46 #include <sys/socket.h>
47 
48 #include "util/operation.hpp"
49 
50 struct addrinfo;
51 
52 namespace datadifferential {
53 namespace util {
54 
55 class Instance
56 {
57 private:
58   enum connection_state_t {
59     NOT_WRITING,
60     NEXT_CONNECT_ADDRINFO,
61     CONNECT,
62     CONNECTING,
63     CONNECTED,
64     WRITING,
65     READING,
66     FINISHED
67   };
68   std::string _last_error;
69 
70 public: // Callbacks
71   class Finish {
72 
73   public:
~Finish()74     virtual ~Finish() { }
75 
76     virtual bool call(const bool, const std::string &)= 0;
77   };
78 
79 
80 public:
81   Instance(const std::string& hostname_arg, const std::string& service_arg);
82 
83   Instance(const std::string& hostname_arg, const in_port_t port_arg);
84 
85   ~Instance();
86 
87   bool run();
88 
set_finish(Finish * arg)89   void set_finish(Finish *arg)
90   {
91     _finish_fn= arg;
92   }
93 
push(util::Operation * next)94   void push(util::Operation *next)
95   {
96     _operations.push_back(next);
97   }
98 
99 private:
100   void close_socket();
101 
102   void free_addrinfo();
103 
104   bool more_to_read() const;
105 
106   std::string _host;
107   std::string _service;
108   int _sockfd;
109   connection_state_t state;
110   struct addrinfo *_addrinfo;
111   struct addrinfo *_addrinfo_next;
112   Finish *_finish_fn;
113   Operation::vector _operations;
114 };
115 
116 } /* namespace util */
117 } /* namespace datadifferential */
118