#if HAVE_CONFIG_H
#include <config.h>
#endif

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <stdio.h>

#include "bgpq3.h"
#include "sx_report.h"

int
bgpq3_print_cisco_aspath(FILE* f, struct bgpq_expander* b)
{ 
	int nc=0, i, j;
	fprintf(f,"no ip as-path access-list %s\n", b->name?b->name:"NN");
	if(b->asn32s[0][b->asnumber/8]&(0x80>>(b->asnumber%8))) { 
		fprintf(f,"ip as-path access-list %s permit ^%i(_%i)*$\n",
			b->name?b->name:"NN",b->asnumber,b->asnumber);
	};
	for(i=0;i<8192;i++) { 
		for(j=0;j<8;j++) { 
			if(b->asn32s[0][i]&(0x80>>j)) { 
				if(i*8+j==b->asnumber) continue;
				if(!nc) { 
					fprintf(f,"ip as-path access-list %s permit"
						" ^%i(_[0-9]+)*_(%i", b->name?b->name:"NN", 
						b->asnumber,i*8+j);
				} else { 
					fprintf(f,"|%i",i*8+j);
				}
				nc++;
				if(nc==b->aswidth) { 
					fprintf(f,")$\n");
					nc=0;
				};
			};
		};
	};
	if(nc) fprintf(f,")$\n");
	return 0;
};
int
bgpq3_print_cisco_oaspath(FILE* f, struct bgpq_expander* b)
{ 
	int nc=0, i, j;
	fprintf(f,"no ip as-path access-list %s\n", b->name?b->name:"NN");
	if(b->asn32s[0][b->asnumber/8]&(0x80>>(b->asnumber%8))) { 
		fprintf(f,"ip as-path access-list %s permit ^(_%i)*$\n",
			b->name?b->name:"NN",b->asnumber);
	};
	for(i=0;i<8192;i++) { 
		for(j=0;j<8;j++) { 
			if(b->asn32s[0][i]&(0x80>>j)) { 
				if(i*8+j==b->asnumber) continue;
				if(!nc) { 
					fprintf(f,"ip as-path access-list %s permit"
						" ^(_[0-9]+)*_(%i", b->name?b->name:"NN", 
						i*8+j);
				} else { 
					fprintf(f,"|%i",i*8+j);
				}
				nc++;
				if(nc==b->aswidth) { 
					fprintf(f,")$\n");
					nc=0;
				};
			};
		};
	};
	if(nc) fprintf(f,")$\n");
	return 0;
};
		
int
bgpq3_print_juniper_aspath(FILE* f, struct bgpq_expander* b)
{ 
	int nc=0, lineNo=0, i, j, k;
	fprintf(f,"policy-options {\nreplace:\n as-path-group %s {\n", 
		b->name?b->name:"NN");

	if(b->asn32s[b->asnumber/65536] && 
		b->asn32s[b->asnumber/65535][(b->asnumber/65536)/8]&
		(0x80>>(b->asnumber%8))) { 
		fprintf(f,"  as-path a%i \"^%u(%u)*$\";\n", lineNo, b->asnumber,
			b->asnumber);
		lineNo++;
	};
	for(k=0;k<65536;k++) { 
		if(!b->asn32s[k]) continue;
		for(i=0;i<8192;i++) { 
			for(j=0;j<8;j++) { 
				if(b->asn32s[k][i]&(0x80>>j)) { 
					if(k*65536+i*8+j==b->asnumber) continue;
					if(!nc) { 
						fprintf(f,"  as-path a%i \"^%u(.)*(%u",
							lineNo,b->asnumber,k*65536+i*8+j);
					} else { 
						fprintf(f,"|%u",k*65536+i*8+j);
					};
					nc++;
					if(nc==b->aswidth) { 
						fprintf(f,")$\";\n");
						nc=0;
						lineNo++;
					};
				};
			};
		};
	};
	if(nc) fprintf(f,")$\";\n");
	fprintf(f," }\n}\n");
	return 0;
};

