[mirror_frr.git] / pimd / pim_zlookup.c

/*
  PIM for Quagga
  Copyright (C) 2008  Everton da Silva Marques

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  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; see the file COPYING; if not, write to the
  Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston,
  MA 02110-1301 USA
  
*/

#include <zebra.h>
#include "zebra/rib.h"

#include "log.h"
#include "prefix.h"
#include "zclient.h"
#include "stream.h"
#include "network.h"
#include "thread.h"
#include "prefix.h"
#include "vty.h"

#include "pimd.h"
#include "pim_iface.h"
#include "pim_pim.h"
#include "pim_str.h"
#include "pim_oil.h"
#include "pim_zlookup.h"

extern int zclient_debug;

static struct zclient *zlookup = NULL;

static void zclient_lookup_sched(struct zclient *zlookup, int delay);

/* Connect to zebra for nexthop lookup. */
static int zclient_lookup_connect(struct thread *t)
{
  struct zclient *zlookup;

  zlookup = THREAD_ARG(t);
  zlookup->t_connect = NULL;

  if (zlookup->sock >= 0) {
    return 0;
  }

  if (zclient_socket_connect(zlookup) < 0) {
    ++zlookup->fail;
    zlog_warn("%s: failure connecting zclient socket: failures=%d",
	      __PRETTY_FUNCTION__, zlookup->fail);
  }
  else {
    zlookup->fail = 0; /* reset counter on connection */
  }

  zassert(!zlookup->t_connect);
  if (zlookup->sock < 0) {
    /* Since last connect failed, retry within 10 secs */
    zclient_lookup_sched(zlookup, 10);
    return -1;
  }

  return 0;
}

/* Schedule connection with delay. */
static void zclient_lookup_sched(struct zclient *zlookup, int delay)
{
  zassert(!zlookup->t_connect);

  THREAD_TIMER_ON(master, zlookup->t_connect,
		  zclient_lookup_connect,
		  zlookup, delay);

  zlog_notice("%s: zclient lookup connection scheduled for %d seconds",
	      __PRETTY_FUNCTION__, delay);
}

/* Schedule connection for now. */
static void zclient_lookup_sched_now(struct zclient *zlookup)
{
  zassert(!zlookup->t_connect);

  zlookup->t_connect = thread_add_event(master, zclient_lookup_connect,
					zlookup, 0);

  zlog_notice("%s: zclient lookup immediate connection scheduled",
	      __PRETTY_FUNCTION__);
}

/* Schedule reconnection, if needed. */
static void zclient_lookup_reconnect(struct zclient *zlookup)
{
  if (zlookup->t_connect) {
    return;
  }

  zclient_lookup_sched_now(zlookup);
}

static void zclient_lookup_failed(struct zclient *zlookup)
{
  if (zlookup->sock >= 0) {
    if (close(zlookup->sock)) {
      zlog_warn("%s: closing fd=%d: errno=%d %s", __func__, zlookup->sock,
		errno, safe_strerror(errno));
    }
    zlookup->sock = -1;
  }

  zclient_lookup_reconnect(zlookup);
}

void
zclient_lookup_new (void)
{
  zlookup = zclient_new (master);
  if (!zlookup) {
    zlog_err("%s: zclient_new() failure",
	     __PRETTY_FUNCTION__);
    return;
  }

  zlookup->sock = -1;
  zlookup->ibuf = stream_new(ZEBRA_MAX_PACKET_SIZ);
  zlookup->obuf = stream_new(ZEBRA_MAX_PACKET_SIZ);
  zlookup->t_connect = 0;

  zclient_lookup_sched_now(zlookup);

  zlog_notice("%s: zclient lookup socket initialized",
	      __PRETTY_FUNCTION__);

}

static int zclient_read_nexthop(struct zclient *zlookup,
				struct pim_zlookup_nexthop nexthop_tab[],
				const int tab_size,
				struct in_addr addr)
{
  int num_ifindex = 0;
  struct stream *s;
  const uint16_t MIN_LEN = 10; /* getipv4=4 getc=1 getl=4 getc=1 */
  uint16_t length;
  u_char marker;
  u_char version;
  vrf_id_t vrf_id;
  uint16_t command = 0;
  struct in_addr raddr;
  uint8_t distance;
  uint32_t metric;
  int nexthop_num;
  int i, err;

  if (PIM_DEBUG_ZEBRA) {
    char addr_str[100];
    pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
    zlog_debug("%s: addr=%s", 
	       __PRETTY_FUNCTION__,
	       addr_str);
  }

  s = zlookup->ibuf;

