1 | <?php |
||
22 | class EcAdapter implements EcAdapterInterface |
||
23 | { |
||
24 | /** |
||
25 | * @var Math |
||
26 | */ |
||
27 | private $math; |
||
28 | |||
29 | /** |
||
30 | * @var GeneratorPoint |
||
31 | */ |
||
32 | private $generator; |
||
33 | |||
34 | /** |
||
35 | * @param Math $math |
||
36 | * @param GeneratorPoint $generator |
||
37 | */ |
||
38 | 13 | public function __construct(Math $math, GeneratorPoint $generator) |
|
43 | |||
44 | /** |
||
45 | * @return Math |
||
46 | */ |
||
47 | 321 | public function getMath() |
|
51 | |||
52 | /** |
||
53 | * @return GeneratorPoint |
||
54 | */ |
||
55 | 160 | public function getGenerator() |
|
59 | |||
60 | /** |
||
61 | * @param \GMP $scalar |
||
62 | * @param bool|false $compressed |
||
63 | * @return PrivateKey |
||
64 | */ |
||
65 | 61 | public function getPrivateKey(\GMP $scalar, $compressed = false) |
|
69 | |||
70 | /** |
||
71 | * @param PointInterface $point |
||
72 | * @param bool|false $compressed |
||
73 | * @return PublicKey |
||
74 | */ |
||
75 | 71 | public function getPublicKey(PointInterface $point, $compressed = false) |
|
79 | |||
80 | /** |
||
81 | * @param \GMP $r |
||
82 | * @param \GMP $s |
||
83 | * @return Signature |
||
84 | */ |
||
85 | public function getSignature(\GMP $r, \GMP $s) |
||
89 | |||
90 | /** |
||
91 | * @param BufferInterface $messageHash |
||
92 | * @param PublicKey $publicKey |
||
93 | * @param Signature $signature |
||
94 | * @return bool |
||
95 | */ |
||
96 | 98 | private function doVerify(BufferInterface $messageHash, PublicKey $publicKey, Signature $signature) |
|
102 | |||
103 | /** |
||
104 | * @param BufferInterface $messageHash |
||
105 | * @param PublicKeyInterface $publicKey |
||
106 | * @param SignatureInterface $signature |
||
107 | * @return bool |
||
108 | */ |
||
109 | 98 | public function verify(BufferInterface $messageHash, PublicKeyInterface $publicKey, SignatureInterface $signature) |
|
115 | |||
116 | /** |
||
117 | * @param BufferInterface $messageHash |
||
118 | * @param PrivateKey $privateKey |
||
119 | * @param RbgInterface|null $rbg |
||
120 | * @return Signature |
||
121 | */ |
||
122 | 34 | private function doSign(BufferInterface $messageHash, PrivateKey $privateKey, RbgInterface $rbg = null) |
|
123 | { |
||
124 | 34 | $rbg = $rbg ?: new Rfc6979($this, $privateKey, $messageHash); |
|
125 | 34 | $randomK = gmp_init($rbg->bytes(32)->getHex(), 16); |
|
126 | 34 | $hash = gmp_init($messageHash->getHex(), 16); |
|
127 | |||
128 | 34 | $signer = new Signer($this->math); |
|
129 | 34 | $signature = $signer->sign($privateKey, $hash, $randomK); |
|
130 | 34 | $s = $signature->getS(); |
|
131 | |||
132 | // if s is less than half the curve order, invert s |
||
133 | 34 | if (!$this->validateSignatureElement($s, true)) { |
|
134 | 16 | $s = $this->math->sub($this->generator->getOrder(), $s); |
|
135 | } |
||
136 | |||
137 | 34 | return new Signature($this, $signature->getR(), $s); |
|
138 | } |
||
139 | |||
140 | /** |
||
141 | * @param BufferInterface $messageHash |
||
142 | * @param PrivateKeyInterface $privateKey |
||
143 | * @param RbgInterface $rbg |
||
144 | * @return SignatureInterface |
||
145 | * @throws \BitWasp\Bitcoin\Exceptions\RandomBytesFailure |
||
146 | */ |
||
147 | 34 | public function sign(BufferInterface $messageHash, PrivateKeyInterface $privateKey, RbgInterface $rbg = null) |
|
148 | { |
||
149 | /** @var PrivateKey $privateKey */ |
||
150 | 34 | return $this->doSign($messageHash, $privateKey, $rbg); |
|
151 | } |
||
152 | |||
153 | /** |
||
154 | * @param BufferInterface $messageHash |
||
155 | * @param CompactSignatureInterface $signature |
||
156 | * @return PublicKey |
||
157 | * @throws \Exception |
||
158 | */ |
||
159 | 8 | public function recover(BufferInterface $messageHash, CompactSignatureInterface $signature) |
|
160 | { |
||
161 | 8 | $math = $this->getMath(); |
|
162 | 8 | $G = $this->getGenerator(); |
|
163 | |||
164 | 8 | $zero = gmp_init(0); |
|
165 | 8 | $one = gmp_init(1); |
|
166 | |||
167 | 8 | $r = $signature->getR(); |
|
168 | 8 | $s = $signature->getS(); |
|
169 | 8 | $recGMP = gmp_init($signature->getRecoveryId(), 10); |
|
170 | 8 | $isYEven = $math->cmp($math->bitwiseAnd($recGMP, $one), $zero) !== 0; |
|
171 | 8 | $isSecondKey = $math->cmp($math->bitwiseAnd($recGMP, gmp_init(2)), $zero) !== 0; |
|
172 | 8 | $curve = $G->getCurve(); |
|
173 | |||
174 | // Precalculate (p + 1) / 4 where p is the field order |
||
175 | 8 | $p_over_four = $math->div($math->add($curve->getPrime(), $one), gmp_init(4)); |
|
176 | |||
177 | // 1.1 Compute x |
||
178 | 8 | if (!$isSecondKey) { |
|
179 | 8 | $x = $r; |
|
180 | } else { |
||
181 | 2 | $x = $math->add($r, $G->getOrder()); |
|
182 | } |
||
183 | |||
184 | // 1.3 Convert x to point |
||
185 | 8 | $alpha = $math->mod($math->add($math->add($math->pow($x, 3), $math->mul($curve->getA(), $x)), $curve->getB()), $curve->getPrime()); |
|
186 | 8 | $beta = $math->powmod($alpha, $p_over_four, $curve->getPrime()); |
|
187 | |||
188 | // If beta is even, but y isn't or vice versa, then convert it, |
||
189 | // otherwise we're done and y=beta. |
||
190 | 8 | if ($math->isEven($beta) === $isYEven) { |
|
191 | 7 | $y = $math->sub($curve->getPrime(), $beta); |
|
192 | } else { |
||
193 | 7 | $y = $beta; |
|
194 | } |
||
195 | |||
196 | // 1.4 Check that nR is at infinity (implicitly done in constructor) |
||
197 | 8 | $R = $G->getCurve()->getPoint($x, $y); |
|
198 | |||
199 | 8 | $point_negate = function (PointInterface $p) use ($math, $G) { |
|
200 | 8 | return $G->getCurve()->getPoint($p->getX(), $math->mul($p->getY(), gmp_init('-1', 10))); |
|
201 | 8 | }; |
|
202 | |||
203 | // 1.6.1 Compute a candidate public key Q = r^-1 (sR - eG) |
||
204 | 8 | $rInv = $math->inverseMod($r, $G->getOrder()); |
|
205 | 8 | $eGNeg = $point_negate($G->mul($messageHash->getGmp())); |
|
206 | 8 | $Q = $R->mul($s)->add($eGNeg)->mul($rInv); |
|
207 | |||
208 | // 1.6.2 Test Q as a public key |
||
209 | 8 | $Qk = new PublicKey($this, $Q, $signature->isCompressed()); |
|
210 | 8 | if ($this->verify($messageHash, $Qk, $signature->convert())) { |
|
211 | 6 | return $Qk; |
|
212 | } |
||
213 | |||
214 | 2 | throw new \Exception('Unable to recover public key'); |
|
215 | } |
||
216 | |||
217 | /** |
||
218 | * Attempt to calculate the public key recovery param by trial and error |
||
219 | * |
||
220 | * @param \GMP $r |
||
221 | * @param \GMP $s |
||
222 | * @param BufferInterface $messageHash |
||
223 | * @param PublicKey $publicKey |
||
224 | * @return int |
||
225 | * @throws \Exception |
||
226 | */ |
||
227 | 7 | public function calcPubKeyRecoveryParam(\GMP $r, \GMP $s, BufferInterface $messageHash, PublicKey $publicKey) |
|
228 | { |
||
229 | 7 | $Q = $publicKey->getPoint(); |
|
230 | 7 | for ($i = 0; $i < 4; $i++) { |
|
231 | try { |
||
232 | 7 | $recover = $this->recover($messageHash, new CompactSignature($this, $r, $s, $i, $publicKey->isCompressed())); |
|
233 | 5 | if ($Q->equals($recover->getPoint())) { |
|
234 | 5 | return $i; |
|
235 | } |
||
236 | 2 | } catch (\Exception $e) { |
|
237 | 2 | continue; |
|
238 | } |
||
239 | } |
||
240 | |||
241 | 2 | throw new \Exception('Failed to find valid recovery factor'); |
|
242 | } |
||
243 | |||
244 | /** |
||
245 | * @param BufferInterface $messageHash |
||
246 | * @param PrivateKey $privateKey |
||
247 | * @param RbgInterface|null $rbg |
||
248 | * @return CompactSignature |
||
249 | * @throws \Exception |
||
250 | */ |
||
251 | 5 | private function doSignCompact(BufferInterface $messageHash, PrivateKey $privateKey, RbgInterface $rbg = null) |
|
252 | { |
||
253 | 5 | $sign = $this->sign($messageHash, $privateKey, $rbg); |
|
254 | |||
255 | // calculate the recovery param |
||
256 | // there should be a way to get this when signing too, but idk how ... |
||
257 | 5 | return new CompactSignature( |
|
258 | 5 | $this, |
|
259 | 5 | $sign->getR(), |
|
260 | 5 | $sign->getS(), |
|
261 | 5 | $this->calcPubKeyRecoveryParam($sign->getR(), $sign->getS(), $messageHash, $privateKey->getPublicKey()), |
|
262 | 5 | $privateKey->isCompressed() |
|
263 | ); |
||
264 | } |
||
265 | |||
266 | /** |
||
267 | * @param PrivateKeyInterface $privateKey |
||
268 | * @param BufferInterface $messageHash |
||
269 | * @param RbgInterface $rbg |
||
270 | * @return CompactSignature |
||
271 | */ |
||
272 | 5 | public function signCompact(BufferInterface $messageHash, PrivateKeyInterface $privateKey, RbgInterface $rbg = null) |
|
277 | |||
278 | /** |
||
279 | * @param BufferInterface $privateKey |
||
280 | * @return bool |
||
281 | */ |
||
282 | 63 | public function validatePrivateKey(BufferInterface $privateKey) |
|
288 | |||
289 | /** |
||
290 | * @param \GMP $element |
||
291 | * @param bool $half |
||
292 | * @return bool |
||
293 | */ |
||
294 | 37 | public function validateSignatureElement(\GMP $element, $half = false) |
|
304 | |||
305 | /** |
||
306 | * @param BufferInterface $publicKey |
||
307 | * @return PublicKeyInterface |
||
308 | * @throws \Exception |
||
309 | */ |
||
310 | 159 | public function publicKeyFromBuffer(BufferInterface $publicKey) |
|
341 | } |
||
342 |
This check looks for parameters that are defined as one type in their type hint or doc comment but seem to be used as a narrower type, i.e an implementation of an interface or a subclass.
Consider changing the type of the parameter or doing an instanceof check before assuming your parameter is of the expected type.