int
bgpq3_print_juniper_oaspath(FILE* f, struct bgpq_expander* b)
{ 
	int nc=0, lineNo=0, i, j, k;
	fprintf(f,"policy-options {\nreplace:\n as-path-group %s {\n", 
		b->name?b->name:"NN");

	if(b->asn32s[b->asnumber/65536] && 
		b->asn32s[b->asnumber/65536][(b->asnumber/65536)/8]&
		(0x80>>(b->asnumber%8))) { 
		fprintf(f,"  as-path a%i \"^%u(%u)*$\";\n", lineNo, b->asnumber,
			b->asnumber);
		lineNo++;
	};
	for(k=0;k<65536;k++) { 
		if(!b->asn32s[k]) continue;

		for(i=0;i<8192;i++) { 
			for(j=0;j<8;j++) { 
				if(b->asn32s[k][i]&(0x80>>j)) { 
					if(k*65536+i*8+j==b->asnumber) continue;
					if(!nc) { 
						fprintf(f,"  as-path a%i \"^(.)*(%u",
							lineNo,k*65536+i*8+j);
					} else { 
						fprintf(f,"|%u",k*65536+i*8+j);
					}
					nc++;
					if(nc==b->aswidth) { 
						fprintf(f,")$\";\n");
						nc=0;
						lineNo++;
					};
				};
			};
		};
	};
	if(nc) fprintf(f,")$\";\n");
	fprintf(f," }\n}\n");
	return 0;
};

int
bgpq3_print_aspath(FILE* f, struct bgpq_expander* b)
{ 
	if(b->vendor==V_JUNIPER) { 
		return bgpq3_print_juniper_aspath(f,b);
	} else if(b->vendor==V_CISCO) { 
		return bgpq3_print_cisco_aspath(f,b);
	} else { 
		sx_report(SX_FATAL,"Unknown vendor %i\n", b->vendor);
	};
	return 0;
};

int
bgpq3_print_oaspath(FILE* f, struct bgpq_expander* b)
{ 
	if(b->vendor==V_JUNIPER) { 
		return bgpq3_print_juniper_oaspath(f,b);
	} else if(b->vendor==V_CISCO) { 
		return bgpq3_print_cisco_oaspath(f,b);
	} else { 
		sx_report(SX_FATAL,"Unknown vendor %i\n", b->vendor);
	};
	return 0;
};

void
bgpq3_print_jprefix(struct sx_radix_node* n, void* ff)
{ 
	char prefix[128];
	FILE* f=(FILE*)ff;
	if(n->isGlue) return;
	if(!f) f=stdout;
	sx_prefix_snprintf(&n->prefix,prefix,sizeof(prefix));
	fprintf(f,"    %s;\n",prefix);
};

static char* bname=NULL;

void
bgpq3_print_cprefix(struct sx_radix_node* n, void* ff)
{ 
	char prefix[128];
	FILE* f=(FILE*)ff;
	if(!f) f=stdout;
	if(n->isGlue) goto checkSon;
	sx_prefix_snprintf(&n->prefix,prefix,sizeof(prefix));
	if(n->isAggregate) { 
		if(n->aggregateLow>n->prefix.masklen) { 
			fprintf(f,"%s prefix-list %s permit %s ge %u le %u\n",
				n->prefix.family==AF_INET?"ip":"ipv6",bname?bname:"NN",prefix,
				n->aggregateLow,n->aggregateHi);
		} else { 
			fprintf(f,"%s prefix-list %s permit %s le %u\n",
				n->prefix.family==AF_INET?"ip":"ipv6",bname?bname:"NN",prefix,
				n->aggregateHi);
		};
	} else { 
		fprintf(f,"%s prefix-list %s permit %s\n",
			(n->prefix.family==AF_INET)?"ip":"ipv6",bname?bname:"NN",prefix);
	};
checkSon:
	if(n->son) 
		bgpq3_print_cprefix(n->son,ff);
};

int
bgpq3_print_juniper_prefixlist(FILE* f, struct bgpq_expander* b)
{ 
	fprintf(f,"policy-options {\nreplace:\n prefix-list %s {\n",
		b->name?b->name:"NN");
	sx_radix_tree_foreach(b->tree,bgpq3_print_jprefix,f);
	fprintf(f," }\n}\n");
	return 0;
};

int
bgpq3_print_cisco_prefixlist(FILE* f, struct bgpq_expander* b)
{ 
	bname=b->name;
	fprintf(f,"no %s prefix-list %s\n",
		(b->family==AF_INET)?"ip":"ipv6",bname?bname:"NN");
	sx_radix_tree_foreach(b->tree,bgpq3_print_cprefix,f);
	return 0;
};

int
bgpq3_print_prefixlist(FILE* f, struct bgpq_expander* b)
{ 
	switch(b->vendor) { 
		case V_JUNIPER: return bgpq3_print_juniper_prefixlist(f,b);
		case V_CISCO: return bgpq3_print_cisco_prefixlist(f,b);
	};
	return 0;
};