  while (command != ZEBRA_IPV4_NEXTHOP_LOOKUP_MRIB)
    {
      stream_reset(s);
      err = zclient_read_header (s, zlookup->sock, &length, &marker, &version,
				 &vrf_id, &command);
      if (err < 0) {
	zlog_err("%s %s: zclient_read_header() failed",
		 __FILE__, __PRETTY_FUNCTION__);
	zclient_lookup_failed(zlookup);
	return -1;
      }

      if (length < MIN_LEN) {
	zlog_err("%s %s: failure reading zclient lookup socket: len=%d < MIN_LEN=%d",
		 __FILE__, __PRETTY_FUNCTION__, length, MIN_LEN);
	zclient_lookup_failed(zlookup);
	return -2;
      }
    }

  raddr.s_addr = stream_get_ipv4(s);

  if (raddr.s_addr != addr.s_addr) {
    char addr_str[100];
    char raddr_str[100];
    pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
    pim_inet4_dump("<raddr?>", raddr, raddr_str, sizeof(raddr_str));
    zlog_warn("%s: address mismatch: addr=%s raddr=%s", 
	       __PRETTY_FUNCTION__,
	       addr_str, raddr_str);
    /* warning only */
  }

  distance = stream_getc(s);
  metric = stream_getl(s);
  nexthop_num = stream_getc(s);

  if (nexthop_num < 1) {
    zlog_err("%s: socket %d bad nexthop_num=%d",
            __func__, zlookup->sock, nexthop_num);
    return -6;
  }

  for (i = 0; i < nexthop_num; ++i) {
    enum nexthop_types_t nexthop_type;
    struct pim_neighbor *nbr;

    nexthop_type = stream_getc(s);
    if (num_ifindex >= tab_size) {
      char addr_str[100];
      pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
      zlog_warn("%s %s: found too many nexthop ifindexes (%d > %d) for address %s",
		__FILE__, __PRETTY_FUNCTION__,
		(num_ifindex + 1), tab_size, addr_str);
      return num_ifindex;
    }
    switch (nexthop_type) {
    case NEXTHOP_TYPE_IFINDEX:
    case NEXTHOP_TYPE_IPV4_IFINDEX:
    case NEXTHOP_TYPE_IPV4:
      nexthop_tab[num_ifindex].nexthop_addr.family = AF_INET;
      if (nexthop_type == NEXTHOP_TYPE_IPV4_IFINDEX ||
	  nexthop_type == NEXTHOP_TYPE_IPV4) {
	nexthop_tab[num_ifindex].nexthop_addr.u.prefix4.s_addr = stream_get_ipv4(s);
      }
      else {
	nexthop_tab[num_ifindex].nexthop_addr.u.prefix4.s_addr = PIM_NET_INADDR_ANY;
      }
      nexthop_tab[num_ifindex].ifindex           = stream_getl(s);
      nexthop_tab[num_ifindex].protocol_distance = distance;
      nexthop_tab[num_ifindex].route_metric      = metric;
      ++num_ifindex;
      break;
    case NEXTHOP_TYPE_IPV6_IFINDEX:
      nexthop_tab[num_ifindex].nexthop_addr.family = AF_INET6;
      stream_get (&nexthop_tab[num_ifindex].nexthop_addr.u.prefix6, s, 16);
      nexthop_tab[num_ifindex].ifindex = stream_getl (s);
      nbr = pim_neighbor_find_if (if_lookup_by_index_vrf (nexthop_tab[num_ifindex].ifindex, VRF_DEFAULT));
      if (nbr)
        {
          nexthop_tab[num_ifindex].nexthop_addr.family = AF_INET;
          nexthop_tab[num_ifindex].nexthop_addr.u.prefix4 = nbr->source_addr;
        }
        ++num_ifindex;
      break;
    default:
      /* do nothing */
      {
	char addr_str[100];
	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	zlog_warn("%s %s: found non-ifindex nexthop type=%d for address %s",
		 __FILE__, __PRETTY_FUNCTION__,
		  nexthop_type, addr_str);
      }
      break;
    }
  }

  return num_ifindex;
}

static int
zclient_lookup_nexthop_once (struct pim_zlookup_nexthop nexthop_tab[],
			     const int tab_size,
			     struct in_addr addr)
{
  struct stream *s;
  int ret;

  if (PIM_DEBUG_ZEBRA) {
    char addr_str[100];
    pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
    zlog_debug("%s: addr=%s", 
	       __PRETTY_FUNCTION__,
	       addr_str);
  }

  /* Check socket. */
  if (zlookup->sock < 0) {
    zlog_err("%s %s: zclient lookup socket is not connected",
	     __FILE__, __PRETTY_FUNCTION__);
    zclient_lookup_failed(zlookup);
    return -1;
  }
  
  s = zlookup->obuf;
  stream_reset(s);
  zclient_create_header(s, ZEBRA_IPV4_NEXTHOP_LOOKUP_MRIB, VRF_DEFAULT);
  stream_put_in_addr(s, &addr);
  stream_putw_at(s, 0, stream_get_endp(s));
  
