/*
  * Copyright (c) 2016 MediaTek Inc.  All rights reserved.
  *
  * This software is available to you under a choice of one of two
  * licenses.  You may choose to be licensed under the terms of the GNU
  * General Public License (GPL) Version 2, available from the file
  * COPYING in the main directory of this source tree, or the
  * BSD license below:
  *
  *     Redistribution and use in source and binary forms, with or
  *     without modification, are permitted provided that the following
  *     conditions are met:
  *
  *      - Redistributions of source code must retain the above
  *        copyright notice, this list of conditions and the following
  *        disclaimer.
  *
  *      - Redistributions in binary form must reproduce the above
  *        copyright notice, this list of conditions and the following
  *        disclaimer in the documentation and/or other materials
  *        provided with the distribution.
  *
  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  * SOFTWARE.
  */


#ifndef __RTMP_IGMP_SNOOP_H__
#define __RTMP_IGMP_SNOOP_H__

#include "link_list.h"

#define IGMP_PROTOCOL_DESCRIPTOR	0x02
#define IGMP_V1_MEMBERSHIP_REPORT	0x12
#define IGMP_V2_MEMBERSHIP_REPORT	0x16
#define IGMP_LEAVE_GROUP			0x17
#define IGMP_V3_MEMBERSHIP_REPORT	0x22

#define MLD_V1_LISTENER_REPORT		131
#define MLD_V1_LISTENER_DONE		132
#define MLD_V2_LISTERNER_REPORT		143

#define IGMPMAC_TB_ENTRY_AGEOUT_TIME (120 * OS_HZ)

#define MULTICAST_ADDR_HASH_INDEX(Addr)      (MAC_ADDR_HASH(Addr) & (MAX_LEN_OF_MULTICAST_FILTER_HASH_TABLE - 1))

#define IS_MULTICAST_MAC_ADDR(Addr)			((((Addr[0]) & 0x01) == 0x01) && ((Addr[0]) != 0xff))
#define IS_IPV6_MULTICAST_MAC_ADDR(Addr)	((((Addr[0]) & 0x01) == 0x01) && ((Addr[0]) == 0x33))
#define IS_BROADCAST_MAC_ADDR(Addr)			((((Addr[0]) & 0xff) == 0xff))

#define IGMP_NONE		0
#define IGMP_PKT		1
#define IGMP_IN_GROUP	2

VOID MulticastFilterTableInit(IN PRTMP_ADAPTER pAd,
			      IN PMULTICAST_FILTER_TABLE * ppMulticastFilterTable);

VOID MultiCastFilterTableReset(IN PMULTICAST_FILTER_TABLE * ppMulticastFilterTable);

BOOLEAN MulticastFilterTableInsertEntry(IN PRTMP_ADAPTER pAd,
					IN PUCHAR pGrpId,
					IN PUCHAR pMemberAddr,
					IN PNET_DEV dev, IN MulticastFilterEntryType type);

BOOLEAN MulticastFilterTableDeleteEntry(IN PRTMP_ADAPTER pAd,
					IN PUCHAR pGrpId, IN PUCHAR pMemberAddr, IN PNET_DEV dev);

PMULTICAST_FILTER_TABLE_ENTRY MulticastFilterTableLookup(IN PMULTICAST_FILTER_TABLE
							 pMulticastFilterTable, IN PUCHAR pAddr,
							 IN PNET_DEV dev);

BOOLEAN isIgmpPkt(IN PUCHAR pDstMacAddr, IN PUCHAR pIpHeader);

VOID IGMPSnooping(IN PRTMP_ADAPTER pAd,
		  IN PUCHAR pDstMacAddr,
		  IN PUCHAR pSrcMacAddr, IN PUCHAR pIpHeader, IN PNET_DEV pDev);

BOOLEAN isMldPkt(IN PUCHAR pDstMacAddr,
		 IN PUCHAR pIpHeader, OUT UINT8 * pProtoType, OUT PUCHAR * pMldHeader);

BOOLEAN IPv6MulticastFilterExcluded(IN PUCHAR pDstMacAddr, IN PUCHAR pIpHeader);

VOID MLDSnooping(IN PRTMP_ADAPTER pAd,
		 IN PUCHAR pDstMacAddr,
		 IN PUCHAR pSrcMacAddr, IN PUCHAR pIpHeader, IN PNET_DEV pDev);

UCHAR IgmpMemberCnt(IN PLIST_HEADER pList);

VOID IgmpGroupDelMembers(IN PRTMP_ADAPTER pAd, IN PUCHAR pMemberAddr, IN PNET_DEV pDev);

INT Set_IgmpSn_Enable_Proc(IN PRTMP_ADAPTER pAd, IN PSTRING arg);

INT Set_IgmpSn_AddEntry_Proc(IN PRTMP_ADAPTER pAd, IN PSTRING arg);

INT Set_IgmpSn_DelEntry_Proc(IN PRTMP_ADAPTER pAd, IN PSTRING arg);

INT Set_IgmpSn_TabDisplay_Proc(IN PRTMP_ADAPTER pAd, IN PSTRING arg);

void rtmp_read_igmp_snoop_from_file(IN PRTMP_ADAPTER pAd, PSTRING tmpbuf, PSTRING buffer);

NDIS_STATUS IgmpPktInfoQuery(IN PRTMP_ADAPTER pAd,
			     IN PUCHAR pSrcBufVA,
			     IN PNDIS_PACKET pPacket,
			     IN struct wifi_dev *wdev,
			     OUT INT *pInIgmpGroup,
			     OUT PMULTICAST_FILTER_TABLE_ENTRY *ppGroupEntry);

NDIS_STATUS IgmpPktClone(IN PRTMP_ADAPTER pAd,
			 IN PNDIS_PACKET pPacket,
			 IN INT IgmpPktInGroup,
			 IN PMULTICAST_FILTER_TABLE_ENTRY pGroupEntry,
			 IN UCHAR QueIdx, IN UINT8 UserPriority, IN PNET_DEV pNetDev);

#endif /* __RTMP_IGMP_SNOOP_H__ */
