示例#1
0
def retrieval(exactkw, trapdoor_set, key):
    results = []
    with open('D:\\Fuzzy Keywords Search\\encrypted_index.csv', 'r') as f:
        reader = csv.DictReader(f)
        for item in reader:
            kw = AES.decrypt(item['Keyword'], key)
            if kw == exactkw:
                de_fid = AES.decrypt(item['FIDS'], key)
                de_fid = de_fid.split('\x00')[0]
                de_fid = de_fid.split('|')
                for i in de_fid:
                    if i != 'NULL':
                        results.append(i)
                break
            else:
                ba = bitarray(item['BloomFilter'])
                fliter = BloomFilter(capacity=100)
                fliter.bitarray = ba
                flag = False
                for i in trapdoor_set:
                    if i in fliter:
                        flag = True
                if flag:
                    de_fid = AES.decrypt(item['FIDS'], key)
                    de_fid = de_fid.split('\x00')[0]
                    de_fid = de_fid.split('|')
                    for i in de_fid:
                        if i != 'NULL':
                            results.append(i)

    if results:
        # 列表结果去重
        temp_list = []  # 定义一个临时空列表,用于保存临时数据。
        for i in results:  # 遍历原列表,判断如果元素不在临时列表,就追加进去,如果在,就不加。
            if i not in temp_list:
                temp_list.append(i)
        return temp_list
    else:
        return '检索失败'
示例#2
0
def search(client_socket):
    while True:
        # 接受检索的关键字及陷门集合
        exact_kw = client_socket.recv(1024).decode('utf-8')
        if exact_kw == 'exit':
            break
        trapdoor_string = client_socket.recv(1024 * 1024).decode("utf-8")
        trapdoor_list = trapdoor_string.split(' ')

        # 进行关键字检索
        print("检索中......")
        results = []
        with open('D:\\Fuzzy Keywords Search\\Sever\\encrypted_index.csv',
                  'r') as f:
            reader = csv.DictReader(f)
            for item in reader:
                if item['Keyword'] == exact_kw:
                    print('关键字检索成功...')
                    results.append(item['FIDS'])
                    break
                else:
                    ba = bitarray(item['BloomFilter'])
                    fliter = BloomFilter(capacity=100)
                    fliter.bitarray = ba
                    flag = False
                    for i in trapdoor_list:
                        if i in fliter:
                            flag = True
                    if flag:
                        results.append(item['FIDS'])

        if not results:
            print('检索失败')
            client_socket.send('wrong'.encode())
            continue
        else:
            results_string = ' '.join(results)
            client_socket.send(results_string.encode())
            print('关键字检索成功...')