  ret = writen(zlookup->sock, s->data, stream_get_endp(s));
  if (ret < 0) {
    zlog_err("%s %s: writen() failure writing to zclient lookup socket",
	     __FILE__, __PRETTY_FUNCTION__);
    zclient_lookup_failed(zlookup);
    return -2;
  }
  if (ret == 0) {
    zlog_err("%s %s: connection closed on zclient lookup socket",
	     __FILE__, __PRETTY_FUNCTION__);
    zclient_lookup_failed(zlookup);
    return -3;
  }
  
  return zclient_read_nexthop(zlookup, nexthop_tab,
			      tab_size, addr);
}

int
zclient_lookup_nexthop (struct pim_zlookup_nexthop nexthop_tab[],
			const int tab_size,
			struct in_addr addr,
			int max_lookup)
{
  int lookup;
  uint32_t route_metric = 0xFFFFFFFF;
  uint8_t  protocol_distance = 0xFF;

  qpim_nexthop_lookups++;

  for (lookup = 0; lookup < max_lookup; ++lookup) {
    int num_ifindex;
    int first_ifindex;
    struct prefix nexthop_addr;

    num_ifindex = zclient_lookup_nexthop_once(nexthop_tab,
					      tab_size, addr);
    if (num_ifindex < 1) {
      if (PIM_DEBUG_ZEBRA) {
	char addr_str[100];
	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	zlog_debug("%s %s: lookup=%d/%d: could not find nexthop ifindex for address %s",
		   __FILE__, __PRETTY_FUNCTION__,
		   lookup, max_lookup, addr_str);
      }
      return -1;
    }

    if (lookup < 1) {
      /* this is the non-recursive lookup - save original metric/distance */
      route_metric = nexthop_tab[0].route_metric;
      protocol_distance = nexthop_tab[0].protocol_distance;
    }
    
    /*
     * FIXME: Non-recursive nexthop ensured only for first ifindex.
     * However, recursive route lookup should really be fixed in zebra daemon.
     * See also TODO T24.
     *
     * So Zebra for NEXTHOP_TYPE_IPV4 returns the ifindex now since
     * it was being stored.  This Doesn't solve all cases of
     * recursive lookup but for the most common types it does.
     */
    first_ifindex = nexthop_tab[0].ifindex;
    nexthop_addr = nexthop_tab[0].nexthop_addr;
    if (first_ifindex > 0) {
      /* found: first ifindex is non-recursive nexthop */

      if (lookup > 0) {
	/* Report non-recursive success after first lookup */
	if (PIM_DEBUG_ZEBRA) {
	  char addr_str[100];
	  pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	  zlog_debug("%s %s: lookup=%d/%d: found non-recursive ifindex=%d for address %s dist=%d met=%d",
		     __FILE__, __PRETTY_FUNCTION__,
		     lookup, max_lookup, first_ifindex, addr_str,
		     nexthop_tab[0].protocol_distance,
		     nexthop_tab[0].route_metric);
	}

	/* use last address as nexthop address */
	nexthop_tab[0].nexthop_addr.u.prefix4 = addr;

	/* report original route metric/distance */
	nexthop_tab[0].route_metric = route_metric;
	nexthop_tab[0].protocol_distance = protocol_distance;
      }

      return num_ifindex;
    }

    if (PIM_DEBUG_ZEBRA) {
      char addr_str[100];
      char nexthop_str[100];
      pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
      pim_addr_dump("<nexthop?>", &nexthop_addr, nexthop_str, sizeof(nexthop_str));
      zlog_debug("%s %s: lookup=%d/%d: zebra returned recursive nexthop %s for address %s dist=%d met=%d",
		__FILE__, __PRETTY_FUNCTION__,
		lookup, max_lookup, nexthop_str, addr_str,
		nexthop_tab[0].protocol_distance,
		nexthop_tab[0].route_metric);
    }

    addr = nexthop_addr.u.prefix4; /* use nexthop addr for recursive lookup */

  } /* for (max_lookup) */

  if (PIM_DEBUG_ZEBRA) {
    char addr_str[100];
    pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
    zlog_warn("%s %s: lookup=%d/%d: failure searching recursive nexthop ifindex for address %s",
	      __FILE__, __PRETTY_FUNCTION__,
	      lookup, max_lookup, addr_str);
  }

  return -2;
}

void
pim_zlookup_show_ip_multicast (struct vty *vty)
{
  vty_out(vty, "Zclient lookup socket: ");
  if (zlookup) {
    vty_out(vty, "%d failures=%d%s", zlookup->sock,
            zlookup->fail, VTY_NEWLINE);
  }
  else {
    vty_out(vty, "<null zclient>%s", VTY_NEWLINE);
  }
}

