diff options
author | Mike Buland <mike@xagasoft.com> | 2024-09-26 14:43:22 -0700 |
---|---|---|
committer | Mike Buland <mike@xagasoft.com> | 2024-09-26 14:43:22 -0700 |
commit | 76821551f312dd447a03748a01670f3718cd8345 (patch) | |
tree | aca702a69c9eb3f7dd8080dcd20d7f238bbbaa81 /src/tools | |
parent | d5e966a0563e3f01c02c3cfb1f3a83fec852168b (diff) | |
download | libbu++-76821551f312dd447a03748a01670f3718cd8345.tar.gz libbu++-76821551f312dd447a03748a01670f3718cd8345.tar.bz2 libbu++-76821551f312dd447a03748a01670f3718cd8345.tar.xz libbu++-76821551f312dd447a03748a01670f3718cd8345.zip |
Basic update to new API for existing components.
This may not all work yet, but it all compiles!
Diffstat (limited to 'src/tools')
-rw-r--r-- | src/tools/myriadfs.cpp | 269 |
1 files changed, 269 insertions, 0 deletions
diff --git a/src/tools/myriadfs.cpp b/src/tools/myriadfs.cpp new file mode 100644 index 0000000..587bc89 --- /dev/null +++ b/src/tools/myriadfs.cpp | |||
@@ -0,0 +1,269 @@ | |||
1 | /* | ||
2 | * Copyright (C) 2007-2023 Xagasoft, All rights reserved. | ||
3 | * | ||
4 | * This file is part of the libbu++ library and is released under the | ||
5 | * terms of the license contained in the file LICENSE. | ||
6 | */ | ||
7 | |||
8 | #include "bu/sio.h" | ||
9 | #include "bu/streamstack.h" | ||
10 | #include "bu/file.h" | ||
11 | #include "bu/myriadfs.h" | ||
12 | #include "bu/myriadstream.h" | ||
13 | #include "bu/optparser.h" | ||
14 | |||
15 | enum Mode | ||
16 | { | ||
17 | modeList, | ||
18 | modeCat, | ||
19 | modeCopyIn, | ||
20 | modeCopyOut, | ||
21 | modeMkdir, | ||
22 | modeInitialize, | ||
23 | modeRm, | ||
24 | |||
25 | modeNone | ||
26 | }; | ||
27 | |||
28 | Bu::Formatter &operator>>( Bu::Formatter &f, Mode & /*e*/ ) | ||
29 | { | ||
30 | Bu::sio << "Uh oh, the formatter was called..." << Bu::sio.nl; | ||
31 | return f; | ||
32 | } | ||
33 | |||
34 | class Options : public Bu::OptParser | ||
35 | { | ||
36 | public: | ||
37 | Options( int argc, char *argv[] ) : | ||
38 | eMode( modeNone ), | ||
39 | iBlockSize( 64 ) | ||
40 | { | ||
41 | addHelpBanner("Options:"); | ||
42 | addOption( sFile, 'f', "file", "Myriadfs file"); | ||
43 | addOption( iBlockSize, 'b', "block-size", | ||
44 | "Specify the block size when initializing a new MyriadFs"); | ||
45 | |||
46 | setNonOption( Bu::slot( this, &Options::nonOption ) ); | ||
47 | |||
48 | addHelpOption(); | ||
49 | |||
50 | parse( argc, argv ); | ||
51 | } | ||
52 | |||
53 | int nonOption( Bu::Array<Bu::String> aParams ) | ||
54 | { | ||
55 | if( eMode == modeNone ) | ||
56 | { | ||
57 | //Bu::println("Checking mode"); | ||
58 | // First param, must be the mode | ||
59 | if( aParams[0] == "ls" ) | ||
60 | eMode = modeList; | ||
61 | else if( aParams[0] == "cat" ) | ||
62 | eMode = modeCat; | ||
63 | else if( aParams[0] == "cp-in" ) | ||
64 | eMode = modeCopyIn; | ||
65 | else if( aParams[0] == "cp-out" ) | ||
66 | eMode = modeCopyOut; | ||
67 | else if( aParams[0] == "mkdir" ) | ||
68 | eMode = modeMkdir; | ||
69 | else if( aParams[0] == "initialize" ) | ||
70 | eMode = modeInitialize; | ||
71 | else if( aParams[0] == "rm" ) | ||
72 | eMode = modeRm; | ||
73 | else | ||
74 | Bu::println("Unknown command, try --help"); | ||
75 | return 0; | ||
76 | } else { | ||
77 | lParams.append( aParams[0] ); | ||
78 | } | ||
79 | //Bu::println("Got option: \"%1\"").arg( aParams[0] ); | ||
80 | return 0; | ||
81 | } | ||
82 | |||
83 | Mode eMode; | ||
84 | Bu::String sFile; | ||
85 | Bu::StringList lParams; | ||
86 | int iBlockSize; | ||
87 | }; | ||
88 | |||
89 | int main( int argc, char *argv[] ) | ||
90 | { | ||
91 | Options opt( argc, argv ); | ||
92 | |||
93 | if( opt.sFile.isEmpty() ) | ||
94 | { | ||
95 | Bu::println("You must specify a MyriadFs stream (see -f).\n"); | ||
96 | return 1; | ||
97 | } | ||
98 | |||
99 | if( opt.eMode == modeNone ) | ||
100 | { | ||
101 | Bu::println("Specify an operation to perfrom.\n"); | ||
102 | return 1; | ||
103 | } | ||
104 | |||
105 | int iFlags = Bu::File::ReadWrite; | ||
106 | |||
107 | if( opt.eMode == modeInitialize ) | ||
108 | { | ||
109 | iFlags |= Bu::File::Create|Bu::File::Truncate; | ||
110 | } | ||
111 | |||
112 | Bu::File fFs( opt.sFile, iFlags ); | ||
113 | Bu::MyriadFs mFs( fFs, opt.iBlockSize ); | ||
114 | |||
115 | switch( opt.eMode ) | ||
116 | { | ||
117 | case modeList: | ||
118 | { | ||
119 | Bu::String sPath = "/"; | ||
120 | if( opt.lParams.getSize() > 0 ) | ||
121 | { | ||
122 | sPath = opt.lParams.first(); | ||
123 | } | ||
124 | Bu::MyriadFs::Dir lEnt = mFs.readDir( sPath ); | ||
125 | for( Bu::MyriadFs::Dir::iterator i = lEnt.begin(); i; i++ ) | ||
126 | { | ||
127 | Bu::String sPerm; | ||
128 | switch( (*i).uPerms&Bu::MyriadFs::typeMask ) | ||
129 | { | ||
130 | case Bu::MyriadFs::typeDir: sPerm += "d"; break; | ||
131 | case Bu::MyriadFs::typeChrDev: sPerm += "c"; break; | ||
132 | case Bu::MyriadFs::typeBlkDev: sPerm += "b"; break; | ||
133 | case Bu::MyriadFs::typeSymLink: sPerm += "l"; break; | ||
134 | case Bu::MyriadFs::typeSocket: sPerm += "s"; break; | ||
135 | default: sPerm += "-"; break; | ||
136 | } | ||
137 | sPerm += ((*i).uPerms&Bu::MyriadFs::permUsrR)?"r":"-"; | ||
138 | sPerm += ((*i).uPerms&Bu::MyriadFs::permUsrW)?"w":"-"; | ||
139 | sPerm += ((*i).uPerms&Bu::MyriadFs::permUsrX)?"x":"-"; | ||
140 | sPerm += ((*i).uPerms&Bu::MyriadFs::permGrpR)?"r":"-"; | ||
141 | sPerm += ((*i).uPerms&Bu::MyriadFs::permGrpW)?"w":"-"; | ||
142 | sPerm += ((*i).uPerms&Bu::MyriadFs::permGrpX)?"x":"-"; | ||
143 | sPerm += ((*i).uPerms&Bu::MyriadFs::permOthR)?"r":"-"; | ||
144 | sPerm += ((*i).uPerms&Bu::MyriadFs::permOthW)?"w":"-"; | ||
145 | sPerm += ((*i).uPerms&Bu::MyriadFs::permOthX)?"x":"-"; | ||
146 | Bu::println("%1 %2 %3:%4 %5 %6") | ||
147 | .arg( sPerm ) | ||
148 | .arg( (*i).iNode ) | ||
149 | .arg( (*i).iUser ) | ||
150 | .arg( (*i).iGroup ) | ||
151 | .arg( (*i).iSize ) | ||
152 | .arg( (*i).sName ); | ||
153 | } | ||
154 | } | ||
155 | break; | ||
156 | |||
157 | case modeCat: | ||
158 | { | ||
159 | if( opt.lParams.isEmpty() ) | ||
160 | { | ||
161 | Bu::println("Specify at least one file."); | ||
162 | return 1; | ||
163 | } | ||
164 | int iBlockSize = 1024*1024; | ||
165 | char *pBuf = new char[iBlockSize]; | ||
166 | for( Bu::StringList::iterator i = opt.lParams.begin(); i; i++ ) | ||
167 | { | ||
168 | Bu::MyriadStream ms = mFs.open( *i, Bu::MyriadFs::Read ); | ||
169 | int iRead = 0; | ||
170 | do | ||
171 | { | ||
172 | iRead = ms.read( pBuf, iBlockSize ); | ||
173 | if( iRead > 0 ) | ||
174 | { | ||
175 | Bu::sioRaw.write( pBuf, iRead ); | ||
176 | } | ||
177 | } while( iRead == iBlockSize ); | ||
178 | } | ||
179 | delete[] pBuf; | ||
180 | } | ||
181 | break; | ||
182 | |||
183 | case modeCopyIn: | ||
184 | { | ||
185 | if( opt.lParams.getSize() != 2 ) | ||
186 | { | ||
187 | Bu::println("Specify a source file and destination file."); | ||
188 | return 1; | ||
189 | } | ||
190 | int iBlockSize = 1024*1024; | ||
191 | char *pBuf = new char[iBlockSize]; | ||
192 | Bu::File fs = Bu::File( | ||
193 | opt.lParams.first(), Bu::File::Read ); | ||
194 | Bu::MyriadStream ms = mFs.open( | ||
195 | opt.lParams.last(), Bu::MyriadFs::WriteNew ); | ||
196 | int iRead = 0; | ||
197 | do | ||
198 | { | ||
199 | iRead = fs.read( pBuf, iBlockSize ); | ||
200 | if( iRead > 0 ) | ||
201 | { | ||
202 | ms.write( pBuf, iRead ); | ||
203 | } | ||
204 | } while( iRead == iBlockSize ); | ||
205 | delete[] pBuf; | ||
206 | } | ||
207 | break; | ||
208 | |||
209 | case modeCopyOut: | ||
210 | { | ||
211 | if( opt.lParams.getSize() != 2 ) | ||
212 | { | ||
213 | Bu::println("Specify a source file and destination file."); | ||
214 | return 1; | ||
215 | } | ||
216 | int iBlockSize = 1024*1024; | ||
217 | char *pBuf = new char[iBlockSize]; | ||
218 | Bu::MyriadStream ms = mFs.open( | ||
219 | opt.lParams.first(), Bu::MyriadFs::Read ); | ||
220 | Bu::File fs = Bu::File( | ||
221 | opt.lParams.last(), Bu::File::WriteNew ); | ||
222 | int iRead = 0; | ||
223 | do | ||
224 | { | ||
225 | iRead = ms.read( pBuf, iBlockSize ); | ||
226 | if( iRead > 0 ) | ||
227 | { | ||
228 | fs.write( pBuf, iRead ); | ||
229 | } | ||
230 | } while( iRead == iBlockSize ); | ||
231 | delete[] pBuf; | ||
232 | } | ||
233 | break; | ||
234 | |||
235 | case modeMkdir: | ||
236 | { | ||
237 | if( opt.lParams.isEmpty() ) | ||
238 | { | ||
239 | Bu::println("Specify at least one directory."); | ||
240 | return 1; | ||
241 | } | ||
242 | for( Bu::StringList::iterator i = opt.lParams.begin(); i; i++ ) | ||
243 | { | ||
244 | mFs.mkDir( *i, 0777 ); | ||
245 | } | ||
246 | } | ||
247 | break; | ||
248 | |||
249 | case modeInitialize: | ||
250 | Bu::println("MyriadFs initialized.\n"); | ||
251 | break; | ||
252 | |||
253 | case modeRm: | ||
254 | { | ||
255 | if( opt.lParams.getSize() != 1 ) | ||
256 | { | ||
257 | Bu::println("Specify a file path."); | ||
258 | return 1; | ||
259 | } | ||
260 | mFs.unlink( opt.lParams.first() ); | ||
261 | } | ||
262 | break; | ||
263 | |||
264 | case modeNone: | ||
265 | break; | ||
266 | } | ||
267 | |||
268 | return 0; | ||
269 | } | ||