Compare commits
27 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| e2a246220e | |||
| 98d6656683 | |||
| c307953fce | |||
| 68f541f00b | |||
| 29c71f1ca9 | |||
| 76ddbdd3b4 | |||
| cae1026f35 | |||
| d07c934ea7 | |||
| d829c18e76 | |||
| e41ed80d05 | |||
| 3e691b61c0 | |||
| 0fb7a9e93d | |||
| 1b4205b04a | |||
| 8e00c8b4a0 | |||
| f3ca98be80 | |||
| 96fc8bfc7b | |||
| 58098aae0f | |||
| c4ad797905 | |||
| a61d8c5c4f | |||
| af579f8311 | |||
| fbdcdbf7ec | |||
| e6870a08d6 | |||
| fbee0ca0e8 | |||
| 65b42475a6 | |||
| af09b763a6 | |||
| 5680c6c554 | |||
| 2caaebfc33 |
@@ -8,11 +8,13 @@ Syncthing devices in the same way a client-server file sharing app accesses its
|
||||
|
||||
This is a client-oriented implementation, designed to work online by downloading and
|
||||
uploading files from an active device on the network (instead of synchronizing a local copy of
|
||||
the entire repository). This is quite different from the way the [syncthing-android][2] works,
|
||||
the entire repository).
|
||||
Due to that, you will see a sync progress of 0% at other devices (and this is expected).
|
||||
This is quite different from the way the [syncthing-android][2] works,
|
||||
and it's useful for those devices that cannot or do not wish to download the entire repository (for
|
||||
example, mobile devices with limited storage available, wishing to access a syncthing share).
|
||||
|
||||
This project is based on [syncthing-java][3], a java implementation of Syncthing protocols.
|
||||
This project is based on syncthing-java (which is in this repository too), a java implementation of Syncthing protocols.
|
||||
|
||||
Due to the behaviour of this App and the [behaviour of the Syncthing Server](https://github.com/syncthing/syncthing/issues/5224),
|
||||
you can't reconnect for some minutes if the App was killed (due to removing from the recent App list) or the connection was interrupted.
|
||||
@@ -24,6 +26,7 @@ This does not apply to local discovery connections.
|
||||
## Translations
|
||||
|
||||
The project is translated on [Transifex](https://www.transifex.com/syncthing/syncthing-lite/).
|
||||
Requests for new languages are always accepted (but this happens manually because there is no option to accept it automatically).
|
||||
|
||||
## Building
|
||||
|
||||
|
||||
@@ -0,0 +1,8 @@
|
||||
# Releasing
|
||||
|
||||
- do tests
|
||||
- update translations using ``tx pull -a -af`` (as extra merge request or branch for the case it does not build correctly)
|
||||
- update the version name and version code of the app
|
||||
- update the changelog at [app/src/main/play/en-GB/whatsnew](https://github.com/syncthing/syncthing-lite/blob/master/app/src/main/play/en-GB/whatsnew)
|
||||
- create a tag/ release in GitHub with an changelog; The tag name should be the version number
|
||||
- F-Droid picks up the release by the tag; additonally, the tag triggers a CI build which uploads the generated APK to Google Play
|
||||
+10
-5
@@ -19,8 +19,8 @@ android {
|
||||
applicationId "net.syncthing.lite"
|
||||
minSdkVersion 21
|
||||
targetSdkVersion 26
|
||||
versionCode 12
|
||||
versionName "0.3.2"
|
||||
versionCode 15
|
||||
versionName "0.3.5"
|
||||
multiDexEnabled true
|
||||
playAccountConfig = playAccountConfigs.defaultAccountConfig
|
||||
}
|
||||
@@ -60,11 +60,18 @@ android {
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
play {
|
||||
jsonFile = file(System.getenv("SYNCTHING_RELEASE_PLAY_ACCOUNT_CONFIG_FILE") ?: 'keys.json')
|
||||
uploadImages = true
|
||||
track = 'production'
|
||||
}
|
||||
|
||||
dependencies {
|
||||
implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
|
||||
implementation "org.jetbrains.anko:anko-commons:$anko_version"
|
||||
implementation "org.jetbrains.anko:anko-coroutines:$anko_version"
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:0.30.2'
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-android:1.0.0'
|
||||
implementation "com.android.support:design:$support_version"
|
||||
implementation "com.android.support:preference-v14:$support_version"
|
||||
implementation "com.android.support:support-v4:$support_version"
|
||||
@@ -82,11 +89,9 @@ dependencies {
|
||||
*/
|
||||
implementation(project(':syncthing-client')) {
|
||||
exclude group: 'commons-logging', module: 'commons-logging'
|
||||
exclude group: 'org.apache.httpcomponents', module: 'httpclient'
|
||||
exclude group: 'org.slf4j'
|
||||
exclude group: 'ch.qos.logback'
|
||||
}
|
||||
implementation 'org.apache.httpcomponents:httpclient-android:4.3.5.1'
|
||||
|
||||
implementation 'sk.baka.slf4j:slf4j-handroid:1.7.26'
|
||||
implementation 'com.google.zxing:android-integration:3.3.0'
|
||||
|
||||
@@ -4,6 +4,7 @@
|
||||
<uses-permission android:name="android.permission.INTERNET"/>
|
||||
|
||||
<application
|
||||
android:name=".android.Application"
|
||||
android:allowBackup="true"
|
||||
android:icon="@mipmap/ic_launcher"
|
||||
android:label="@string/app_name"
|
||||
@@ -22,14 +23,10 @@
|
||||
<activity android:name=".activities.FolderBrowserActivity"
|
||||
android:parentActivityName=".activities.MainActivity"/>
|
||||
<provider
|
||||
android:name="android.support.v4.content.FileProvider"
|
||||
android:name=".library.CacheFileProvider"
|
||||
android:authorities="net.syncthing.lite.fileprovider"
|
||||
android:grantUriPermissions="true"
|
||||
android:exported="false">
|
||||
<meta-data
|
||||
android:name="android.support.FILE_PROVIDER_PATHS"
|
||||
android:resource="@xml/filepaths" />
|
||||
</provider>
|
||||
android:exported="false" />
|
||||
<provider
|
||||
android:name=".library.SyncthingProvider"
|
||||
android:authorities="net.syncthing.lite.documents"
|
||||
|
||||
@@ -5,8 +5,10 @@ import android.content.Intent
|
||||
import android.databinding.DataBindingUtil
|
||||
import android.os.Bundle
|
||||
import android.util.Log
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.CompletableDeferred
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.bep.IndexBrowser
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
import net.syncthing.java.core.beans.FolderInfo
|
||||
@@ -16,10 +18,11 @@ import net.syncthing.lite.R
|
||||
import net.syncthing.lite.adapters.FolderContentsAdapter
|
||||
import net.syncthing.lite.adapters.FolderContentsListener
|
||||
import net.syncthing.lite.databinding.ActivityFolderBrowserBinding
|
||||
import net.syncthing.lite.dialogs.FileMenuDialogFragment
|
||||
import net.syncthing.lite.dialogs.FileUploadDialog
|
||||
import net.syncthing.lite.dialogs.ReconnectIssueDialogFragment
|
||||
import net.syncthing.lite.dialogs.downloadfile.DownloadFileDialogFragment
|
||||
import org.jetbrains.anko.custom.async
|
||||
import org.jetbrains.anko.doAsync
|
||||
|
||||
class FolderBrowserActivity : SyncthingActivity() {
|
||||
|
||||
@@ -44,9 +47,19 @@ class FolderBrowserActivity : SyncthingActivity() {
|
||||
override fun onItemClicked(fileInfo: FileInfo) {
|
||||
navigateToFolder(fileInfo)
|
||||
}
|
||||
|
||||
override fun onItemLongClicked(fileInfo: FileInfo): Boolean {
|
||||
return if (fileInfo.type == FileInfo.FileType.FILE) {
|
||||
FileMenuDialogFragment.newInstance(fileInfo).show(supportFragmentManager)
|
||||
|
||||
true
|
||||
} else {
|
||||
false
|
||||
}
|
||||
}
|
||||
}
|
||||
val folder = intent.getStringExtra(EXTRA_FOLDER_NAME)
|
||||
libraryHandler?.syncthingClient {
|
||||
libraryHandler.syncthingClient {
|
||||
indexBrowser = it.indexHandler.newIndexBrowser(folder, true, true)
|
||||
indexBrowser.setOnFolderChangedListener(this::onFolderChanged)
|
||||
}
|
||||
@@ -69,14 +82,16 @@ class FolderBrowserActivity : SyncthingActivity() {
|
||||
|
||||
override fun onActivityResult(requestCode: Int, resultCode: Int, intent: Intent?) {
|
||||
if (requestCode == REQUEST_SELECT_UPLOAD_FILE && resultCode == Activity.RESULT_OK) {
|
||||
libraryHandler?.syncthingClient { syncthingClient ->
|
||||
async (UI) {
|
||||
libraryHandler.syncthingClient { syncthingClient ->
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
// FIXME: it would be better if the dialog would use the library handler
|
||||
FileUploadDialog(this@FolderBrowserActivity, syncthingClient, intent!!.data,
|
||||
indexBrowser.folder, indexBrowser.currentPath,
|
||||
{ showFolderListView(indexBrowser.currentPath) }).show()
|
||||
}
|
||||
}
|
||||
} else {
|
||||
super.onActivityResult(requestCode, resultCode, intent)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -91,7 +106,7 @@ class FolderBrowserActivity : SyncthingActivity() {
|
||||
finish()
|
||||
} else {
|
||||
if (fileInfo.isDirectory()) {
|
||||
async {
|
||||
doAsync {
|
||||
indexBrowser.navigateTo(fileInfo)
|
||||
}
|
||||
|
||||
@@ -108,32 +123,33 @@ class FolderBrowserActivity : SyncthingActivity() {
|
||||
}
|
||||
|
||||
private fun onFolderChanged() {
|
||||
runOnUiThread {
|
||||
binding.isLoading = false
|
||||
GlobalScope.launch {
|
||||
val list = indexBrowser.listFiles()
|
||||
|
||||
async {
|
||||
val list = indexBrowser.listFiles()
|
||||
Log.i("navigateToFolder", "list for path = '" + indexBrowser.currentPath + "' list = " + list.size + " records")
|
||||
Log.d("navigateToFolder", "list for path = '" + indexBrowser.currentPath + "' list = " + list)
|
||||
assert(!list.isEmpty())//list must contain at least the 'parent' path
|
||||
|
||||
async (UI) {
|
||||
Log.i("navigateToFolder", "list for path = '" + indexBrowser.currentPath + "' list = " + list.size + " records")
|
||||
Log.d("navigateToFolder", "list for path = '" + indexBrowser.currentPath + "' list = " + list)
|
||||
assert(!list.isEmpty())//list must contain at least the 'parent' path
|
||||
adapter.data = list
|
||||
binding.listView.scrollToPosition(0)
|
||||
if (indexBrowser.isRoot())
|
||||
libraryHandler?.folderBrowser {
|
||||
val title = it.getFolderInfo(indexBrowser.folder)?.label
|
||||
val title = if (indexBrowser.isRoot()) {
|
||||
val result = CompletableDeferred<String?>()
|
||||
|
||||
async(UI) {
|
||||
supportActionBar?.title = title
|
||||
}
|
||||
}
|
||||
else
|
||||
supportActionBar?.title = indexBrowser.currentPathInfo().fileName
|
||||
libraryHandler.folderBrowser {
|
||||
result.complete(it.getFolderInfo(indexBrowser.folder)?.label)
|
||||
}
|
||||
|
||||
result.await()
|
||||
} else {
|
||||
indexBrowser.currentPathInfo().fileName
|
||||
}
|
||||
|
||||
runOnUiThread {
|
||||
binding.isLoading = false
|
||||
adapter.data = list
|
||||
binding.listView.scrollToPosition(0)
|
||||
supportActionBar?.title = title
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun updateFolderListView() {
|
||||
showFolderListView(indexBrowser.currentPath)
|
||||
|
||||
@@ -4,8 +4,6 @@ import android.arch.lifecycle.Observer
|
||||
import android.content.Intent
|
||||
import android.databinding.DataBindingUtil
|
||||
import android.os.Bundle
|
||||
import android.os.Handler
|
||||
import android.os.Looper
|
||||
import android.support.v4.app.Fragment
|
||||
import android.support.v4.content.ContextCompat
|
||||
import android.text.Html
|
||||
@@ -15,8 +13,9 @@ import android.view.ViewGroup
|
||||
import android.widget.Button
|
||||
import com.github.paolorotolo.appintro.AppIntro
|
||||
import com.google.zxing.integration.android.IntentIntegrator
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.lite.R
|
||||
import net.syncthing.lite.databinding.FragmentIntroOneBinding
|
||||
@@ -101,18 +100,18 @@ class IntroActivity : AppIntro() {
|
||||
|
||||
override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View {
|
||||
binding = DataBindingUtil.inflate(inflater, R.layout.fragment_intro_two, container, false)
|
||||
binding.enterDeviceId!!.scanQrCode.setOnClickListener {
|
||||
binding.enterDeviceId.scanQrCode.setOnClickListener {
|
||||
FragmentIntentIntegrator(this@IntroFragmentTwo).initiateScan()
|
||||
}
|
||||
binding.enterDeviceId!!.scanQrCode.setImageResource(R.drawable.ic_qr_code_white_24dp)
|
||||
binding.enterDeviceId.scanQrCode.setImageResource(R.drawable.ic_qr_code_white_24dp)
|
||||
return binding.root
|
||||
}
|
||||
|
||||
override fun onActivityResult(requestCode: Int, resultCode: Int, intent: Intent?) {
|
||||
val scanResult = IntentIntegrator.parseActivityResult(requestCode, resultCode, intent)
|
||||
if (scanResult?.contents != null && scanResult.contents.isNotBlank()) {
|
||||
binding.enterDeviceId!!.deviceId.setText(scanResult.contents)
|
||||
binding.enterDeviceId!!.deviceIdHolder.isErrorEnabled = false
|
||||
binding.enterDeviceId.deviceId.setText(scanResult.contents)
|
||||
binding.enterDeviceId.deviceIdHolder.isErrorEnabled = false
|
||||
}
|
||||
}
|
||||
|
||||
@@ -122,11 +121,11 @@ class IntroActivity : AppIntro() {
|
||||
*/
|
||||
fun isDeviceIdValid(): Boolean {
|
||||
return try {
|
||||
val deviceId = binding.enterDeviceId!!.deviceId.text.toString()
|
||||
val deviceId = binding.enterDeviceId.deviceId.text.toString()
|
||||
Util.importDeviceId(libraryHandler, context, deviceId, { })
|
||||
true
|
||||
} catch (e: IOException) {
|
||||
binding.enterDeviceId!!.deviceId.error = getString(R.string.invalid_device_id)
|
||||
binding.enterDeviceId.deviceId.error = getString(R.string.invalid_device_id)
|
||||
false
|
||||
}
|
||||
}
|
||||
@@ -183,7 +182,7 @@ class IntroActivity : AppIntro() {
|
||||
binding = DataBindingUtil.inflate(inflater, R.layout.fragment_intro_three, container, false)
|
||||
|
||||
libraryHandler.library { config, client, _ ->
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
client.addOnConnectionChangedListener(this@IntroFragmentThree::onConnectionChanged)
|
||||
val deviceId = config.localDeviceId.deviceId
|
||||
val desc = activity?.getString(R.string.intro_page_three_description, "<b>$deviceId</b>")
|
||||
@@ -196,7 +195,7 @@ class IntroActivity : AppIntro() {
|
||||
|
||||
private fun onConnectionChanged(deviceId: DeviceId) {
|
||||
libraryHandler.library { config, client, _ ->
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
if (config.folders.isNotEmpty()) {
|
||||
client.removeOnConnectionChangedListener(this@IntroFragmentThree::onConnectionChanged)
|
||||
(activity as IntroActivity?)?.onDonePressed(this@IntroFragmentThree)
|
||||
|
||||
@@ -8,8 +8,9 @@ import android.support.v4.app.Fragment
|
||||
import android.support.v7.app.ActionBarDrawerToggle
|
||||
import android.view.Gravity
|
||||
import android.view.MenuItem
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.lite.R
|
||||
import net.syncthing.lite.databinding.ActivityMainBinding
|
||||
import net.syncthing.lite.dialogs.DeviceIdDialogFragment
|
||||
@@ -101,7 +102,7 @@ class MainActivity : SyncthingActivity() {
|
||||
}
|
||||
|
||||
private fun cleanCacheAndIndex() {
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
libraryHandler.syncthingClient { it.clearCacheAndIndex() }
|
||||
recreate()
|
||||
}
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
package net.syncthing.lite.activities
|
||||
|
||||
import android.app.AlertDialog
|
||||
import android.content.Context
|
||||
import android.databinding.DataBindingUtil
|
||||
import android.os.Bundle
|
||||
import android.support.design.widget.Snackbar
|
||||
|
||||
@@ -50,6 +50,10 @@ class FolderContentsAdapter: RecyclerView.Adapter<FolderContentsViewHolder>() {
|
||||
listener?.onItemClicked(fileInfo)
|
||||
}
|
||||
|
||||
binding.root.setOnLongClickListener {
|
||||
listener?.onItemLongClicked(fileInfo) ?: false
|
||||
}
|
||||
|
||||
binding.executePendingBindings()
|
||||
}
|
||||
|
||||
@@ -59,6 +63,7 @@ class FolderContentsAdapter: RecyclerView.Adapter<FolderContentsViewHolder>() {
|
||||
|
||||
interface FolderContentsListener {
|
||||
fun onItemClicked(fileInfo: FileInfo)
|
||||
fun onItemLongClicked(fileInfo: FileInfo): Boolean
|
||||
}
|
||||
|
||||
class FolderContentsViewHolder(val binding: ListviewFileBinding): RecyclerView.ViewHolder(binding.root)
|
||||
class FolderContentsViewHolder(val binding: ListviewFileBinding): RecyclerView.ViewHolder(binding.root)
|
||||
|
||||
@@ -0,0 +1,52 @@
|
||||
package net.syncthing.lite.android
|
||||
|
||||
import android.app.Application
|
||||
import android.content.ClipData
|
||||
import android.content.ClipboardManager
|
||||
import android.content.Context
|
||||
import android.util.Log
|
||||
import net.syncthing.lite.BuildConfig
|
||||
import org.jetbrains.anko.defaultSharedPreferences
|
||||
import java.io.PrintWriter
|
||||
import java.io.StringWriter
|
||||
|
||||
class Application: Application() {
|
||||
companion object {
|
||||
private const val LOG_TAG = "Application"
|
||||
private const val PREF_ENABLE_CRASH_HANDLER = "crash_handler"
|
||||
}
|
||||
|
||||
override fun onCreate() {
|
||||
super.onCreate()
|
||||
|
||||
val clipboard = getSystemService(Context.CLIPBOARD_SERVICE) as ClipboardManager
|
||||
val defaultHandler = Thread.getDefaultUncaughtExceptionHandler()
|
||||
|
||||
if (defaultHandler == null) {
|
||||
Log.w(LOG_TAG, "could not get default crash handler")
|
||||
}
|
||||
|
||||
Thread.setDefaultUncaughtExceptionHandler { thread, ex ->
|
||||
Log.w(LOG_TAG, "app crashed", ex)
|
||||
|
||||
val enableCustomCrashHandling = defaultSharedPreferences.getBoolean(PREF_ENABLE_CRASH_HANDLER, false)
|
||||
|
||||
if (enableCustomCrashHandling) {
|
||||
clipboard.primaryClip = ClipData.newPlainText(
|
||||
"stacktrace",
|
||||
StringWriter().apply {
|
||||
append("Version: ").append(BuildConfig.VERSION_NAME).append('\n')
|
||||
append(Log.getStackTraceString(ex)).append('\n')
|
||||
ex.printStackTrace(PrintWriter(this))
|
||||
}.buffer.toString()
|
||||
)
|
||||
}
|
||||
|
||||
if (defaultHandler != null) {
|
||||
defaultHandler.uncaughtException(thread, ex)
|
||||
} else {
|
||||
System.exit(1)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -17,8 +17,9 @@ import android.widget.Toast
|
||||
import com.google.zxing.BarcodeFormat
|
||||
import com.google.zxing.WriterException
|
||||
import com.google.zxing.qrcode.QRCodeWriter
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.lite.R
|
||||
import net.syncthing.lite.databinding.DialogDeviceIdBinding
|
||||
import net.syncthing.lite.fragments.SyncthingDialogFragment
|
||||
@@ -62,7 +63,7 @@ class DeviceIdDialogFragment: SyncthingDialogFragment() {
|
||||
))
|
||||
}
|
||||
|
||||
async (UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
binding.deviceId.text = deviceId.deviceId
|
||||
binding.deviceId.visibility = View.VISIBLE
|
||||
|
||||
@@ -83,7 +84,7 @@ class DeviceIdDialogFragment: SyncthingDialogFragment() {
|
||||
}
|
||||
}
|
||||
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
binding.flipper.displayedChild = 1
|
||||
binding.qrCode.setImageBitmap(bmp)
|
||||
}
|
||||
|
||||
@@ -0,0 +1,79 @@
|
||||
package net.syncthing.lite.dialogs
|
||||
|
||||
import android.app.Activity
|
||||
import android.content.Intent
|
||||
import android.os.Bundle
|
||||
import android.support.design.widget.BottomSheetDialogFragment
|
||||
import android.support.v4.app.FragmentManager
|
||||
import android.view.LayoutInflater
|
||||
import android.view.View
|
||||
import android.view.ViewGroup
|
||||
import android.webkit.MimeTypeMap
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
import net.syncthing.lite.databinding.DialogFileBinding
|
||||
import net.syncthing.lite.dialogs.downloadfile.DownloadFileDialogFragment
|
||||
import net.syncthing.lite.dialogs.downloadfile.DownloadFileSpec
|
||||
import org.apache.commons.io.FilenameUtils
|
||||
|
||||
class FileMenuDialogFragment: BottomSheetDialogFragment() {
|
||||
companion object {
|
||||
private const val ARG_FILE_SPEC = "file spec"
|
||||
private const val TAG = "DownloadFileDialog"
|
||||
private const val REQ_SAVE_AS = 1
|
||||
|
||||
fun newInstance(fileInfo: FileInfo) = newInstance(DownloadFileSpec(
|
||||
folder = fileInfo.folder,
|
||||
path = fileInfo.path,
|
||||
fileName = fileInfo.fileName
|
||||
))
|
||||
|
||||
fun newInstance(fileSpec: DownloadFileSpec) = FileMenuDialogFragment().apply {
|
||||
arguments = Bundle().apply {
|
||||
putSerializable(ARG_FILE_SPEC, fileSpec)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
val fileSpec: DownloadFileSpec by lazy {
|
||||
arguments!!.getSerializable(ARG_FILE_SPEC) as DownloadFileSpec
|
||||
}
|
||||
|
||||
override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View? {
|
||||
val binding = DialogFileBinding.inflate(inflater, container, false)
|
||||
|
||||
binding.filename = fileSpec.fileName
|
||||
|
||||
binding.saveAsButton.setOnClickListener {
|
||||
startActivityForResult(
|
||||
Intent(Intent.ACTION_CREATE_DOCUMENT).apply {
|
||||
addCategory(Intent.CATEGORY_OPENABLE)
|
||||
|
||||
type = MimeTypeMap.getSingleton().getMimeTypeFromExtension(
|
||||
FilenameUtils.getExtension(fileSpec.fileName)
|
||||
)
|
||||
|
||||
putExtra(Intent.EXTRA_TITLE, fileSpec.fileName)
|
||||
},
|
||||
REQ_SAVE_AS
|
||||
)
|
||||
}
|
||||
|
||||
return binding.root
|
||||
}
|
||||
|
||||
override fun onActivityResult(requestCode: Int, resultCode: Int, data: Intent?) {
|
||||
when (requestCode) {
|
||||
REQ_SAVE_AS -> {
|
||||
if (resultCode == Activity.RESULT_OK) {
|
||||
DownloadFileDialogFragment.newInstance(fileSpec, data!!.data!!).show(fragmentManager)
|
||||
dismiss()
|
||||
}
|
||||
}
|
||||
else -> super.onActivityResult(requestCode, resultCode, data)
|
||||
}
|
||||
}
|
||||
|
||||
fun show(fragmentManager: FragmentManager?) {
|
||||
show(fragmentManager, TAG)
|
||||
}
|
||||
}
|
||||
@@ -3,8 +3,6 @@ package net.syncthing.lite.dialogs
|
||||
import android.app.ProgressDialog
|
||||
import android.content.Context
|
||||
import android.net.Uri
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import net.syncthing.java.bep.BlockPusher
|
||||
import net.syncthing.java.client.SyncthingClient
|
||||
import net.syncthing.lite.R
|
||||
|
||||
+42
-18
@@ -7,15 +7,16 @@ import android.arch.lifecycle.ViewModelProviders
|
||||
import android.content.ActivityNotFoundException
|
||||
import android.content.DialogInterface
|
||||
import android.content.Intent
|
||||
import android.net.Uri
|
||||
import android.os.Bundle
|
||||
import android.support.v4.app.DialogFragment
|
||||
import android.support.v4.app.FragmentManager
|
||||
import android.support.v4.content.FileProvider
|
||||
import android.util.Log
|
||||
import android.webkit.MimeTypeMap
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
import net.syncthing.lite.BuildConfig
|
||||
import net.syncthing.lite.R
|
||||
import net.syncthing.lite.library.CacheFileProviderUrl
|
||||
import net.syncthing.lite.library.LibraryHandler
|
||||
import org.apache.commons.io.FilenameUtils
|
||||
import org.jetbrains.anko.newTask
|
||||
@@ -24,6 +25,7 @@ import org.jetbrains.anko.toast
|
||||
class DownloadFileDialogFragment : DialogFragment() {
|
||||
companion object {
|
||||
private const val ARG_FILE_SPEC = "file spec"
|
||||
private const val ARG_SAVE_AS_URI = "save as"
|
||||
private const val TAG = "DownloadFileDialog"
|
||||
|
||||
fun newInstance(fileInfo: FileInfo) = newInstance(DownloadFileSpec(
|
||||
@@ -32,9 +34,16 @@ class DownloadFileDialogFragment : DialogFragment() {
|
||||
fileName = fileInfo.fileName
|
||||
))
|
||||
|
||||
fun newInstance(fileSpec: DownloadFileSpec) = DownloadFileDialogFragment().apply {
|
||||
fun newInstance(
|
||||
fileSpec: DownloadFileSpec,
|
||||
outputUri: Uri? = null
|
||||
) = DownloadFileDialogFragment().apply {
|
||||
arguments = Bundle().apply {
|
||||
putSerializable(ARG_FILE_SPEC, fileSpec)
|
||||
|
||||
if (outputUri != null) {
|
||||
putParcelable(ARG_SAVE_AS_URI, outputUri)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -45,11 +54,17 @@ class DownloadFileDialogFragment : DialogFragment() {
|
||||
|
||||
override fun onCreateDialog(savedInstanceState: Bundle?): Dialog {
|
||||
val fileSpec = arguments!!.getSerializable(ARG_FILE_SPEC) as DownloadFileSpec
|
||||
val outputUri = if (arguments!!.containsKey(ARG_SAVE_AS_URI))
|
||||
arguments!!.getParcelable(ARG_SAVE_AS_URI) as Uri
|
||||
else
|
||||
null
|
||||
|
||||
model.init(
|
||||
libraryHandler = LibraryHandler(context!!),
|
||||
fileSpec = fileSpec,
|
||||
externalCacheDir = context!!.externalCacheDir
|
||||
externalCacheDir = context!!.externalCacheDir,
|
||||
outputUri = outputUri,
|
||||
contentResolver = context!!.contentResolver
|
||||
)
|
||||
|
||||
val progressDialog = ProgressDialog(context).apply {
|
||||
@@ -73,22 +88,31 @@ class DownloadFileDialogFragment : DialogFragment() {
|
||||
is DownloadFileStatusDone -> {
|
||||
dismissAllowingStateLoss()
|
||||
|
||||
try {
|
||||
context!!.startActivity(
|
||||
Intent(Intent.ACTION_VIEW)
|
||||
.setDataAndType(
|
||||
FileProvider.getUriForFile(context!!, "net.syncthing.lite.fileprovider", status.file),
|
||||
MimeTypeMap.getSingleton().getMimeTypeFromExtension(FilenameUtils.getExtension(fileSpec.fileName))
|
||||
)
|
||||
.newTask()
|
||||
.addFlags(Intent.FLAG_GRANT_READ_URI_PERMISSION)
|
||||
)
|
||||
} catch (e: ActivityNotFoundException) {
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.w(TAG, "No handler found for file " + status.file.name, e)
|
||||
}
|
||||
if (outputUri == null) {
|
||||
val mimeType = MimeTypeMap.getSingleton().getMimeTypeFromExtension(FilenameUtils.getExtension(fileSpec.fileName))
|
||||
|
||||
context!!.toast(R.string.toast_open_file_failed)
|
||||
try {
|
||||
context!!.startActivity(
|
||||
Intent(Intent.ACTION_VIEW)
|
||||
.setDataAndType(
|
||||
CacheFileProviderUrl.fromFile(
|
||||
filename = fileSpec.fileName,
|
||||
mimeType = mimeType,
|
||||
file = status.file,
|
||||
context = context!!
|
||||
).serialized,
|
||||
mimeType
|
||||
)
|
||||
.newTask()
|
||||
.addFlags(Intent.FLAG_GRANT_READ_URI_PERMISSION)
|
||||
)
|
||||
} catch (e: ActivityNotFoundException) {
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.w(TAG, "No handler found for file " + status.file.name, e)
|
||||
}
|
||||
|
||||
context!!.toast(R.string.toast_open_file_failed)
|
||||
}
|
||||
}
|
||||
}
|
||||
is DownloadFileStatusFailed -> {
|
||||
|
||||
+32
-5
@@ -2,12 +2,17 @@ package net.syncthing.lite.dialogs.downloadfile
|
||||
|
||||
import android.arch.lifecycle.LiveData
|
||||
import android.arch.lifecycle.MutableLiveData
|
||||
import android.arch.lifecycle.ViewModel;
|
||||
import android.arch.lifecycle.ViewModel
|
||||
import android.content.ContentResolver
|
||||
import android.net.Uri
|
||||
import android.support.v4.os.CancellationSignal
|
||||
import android.util.Log
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.lite.BuildConfig
|
||||
import net.syncthing.lite.library.DownloadFileTask
|
||||
import net.syncthing.lite.library.LibraryHandler
|
||||
import org.apache.commons.io.FileUtils
|
||||
import java.io.File
|
||||
|
||||
class DownloadFileDialogViewModel : ViewModel() {
|
||||
@@ -20,7 +25,13 @@ class DownloadFileDialogViewModel : ViewModel() {
|
||||
private val cancellationSignal = CancellationSignal()
|
||||
val status: LiveData<DownloadFileStatus> = statusInternal
|
||||
|
||||
fun init(libraryHandler: LibraryHandler, fileSpec: DownloadFileSpec, externalCacheDir: File) {
|
||||
fun init(
|
||||
libraryHandler: LibraryHandler,
|
||||
fileSpec: DownloadFileSpec,
|
||||
externalCacheDir: File,
|
||||
outputUri: Uri?,
|
||||
contentResolver: ContentResolver
|
||||
) {
|
||||
if (isInitialized) {
|
||||
return
|
||||
}
|
||||
@@ -54,10 +65,26 @@ class DownloadFileDialogViewModel : ViewModel() {
|
||||
statusInternal.value = DownloadFileStatusRunning(newProgress)
|
||||
}
|
||||
},
|
||||
onComplete = {
|
||||
statusInternal.value = DownloadFileStatusDone(it)
|
||||
|
||||
onComplete = { file ->
|
||||
libraryHandler.stop()
|
||||
|
||||
GlobalScope.launch {
|
||||
try {
|
||||
if (outputUri != null) {
|
||||
contentResolver.openOutputStream(outputUri).use { outputStream ->
|
||||
FileUtils.copyFile(file, outputStream)
|
||||
}
|
||||
}
|
||||
|
||||
statusInternal.postValue(DownloadFileStatusDone(file))
|
||||
} catch (ex: Exception) {
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.w(TAG, "downloading file failed", ex)
|
||||
}
|
||||
|
||||
statusInternal.postValue(DownloadFileStatusFailed)
|
||||
}
|
||||
}
|
||||
},
|
||||
onError = {
|
||||
statusInternal.value = DownloadFileStatusFailed
|
||||
|
||||
@@ -10,8 +10,9 @@ import android.view.View
|
||||
import android.view.ViewGroup
|
||||
import android.view.inputmethod.InputMethodManager
|
||||
import com.google.zxing.integration.android.IntentIntegrator
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.core.beans.DeviceInfo
|
||||
import net.syncthing.lite.R
|
||||
import net.syncthing.lite.adapters.DeviceAdapterListener
|
||||
@@ -38,12 +39,12 @@ class DevicesFragment : SyncthingFragment() {
|
||||
|
||||
override fun onResume() {
|
||||
super.onResume()
|
||||
libraryHandler?.syncthingClient { it.addOnConnectionChangedListener { _ -> updateDeviceList() } }
|
||||
libraryHandler.syncthingClient { it.addOnConnectionChangedListener { _ -> updateDeviceList() } }
|
||||
}
|
||||
|
||||
override fun onPause() {
|
||||
super.onPause()
|
||||
libraryHandler?.syncthingClient { it.removeOnConnectionChangedListener{ _ -> updateDeviceList() } }
|
||||
libraryHandler.syncthingClient { it.removeOnConnectionChangedListener{ _ -> updateDeviceList() } }
|
||||
}
|
||||
|
||||
override fun onLibraryLoaded() {
|
||||
@@ -60,10 +61,12 @@ class DevicesFragment : SyncthingFragment() {
|
||||
.setTitle(getString(R.string.remove_device_title, deviceInfo.name))
|
||||
.setMessage(getString(R.string.remove_device_message, deviceInfo.deviceId.deviceId.substring(0, 7)))
|
||||
.setPositiveButton(android.R.string.yes) { _, _ ->
|
||||
libraryHandler?.configuration { config ->
|
||||
libraryHandler.library { config, syncthingClient, _ ->
|
||||
config.peers = config.peers.filterNot { it.deviceId == deviceInfo.deviceId }.toSet()
|
||||
config.persistLater()
|
||||
updateDeviceList()
|
||||
|
||||
syncthingClient.disconnectFromRemovedDevices()
|
||||
}
|
||||
}
|
||||
.setNegativeButton(android.R.string.no, null)
|
||||
@@ -76,7 +79,7 @@ class DevicesFragment : SyncthingFragment() {
|
||||
|
||||
private fun updateDeviceList() {
|
||||
libraryHandler.syncthingClient { syncthingClient ->
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
adapter.data = syncthingClient.getPeerStatus()
|
||||
binding.isEmpty = adapter.data.isEmpty()
|
||||
}
|
||||
|
||||
@@ -6,8 +6,9 @@ import android.util.Log
|
||||
import android.view.LayoutInflater
|
||||
import android.view.View
|
||||
import android.view.ViewGroup
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.core.beans.FolderInfo
|
||||
import net.syncthing.java.core.beans.FolderStats
|
||||
import net.syncthing.lite.activities.FolderBrowserActivity
|
||||
@@ -39,7 +40,7 @@ class FoldersFragment : SyncthingFragment() {
|
||||
libraryHandler.folderBrowser { folderBrowser ->
|
||||
val list = folderBrowser.folderInfoAndStatsList()
|
||||
|
||||
async (UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
Log.i(TAG, "list folders = " + list + " (" + list.size + " records)")
|
||||
val adapter = FoldersListAdapter().apply { data = list }
|
||||
binding.list.adapter = adapter
|
||||
|
||||
@@ -18,9 +18,9 @@ class SettingsFragment : PreferenceFragmentCompat() {
|
||||
val versionName = activity.packageManager.getPackageInfo(activity.packageName, 0)?.versionName
|
||||
appVersion.summary = versionName
|
||||
|
||||
activity.libraryHandler?.configuration { localDeviceName.text = it.localDeviceName }
|
||||
activity.libraryHandler.configuration { localDeviceName.text = it.localDeviceName }
|
||||
localDeviceName.setOnPreferenceChangeListener { _, _ ->
|
||||
activity.libraryHandler?.configuration { conf ->
|
||||
activity.libraryHandler.configuration { conf ->
|
||||
conf.localDeviceName = localDeviceName.text
|
||||
conf.persistLater()
|
||||
}
|
||||
@@ -28,4 +28,4 @@ class SettingsFragment : PreferenceFragmentCompat() {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,105 @@
|
||||
package net.syncthing.lite.library
|
||||
|
||||
import android.content.ContentProvider
|
||||
import android.content.ContentValues
|
||||
import android.content.Context
|
||||
import android.database.Cursor
|
||||
import android.database.MatrixCursor
|
||||
import android.net.Uri
|
||||
import android.os.ParcelFileDescriptor
|
||||
import android.provider.OpenableColumns
|
||||
import java.io.File
|
||||
import java.io.IOException
|
||||
|
||||
class CacheFileProvider: ContentProvider() {
|
||||
companion object {
|
||||
const val AUTHORITY = "net.syncthing.lite.fileprovider"
|
||||
}
|
||||
|
||||
override fun onCreate() = true
|
||||
|
||||
override fun insert(uri: Uri?, values: ContentValues?): Uri {
|
||||
throw NotImplementedError()
|
||||
}
|
||||
|
||||
override fun update(uri: Uri?, values: ContentValues?, selection: String?, selectionArgs: Array<out String>?): Int {
|
||||
throw NotImplementedError()
|
||||
}
|
||||
|
||||
override fun delete(uri: Uri?, selection: String?, selectionArgs: Array<out String>?): Int {
|
||||
throw NotImplementedError()
|
||||
}
|
||||
|
||||
override fun query(uri: Uri, projection: Array<out String>?, selection: String?, selectionArgs: Array<out String>?, sortOrder: String?): Cursor {
|
||||
val url = CacheFileProviderUrl.fromUri(uri)
|
||||
val file = url.getFile(context)
|
||||
|
||||
val resultProjection = projection ?: arrayOf(OpenableColumns.DISPLAY_NAME, OpenableColumns.SIZE)
|
||||
val resultCursor = MatrixCursor(resultProjection)
|
||||
|
||||
if (file.exists()) {
|
||||
val builder = resultCursor.newRow()
|
||||
|
||||
for (row in resultProjection) {
|
||||
when (row) {
|
||||
OpenableColumns.DISPLAY_NAME -> builder.add(url.filename)
|
||||
OpenableColumns.SIZE -> builder.add(file.length())
|
||||
else -> builder.add(null)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return resultCursor
|
||||
}
|
||||
|
||||
override fun getType(uri: Uri): String = CacheFileProviderUrl.fromUri(uri).mimeType
|
||||
|
||||
override fun openFile(uri: Uri, mode: String): ParcelFileDescriptor {
|
||||
if (mode == "r") {
|
||||
val url = CacheFileProviderUrl.fromUri(uri)
|
||||
val file = url.getFile(context)
|
||||
|
||||
return ParcelFileDescriptor.open(file, ParcelFileDescriptor.MODE_READ_ONLY)
|
||||
} else {
|
||||
throw IOException("illegal mode")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
data class CacheFileProviderUrl(
|
||||
val pathInCacheDirectory: String,
|
||||
val filename: String,
|
||||
val mimeType: String
|
||||
) {
|
||||
companion object {
|
||||
private const val PATH = "path"
|
||||
private const val FILENAME = "filename"
|
||||
private const val MIME_TYPE = "mimeType"
|
||||
|
||||
fun fromUri(uri: Uri) = CacheFileProviderUrl(
|
||||
pathInCacheDirectory = uri.getQueryParameter(PATH),
|
||||
filename = uri.getQueryParameter(FILENAME),
|
||||
mimeType = uri.getQueryParameter(MIME_TYPE)
|
||||
)
|
||||
|
||||
fun fromFile(file: File, filename: String, mimeType: String, context: Context) = CacheFileProviderUrl(
|
||||
filename = filename,
|
||||
mimeType = mimeType,
|
||||
pathInCacheDirectory = file.toRelativeString(context.externalCacheDir)
|
||||
)
|
||||
}
|
||||
|
||||
val serialized: Uri by lazy {
|
||||
Uri.Builder()
|
||||
.scheme("content")
|
||||
.authority(CacheFileProvider.AUTHORITY)
|
||||
.appendQueryParameter(PATH, pathInCacheDirectory)
|
||||
.appendQueryParameter(FILENAME, filename)
|
||||
.appendQueryParameter(MIME_TYPE, mimeType)
|
||||
.build()
|
||||
}
|
||||
|
||||
fun getFile(context: Context): File {
|
||||
return File(context.externalCacheDir, pathInCacheDirectory)
|
||||
}
|
||||
}
|
||||
@@ -4,8 +4,9 @@ import android.os.Handler
|
||||
import android.os.Looper
|
||||
import android.support.v4.os.CancellationSignal
|
||||
import android.util.Log
|
||||
import kotlinx.coroutines.experimental.launch
|
||||
import kotlinx.coroutines.experimental.suspendCancellableCoroutine
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import kotlinx.coroutines.suspendCancellableCoroutine
|
||||
import net.syncthing.java.bep.BlockPullerStatus
|
||||
import net.syncthing.java.client.SyncthingClient
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
@@ -13,6 +14,8 @@ import net.syncthing.lite.BuildConfig
|
||||
import org.apache.commons.io.FileUtils
|
||||
import java.io.File
|
||||
import java.io.IOException
|
||||
import kotlin.coroutines.resume
|
||||
import kotlin.coroutines.resumeWithException
|
||||
|
||||
class DownloadFileTask(private val fileStorageDirectory: File,
|
||||
syncthingClient: SyncthingClient,
|
||||
@@ -58,7 +61,7 @@ class DownloadFileTask(private val fileStorageDirectory: File,
|
||||
init {
|
||||
val file = DownloadFilePath(fileStorageDirectory, fileInfo.hash!!)
|
||||
|
||||
launch {
|
||||
GlobalScope.launch {
|
||||
if (file.targetFile.exists()) {
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.d(TAG, "there is already a file")
|
||||
@@ -69,43 +72,35 @@ class DownloadFileTask(private val fileStorageDirectory: File,
|
||||
return@launch
|
||||
}
|
||||
|
||||
syncthingClient.getBlockPuller(fileInfo.folder, { blockPuller ->
|
||||
val job = launch {
|
||||
try {
|
||||
if (!file.filesDirectory.isDirectory) {
|
||||
if (!file.filesDirectory.mkdirs()) {
|
||||
throw IOException("could not create output directory")
|
||||
}
|
||||
}
|
||||
|
||||
// download the file to a temp location
|
||||
val inputStream = blockPuller.pullFileCoroutine(fileInfo, this@DownloadFileTask::callProgress)
|
||||
|
||||
try {
|
||||
FileUtils.copyInputStreamToFile(inputStream, file.tempFile)
|
||||
file.tempFile.renameTo(file.targetFile)
|
||||
} finally {
|
||||
file.tempFile.delete()
|
||||
}
|
||||
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.i(TAG, "Downloaded file $fileInfo")
|
||||
}
|
||||
|
||||
callComplete(file.targetFile)
|
||||
} catch (e: Exception) {
|
||||
callError(e)
|
||||
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.w(TAG, "Failed to download file $fileInfo", e)
|
||||
}
|
||||
try {
|
||||
if (!file.filesDirectory.isDirectory) {
|
||||
if (!file.filesDirectory.mkdirs()) {
|
||||
throw IOException("could not create output directory")
|
||||
}
|
||||
}
|
||||
|
||||
cancellationSignal.setOnCancelListener {
|
||||
job.cancel()
|
||||
// download the file to a temp location
|
||||
val inputStream = syncthingClient.pullFile(fileInfo, this@DownloadFileTask::callProgress)
|
||||
|
||||
try {
|
||||
FileUtils.copyInputStreamToFile(inputStream, file.tempFile)
|
||||
file.tempFile.renameTo(file.targetFile)
|
||||
} finally {
|
||||
file.tempFile.delete()
|
||||
}
|
||||
}, { callError(IOException("could not get block puller for file")) })
|
||||
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.i(TAG, "Downloaded file $fileInfo")
|
||||
}
|
||||
|
||||
callComplete(file.targetFile)
|
||||
} catch (e: Exception) {
|
||||
callError(e)
|
||||
|
||||
if (BuildConfig.DEBUG) {
|
||||
Log.w(TAG, "Failed to download file $fileInfo", e)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -6,8 +6,9 @@ import android.content.Context
|
||||
import android.os.Handler
|
||||
import android.os.Looper
|
||||
import android.util.Log
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.bep.FolderBrowser
|
||||
import net.syncthing.java.client.SyncthingClient
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
@@ -35,7 +36,7 @@ class LibraryHandler(context: Context,
|
||||
|
||||
private val libraryManager = DefaultLibraryManager.with(context)
|
||||
private val isStarted = AtomicBoolean(false)
|
||||
private val isListeningPortTakenInternal = MutableLiveData<Boolean>().apply { value = false }
|
||||
private val isListeningPortTakenInternal = MutableLiveData<Boolean>().apply { postValue(false) }
|
||||
|
||||
val isListeningPortTaken: LiveData<Boolean> = isListeningPortTakenInternal
|
||||
|
||||
@@ -88,7 +89,7 @@ class LibraryHandler(context: Context,
|
||||
private fun onIndexRecordAcquired(folderInfo: FolderInfo, newRecords: List<FileInfo>, indexInfo: IndexInfo) {
|
||||
Log.i(TAG, "handleIndexRecordEvent trigger folder list update from index record acquired")
|
||||
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
onIndexUpdateProgressListener(folderInfo, (indexInfo.getCompleted() * 100).toInt())
|
||||
}
|
||||
}
|
||||
@@ -96,7 +97,7 @@ class LibraryHandler(context: Context,
|
||||
private fun onRemoteIndexAcquired(folderInfo: FolderInfo) {
|
||||
Log.i(TAG, "handleIndexAcquiredEvent trigger folder list update from index acquired")
|
||||
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
onIndexUpdateCompleteListener(folderInfo)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -3,7 +3,8 @@ package net.syncthing.lite.library
|
||||
import android.os.Handler
|
||||
import android.os.Looper
|
||||
import java.util.concurrent.Executors
|
||||
import kotlin.coroutines.experimental.suspendCoroutine
|
||||
import kotlin.coroutines.resume
|
||||
import kotlin.coroutines.suspendCoroutine
|
||||
|
||||
/**
|
||||
* This class manages the access to an LibraryInstance
|
||||
|
||||
@@ -8,8 +8,8 @@ import android.provider.DocumentsContract.Document
|
||||
import android.provider.DocumentsContract.Root
|
||||
import android.provider.DocumentsProvider
|
||||
import android.util.Log
|
||||
import kotlinx.coroutines.experimental.cancel
|
||||
import kotlinx.coroutines.experimental.runBlocking
|
||||
import kotlinx.coroutines.cancel
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import net.syncthing.java.bep.IndexBrowser
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
import net.syncthing.java.core.beans.FolderInfo
|
||||
|
||||
@@ -5,6 +5,8 @@ import android.net.Uri
|
||||
import android.os.Handler
|
||||
import android.os.Looper
|
||||
import android.util.Log
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.bep.BlockPusher
|
||||
import net.syncthing.java.client.SyncthingClient
|
||||
import net.syncthing.java.core.utils.PathUtils
|
||||
@@ -31,22 +33,28 @@ class UploadFileTask(context: Context, syncthingClient: SyncthingClient,
|
||||
|
||||
init {
|
||||
Log.i(TAG, "Uploading file $localFile to folder $syncthingFolder:$syncthingPath")
|
||||
syncthingClient.getBlockPusher(syncthingFolder, { blockPusher ->
|
||||
val observer = blockPusher.pushFile(uploadStream, syncthingFolder, syncthingPath)
|
||||
|
||||
handler.post { onProgress(observer) }
|
||||
GlobalScope.launch {
|
||||
try {
|
||||
val blockPusher = syncthingClient.getBlockPusher(folderId = syncthingFolder)
|
||||
val observer = blockPusher.pushFile(uploadStream, syncthingFolder, syncthingPath)
|
||||
|
||||
while (!observer.isCompleted()) {
|
||||
if (isCancelled)
|
||||
return@getBlockPusher
|
||||
|
||||
observer.waitForProgressUpdate()
|
||||
Log.i(TAG, "upload progress = ${observer.progressPercentage()}%")
|
||||
handler.post { onProgress(observer) }
|
||||
|
||||
while (!observer.isCompleted()) {
|
||||
if (isCancelled)
|
||||
return@launch
|
||||
|
||||
observer.waitForProgressUpdate()
|
||||
Log.i(TAG, "upload progress = ${observer.progressPercentage()}%")
|
||||
handler.post { onProgress(observer) }
|
||||
}
|
||||
IOUtils.closeQuietly(uploadStream)
|
||||
handler.post { onComplete() }
|
||||
} catch (ex: Exception) {
|
||||
handler.post { onError() }
|
||||
}
|
||||
IOUtils.closeQuietly(uploadStream)
|
||||
handler.post { onComplete() }
|
||||
}, { handler.post { onError() } })
|
||||
}
|
||||
}
|
||||
|
||||
fun cancel() {
|
||||
|
||||
@@ -4,8 +4,9 @@ import android.content.Context
|
||||
import android.net.Uri
|
||||
import android.os.Build
|
||||
import android.provider.OpenableColumns
|
||||
import kotlinx.coroutines.experimental.android.UI
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.beans.DeviceInfo
|
||||
import net.syncthing.lite.R
|
||||
@@ -43,16 +44,17 @@ object Util {
|
||||
fun importDeviceId(libraryHandler: LibraryHandler?, context: Context?, deviceId: String,
|
||||
onComplete: () -> Unit) {
|
||||
val deviceId2 = DeviceId(deviceId.toUpperCase(Locale.US))
|
||||
libraryHandler?.configuration { configuration ->
|
||||
libraryHandler?.library { configuration, syncthingClient, _ ->
|
||||
if (!configuration.peerIds.contains(deviceId2)) {
|
||||
configuration.peers = configuration.peers + DeviceInfo(deviceId2, null)
|
||||
configuration.persistLater()
|
||||
async(UI) {
|
||||
syncthingClient.connectToNewlyAddedDevices()
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
context?.toast(context.getString(R.string.device_import_success, deviceId2.shortId))
|
||||
onComplete()
|
||||
}
|
||||
} else {
|
||||
async(UI) {
|
||||
GlobalScope.launch (Dispatchers.Main) {
|
||||
context?.toast(context.getString(R.string.device_already_known, deviceId2.shortId))
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1 @@
|
||||
googleplay@nutomic.com
|
||||
@@ -0,0 +1 @@
|
||||
https://syncthing.net
|
||||
@@ -0,0 +1 @@
|
||||
en-GB
|
||||
@@ -0,0 +1,5 @@
|
||||
This project is an Android app, that works as a client for a Syncthing share (accessing Syncthing devices in the same way a client-server file sharing app access its proprietary server).
|
||||
|
||||
This is a client-oriented implementation, designed to work online by downloading and uploading files from an active device on the network (instead of synchronizing a local copy of the entire repository). This is quite different from the way the syncthing-android works, and its useful from those devices that cannot or wish not to download the entire repository (for example, mobile devices with limited storage available, wishing to access a syncthing share).
|
||||
|
||||
Source code: https://github.com/syncthing/syncthing-lite
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 73 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 40 KiB |
@@ -0,0 +1 @@
|
||||
A browser app for Syncthing-compatible shares
|
||||
@@ -0,0 +1 @@
|
||||
Syncthing Lite
|
||||
@@ -0,0 +1,3 @@
|
||||
- new connection handling
|
||||
- option for users to get detailed crash reports
|
||||
- bugfixes
|
||||
@@ -0,0 +1,9 @@
|
||||
<vector xmlns:android="http://schemas.android.com/apk/res/android"
|
||||
android:width="24dp"
|
||||
android:height="24dp"
|
||||
android:viewportWidth="24.0"
|
||||
android:viewportHeight="24.0">
|
||||
<path
|
||||
android:fillColor="#FF000000"
|
||||
android:pathData="M17,3L5,3c-1.11,0 -2,0.9 -2,2v14c0,1.1 0.89,2 2,2h14c1.1,0 2,-0.9 2,-2L21,7l-4,-4zM12,19c-1.66,0 -3,-1.34 -3,-3s1.34,-3 3,-3 3,1.34 3,3 -1.34,3 -3,3zM15,9L5,9L5,5h10v4z"/>
|
||||
</vector>
|
||||
@@ -0,0 +1,35 @@
|
||||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<layout xmlns:android="http://schemas.android.com/apk/res/android"
|
||||
xmlns:tools="http://schemas.android.com/tools">
|
||||
|
||||
<data>
|
||||
<variable
|
||||
name="filename"
|
||||
type="String" />
|
||||
</data>
|
||||
|
||||
<LinearLayout
|
||||
android:orientation="vertical"
|
||||
android:layout_width="match_parent"
|
||||
android:layout_height="wrap_content">
|
||||
|
||||
<TextView
|
||||
android:padding="8dp"
|
||||
android:textAppearance="?android:textAppearanceMedium"
|
||||
android:text="@{filename}"
|
||||
tools:text="Filename.type"
|
||||
android:layout_width="match_parent"
|
||||
android:layout_height="wrap_content" />
|
||||
|
||||
<Button
|
||||
android:drawableStart="@drawable/ic_save_black_24dp"
|
||||
android:id="@+id/save_as_button"
|
||||
android:padding="8dp"
|
||||
android:gravity="start|center_vertical"
|
||||
android:text="@string/dialog_file_save_as"
|
||||
android:background="?selectableItemBackground"
|
||||
android:layout_width="match_parent"
|
||||
android:layout_height="wrap_content" />
|
||||
|
||||
</LinearLayout>
|
||||
</layout>
|
||||
@@ -0,0 +1,5 @@
|
||||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<adaptive-icon xmlns:android="http://schemas.android.com/apk/res/android">
|
||||
<background android:drawable="@mipmap/ic_background"/>
|
||||
<foreground android:drawable="@mipmap/ic_foreground"/>
|
||||
</adaptive-icon>
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 116 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 7.6 KiB |
@@ -19,7 +19,7 @@
|
||||
<string name="last_modified_time">Última modificación: %1$s</string>
|
||||
<string name="remove_device_title">¿Quitar dispositivo %1$s?</string>
|
||||
<string name="remove_device_message">¿Quitar dispositivo %1$s de la lista de dispositivos conocidos?</string>
|
||||
<string name="device_import_success">Dispositivo %1$simportado con éxito</string>
|
||||
<string name="device_import_success">Dispositivo %1$s importado con éxito</string>
|
||||
<string name="device_already_known">Dispositivo %1$s ya está presente</string>
|
||||
<string name="folders_label">Carpetas</string>
|
||||
<string name="devices_label">Dispositivos</string>
|
||||
|
||||
@@ -40,5 +40,11 @@
|
||||
<string name="settings_app_version_title">Verzió</string>
|
||||
<string name="settings_local_device_name">Helyi eszköz neve</string>
|
||||
<string name="settings_local_device_summary">Név amit a többi eszköz fog látni</string>
|
||||
<string name="settings_shutdown_delay_title">Leállítás késleltetés</string>
|
||||
<string name="settings_shutdown_delay_summary">A Syncthing leállítása ennyi idő elteltével a kliens utolsó csatlakozása után</string>
|
||||
<string name="device_id_dialog_title">Eszközazonosító megadása</string>
|
||||
</resources>
|
||||
<string name="settings_shutdown_delay_10_seconds">10 másodperc</string>
|
||||
<string name="settings_shutdown_delay_30_seconds">30 másodperc</string>
|
||||
<string name="settings_shutdown_delay_1_minute">1 perc</string>
|
||||
<string name="settings_shutdown_delay_5_minutes">5 perc</string>
|
||||
</resources>
|
||||
|
||||
@@ -0,0 +1,2 @@
|
||||
<resources>
|
||||
</resources>
|
||||
@@ -3,33 +3,47 @@
|
||||
<string name="folder_list_empty_message">没有可用的文件夹</string>
|
||||
<string name="clear_local_cache_index_label">清除本地缓存/索引</string>
|
||||
<string name="devices_list_view_empty_message">没有可用的设备</string>
|
||||
<string name="invalid_device_id">无效的设备 ID</string>
|
||||
<string name="dialog_downloading_file">正下载文件 %1$s</string>
|
||||
<string name="invalid_device_id">错误:无效的设备 ID</string>
|
||||
<string name="dialog_downloading_file">正在下载文件 %1$s</string>
|
||||
<string name="toast_file_download_failed">下载文件失败</string>
|
||||
<string name="toast_open_file_failed">没有找到兼容的程序</string>
|
||||
<string name="toast_open_file_failed">未找到兼容的应用</string>
|
||||
<string name="toast_file_upload_failed">上传文件失败</string>
|
||||
<string name="toast_upload_complete">文件上传完成</string>
|
||||
<string name="dialog_uploading_file">正上传文件 %1$s</string>
|
||||
<string name="dialog_uploading_file">正在上传文件 %1$s</string>
|
||||
<string name="clear_cache_and_index_title">确定清除本地缓存和索引?</string>
|
||||
<string name="clear_cache_and_index_body">确定清除全部本地缓存数据和索引数据?</string>
|
||||
<string name="loading_config_starting_syncthing_client">载入配置,正在启动 syncthing 客户端</string>
|
||||
<string name="remove_device_title">移除设备: %1$s</string>
|
||||
<string name="device_import_success">成功导入的设备: %1$s</string>
|
||||
<string name="device_already_known">已经存在的设备: %1$s</string>
|
||||
<string name="loading_config_starting_syncthing_client">正在载入配置,启动 syncthing 客户端……</string>
|
||||
<string name="last_modified_time">最后修改:%1$s</string>
|
||||
<string name="remove_device_title">移除设备: %1$s?</string>
|
||||
<string name="remove_device_message">从已知设备列表中移除 %1$s?</string>
|
||||
<string name="device_import_success">已成功导入设备 %1$s</string>
|
||||
<string name="device_already_known">设备已存在 %1$s</string>
|
||||
<string name="folders_label">文件夹</string>
|
||||
<string name="devices_label">设备</string>
|
||||
<string name="folder_label_format">%1$s(%2$s)</string>
|
||||
<string name="folder_content_info">%1$s,%2$d个文件,%3$d个目录</string>
|
||||
<string name="file_info">%1$s,最后修改 %2$s</string>
|
||||
<string name="show_device_id">显示设备 ID</string>
|
||||
<string name="device_id">设备 ID</string>
|
||||
<string name="device_id_copied">设备 ID 已复制到剪贴板</string>
|
||||
<string name="share_device_id_chooser">分享设备 ID 于</string>
|
||||
<string name="other_syncthing_instance_title">另一个 Syncthing 实例正在运行</string>
|
||||
<string name="other_syncthing_instance_message">本地发现将无法工作。停止其他 Syncthing 实例以启用本地发现。</string>
|
||||
<string name="intro_page_one_title">欢迎使用 Syncthing Lite</string>
|
||||
<string name="intro_page_two_title">添加一个设备</string>
|
||||
<string name="intro_page_three_title">分享您的文件夹</string>
|
||||
<string name="intro_page_two_description">输入一个 Syncthing 设备 ID,或者通过 QR 码扫描一个设备 ID</string>
|
||||
<string name="settings">设定</string>
|
||||
<string name="intro_page_one_description">Syncthing 以开放、可靠并去中心化的软件替换掉封闭的云服务。您的数据仍由您拥有,您可以选择它们的存储位置,如果要共享给第三方您还可以选择如何在互联网上传输它们。</string>
|
||||
<string name="intro_page_two_title">添加设备</string>
|
||||
<string name="intro_page_three_title">共享您的文件夹</string>
|
||||
<string name="intro_page_two_description">输入 Syncthing 设备 ID,或者通过 QR 码扫描设备 ID</string>
|
||||
<string name="intro_page_three_description">已接受 ID 为%1$s 的设备,并与它共享了一个文件夹。设备连接可能需要花上数分钟。</string>
|
||||
<string name="settings">设置</string>
|
||||
<string name="settings_app_version_title">应用版本</string>
|
||||
<string name="settings_local_device_name">本地设备名称</string>
|
||||
<string name="settings_local_device_summary">其他设备将会看到这台设备的名字</string>
|
||||
<string name="settings_local_device_summary">此设备将被其他设备看到的名称</string>
|
||||
<string name="settings_shutdown_delay_title">关闭延迟</string>
|
||||
<string name="settings_shutdown_delay_summary">关闭 Syncthing 客户端与其最后使用之间的时间</string>
|
||||
<string name="device_id_dialog_title">输入设备 ID</string>
|
||||
</resources>
|
||||
<string name="settings_shutdown_delay_10_seconds">10 秒</string>
|
||||
<string name="settings_shutdown_delay_30_seconds">30 秒</string>
|
||||
<string name="settings_shutdown_delay_1_minute">1 分钟</string>
|
||||
<string name="settings_shutdown_delay_5_minutes">5 分钟</string>
|
||||
</resources>
|
||||
|
||||
@@ -42,6 +42,8 @@
|
||||
<string name="settings_local_device_summary">The name that other devices will see for this device</string>
|
||||
<string name="settings_shutdown_delay_title">Shutdown delay</string>
|
||||
<string name="settings_shutdown_delay_summary">Time before shuting down the Syncthing client after its last usage</string>
|
||||
<string name="settings_crash_handler_title">Custom Crash-Handler</string>
|
||||
<string name="settings_crash_handler_summary">Copy the error message to the clipboard when the App crashes</string>
|
||||
<string name="device_id_dialog_title">Enter Device ID</string>
|
||||
<string name="settings_shutdown_delay_10_seconds">10 seconds</string>
|
||||
<string name="settings_shutdown_delay_30_seconds">30 seconds</string>
|
||||
@@ -53,4 +55,5 @@
|
||||
or the connection was interrupted.
|
||||
This does not apply to local discovery connections.
|
||||
</string>
|
||||
<string name="dialog_file_save_as">Save as</string>
|
||||
</resources>
|
||||
|
||||
@@ -1,4 +0,0 @@
|
||||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<paths>
|
||||
<external-cache-path name="files" path="/" />
|
||||
</paths>
|
||||
@@ -24,6 +24,11 @@
|
||||
|
||||
-->
|
||||
|
||||
<CheckBoxPreference
|
||||
android:key="crash_handler"
|
||||
android:title="@string/settings_crash_handler_title"
|
||||
android:summary="@string/settings_crash_handler_summary" />
|
||||
|
||||
<Preference
|
||||
android:key="app_version"
|
||||
android:title="@string/settings_app_version_title"/>
|
||||
|
||||
+2
-2
@@ -1,10 +1,10 @@
|
||||
// Top-level build file where you can add configuration options common to all sub-projects/modules.
|
||||
|
||||
buildscript {
|
||||
ext.kotlin_version = '1.2.61'
|
||||
ext.kotlin_version = '1.3.0'
|
||||
ext.support_version = '27.0.2'
|
||||
ext.build_tools_version = '3.2.0'
|
||||
ext.anko_version = '0.10.7'
|
||||
ext.anko_version = '0.10.8'
|
||||
ext.protobuf_lite_version = '3.0.1'
|
||||
repositories {
|
||||
mavenLocal()
|
||||
|
||||
@@ -1,54 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -e
|
||||
|
||||
NEW_VERSION_NAME=$1
|
||||
OLD_VERSION_NAME=$(grep "versionName" "app/build.gradle" | awk '{print $2}' | tr -d "\"")
|
||||
if [[ -z ${NEW_VERSION_NAME} ]]
|
||||
then
|
||||
echo "New version name is empty. Please set a new version. Current version: $OLD_VERSION_NAME"
|
||||
exit
|
||||
fi
|
||||
|
||||
echo "
|
||||
|
||||
Updating Translations
|
||||
-----------------------------
|
||||
"
|
||||
tx push -s
|
||||
# Force push/pull to make sure this is executed. Apparently tx only compares timestamps, not file
|
||||
# contents. So if a file was `touch`ed, it won't be updated by default.
|
||||
tx pull -a -f
|
||||
git add -A "app/src/main/res/values-*/strings.xml"
|
||||
if ! git diff --cached --exit-code;
|
||||
then
|
||||
git commit -m "Imported translations"
|
||||
fi
|
||||
|
||||
echo "
|
||||
|
||||
Updating Version
|
||||
-----------------------------
|
||||
"
|
||||
OLD_VERSION_CODE=$(grep "versionCode" "app/build.gradle" -m 1 | awk '{print $2}')
|
||||
NEW_VERSION_CODE=$(($OLD_VERSION_CODE + 1))
|
||||
sed -i "s/versionCode $OLD_VERSION_CODE/versionCode $NEW_VERSION_CODE/" "app/build.gradle"
|
||||
sed -i "s/versionName \"$OLD_VERSION_NAME\"/versionName \"$NEW_VERSION_NAME\"/" "app/build.gradle"
|
||||
|
||||
LIBRARY_NAME="com.github.Nutomic:syncthing-java"
|
||||
sed -i "s/$LIBRARY_NAME:$OLD_VERSION_NAME/$LIBRARY_NAME:$NEW_VERSION_NAME/" "app/build.gradle"
|
||||
|
||||
git add "app/build.gradle"
|
||||
git commit -m "Version $NEW_VERSION_NAME"
|
||||
git tag ${NEW_VERSION_NAME}
|
||||
|
||||
echo "
|
||||
|
||||
Running Lint
|
||||
-----------------------------
|
||||
"
|
||||
./gradlew clean lintVitalRelease
|
||||
|
||||
echo "
|
||||
Update ready.
|
||||
"
|
||||
@@ -1,37 +0,0 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
set -e
|
||||
|
||||
version=$(git describe --tags)
|
||||
regex='^[0-9]+\.[0-9]+\.[0-9]+$'
|
||||
if [[ ! ${version} =~ $regex ]]
|
||||
then
|
||||
echo "Current commit is not a release"
|
||||
exit;
|
||||
fi
|
||||
|
||||
echo "
|
||||
|
||||
Pushing to Github
|
||||
-----------------------------
|
||||
"
|
||||
git push
|
||||
git push --tags
|
||||
|
||||
echo "
|
||||
|
||||
Push to Google Play
|
||||
-----------------------------
|
||||
"
|
||||
|
||||
read -s -p "Enter signing password: " password
|
||||
|
||||
SIGNING_PASSWORD=${password} ./gradlew assembleRelease
|
||||
|
||||
# Upload apk and listing to Google Play
|
||||
SIGNING_PASSWORD=${password} ./gradlew publishRelease
|
||||
|
||||
echo "
|
||||
|
||||
Release published!
|
||||
"
|
||||
+1
-1
@@ -1 +1 @@
|
||||
include ':app', ':syncthing-repository-android', ':syncthing-repository-default', ':syncthing-relay-client', ':syncthing-bep', ':syncthing-core', ':syncthing-client', ':syncthing-discovery', ':syncthing-client-cli', ':syncthing-http-relay-client'
|
||||
include ':app', ':syncthing-repository-android', ':syncthing-repository-default', ':syncthing-relay-client', ':syncthing-bep', ':syncthing-core', ':syncthing-client', ':syncthing-discovery', ':syncthing-client-cli'
|
||||
|
||||
@@ -6,10 +6,9 @@ dependencies {
|
||||
compile fileTree(dir: 'libs', include: ['*.jar'])
|
||||
compile project(':syncthing-core')
|
||||
compile project(':syncthing-relay-client')
|
||||
compile project(':syncthing-http-relay-client')
|
||||
compile "net.jpountz.lz4:lz4:1.3.0"
|
||||
implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:0.30.2'
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:1.0.0'
|
||||
implementation "com.google.protobuf:protobuf-lite:$protobuf_lite_version"
|
||||
}
|
||||
|
||||
|
||||
@@ -15,58 +15,46 @@
|
||||
package net.syncthing.java.bep
|
||||
|
||||
import com.google.protobuf.ByteString
|
||||
import kotlinx.coroutines.experimental.*
|
||||
import kotlinx.coroutines.experimental.channels.Channel
|
||||
import net.syncthing.java.bep.BlockExchangeProtos.ErrorCode
|
||||
import net.syncthing.java.bep.BlockExchangeProtos.Request
|
||||
import kotlinx.coroutines.*
|
||||
import kotlinx.coroutines.channels.Channel
|
||||
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
|
||||
import net.syncthing.java.bep.utils.longSumBy
|
||||
import net.syncthing.java.core.beans.BlockInfo
|
||||
import net.syncthing.java.core.beans.FileBlocks
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
import net.syncthing.java.core.interfaces.TempRepository
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import org.apache.commons.io.FileUtils
|
||||
import org.bouncycastle.util.encoders.Hex
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.*
|
||||
import java.lang.Exception
|
||||
import java.security.MessageDigest
|
||||
import java.util.*
|
||||
import kotlin.collections.HashMap
|
||||
|
||||
class BlockPuller internal constructor(private val connectionHandler: ConnectionHandler,
|
||||
private val indexHandler: IndexHandler,
|
||||
private val responseHandler: ResponseHandler,
|
||||
private val tempRepository: TempRepository) {
|
||||
|
||||
object BlockPuller {
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
|
||||
fun pullFileSync(
|
||||
suspend fun pullFile(
|
||||
fileInfo: FileInfo,
|
||||
progressListener: (status: BlockPullerStatus) -> Unit = { }
|
||||
progressListener: (status: BlockPullerStatus) -> Unit = { },
|
||||
connections: List<ConnectionActorWrapper>,
|
||||
indexHandler: IndexHandler,
|
||||
tempRepository: TempRepository
|
||||
): InputStream {
|
||||
return runBlocking {
|
||||
pullFileCoroutine(fileInfo, progressListener)
|
||||
val connectionHelper = MultiConnectionHelper(connections) {
|
||||
it.hasFolder(fileInfo.folder)
|
||||
}
|
||||
}
|
||||
|
||||
suspend fun pullFileCoroutine(
|
||||
fileInfo: FileInfo,
|
||||
progressListener: (status: BlockPullerStatus) -> Unit = { }
|
||||
): InputStream {
|
||||
val fileBlocks = indexHandler.waitForRemoteIndexAcquired(connectionHandler)
|
||||
.getFileInfoAndBlocksByPath(fileInfo.folder, fileInfo.path)
|
||||
?.value
|
||||
?: throw IOException("file not found in local index for folder = ${fileInfo.folder} path = ${fileInfo.path}")
|
||||
logger.info("pulling file = {}", fileBlocks)
|
||||
NetworkUtils.assertProtocol(connectionHandler.hasFolder(fileBlocks.folder), { "supplied connection handler $connectionHandler will not share folder ${fileBlocks.folder}" })
|
||||
// fail early if there is no matching connection
|
||||
connectionHelper.pickConnection()
|
||||
|
||||
val (newFileInfo, fileBlocks) = indexHandler.getFileInfoAndBlocksByPath(fileInfo.folder, fileInfo.path) ?: throw FileNotFoundException()
|
||||
|
||||
// the file could have changed since the caller read it
|
||||
// this would save the file using a wrong name, so throw here
|
||||
if (fileBlocks.hash != fileInfo.hash) {
|
||||
throw IllegalStateException("the current file entry hash does not match the hash of the provided one")
|
||||
}
|
||||
|
||||
logger.info("pulling file = {}", fileBlocks)
|
||||
|
||||
val blockTempIdByHash = Collections.synchronizedMap(HashMap<String, String>())
|
||||
|
||||
var status = BlockPullerStatus(
|
||||
@@ -75,6 +63,47 @@ class BlockPuller internal constructor(private val connectionHandler: Connection
|
||||
totalFileSize = fileBlocks.size
|
||||
)
|
||||
|
||||
suspend fun pullBlock(fileBlocks: FileBlocks, block: BlockInfo, timeoutInMillis: Long, connectionActorWrapper: ConnectionActorWrapper): ByteArray {
|
||||
logger.debug("sent message for block, hash = {}", block.hash)
|
||||
|
||||
val response =
|
||||
withTimeout(timeoutInMillis) {
|
||||
try {
|
||||
connectionActorWrapper.sendRequest(
|
||||
BlockExchangeProtos.Request.newBuilder()
|
||||
.setFolder(fileBlocks.folder)
|
||||
.setName(fileBlocks.path)
|
||||
.setOffset(block.offset)
|
||||
.setSize(block.size)
|
||||
.setHash(ByteString.copyFrom(Hex.decode(block.hash)))
|
||||
.buildPartial()
|
||||
)
|
||||
} catch (ex: TimeoutCancellationException) {
|
||||
// It seems like the TimeoutCancellationException
|
||||
// is handled differently so that the timeout is ignored.
|
||||
// Due to that, it's converted to an IOException.
|
||||
|
||||
throw IOException("timeout during requesting block")
|
||||
}
|
||||
}
|
||||
|
||||
if (response.code != BlockExchangeProtos.ErrorCode.NO_ERROR) {
|
||||
// the server does not have/ want to provide this file -> don't ask him again
|
||||
connectionHelper.disableConnection(connectionActorWrapper)
|
||||
|
||||
throw IOException("received error response ${response.code}")
|
||||
}
|
||||
|
||||
val data = response.data.toByteArray()
|
||||
val hash = Hex.toHexString(MessageDigest.getInstance("SHA-256").digest(data))
|
||||
|
||||
if (hash != block.hash) {
|
||||
throw IllegalStateException("expected block with hash ${block.hash}, but got block with hash $hash")
|
||||
}
|
||||
|
||||
return data
|
||||
}
|
||||
|
||||
try {
|
||||
val reportProgressLock = Object()
|
||||
|
||||
@@ -94,9 +123,31 @@ class BlockPuller internal constructor(private val connectionHandler: Connection
|
||||
repeat(4 /* 4 blocks per time */) { workerNumber ->
|
||||
async {
|
||||
for (block in pipe) {
|
||||
logger.debug("request block with hash = {} from worker {}", block.hash, workerNumber)
|
||||
logger.debug("message block with hash = {} from worker {}", block.hash, workerNumber)
|
||||
|
||||
val blockContent = pullBlock(fileBlocks, block, 1000 * 60 /* 60 seconds timeout per block */)
|
||||
lateinit var blockContent: ByteArray
|
||||
|
||||
val attempts = 0..4
|
||||
|
||||
for (attempt in attempts) {
|
||||
try {
|
||||
blockContent = pullBlock(fileBlocks, block, 1000 * 60 /* 60 seconds timeout per block */, connectionHelper.pickConnection())
|
||||
|
||||
break
|
||||
} catch (ex: IOException) {
|
||||
if (attempt == attempts.last) {
|
||||
throw ex
|
||||
} else {
|
||||
// will retry after a pause
|
||||
// 0: 300 ms after the first attempt
|
||||
// 1: 1200 ms after the second attempt
|
||||
// 2: 2700 ms after the third attempt
|
||||
// 3: 4800 ms after the third attempt
|
||||
// total: 9000 ms
|
||||
delay((attempt + 1) * (attempt + 1) * 300L)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
blockTempIdByHash[block.hash] = tempRepository.pushTempData(blockContent)
|
||||
|
||||
@@ -138,57 +189,6 @@ class BlockPuller internal constructor(private val connectionHandler: Connection
|
||||
throw ex
|
||||
}
|
||||
}
|
||||
|
||||
private suspend fun pullBlock(fileBlocks: FileBlocks, block: BlockInfo, timeoutInMillis: Long): ByteArray {
|
||||
logger.debug("sent request for block, hash = {}", block.hash)
|
||||
|
||||
val response =
|
||||
withTimeout(timeoutInMillis) {
|
||||
try {
|
||||
doRequest(
|
||||
Request.newBuilder()
|
||||
.setFolder(fileBlocks.folder)
|
||||
.setName(fileBlocks.path)
|
||||
.setOffset(block.offset)
|
||||
.setSize(block.size)
|
||||
.setHash(ByteString.copyFrom(Hex.decode(block.hash)))
|
||||
)
|
||||
} catch (ex: TimeoutCancellationException) {
|
||||
// It seems like the TimeoutCancellationException
|
||||
// is handled differently so that the timeout is ignored.
|
||||
// Due to that, it's converted to an IOException.
|
||||
|
||||
throw IOException("timeout during requesting block")
|
||||
}
|
||||
}
|
||||
|
||||
NetworkUtils.assertProtocol(response.code == ErrorCode.NO_ERROR) {
|
||||
"received error response, code = ${response.code}"
|
||||
}
|
||||
|
||||
val data = response.data.toByteArray()
|
||||
val hash = Hex.toHexString(MessageDigest.getInstance("SHA-256").digest(data))
|
||||
|
||||
if (hash != block.hash) {
|
||||
throw IllegalStateException("expected block with hash ${block.hash}, but got block with hash $hash")
|
||||
}
|
||||
|
||||
return data
|
||||
}
|
||||
|
||||
private suspend fun doRequest(request: Request.Builder): BlockExchangeProtos.Response {
|
||||
return suspendCancellableCoroutine { continuation ->
|
||||
val requestId = responseHandler.registerListener { response ->
|
||||
continuation.resume(response)
|
||||
}
|
||||
|
||||
connectionHandler.sendMessage(
|
||||
request
|
||||
.setId(requestId)
|
||||
.build()
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
data class BlockPullerStatus(
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
/*
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
@@ -14,15 +15,15 @@
|
||||
package net.syncthing.java.bep
|
||||
|
||||
import com.google.protobuf.ByteString
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.async
|
||||
import net.syncthing.java.bep.BlockExchangeProtos.Vector
|
||||
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
|
||||
import net.syncthing.java.core.beans.*
|
||||
import net.syncthing.java.core.beans.FileInfo.Version
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.core.utils.BlockUtils
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import net.syncthing.java.core.utils.submitLogging
|
||||
import org.apache.commons.io.IOUtils
|
||||
import org.apache.commons.lang3.tuple.Pair
|
||||
import org.bouncycastle.util.encoders.Hex
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.Closeable
|
||||
@@ -32,36 +33,35 @@ import java.nio.ByteBuffer
|
||||
import java.security.MessageDigest
|
||||
import java.util.*
|
||||
import java.util.concurrent.ConcurrentHashMap
|
||||
import java.util.concurrent.ExecutionException
|
||||
import java.util.concurrent.Executors
|
||||
import java.util.concurrent.Future
|
||||
import java.util.concurrent.atomic.AtomicBoolean
|
||||
import java.util.concurrent.atomic.AtomicReference
|
||||
|
||||
class BlockPusher internal constructor(private val localDeviceId: DeviceId,
|
||||
private val connectionHandler: ConnectionHandler,
|
||||
private val indexHandler: IndexHandler) {
|
||||
// TODO: refactor this
|
||||
class BlockPusher(private val localDeviceId: DeviceId,
|
||||
private val connectionHandler: ConnectionActorWrapper,
|
||||
private val indexHandler: IndexHandler,
|
||||
private val requestHandlerRegistry: RequestHandlerRegistry) {
|
||||
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
|
||||
|
||||
fun pushDelete(folderId: String, targetPath: String): IndexEditObserver {
|
||||
suspend fun pushDelete(folderId: String, targetPath: String): BlockExchangeProtos.IndexUpdate {
|
||||
val fileInfo = indexHandler.waitForRemoteIndexAcquired(connectionHandler).getFileInfoByPath(folderId, targetPath)!!
|
||||
NetworkUtils.assertProtocol(connectionHandler.hasFolder(fileInfo.folder), {"supplied connection handler $connectionHandler will not share folder ${fileInfo.folder}"})
|
||||
return IndexEditObserver(sendIndexUpdate(folderId, BlockExchangeProtos.FileInfo.newBuilder()
|
||||
return sendIndexUpdate(folderId, BlockExchangeProtos.FileInfo.newBuilder()
|
||||
.setName(targetPath)
|
||||
.setType(BlockExchangeProtos.FileInfoType.valueOf(fileInfo.type.name))
|
||||
.setDeleted(true), fileInfo.versionList))
|
||||
.setDeleted(true), fileInfo.versionList)
|
||||
}
|
||||
|
||||
fun pushDir(folder: String, path: String): IndexEditObserver {
|
||||
suspend fun pushDir(folder: String, path: String): BlockExchangeProtos.IndexUpdate {
|
||||
NetworkUtils.assertProtocol(connectionHandler.hasFolder(folder), {"supplied connection handler $connectionHandler will not share folder $folder"})
|
||||
return IndexEditObserver(sendIndexUpdate(folder, BlockExchangeProtos.FileInfo.newBuilder()
|
||||
return sendIndexUpdate(folder, BlockExchangeProtos.FileInfo.newBuilder()
|
||||
.setName(path)
|
||||
.setType(BlockExchangeProtos.FileInfoType.DIRECTORY), null))
|
||||
.setType(BlockExchangeProtos.FileInfoType.DIRECTORY), null)
|
||||
}
|
||||
|
||||
fun pushFile(inputStream: InputStream, folderId: String, targetPath: String): FileUploadObserver {
|
||||
suspend fun pushFile(inputStream: InputStream, folderId: String, targetPath: String): FileUploadObserver {
|
||||
val fileInfo = indexHandler.waitForRemoteIndexAcquired(connectionHandler).getFileInfoByPath(folderId, targetPath)
|
||||
NetworkUtils.assertProtocol(connectionHandler.hasFolder(folderId), {"supplied connection handler $connectionHandler will not share folder $folderId"})
|
||||
assert(fileInfo == null || fileInfo.folder == folderId)
|
||||
@@ -73,38 +73,33 @@ class BlockPusher internal constructor(private val localDeviceId: DeviceId,
|
||||
val uploadError = AtomicReference<Exception>()
|
||||
val isCompleted = AtomicBoolean(false)
|
||||
val updateLock = Object()
|
||||
val listener = {request: BlockExchangeProtos.Request ->
|
||||
if (request.folder == folderId && request.name == targetPath) {
|
||||
val requestFilter = RequestHandlerFilter(
|
||||
deviceId = connectionHandler.deviceId,
|
||||
folderId = folderId,
|
||||
path = targetPath
|
||||
)
|
||||
|
||||
requestHandlerRegistry.registerListener(requestFilter) { request ->
|
||||
GlobalScope.async {
|
||||
val hash = Hex.toHexString(request.hash.toByteArray())
|
||||
logger.debug("handling block request = {}:{}-{} ({})", request.name, request.offset, request.size, hash)
|
||||
val data = dataSource.getBlock(request.offset, request.size, hash)
|
||||
val future = connectionHandler.sendMessage(BlockExchangeProtos.Response.newBuilder()
|
||||
|
||||
sentBlocks.add(hash)
|
||||
synchronized(updateLock) {
|
||||
updateLock.notifyAll()
|
||||
}
|
||||
|
||||
BlockExchangeProtos.Response.newBuilder()
|
||||
.setCode(BlockExchangeProtos.ErrorCode.NO_ERROR)
|
||||
.setData(ByteString.copyFrom(data))
|
||||
.setId(request.id)
|
||||
.build())
|
||||
monitoringProcessExecutorService.submitLogging {
|
||||
try {
|
||||
future.get()
|
||||
sentBlocks.add(hash)
|
||||
synchronized(updateLock) {
|
||||
updateLock.notifyAll()
|
||||
}
|
||||
//TODO retry on error, register error and throw on watcher
|
||||
} catch (ex: InterruptedException) {
|
||||
//return and do nothing
|
||||
} catch (ex: ExecutionException) {
|
||||
uploadError.set(ex)
|
||||
synchronized(updateLock) {
|
||||
updateLock.notifyAll()
|
||||
}
|
||||
}
|
||||
}
|
||||
.build()
|
||||
}
|
||||
}
|
||||
connectionHandler.registerOnRequestMessageReceivedListeners(listener)
|
||||
|
||||
logger.debug("send index update for file = {}", targetPath)
|
||||
val indexListener = { folderInfo: FolderInfo, newRecords: List<FileInfo>, indexInfo: IndexInfo ->
|
||||
val indexListener = { folderInfo: FolderInfo, newRecords: List<FileInfo>, _: IndexInfo ->
|
||||
if (folderInfo.folderId == folderId) {
|
||||
for (fileInfo2 in newRecords) {
|
||||
if (fileInfo2.path == targetPath && fileInfo2.hash == dataSource.getHash()) { //TODO check not invalid
|
||||
@@ -122,7 +117,7 @@ class BlockPusher internal constructor(private val localDeviceId: DeviceId,
|
||||
.setName(targetPath)
|
||||
.setSize(fileSize)
|
||||
.setType(BlockExchangeProtos.FileInfoType.FILE)
|
||||
.addAllBlocks(dataSource.blocks), fileInfo?.versionList).right
|
||||
.addAllBlocks(dataSource.blocks), fileInfo?.versionList)
|
||||
return object : FileUploadObserver() {
|
||||
|
||||
override fun progressPercentage() = if (isCompleted.get()) 100 else (sentBlocks.size.toFloat() / dataSource.getHashes().size).toInt()
|
||||
@@ -134,7 +129,7 @@ class BlockPusher internal constructor(private val localDeviceId: DeviceId,
|
||||
logger.debug("closing upload process")
|
||||
monitoringProcessExecutorService.shutdown()
|
||||
indexHandler.unregisterOnIndexRecordAcquiredListener(indexListener)
|
||||
connectionHandler.unregisterOnRequestMessageReceivedListeners(listener)
|
||||
requestHandlerRegistry.unregisterListener(requestFilter)
|
||||
val fileInfo1 = indexHandler.pushRecord(indexUpdate.folder, indexUpdate.filesList.single())
|
||||
logger.info("sent file info record = {}", fileInfo1)
|
||||
}
|
||||
@@ -153,8 +148,8 @@ class BlockPusher internal constructor(private val localDeviceId: DeviceId,
|
||||
}
|
||||
}
|
||||
|
||||
private fun sendIndexUpdate(folderId: String, fileInfoBuilder: BlockExchangeProtos.FileInfo.Builder,
|
||||
oldVersions: Iterable<Version>?): Pair<Future<*>, BlockExchangeProtos.IndexUpdate> {
|
||||
private suspend fun sendIndexUpdate(folderId: String, fileInfoBuilder: BlockExchangeProtos.FileInfo.Builder,
|
||||
oldVersions: Iterable<Version>?): BlockExchangeProtos.IndexUpdate {
|
||||
run {
|
||||
val nextSequence = indexHandler.sequencer().nextSequence()
|
||||
val list = oldVersions ?: emptyList()
|
||||
@@ -183,7 +178,10 @@ class BlockPusher internal constructor(private val localDeviceId: DeviceId,
|
||||
.addFiles(fileInfo)
|
||||
.build()
|
||||
logger.debug("index update = {}", fileInfo)
|
||||
return Pair.of(connectionHandler.sendMessage(indexUpdate), indexUpdate)
|
||||
|
||||
connectionHandler.sendIndexUpdate(indexUpdate)
|
||||
|
||||
return indexUpdate
|
||||
}
|
||||
|
||||
abstract inner class FileUploadObserver : Closeable {
|
||||
@@ -204,33 +202,6 @@ class BlockPusher internal constructor(private val localDeviceId: DeviceId,
|
||||
}
|
||||
}
|
||||
|
||||
inner class IndexEditObserver(private val future: Future<*>, private val indexUpdate: BlockExchangeProtos.IndexUpdate) : Closeable {
|
||||
|
||||
//throw exception if job has errors
|
||||
@Throws(InterruptedException::class, ExecutionException::class)
|
||||
fun isCompleted(): Boolean {
|
||||
return if (future.isDone) {
|
||||
future.get()
|
||||
true
|
||||
} else {
|
||||
false
|
||||
}
|
||||
}
|
||||
|
||||
constructor(pair: Pair<Future<*>, BlockExchangeProtos.IndexUpdate>) : this(pair.left, pair.right)
|
||||
|
||||
@Throws(InterruptedException::class, ExecutionException::class)
|
||||
fun waitForComplete() {
|
||||
future.get()
|
||||
}
|
||||
|
||||
@Throws(IOException::class)
|
||||
override fun close() {
|
||||
indexHandler.pushRecord(indexUpdate.folder, indexUpdate.filesList.single())
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private class DataSource @Throws(IOException::class) constructor(private val inputStream: InputStream) {
|
||||
|
||||
var size: Long = 0
|
||||
|
||||
@@ -1,517 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep
|
||||
|
||||
import com.google.protobuf.ByteString
|
||||
import com.google.protobuf.MessageLite
|
||||
import net.jpountz.lz4.LZ4Factory
|
||||
import net.syncthing.java.bep.BlockExchangeProtos.*
|
||||
import net.syncthing.java.client.protocol.rp.RelayClient
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.beans.DeviceInfo
|
||||
import net.syncthing.java.core.beans.FolderInfo
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.core.interfaces.TempRepository
|
||||
import net.syncthing.java.core.security.KeystoreHandler
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import net.syncthing.java.core.utils.submitLogging
|
||||
import net.syncthing.java.httprelay.HttpRelayClient
|
||||
import org.apache.commons.io.IOUtils
|
||||
import org.apache.commons.lang3.tuple.Pair
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.Closeable
|
||||
import java.io.DataInputStream
|
||||
import java.io.DataOutputStream
|
||||
import java.io.IOException
|
||||
import java.lang.reflect.InvocationTargetException
|
||||
import java.nio.ByteBuffer
|
||||
import java.security.cert.CertificateException
|
||||
import java.util.concurrent.ConcurrentHashMap
|
||||
import java.util.concurrent.Executors
|
||||
import java.util.concurrent.Future
|
||||
import java.util.concurrent.TimeUnit
|
||||
import javax.net.ssl.SSLSocket
|
||||
|
||||
class ConnectionHandler(private val configuration: Configuration, val address: DeviceAddress,
|
||||
private val indexHandler: IndexHandler,
|
||||
private val tempRepository: TempRepository,
|
||||
private val onNewFolderSharedListener: (ConnectionHandler, FolderInfo) -> Unit,
|
||||
private val onConnectionChangedListener: (ConnectionHandler) -> Unit) : Closeable {
|
||||
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
|
||||
private val outExecutorService = Executors.newSingleThreadExecutor()
|
||||
private val inExecutorService = Executors.newSingleThreadExecutor()
|
||||
private val messageProcessingService = Executors.newCachedThreadPool()
|
||||
private val periodicExecutorService = Executors.newSingleThreadScheduledExecutor()
|
||||
private lateinit var socket: SSLSocket
|
||||
private var inputStream: DataInputStream? = null
|
||||
private var outputStream: DataOutputStream? = null
|
||||
private var lastActive = Long.MIN_VALUE
|
||||
internal var clusterConfigInfo: ClusterConfigInfo? = null
|
||||
private set
|
||||
private val clusterConfigWaitingLock = Object()
|
||||
private val responseHandler = ResponseHandler()
|
||||
private val blockPuller = BlockPuller(this, indexHandler, responseHandler, tempRepository)
|
||||
private val blockPusher = BlockPusher(configuration.localDeviceId, this, indexHandler)
|
||||
private val onRequestMessageReceivedListeners = mutableSetOf<(Request) -> Unit>()
|
||||
private var isClosed = false
|
||||
var isConnected = false
|
||||
private set
|
||||
|
||||
fun deviceId(): DeviceId = address.deviceId()
|
||||
|
||||
private fun checkNotClosed() {
|
||||
NetworkUtils.assertProtocol(!isClosed, {"connection $this closed"})
|
||||
}
|
||||
|
||||
internal fun registerOnRequestMessageReceivedListeners(listener: (Request) -> Unit) {
|
||||
onRequestMessageReceivedListeners.add(listener)
|
||||
}
|
||||
|
||||
internal fun unregisterOnRequestMessageReceivedListeners(listener: (Request) -> Unit) {
|
||||
assert(onRequestMessageReceivedListeners.contains(listener))
|
||||
onRequestMessageReceivedListeners.remove(listener)
|
||||
}
|
||||
|
||||
@Throws(IOException::class, KeystoreHandler.CryptoException::class)
|
||||
fun connect(): ConnectionHandler {
|
||||
checkNotClosed()
|
||||
assert(!isConnected, {"already connected!"})
|
||||
logger.info("connecting to {}", address.address)
|
||||
|
||||
val keystoreHandler = KeystoreHandler.Loader().loadKeystore(configuration)
|
||||
|
||||
socket = when (address.getType()) {
|
||||
DeviceAddress.AddressType.TCP -> {
|
||||
logger.debug("opening tcp ssl connection")
|
||||
keystoreHandler.createSocket(address.getSocketAddress(), KeystoreHandler.BEP)
|
||||
}
|
||||
DeviceAddress.AddressType.RELAY -> {
|
||||
logger.debug("opening relay connection")
|
||||
keystoreHandler.wrapSocket(RelayClient(configuration).openRelayConnection(address), KeystoreHandler.BEP)
|
||||
}
|
||||
DeviceAddress.AddressType.HTTP_RELAY, DeviceAddress.AddressType.HTTPS_RELAY -> {
|
||||
logger.debug("opening http relay connection")
|
||||
keystoreHandler.wrapSocket(HttpRelayClient().openRelayConnection(address), KeystoreHandler.BEP)
|
||||
}
|
||||
else -> throw UnsupportedOperationException("unsupported address type = " + address.getType())
|
||||
}
|
||||
inputStream = DataInputStream(socket.inputStream)
|
||||
outputStream = DataOutputStream(socket.outputStream)
|
||||
|
||||
sendHelloMessage(BlockExchangeProtos.Hello.newBuilder()
|
||||
.setClientName(configuration.clientName)
|
||||
.setClientVersion(configuration.clientVersion)
|
||||
.setDeviceName(configuration.localDeviceName)
|
||||
.build().toByteArray())
|
||||
markActivityOnSocket()
|
||||
|
||||
receiveHelloMessage()
|
||||
try {
|
||||
keystoreHandler.checkSocketCertificate(socket, address.deviceId())
|
||||
} catch (e: CertificateException) {
|
||||
throw IOException(e)
|
||||
}
|
||||
|
||||
run {
|
||||
val clusterConfigBuilder = ClusterConfig.newBuilder()
|
||||
for (folder in configuration.folders) {
|
||||
val folderBuilder = Folder.newBuilder()
|
||||
.setId(folder.folderId)
|
||||
.setLabel(folder.label)
|
||||
run {
|
||||
//our device
|
||||
val deviceBuilder = Device.newBuilder()
|
||||
.setId(ByteString.copyFrom(configuration.localDeviceId.toHashData()))
|
||||
.setIndexId(indexHandler.sequencer().indexId())
|
||||
.setMaxSequence(indexHandler.sequencer().currentSequence())
|
||||
folderBuilder.addDevices(deviceBuilder)
|
||||
}
|
||||
run {
|
||||
//other device
|
||||
val deviceBuilder = Device.newBuilder()
|
||||
.setId(ByteString.copyFrom(DeviceId(address.deviceId).toHashData()))
|
||||
val indexSequenceInfo = indexHandler.indexRepository.findIndexInfoByDeviceAndFolder(address.deviceId(), folder.folderId)
|
||||
indexSequenceInfo?.let {
|
||||
deviceBuilder
|
||||
.setIndexId(indexSequenceInfo.indexId)
|
||||
.setMaxSequence(indexSequenceInfo.localSequence)
|
||||
logger.info("send delta index info device = {} index = {} max (local) sequence = {}",
|
||||
indexSequenceInfo.deviceId,
|
||||
indexSequenceInfo.indexId,
|
||||
indexSequenceInfo.localSequence)
|
||||
}
|
||||
folderBuilder.addDevices(deviceBuilder)
|
||||
}
|
||||
clusterConfigBuilder.addFolders(folderBuilder)
|
||||
//TODO other devices??
|
||||
}
|
||||
sendMessage(clusterConfigBuilder.build())
|
||||
}
|
||||
synchronized(clusterConfigWaitingLock) {
|
||||
startMessageListenerService()
|
||||
while (clusterConfigInfo == null && !isClosed) {
|
||||
logger.debug("wait for cluster config")
|
||||
try {
|
||||
clusterConfigWaitingLock.wait()
|
||||
} catch (e: InterruptedException) {
|
||||
throw IOException(e)
|
||||
}
|
||||
}
|
||||
if (clusterConfigInfo == null) {
|
||||
throw IOException("unable to retrieve cluster config from peer!")
|
||||
}
|
||||
}
|
||||
for (folder in configuration.folders) {
|
||||
if (hasFolder(folder.folderId)) {
|
||||
sendIndexMessage(folder.folderId)
|
||||
}
|
||||
}
|
||||
periodicExecutorService.scheduleWithFixedDelay({ this.sendPing() }, 90, 90, TimeUnit.SECONDS)
|
||||
isConnected = true
|
||||
onConnectionChangedListener(this)
|
||||
return this
|
||||
}
|
||||
|
||||
fun getBlockPuller(): BlockPuller {
|
||||
return blockPuller
|
||||
}
|
||||
|
||||
fun getBlockPusher(): BlockPusher {
|
||||
return blockPusher
|
||||
}
|
||||
|
||||
private fun sendIndexMessage(folderId: String) {
|
||||
sendMessage(Index.newBuilder()
|
||||
.setFolder(folderId)
|
||||
.build())
|
||||
}
|
||||
|
||||
fun closeBg() {
|
||||
Thread { close() }.start()
|
||||
}
|
||||
|
||||
/**
|
||||
* Receive hello message and save device name to configuration.
|
||||
*/
|
||||
@Throws(IOException::class)
|
||||
private fun receiveHelloMessage() {
|
||||
val magic = inputStream!!.readInt()
|
||||
NetworkUtils.assertProtocol(magic == MAGIC, {"magic mismatch, expected $MAGIC, got $magic"})
|
||||
val length = inputStream!!.readShort().toInt()
|
||||
NetworkUtils.assertProtocol(length > 0, {"invalid lenght, must be >0, got $length"})
|
||||
val buffer = ByteArray(length)
|
||||
inputStream!!.readFully(buffer)
|
||||
val hello = BlockExchangeProtos.Hello.parseFrom(buffer)
|
||||
logger.info("Received hello message, deviceName=${hello.deviceName}, clientName=${hello.clientName}, clientVersion=${hello.clientVersion}")
|
||||
configuration.peers = configuration.peers.map { peer ->
|
||||
if (peer.deviceId == deviceId()) {
|
||||
DeviceInfo(deviceId(), hello.deviceName)
|
||||
} else {
|
||||
peer
|
||||
}
|
||||
}.toSet()
|
||||
configuration.persistLater()
|
||||
}
|
||||
|
||||
private fun sendHelloMessage(payload: ByteArray): Future<*> {
|
||||
return outExecutorService.submitLogging {
|
||||
try {
|
||||
logger.debug("Sending hello message")
|
||||
val header = ByteBuffer.allocate(6)
|
||||
header.putInt(MAGIC)
|
||||
header.putShort(payload.size.toShort())
|
||||
outputStream!!.write(header.array())
|
||||
outputStream!!.write(payload)
|
||||
outputStream!!.flush()
|
||||
} catch (ex: IOException) {
|
||||
if (outExecutorService.isShutdown) {
|
||||
return@submitLogging
|
||||
}
|
||||
logger.error("error writing to output stream", ex)
|
||||
closeBg()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun sendPing(): Future<*> {
|
||||
return sendMessage(Ping.newBuilder().build())
|
||||
}
|
||||
|
||||
private fun markActivityOnSocket() {
|
||||
lastActive = System.currentTimeMillis()
|
||||
}
|
||||
|
||||
@Throws(IOException::class)
|
||||
private fun receiveMessage(): Pair<BlockExchangeProtos.MessageType, MessageLite> {
|
||||
var headerLength = inputStream!!.readShort().toInt()
|
||||
while (headerLength == 0) {
|
||||
logger.warn("got headerLength == 0, skipping short")
|
||||
headerLength = inputStream!!.readShort().toInt()
|
||||
}
|
||||
markActivityOnSocket()
|
||||
NetworkUtils.assertProtocol(headerLength > 0, {"invalid lenght, must be >0, got $headerLength"})
|
||||
val headerBuffer = ByteArray(headerLength)
|
||||
inputStream!!.readFully(headerBuffer)
|
||||
val header = BlockExchangeProtos.Header.parseFrom(headerBuffer)
|
||||
var messageLength = 0
|
||||
while (messageLength == 0) {
|
||||
logger.warn("received readInt() == 0, expecting 'bep message header length' (int >0), ignoring (keepalive?)")
|
||||
messageLength = inputStream!!.readInt()
|
||||
}
|
||||
NetworkUtils.assertProtocol(messageLength >= 0, {"invalid lenght, must be >=0, got $messageLength"})
|
||||
var messageBuffer = ByteArray(messageLength)
|
||||
inputStream!!.readFully(messageBuffer)
|
||||
markActivityOnSocket()
|
||||
if (header.compression == BlockExchangeProtos.MessageCompression.LZ4) {
|
||||
val uncompressedLength = ByteBuffer.wrap(messageBuffer).int
|
||||
messageBuffer = LZ4Factory.fastestInstance().fastDecompressor().decompress(messageBuffer, 4, uncompressedLength)
|
||||
}
|
||||
val messageTypeInfo = messageTypesByProtoMessageType[header.type]
|
||||
NetworkUtils.assertProtocol(messageTypeInfo != null, {"unsupported message type = ${header.type}"})
|
||||
try {
|
||||
val message = messageTypeInfo!!.parseFrom(messageBuffer)
|
||||
return Pair.of(header.type, message)
|
||||
} catch (e: Exception) {
|
||||
when (e) {
|
||||
is IllegalAccessException, is IllegalArgumentException, is InvocationTargetException, is NoSuchMethodException, is SecurityException ->
|
||||
throw IOException(e)
|
||||
else -> throw e
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
internal fun sendMessage(message: MessageLite): Future<*> {
|
||||
checkNotClosed()
|
||||
val messageTypeInfo = messageTypesByJavaClass[message.javaClass]
|
||||
messageTypeInfo!!
|
||||
val header = BlockExchangeProtos.Header.newBuilder()
|
||||
.setCompression(BlockExchangeProtos.MessageCompression.NONE)
|
||||
// invert map
|
||||
.setType(messageTypeInfo.protoMessageType)
|
||||
.build()
|
||||
val headerData = header.toByteArray()
|
||||
val messageData = message.toByteArray() //TODO compression
|
||||
return outExecutorService.submit<Any> {
|
||||
try {
|
||||
logger.debug("sending message type = {} {}", header.type, getIdForMessage(message))
|
||||
markActivityOnSocket()
|
||||
outputStream!!.writeShort(headerData.size)
|
||||
outputStream!!.write(headerData)
|
||||
outputStream!!.writeInt(messageData.size)//with compression, check this
|
||||
outputStream!!.write(messageData)
|
||||
outputStream!!.flush()
|
||||
markActivityOnSocket()
|
||||
} catch (ex: IOException) {
|
||||
if (!outExecutorService.isShutdown) {
|
||||
logger.error("error writing to output stream", ex)
|
||||
closeBg()
|
||||
}
|
||||
throw ex
|
||||
}
|
||||
|
||||
null
|
||||
}
|
||||
}
|
||||
|
||||
override fun close() {
|
||||
if (!isClosed) {
|
||||
sendMessage(Close.getDefaultInstance())
|
||||
isClosed = true
|
||||
isConnected = false
|
||||
periodicExecutorService.shutdown()
|
||||
outExecutorService.shutdown()
|
||||
inExecutorService.shutdown()
|
||||
messageProcessingService.shutdown()
|
||||
assert(onRequestMessageReceivedListeners.isEmpty())
|
||||
if (outputStream != null) {
|
||||
IOUtils.closeQuietly(outputStream)
|
||||
outputStream = null
|
||||
}
|
||||
if (inputStream != null) {
|
||||
IOUtils.closeQuietly(inputStream)
|
||||
inputStream = null
|
||||
}
|
||||
try {
|
||||
IOUtils.closeQuietly(socket)
|
||||
} catch (ex: Exception) {
|
||||
// ignore this
|
||||
// this can throw an exception if socket was not yet initialized/ set
|
||||
// as Kotlin does an check about this, the closeQuietly does not catch it
|
||||
}
|
||||
logger.info("closed connection {}", address)
|
||||
synchronized(clusterConfigWaitingLock) {
|
||||
clusterConfigWaitingLock.notifyAll()
|
||||
}
|
||||
onConnectionChangedListener(this)
|
||||
try {
|
||||
periodicExecutorService.awaitTermination(2, TimeUnit.SECONDS)
|
||||
outExecutorService.awaitTermination(2, TimeUnit.SECONDS)
|
||||
inExecutorService.awaitTermination(2, TimeUnit.SECONDS)
|
||||
messageProcessingService.awaitTermination(2, TimeUnit.SECONDS)
|
||||
} catch (ex: InterruptedException) {
|
||||
logger.warn("", ex)
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* return time elapsed since last activity on socket, inputStream millis
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
fun getLastActive(): Long {
|
||||
return System.currentTimeMillis() - lastActive
|
||||
}
|
||||
|
||||
private fun startMessageListenerService() {
|
||||
inExecutorService.submitLogging {
|
||||
try {
|
||||
while (!Thread.interrupted()) {
|
||||
val message = receiveMessage()
|
||||
messageProcessingService.submitLogging {
|
||||
logger.debug("received message type = {} {}", message.left, getIdForMessage(message.right))
|
||||
when (message.left) {
|
||||
BlockExchangeProtos.MessageType.INDEX -> {
|
||||
val index = message.value as Index
|
||||
indexHandler.handleIndexMessageReceivedEvent(index.folder, index.filesList, this)
|
||||
}
|
||||
BlockExchangeProtos.MessageType.INDEX_UPDATE -> {
|
||||
val update = message.value as IndexUpdate
|
||||
indexHandler.handleIndexMessageReceivedEvent(update.folder, update.filesList, this)
|
||||
}
|
||||
BlockExchangeProtos.MessageType.REQUEST -> {
|
||||
onRequestMessageReceivedListeners.forEach { it(message.value as Request) }
|
||||
}
|
||||
BlockExchangeProtos.MessageType.RESPONSE -> {
|
||||
responseHandler.handleResponse(message.value as Response)
|
||||
}
|
||||
BlockExchangeProtos.MessageType.PING -> logger.debug("ping message received")
|
||||
BlockExchangeProtos.MessageType.CLOSE -> {
|
||||
val close = message.value as BlockExchangeProtos.Close
|
||||
logger.info("received close message, reason=${close.reason}")
|
||||
closeBg()
|
||||
}
|
||||
BlockExchangeProtos.MessageType.CLUSTER_CONFIG -> {
|
||||
NetworkUtils.assertProtocol(clusterConfigInfo == null, {"received cluster config message twice!"})
|
||||
clusterConfigInfo = ClusterConfigInfo()
|
||||
val clusterConfig = message.value as ClusterConfig
|
||||
for (folder in clusterConfig.foldersList ?: emptyList()) {
|
||||
val folderInfo = ClusterConfigFolderInfo(folder.id, folder.label)
|
||||
val devicesById = (folder.devicesList ?: emptyList())
|
||||
.associateBy { input ->
|
||||
DeviceId.fromHashData(input.id!!.toByteArray())
|
||||
}
|
||||
val otherDevice = devicesById[address.deviceId()]
|
||||
val ourDevice = devicesById[configuration.localDeviceId]
|
||||
if (otherDevice != null) {
|
||||
folderInfo.isAnnounced = true
|
||||
}
|
||||
if (ourDevice != null) {
|
||||
folderInfo.isShared = true
|
||||
logger.info("folder shared from device = {} folder = {}", address.deviceId, folderInfo)
|
||||
val folderIds = configuration.folders.map { it.folderId }
|
||||
if (!folderIds.contains(folderInfo.folderId)) {
|
||||
val fi = FolderInfo(folderInfo.folderId, folderInfo.label)
|
||||
configuration.folders = configuration.folders + fi
|
||||
onNewFolderSharedListener(this, fi)
|
||||
logger.info("new folder shared = {}", folderInfo)
|
||||
}
|
||||
} else {
|
||||
logger.info("folder not shared from device = {} folder = {}", address.deviceId, folderInfo)
|
||||
}
|
||||
clusterConfigInfo!!.putFolderInfo(folderInfo)
|
||||
}
|
||||
configuration.persistLater()
|
||||
indexHandler.handleClusterConfigMessageProcessedEvent(clusterConfig)
|
||||
synchronized(clusterConfigWaitingLock) {
|
||||
clusterConfigWaitingLock.notifyAll()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (ex: IOException) {
|
||||
if (inExecutorService.isShutdown) {
|
||||
return@submitLogging
|
||||
}
|
||||
logger.error("error receiving message", ex)
|
||||
closeBg()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
override fun toString(): String {
|
||||
return "ConnectionHandler{" + "address=" + address + ", lastActive=" + getLastActive() / 1000.0 + "secs ago}"
|
||||
}
|
||||
|
||||
internal inner class ClusterConfigInfo {
|
||||
|
||||
private val folderInfoById = ConcurrentHashMap<String, ClusterConfigFolderInfo>()
|
||||
|
||||
fun getSharedFolders(): Set<String> = folderInfoById.values.filter { it.isShared }.map { it.folderId }.toSet()
|
||||
|
||||
fun putFolderInfo(folderInfo: ClusterConfigFolderInfo) {
|
||||
folderInfoById[folderInfo.folderId] = folderInfo
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
fun hasFolder(folder: String): Boolean {
|
||||
return clusterConfigInfo!!.getSharedFolders().contains(folder)
|
||||
}
|
||||
|
||||
companion object {
|
||||
|
||||
private const val MAGIC = 0x2EA7D90B
|
||||
|
||||
private val messageTypes = listOf(
|
||||
MessageTypeInfo(MessageType.CLOSE, Close::class.java) { Close.parseFrom(it) },
|
||||
MessageTypeInfo(MessageType.CLUSTER_CONFIG, ClusterConfig::class.java) { ClusterConfig.parseFrom(it) },
|
||||
MessageTypeInfo(MessageType.DOWNLOAD_PROGRESS, DownloadProgress::class.java) { DownloadProgress.parseFrom(it) },
|
||||
MessageTypeInfo(MessageType.INDEX, Index::class.java) { Index.parseFrom(it) },
|
||||
MessageTypeInfo(MessageType.INDEX_UPDATE, IndexUpdate::class.java) { IndexUpdate.parseFrom(it) },
|
||||
MessageTypeInfo(MessageType.PING, Ping::class.java) { Ping.parseFrom(it) },
|
||||
MessageTypeInfo(MessageType.REQUEST, Request::class.java) { Request.parseFrom(it) },
|
||||
MessageTypeInfo(MessageType.RESPONSE, Response::class.java) { Response.parseFrom(it) }
|
||||
)
|
||||
|
||||
private val messageTypesByProtoMessageType = messageTypes.map { it.protoMessageType to it }.toMap()
|
||||
private val messageTypesByJavaClass = messageTypes.map { it.javaClass to it }.toMap()
|
||||
|
||||
/**
|
||||
* get id for message bean/instance, for log tracking
|
||||
*
|
||||
* @param message
|
||||
* @return id for message bean
|
||||
*/
|
||||
private fun getIdForMessage(message: MessageLite): String {
|
||||
return when (message) {
|
||||
is Request -> Integer.toString(message.id)
|
||||
is Response -> Integer.toString(message.id)
|
||||
else -> Integer.toString(Math.abs(message.hashCode()))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
data class MessageTypeInfo(
|
||||
val protoMessageType: MessageType,
|
||||
val javaClass: Class<out MessageLite>,
|
||||
val parseFrom: (data: ByteArray) -> MessageLite
|
||||
)
|
||||
}
|
||||
@@ -55,7 +55,7 @@ class IndexBrowser internal constructor(private val indexRepository: IndexReposi
|
||||
}
|
||||
}
|
||||
|
||||
internal fun onIndexChangedevent(folder: String, newRecord: FileInfo) {
|
||||
internal fun onIndexChangedevent(folder: String) {
|
||||
if (folder == this.folder) {
|
||||
preloadFileInfoForCurrentPath()
|
||||
}
|
||||
|
||||
@@ -13,18 +13,18 @@
|
||||
*/
|
||||
package net.syncthing.java.bep
|
||||
|
||||
import net.syncthing.java.bep.connectionactor.ClusterConfigInfo
|
||||
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
|
||||
import net.syncthing.java.core.beans.*
|
||||
import net.syncthing.java.core.beans.FileInfo.Version
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.core.interfaces.IndexRepository
|
||||
import net.syncthing.java.core.interfaces.Sequencer
|
||||
import net.syncthing.java.core.interfaces.TempRepository
|
||||
import net.syncthing.java.core.utils.BlockUtils
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import net.syncthing.java.core.utils.awaitTerminationSafe
|
||||
import net.syncthing.java.core.utils.submitLogging
|
||||
import net.syncthing.java.core.utils.trySubmitLogging
|
||||
import org.apache.commons.lang3.tuple.Pair
|
||||
import org.apache.http.util.TextUtils
|
||||
import org.bouncycastle.util.encoders.Hex
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.Closeable
|
||||
@@ -95,9 +95,9 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
}
|
||||
}
|
||||
|
||||
internal fun isRemoteIndexAcquired(clusterConfigInfo: ConnectionHandler.ClusterConfigInfo, peerDeviceId: DeviceId): Boolean {
|
||||
internal fun isRemoteIndexAcquired(clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId): Boolean {
|
||||
var ready = true
|
||||
for (folder in clusterConfigInfo.getSharedFolders()) {
|
||||
for (folder in clusterConfigInfo.sharedFolderIds) {
|
||||
val indexSequenceInfo = indexRepository.findIndexInfoByDeviceAndFolder(peerDeviceId, folder)
|
||||
if (indexSequenceInfo == null || indexSequenceInfo.localSequence < indexSequenceInfo.maxSequence) {
|
||||
logger.debug("waiting for index on folder = {} sequenceInfo = {}", folder, indexSequenceInfo)
|
||||
@@ -108,12 +108,12 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
}
|
||||
|
||||
@Throws(InterruptedException::class)
|
||||
fun waitForRemoteIndexAcquired(connectionHandler: ConnectionHandler, timeoutSecs: Long? = null): IndexHandler {
|
||||
fun waitForRemoteIndexAcquired(connectionHandler: ConnectionActorWrapper, timeoutSecs: Long? = null): IndexHandler {
|
||||
val timeoutMillis = (timeoutSecs ?: DEFAULT_INDEX_TIMEOUT) * 1000
|
||||
synchronized(indexWaitLock) {
|
||||
while (!isRemoteIndexAcquired(connectionHandler.clusterConfigInfo!!, connectionHandler.deviceId())) {
|
||||
while (!isRemoteIndexAcquired(connectionHandler.getClusterConfig(), connectionHandler.deviceId)) {
|
||||
indexWaitLock.wait(timeoutMillis)
|
||||
NetworkUtils.assertProtocol(connectionHandler.getLastActive() < timeoutMillis || lastActive() < timeoutMillis,
|
||||
NetworkUtils.assertProtocol(/* TODO connectionHandler.getLastActive() < timeoutMillis || */ lastActive() < timeoutMillis,
|
||||
{"unable to acquire index from connection $connectionHandler, timeout reached!"})
|
||||
}
|
||||
}
|
||||
@@ -138,8 +138,8 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
}
|
||||
}
|
||||
|
||||
fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, connectionHandler: ConnectionHandler) {
|
||||
indexMessageProcessor.handleIndexMessageReceivedEvent(folderId, filesList, connectionHandler)
|
||||
internal fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId) {
|
||||
indexMessageProcessor.handleIndexMessageReceivedEvent(folderId, filesList, clusterConfigInfo, peerDeviceId)
|
||||
}
|
||||
|
||||
fun pushRecord(folder: String, bepFileInfo: BlockExchangeProtos.FileInfo): FileInfo? {
|
||||
@@ -209,14 +209,14 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
private fun addRecord(record: FileInfo, fileBlocks: FileBlocks?): FileInfo? {
|
||||
synchronized(writeAccessLock) {
|
||||
val lastModified = indexRepository.findFileInfoLastModified(record.folder, record.path)
|
||||
return if (lastModified != null && !record.lastModified.after(lastModified)) {
|
||||
return if (lastModified != null && record.lastModified < lastModified) {
|
||||
logger.trace("discarding record = {}, modified before local record", record)
|
||||
null
|
||||
} else {
|
||||
indexRepository.updateFileInfo(record, fileBlocks)
|
||||
logger.trace("loaded new record = {}", record)
|
||||
indexBrowsers.forEach {
|
||||
it.onIndexChangedevent(record.folder, record)
|
||||
it.onIndexChangedevent(record.folder)
|
||||
}
|
||||
record
|
||||
}
|
||||
@@ -234,9 +234,9 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
} else {
|
||||
assert(fileInfo.isFile())
|
||||
val fileBlocks = indexRepository.findFileBlocks(folder, path)
|
||||
checkNotNull(fileBlocks, {"file blocks not found for file info = $fileInfo"})
|
||||
checkNotNull(fileBlocks) {"file blocks not found for file info = $fileInfo"}
|
||||
|
||||
FileInfo.checkBlocks(fileInfo, fileBlocks!!)
|
||||
FileInfo.checkBlocks(fileInfo, fileBlocks)
|
||||
|
||||
Pair.of(fileInfo, fileBlocks)
|
||||
}
|
||||
@@ -244,7 +244,7 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
|
||||
private fun updateFolderInfo(folder: String, label: String?): FolderInfo {
|
||||
var folderInfo: FolderInfo? = folderInfoByFolder[folder]
|
||||
if (folderInfo == null || !TextUtils.isEmpty(label)) {
|
||||
if (folderInfo == null || label.isNullOrEmpty()) {
|
||||
folderInfo = FolderInfo(folder, label)
|
||||
folderInfoByFolder.put(folderInfo.folderId, folderInfo)
|
||||
}
|
||||
@@ -293,11 +293,9 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
// private final int MIN_DELAY = 0, MAX_DELAY = 5000, MAX_RECORD_PER_PROCESS = 16, DELAY_FACTOR = 1;
|
||||
private var startTime: Long? = null
|
||||
|
||||
fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, connectionHandler: ConnectionHandler) {
|
||||
fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId) {
|
||||
logger.info("received index message event, preparing (queued records = {} event record count = {})", queuedRecords, filesList.size)
|
||||
markActive()
|
||||
val clusterConfigInfo = connectionHandler.clusterConfigInfo
|
||||
val peerDeviceId = connectionHandler.deviceId()
|
||||
// List<BlockExchangeProtos.FileInfo> fileList = event.getFilesList();
|
||||
// for (int index = 0; index < fileList.size(); index += MAX_RECORD_PER_PROCESS) {
|
||||
// BlockExchangeProtos.IndexUpdate data = BlockExchangeProtos.IndexUpdate.newBuilder()
|
||||
@@ -321,23 +319,23 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
}
|
||||
}
|
||||
|
||||
private fun processBg(data: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ConnectionHandler.ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
private fun processBg(data: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
logger.debug("received index message event, queuing for processing")
|
||||
queuedMessages++
|
||||
queuedRecords += data.filesCount.toLong()
|
||||
executorService.submitLogging(object : ProcessingRunnable() {
|
||||
executorService.trySubmitLogging(object : ProcessingRunnable() {
|
||||
override fun runProcess() {
|
||||
doHandleIndexMessageReceivedEvent(data, clusterConfigInfo, peerDeviceId)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
private fun storeAndProcessBg(data: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ConnectionHandler.ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
private fun storeAndProcessBg(data: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
val key = tempRepository.pushTempData(data.toByteArray())
|
||||
logger.debug("received index message event, stored to temp record {}, queuing for processing", key)
|
||||
queuedMessages++
|
||||
queuedRecords += data.filesCount.toLong()
|
||||
executorService.submitLogging(object : ProcessingRunnable() {
|
||||
executorService.trySubmitLogging(object : ProcessingRunnable() {
|
||||
override fun runProcess() {
|
||||
try {
|
||||
doHandleIndexMessageReceivedEvent(key, clusterConfigInfo, peerDeviceId)
|
||||
@@ -370,7 +368,7 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
// return fileSequence < localSequence;
|
||||
// }
|
||||
@Throws(IOException::class)
|
||||
protected fun doHandleIndexMessageReceivedEvent(key: String, clusterConfigInfo: ConnectionHandler.ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
protected fun doHandleIndexMessageReceivedEvent(key: String, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
logger.debug("processing index message event from temp record {}", key)
|
||||
markActive()
|
||||
val data = tempRepository.popTempData(key)
|
||||
@@ -378,7 +376,7 @@ class IndexHandler(private val configuration: Configuration, val indexRepository
|
||||
doHandleIndexMessageReceivedEvent(message, clusterConfigInfo, peerDeviceId)
|
||||
}
|
||||
|
||||
protected fun doHandleIndexMessageReceivedEvent(message: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ConnectionHandler.ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
protected fun doHandleIndexMessageReceivedEvent(message: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
|
||||
// synchronized (writeAccessLock) {
|
||||
// if (addProcessingDelayForInterface) {
|
||||
// delay = Math.min(MAX_DELAY, Math.max(MIN_DELAY, lastRecordProcessingTime * DELAY_FACTOR));
|
||||
|
||||
@@ -0,0 +1,49 @@
|
||||
/*
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep
|
||||
|
||||
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
|
||||
import java.io.IOException
|
||||
import java.util.*
|
||||
|
||||
class MultiConnectionHelper (
|
||||
initialConnections: List<ConnectionActorWrapper>,
|
||||
private val connectionFilter: (ConnectionActorWrapper) -> Boolean
|
||||
) {
|
||||
companion object {
|
||||
private val random = Random()
|
||||
}
|
||||
|
||||
private val usableConnections = initialConnections.toMutableList()
|
||||
|
||||
fun pickConnection(): ConnectionActorWrapper {
|
||||
val possibleConnections = synchronized(usableConnections) {
|
||||
usableConnections.filter { it.isConnected and connectionFilter(it) }
|
||||
}
|
||||
|
||||
if (possibleConnections.isEmpty()) {
|
||||
throw IOException("no matching connection is available")
|
||||
} else if (possibleConnections.size == 1) {
|
||||
return possibleConnections.first()
|
||||
} else {
|
||||
return possibleConnections[random.nextInt(possibleConnections.size)]
|
||||
}
|
||||
}
|
||||
|
||||
fun disableConnection(wrapper: ConnectionActorWrapper) {
|
||||
synchronized(usableConnections) {
|
||||
usableConnections.remove(wrapper)
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,54 @@
|
||||
package net.syncthing.java.bep
|
||||
|
||||
import kotlinx.coroutines.Deferred
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import java.io.IOException
|
||||
|
||||
class RequestHandlerRegistry {
|
||||
private val listeners = mutableMapOf<RequestHandlerFilter, (BlockExchangeProtos.Request) -> Deferred<BlockExchangeProtos.Response>>()
|
||||
|
||||
suspend fun handleRequest(source: DeviceId, request: BlockExchangeProtos.Request): BlockExchangeProtos.Response {
|
||||
val rule = RequestHandlerFilter(
|
||||
deviceId = source,
|
||||
folderId = request.folder,
|
||||
path = request.name
|
||||
)
|
||||
|
||||
val matchingListener = synchronized(listeners) {
|
||||
listeners[rule]
|
||||
}
|
||||
|
||||
if (matchingListener != null) {
|
||||
return matchingListener(request).await()
|
||||
} else {
|
||||
return BlockExchangeProtos.Response.newBuilder()
|
||||
.setId(request.id)
|
||||
.setCode(BlockExchangeProtos.ErrorCode.GENERIC)
|
||||
.build()
|
||||
}
|
||||
}
|
||||
|
||||
fun registerListener(filter: RequestHandlerFilter, listener: (BlockExchangeProtos.Request) -> Deferred<BlockExchangeProtos.Response>) {
|
||||
synchronized(listeners) {
|
||||
val oldListener = listeners[filter]
|
||||
|
||||
if (oldListener != null) {
|
||||
throw IOException("there is already an listener for this filter")
|
||||
}
|
||||
|
||||
listeners[filter] = listener
|
||||
}
|
||||
}
|
||||
|
||||
fun unregisterListener(filter: RequestHandlerFilter) {
|
||||
synchronized(listeners) {
|
||||
listeners.remove(filter)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
data class RequestHandlerFilter(
|
||||
val deviceId: DeviceId,
|
||||
val folderId: String,
|
||||
val path: String
|
||||
)
|
||||
@@ -1,50 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep
|
||||
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.util.*
|
||||
import java.util.concurrent.atomic.AtomicInteger
|
||||
import kotlin.collections.HashMap
|
||||
|
||||
class ResponseHandler {
|
||||
companion object {
|
||||
private val logger = LoggerFactory.getLogger(ResponseHandler::class.java)
|
||||
}
|
||||
|
||||
private val responseListeners = Collections.synchronizedMap(HashMap<Int, (BlockExchangeProtos.Response) -> Unit>())
|
||||
private val nextRequestId = AtomicInteger(0)
|
||||
|
||||
fun registerListener(listener: (BlockExchangeProtos.Response) -> Unit): Int {
|
||||
val requestId = nextRequestId.getAndIncrement()
|
||||
|
||||
responseListeners[requestId] = listener
|
||||
|
||||
return requestId
|
||||
}
|
||||
|
||||
fun unregisterListener(requestId: Int) {
|
||||
responseListeners.remove(requestId)
|
||||
}
|
||||
|
||||
fun handleResponse(response: BlockExchangeProtos.Response) {
|
||||
val listener = responseListeners.remove(response.id)
|
||||
|
||||
if (listener != null) {
|
||||
listener(response)
|
||||
} else {
|
||||
logger.warn("received response for {} without associated handler", response.id)
|
||||
}
|
||||
}
|
||||
}
|
||||
+139
@@ -0,0 +1,139 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import com.google.protobuf.ByteString
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
import net.syncthing.java.bep.IndexHandler
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.beans.FolderInfo
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import org.slf4j.LoggerFactory
|
||||
|
||||
object ClusterConfigHandler {
|
||||
private val logger = LoggerFactory.getLogger(ClusterConfigHandler::class.java)
|
||||
|
||||
fun buildClusterConfig(
|
||||
configuration: Configuration,
|
||||
indexHandler: IndexHandler,
|
||||
deviceId: DeviceId
|
||||
): BlockExchangeProtos.ClusterConfig {
|
||||
val builder = BlockExchangeProtos.ClusterConfig.newBuilder()
|
||||
|
||||
for (folder in configuration.folders) {
|
||||
val folderBuilder = BlockExchangeProtos.Folder.newBuilder()
|
||||
.setId(folder.folderId)
|
||||
.setLabel(folder.label)
|
||||
|
||||
// add this device
|
||||
folderBuilder.addDevices(
|
||||
BlockExchangeProtos.Device.newBuilder()
|
||||
.setId(ByteString.copyFrom(configuration.localDeviceId.toHashData()))
|
||||
.setIndexId(indexHandler.sequencer().indexId())
|
||||
.setMaxSequence(indexHandler.sequencer().currentSequence())
|
||||
)
|
||||
|
||||
// add other device
|
||||
val indexSequenceInfo = indexHandler.indexRepository.findIndexInfoByDeviceAndFolder(deviceId, folder.folderId)
|
||||
|
||||
folderBuilder.addDevices(
|
||||
BlockExchangeProtos.Device.newBuilder()
|
||||
.setId(ByteString.copyFrom(deviceId.toHashData()))
|
||||
.apply {
|
||||
indexSequenceInfo?.let {
|
||||
setIndexId(indexSequenceInfo.indexId)
|
||||
setMaxSequence(indexSequenceInfo.localSequence)
|
||||
|
||||
logger.info("send delta index info device = {} index = {} max (local) sequence = {}",
|
||||
indexSequenceInfo.deviceId,
|
||||
indexSequenceInfo.indexId,
|
||||
indexSequenceInfo.localSequence)
|
||||
}
|
||||
}
|
||||
)
|
||||
|
||||
builder.addFolders(folderBuilder)
|
||||
|
||||
// TODO: add the other devices to the cluster config
|
||||
}
|
||||
|
||||
return builder.build()
|
||||
}
|
||||
|
||||
// TODO: understand this
|
||||
internal fun handleReceivedClusterConfig(
|
||||
clusterConfig: BlockExchangeProtos.ClusterConfig,
|
||||
configuration: Configuration,
|
||||
otherDeviceId: DeviceId,
|
||||
indexHandler: IndexHandler
|
||||
): ClusterConfigInfo {
|
||||
val folderInfoList = mutableListOf<ClusterConfigFolderInfo>()
|
||||
val newSharedFolders = mutableListOf<FolderInfo>()
|
||||
|
||||
for (folder in clusterConfig.foldersList ?: emptyList()) {
|
||||
var folderInfo = ClusterConfigFolderInfo(folder.id, folder.label)
|
||||
val devicesById = (folder.devicesList ?: emptyList())
|
||||
.associateBy { input ->
|
||||
DeviceId.fromHashData(input.id!!.toByteArray())
|
||||
}
|
||||
val otherDevice = devicesById[otherDeviceId]
|
||||
val ourDevice = devicesById[configuration.localDeviceId]
|
||||
if (otherDevice != null) {
|
||||
folderInfo = folderInfo.copy(isAnnounced = true)
|
||||
}
|
||||
if (ourDevice != null) {
|
||||
folderInfo = folderInfo.copy(isShared = true)
|
||||
logger.info("folder shared from device = {} folder = {}", otherDeviceId, folderInfo)
|
||||
val folderIds = configuration.folders.map { it.folderId }
|
||||
if (!folderIds.contains(folderInfo.folderId)) {
|
||||
val fi = FolderInfo(folderInfo.folderId, folderInfo.label)
|
||||
configuration.folders = configuration.folders + fi
|
||||
newSharedFolders.add(fi)
|
||||
logger.info("new folder shared = {}", folderInfo)
|
||||
}
|
||||
} else {
|
||||
logger.info("folder not shared from device = {} folder = {}", otherDeviceId, folderInfo)
|
||||
}
|
||||
|
||||
folderInfoList.add(folderInfo)
|
||||
}
|
||||
configuration.persistLater()
|
||||
indexHandler.handleClusterConfigMessageProcessedEvent(clusterConfig)
|
||||
|
||||
return ClusterConfigInfo(folderInfoList, newSharedFolders)
|
||||
}
|
||||
}
|
||||
|
||||
class ClusterConfigInfo (val folderInfo: List<ClusterConfigFolderInfo>, val newSharedFolders: List<FolderInfo>) {
|
||||
companion object {
|
||||
val dummy = ClusterConfigInfo(folderInfo = emptyList(), newSharedFolders = emptyList())
|
||||
}
|
||||
|
||||
val folderInfoById = folderInfo.associateBy { it.folderId }
|
||||
val sharedFolderIds: Set<String> by lazy {
|
||||
folderInfo.filter { it.isShared }.map { it.folderId }.toSet()
|
||||
}
|
||||
}
|
||||
|
||||
data class ClusterConfigFolderInfo(
|
||||
val folderId: String,
|
||||
val label: String = folderId,
|
||||
val isAnnounced: Boolean = false,
|
||||
val isShared: Boolean = false
|
||||
) {
|
||||
init {
|
||||
assert(folderId.isNotEmpty())
|
||||
}
|
||||
}
|
||||
+29
@@ -0,0 +1,29 @@
|
||||
/*
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import kotlinx.coroutines.CompletableDeferred
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
|
||||
sealed class ConnectionAction
|
||||
object CloseConnectionAction: ConnectionAction()
|
||||
class SendRequestConnectionAction(
|
||||
val request: BlockExchangeProtos.Request,
|
||||
val completableDeferred: CompletableDeferred<BlockExchangeProtos.Response>
|
||||
): ConnectionAction()
|
||||
class ConfirmIsConnectedAction(val completableDeferred: CompletableDeferred<ClusterConfigInfo>): ConnectionAction()
|
||||
class SendIndexUpdateAction(
|
||||
val message: BlockExchangeProtos.IndexUpdate,
|
||||
val completableDeferred: CompletableDeferred<Unit?>
|
||||
): ConnectionAction()
|
||||
+204
@@ -0,0 +1,204 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import kotlinx.coroutines.*
|
||||
import kotlinx.coroutines.channels.*
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
import net.syncthing.java.bep.IndexHandler
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.IOException
|
||||
|
||||
object ConnectionActorGenerator {
|
||||
private val closed = Channel<ConnectionAction>().apply { cancel() }
|
||||
private val logger = LoggerFactory.getLogger(ConnectionActorGenerator::class.java)
|
||||
|
||||
private fun deviceAddressesGenerator(deviceAddress: ReceiveChannel<DeviceAddress>) = GlobalScope.produce<List<DeviceAddress>> (capacity = Channel.CONFLATED) {
|
||||
val addresses = mutableMapOf<String, DeviceAddress>()
|
||||
|
||||
deviceAddress.consumeEach { address ->
|
||||
val isNew = addresses[address.address] == null
|
||||
|
||||
addresses[address.address] = address
|
||||
|
||||
if (isNew) {
|
||||
send(
|
||||
addresses.values.sortedBy { it.score }
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun <T> waitForFirstValue(source: ReceiveChannel<T>, time: Long) = GlobalScope.produce<T> {
|
||||
source.consume {
|
||||
val firstValue = source.receive()
|
||||
var lastValue = firstValue
|
||||
|
||||
try {
|
||||
withTimeout(time) {
|
||||
while (true) {
|
||||
lastValue = source.receive()
|
||||
}
|
||||
}
|
||||
|
||||
throw IllegalStateException()
|
||||
} catch (ex: TimeoutCancellationException) {
|
||||
// this is expected here
|
||||
}
|
||||
|
||||
send(lastValue)
|
||||
|
||||
// other values without delay
|
||||
for (value in source) {
|
||||
send(value)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fun generateConnectionActors(
|
||||
deviceAddress: ReceiveChannel<DeviceAddress>,
|
||||
configuration: Configuration,
|
||||
indexHandler: IndexHandler,
|
||||
requestHandler: (BlockExchangeProtos.Request) -> Deferred<BlockExchangeProtos.Response>
|
||||
) = generateConnectionActorsFromDeviceAddressList(
|
||||
deviceAddressSource = waitForFirstValue(
|
||||
source = deviceAddressesGenerator(deviceAddress),
|
||||
time = 1000
|
||||
),
|
||||
configuration = configuration,
|
||||
indexHandler = indexHandler,
|
||||
requestHandler = requestHandler
|
||||
)
|
||||
|
||||
fun generateConnectionActorsFromDeviceAddressList(
|
||||
deviceAddressSource: ReceiveChannel<List<DeviceAddress>>,
|
||||
configuration: Configuration,
|
||||
indexHandler: IndexHandler,
|
||||
requestHandler: (BlockExchangeProtos.Request) -> Deferred<BlockExchangeProtos.Response>
|
||||
) = GlobalScope.produce<Pair<SendChannel<ConnectionAction>, ClusterConfigInfo>> {
|
||||
var currentActor: SendChannel<ConnectionAction> = closed
|
||||
var currentDeviceAddress: DeviceAddress? = null
|
||||
|
||||
suspend fun closeCurrent() {
|
||||
if (currentActor != closed) {
|
||||
currentActor.close()
|
||||
currentActor = closed
|
||||
send(currentActor to ClusterConfigInfo.dummy)
|
||||
}
|
||||
}
|
||||
|
||||
suspend fun tryConnectingToAddressHandleBaseErrors(deviceAddress: DeviceAddress) = try {
|
||||
val newActor = ConnectionActor.createInstance(deviceAddress, configuration, indexHandler, requestHandler)
|
||||
val clusterConfig = ConnectionActorUtil.waitUntilConnected(newActor)
|
||||
|
||||
newActor to clusterConfig
|
||||
} catch (ex: Exception) {
|
||||
logger.warn("failed to connect to $deviceAddress", ex)
|
||||
|
||||
when (ex) {
|
||||
is IOException -> {/* expected -> ignore */}
|
||||
is InterruptedException -> {/* expected -> ignore */}
|
||||
else -> throw ex
|
||||
}
|
||||
|
||||
null
|
||||
}
|
||||
|
||||
suspend fun dispatchConnection(
|
||||
connection: SendChannel<ConnectionAction>,
|
||||
clusterConfig: ClusterConfigInfo,
|
||||
deviceAddress: DeviceAddress
|
||||
) {
|
||||
currentActor = connection
|
||||
currentDeviceAddress = deviceAddress
|
||||
|
||||
send(connection to clusterConfig)
|
||||
}
|
||||
|
||||
suspend fun tryConnectingToAddress(deviceAddress: DeviceAddress): Boolean {
|
||||
closeCurrent()
|
||||
|
||||
var connection = tryConnectingToAddressHandleBaseErrors(deviceAddress) ?: return false
|
||||
|
||||
if (connection.second.newSharedFolders.isNotEmpty()) {
|
||||
logger.debug("connected to $deviceAddress with new folders -> reconnect")
|
||||
// reconnect to send new cluster config
|
||||
connection.first.close()
|
||||
connection = tryConnectingToAddressHandleBaseErrors(deviceAddress) ?: return false
|
||||
}
|
||||
|
||||
logger.debug("connected to $deviceAddress")
|
||||
|
||||
dispatchConnection(connection.first, connection.second, deviceAddress)
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
fun isConnected() = !currentActor.isClosedForSend
|
||||
|
||||
invokeOnClose {
|
||||
currentActor.close()
|
||||
}
|
||||
|
||||
val reconnectTicker = ticker(delayMillis = 30 * 1000, initialDelayMillis = 0)
|
||||
|
||||
deviceAddressSource.consume {
|
||||
var lastDeviceAddressList: List<DeviceAddress> = emptyList()
|
||||
|
||||
while (true) {
|
||||
if (isConnected()) {
|
||||
lastDeviceAddressList = deviceAddressSource.poll() ?: lastDeviceAddressList
|
||||
|
||||
if (lastDeviceAddressList.isNotEmpty()) {
|
||||
if (reconnectTicker.poll() != null) {
|
||||
if (currentDeviceAddress != lastDeviceAddressList.first()) {
|
||||
val oldDeviceAddress = currentDeviceAddress!!
|
||||
|
||||
if (!tryConnectingToAddress(lastDeviceAddressList.first())) {
|
||||
tryConnectingToAddress(oldDeviceAddress)
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
closeCurrent()
|
||||
}
|
||||
|
||||
delay(500) // don't take too much CPU
|
||||
} else /* is not connected */ {
|
||||
// get the new list version if there is any
|
||||
lastDeviceAddressList = deviceAddressSource.poll() ?: lastDeviceAddressList
|
||||
|
||||
// try all addresses
|
||||
for (address in lastDeviceAddressList) {
|
||||
if (tryConnectingToAddress(address)) {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
// reset countdown before trying other connection if it would be time now
|
||||
// this does not reset if it has not counted down the whole time yet
|
||||
reconnectTicker.poll()
|
||||
|
||||
// wait for new device address list but not more than 15 seconds before the next iteration
|
||||
lastDeviceAddressList = withTimeoutOrNull(15 * 1000) {
|
||||
deviceAddressSource.receive()
|
||||
} ?: lastDeviceAddressList
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
+45
@@ -0,0 +1,45 @@
|
||||
/*
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import kotlinx.coroutines.CompletableDeferred
|
||||
import kotlinx.coroutines.channels.SendChannel
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
|
||||
object ConnectionActorUtil {
|
||||
suspend fun waitUntilConnected(actor: SendChannel<ConnectionAction>): ClusterConfigInfo {
|
||||
val deferred = CompletableDeferred<ClusterConfigInfo>()
|
||||
|
||||
actor.send(ConfirmIsConnectedAction(deferred))
|
||||
actor.invokeOnClose { deferred.cancel() }
|
||||
|
||||
return deferred.await()
|
||||
}
|
||||
|
||||
suspend fun sendRequest(request: BlockExchangeProtos.Request, actor: SendChannel<ConnectionAction>): BlockExchangeProtos.Response {
|
||||
val deferred = CompletableDeferred<BlockExchangeProtos.Response>()
|
||||
|
||||
actor.send(SendRequestConnectionAction(request, deferred))
|
||||
|
||||
return deferred.await()
|
||||
}
|
||||
|
||||
suspend fun sendIndexUpdate(update: BlockExchangeProtos.IndexUpdate, actor: SendChannel<ConnectionAction>) {
|
||||
val deferred = CompletableDeferred<Unit?>()
|
||||
|
||||
actor.send(SendIndexUpdateAction(update, deferred))
|
||||
|
||||
deferred.await()
|
||||
}
|
||||
}
|
||||
+86
@@ -0,0 +1,86 @@
|
||||
/*
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import kotlinx.coroutines.*
|
||||
import kotlinx.coroutines.channels.ReceiveChannel
|
||||
import kotlinx.coroutines.channels.SendChannel
|
||||
import kotlinx.coroutines.channels.consumeEach
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import java.io.IOException
|
||||
|
||||
class ConnectionActorWrapper (
|
||||
private val source: ReceiveChannel<Pair<SendChannel<ConnectionAction>, ClusterConfigInfo>>,
|
||||
val deviceId: DeviceId,
|
||||
val connectivityChangeListener: () -> Unit
|
||||
) {
|
||||
private val job = Job()
|
||||
|
||||
private var currentConnectionActor: SendChannel<ConnectionAction>? = null
|
||||
private var clusterConfigInfo: ClusterConfigInfo? = null
|
||||
|
||||
var isConnected = false
|
||||
get() = currentConnectionActor?.isClosedForSend == false
|
||||
|
||||
init {
|
||||
GlobalScope.launch (job) {
|
||||
source.consumeEach { (connectionActor, clusterConfig) ->
|
||||
currentConnectionActor = connectionActor
|
||||
clusterConfigInfo = clusterConfig
|
||||
}
|
||||
}
|
||||
|
||||
// this is a very simple solution but it does its job
|
||||
GlobalScope.launch (job) {
|
||||
var previousConnected = false
|
||||
|
||||
while (isActive) {
|
||||
val nowConnected = isConnected
|
||||
|
||||
if (previousConnected != nowConnected) {
|
||||
previousConnected = nowConnected
|
||||
|
||||
connectivityChangeListener()
|
||||
}
|
||||
|
||||
delay(200)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
suspend fun sendRequest(request: BlockExchangeProtos.Request) = ConnectionActorUtil.sendRequest(
|
||||
request,
|
||||
currentConnectionActor ?: throw IOException("not connected")
|
||||
)
|
||||
|
||||
suspend fun sendIndexUpdate(update: BlockExchangeProtos.IndexUpdate) = ConnectionActorUtil.sendIndexUpdate(
|
||||
update,
|
||||
currentConnectionActor ?: throw IOException("not connected")
|
||||
)
|
||||
|
||||
fun hasFolder(folderId: String) = clusterConfigInfo?.sharedFolderIds?.contains(folderId) ?: false
|
||||
|
||||
fun getClusterConfig() = clusterConfigInfo ?: throw IOException("not connected")
|
||||
|
||||
fun shutdown() {
|
||||
job.cancel()
|
||||
}
|
||||
|
||||
// this triggers a disconnection
|
||||
// the ConnectionActorGenerator will reconnect soon
|
||||
fun reconnect() {
|
||||
currentConnectionActor?.close()
|
||||
}
|
||||
}
|
||||
+5
-9
@@ -1,5 +1,6 @@
|
||||
/*
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
@@ -11,13 +12,8 @@
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep
|
||||
|
||||
internal data class ClusterConfigFolderInfo(val folderId: String, var label: String = folderId,
|
||||
var isAnnounced: Boolean = false, var isShared: Boolean = false) {
|
||||
|
||||
init {
|
||||
assert(folderId.isNotEmpty())
|
||||
}
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
object ConnectionConstants {
|
||||
const val MAGIC = 0x2EA7D90B
|
||||
}
|
||||
+95
@@ -0,0 +1,95 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.beans.DeviceInfo
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.DataInputStream
|
||||
import java.io.DataOutputStream
|
||||
import java.nio.ByteBuffer
|
||||
|
||||
object HelloMessageHandler {
|
||||
private val logger = LoggerFactory.getLogger(HelloMessageHandler::class.java)
|
||||
|
||||
fun sendHelloMessage(configuration: Configuration, outputStream: DataOutputStream) {
|
||||
sendHelloMessage(
|
||||
BlockExchangeProtos.Hello.newBuilder()
|
||||
.setClientName(configuration.clientName)
|
||||
.setClientVersion(configuration.clientVersion)
|
||||
.setDeviceName(configuration.localDeviceName)
|
||||
.build(),
|
||||
outputStream
|
||||
)
|
||||
}
|
||||
|
||||
private fun sendHelloMessage(message: BlockExchangeProtos.Hello, outputStream: DataOutputStream) {
|
||||
sendHelloMessage(message.toByteArray(), outputStream)
|
||||
}
|
||||
|
||||
private fun sendHelloMessage(payload: ByteArray, outputStream: DataOutputStream) {
|
||||
logger.debug("Sending hello message")
|
||||
|
||||
outputStream.apply {
|
||||
write(
|
||||
ByteBuffer.allocate(6).apply {
|
||||
putInt(ConnectionConstants.MAGIC)
|
||||
putShort(payload.size.toShort())
|
||||
}.array()
|
||||
)
|
||||
write(payload)
|
||||
flush()
|
||||
}
|
||||
}
|
||||
|
||||
fun receiveHelloMessage(
|
||||
inputStream: DataInputStream
|
||||
): BlockExchangeProtos.Hello {
|
||||
val magic = inputStream.readInt()
|
||||
NetworkUtils.assertProtocol(magic == ConnectionConstants.MAGIC) {"magic mismatch, got $magic"}
|
||||
|
||||
val length = inputStream.readShort().toInt()
|
||||
NetworkUtils.assertProtocol(length > 0) {"invalid length, must be > 0, got $length"}
|
||||
|
||||
return BlockExchangeProtos.Hello.parseFrom(
|
||||
ByteArray(length).apply {
|
||||
inputStream.readFully(this)
|
||||
}
|
||||
)
|
||||
}
|
||||
|
||||
fun processHelloMessage(
|
||||
hello: BlockExchangeProtos.Hello,
|
||||
configuration: Configuration,
|
||||
deviceId: DeviceId
|
||||
) {
|
||||
logger.info("Received hello message, deviceName=${hello.deviceName}, clientName=${hello.clientName}, clientVersion=${hello.clientVersion}")
|
||||
|
||||
// update the local device name
|
||||
// TODO: this could need some locking
|
||||
configuration.peers = configuration.peers.map { peer ->
|
||||
if (peer.deviceId == deviceId) {
|
||||
DeviceInfo(deviceId, hello.deviceName)
|
||||
} else {
|
||||
peer
|
||||
}
|
||||
}.toSet()
|
||||
|
||||
configuration.persistLater()
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,225 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import com.google.protobuf.MessageLite
|
||||
import kotlinx.coroutines.*
|
||||
import kotlinx.coroutines.channels.Channel
|
||||
import kotlinx.coroutines.channels.SendChannel
|
||||
import kotlinx.coroutines.channels.consumeEach
|
||||
import kotlinx.coroutines.sync.Mutex
|
||||
import kotlinx.coroutines.sync.withLock
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
import net.syncthing.java.bep.IndexHandler
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.core.security.KeystoreHandler
|
||||
import java.io.DataInputStream
|
||||
import java.io.DataOutputStream
|
||||
import java.io.IOException
|
||||
import java.util.*
|
||||
|
||||
object ConnectionActor {
|
||||
fun createInstance(
|
||||
address: DeviceAddress,
|
||||
configuration: Configuration,
|
||||
indexHandler: IndexHandler,
|
||||
requestHandler: (BlockExchangeProtos.Request) -> Deferred<BlockExchangeProtos.Response>
|
||||
): SendChannel<ConnectionAction> {
|
||||
val channel = Channel<ConnectionAction>(Channel.RENDEZVOUS)
|
||||
|
||||
GlobalScope.async (Dispatchers.IO) {
|
||||
OpenConnection.openSocketConnection(address, configuration).use { socket ->
|
||||
val inputStream = DataInputStream(socket.inputStream)
|
||||
val outputStream = DataOutputStream(socket.outputStream)
|
||||
|
||||
val helloMessage = coroutineScope {
|
||||
async { HelloMessageHandler.sendHelloMessage(configuration, outputStream) }
|
||||
async { HelloMessageHandler.receiveHelloMessage(inputStream) }.await()
|
||||
}
|
||||
|
||||
// the hello message exchange should happen before the certificate validation
|
||||
KeystoreHandler.assertSocketCertificateValid(socket, address.deviceId)
|
||||
|
||||
// now (after the validation) use the content of the hello message
|
||||
HelloMessageHandler.processHelloMessage(helloMessage, configuration, address.deviceId)
|
||||
|
||||
// helpers for messages
|
||||
val sendPostAuthMessageLock = Mutex()
|
||||
val receivePostAuthMessageLock = Mutex()
|
||||
|
||||
suspend fun sendPostAuthMessage(message: MessageLite) = sendPostAuthMessageLock.withLock {
|
||||
PostAuthenticationMessageHandler.sendMessage(outputStream, message, markActivityOnSocket = {})
|
||||
}
|
||||
|
||||
suspend fun receivePostAuthMessage() = receivePostAuthMessageLock.withLock {
|
||||
PostAuthenticationMessageHandler.receiveMessage(inputStream, markActivityOnSocket = {})
|
||||
}
|
||||
|
||||
// cluster config exchange
|
||||
val clusterConfig = coroutineScope {
|
||||
launch { sendPostAuthMessage(ClusterConfigHandler.buildClusterConfig(configuration, indexHandler, address.deviceId)) }
|
||||
async { receivePostAuthMessage() }.await()
|
||||
}.second
|
||||
|
||||
if (!(clusterConfig is BlockExchangeProtos.ClusterConfig)) {
|
||||
throw IOException("first message was not a cluster config message")
|
||||
}
|
||||
|
||||
val clusterConfigInfo = ClusterConfigHandler.handleReceivedClusterConfig(
|
||||
clusterConfig = clusterConfig,
|
||||
configuration = configuration,
|
||||
otherDeviceId = address.deviceId,
|
||||
indexHandler = indexHandler
|
||||
)
|
||||
|
||||
fun hasFolder(folder: String) = clusterConfigInfo.sharedFolderIds.contains(folder)
|
||||
|
||||
val messageListeners = Collections.synchronizedMap(mutableMapOf<Int, CompletableDeferred<BlockExchangeProtos.Response>>())
|
||||
|
||||
try {
|
||||
launch {
|
||||
while (isActive) {
|
||||
val message = receivePostAuthMessage().second
|
||||
|
||||
when (message) {
|
||||
is BlockExchangeProtos.Response -> {
|
||||
val listener = messageListeners.remove(message.id)
|
||||
listener
|
||||
?: throw IOException("got response ${message.id} but there is no response listener")
|
||||
listener.complete(message)
|
||||
}
|
||||
is BlockExchangeProtos.Index -> {
|
||||
indexHandler.handleIndexMessageReceivedEvent(
|
||||
folderId = message.folder,
|
||||
filesList = message.filesList,
|
||||
clusterConfigInfo = clusterConfigInfo,
|
||||
peerDeviceId = address.deviceId
|
||||
)
|
||||
}
|
||||
is BlockExchangeProtos.IndexUpdate -> {
|
||||
indexHandler.handleIndexMessageReceivedEvent(
|
||||
folderId = message.folder,
|
||||
filesList = message.filesList,
|
||||
clusterConfigInfo = clusterConfigInfo,
|
||||
peerDeviceId = address.deviceId
|
||||
)
|
||||
}
|
||||
is BlockExchangeProtos.Request -> {
|
||||
launch {
|
||||
val response = requestHandler(message).await()
|
||||
|
||||
try {
|
||||
sendPostAuthMessage(response)
|
||||
} catch (ex: IOException) {
|
||||
// the connection was closed in the time between - ignore it
|
||||
}
|
||||
}
|
||||
}
|
||||
is BlockExchangeProtos.Ping -> { /* nothing to do */
|
||||
}
|
||||
is BlockExchangeProtos.ClusterConfig -> throw IOException("received cluster config twice")
|
||||
is BlockExchangeProtos.Close -> socket.close()
|
||||
else -> throw IOException("unsupported message type ${message.javaClass}")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// send index messages - TODO: Why?
|
||||
for (folder in configuration.folders) {
|
||||
if (hasFolder(folder.folderId)) {
|
||||
sendPostAuthMessage(
|
||||
BlockExchangeProtos.Index.newBuilder()
|
||||
.setFolder(folder.folderId)
|
||||
.build()
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
launch {
|
||||
// send ping all 90 seconds
|
||||
// TODO: only send when there were no messages for 90 seconds
|
||||
|
||||
while (isActive) {
|
||||
delay(90 * 1000)
|
||||
|
||||
launch { sendPostAuthMessage(BlockExchangeProtos.Ping.getDefaultInstance()) }
|
||||
}
|
||||
}
|
||||
|
||||
var nextRequestId = 0
|
||||
|
||||
channel.consumeEach { action ->
|
||||
when (action) {
|
||||
CloseConnectionAction -> throw InterruptedException()
|
||||
is SendRequestConnectionAction -> {
|
||||
val requestId = nextRequestId++
|
||||
|
||||
messageListeners[requestId] = action.completableDeferred
|
||||
|
||||
// async to allow handling the next action faster
|
||||
async {
|
||||
try {
|
||||
sendPostAuthMessage(
|
||||
action.request.toBuilder()
|
||||
.setId(requestId)
|
||||
.build()
|
||||
)
|
||||
} catch (ex: Exception) {
|
||||
action.completableDeferred.cancel(ex)
|
||||
}
|
||||
}
|
||||
}
|
||||
is ConfirmIsConnectedAction -> {
|
||||
action.completableDeferred.complete(clusterConfigInfo)
|
||||
|
||||
// otherwise, Kotlin would warn that the return
|
||||
// type does not match to the other branches
|
||||
null
|
||||
}
|
||||
is SendIndexUpdateAction -> {
|
||||
async {
|
||||
try {
|
||||
sendPostAuthMessage(action.message)
|
||||
} catch (ex: Exception) {
|
||||
action.completableDeferred.cancel(ex)
|
||||
}
|
||||
}
|
||||
}
|
||||
}.let { /* prevents compiling if one action is not handled */ }
|
||||
}
|
||||
} finally {
|
||||
// send close message
|
||||
withContext(NonCancellable) {
|
||||
if (socket.isConnected) {
|
||||
sendPostAuthMessage(BlockExchangeProtos.Close.getDefaultInstance())
|
||||
}
|
||||
}
|
||||
|
||||
// cancel all pending listeners
|
||||
messageListeners.values.forEach { it.cancel() }
|
||||
}
|
||||
}
|
||||
}.invokeOnCompletion { ex ->
|
||||
if (ex != null) {
|
||||
channel.cancel(ex)
|
||||
} else {
|
||||
channel.cancel()
|
||||
}
|
||||
}
|
||||
|
||||
return channel
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,46 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import com.google.protobuf.MessageLite
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
|
||||
object MessageTypes {
|
||||
val messageTypes = listOf(
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.CLOSE, BlockExchangeProtos.Close::class.java) { BlockExchangeProtos.Close.parseFrom(it) },
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.CLUSTER_CONFIG, BlockExchangeProtos.ClusterConfig::class.java) { BlockExchangeProtos.ClusterConfig.parseFrom(it) },
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.DOWNLOAD_PROGRESS, BlockExchangeProtos.DownloadProgress::class.java) { BlockExchangeProtos.DownloadProgress.parseFrom(it) },
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.INDEX, BlockExchangeProtos.Index::class.java) { BlockExchangeProtos.Index.parseFrom(it) },
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.INDEX_UPDATE, BlockExchangeProtos.IndexUpdate::class.java) { BlockExchangeProtos.IndexUpdate.parseFrom(it) },
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.PING, BlockExchangeProtos.Ping::class.java) { BlockExchangeProtos.Ping.parseFrom(it) },
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.REQUEST, BlockExchangeProtos.Request::class.java) { BlockExchangeProtos.Request.parseFrom(it) },
|
||||
MessageTypeInfo(BlockExchangeProtos.MessageType.RESPONSE, BlockExchangeProtos.Response::class.java) { BlockExchangeProtos.Response.parseFrom(it) }
|
||||
)
|
||||
|
||||
val messageTypesByProtoMessageType = messageTypes.map { it.protoMessageType to it }.toMap()
|
||||
val messageTypesByJavaClass = messageTypes.map { it.javaClass to it }.toMap()
|
||||
|
||||
fun getIdForMessage(message: MessageLite) = when (message) {
|
||||
is BlockExchangeProtos.Request -> Integer.toString(message.id)
|
||||
is BlockExchangeProtos.Response -> Integer.toString(message.id)
|
||||
else -> Integer.toString(Math.abs(message.hashCode()))
|
||||
}
|
||||
}
|
||||
|
||||
data class MessageTypeInfo(
|
||||
val protoMessageType: BlockExchangeProtos.MessageType,
|
||||
val javaClass: Class<out MessageLite>,
|
||||
val parseFrom: (data: ByteArray) -> MessageLite
|
||||
)
|
||||
+45
@@ -0,0 +1,45 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import net.syncthing.java.client.protocol.rp.RelayClient
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.core.security.KeystoreHandler
|
||||
import org.slf4j.LoggerFactory
|
||||
import javax.net.ssl.SSLSocket
|
||||
|
||||
object OpenConnection {
|
||||
private val logger = LoggerFactory.getLogger(OpenConnection::class.java)
|
||||
|
||||
fun openSocketConnection(
|
||||
address: DeviceAddress,
|
||||
configuration: Configuration
|
||||
): SSLSocket {
|
||||
val keystoreHandler = KeystoreHandler.Loader().loadKeystore(configuration)
|
||||
|
||||
return when (address.type) {
|
||||
DeviceAddress.AddressType.TCP -> {
|
||||
logger.debug("opening tcp ssl connection")
|
||||
keystoreHandler.createSocket(address.getSocketAddress())
|
||||
}
|
||||
DeviceAddress.AddressType.RELAY -> {
|
||||
logger.debug("opening relay connection")
|
||||
keystoreHandler.wrapSocket(RelayClient(configuration).openRelayConnection(address))
|
||||
}
|
||||
else -> throw UnsupportedOperationException("unsupported address type ${address.type}")
|
||||
}
|
||||
}
|
||||
}
|
||||
+140
@@ -0,0 +1,140 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.bep.connectionactor
|
||||
|
||||
import com.google.protobuf.MessageLite
|
||||
import net.jpountz.lz4.LZ4Factory
|
||||
import net.syncthing.java.bep.BlockExchangeProtos
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.DataInputStream
|
||||
import java.io.DataOutputStream
|
||||
import java.io.IOException
|
||||
import java.lang.reflect.InvocationTargetException
|
||||
import java.nio.ByteBuffer
|
||||
|
||||
object PostAuthenticationMessageHandler {
|
||||
private val logger = LoggerFactory.getLogger(PostAuthenticationMessageHandler::class.java)
|
||||
|
||||
fun sendMessage(
|
||||
outputStream: DataOutputStream,
|
||||
message: MessageLite,
|
||||
markActivityOnSocket: () -> Unit
|
||||
) {
|
||||
val messageTypeInfo = MessageTypes.messageTypesByJavaClass[message.javaClass]!!
|
||||
val header = BlockExchangeProtos.Header.newBuilder()
|
||||
.setCompression(BlockExchangeProtos.MessageCompression.NONE)
|
||||
.setType(messageTypeInfo.protoMessageType)
|
||||
.build()
|
||||
val headerData = header.toByteArray()
|
||||
val messageData = message.toByteArray() //TODO support compression
|
||||
|
||||
logger.debug("sending message type = {} {}", header.type, MessageTypes.getIdForMessage(message))
|
||||
markActivityOnSocket()
|
||||
|
||||
outputStream.apply {
|
||||
writeShort(headerData.size)
|
||||
write(headerData)
|
||||
writeInt(messageData.size)
|
||||
write(messageData)
|
||||
flush()
|
||||
}
|
||||
|
||||
markActivityOnSocket()
|
||||
}
|
||||
|
||||
fun receiveMessage(
|
||||
inputStream: DataInputStream,
|
||||
markActivityOnSocket: () -> Unit
|
||||
): Pair<BlockExchangeProtos.MessageType, MessageLite> {
|
||||
val header = BlockExchangeProtos.Header.parseFrom(readHeader(
|
||||
inputStream = inputStream,
|
||||
retryReadingLength = true,
|
||||
markActivityOnSocket = markActivityOnSocket
|
||||
))
|
||||
|
||||
var messageBuffer = readMessage(
|
||||
inputStream = inputStream,
|
||||
retryReadingLength = true,
|
||||
markActivityOnSocket = markActivityOnSocket
|
||||
)
|
||||
|
||||
if (header.compression == BlockExchangeProtos.MessageCompression.LZ4) {
|
||||
val uncompressedLength = ByteBuffer.wrap(messageBuffer).int
|
||||
messageBuffer = LZ4Factory.fastestInstance().fastDecompressor().decompress(messageBuffer, 4, uncompressedLength)
|
||||
}
|
||||
|
||||
val messageTypeInfo = MessageTypes.messageTypesByProtoMessageType[header.type]
|
||||
NetworkUtils.assertProtocol(messageTypeInfo != null) {"unsupported message type = ${header.type}"}
|
||||
|
||||
try {
|
||||
return header.type to messageTypeInfo!!.parseFrom(messageBuffer)
|
||||
} catch (e: Exception) {
|
||||
when (e) {
|
||||
is IllegalAccessException, is IllegalArgumentException, is InvocationTargetException, is NoSuchMethodException, is SecurityException ->
|
||||
throw IOException(e)
|
||||
else -> throw e
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun readHeader(
|
||||
inputStream: DataInputStream,
|
||||
markActivityOnSocket: () -> Unit,
|
||||
retryReadingLength: Boolean
|
||||
): ByteArray {
|
||||
var headerLength = inputStream.readShort().toInt()
|
||||
|
||||
// TODO: what is this good for?
|
||||
if (retryReadingLength) {
|
||||
while (headerLength == 0) {
|
||||
logger.warn("got headerLength == 0, skipping short")
|
||||
headerLength = inputStream.readShort().toInt()
|
||||
}
|
||||
}
|
||||
|
||||
markActivityOnSocket()
|
||||
|
||||
NetworkUtils.assertProtocol(headerLength > 0) {"invalid length, must be > 0, got $headerLength"}
|
||||
|
||||
return ByteArray(headerLength).apply {
|
||||
inputStream.readFully(this)
|
||||
}
|
||||
}
|
||||
|
||||
private fun readMessage(
|
||||
inputStream: DataInputStream,
|
||||
markActivityOnSocket: () -> Unit,
|
||||
retryReadingLength: Boolean
|
||||
): ByteArray {
|
||||
var messageLength = inputStream.readInt()
|
||||
|
||||
// TODO: what is this good for?
|
||||
if (retryReadingLength) {
|
||||
while (messageLength == 0) {
|
||||
logger.warn("received readInt() == 0, expecting 'bep message header length' (int >0), ignoring (keepalive?)")
|
||||
messageLength = inputStream.readInt()
|
||||
}
|
||||
}
|
||||
|
||||
NetworkUtils.assertProtocol(messageLength >= 0) {"invalid length, must be >= 0, got $messageLength"}
|
||||
|
||||
val messageBuffer = ByteArray(messageLength)
|
||||
inputStream.readFully(messageBuffer)
|
||||
markActivityOnSocket()
|
||||
|
||||
return messageBuffer
|
||||
}
|
||||
}
|
||||
@@ -7,6 +7,7 @@ dependencies {
|
||||
compile project(':syncthing-repository-default')
|
||||
compile "commons-cli:commons-cli:1.4"
|
||||
implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:1.0.0'
|
||||
}
|
||||
|
||||
run {
|
||||
|
||||
@@ -13,12 +13,13 @@
|
||||
*/
|
||||
package net.syncthing.java.client.cli
|
||||
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import net.syncthing.java.client.SyncthingClient
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.beans.DeviceInfo
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.repository.repo.SqlRepository
|
||||
import net.syncthing.java.client.SyncthingClient
|
||||
import org.apache.commons.cli.*
|
||||
import org.apache.commons.io.FileUtils
|
||||
import org.slf4j.LoggerFactory
|
||||
@@ -91,28 +92,23 @@ class Main(private val commandLine: CommandLine) {
|
||||
System.out.println("file path = $folderAndPath")
|
||||
val folder = folderAndPath.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[0]
|
||||
val path = folderAndPath.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[1]
|
||||
val latch = CountDownLatch(1)
|
||||
val fileInfo = FileInfo(folder = folder, path = path, type = FileInfo.FileType.FILE)
|
||||
syncthingClient.getBlockPuller(folder, { blockPuller ->
|
||||
try {
|
||||
val inputStream = blockPuller.pullFileSync(fileInfo)
|
||||
val fileName = syncthingClient.indexHandler.getFileInfoByPath(folder, path)!!.fileName
|
||||
val file =
|
||||
if (commandLine.hasOption("o")) {
|
||||
val param = File(commandLine.getOptionValue("o"))
|
||||
if (param.isDirectory) File(param, fileName) else param
|
||||
} else {
|
||||
File(fileName)
|
||||
}
|
||||
FileUtils.copyInputStreamToFile(inputStream, file)
|
||||
System.out.println("saved file to = $file.absolutePath")
|
||||
} catch (e: InterruptedException) {
|
||||
logger.warn("", e)
|
||||
} catch (e: IOException) {
|
||||
logger.warn("", e)
|
||||
try {
|
||||
val inputStream = syncthingClient.pullFileSync(fileInfo)
|
||||
val fileName = syncthingClient.indexHandler.getFileInfoByPath(folder, path)!!.fileName
|
||||
val file = if (commandLine.hasOption("o")) {
|
||||
val param = File(commandLine.getOptionValue("o"))
|
||||
if (param.isDirectory) File(param, fileName) else param
|
||||
} else {
|
||||
File(fileName)
|
||||
}
|
||||
}, { logger.warn("Failed to pull file") })
|
||||
latch.await()
|
||||
FileUtils.copyInputStreamToFile(inputStream, file)
|
||||
System.out.println("saved file to = $file.absolutePath")
|
||||
} catch (e: InterruptedException) {
|
||||
logger.warn("", e)
|
||||
} catch (e: IOException) {
|
||||
logger.warn("", e)
|
||||
}
|
||||
}
|
||||
"P" -> {
|
||||
var path = option.value
|
||||
@@ -122,20 +118,20 @@ class Main(private val commandLine: CommandLine) {
|
||||
val folder = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[0]
|
||||
path = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[1]
|
||||
val latch = CountDownLatch(1)
|
||||
syncthingClient.getBlockPusher(folder, { blockPusher ->
|
||||
val observer = blockPusher.pushFile(FileInputStream(file), folder, path)
|
||||
while (!observer.isCompleted()) {
|
||||
try {
|
||||
observer.waitForProgressUpdate()
|
||||
} catch (e: InterruptedException) {
|
||||
logger.warn("", e)
|
||||
}
|
||||
val blockPusher = syncthingClient.getBlockPusher(folder)
|
||||
|
||||
System.out.println("upload progress ${observer.progressPercentage()}%")
|
||||
val observer = runBlocking {
|
||||
blockPusher.pushFile(FileInputStream(file), folder, path)
|
||||
}
|
||||
while (!observer.isCompleted()) {
|
||||
try {
|
||||
observer.waitForProgressUpdate()
|
||||
} catch (e: InterruptedException) {
|
||||
logger.warn("", e)
|
||||
}
|
||||
latch.countDown()
|
||||
}, { logger.warn("Failed to upload file") })
|
||||
latch.await()
|
||||
|
||||
System.out.println("upload progress ${observer.progressPercentage()}%")
|
||||
}
|
||||
System.out.println("uploaded file to network")
|
||||
}
|
||||
"D" -> {
|
||||
@@ -143,17 +139,16 @@ class Main(private val commandLine: CommandLine) {
|
||||
val folder = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[0]
|
||||
path = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[1]
|
||||
System.out.println("delete path = $path")
|
||||
val latch = CountDownLatch(1)
|
||||
syncthingClient.getBlockPusher(folder, { blockPusher ->
|
||||
try {
|
||||
blockPusher.pushDelete(folder, path).waitForComplete()
|
||||
} catch (e: InterruptedException) {
|
||||
logger.warn("", e)
|
||||
}
|
||||
try {
|
||||
val blockPusher = syncthingClient.getBlockPusher(folder)
|
||||
|
||||
latch.countDown()
|
||||
}, { System.out.println("Failed to delete path") })
|
||||
latch.await()
|
||||
runBlocking {
|
||||
blockPusher.pushDelete(folder, path)
|
||||
}
|
||||
} catch (e: InterruptedException) {
|
||||
logger.warn("", e)
|
||||
System.out.println("Failed to delete path")
|
||||
}
|
||||
System.out.println("deleted path")
|
||||
}
|
||||
"M" -> {
|
||||
@@ -161,17 +156,16 @@ class Main(private val commandLine: CommandLine) {
|
||||
val folder = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[0]
|
||||
path = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[1]
|
||||
System.out.println("dir path = $path")
|
||||
val latch = CountDownLatch(1)
|
||||
syncthingClient.getBlockPusher(folder, { blockPusher ->
|
||||
try {
|
||||
blockPusher.pushDir(folder, path).waitForComplete()
|
||||
} catch (e: InterruptedException) {
|
||||
logger.warn("", e)
|
||||
}
|
||||
try {
|
||||
val blockPusher = syncthingClient.getBlockPusher(folder)
|
||||
|
||||
latch.countDown()
|
||||
}, { System.out.println("Failed to push directory") })
|
||||
latch.await()
|
||||
runBlocking {
|
||||
blockPusher.pushDir(folder, path)
|
||||
}
|
||||
} catch (e: InterruptedException) {
|
||||
System.out.println("Failed to push directory")
|
||||
logger.warn("", e)
|
||||
}
|
||||
System.out.println("uploaded dir to network")
|
||||
}
|
||||
"L" -> {
|
||||
|
||||
@@ -6,4 +6,5 @@ dependencies {
|
||||
compile project(':syncthing-bep')
|
||||
compile project(':syncthing-discovery')
|
||||
implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:1.0.0'
|
||||
}
|
||||
|
||||
@@ -0,0 +1,49 @@
|
||||
/*
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.client
|
||||
|
||||
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
|
||||
class Connections (val generate: (DeviceId) -> ConnectionActorWrapper) {
|
||||
private val map = mutableMapOf<DeviceId, ConnectionActorWrapper>()
|
||||
|
||||
fun getByDeviceId(deviceId: DeviceId): ConnectionActorWrapper {
|
||||
return synchronized(map) {
|
||||
val oldEntry = map[deviceId]
|
||||
|
||||
if (oldEntry != null) {
|
||||
return oldEntry
|
||||
} else {
|
||||
val newEntry = generate(deviceId)
|
||||
|
||||
map[deviceId] = newEntry
|
||||
|
||||
return newEntry
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fun shutdown() {
|
||||
synchronized(map) {
|
||||
map.values.forEach { it.shutdown() }
|
||||
}
|
||||
}
|
||||
|
||||
fun reconnectAllConnections() {
|
||||
synchronized(map) {
|
||||
map.values.forEach { it.reconnect() }
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -13,56 +13,63 @@
|
||||
*/
|
||||
package net.syncthing.java.client
|
||||
|
||||
import net.syncthing.java.bep.BlockPuller
|
||||
import net.syncthing.java.bep.BlockPusher
|
||||
import net.syncthing.java.bep.ConnectionHandler
|
||||
import net.syncthing.java.bep.IndexHandler
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.async
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import net.syncthing.java.bep.*
|
||||
import net.syncthing.java.bep.connectionactor.ConnectionActorGenerator
|
||||
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.beans.DeviceInfo
|
||||
import net.syncthing.java.core.beans.FileInfo
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.core.interfaces.IndexRepository
|
||||
import net.syncthing.java.core.interfaces.TempRepository
|
||||
import net.syncthing.java.core.security.KeystoreHandler
|
||||
import net.syncthing.java.core.utils.awaitTerminationSafe
|
||||
import net.syncthing.java.discovery.DiscoveryHandler
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.Closeable
|
||||
import java.io.IOException
|
||||
import java.util.Collections
|
||||
import java.util.TreeSet
|
||||
import java.util.concurrent.Executors
|
||||
import java.util.concurrent.TimeUnit
|
||||
import java.util.concurrent.atomic.AtomicBoolean
|
||||
import kotlin.collections.ArrayList
|
||||
import java.io.InputStream
|
||||
import java.util.*
|
||||
|
||||
class SyncthingClient(
|
||||
private val configuration: Configuration,
|
||||
private val repository: IndexRepository,
|
||||
private val tempRepository: TempRepository
|
||||
) : Closeable {
|
||||
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
val discoveryHandler: DiscoveryHandler
|
||||
val indexHandler: IndexHandler
|
||||
private val connections = Collections.synchronizedSet(createConnectionsSet())
|
||||
private val connectByDeviceIdLocks = Collections.synchronizedMap(HashMap<DeviceId, Object>())
|
||||
val indexHandler = IndexHandler(configuration, repository, tempRepository)
|
||||
val discoveryHandler = DiscoveryHandler(configuration)
|
||||
private val onConnectionChangedListeners = Collections.synchronizedList(mutableListOf<(DeviceId) -> Unit>())
|
||||
private var connectDevicesScheduler = Executors.newSingleThreadScheduledExecutor()
|
||||
|
||||
private fun createConnectionsSet() = TreeSet<ConnectionHandler>(compareBy { it.address.score })
|
||||
|
||||
init {
|
||||
indexHandler = IndexHandler(configuration, repository, tempRepository)
|
||||
discoveryHandler = DiscoveryHandler(configuration)
|
||||
connectDevicesScheduler.scheduleAtFixedRate(this::updateIndexFromPeers, 0, 15, TimeUnit.SECONDS)
|
||||
}
|
||||
private val requestHandlerRegistry = RequestHandlerRegistry()
|
||||
private val connections = Connections(
|
||||
generate = { deviceId ->
|
||||
ConnectionActorWrapper(
|
||||
source = ConnectionActorGenerator.generateConnectionActors(
|
||||
deviceAddress = discoveryHandler.devicesAddressesManager.getDeviceAddressManager(deviceId).streamCurrentDeviceAddresses(),
|
||||
requestHandler = { request ->
|
||||
GlobalScope.async {
|
||||
requestHandlerRegistry.handleRequest(
|
||||
source = deviceId,
|
||||
request = request
|
||||
)
|
||||
}
|
||||
},
|
||||
indexHandler = indexHandler,
|
||||
configuration = configuration
|
||||
),
|
||||
deviceId = deviceId,
|
||||
connectivityChangeListener = {
|
||||
synchronized(onConnectionChangedListeners) {
|
||||
onConnectionChangedListeners.forEach { it(deviceId) }
|
||||
}
|
||||
}
|
||||
)
|
||||
}
|
||||
)
|
||||
|
||||
fun clearCacheAndIndex() {
|
||||
indexHandler.clearIndex()
|
||||
configuration.folders = emptySet()
|
||||
configuration.persistLater()
|
||||
updateIndexFromPeers()
|
||||
connections.reconnectAllConnections()
|
||||
}
|
||||
|
||||
fun addOnConnectionChangedListener(listener: (DeviceId) -> Unit) {
|
||||
@@ -74,158 +81,61 @@ class SyncthingClient(
|
||||
onConnectionChangedListeners.remove(listener)
|
||||
}
|
||||
|
||||
@Throws(IOException::class, KeystoreHandler.CryptoException::class)
|
||||
private fun openConnection(deviceAddress: DeviceAddress): ConnectionHandler {
|
||||
logger.debug("Connecting to ${deviceAddress.deviceId}, active connections: ${connections.map { it.deviceId().deviceId }}")
|
||||
val connectionHandler = ConnectionHandler(
|
||||
configuration, deviceAddress, indexHandler, tempRepository, { connectionHandler, _ ->
|
||||
connectionHandler.close()
|
||||
openConnection(deviceAddress)
|
||||
},
|
||||
{connection ->
|
||||
if (!connection.isConnected) {
|
||||
connections.remove(connection)
|
||||
}
|
||||
onConnectionChangedListeners.forEach { it(connection.deviceId()) }
|
||||
})
|
||||
private fun getConnections() = configuration.peerIds.map { connections.getByDeviceId(it) }
|
||||
|
||||
try {
|
||||
connectionHandler.connect()
|
||||
} catch (ex: Exception) {
|
||||
connectionHandler.closeBg()
|
||||
|
||||
throw ex
|
||||
}
|
||||
|
||||
connections.add(connectionHandler)
|
||||
|
||||
return connectionHandler
|
||||
init {
|
||||
discoveryHandler.newDeviceAddressSupplier() // starts the discovery
|
||||
getConnections()
|
||||
}
|
||||
|
||||
/**
|
||||
* Takes discovered addresses from [[DiscoveryHandler]] and connects to devices.
|
||||
*
|
||||
* We need to make sure that we are only connecting once to each device.
|
||||
*/
|
||||
private fun getPeerConnections(listener: (connection: ConnectionHandler) -> Unit, completeListener: () -> Unit) {
|
||||
// create an copy to prevent dispatching an action two times
|
||||
val connectionsWhichWereDispatched = createConnectionsSet()
|
||||
|
||||
synchronized (connections) {
|
||||
connectionsWhichWereDispatched.addAll(connections)
|
||||
}
|
||||
|
||||
connectionsWhichWereDispatched.forEach { listener(it) }
|
||||
|
||||
discoveryHandler.newDeviceAddressSupplier()
|
||||
.takeWhile { it != null }
|
||||
.filterNotNull()
|
||||
.groupBy { it.deviceId() }
|
||||
.filterNot { it.value.isEmpty() }
|
||||
.forEach { (deviceId, addresses) ->
|
||||
// create an lock per device id to prevent multiple connections to one device
|
||||
|
||||
synchronized (connectByDeviceIdLocks) {
|
||||
if (connectByDeviceIdLocks[deviceId] == null) {
|
||||
connectByDeviceIdLocks[deviceId] = Object()
|
||||
}
|
||||
}
|
||||
|
||||
synchronized (connectByDeviceIdLocks[deviceId]!!) {
|
||||
val existingConnection = connections.find { it.deviceId() == deviceId && it.isConnected }
|
||||
|
||||
if (existingConnection != null) {
|
||||
connectionsWhichWereDispatched.add(existingConnection)
|
||||
listener(existingConnection)
|
||||
|
||||
return@synchronized
|
||||
}
|
||||
|
||||
// try to use all addresses
|
||||
for (address in addresses.distinctBy { it.address }) {
|
||||
try {
|
||||
val newConnection = openConnection(address)
|
||||
|
||||
connectionsWhichWereDispatched.add(newConnection)
|
||||
listener(newConnection)
|
||||
|
||||
break // it worked, no need to try more
|
||||
} catch (e: IOException) {
|
||||
logger.warn("error connecting to device = $address", e)
|
||||
} catch (e: KeystoreHandler.CryptoException) {
|
||||
logger.warn("error connecting to device = $address", e)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// use all connections which were added in the time between and were not added by this function call
|
||||
val newConnectionsBackup = createConnectionsSet()
|
||||
|
||||
synchronized (connections) {
|
||||
newConnectionsBackup.addAll(connections)
|
||||
}
|
||||
|
||||
connectionsWhichWereDispatched.forEach { newConnectionsBackup.remove(it) }
|
||||
|
||||
newConnectionsBackup.forEach { listener(it) }
|
||||
|
||||
completeListener()
|
||||
fun connectToNewlyAddedDevices() {
|
||||
getConnections()
|
||||
}
|
||||
|
||||
private fun updateIndexFromPeers() {
|
||||
getPeerConnections({ connection ->
|
||||
try {
|
||||
indexHandler.waitForRemoteIndexAcquired(connection)
|
||||
} catch (ex: InterruptedException) {
|
||||
logger.warn("exception while waiting for index", ex)
|
||||
}
|
||||
}, {})
|
||||
fun disconnectFromRemovedDevices() {
|
||||
// TODO: implement this
|
||||
}
|
||||
|
||||
private fun getConnectionForFolder(folder: String, listener: (connection: ConnectionHandler) -> Unit,
|
||||
errorListener: () -> Unit) {
|
||||
val isConnected = AtomicBoolean(false)
|
||||
getPeerConnections({ connection ->
|
||||
if (connection.hasFolder(folder) && !isConnected.get()) {
|
||||
listener(connection)
|
||||
isConnected.set(true)
|
||||
}
|
||||
}, {
|
||||
if (!isConnected.get()) {
|
||||
errorListener()
|
||||
}
|
||||
})
|
||||
fun getActiveConnectionsForFolder(folderId: String) = configuration.peerIds
|
||||
.map { connections.getByDeviceId(it) }
|
||||
.filter { it.isConnected && it.hasFolder(folderId) }
|
||||
|
||||
suspend fun pullFile(
|
||||
fileInfo: FileInfo,
|
||||
progressListener: (status: BlockPullerStatus) -> Unit = { }
|
||||
): InputStream = BlockPuller.pullFile(
|
||||
fileInfo = fileInfo,
|
||||
progressListener = progressListener,
|
||||
connections = getConnections(),
|
||||
indexHandler = indexHandler,
|
||||
tempRepository = tempRepository
|
||||
)
|
||||
|
||||
fun pullFileSync(fileInfo: FileInfo) = runBlocking { pullFile(fileInfo) }
|
||||
|
||||
fun getBlockPusher(folderId: String): BlockPusher {
|
||||
val connection = getActiveConnectionsForFolder(folderId).first()
|
||||
|
||||
return BlockPusher(
|
||||
localDeviceId = connection.deviceId,
|
||||
connectionHandler = connection,
|
||||
indexHandler = indexHandler,
|
||||
requestHandlerRegistry = requestHandlerRegistry
|
||||
)
|
||||
}
|
||||
|
||||
fun getBlockPuller(folderId: String, listener: (BlockPuller) -> Unit, errorListener: () -> Unit) {
|
||||
getConnectionForFolder(folderId, { connection ->
|
||||
listener(connection.getBlockPuller())
|
||||
}, errorListener)
|
||||
}
|
||||
|
||||
fun getBlockPusher(folderId: String, listener: (BlockPusher) -> Unit, errorListener: () -> Unit) {
|
||||
getConnectionForFolder(folderId, { connection ->
|
||||
listener(connection.getBlockPusher())
|
||||
}, errorListener)
|
||||
}
|
||||
|
||||
fun getPeerStatus(): List<DeviceInfo> {
|
||||
return configuration.peers.map { device ->
|
||||
val isConnected = connections.find { it.deviceId() == device.deviceId }?.isConnected ?: false
|
||||
device.copy(isConnected = isConnected)
|
||||
}
|
||||
fun getPeerStatus() = configuration.peers.map { device ->
|
||||
device.copy(
|
||||
isConnected = connections.getByDeviceId(device.deviceId).isConnected
|
||||
)
|
||||
}
|
||||
|
||||
override fun close() {
|
||||
connectDevicesScheduler.awaitTerminationSafe()
|
||||
discoveryHandler.close()
|
||||
// Create copy of list, because it will be modified by handleConnectionClosedEvent(), causing ConcurrentModificationException.
|
||||
ArrayList(connections).forEach{it.close()}
|
||||
indexHandler.close()
|
||||
repository.close()
|
||||
tempRepository.close()
|
||||
connections.shutdown()
|
||||
assert(onConnectionChangedListeners.isEmpty())
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -9,7 +9,6 @@ dependencies {
|
||||
compile "org.slf4j:slf4j-api:1.7.25"
|
||||
compile "ch.qos.logback:logback-classic:1.2.3"
|
||||
compile "com.google.code.gson:gson:2.8.2"
|
||||
compile "org.apache.httpcomponents:httpclient:4.5.4"
|
||||
compile "org.bouncycastle:bcmail-jdk15on:1.59"
|
||||
implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
|
||||
}
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
/*
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
@@ -18,44 +18,38 @@ import java.net.InetSocketAddress
|
||||
import java.net.UnknownHostException
|
||||
import java.util.*
|
||||
|
||||
/**
|
||||
*
|
||||
* TODO: this class cant use [[DeviceId]] because [[GlobalDiscoveryHandler.pickAnnounceServers]] uses that field for discovery server URLs.
|
||||
*/
|
||||
class DeviceAddress private constructor(val deviceId: String, private val instanceId: Long?, val address: String, producer: AddressProducer?, score: Int?, lastModified: Date?) {
|
||||
// TODO: this should use a data class, but the custom equals prevents it
|
||||
class DeviceAddress private constructor(val deviceId: DeviceId, private val instanceId: Long?, val address: String, producer: AddressProducer?, score: Int?, lastModified: Date?) {
|
||||
private val producer = producer ?: AddressProducer.UNKNOWN
|
||||
val score = score ?: Integer.MAX_VALUE
|
||||
private val lastModified = lastModified ?: Date()
|
||||
|
||||
@Deprecated(message = "should use deviceIdObject instead")
|
||||
fun deviceId() = DeviceId(deviceId)
|
||||
|
||||
val deviceIdObject: DeviceId by lazy { DeviceId(deviceId) }
|
||||
|
||||
@Throws(UnknownHostException::class)
|
||||
private fun getInetAddress(): InetAddress = InetAddress.getByName(address.replaceFirst("^[^:]+://".toRegex(), "").replaceFirst("(:[0-9]+)?(/.*)?$".toRegex(), ""))
|
||||
|
||||
private fun getPort(): Int = if (address.matches("^[a-z]+://[^:]+:([0-9]+).*".toRegex())) {
|
||||
private val port: Int by lazy {
|
||||
if (address.matches("^[a-z]+://[^:]+:([0-9]+).*".toRegex())) {
|
||||
Integer.parseInt(address.replaceFirst("^[a-z]+://[^:]+:([0-9]+).*".toRegex(), "$1"))
|
||||
} else {
|
||||
DEFAULT_PORT_BY_PROTOCOL[getType()]!!
|
||||
DEFAULT_PORT_BY_PROTOCOL[type]!!
|
||||
}
|
||||
}
|
||||
|
||||
fun getType(): AddressType = when {
|
||||
address.isEmpty() -> AddressType.NULL
|
||||
address.startsWith("tcp://") -> AddressType.TCP
|
||||
address.startsWith("relay://") -> AddressType.RELAY
|
||||
address.startsWith("relay-http://") -> AddressType.HTTP_RELAY
|
||||
address.startsWith("relay-https://") -> AddressType.HTTPS_RELAY
|
||||
else -> AddressType.OTHER
|
||||
val type: AddressType by lazy {
|
||||
when {
|
||||
address.isEmpty() -> AddressType.NULL
|
||||
address.startsWith("tcp://") -> AddressType.TCP
|
||||
address.startsWith("relay://") -> AddressType.RELAY
|
||||
else -> AddressType.OTHER
|
||||
}
|
||||
}
|
||||
|
||||
@Throws(UnknownHostException::class)
|
||||
fun getSocketAddress(): InetSocketAddress = InetSocketAddress(getInetAddress(), getPort())
|
||||
fun getSocketAddress(): InetSocketAddress = InetSocketAddress(getInetAddress(), port)
|
||||
|
||||
fun isWorking(): Boolean = score < Integer.MAX_VALUE
|
||||
|
||||
constructor(deviceId: String, address: String) : this(deviceId, null, address, null, null, null)
|
||||
constructor(deviceId: String, address: String) : this(DeviceId(deviceId), null, address, null, null, null)
|
||||
|
||||
fun containsUriParamValue(key: String): Boolean {
|
||||
return !getUriParam(key).isNullOrEmpty()
|
||||
@@ -79,7 +73,7 @@ class DeviceAddress private constructor(val deviceId: String, private val instan
|
||||
}
|
||||
|
||||
enum class AddressType {
|
||||
TCP, RELAY, OTHER, NULL, HTTP_RELAY, HTTPS_RELAY
|
||||
TCP, RELAY, OTHER, NULL
|
||||
}
|
||||
|
||||
enum class AddressProducer {
|
||||
@@ -97,18 +91,18 @@ class DeviceAddress private constructor(val deviceId: String, private val instan
|
||||
return hash
|
||||
}
|
||||
|
||||
override fun equals(obj: Any?): Boolean {
|
||||
if (this === obj) {
|
||||
override fun equals(other: Any?): Boolean {
|
||||
if (this === other) {
|
||||
return true
|
||||
}
|
||||
if (obj == null) {
|
||||
if (other == null) {
|
||||
return false
|
||||
}
|
||||
if (javaClass != obj.javaClass) {
|
||||
if (javaClass != other.javaClass) {
|
||||
return false
|
||||
}
|
||||
val other = obj as DeviceAddress?
|
||||
if (this.deviceId != other!!.deviceId) {
|
||||
other as DeviceAddress
|
||||
if (this.deviceId != other.deviceId) {
|
||||
return false
|
||||
}
|
||||
return this.address == other.address
|
||||
@@ -120,7 +114,7 @@ class DeviceAddress private constructor(val deviceId: String, private val instan
|
||||
|
||||
class Builder {
|
||||
|
||||
private var deviceId: String? = null
|
||||
private var deviceId: DeviceId? = null
|
||||
private var instanceId: Long? = null
|
||||
private var address: String? = null
|
||||
private var producer: AddressProducer? = null
|
||||
@@ -129,7 +123,7 @@ class DeviceAddress private constructor(val deviceId: String, private val instan
|
||||
|
||||
constructor()
|
||||
|
||||
internal constructor(deviceId: String, instanceId: Long?, address: String, producer: AddressProducer, score: Int?, lastModified: Date) {
|
||||
internal constructor(deviceId: DeviceId, instanceId: Long?, address: String, producer: AddressProducer, score: Int?, lastModified: Date) {
|
||||
this.deviceId = deviceId
|
||||
this.instanceId = instanceId
|
||||
this.address = address
|
||||
@@ -147,11 +141,11 @@ class DeviceAddress private constructor(val deviceId: String, private val instan
|
||||
return this
|
||||
}
|
||||
|
||||
fun getDeviceId(): String? {
|
||||
fun getDeviceId(): DeviceId? {
|
||||
return deviceId
|
||||
}
|
||||
|
||||
fun setDeviceId(deviceId: String): Builder {
|
||||
fun setDeviceId(deviceId: DeviceId): Builder {
|
||||
this.deviceId = deviceId
|
||||
return this
|
||||
}
|
||||
@@ -200,8 +194,7 @@ class DeviceAddress private constructor(val deviceId: String, private val instan
|
||||
companion object {
|
||||
private val DEFAULT_PORT_BY_PROTOCOL = mapOf(
|
||||
AddressType.TCP to 22000,
|
||||
AddressType.RELAY to 22067,
|
||||
AddressType.HTTP_RELAY to 80,
|
||||
AddressType.HTTPS_RELAY to 443)
|
||||
AddressType.RELAY to 22067
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -4,7 +4,6 @@ import com.google.gson.stream.JsonReader
|
||||
import com.google.gson.stream.JsonWriter
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import org.apache.commons.codec.binary.Base32
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.IOException
|
||||
|
||||
data class DeviceId @Throws(IOException::class) constructor(val deviceId: String) {
|
||||
|
||||
@@ -14,8 +14,7 @@
|
||||
package net.syncthing.java.core.beans
|
||||
|
||||
import org.apache.commons.io.FileUtils
|
||||
|
||||
import java.util.Date
|
||||
import java.util.*
|
||||
|
||||
class FolderStats private constructor(val fileCount: Long, val dirCount: Long, val size: Long, val lastUpdate: Date, folder: String, label: String?) : FolderInfo(folder, label) {
|
||||
|
||||
|
||||
@@ -12,7 +12,8 @@ data class Config(
|
||||
val folders: Set<FolderInfo>,
|
||||
val localDeviceName: String,
|
||||
val localDeviceId: String,
|
||||
val discoveryServers: Set<String>,
|
||||
val customDiscoveryServers: Set<DiscoveryServer>,
|
||||
val useDefaultDiscoveryServers: Boolean,
|
||||
val keystoreAlgorithm: String,
|
||||
val keystoreData: String
|
||||
) {
|
||||
@@ -21,7 +22,8 @@ data class Config(
|
||||
private const val FOLDERS = "folders"
|
||||
private const val LOCAL_DEVICE_NAME = "localDeviceName"
|
||||
private const val LOCAL_DEVICE_ID = "localDeviceId"
|
||||
private const val DISCOVERY_SERVERS = "discoveryServers"
|
||||
private const val USE_DEFAULT_DISCOVERY_SERVERS = "useDefaultDiscoveryServers"
|
||||
private const val CUSTOM_DISCOVERY_SERVERS = "customDiscoveryServers"
|
||||
private const val KEYSTORE_ALGORITHM = "keystoreAlgorithm"
|
||||
private const val KEYSTORE_DATA = "keystoreData"
|
||||
|
||||
@@ -30,7 +32,8 @@ data class Config(
|
||||
var folders: Set<FolderInfo>? = null
|
||||
var localDeviceName: String? = null
|
||||
var localDeviceId: String? = null
|
||||
var discoveryServers: Set<String>? = null
|
||||
var customDiscoveryServers = emptySet<DiscoveryServer>() // this field was added later, so it needs an default value
|
||||
var useDefaultDiscoveryServers = true // this field was added later, so it needs an default value
|
||||
var keystoreAlgorithm: String? = null
|
||||
var keystoreData: String? = null
|
||||
|
||||
@@ -61,17 +64,16 @@ data class Config(
|
||||
}
|
||||
LOCAL_DEVICE_NAME -> localDeviceName = reader.nextString()
|
||||
LOCAL_DEVICE_ID -> localDeviceId = reader.nextString()
|
||||
DISCOVERY_SERVERS -> {
|
||||
val newDiscoveryServers = HashSet<String>()
|
||||
|
||||
reader.beginArray()
|
||||
while (reader.hasNext()) {
|
||||
newDiscoveryServers.add(reader.nextString())
|
||||
CUSTOM_DISCOVERY_SERVERS -> {
|
||||
customDiscoveryServers = mutableSetOf<DiscoveryServer>().apply {
|
||||
reader.beginArray()
|
||||
while (reader.hasNext()) {
|
||||
add(DiscoveryServer.parse(reader))
|
||||
}
|
||||
reader.endArray()
|
||||
}
|
||||
reader.endArray()
|
||||
|
||||
discoveryServers = Collections.unmodifiableSet(newDiscoveryServers)
|
||||
}
|
||||
USE_DEFAULT_DISCOVERY_SERVERS -> useDefaultDiscoveryServers = reader.nextBoolean()
|
||||
KEYSTORE_ALGORITHM -> keystoreAlgorithm = reader.nextString()
|
||||
KEYSTORE_DATA -> keystoreData = reader.nextString()
|
||||
else -> reader.skipValue()
|
||||
@@ -84,7 +86,8 @@ data class Config(
|
||||
folders = folders!!,
|
||||
localDeviceName = localDeviceName!!,
|
||||
localDeviceId = localDeviceId!!,
|
||||
discoveryServers = discoveryServers!!,
|
||||
customDiscoveryServers = customDiscoveryServers,
|
||||
useDefaultDiscoveryServers = useDefaultDiscoveryServers,
|
||||
keystoreAlgorithm = keystoreAlgorithm!!,
|
||||
keystoreData = keystoreData!!
|
||||
)
|
||||
@@ -105,10 +108,12 @@ data class Config(
|
||||
writer.name(LOCAL_DEVICE_NAME).value(localDeviceName)
|
||||
writer.name(LOCAL_DEVICE_ID).value(localDeviceId)
|
||||
|
||||
writer.name(DISCOVERY_SERVERS).beginArray()
|
||||
discoveryServers.forEach { writer.value(it) }
|
||||
writer.name(CUSTOM_DISCOVERY_SERVERS).beginArray()
|
||||
customDiscoveryServers.forEach { it.serialize(writer) }
|
||||
writer.endArray()
|
||||
|
||||
writer.name(USE_DEFAULT_DISCOVERY_SERVERS).value(useDefaultDiscoveryServers)
|
||||
|
||||
writer.name(KEYSTORE_ALGORITHM).value(keystoreAlgorithm)
|
||||
writer.name(KEYSTORE_DATA).value(keystoreData)
|
||||
|
||||
@@ -117,5 +122,6 @@ data class Config(
|
||||
|
||||
// Exclude keystoreData from toString()
|
||||
override fun toString() = "Config(peers=$peers, folders=$folders, localDeviceName=$localDeviceName, " +
|
||||
"localDeviceId=$localDeviceId, discoveryServers=$discoveryServers, keystoreAlgorithm=$keystoreAlgorithm)"
|
||||
"localDeviceId=$localDeviceId, customDiscoveryServers=$customDiscoveryServers, " +
|
||||
"useDefaultDiscoveryServers=$useDefaultDiscoveryServers, keystoreAlgorithm=$keystoreAlgorithm)"
|
||||
}
|
||||
|
||||
+6
-22
@@ -38,26 +38,15 @@ class Configuration(configFolder: File = DefaultConfigFolder) {
|
||||
isSaved = false
|
||||
config = Config(peers = setOf(), folders = setOf(),
|
||||
localDeviceName = localDeviceName,
|
||||
discoveryServers = Companion.DiscoveryServers,
|
||||
localDeviceId = keystoreData.first.deviceId,
|
||||
keystoreData = Base64.toBase64String(keystoreData.second),
|
||||
keystoreAlgorithm = keystoreData.third)
|
||||
keystoreAlgorithm = keystoreData.third,
|
||||
customDiscoveryServers = emptySet(),
|
||||
useDefaultDiscoveryServers = true
|
||||
)
|
||||
persistNow()
|
||||
} else {
|
||||
config = Config.parse(JsonReader(StringReader(configFile.readText())))
|
||||
|
||||
// automatic migration if the old config was used
|
||||
if (config.discoveryServers == OldDiscoveryServers) {
|
||||
config = Config(
|
||||
peers = config.peers,
|
||||
folders = config.folders,
|
||||
localDeviceName = config.localDeviceName,
|
||||
localDeviceId = config.localDeviceId,
|
||||
discoveryServers = Companion.DiscoveryServers,
|
||||
keystoreAlgorithm = config.keystoreAlgorithm,
|
||||
keystoreData = config.keystoreData
|
||||
)
|
||||
}
|
||||
}
|
||||
logger.debug("Loaded config = $config")
|
||||
}
|
||||
@@ -66,11 +55,6 @@ class Configuration(configFolder: File = DefaultConfigFolder) {
|
||||
private val DefaultConfigFolder = File(System.getProperty("user.home"), ".config/syncthing-java/")
|
||||
private const val ConfigFileName = "config.json"
|
||||
private const val DatabaseFolderName = "database"
|
||||
private val DiscoveryServers = setOf(
|
||||
"discovery.syncthing.net", "discovery-v4.syncthing.net", "discovery-v6.syncthing.net")
|
||||
private val OldDiscoveryServers = setOf(
|
||||
"discovery-v4-1.syncthing.net", "discovery-v4-2.syncthing.net", "discovery-v4-3.syncthing.net",
|
||||
"discovery-v6-1.syncthing.net", "discovery-v6-2.syncthing.net", "discovery-v6-3.syncthing.net")
|
||||
}
|
||||
|
||||
val instanceId = Math.abs(Random().nextLong())
|
||||
@@ -78,8 +62,8 @@ class Configuration(configFolder: File = DefaultConfigFolder) {
|
||||
val localDeviceId: DeviceId
|
||||
get() = DeviceId(config.localDeviceId)
|
||||
|
||||
val discoveryServers: Set<String>
|
||||
get() = config.discoveryServers
|
||||
val discoveryServers: Set<DiscoveryServer>
|
||||
get() = config.customDiscoveryServers + (if (config.useDefaultDiscoveryServers) DiscoveryServer.defaultDiscoveryServers else emptySet())
|
||||
|
||||
val keystoreData: ByteArray
|
||||
get() = Base64.decode(config.keystoreData)
|
||||
|
||||
+86
@@ -0,0 +1,86 @@
|
||||
package net.syncthing.java.core.configuration
|
||||
|
||||
import com.google.gson.stream.JsonReader
|
||||
import com.google.gson.stream.JsonWriter
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
|
||||
data class DiscoveryServer(
|
||||
val hostname: String,
|
||||
val useForLookup: Boolean,
|
||||
val useForAnnounce: Boolean,
|
||||
val deviceId: DeviceId?
|
||||
) {
|
||||
companion object {
|
||||
private const val JSON_HOSTNAME = "host"
|
||||
private const val JSON_LOOKUP = "lookup"
|
||||
private const val JSON_ANNOUNCE = "announce"
|
||||
private const val JSON_DEVICE_ID = "deviceId"
|
||||
|
||||
// from https://github.com/syncthing/syncthing/blob/add12b43aa0bdf5e67d8f421c57a5ecafb3d25fa/lib/config/config.go#L50-L64
|
||||
// (if you update it, use the most recent commit)
|
||||
private val serverDeviceId = DeviceId("LYXKCHX-VI3NYZR-ALCJBHF-WMZYSPK-QG6QJA3-MPFYMSO-U56GTUK-NA2MIAW")
|
||||
|
||||
private val lookupServer = DiscoveryServer(
|
||||
hostname = "discovery.syncthing.net",
|
||||
useForLookup = true,
|
||||
useForAnnounce = false,
|
||||
deviceId = serverDeviceId
|
||||
)
|
||||
|
||||
private val announceIpV4Server = DiscoveryServer(
|
||||
hostname = "discovery-v4.syncthing.net",
|
||||
useForLookup = false,
|
||||
useForAnnounce = true,
|
||||
deviceId = serverDeviceId
|
||||
)
|
||||
|
||||
private val announceIpV6Server = DiscoveryServer(
|
||||
hostname = "discovery-v6.syncthing.net",
|
||||
useForLookup = false,
|
||||
useForAnnounce = true,
|
||||
deviceId = serverDeviceId
|
||||
)
|
||||
|
||||
val defaultDiscoveryServers = setOf(lookupServer, announceIpV4Server, announceIpV6Server)
|
||||
|
||||
fun parse(reader: JsonReader): DiscoveryServer {
|
||||
var hostname: String? = null
|
||||
var useForLookup: Boolean? = null
|
||||
var useForAnnounce: Boolean? = null
|
||||
var deviceId: DeviceId? = null
|
||||
|
||||
reader.beginObject()
|
||||
while (reader.hasNext()) {
|
||||
when (reader.nextName()) {
|
||||
JSON_HOSTNAME -> hostname = reader.nextString()
|
||||
JSON_LOOKUP -> useForLookup = reader.nextBoolean()
|
||||
JSON_ANNOUNCE -> useForAnnounce = reader.nextBoolean()
|
||||
JSON_DEVICE_ID -> deviceId = DeviceId(reader.nextString())
|
||||
else -> reader.skipValue()
|
||||
}
|
||||
}
|
||||
reader.endObject()
|
||||
|
||||
return DiscoveryServer(
|
||||
hostname = hostname!!,
|
||||
useForLookup = useForLookup!!,
|
||||
useForAnnounce = useForAnnounce!!,
|
||||
deviceId = deviceId
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
fun serialize(writer: JsonWriter) {
|
||||
writer.beginObject()
|
||||
|
||||
writer.name(JSON_HOSTNAME).value(hostname)
|
||||
writer.name(JSON_LOOKUP).value(useForLookup)
|
||||
writer.name(JSON_ANNOUNCE).value(useForAnnounce)
|
||||
|
||||
if (deviceId != null) {
|
||||
writer.name(JSON_DEVICE_ID).value(deviceId.deviceId)
|
||||
}
|
||||
|
||||
writer.endObject()
|
||||
}
|
||||
}
|
||||
+32
-21
@@ -44,8 +44,6 @@ import javax.security.auth.x500.X500Principal
|
||||
|
||||
class KeystoreHandler private constructor(private val keyStore: KeyStore) {
|
||||
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
|
||||
class CryptoException internal constructor(t: Throwable) : GeneralSecurityException(t)
|
||||
|
||||
private val socketFactory: SSLSocketFactory
|
||||
@@ -79,7 +77,7 @@ class KeystoreHandler private constructor(private val keyStore: KeyStore) {
|
||||
}
|
||||
|
||||
@Throws(CryptoException::class, IOException::class)
|
||||
private fun wrapSocket(socket: Socket, isServerSocket: Boolean, protocol: String): SSLSocket {
|
||||
private fun wrapSocket(socket: Socket, isServerSocket: Boolean): SSLSocket {
|
||||
try {
|
||||
logger.debug("wrapping plain socket, server mode = {}", isServerSocket)
|
||||
val sslSocket = socketFactory.createSocket(socket, null, socket.port, true) as SSLSocket
|
||||
@@ -100,7 +98,7 @@ class KeystoreHandler private constructor(private val keyStore: KeyStore) {
|
||||
}
|
||||
|
||||
@Throws(CryptoException::class, IOException::class)
|
||||
fun createSocket(relaySocketAddress: InetSocketAddress, protocol: String): SSLSocket {
|
||||
fun createSocket(relaySocketAddress: InetSocketAddress): SSLSocket {
|
||||
try {
|
||||
val socket = socketFactory.createSocket() as SSLSocket
|
||||
socket.connect(relaySocketAddress, SOCKET_TIMEOUT)
|
||||
@@ -116,24 +114,9 @@ class KeystoreHandler private constructor(private val keyStore: KeyStore) {
|
||||
}
|
||||
}
|
||||
|
||||
@Throws(SSLPeerUnverifiedException::class, CertificateException::class)
|
||||
fun checkSocketCertificate(socket: SSLSocket, deviceId: DeviceId) {
|
||||
val session = socket.session
|
||||
val certs = session.peerCertificates.toList()
|
||||
val certificateFactory = CertificateFactory.getInstance("X.509")
|
||||
val certPath = certificateFactory.generateCertPath(certs)
|
||||
val certificate = certPath.certificates[0]
|
||||
NetworkUtils.assertProtocol(certificate is X509Certificate)
|
||||
val derData = certificate.encoded
|
||||
val deviceIdFromCertificate = derDataToDeviceId(derData)
|
||||
logger.trace("remote pem certificate =\n{}", derToPem(derData))
|
||||
NetworkUtils.assertProtocol(deviceIdFromCertificate == deviceId, {"device id mismatch! expected = $deviceId, got = $deviceIdFromCertificate"})
|
||||
logger.debug("remote ssl certificate match deviceId = {}", deviceId)
|
||||
}
|
||||
|
||||
@Throws(CryptoException::class, IOException::class)
|
||||
fun wrapSocket(relayConnection: RelayConnection, protocol: String): SSLSocket {
|
||||
return wrapSocket(relayConnection.getSocket(), relayConnection.isServerSocket(), protocol)
|
||||
fun wrapSocket(relayConnection: RelayConnection): SSLSocket {
|
||||
return wrapSocket(relayConnection.getSocket(), relayConnection.isServerSocket())
|
||||
}
|
||||
|
||||
class Loader {
|
||||
@@ -269,6 +252,34 @@ class KeystoreHandler private constructor(private val keyStore: KeyStore) {
|
||||
|
||||
const val BEP = "bep/1.0"
|
||||
const val RELAY = "bep-relay"
|
||||
|
||||
private val logger = LoggerFactory.getLogger(KeystoreHandler::class.java)
|
||||
|
||||
@Throws(SSLPeerUnverifiedException::class, CertificateException::class)
|
||||
fun assertSocketCertificateValid(socket: SSLSocket, deviceId: DeviceId) {
|
||||
val session = socket.session
|
||||
val certs = session.peerCertificates.toList()
|
||||
val certificateFactory = CertificateFactory.getInstance("X.509")
|
||||
val certPath = certificateFactory.generateCertPath(certs)
|
||||
val certificate = certPath.certificates[0]
|
||||
|
||||
assertSocketCertificateValid(certificate, deviceId)
|
||||
}
|
||||
|
||||
@Throws(SSLPeerUnverifiedException::class, CertificateException::class)
|
||||
fun assertSocketCertificateValid(certificate: Certificate, deviceId: DeviceId) {
|
||||
NetworkUtils.assertProtocol(certificate is X509Certificate)
|
||||
|
||||
val derData = certificate.encoded
|
||||
val deviceIdFromCertificate = derDataToDeviceId(derData)
|
||||
logger.trace("remote pem certificate =\n{}", derToPem(derData))
|
||||
|
||||
NetworkUtils.assertProtocol(deviceIdFromCertificate == deviceId) {
|
||||
"device id mismatch! expected = $deviceId, got = $deviceIdFromCertificate"
|
||||
}
|
||||
|
||||
logger.debug("remote ssl certificate match deviceId = {}", deviceId)
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -16,6 +16,7 @@ package net.syncthing.java.core.utils
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.util.concurrent.ExecutorService
|
||||
import java.util.concurrent.Future
|
||||
import java.util.concurrent.RejectedExecutionException
|
||||
import java.util.concurrent.TimeUnit
|
||||
|
||||
private val logger = LoggerFactory.getLogger(ExecutorService::class.java)
|
||||
@@ -45,3 +46,11 @@ fun <T> ExecutorService.submitLogging(runnable: () -> T): Future<T> {
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
fun ExecutorService.trySubmitLogging(runnable: Runnable) {
|
||||
try {
|
||||
submitLogging(runnable)
|
||||
} catch (ex: RejectedExecutionException) {
|
||||
logger.warn("could not submit task", ex)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
* Copyright (C) 2018 Jonas Lochmann
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
@@ -13,36 +14,90 @@
|
||||
*/
|
||||
package net.syncthing.java.core.utils
|
||||
|
||||
import org.apache.commons.io.FilenameUtils
|
||||
|
||||
object PathUtils {
|
||||
|
||||
val ROOT_PATH = ""
|
||||
val PATH_SEPARATOR = "/"
|
||||
val PARENT_PATH = ".."
|
||||
|
||||
private fun normalizePath(path: String): String {
|
||||
return FilenameUtils.normalizeNoEndSeparator(path, true).replaceFirst(("^" + PATH_SEPARATOR).toRegex(), "")
|
||||
}
|
||||
const val ROOT_PATH = ""
|
||||
const val PATH_SEPARATOR = "/"
|
||||
const val PATH_SEPARATOR_WIN = "\\"
|
||||
const val PARENT_PATH = ".."
|
||||
const val CURRENT_PATH = "."
|
||||
|
||||
fun isRoot(path: String): Boolean {
|
||||
return path.isEmpty()
|
||||
}
|
||||
|
||||
private fun containsRelativeElements(path: String): Boolean {
|
||||
val pathSegments = path.split(PATH_SEPARATOR)
|
||||
|
||||
return pathSegments.contains(PARENT_PATH) or pathSegments.contains(CURRENT_PATH)
|
||||
}
|
||||
|
||||
private fun isTrimmed(value: String) = value.trim() == value
|
||||
private fun containsWindowsPathSeparator(path: String) = path.contains(PATH_SEPARATOR_WIN)
|
||||
private fun startsWithPathSeperator(path: String) = path.startsWith(PATH_SEPARATOR)
|
||||
private fun isValidPath(path: String) = (!containsRelativeElements(path)) and
|
||||
(!containsWindowsPathSeparator(path)) and
|
||||
path.isNotEmpty() and
|
||||
(!startsWithPathSeperator(path)) and
|
||||
isTrimmed(path)
|
||||
|
||||
private fun containsPathSeparator(file: String) = file.contains(PATH_SEPARATOR) or file.contains(PATH_SEPARATOR_WIN)
|
||||
private fun isFilenameValid(file: String) = file.isNotBlank() and
|
||||
(!containsPathSeparator(file)) and
|
||||
isTrimmed(file)
|
||||
|
||||
private fun assertPathValid(path: String) {
|
||||
if (!isValidPath(path)) {
|
||||
throw IllegalArgumentException("provided path is invalid")
|
||||
}
|
||||
}
|
||||
|
||||
private fun assertFilenameValid(filename: String) {
|
||||
if (!isFilenameValid(filename)) {
|
||||
throw IllegalArgumentException("provided filename is invalid")
|
||||
}
|
||||
}
|
||||
|
||||
fun isParent(path: String): Boolean {
|
||||
return path == PARENT_PATH
|
||||
}
|
||||
|
||||
fun getParentPath(path: String): String {
|
||||
assert(!isRoot(path), {"cannot get parent of root path"})
|
||||
return normalizePath(path + PATH_SEPARATOR + PARENT_PATH)
|
||||
assertPathValid(path)
|
||||
|
||||
val pathWithoutSuffix = path.removeSuffix(PATH_SEPARATOR)
|
||||
val previousSeparator = pathWithoutSuffix.lastIndexOf(PATH_SEPARATOR)
|
||||
|
||||
return if (previousSeparator == -1) {
|
||||
ROOT_PATH
|
||||
} else {
|
||||
pathWithoutSuffix.substring(0, previousSeparator)
|
||||
}
|
||||
}
|
||||
|
||||
fun getFileName(path: String): String {
|
||||
return FilenameUtils.getName(path)
|
||||
if (path.isEmpty()) {
|
||||
// this is required for IndexHandler.ROOT_FILE_INFO
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
assertPathValid(path)
|
||||
|
||||
val pathWithoutSuffix = path.removeSuffix(PATH_SEPARATOR)
|
||||
val previousSeparator = pathWithoutSuffix.lastIndexOf(PATH_SEPARATOR)
|
||||
|
||||
return if (previousSeparator == -1) {
|
||||
// the file is in the root directory
|
||||
pathWithoutSuffix
|
||||
} else {
|
||||
pathWithoutSuffix.substring(previousSeparator + 1)
|
||||
}
|
||||
}
|
||||
|
||||
fun buildPath(dir: String, file: String): String {
|
||||
return normalizePath(dir + PATH_SEPARATOR + file)
|
||||
assertPathValid(dir)
|
||||
assertFilenameValid(file)
|
||||
|
||||
return dir.removeSuffix(PATH_SEPARATOR) + file
|
||||
}
|
||||
}
|
||||
|
||||
@@ -8,7 +8,7 @@ dependencies {
|
||||
compile "commons-cli:commons-cli:1.4"
|
||||
implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
|
||||
implementation "com.google.protobuf:protobuf-lite:$protobuf_lite_version"
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:0.30.2'
|
||||
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:1.0.0'
|
||||
}
|
||||
|
||||
run {
|
||||
|
||||
+4
-4
@@ -14,10 +14,10 @@
|
||||
*/
|
||||
package net.syncthing.java.discovery
|
||||
|
||||
import kotlinx.coroutines.experimental.CancellationException
|
||||
import kotlinx.coroutines.experimental.runBlocking
|
||||
import kotlinx.coroutines.experimental.selects.select
|
||||
import kotlinx.coroutines.experimental.withTimeout
|
||||
import kotlinx.coroutines.CancellationException
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import kotlinx.coroutines.selects.select
|
||||
import kotlinx.coroutines.withTimeout
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import org.slf4j.LoggerFactory
|
||||
|
||||
+14
-3
@@ -13,22 +13,33 @@
|
||||
*/
|
||||
package net.syncthing.java.discovery
|
||||
|
||||
import kotlinx.coroutines.experimental.channels.Channel
|
||||
import kotlinx.coroutines.experimental.channels.ReceiveChannel
|
||||
import kotlinx.coroutines.channels.Channel
|
||||
import kotlinx.coroutines.channels.ReceiveChannel
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
|
||||
class DeviceAddressesManager (val deviceId: DeviceId) {
|
||||
companion object {
|
||||
private const val MAX_ADDRESSES_PER_TYPE = 16
|
||||
}
|
||||
|
||||
private val lock = Object()
|
||||
private val deviceAddressesCache = mutableListOf<DeviceAddress>()
|
||||
private val listeners = mutableListOf<(DeviceAddress) -> Unit>()
|
||||
|
||||
fun putAddress(address: DeviceAddress) {
|
||||
if (address.deviceIdObject != deviceId) {
|
||||
if (address.deviceId != deviceId) {
|
||||
throw IllegalArgumentException()
|
||||
}
|
||||
|
||||
synchronized(lock) {
|
||||
val otherAddressesOfSameType = deviceAddressesCache.filter { it.type == address.type }
|
||||
|
||||
if (otherAddressesOfSameType.size == MAX_ADDRESSES_PER_TYPE) {
|
||||
// forget the oldest one of the same type
|
||||
deviceAddressesCache.remove(otherAddressesOfSameType.first())
|
||||
}
|
||||
|
||||
deviceAddressesCache.add(address)
|
||||
listeners.forEach { it(address) }
|
||||
}
|
||||
|
||||
+9
-7
@@ -14,8 +14,9 @@
|
||||
*/
|
||||
package net.syncthing.java.discovery
|
||||
|
||||
import kotlinx.coroutines.experimental.GlobalScope
|
||||
import kotlinx.coroutines.experimental.launch
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.channels.consumeEach
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
@@ -39,7 +40,7 @@ class DiscoveryHandler(private val configuration: Configuration) : Closeable {
|
||||
}, { deviceId ->
|
||||
onMessageFromUnknownDeviceListeners.forEach { listener -> listener(deviceId) }
|
||||
})
|
||||
private val devicesAddressesManager = DevicesAddressesManager()
|
||||
val devicesAddressesManager = DevicesAddressesManager()
|
||||
private var isClosed = false
|
||||
private val onMessageFromUnknownDeviceListeners = Collections.synchronizedSet(HashSet<(DeviceId) -> Unit>())
|
||||
|
||||
@@ -74,17 +75,18 @@ class DiscoveryHandler(private val configuration: Configuration) : Closeable {
|
||||
val peers = configuration.peerIds
|
||||
//do not process address already processed
|
||||
list.filter { deviceAddress ->
|
||||
!peers.contains(deviceAddress.deviceIdObject)
|
||||
!peers.contains(deviceAddress.deviceId)
|
||||
}
|
||||
|
||||
AddressRanker.pingAddresses(list)
|
||||
.forEach { putDeviceAddress(it) }
|
||||
AddressRanker.pingAddressesChannel(list).consumeEach {
|
||||
putDeviceAddress(it)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun putDeviceAddress(deviceAddress: DeviceAddress) {
|
||||
devicesAddressesManager.getDeviceAddressManager(
|
||||
deviceId = deviceAddress.deviceIdObject
|
||||
deviceId = deviceAddress.deviceId
|
||||
).putAddress(deviceAddress)
|
||||
}
|
||||
|
||||
|
||||
+15
-13
@@ -14,14 +14,14 @@
|
||||
*/
|
||||
package net.syncthing.java.discovery.protocol
|
||||
|
||||
import kotlinx.coroutines.experimental.GlobalScope
|
||||
import kotlinx.coroutines.experimental.async
|
||||
import kotlinx.coroutines.experimental.coroutineScope
|
||||
import kotlinx.coroutines.experimental.launch
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.async
|
||||
import kotlinx.coroutines.coroutineScope
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import net.syncthing.java.discovery.utils.AddressRanker
|
||||
import net.syncthing.java.core.configuration.DiscoveryServer
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.IOException
|
||||
|
||||
@@ -38,7 +38,7 @@ internal class GlobalDiscoveryHandler(private val configuration: Configuration)
|
||||
}
|
||||
|
||||
suspend fun query(deviceIds: Collection<DeviceId>): List<DeviceAddress> {
|
||||
val discoveryServers = pickAnnounceServers()
|
||||
val discoveryServers = getLookupServers()
|
||||
|
||||
return coroutineScope {
|
||||
deviceIds
|
||||
@@ -57,15 +57,13 @@ internal class GlobalDiscoveryHandler(private val configuration: Configuration)
|
||||
}
|
||||
|
||||
suspend fun query(deviceId: DeviceId) = queryAnnounceServers(
|
||||
servers = pickAnnounceServers(),
|
||||
servers = getLookupServers(),
|
||||
deviceId = deviceId
|
||||
)
|
||||
|
||||
suspend fun pickAnnounceServers() = AddressRanker
|
||||
.pingAddresses(configuration.discoveryServers.map { DeviceAddress(it, "tcp://$it:443") })
|
||||
.map { it.deviceId }
|
||||
fun getLookupServers() = configuration.discoveryServers.filter { it.useForLookup }
|
||||
|
||||
suspend fun queryAnnounceServers(servers: List<String>, deviceId: DeviceId) = coroutineScope {
|
||||
suspend fun queryAnnounceServers(servers: List<DiscoveryServer>, deviceId: DeviceId) = coroutineScope {
|
||||
servers
|
||||
.map { server ->
|
||||
async {
|
||||
@@ -91,9 +89,13 @@ internal class GlobalDiscoveryHandler(private val configuration: Configuration)
|
||||
}
|
||||
|
||||
companion object {
|
||||
suspend fun queryAnnounceServer(server: String, deviceId: DeviceId) =
|
||||
suspend fun queryAnnounceServer(server: DiscoveryServer, deviceId: DeviceId) =
|
||||
GlobalDiscoveryUtil
|
||||
.queryAnnounceServer(server, deviceId)
|
||||
.queryAnnounceServer(
|
||||
server = server.hostname,
|
||||
requestedDeviceId = deviceId,
|
||||
serverDeviceId = server.deviceId
|
||||
)
|
||||
.addresses.map { DeviceAddress(deviceId.deviceId, it) }
|
||||
}
|
||||
}
|
||||
|
||||
+29
-5
@@ -15,15 +15,17 @@
|
||||
package net.syncthing.java.discovery.protocol
|
||||
|
||||
import com.google.gson.stream.JsonReader
|
||||
import kotlinx.coroutines.experimental.Dispatchers
|
||||
import kotlinx.coroutines.experimental.withContext
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.withContext
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.security.KeystoreHandler
|
||||
import java.io.BufferedInputStream
|
||||
import java.io.IOException
|
||||
import java.io.InputStreamReader
|
||||
import java.net.HttpURLConnection
|
||||
import java.net.URL
|
||||
import java.security.SecureRandom
|
||||
import java.security.cert.CertificateException
|
||||
import java.security.cert.X509Certificate
|
||||
import javax.net.ssl.HostnameVerifier
|
||||
import javax.net.ssl.HttpsURLConnection
|
||||
@@ -34,11 +36,33 @@ object GlobalDiscoveryUtil {
|
||||
private fun queryAnnounceServerUrl(server: String, deviceId: DeviceId) =
|
||||
"https://$server/v2/?device=${deviceId.deviceId}"
|
||||
|
||||
suspend fun queryAnnounceServer(server: String, deviceId: DeviceId): AnnouncementMessage {
|
||||
suspend fun queryAnnounceServer(
|
||||
server: String,
|
||||
requestedDeviceId: DeviceId,
|
||||
serverDeviceId: DeviceId?
|
||||
): AnnouncementMessage {
|
||||
return withContext(Dispatchers.IO) {
|
||||
val url = URL(queryAnnounceServerUrl(server, deviceId))
|
||||
val url = URL(queryAnnounceServerUrl(server, requestedDeviceId))
|
||||
val connection = (url.openConnection() as HttpsURLConnection).apply {
|
||||
hostnameVerifier = HostnameVerifier { _, _ -> true }
|
||||
hostnameVerifier = HostnameVerifier { _, session ->
|
||||
try {
|
||||
if (serverDeviceId != null) {
|
||||
if (session.peerCertificates.isEmpty()) {
|
||||
throw IOException("no certificate found")
|
||||
}
|
||||
|
||||
KeystoreHandler.assertSocketCertificateValid(session.peerCertificates.first(), serverDeviceId)
|
||||
}
|
||||
|
||||
true
|
||||
} catch (ex: Exception) {
|
||||
when (ex) {
|
||||
is IOException -> false
|
||||
is CertificateException -> false
|
||||
else -> throw ex
|
||||
}
|
||||
}
|
||||
}
|
||||
sslSocketFactory = SSLContext.getInstance("SSL").apply {
|
||||
init(null, arrayOf(object: X509TrustManager {
|
||||
override fun checkClientTrusted(p0: Array<out X509Certificate>?, p1: String?) {
|
||||
|
||||
+4
-4
@@ -14,10 +14,10 @@
|
||||
*/
|
||||
package net.syncthing.java.discovery.protocol
|
||||
|
||||
import kotlinx.coroutines.experimental.GlobalScope
|
||||
import kotlinx.coroutines.experimental.Job
|
||||
import kotlinx.coroutines.experimental.channels.consumeEach
|
||||
import kotlinx.coroutines.experimental.launch
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.Job
|
||||
import kotlinx.coroutines.channels.consumeEach
|
||||
import kotlinx.coroutines.launch
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.configuration.Configuration
|
||||
import org.slf4j.LoggerFactory
|
||||
|
||||
+7
-7
@@ -15,11 +15,11 @@
|
||||
package net.syncthing.java.discovery.protocol
|
||||
|
||||
import com.google.protobuf.ByteString
|
||||
import kotlinx.coroutines.experimental.Dispatchers
|
||||
import kotlinx.coroutines.experimental.GlobalScope
|
||||
import kotlinx.coroutines.experimental.channels.ReceiveChannel
|
||||
import kotlinx.coroutines.experimental.channels.produce
|
||||
import kotlinx.coroutines.experimental.withContext
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.GlobalScope
|
||||
import kotlinx.coroutines.channels.ReceiveChannel
|
||||
import kotlinx.coroutines.channels.produce
|
||||
import kotlinx.coroutines.withContext
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceId
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
@@ -77,7 +77,7 @@ object LocalDiscoveryUtil {
|
||||
// discovery announcement is to be used.
|
||||
DeviceAddress.Builder()
|
||||
.setAddress(address.replaceFirst("tcp://(0.0.0.0|):".toRegex(), "tcp://$sourceAddress:"))
|
||||
.setDeviceId(deviceId.deviceId)
|
||||
.setDeviceId(deviceId)
|
||||
.setInstanceId(announce.instanceId)
|
||||
.setProducer(DeviceAddress.AddressProducer.LOCAL_DISCOVERY)
|
||||
.build()
|
||||
@@ -135,7 +135,7 @@ object LocalDiscoveryUtil {
|
||||
data class LocalDiscoveryMessage(val deviceId: DeviceId, val addresses: List<DeviceAddress>) {
|
||||
init {
|
||||
addresses.forEach { address ->
|
||||
if (address.deviceIdObject != deviceId) {
|
||||
if (address.deviceId != deviceId) {
|
||||
throw IllegalArgumentException()
|
||||
}
|
||||
}
|
||||
|
||||
+27
-27
@@ -14,7 +14,9 @@
|
||||
*/
|
||||
package net.syncthing.java.discovery.utils
|
||||
|
||||
import kotlinx.coroutines.experimental.*
|
||||
import kotlinx.coroutines.*
|
||||
import kotlinx.coroutines.channels.produce
|
||||
import kotlinx.coroutines.channels.toList
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceAddress.AddressType
|
||||
import org.slf4j.LoggerFactory
|
||||
@@ -26,49 +28,47 @@ object AddressRanker {
|
||||
private const val TCP_CONNECTION_TIMEOUT = 5000
|
||||
private val BASE_SCORE_MAP = mapOf(
|
||||
AddressType.TCP to 0,
|
||||
AddressType.RELAY to 2000,
|
||||
AddressType.HTTP_RELAY to 1000 * 2000,
|
||||
AddressType.HTTPS_RELAY to 1000 * 2000
|
||||
AddressType.RELAY to 2000
|
||||
)
|
||||
private val ACCEPTED_ADDRESS_TYPES = BASE_SCORE_MAP.keys
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
|
||||
suspend fun pingAddresses(sourceAddresses: List<DeviceAddress>) = coroutineScope {
|
||||
addHttpRelays(sourceAddresses)
|
||||
.filter { ACCEPTED_ADDRESS_TYPES.contains(it.getType()) }
|
||||
.toList() // the following should happen parallel
|
||||
.map {
|
||||
fun pingAddressesChannel(sourceAddresses: List<DeviceAddress>) = GlobalScope.produce<DeviceAddress> {
|
||||
sourceAddresses
|
||||
.filter { ACCEPTED_ADDRESS_TYPES.contains(it.type) }
|
||||
.toList()
|
||||
.map { address ->
|
||||
async {
|
||||
try {
|
||||
withTimeout(TCP_CONNECTION_TIMEOUT * 2L) {
|
||||
val addressWithScore = withTimeout(TCP_CONNECTION_TIMEOUT * 2L) {
|
||||
// this nested async ensures that cancelling/ the timeout has got an effect without delay
|
||||
GlobalScope.async (Dispatchers.IO) {
|
||||
pingAddressSync(it)
|
||||
pingAddressSync(address)
|
||||
}.await()
|
||||
}
|
||||
|
||||
if (addressWithScore != null) {
|
||||
send(addressWithScore)
|
||||
}
|
||||
} catch (ex: Exception) {
|
||||
logger.warn("Failed to ping device", ex)
|
||||
|
||||
null
|
||||
}
|
||||
|
||||
null
|
||||
}
|
||||
}
|
||||
.map { it.await() }
|
||||
.filterNotNull()
|
||||
.sortedBy { it.score }
|
||||
|
||||
close()
|
||||
}
|
||||
|
||||
private fun getHttpRelays(list: List<DeviceAddress>) = list
|
||||
.asSequence()
|
||||
.filter { address ->
|
||||
address.getType() == AddressType.RELAY && address.containsUriParamValue("httpUrl")
|
||||
}
|
||||
.map { address ->
|
||||
val httpUrl = address.getUriParam("httpUrl")
|
||||
address.copyBuilder().setAddress("relay-" + httpUrl!!).build()
|
||||
}
|
||||
|
||||
private fun addHttpRelays(list: List<DeviceAddress>) = getHttpRelays(list) + list
|
||||
@Deprecated(
|
||||
message = "This is slower than the version which returns the channel",
|
||||
replaceWith = ReplaceWith("pingAddressesChannel")
|
||||
)
|
||||
suspend fun pingAddressesReturnAllResultsAtOnce(sourceAddresses: List<DeviceAddress>) = pingAddressesChannel(sourceAddresses)
|
||||
.toList()
|
||||
.sortedBy { it.score }
|
||||
|
||||
private fun pingAddressSync(deviceAddress: DeviceAddress): DeviceAddress? {
|
||||
val startTime = System.currentTimeMillis()
|
||||
@@ -84,7 +84,7 @@ object AddressRanker {
|
||||
}
|
||||
|
||||
val ping = (System.currentTimeMillis() - startTime).toInt()
|
||||
val baseScore = BASE_SCORE_MAP[deviceAddress.getType()] ?: 0
|
||||
val baseScore = BASE_SCORE_MAP[deviceAddress.type] ?: 0
|
||||
|
||||
return deviceAddress.copyBuilder().setScore(ping + baseScore).build()
|
||||
}
|
||||
|
||||
@@ -1,37 +0,0 @@
|
||||
apply plugin: 'java-library'
|
||||
apply plugin: 'kotlin'
|
||||
apply plugin: 'com.google.protobuf'
|
||||
|
||||
dependencies {
|
||||
compile project(':syncthing-relay-client')
|
||||
implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
|
||||
implementation "com.google.protobuf:protobuf-lite:$protobuf_lite_version"
|
||||
}
|
||||
|
||||
protobuf {
|
||||
protoc {
|
||||
artifact = "com.google.protobuf:protoc:3.5.1-1"
|
||||
}
|
||||
plugins {
|
||||
javalite {
|
||||
// The codegen for lite comes as a separate artifact
|
||||
artifact = "com.google.protobuf:protoc-gen-javalite:3.0.0"
|
||||
}
|
||||
}
|
||||
generateProtoTasks {
|
||||
all().each { task ->
|
||||
task.builtins {
|
||||
// In most cases you don't need the full Java output
|
||||
// if you use the lite output.
|
||||
remove java
|
||||
}
|
||||
task.plugins {
|
||||
javalite { }
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Workaround for https://github.com/google/protobuf-gradle-plugin/issues/100
|
||||
compileKotlin.dependsOn('generateProto')
|
||||
sourceSets.main.kotlin.srcDirs += file("${protobuf.generatedFilesBaseDir}/main/javalite")
|
||||
-31
@@ -1,31 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.httprelay
|
||||
|
||||
import net.syncthing.java.core.beans.DeviceAddress
|
||||
import net.syncthing.java.core.beans.DeviceAddress.AddressType
|
||||
import org.slf4j.LoggerFactory
|
||||
|
||||
class HttpRelayClient {
|
||||
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
|
||||
fun openRelayConnection(deviceAddress: DeviceAddress): HttpRelayConnection {
|
||||
assert(setOf(AddressType.HTTP_RELAY, AddressType.HTTPS_RELAY).contains(deviceAddress.getType()))
|
||||
val httpRelayServerUrl = deviceAddress.address.replaceFirst("^relay-".toRegex(), "")
|
||||
val deviceId = deviceAddress.deviceId
|
||||
logger.info("open http relay connection, relay url = {}, target device id = {}", httpRelayServerUrl, deviceId)
|
||||
return HttpRelayConnection(httpRelayServerUrl, deviceId)
|
||||
}
|
||||
}
|
||||
-304
@@ -1,304 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2016 Davide Imbriaco
|
||||
*
|
||||
* This Java file is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package net.syncthing.java.httprelay
|
||||
|
||||
import com.google.protobuf.ByteString
|
||||
import net.syncthing.java.core.interfaces.RelayConnection
|
||||
import net.syncthing.java.core.utils.NetworkUtils
|
||||
import net.syncthing.java.core.utils.submitLogging
|
||||
import org.apache.http.HttpStatus
|
||||
import org.apache.http.client.methods.HttpPost
|
||||
import org.apache.http.entity.ByteArrayEntity
|
||||
import org.apache.http.impl.client.HttpClients
|
||||
import org.apache.http.util.EntityUtils
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.io.*
|
||||
import java.net.*
|
||||
import java.util.concurrent.ExecutionException
|
||||
import java.util.concurrent.Executors
|
||||
import java.util.concurrent.LinkedBlockingQueue
|
||||
import java.util.concurrent.TimeUnit
|
||||
|
||||
class HttpRelayConnection internal constructor(private val httpRelayServerUrl: String, deviceId: String) : RelayConnection, Closeable {
|
||||
|
||||
private val logger = LoggerFactory.getLogger(javaClass)
|
||||
private val outgoingExecutorService = Executors.newSingleThreadExecutor()
|
||||
private val incomingExecutorService = Executors.newSingleThreadExecutor()
|
||||
private val flusherStreamService = Executors.newSingleThreadScheduledExecutor()
|
||||
private var peerToRelaySequence: Long = 0
|
||||
private var relayToPeerSequence: Long = 0
|
||||
private val sessionId: String
|
||||
private val incomingDataQueue = LinkedBlockingQueue<ByteArray>()
|
||||
private val socket: Socket
|
||||
private val isServerSocket: Boolean
|
||||
private val inputStream: InputStream
|
||||
private val outputStream: OutputStream
|
||||
|
||||
var isClosed = false
|
||||
private set
|
||||
|
||||
override fun getSocket() = socket
|
||||
|
||||
override fun isServerSocket() = isServerSocket
|
||||
|
||||
init {
|
||||
val serverMessage = sendMessage(HttpRelayProtos.HttpRelayPeerMessage.newBuilder()
|
||||
.setMessageType(HttpRelayProtos.HttpRelayPeerMessageType.CONNECT)
|
||||
.setDeviceId(deviceId))
|
||||
assert(serverMessage.messageType == HttpRelayProtos.HttpRelayServerMessageType.PEER_CONNECTED)
|
||||
assert(!serverMessage.sessionId.isNullOrEmpty())
|
||||
sessionId = serverMessage.sessionId
|
||||
isServerSocket = serverMessage.isServerSocket
|
||||
outputStream = object : OutputStream() {
|
||||
|
||||
private var buffer = ByteArrayOutputStream()
|
||||
private var lastFlush = System.currentTimeMillis()
|
||||
|
||||
init {
|
||||
flusherStreamService.scheduleWithFixedDelay({
|
||||
if (System.currentTimeMillis() - lastFlush > 1000) {
|
||||
try {
|
||||
flush()
|
||||
} catch (ex: IOException) {
|
||||
logger.warn("", ex)
|
||||
}
|
||||
|
||||
}
|
||||
}, 1, 1, TimeUnit.SECONDS)
|
||||
}
|
||||
|
||||
@Synchronized
|
||||
@Throws(IOException::class)
|
||||
override fun write(i: Int) {
|
||||
NetworkUtils.assertProtocol(!this@HttpRelayConnection.isClosed)
|
||||
buffer.write(i)
|
||||
}
|
||||
|
||||
@Synchronized
|
||||
@Throws(IOException::class)
|
||||
override fun write(bytes: ByteArray, offset: Int, size: Int) {
|
||||
NetworkUtils.assertProtocol(!this@HttpRelayConnection.isClosed)
|
||||
buffer.write(bytes, offset, size)
|
||||
}
|
||||
|
||||
@Synchronized
|
||||
@Throws(IOException::class)
|
||||
override fun flush() {
|
||||
val data = buffer.toByteArray().copyOf().toList()
|
||||
buffer = ByteArrayOutputStream()
|
||||
try {
|
||||
if (!data.isEmpty()) {
|
||||
outgoingExecutorService.submit {
|
||||
sendMessage(HttpRelayProtos.HttpRelayPeerMessage.newBuilder()
|
||||
.setMessageType(HttpRelayProtos.HttpRelayPeerMessageType.PEER_TO_RELAY)
|
||||
.setSequence(++peerToRelaySequence)
|
||||
.setData(data as ByteString))
|
||||
}.get()
|
||||
}
|
||||
lastFlush = System.currentTimeMillis()
|
||||
} catch (ex: InterruptedException) {
|
||||
logger.error("error", ex)
|
||||
closeBg()
|
||||
throw IOException(ex)
|
||||
} catch (ex: ExecutionException) {
|
||||
logger.error("error", ex)
|
||||
closeBg()
|
||||
throw IOException(ex)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Synchronized
|
||||
@Throws(IOException::class)
|
||||
override fun write(bytes: ByteArray) {
|
||||
NetworkUtils.assertProtocol(!this@HttpRelayConnection.isClosed)
|
||||
buffer.write(bytes)
|
||||
}
|
||||
|
||||
}
|
||||
incomingExecutorService.submitLogging {
|
||||
while (!isClosed) {
|
||||
val serverMessage1 =
|
||||
try {
|
||||
sendMessage(HttpRelayProtos.HttpRelayPeerMessage.newBuilder().setMessageType(HttpRelayProtos.HttpRelayPeerMessageType.WAIT_FOR_DATA))
|
||||
} catch (e: IOException) {
|
||||
logger.warn("Failed to send relay message", e)
|
||||
return@submitLogging
|
||||
}
|
||||
if (isClosed) {
|
||||
return@submitLogging
|
||||
}
|
||||
NetworkUtils.assertProtocol(serverMessage1.messageType == HttpRelayProtos.HttpRelayServerMessageType.RELAY_TO_PEER)
|
||||
NetworkUtils.assertProtocol(serverMessage1.sequence == relayToPeerSequence + 1)
|
||||
if (!serverMessage1.data.isEmpty) {
|
||||
incomingDataQueue.add(serverMessage1.data.toByteArray())
|
||||
}
|
||||
relayToPeerSequence = serverMessage1.sequence
|
||||
}
|
||||
}
|
||||
inputStream = object : InputStream() {
|
||||
|
||||
private var noMoreData = false
|
||||
private var byteArrayInputStream = ByteArrayInputStream(ByteArray(0))
|
||||
|
||||
@Throws(IOException::class)
|
||||
override fun read(): Int {
|
||||
NetworkUtils.assertProtocol(!this@HttpRelayConnection.isClosed)
|
||||
if (noMoreData) {
|
||||
return -1
|
||||
}
|
||||
var bite = -1
|
||||
while (bite == -1) {
|
||||
bite = byteArrayInputStream.read()
|
||||
try {
|
||||
val data = incomingDataQueue.poll(1, TimeUnit.SECONDS)
|
||||
if (data == null) {
|
||||
//continue
|
||||
} else if (data.contentEquals(STREAM_CLOSED)) {
|
||||
noMoreData = true
|
||||
return -1
|
||||
} else {
|
||||
byteArrayInputStream = ByteArrayInputStream(data)
|
||||
}
|
||||
} catch (ex: InterruptedException) {
|
||||
logger.warn("", ex)
|
||||
}
|
||||
|
||||
}
|
||||
return bite
|
||||
}
|
||||
|
||||
}
|
||||
socket = object : Socket() {
|
||||
override fun isClosed(): Boolean {
|
||||
return this@HttpRelayConnection.isClosed
|
||||
}
|
||||
|
||||
override fun isConnected(): Boolean {
|
||||
return !isClosed
|
||||
}
|
||||
|
||||
@Throws(IOException::class)
|
||||
override fun shutdownOutput() {
|
||||
logger.debug("shutdownOutput")
|
||||
outputStream.flush()
|
||||
}
|
||||
|
||||
@Throws(IOException::class)
|
||||
override fun shutdownInput() {
|
||||
logger.debug("shutdownInput")
|
||||
//do nothing
|
||||
}
|
||||
|
||||
@Synchronized
|
||||
@Throws(IOException::class)
|
||||
override fun close() {
|
||||
logger.debug("received close on socket adapter")
|
||||
this@HttpRelayConnection.close()
|
||||
}
|
||||
|
||||
@Throws(IOException::class)
|
||||
override fun getOutputStream(): OutputStream {
|
||||
return this@HttpRelayConnection.outputStream
|
||||
}
|
||||
|
||||
@Throws(IOException::class)
|
||||
override fun getInputStream(): InputStream {
|
||||
return this@HttpRelayConnection.inputStream
|
||||
}
|
||||
|
||||
@Throws(UnknownHostException::class)
|
||||
override fun getRemoteSocketAddress(): SocketAddress {
|
||||
return InetSocketAddress(inetAddress, port)
|
||||
}
|
||||
|
||||
override fun getPort(): Int {
|
||||
return 22067
|
||||
}
|
||||
|
||||
@Throws(UnknownHostException::class)
|
||||
override fun getInetAddress(): InetAddress {
|
||||
return InetAddress.getByName(URI.create(this@HttpRelayConnection.httpRelayServerUrl).host)
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
private fun closeBg() {
|
||||
|
||||
Thread { close() }.start()
|
||||
}
|
||||
|
||||
@Throws(IOException::class)
|
||||
private fun sendMessage(peerMessageBuilder: HttpRelayProtos.HttpRelayPeerMessage.Builder): HttpRelayProtos.HttpRelayServerMessage {
|
||||
if (!sessionId.isEmpty()) {
|
||||
peerMessageBuilder.sessionId = sessionId
|
||||
}
|
||||
logger.debug("send http relay peer message = {} session id = {} sequence = {}", peerMessageBuilder.messageType, peerMessageBuilder.sessionId, peerMessageBuilder.sequence)
|
||||
val httpClient = HttpClients.custom()
|
||||
// .setSSLSocketFactory(new SSLConnectionSocketFactory(new SSLContextBuilder().loadTrustMaterial(null, new TrustSelfSignedStrategy()).build(), SSLConnectionSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER))
|
||||
.build()
|
||||
val httpPost = HttpPost(httpRelayServerUrl)
|
||||
httpPost.entity = ByteArrayEntity(peerMessageBuilder.build().toByteArray())
|
||||
val serverMessage = httpClient.execute(httpPost) { response ->
|
||||
NetworkUtils.assertProtocol(response.statusLine.statusCode == HttpStatus.SC_OK, {"http error ${response.statusLine}"})
|
||||
HttpRelayProtos.HttpRelayServerMessage.parseFrom(EntityUtils.toByteArray(response.entity))
|
||||
}
|
||||
logger.debug("received http relay server message = {}", serverMessage.messageType)
|
||||
NetworkUtils.assertProtocol(serverMessage.messageType != HttpRelayProtos.HttpRelayServerMessageType.ERROR, {"server error : ${serverMessage.data.toStringUtf8()}"})
|
||||
return serverMessage
|
||||
}
|
||||
|
||||
override fun close() {
|
||||
if (!isClosed) {
|
||||
isClosed = true
|
||||
logger.info("closing http relay connection {} : {}", httpRelayServerUrl, sessionId)
|
||||
flusherStreamService.shutdown()
|
||||
if (!sessionId.isEmpty()) {
|
||||
try {
|
||||
outputStream.flush()
|
||||
sendMessage(HttpRelayProtos.HttpRelayPeerMessage.newBuilder().setMessageType(HttpRelayProtos.HttpRelayPeerMessageType.PEER_CLOSING))
|
||||
} catch (ex: IOException) {
|
||||
logger.warn("error closing http relay connection", ex)
|
||||
}
|
||||
|
||||
}
|
||||
incomingExecutorService.shutdown()
|
||||
outgoingExecutorService.shutdown()
|
||||
try {
|
||||
incomingExecutorService.awaitTermination(1, TimeUnit.SECONDS)
|
||||
} catch (ex: InterruptedException) {
|
||||
logger.warn("", ex)
|
||||
}
|
||||
|
||||
try {
|
||||
outgoingExecutorService.awaitTermination(1, TimeUnit.SECONDS)
|
||||
} catch (ex: InterruptedException) {
|
||||
logger.warn("", ex)
|
||||
}
|
||||
|
||||
try {
|
||||
flusherStreamService.awaitTermination(1, TimeUnit.SECONDS)
|
||||
} catch (ex: InterruptedException) {
|
||||
logger.warn("", ex)
|
||||
}
|
||||
|
||||
incomingDataQueue.add(STREAM_CLOSED)
|
||||
}
|
||||
}
|
||||
|
||||
companion object {
|
||||
private val STREAM_CLOSED = "STREAM_CLOSED".toByteArray()
|
||||
}
|
||||
}
|
||||
@@ -1,34 +0,0 @@
|
||||
package net.syncthing.java.httprelay;
|
||||
|
||||
option optimize_for = LITE_RUNTIME;
|
||||
|
||||
message HttpRelayPeerMessage{
|
||||
optional HttpRelayPeerMessageType message_type = 1;
|
||||
optional string session_id = 2;
|
||||
optional string device_id = 3;
|
||||
optional int64 sequence = 4;
|
||||
optional bytes data = 5;
|
||||
}
|
||||
|
||||
message HttpRelayServerMessage{
|
||||
optional HttpRelayServerMessageType message_type = 1;
|
||||
optional string session_id = 2;
|
||||
optional bool is_server_socket = 3;
|
||||
optional int64 sequence = 4;
|
||||
optional bytes data = 5;
|
||||
}
|
||||
|
||||
enum HttpRelayPeerMessageType {
|
||||
CONNECT = 0;
|
||||
PEER_TO_RELAY = 1;
|
||||
WAIT_FOR_DATA = 2;
|
||||
PEER_CLOSING = 3;
|
||||
}
|
||||
|
||||
enum HttpRelayServerMessageType {
|
||||
PEER_CONNECTED = 0;
|
||||
DATA_ACCEPTED = 1;
|
||||
RELAY_TO_PEER = 2;
|
||||
SERVER_CLOSING = 3;
|
||||
ERROR = 4;
|
||||
}
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user