int
pim_zlookup_sg_statistics (struct channel_oil *c_oil)
{
  struct stream *s = zlookup->obuf;
  uint16_t command = 0;
  unsigned long long lastused;
  struct prefix_sg sg;
  int count = 0;
  int ret;
  struct interface *ifp = pim_if_find_by_vif_index (c_oil->oil.mfcc_parent);

  if (PIM_DEBUG_ZEBRA)
    zlog_debug ("Sending Request for New Channel Oil Information");

  stream_reset (s);
  zclient_create_header (s, ZEBRA_IPMR_ROUTE_STATS, VRF_DEFAULT);
  stream_put_in_addr (s, &c_oil->oil.mfcc_origin);
  stream_put_in_addr (s, &c_oil->oil.mfcc_mcastgrp);
  stream_putl (s,  ifp->ifindex);
  stream_putw_at(s, 0, stream_get_endp(s));

  count = stream_get_endp (s);
  ret = writen (zlookup->sock, s->data, count);
  if (ret <= 0)
    {
      zlog_err("%s %s: writen() failure writing to zclient lookup socket",
               __FILE__, __PRETTY_FUNCTION__);
      return -1;
    }

  s = zlookup->ibuf;

  while (command != ZEBRA_IPMR_ROUTE_STATS)
    {
      int err;
      uint16_t length = 0;
      vrf_id_t vrf_id;
      u_char marker;
      u_char version;

      stream_reset (s);
      err = zclient_read_header (s, zlookup->sock, &length, &marker, &version,
				 &vrf_id, &command);
      if (err < 0)
        {
          zlog_err ("%s %s: zclient_read_header() failed",
                 __FILE__, __PRETTY_FUNCTION__);
        zclient_lookup_failed(zlookup);
        return -1;
        }
    }

  sg.src.s_addr = stream_get_ipv4 (s);
  sg.grp.s_addr = stream_get_ipv4 (s);
  if (sg.src.s_addr != c_oil->oil.mfcc_origin.s_addr ||
      sg.grp.s_addr != c_oil->oil.mfcc_mcastgrp.s_addr)
    {
       zlog_err ("%s: Received wrong %s information",
		 __PRETTY_FUNCTION__, pim_str_sg_dump (&sg));
       zclient_lookup_failed (zlookup);
       return -3;
    }

  stream_get (&lastused, s, sizeof (lastused));

  if (PIM_DEBUG_ZEBRA)
    zlog_debug ("Received %lld for %s", lastused, pim_str_sg_dump (&sg));

  c_oil->cc.lastused = lastused;

  return 0;

}
Commit	Line	Data
12e41d03 DL	1	/*
	2	PIM for Quagga
	3	Copyright (C) 2008 Everton da Silva Marques
	4
	5	This program is free software; you can redistribute it and/or modify
	6	it under the terms of the GNU General Public License as published by
	7	the Free Software Foundation; either version 2 of the License, or
	8	(at your option) any later version.
	9
	10	This program is distributed in the hope that it will be useful, but
	11	WITHOUT ANY WARRANTY; without even the implied warranty of
	12	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	13	General Public License for more details.
	14
	15	You should have received a copy of the GNU General Public License
	16	along with this program; see the file COPYING; if not, write to the
	17	Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston,
	18	MA 02110-1301 USA
	19
12e41d03 DL	20	*/
	21
	22	#include <zebra.h>
	23	#include "zebra/rib.h"
	24
	25	#include "log.h"
	26	#include "prefix.h"
	27	#include "zclient.h"
	28	#include "stream.h"
	29	#include "network.h"
	30	#include "thread.h"
e3be0432	31	#include "prefix.h"
05b0d0d0	32	#include "vty.h"
12e41d03 DL	33
12e41d03 DL	34	#include "pimd.h"
e3be0432	35	#include "pim_iface.h"
12e41d03 DL	36	#include "pim_pim.h"
12e41d03 DL	37	#include "pim_str.h"
e3be0432	38	#include "pim_oil.h"
12e41d03 DL	39	#include "pim_zlookup.h"
	40
	41	extern int zclient_debug;
	42
