File: getrecipientidfrommapped.cc

package info (click to toggle)
signalbackup-tools 20250313.1-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 3,752 kB
  • sloc: cpp: 47,042; sh: 477; ansic: 399; ruby: 19; makefile: 3
file content (88 lines) | stat: -rw-r--r-- 3,252 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/*
  Copyright (C) 2022-2025  Selwin van Dijk

  This file is part of signalbackup-tools.

  signalbackup-tools 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 3 of the License, or
  (at your option) any later version.

  signalbackup-tools 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 signalbackup-tools.  If not, see <https://www.gnu.org/licenses/>.
*/

#include "signalbackup.ih"

long long int SignalBackup::getRecipientIdFromUuidMapped(std::string const &uuid, std::map<std::string, long long int> *savedmap,
                                                         bool suppresswarning) const
{
  if (uuid.empty())
  {
    Logger::error("Asked to find recipient._id for empty uuid. Refusing");
    return -1;
  }

  if (savedmap)
    if (auto found = savedmap->find(uuid); found != savedmap->end())
      return found->second;

  // either savedmap does not exist, or uuid is not in there:
  std::string printable_uuid(makePrintable(uuid));
  SqliteDB::QueryResults res;
  if (!d_database.exec("SELECT recipient._id FROM recipient WHERE " + d_recipient_aci + " = ?1 COLLATE NOCASE OR group_id = ?1 COLLATE NOCASE", uuid, &res) ||
      res.rows() != 1 ||
      !res.valueHasType<long long int>(0, 0))
  {
    if (!suppresswarning) // we can suppress this warning, if we are creating the contact after not finding it...
      Logger::warning("Failed to find recipient for uuid: ", printable_uuid);
    return -1;
  }
  //res.prettyPrint();
  if (savedmap)
    (*savedmap)[uuid] = res.getValueAs<long long int>(0, 0);

  return res.getValueAs<long long int>(0, 0);
}

long long int SignalBackup::getRecipientIdFromPhoneMapped(std::string const &phone, std::map<std::string, long long int> *savedmap,
                                                          bool suppresswarning) const
{
  if (phone.empty())
  {
    Logger::error("Asked to find recipient._id for empty e164. Refusing");
    return -1;
  }

  if (savedmap)
    if (auto found = savedmap->find(phone); found != savedmap->end())
      return found->second;

  // either savedmap does not exist, or phone is not in there:
  std::string printable_phone(phone);
  unsigned int offset = 3;
  if (offset < phone.size()) [[likely]]
    std::replace_if(printable_phone.begin() + offset, printable_phone.end(), [](char c){ return std::isdigit(c); }, 'x');
  else
    printable_phone = "xxx";

  SqliteDB::QueryResults res;
  if (!d_database.exec("SELECT recipient._id FROM recipient WHERE " + d_recipient_e164 + " = ? COLLATE NOCASE", phone, &res) ||
      res.rows() != 1 ||
      !res.valueHasType<long long int>(0, 0))
  {
    if (!suppresswarning)
      Logger::warning("Failed to find recipient for phone: ", printable_phone);
    return -1;
  }
  //res.prettyPrint();
  if (savedmap)
    (*savedmap)[phone] = res.getValueAs<long long int>(0, 0);

  return res.getValueAs<long long int>(0, 0);
}