Merge pull request #1503 from valentinewallace/2022-05-onion-msgs
[rust-lightning] / lightning / src / lib.rs
1 // This file is Copyright its original authors, visible in version control
2 // history.
3 //
4 // This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
5 // or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
6 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
7 // You may not use this file except in accordance with one or both of these
8 // licenses.
9
10 #![crate_name = "lightning"]
11
12 //! Rust-Lightning, not Rusty's Lightning!
13 //!
14 //! A full-featured but also flexible lightning implementation, in library form. This allows the
15 //! user (you) to decide how they wish to use it instead of being a fully self-contained daemon.
16 //! This means there is no built-in threading/execution environment and it's up to the user to
17 //! figure out how best to make networking happen/timers fire/things get written to disk/keys get
18 //! generated/etc. This makes it a good candidate for tight integration into an existing wallet
19 //! instead of having a rather-separate lightning appendage to a wallet.
20 //!
21 //! `default` features are:
22 //!
23 //! * `std` - enables functionalities which require `std`, including `std::io` trait implementations and things which utilize time
24 //! * `grind_signatures` - enables generation of [low-r bitcoin signatures](https://bitcoin.stackexchange.com/questions/111660/what-is-signature-grinding),
25 //! which saves 1 byte per signature in 50% of the cases (see [bitcoin PR #13666](https://github.com/bitcoin/bitcoin/pull/13666))
26 //!
27 //! Available features are:
28 //!
29 //! * `std`
30 //! * `grind_signatures`
31 //! * `no-std ` - exposes write trait implementations from the `core2` crate (at least one of `no-std` or `std` are required)
32 //! * Skip logging of messages at levels below the given log level:
33 //!     * `max_level_off`
34 //!     * `max_level_error`
35 //!     * `max_level_warn`
36 //!     * `max_level_info`
37 //!     * `max_level_debug`
38 //!     * `max_level_trace`
39
40 #![cfg_attr(not(any(test, fuzzing, feature = "_test_utils")), deny(missing_docs))]
41 #![cfg_attr(not(any(test, fuzzing, feature = "_test_utils")), forbid(unsafe_code))]
42 #![deny(broken_intra_doc_links)]
43
44 // In general, rust is absolutely horrid at supporting users doing things like,
45 // for example, compiling Rust code for real environments. Disable useless lints
46 // that don't do anything but annoy us and cant actually ever be resolved.
47 #![allow(bare_trait_objects)]
48 #![allow(ellipsis_inclusive_range_patterns)]
49
50 #![cfg_attr(docsrs, feature(doc_auto_cfg))]
51
52 #![cfg_attr(all(not(feature = "std"), not(test)), no_std)]
53
54 #![cfg_attr(all(any(test, feature = "_test_utils"), feature = "_bench_unstable"), feature(test))]
55 #[cfg(all(any(test, feature = "_test_utils"), feature = "_bench_unstable"))] extern crate test;
56
57 #[cfg(not(any(feature = "std", feature = "no-std")))]
58 compile_error!("at least one of the `std` or `no-std` features must be enabled");
59
60 #[cfg(all(fuzzing, test))]
61 compile_error!("Tests will always fail with cfg=fuzzing");
62
63 #[macro_use]
64 extern crate alloc;
65 extern crate bitcoin;
66 #[cfg(any(test, feature = "std"))]
67 extern crate core;
68
69 #[cfg(any(test, feature = "_test_utils"))] extern crate hex;
70 #[cfg(any(test, fuzzing, feature = "_test_utils"))] extern crate regex;
71
72 #[cfg(not(feature = "std"))] extern crate core2;
73
74 #[macro_use]
75 pub mod util;
76 pub mod chain;
77 pub mod ln;
78 pub mod routing;
79 #[allow(unused)]
80 mod onion_message; // To be exposed after sending/receiving OMs is supported in PeerManager.
81
82 #[cfg(feature = "std")]
83 /// Re-export of either `core2::io` or `std::io`, depending on the `std` feature flag.
84 pub use std::io;
85 #[cfg(not(feature = "std"))]
86 /// Re-export of either `core2::io` or `std::io`, depending on the `std` feature flag.
87 pub use core2::io;
88
89 #[cfg(not(feature = "std"))]
90 mod io_extras {
91         use core2::io::{self, Read, Write};
92
93         /// A writer which will move data into the void.
94         pub struct Sink {
95                 _priv: (),
96         }
97
98         /// Creates an instance of a writer which will successfully consume all data.
99         pub const fn sink() -> Sink {
100                 Sink { _priv: () }
101         }
102
103         impl core2::io::Write for Sink {
104                 #[inline]
105                 fn write(&mut self, buf: &[u8]) -> core2::io::Result<usize> {
106                         Ok(buf.len())
107                 }
108
109                 #[inline]
110                 fn flush(&mut self) -> core2::io::Result<()> {
111                         Ok(())
112                 }
113         }
114
115         pub fn copy<R: ?Sized, W: ?Sized>(reader: &mut R, writer: &mut W) -> Result<u64, io::Error>
116                 where
117                 R: Read,
118                 W: Write,
119         {
120                 let mut count = 0;
121                 let mut buf = [0u8; 64];
122
123                 loop {
124                         match reader.read(&mut buf) {
125                                 Ok(0) => break,
126                                 Ok(n) => { writer.write_all(&buf[0..n])?; count += n as u64; },
127                                 Err(ref e) if e.kind() == io::ErrorKind::Interrupted => {},
128                                 Err(e) => return Err(e.into()),
129                         };
130                 }
131                 Ok(count)
132         }
133
134         pub fn read_to_end<D: io::Read>(mut d: D) -> Result<alloc::vec::Vec<u8>, io::Error> {
135                 let mut result = vec![];
136                 let mut buf = [0u8; 64];
137                 loop {
138                         match d.read(&mut buf) {
139                                 Ok(0) => break,
140                                 Ok(n) => result.extend_from_slice(&buf[0..n]),
141                                 Err(ref e) if e.kind() == io::ErrorKind::Interrupted => {},
142                                 Err(e) => return Err(e.into()),
143                         };
144                 }
145                 Ok(result)
146         }
147 }
148
149 #[cfg(feature = "std")]
150 mod io_extras {
151         pub fn read_to_end<D: ::std::io::Read>(mut d: D) -> Result<Vec<u8>, ::std::io::Error> {
152                 let mut buf = Vec::new();
153                 d.read_to_end(&mut buf)?;
154                 Ok(buf)
155         }
156
157         pub use std::io::{copy, sink};
158 }
159
160 mod prelude {
161         #[cfg(feature = "hashbrown")]
162         extern crate hashbrown;
163
164         pub use alloc::{vec, vec::Vec, string::String, collections::VecDeque, boxed::Box};
165         #[cfg(not(feature = "hashbrown"))]
166         pub use std::collections::{HashMap, HashSet, hash_map};
167         #[cfg(feature = "hashbrown")]
168         pub use self::hashbrown::{HashMap, HashSet, hash_map};
169
170         pub use alloc::borrow::ToOwned;
171         pub use alloc::string::ToString;
172 }
173
174 #[cfg(all(feature = "std", test))]
175 mod debug_sync;
176 #[cfg(all(feature = "backtrace", feature = "std", test))]
177 extern crate backtrace;
178
179 #[cfg(feature = "std")]
180 mod sync {
181         #[cfg(test)]
182         pub use debug_sync::*;
183         #[cfg(not(test))]
184         pub use ::std::sync::{Arc, Mutex, Condvar, MutexGuard, RwLock, RwLockReadGuard, RwLockWriteGuard};
185         #[cfg(not(test))]
186         pub use crate::util::fairrwlock::FairRwLock;
187 }
188
189 #[cfg(not(feature = "std"))]
190 mod sync;