05b0d0d0 DS	43	static struct zclient *zlookup = NULL;
05b0d0d0 DS	44
12e41d03 DL	45	static void zclient_lookup_sched(struct zclient *zlookup, int delay);
	46
	47	/* Connect to zebra for nexthop lookup. */
	48	static int zclient_lookup_connect(struct thread *t)
	49	{
	50	struct zclient *zlookup;
	51
	52	zlookup = THREAD_ARG(t);
	53	zlookup->t_connect = NULL;
	54
	55	if (zlookup->sock >= 0) {
	56	return 0;
	57	}
	58
	59	if (zclient_socket_connect(zlookup) < 0) {
	60	++zlookup->fail;
	61	zlog_warn("%s: failure connecting zclient socket: failures=%d",
	62	__PRETTY_FUNCTION__, zlookup->fail);
	63	}
	64	else {
	65	zlookup->fail = 0; /* reset counter on connection */
	66	}
	67
	68	zassert(!zlookup->t_connect);
	69	if (zlookup->sock < 0) {
	70	/* Since last connect failed, retry within 10 secs */
	71	zclient_lookup_sched(zlookup, 10);
	72	return -1;
	73	}
	74
	75	return 0;
	76	}
	77
	78	/* Schedule connection with delay. */
	79	static void zclient_lookup_sched(struct zclient *zlookup, int delay)
	80	{
	81	zassert(!zlookup->t_connect);
	82
	83	THREAD_TIMER_ON(master, zlookup->t_connect,
	84	zclient_lookup_connect,
	85	zlookup, delay);
	86
	87	zlog_notice("%s: zclient lookup connection scheduled for %d seconds",
	88	__PRETTY_FUNCTION__, delay);
	89	}
	90
	91	/* Schedule connection for now. */
	92	static void zclient_lookup_sched_now(struct zclient *zlookup)
	93	{
	94	zassert(!zlookup->t_connect);
	95
	96	zlookup->t_connect = thread_add_event(master, zclient_lookup_connect,
	97	zlookup, 0);
	98
	99	zlog_notice("%s: zclient lookup immediate connection scheduled",
	100	__PRETTY_FUNCTION__);
	101	}
	102
	103	/* Schedule reconnection, if needed. */
	104	static void zclient_lookup_reconnect(struct zclient *zlookup)
	105	{
	106	if (zlookup->t_connect) {
	107	return;
	108	}
109
110	zclient_lookup_sched_now(zlookup);
111	}
112
113	static void zclient_lookup_failed(struct zclient *zlookup)
114	{
115	if (zlookup->sock >= 0) {
116	if (close(zlookup->sock)) {
117	zlog_warn("%s: closing fd=%d: errno=%d %s", __func__, zlookup->sock,
118	errno, safe_strerror(errno));
119	}
120	zlookup->sock = -1;
121	}
122
123	zclient_lookup_reconnect(zlookup);
124	}
125
05b0d0d0 DS	126	void
05b0d0d0 DS	127	zclient_lookup_new (void)
12e41d03	128	{
469351b3	129	zlookup = zclient_new (master);
12e41d03 DL	130	if (!zlookup) {
	131	zlog_err("%s: zclient_new() failure",
	132	__PRETTY_FUNCTION__);
05b0d0d0	133	return;
12e41d03 DL	134	}
	135
	136	zlookup->sock = -1;
	137	zlookup->ibuf = stream_new(ZEBRA_MAX_PACKET_SIZ);
	138	zlookup->obuf = stream_new(ZEBRA_MAX_PACKET_SIZ);
	139	zlookup->t_connect = 0;
	140
	141	zclient_lookup_sched_now(zlookup);
	142
	143	zlog_notice("%s: zclient lookup socket initialized",
	144	__PRETTY_FUNCTION__);
	145
12e41d03 DL	146	}
	147
	148	static int zclient_read_nexthop(struct zclient *zlookup,
	149	struct pim_zlookup_nexthop nexthop_tab[],
	150	const int tab_size,
	151	struct in_addr addr)
	152	{
	153	int num_ifindex = 0;
	154	struct stream *s;
55119089 ND	155	const uint16_t MIN_LEN = 10; /* getipv4=4 getc=1 getl=4 getc=1 */
55119089 ND	156	uint16_t length;
12e41d03 DL	157	u_char marker;
12e41d03 DL	158	u_char version;
e7a2870b	159	vrf_id_t vrf_id;
78d3e257	160	uint16_t command = 0;
12e41d03 DL	161	struct in_addr raddr;
	162	uint8_t distance;
	163	uint32_t metric;
	164	int nexthop_num;
55119089	165	int i, err;
12e41d03 DL	166
	167	if (PIM_DEBUG_ZEBRA) {
	168	char addr_str[100];
	169	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	170	zlog_debug("%s: addr=%s",
	171	__PRETTY_FUNCTION__,
	172	addr_str);
	173	}
	174
	175	s = zlookup->ibuf;
12e41d03	176
78d3e257 DS	177	while (command != ZEBRA_IPV4_NEXTHOP_LOOKUP_MRIB)
	178	{
	179	stream_reset(s);
	180	err = zclient_read_header (s, zlookup->sock, &length, &marker, &version,
	181	&vrf_id, &command);
	182	if (err < 0) {
	183	zlog_err("%s %s: zclient_read_header() failed",
	184	__FILE__, __PRETTY_FUNCTION__);
	185	zclient_lookup_failed(zlookup);
	186	return -1;
	187	}
	188
	189	if (length < MIN_LEN) {
	190	zlog_err("%s %s: failure reading zclient lookup socket: len=%d < MIN_LEN=%d",
	191	__FILE__, __PRETTY_FUNCTION__, length, MIN_LEN);
	192	zclient_lookup_failed(zlookup);
	193	return -2;
	194	}
78d3e257	195	}
12e41d03 DL	196
	197	raddr.s_addr = stream_get_ipv4(s);
	198
	199	if (raddr.s_addr != addr.s_addr) {
	200	char addr_str[100];
	201	char raddr_str[100];
	202	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	203	pim_inet4_dump("<raddr?>", raddr, raddr_str, sizeof(raddr_str));
	204	zlog_warn("%s: address mismatch: addr=%s raddr=%s",
	205	__PRETTY_FUNCTION__,
	206	addr_str, raddr_str);
	207	/* warning only */
	208	}
	209
	210	distance = stream_getc(s);
	211	metric = stream_getl(s);
	212	nexthop_num = stream_getc(s);
	213
	214	if (nexthop_num < 1) {
	215	zlog_err("%s: socket %d bad nexthop_num=%d",
	216	__func__, zlookup->sock, nexthop_num);
	217	return -6;
	218	}
	219
12e41d03 DL	220	for (i = 0; i < nexthop_num; ++i) {
12e41d03 DL	221	enum nexthop_types_t nexthop_type;
12c7b75b	222	struct pim_neighbor *nbr;
12e41d03	223
12e41d03	224	nexthop_type = stream_getc(s);
84a35fdf DS	225	if (num_ifindex >= tab_size) {
	226	char addr_str[100];
	227	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	228	zlog_warn("%s %s: found too many nexthop ifindexes (%d > %d) for address %s",
	229	__FILE__, __PRETTY_FUNCTION__,
	230	(num_ifindex + 1), tab_size, addr_str);
	231	return num_ifindex;
	232	}
12e41d03	233	switch (nexthop_type) {
5b30316e DS	234	case NEXTHOP_TYPE_IFINDEX:
5b30316e DS	235	case NEXTHOP_TYPE_IPV4_IFINDEX:
d4a2bc11	236	case NEXTHOP_TYPE_IPV4:
ba634917	237	nexthop_tab[num_ifindex].nexthop_addr.family = AF_INET;
d4a2bc11 DS	238	if (nexthop_type == NEXTHOP_TYPE_IPV4_IFINDEX \|\|
d4a2bc11 DS	239	nexthop_type == NEXTHOP_TYPE_IPV4) {
ba634917	240	nexthop_tab[num_ifindex].nexthop_addr.u.prefix4.s_addr = stream_get_ipv4(s);
12e41d03 DL	241	}
12e41d03 DL	242	else {
ba634917	243	nexthop_tab[num_ifindex].nexthop_addr.u.prefix4.s_addr = PIM_NET_INADDR_ANY;
12e41d03 DL	244	}
	245	nexthop_tab[num_ifindex].ifindex = stream_getl(s);
	246	nexthop_tab[num_ifindex].protocol_distance = distance;
	247	nexthop_tab[num_ifindex].route_metric = metric;
	248	++num_ifindex;
	249	break;
12c7b75b DS	250	case NEXTHOP_TYPE_IPV6_IFINDEX:
	251	nexthop_tab[num_ifindex].nexthop_addr.family = AF_INET6;
	252	stream_get (&nexthop_tab[num_ifindex].nexthop_addr.u.prefix6, s, 16);
	253	nexthop_tab[num_ifindex].ifindex = stream_getl (s);
	254	nbr = pim_neighbor_find_if (if_lookup_by_index_vrf (nexthop_tab[num_ifindex].ifindex, VRF_DEFAULT));
	255	if (nbr)
	256	{
	257	nexthop_tab[num_ifindex].nexthop_addr.family = AF_INET;
	258	nexthop_tab[num_ifindex].nexthop_addr.u.prefix4 = nbr->source_addr;
	259	}
	260	++num_ifindex;
	261	break;
12e41d03 DL	262	default:
	263	/* do nothing */
	264	{
	265	char addr_str[100];
	266	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	267	zlog_warn("%s %s: found non-ifindex nexthop type=%d for address %s",
	268	__FILE__, __PRETTY_FUNCTION__,
	269	nexthop_type, addr_str);
	270	}
	271	break;
	272	}
	273	}
	274
	275	return num_ifindex;
	276	}
	277
