Mercurial Hosting > luan
annotate src/goodjava/lucene/backup/BackupServer.java @ 2020:163f88f7ee00
for nabble
| author | Franklin Schmidt <fschmidt@gmail.com> | 
|---|---|
| date | Sun, 05 Oct 2025 20:43:42 -0600 | 
| parents | 2dbcc8360a3e | 
| children | 969291201e12 | 
| rev | line source | 
|---|---|
| 1499 | 1 package goodjava.lucene.backup; | 
| 2 | |
| 3 import java.io.File; | |
| 4 import java.io.IOException; | |
| 5 import java.util.Map; | |
| 1690 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 6 import java.net.InetAddress; | 
| 1499 | 7 import java.net.Socket; | 
| 8 import java.net.ServerSocket; | |
| 1690 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 9 import java.net.UnknownHostException; | 
| 1499 | 10 import java.util.concurrent.Executors; | 
| 1502 | 11 import java.util.concurrent.ExecutorService; | 
| 1509 | 12 import javax.net.ssl.SSLSocket; | 
| 1499 | 13 import javax.net.ssl.SSLServerSocket; | 
| 14 import goodjava.util.SoftCacheMap; | |
| 1501 | 15 import goodjava.io.IoUtils; | 
| 1499 | 16 import goodjava.rpc.RpcServer; | 
| 17 import goodjava.rpc.RpcCall; | |
| 1509 | 18 import goodjava.rpc.RpcClient; | 
| 19 import goodjava.rpc.RpcResult; | |
| 1512 | 20 import goodjava.rpc.Rpc; | 
| 1509 | 21 import goodjava.rpc.RpcException; | 
| 1499 | 22 import goodjava.logging.Logger; | 
| 23 import goodjava.logging.LoggerFactory; | |
| 24 | |
| 25 | |
| 1672 | 26 public final class BackupServer { | 
| 1499 | 27 private static final Logger logger = LoggerFactory.getLogger(BackupServer.class); | 
| 28 | |
| 1675 | 29 public static int port = 9102; | 
| 1499 | 30 public static String[] cipherSuites = new String[] { | 
| 31 "TLS_DH_anon_WITH_AES_128_GCM_SHA256", | |
| 32 "TLS_DH_anon_WITH_AES_128_CBC_SHA256", | |
| 33 "TLS_ECDH_anon_WITH_AES_128_CBC_SHA", | |
| 34 "TLS_DH_anon_WITH_AES_128_CBC_SHA", | |
| 35 "TLS_ECDH_anon_WITH_3DES_EDE_CBC_SHA", | |
| 36 "SSL_DH_anon_WITH_3DES_EDE_CBC_SHA", | |
| 37 "TLS_ECDH_anon_WITH_RC4_128_SHA", | |
| 38 "SSL_DH_anon_WITH_RC4_128_MD5", | |
| 39 "SSL_DH_anon_WITH_DES_CBC_SHA", | |
| 40 "SSL_DH_anon_EXPORT_WITH_DES40_CBC_SHA", | |
| 41 "SSL_DH_anon_EXPORT_WITH_RC4_40_MD5", | |
| 42 }; | |
| 43 | |
| 44 private final File backupDir; | |
| 1502 | 45 private static final ExecutorService threadPool = Executors.newCachedThreadPool(); | 
| 1499 | 46 private static final Map<String,Backup> backups = new SoftCacheMap<String,Backup>(); | 
| 47 | |
| 1501 | 48 public BackupServer(File backupDir) throws IOException { | 
| 1499 | 49 this.backupDir = backupDir; | 
| 1501 | 50 IoUtils.mkdirs(backupDir); | 
| 1499 | 51 } | 
| 52 | |
| 53 public synchronized void start() throws IOException { | |
| 54 final ServerSocket ss; | |
| 55 if( cipherSuites == null ) { | |
| 56 ss = new ServerSocket(port); | |
| 57 } else { | |
| 1506 
d80395468b4e
ssl security in code
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1502diff
changeset | 58 ss = IoUtils.getSSLServerSocketFactory().createServerSocket(port); | 
| 1499 | 59 ((SSLServerSocket)ss).setEnabledCipherSuites(cipherSuites); | 
| 60 } | |
| 61 threadPool.execute(new Runnable(){public void run() { | |
| 62 try { | |
| 63 while(!threadPool.isShutdown()) { | |
| 64 final Socket socket = ss.accept(); | |
| 65 threadPool.execute(new Runnable(){public void run() { | |
| 66 handle(socket); | |
| 67 }}); | |
| 68 } | |
| 69 } catch(IOException e) { | |
| 70 logger.error("",e); | |
| 71 } | |
| 72 }}); | |
| 73 logger.info("started server on port "+port); | |
| 74 } | |
| 75 | |
| 1690 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 76 private static String getName(RpcServer rpc,Object[] args) { | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 77 String domain = (String)args[0]; | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 78 InetAddress addr; | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 79 try { | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 80 addr = InetAddress.getByName(domain); | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 81 } catch(UnknownHostException e) { | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 82 rpc.write( new RpcException("domain lookup failed") ); | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 83 rpc.close(); | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 84 return null; | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 85 } | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 86 if( !rpc.socket.getInetAddress().equals(addr) ) { | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 87 rpc.write( new RpcException("domain doesn't match client") ); | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 88 rpc.close(); | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 89 return null; | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 90 } | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 91 String name = (String)args[1]; | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 92 return name==null ? domain : domain + "~" + name; | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 93 } | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 94 | 
| 1697 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 95 private Backup getBackup(String name) { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 96 synchronized(backups) { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 97 Backup backup = backups.get(name); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 98 if( backup == null ) { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 99 backup = new Backup( new File(backupDir,name) ); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 100 backups.put(name,backup); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 101 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 102 return backup; | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 103 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 104 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 105 | 
| 1499 | 106 private void handle(Socket socket) { | 
| 107 RpcServer rpc = new RpcServer(socket); | |
| 1674 | 108 Backup backup = null; | 
| 109 while( !rpc.isClosed() ) { | |
| 1499 | 110 RpcCall call = rpc.read(); | 
| 1674 | 111 if( call == null ) | 
| 112 break; | |
| 113 if( call.cmd.equals("exists") ) { | |
| 1690 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 114 String name = getName(rpc,call.args); | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 115 if( name==null ) return; | 
| 1674 | 116 rpc.write( new RpcResult(new Object[]{new File(backupDir,name).exists()}) ); | 
| 117 } else if( call.cmd.equals("login") ) { | |
| 1690 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 118 String name = getName(rpc,call.args); | 
| 
973d3039c421
backup server checks client domain
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1675diff
changeset | 119 if( name==null ) return; | 
| 1697 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 120 backup = getBackup(name); | 
| 1674 | 121 rpc.write(Rpc.OK); | 
| 122 } else if( backup != null ) { | |
| 123 backup.handle(rpc,call); | |
| 1697 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 124 } else if( call.cmd.equals("copy") ) { | 
| 1698 | 125 try { | 
| 126 if( !IoUtils.getInetAddresses().contains(rpc.socket.getInetAddress()) ) { | |
| 127 rpc.write( new RpcException("only localhost allowed") ); | |
| 128 rpc.close(); | |
| 129 return; | |
| 130 } | |
| 131 } catch(IOException e) { | |
| 132 throw new RuntimeException(e); | |
| 133 } | |
| 1697 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 134 String dirName = (String)call.args[0]; | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 135 copy(new File(dirName)); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 136 rpc.write(Rpc.OK); | 
| 1674 | 137 } else { | 
| 1512 | 138 rpc.write( new RpcException("login expected") ); | 
| 139 rpc.close(); | |
| 140 return; | |
| 141 } | |
| 1499 | 142 } | 
| 143 } | |
| 144 | |
| 1697 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 145 private void copy(File dir) { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 146 try { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 147 IoUtils.deleteRecursively(dir); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 148 IoUtils.mkdirs(dir); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 149 for( File f : backupDir.listFiles() ) { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 150 if( f.isDirectory() && new File(f,"index.json").exists() ) { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 151 String name = f.getName(); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 152 Backup backup = getBackup(name); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 153 backup.copyTo( new File(dir,name) ); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 154 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 155 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 156 } catch(IOException e) { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 157 throw new RuntimeException(e); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 158 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 159 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 160 | 
| 1509 | 161 | 
| 162 // for client | |
| 163 | |
| 164 public static RpcClient rpcClient(String backupDomain) throws IOException { | |
| 165 Socket socket; | |
| 166 if( BackupServer.cipherSuites == null ) { | |
| 167 socket = new Socket(backupDomain,BackupServer.port); | |
| 168 } else { | |
| 169 socket = IoUtils.getSSLSocketFactory().createSocket(backupDomain,BackupServer.port); | |
| 170 ((SSLSocket)socket).setEnabledCipherSuites(BackupServer.cipherSuites); | |
| 171 } | |
| 172 return new RpcClient(socket); | |
| 173 } | |
| 174 | |
| 1697 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 175 public static void copyBackupTo(String dirName) throws IOException, RpcException { | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 176 RpcClient rpc = BackupServer.rpcClient("localhost"); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 177 rpc.write( new RpcCall("copy",dirName) ); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 178 RpcResult result = rpc.read(); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 179 rpc.close(); | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 180 } | 
| 
aff2309ae510
add copy_backups.luan
 Franklin Schmidt <fschmidt@gmail.com> parents: 
1690diff
changeset | 181 | 
| 1499 | 182 } | 
