3 Copyright (C) 2008 Everton da Silva Marques
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.
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.
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,
35 #include "pim_iface.h"
38 #include "pim_register.h"
39 #include "pim_jp_agg.h"
41 void pim_msg_build_header(uint8_t *pim_msg
, size_t pim_msg_size
, uint8_t pim_msg_type
)
43 struct pim_msg_header
*header
= (struct pim_msg_header
*)pim_msg
;
48 header
->ver
= PIM_PROTO_VERSION
;
49 header
->type
= pim_msg_type
;
55 * The checksum for Registers is done only on the first 8 bytes of the packet,
56 * including the PIM header and the next 4 bytes, excluding the data packet portion
58 if (pim_msg_type
== PIM_MSG_TYPE_REGISTER
)
59 header
->checksum
= in_cksum (pim_msg
, PIM_MSG_REGISTER_LEN
);
61 header
->checksum
= in_cksum (pim_msg
, pim_msg_size
);
64 uint8_t *pim_msg_addr_encode_ipv4_ucast(uint8_t *buf
, struct in_addr addr
)
66 buf
[0] = PIM_MSG_ADDRESS_FAMILY_IPV4
; /* addr family */
67 buf
[1] = '\0'; /* native encoding */
68 memcpy(buf
+2, &addr
, sizeof(struct in_addr
));
70 return buf
+ PIM_ENCODED_IPV4_UCAST_SIZE
;
73 uint8_t *pim_msg_addr_encode_ipv4_group(uint8_t *buf
, struct in_addr addr
)
75 buf
[0] = PIM_MSG_ADDRESS_FAMILY_IPV4
; /* addr family */
76 buf
[1] = '\0'; /* native encoding */
77 buf
[2] = '\0'; /* reserved */
78 buf
[3] = 32; /* mask len */
79 memcpy(buf
+4, &addr
, sizeof(struct in_addr
));
81 return buf
+ PIM_ENCODED_IPV4_GROUP_SIZE
;
85 pim_msg_addr_encode_ipv4_source(uint8_t *buf
,
86 struct in_addr addr
, uint8_t bits
)
88 buf
[0] = PIM_MSG_ADDRESS_FAMILY_IPV4
; /* addr family */
89 buf
[1] = '\0'; /* native encoding */
91 buf
[3] = 32; /* mask len */
92 memcpy(buf
+4, &addr
, sizeof(struct in_addr
));
94 return buf
+ PIM_ENCODED_IPV4_SOURCE_SIZE
;
98 * For the given 'struct pim_jp_sources' list
99 * determine the size_t it would take up.
102 pim_msg_get_jp_group_size (struct list
*sources
)
106 size
+= sizeof (struct pim_encoded_group_ipv4
);
107 size
+= 4; // Joined sources (2) + Pruned Sources (2)
109 size
+= sizeof (struct pim_encoded_source_ipv4
) * sources
->count
;
115 pim_msg_build_jp_groups (struct pim_jp_groups
*grp
, struct pim_jp_agg_group
*sgs
)
117 struct listnode
*node
, *nnode
;
118 struct pim_jp_sources
*source
;
119 struct in_addr stosend
;
121 size_t size
= pim_msg_get_jp_group_size (sgs
->sources
);
124 memset (grp
, 0, size
);
125 pim_msg_addr_encode_ipv4_group ((uint8_t *)&grp
->g
, sgs
->group
);
127 for (ALL_LIST_ELEMENTS(sgs
->sources
, node
, nnode
, source
))
129 /* number of joined/pruned sources */
135 if (source
->up
->sg
.src
.s_addr
== INADDR_ANY
)
137 struct pim_rpf
*rpf
= pim_rp_g (source
->up
->sg
.grp
);
138 bits
= PIM_ENCODE_SPARSE_BIT
| PIM_ENCODE_WC_BIT
| PIM_ENCODE_RPT_BIT
;
139 stosend
= rpf
->rpf_addr
.u
.prefix4
;
143 bits
= PIM_ENCODE_SPARSE_BIT
;
144 stosend
= source
->up
->sg
.src
;
147 pim_msg_addr_encode_ipv4_source ((uint8_t *)&grp
->s
[tgroups
], stosend
, bits
);
151 grp
->joins
= htons(grp
->joins
);
152 grp
->prunes
= htons(grp
->prunes
);
154 * This is not implemented correctly at this point in time
158 if (up
->sg
.src
.s_addr
== INADDR_ANY
)
160 struct pim_upstream
*child
;
161 struct listnode
*up_node
;
164 zlog_debug ("%s: Considering (%s) children for (S,G,rpt) prune",
165 __PRETTY_FUNCTION__
, up
->sg_str
);
166 for (ALL_LIST_ELEMENTS_RO (up
->sources
, up_node
, child
))
168 if (child
->sptbit
== PIM_UPSTREAM_SPTBIT_TRUE
)
170 if (!pim_rpf_is_same(&up
->rpf
, &child
->rpf
))
173 if (PIM_DEBUG_PIM_PACKETS
)
174 zlog_debug ("%s: SPT Bit and RPF'(%s) != RPF'(S,G): Add Prune (%s,rpt) to compound message",
175 __PRETTY_FUNCTION__
, up
->sg_str
, child
->sg_str
);
178 if (PIM_DEBUG_PIM_PACKETS
)
179 zlog_debug ("%s: SPT Bit and RPF'(%s) == RPF'(S,G): Not adding Prune for (%s,rpt)",
180 __PRETTY_FUNCTION__
, up
->sg_str
, child
->sg_str
);
182 else if (pim_upstream_is_sg_rpt (child
))
184 if (pim_upstream_empty_inherited_olist (child
))
187 if (PIM_DEBUG_PIM_PACKETS
)
188 zlog_debug ("%s: inherited_olist(%s,rpt) is NULL, Add Prune to compound message",
189 __PRETTY_FUNCTION__
, child
->sg_str
);
191 else if (!pim_rpf_is_same (&up
->rpf
, &child
->rpf
))
194 if (PIM_DEBUG_PIM_PACKETS
)
195 zlog_debug ("%s: RPF'(%s) != RPF'(%s,rpt), Add Prune to compound message",
196 __PRETTY_FUNCTION__
, up
->sg_str
, child
->sg_str
);
199 if (PIM_DEBUG_PIM_PACKETS
)
200 zlog_debug ("%s: RPF'(%s) == RPF'(%s,rpt), Do not add Prune to compound message",
201 __PRETTY_FUNCTION__
, up
->sg_str
, child
->sg_str
);
204 if (PIM_DEBUG_PIM_PACKETS
)
205 zlog_debug ("%s: SPT bit is not set for (%s)",
206 __PRETTY_FUNCTION__
, child
->sg_str
);
209 pim_msg_curr
= pim_msg_addr_encode_ipv4_source (pim_msg_curr
, remain
,
211 PIM_ENCODE_SPARSE_BIT
| PIM_ENCODE_RPT_BIT
);
212 remain
= pim_msg_curr
- pim_msg
;
213 *prunes
= htons(ntohs(*prunes
) + 1);