05b0d0d0 DS	278	static int
	279	zclient_lookup_nexthop_once (struct pim_zlookup_nexthop nexthop_tab[],
	280	const int tab_size,
	281	struct in_addr addr)
12e41d03 DL	282	{
	283	struct stream *s;
	284	int ret;
	285
	286	if (PIM_DEBUG_ZEBRA) {
	287	char addr_str[100];
	288	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	289	zlog_debug("%s: addr=%s",
	290	__PRETTY_FUNCTION__,
	291	addr_str);
	292	}
	293
	294	/* Check socket. */
	295	if (zlookup->sock < 0) {
	296	zlog_err("%s %s: zclient lookup socket is not connected",
	297	__FILE__, __PRETTY_FUNCTION__);
	298	zclient_lookup_failed(zlookup);
	299	return -1;
	300	}
	301
	302	s = zlookup->obuf;
	303	stream_reset(s);
469351b3	304	zclient_create_header(s, ZEBRA_IPV4_NEXTHOP_LOOKUP_MRIB, VRF_DEFAULT);
12e41d03 DL	305	stream_put_in_addr(s, &addr);
	306	stream_putw_at(s, 0, stream_get_endp(s));
	307
	308	ret = writen(zlookup->sock, s->data, stream_get_endp(s));
	309	if (ret < 0) {
	310	zlog_err("%s %s: writen() failure writing to zclient lookup socket",
	311	__FILE__, __PRETTY_FUNCTION__);
	312	zclient_lookup_failed(zlookup);
	313	return -2;
	314	}
	315	if (ret == 0) {
	316	zlog_err("%s %s: connection closed on zclient lookup socket",
	317	__FILE__, __PRETTY_FUNCTION__);
	318	zclient_lookup_failed(zlookup);
	319	return -3;
	320	}
	321
	322	return zclient_read_nexthop(zlookup, nexthop_tab,
	323	tab_size, addr);
	324	}
	325
