Built and tested on VS, works as expected.

ACK.

ti 2. lokak. 2018 klo 23.02 selva.n...@gmail.com kirjoitti:

> From: Selva Nair <selva.n...@gmail.com>
>
> Move writing the message buffer to the interactive service pipe and
> reading acknowledgement to a function.
>
> A minor bug in open_tun where the ack data could be read even after
> a communication error is fixed.
>
> Signed-off-by: Selva Nair <selva.n...@gmail.com>
> ---
>  src/openvpn/route.c |  6 +-----
>  src/openvpn/tun.c   | 34 +++++++++-------------------------
>  src/openvpn/win32.c | 27 ++++++++++++++++++++++-----
>  src/openvpn/win32.h |  9 +++++++++
>  4 files changed, 41 insertions(+), 35 deletions(-)
>
> diff --git a/src/openvpn/route.c b/src/openvpn/route.c
> index ff39230..8a3e8b4 100644
> --- a/src/openvpn/route.c
> +++ b/src/openvpn/route.c
> @@ -2991,16 +2991,12 @@ del_route_ipapi(const struct route_ipv4 *r, const
> struct tuntap *tt)
>  static bool
>  do_route_service(const bool add, const route_message_t *rt, const size_t
> size, HANDLE pipe)
>  {
> -    DWORD len;
>      bool ret = false;
>      ack_message_t ack;
>      struct gc_arena gc = gc_new();
>
> -    if (!WriteFile(pipe, rt, size, &len, NULL)
> -        || !ReadFile(pipe, &ack, sizeof(ack), &len, NULL))
> +    if (!send_msg_iservice(pipe, rt, size, &ack, "ROUTE"))
>      {
> -        msg(M_WARN, "ROUTE: could not talk to service: %s [%lu]",
> -            strerror_win32(GetLastError(), &gc), GetLastError());
>          goto out;
>      }
>
> diff --git a/src/openvpn/tun.c b/src/openvpn/tun.c
> index a2d5315..948fd17 100644
> --- a/src/openvpn/tun.c
> +++ b/src/openvpn/tun.c
> @@ -82,7 +82,6 @@ static DWORD get_adapter_index_flexible(const char
> *name);
>  static bool
>  do_address_service(const bool add, const short family, const struct
> tuntap *tt)
>  {
> -    DWORD len;
>      bool ret = false;
>      ack_message_t ack;
>      struct gc_arena gc = gc_new();
> @@ -115,11 +114,8 @@ do_address_service(const bool add, const short
> family, const struct tuntap *tt)
>          addr.prefix_len = tt->netbits_ipv6;
>      }
>
> -    if (!WriteFile(pipe, &addr, sizeof(addr), &len, NULL)
> -        || !ReadFile(pipe, &ack, sizeof(ack), &len, NULL))
> +    if (!send_msg_iservice(pipe, &addr, sizeof(addr), &ack, "TUN"))
>      {
> -        msg(M_WARN, "TUN: could not talk to service: %s [%lu]",
> -            strerror_win32(GetLastError(), &gc), GetLastError());
>          goto out;
>      }
>
> @@ -141,7 +137,6 @@ out:
>  static bool
>  do_dns6_service(bool add, const struct tuntap *tt)
>  {
> -    DWORD len;
>      bool ret = false;
>      ack_message_t ack;
>      struct gc_arena gc = gc_new();
> @@ -185,11 +180,8 @@ do_dns6_service(bool add, const struct tuntap *tt)
>      msg(D_LOW, "%s IPv6 dns servers on '%s' (if_index = %d) using
> service",
>          (add ? "Setting" : "Deleting"), dns.iface.name, dns.iface.index);
>
> -    if (!WriteFile(pipe, &dns, sizeof(dns), &len, NULL)
> -        || !ReadFile(pipe, &ack, sizeof(ack), &len, NULL))
> +    if (!send_msg_iservice(pipe, &dns, sizeof(dns), &ack, "TUN"))
>      {
> -        msg(M_WARN, "TUN: could not talk to service: %s [%lu]",
> -            strerror_win32(GetLastError(), &gc), GetLastError());
>          goto out;
>      }
>
> @@ -5222,11 +5214,8 @@ service_enable_dhcp(const struct tuntap *tt)
>          .iface = { .index = tt->adapter_index, .name = "" }
>      };
>
> -    if (!WriteFile(pipe, &dhcp, sizeof(dhcp), &len, NULL)
> -        || !ReadFile(pipe, &ack, sizeof(ack), &len, NULL))
> +    if (!send_msg_iservice(pipe, &dhcp, sizeof(dhcp), &ack,
> "Enable_dhcp"))
>      {
> -        msg(M_WARN, "Enable_dhcp: could not talk to service: %s [%lu]",
> -            strerror_win32(GetLastError(), &gc), GetLastError());
>          goto out;
>      }
>
> @@ -5461,18 +5450,16 @@ fork_dhcp_action(struct tuntap *tt)
>  static void
>  register_dns_service(const struct tuntap *tt)
>  {
> -    DWORD len;
>      HANDLE msg_channel = tt->options.msg_channel;
>      ack_message_t ack;
>      struct gc_arena gc = gc_new();
>
>      message_header_t rdns = { msg_register_dns, sizeof(message_header_t),
> 0 };
>
> -    if (!WriteFile(msg_channel, &rdns, sizeof(rdns), &len, NULL)
> -        || !ReadFile(msg_channel, &ack, sizeof(ack), &len, NULL))
> +    if (!send_msg_iservice(msg_channel, &rdns, sizeof(rdns), &ack,
> "Register_dns"))
>      {
> -        msg(M_WARN, "Register_dns: could not talk to service: %s
> [status=0x%lx]",
> -            strerror_win32(GetLastError(), &gc), GetLastError());
> +        gc_free(&gc);
> +        return;
>      }
>
>      else if (ack.error_number != NO_ERROR)
> @@ -5936,14 +5923,11 @@ open_tun(const char *dev, const char *dev_type,
> const char *dev_node, struct tun
>                      .iface = { .index = index, .name = "" }
>                  };
>
> -                if (!WriteFile(tt->options.msg_channel, &msg,
> sizeof(msg), &len, NULL)
> -                    || !ReadFile(tt->options.msg_channel, &ack,
> sizeof(ack), &len, NULL))
> +                if (send_msg_iservice(tt->options.msg_channel, &msg,
> sizeof(msg),
> +                    &ack, "TUN"))
>                  {
> -                    msg(M_WARN, "TUN: could not talk to service: %s
> [%lu]",
> -                        strerror_win32(GetLastError(), &gc),
> GetLastError());
> +                    status = ack.error_number;
>                  }
> -
> -                status = ack.error_number;
>              }
>              else
>              {
> diff --git a/src/openvpn/win32.c b/src/openvpn/win32.c
> index 3905524..e43296e 100644
> --- a/src/openvpn/win32.c
> +++ b/src/openvpn/win32.c
> @@ -1264,7 +1264,6 @@ win_get_tempdir(void)
>  static bool
>  win_block_dns_service(bool add, int index, const HANDLE pipe)
>  {
> -    DWORD len;
>      bool ret = false;
>      ack_message_t ack;
>      struct gc_arena gc = gc_new();
> @@ -1278,11 +1277,8 @@ win_block_dns_service(bool add, int index, const
> HANDLE pipe)
>          .iface = { .index = index, .name = "" }
>      };
>
> -    if (!WriteFile(pipe, &data, sizeof(data), &len, NULL)
> -        || !ReadFile(pipe, &ack, sizeof(ack), &len, NULL))
> +    if (!send_msg_iservice(pipe, &data, sizeof(data), &ack, "Block_DNS"))
>      {
> -        msg(M_WARN, "Block_DNS: could not talk to service: %s [%lu]",
> -            strerror_win32(GetLastError(), &gc), GetLastError());
>          goto out;
>      }
>
> @@ -1473,4 +1469,25 @@ win32_version_string(struct gc_arena *gc, bool
> add_name)
>      return (const char *)out.data;
>  }
>
> +bool
> +send_msg_iservice(HANDLE pipe, const void *data, size_t size,
> +                  ack_message_t *ack, const char *context)
> +{
> +    struct gc_arena gc = gc_new();
> +    DWORD len;
> +    bool ret = true;
> +
> +    if (!WriteFile(pipe, data, size, &len, NULL)
> +        || !ReadFile(pipe, ack, sizeof(*ack), &len, NULL))
> +    {
> +        msg(M_WARN, "%s: could not talk to service: %s [%lu]",
> +            context? context : "Unknown",
> +            strerror_win32(GetLastError(), &gc), GetLastError());
> +        ret = false;
> +    }
> +
> +    gc_free(&gc);
> +    return ret;
> +}
> +
>  #endif /* ifdef _WIN32 */
> diff --git a/src/openvpn/win32.h b/src/openvpn/win32.h
> index 4b99a5e..b5cbe25 100644
> --- a/src/openvpn/win32.h
> +++ b/src/openvpn/win32.h
> @@ -26,6 +26,7 @@
>  #define OPENVPN_WIN32_H
>
>  #include "mtu.h"
> +#include "openvpn-msg.h"
>
>  /* location of executables */
>  #define SYS_PATH_ENV_VAR_NAME "SystemRoot"  /* environmental variable
> name that normally contains the system path */
> @@ -307,5 +308,13 @@ int win32_version_info(void);
>   */
>  const char *win32_version_string(struct gc_arena *gc, bool add_name);
>
> +/*
> + * Send the |size| bytes in buffer |data| to the interactive service
> |pipe|
> + * and read the result in |ack|. Returns false on communication error.
> + * The string in |context| is used to prefix error messages.
> + */
> +bool send_msg_iservice(HANDLE pipe, const void *data, size_t size,
> +                       ack_message_t *ack, const char *context);
> +
>  #endif /* ifndef OPENVPN_WIN32_H */
>  #endif /* ifdef _WIN32 */
> --
> 2.1.4
>
>
>
> _______________________________________________
> Openvpn-devel mailing list
> Openvpn-devel@lists.sourceforge.net
> https://lists.sourceforge.net/lists/listinfo/openvpn-devel
>


-- 
-Lev
_______________________________________________
Openvpn-devel mailing list
Openvpn-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/openvpn-devel

Reply via email to