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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
//===--- SILDefaultWitnessTable.cpp ---------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
// This file defines the SILDefaultWitnessTable class, which is used to provide
// default implementations of protocol requirements for resilient protocols,
// allowing IRGen to generate the appropriate metadata so that the runtime can
// insert those requirements to witness tables that were emitted prior to the
// requirement being added.
//
//===----------------------------------------------------------------------===//
#include "swift/AST/ASTMangler.h"
#include "swift/Basic/Assertions.h"
#include "swift/SIL/SILDefaultWitnessTable.h"
#include "swift/SIL/SILModule.h"
#include "llvm/ADT/SmallString.h"
using namespace swift;
void SILDefaultWitnessTable::addDefaultWitnessTable() {
// Make sure we have not seen this witness table yet.
assert(Mod.DefaultWitnessTableMap.find(Protocol) ==
Mod.DefaultWitnessTableMap.end() && "Attempting to create duplicate "
"default witness table.");
Mod.DefaultWitnessTableMap[Protocol] = this;
Mod.defaultWitnessTables.push_back(this);
}
SILDefaultWitnessTable *
SILDefaultWitnessTable::create(SILModule &M, SILLinkage Linkage,
const ProtocolDecl *Protocol,
ArrayRef<SILDefaultWitnessTable::Entry> entries){
// Allocate the witness table and initialize it.
auto *buf = M.allocate<SILDefaultWitnessTable>(1);
SILDefaultWitnessTable *wt =
::new (buf) SILDefaultWitnessTable(M, Linkage, Protocol, entries);
wt->addDefaultWitnessTable();
// Return the resulting default witness table.
return wt;
}
SILDefaultWitnessTable *
SILDefaultWitnessTable::create(SILModule &M, SILLinkage Linkage,
const ProtocolDecl *Protocol) {
// Allocate the witness table and initialize it.
auto *buf = M.allocate<SILDefaultWitnessTable>(1);
SILDefaultWitnessTable *wt =
::new (buf) SILDefaultWitnessTable(M, Linkage, Protocol);
wt->addDefaultWitnessTable();
// Return the resulting default witness table.
return wt;
}
SILDefaultWitnessTable::
SILDefaultWitnessTable(SILModule &M,
SILLinkage Linkage,
const ProtocolDecl *Protocol,
ArrayRef<Entry> entries)
: Mod(M), Linkage(Linkage), Protocol(Protocol), Entries(),
IsDeclaration(true) {
convertToDefinition(entries);
}
SILDefaultWitnessTable::SILDefaultWitnessTable(SILModule &M,
SILLinkage Linkage,
const ProtocolDecl *Protocol)
: Mod(M), Linkage(Linkage), Protocol(Protocol), Entries(),
IsDeclaration(true) {}
void SILDefaultWitnessTable::
convertToDefinition(ArrayRef<Entry> entries) {
assert(IsDeclaration);
IsDeclaration = false;
Entries = Mod.allocateCopy(entries);
// Bump the reference count of witness functions referenced by this table.
for (auto entry : getEntries()) {
if (entry.isValid() && entry.getKind() == SILWitnessTable::Method) {
if (SILFunction *f = entry.getMethodWitness().Witness)
f->incrementRefCount();
}
}
}
std::string SILDefaultWitnessTable::getUniqueName() const {
Mangle::ASTMangler Mangler;
return Mangler.mangleTypeWithoutPrefix(
getProtocol()->getDeclaredInterfaceType());
}
SILDefaultWitnessTable::~SILDefaultWitnessTable() {
// Drop the reference count of witness functions referenced by this table.
for (auto entry : getEntries()) {
if (entry.isValid() && entry.getKind() == SILWitnessTable::Method) {
if (SILFunction *f = entry.getMethodWitness().Witness)
f->decrementRefCount();
}
}
}
|