05b0d0d0 DS	326	int
	327	zclient_lookup_nexthop (struct pim_zlookup_nexthop nexthop_tab[],
	328	const int tab_size,
	329	struct in_addr addr,
	330	int max_lookup)
12e41d03 DL	331	{
	332	int lookup;
	333	uint32_t route_metric = 0xFFFFFFFF;
	334	uint8_t protocol_distance = 0xFF;
	335
284b4301 DS	336	qpim_nexthop_lookups++;
284b4301 DS	337
12e41d03 DL	338	for (lookup = 0; lookup < max_lookup; ++lookup) {
	339	int num_ifindex;
	340	int first_ifindex;
ba634917	341	struct prefix nexthop_addr;
12e41d03	342
05b0d0d0	343	num_ifindex = zclient_lookup_nexthop_once(nexthop_tab,
7b2c4d16	344	tab_size, addr);
d5a0a629 DS	345	if (num_ifindex < 1) {
	346	if (PIM_DEBUG_ZEBRA) {
	347	char addr_str[100];
	348	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	349	zlog_debug("%s %s: lookup=%d/%d: could not find nexthop ifindex for address %s",
	350	__FILE__, __PRETTY_FUNCTION__,
	351	lookup, max_lookup, addr_str);
	352	}
12e41d03 DL	353	return -1;
	354	}
	355
	356	if (lookup < 1) {
	357	/* this is the non-recursive lookup - save original metric/distance */
	358	route_metric = nexthop_tab[0].route_metric;
	359	protocol_distance = nexthop_tab[0].protocol_distance;
	360	}
	361
	362	/*
d8ba2f4a DS	363	* FIXME: Non-recursive nexthop ensured only for first ifindex.
	364	* However, recursive route lookup should really be fixed in zebra daemon.
	365	* See also TODO T24.
	366	*
	367	* So Zebra for NEXTHOP_TYPE_IPV4 returns the ifindex now since
	368	* it was being stored. This Doesn't solve all cases of
	369	* recursive lookup but for the most common types it does.
12e41d03 DL	370	*/
	371	first_ifindex = nexthop_tab[0].ifindex;
	372	nexthop_addr = nexthop_tab[0].nexthop_addr;
	373	if (first_ifindex > 0) {
	374	/* found: first ifindex is non-recursive nexthop */
	375
d5a0a629	376	if (lookup > 0) {
12e41d03	377	/* Report non-recursive success after first lookup */
d5a0a629 DS	378	if (PIM_DEBUG_ZEBRA) {
	379	char addr_str[100];
	380	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	381	zlog_debug("%s %s: lookup=%d/%d: found non-recursive ifindex=%d for address %s dist=%d met=%d",
	382	__FILE__, __PRETTY_FUNCTION__,
	383	lookup, max_lookup, first_ifindex, addr_str,
	384	nexthop_tab[0].protocol_distance,
	385	nexthop_tab[0].route_metric);
	386	}
12e41d03 DL	387
12e41d03 DL	388	/* use last address as nexthop address */
ba634917	389	nexthop_tab[0].nexthop_addr.u.prefix4 = addr;
12e41d03 DL	390
	391	/* report original route metric/distance */
	392	nexthop_tab[0].route_metric = route_metric;
	393	nexthop_tab[0].protocol_distance = protocol_distance;
	394	}
	395
	396	return num_ifindex;
	397	}
	398
7adf0260	399	if (PIM_DEBUG_ZEBRA) {
12e41d03 DL	400	char addr_str[100];
	401	char nexthop_str[100];
	402	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
ba634917	403	pim_addr_dump("<nexthop?>", &nexthop_addr, nexthop_str, sizeof(nexthop_str));
7adf0260	404	zlog_debug("%s %s: lookup=%d/%d: zebra returned recursive nexthop %s for address %s dist=%d met=%d",
12e41d03 DL	405	__FILE__, __PRETTY_FUNCTION__,
	406	lookup, max_lookup, nexthop_str, addr_str,
	407	nexthop_tab[0].protocol_distance,
	408	nexthop_tab[0].route_metric);
	409	}
	410
ba634917	411	addr = nexthop_addr.u.prefix4; /* use nexthop addr for recursive lookup */
12e41d03 DL	412
	413	} /* for (max_lookup) */
	414
7adf0260 DS	415	if (PIM_DEBUG_ZEBRA) {
	416	char addr_str[100];
	417	pim_inet4_dump("<addr?>", addr, addr_str, sizeof(addr_str));
	418	zlog_warn("%s %s: lookup=%d/%d: failure searching recursive nexthop ifindex for address %s",
	419	__FILE__, __PRETTY_FUNCTION__,
	420	lookup, max_lookup, addr_str);
	421	}
12e41d03 DL	422
	423	return -2;
	424	}
05b0d0d0 DS	425
	426	void
	427	pim_zlookup_show_ip_multicast (struct vty *vty)
	428	{
	429	vty_out(vty, "Zclient lookup socket: ");
	430	if (zlookup) {
	431	vty_out(vty, "%d failures=%d%s", zlookup->sock,
	432	zlookup->fail, VTY_NEWLINE);
	433	}
	434	else {
	435	vty_out(vty, "<null zclient>%s", VTY_NEWLINE);
	436	}
	437	}
e3be0432 DS	438
	439	int
	440	pim_zlookup_sg_statistics (struct channel_oil *c_oil)
	441	{
	442	struct stream *s = zlookup->obuf;
	443	uint16_t command = 0;
	444	unsigned long long lastused;
	445	struct prefix_sg sg;
	446	int count = 0;
	447	int ret;
	448	struct interface *ifp = pim_if_find_by_vif_index (c_oil->oil.mfcc_parent);
	449
	450	if (PIM_DEBUG_ZEBRA)
	451	zlog_debug ("Sending Request for New Channel Oil Information");
	452
	453	stream_reset (s);
	454	zclient_create_header (s, ZEBRA_IPMR_ROUTE_STATS, VRF_DEFAULT);
	455	stream_put_in_addr (s, &c_oil->oil.mfcc_origin);
	456	stream_put_in_addr (s, &c_oil->oil.mfcc_mcastgrp);
	457	stream_putl (s, ifp->ifindex);
	458	stream_putw_at(s, 0, stream_get_endp(s));
	459
	460	count = stream_get_endp (s);
	461	ret = writen (zlookup->sock, s->data, count);
	462	if (ret <= 0)
	463	{
	464	zlog_err("%s %s: writen() failure writing to zclient lookup socket",
	465	__FILE__, __PRETTY_FUNCTION__);
	466	return -1;
	467	}
	468
	469	s = zlookup->ibuf;
	470
	471	while (command != ZEBRA_IPMR_ROUTE_STATS)
	472	{
	473	int err;
	474	uint16_t length = 0;
	475	vrf_id_t vrf_id;
	476	u_char marker;
	477	u_char version;
	478
	479	stream_reset (s);
	480	err = zclient_read_header (s, zlookup->sock, &length, &marker, &version,
	481	&vrf_id, &command);
	482	if (err < 0)
	483	{
	484	zlog_err ("%s %s: zclient_read_header() failed",
	485	__FILE__, __PRETTY_FUNCTION__);
	486	zclient_lookup_failed(zlookup);
	487	return -1;
	488	}
	489	}
	490
	491	sg.src.s_addr = stream_get_ipv4 (s);
	492	sg.grp.s_addr = stream_get_ipv4 (s);
	493	if (sg.src.s_addr != c_oil->oil.mfcc_origin.s_addr \|\|
	494	sg.grp.s_addr != c_oil->oil.mfcc_mcastgrp.s_addr)
	495	{
	496	zlog_err ("%s: Received wrong %s information",
	497	__PRETTY_FUNCTION__, pim_str_sg_dump (&sg));
	498	zclient_lookup_failed (zlookup);
	499	return -3;
	500	}
	501
502	stream_get (&lastused, s, sizeof (lastused));
503
8d63c2d9 DS	504	if (PIM_DEBUG_ZEBRA)
8d63c2d9 DS	505	zlog_debug ("Received %lld for %s", lastused, pim_str_sg_dump (&sg));
e3be0432 DS	506
	507	c_oil->cc.lastused = lastused;
	508
	509	return 0;
	510
	511	}