It looks like a solution:

        try {
            getDropTarget().addDropTargetListener(new DropTargetAdapter() {
                // save restore need to be done only for a particular dnd
                public void dragEnter(DropTargetDragEvent dtde) {
                    dragInProgress = true;
                    selectedPaths = getSelectionPaths();
                }
                public void dragExit(DropTargetEvent dte) {
                    dragInProgress = false;
                }
                public void drop(DropTargetDropEvent dtde) {
                    dragInProgress = false;
                    setSelectionPaths(selectedPaths);
                }
                TreePath[] selectedPaths;
                });
        } catch (TooManyListenersException tmle) {
            // should not happen... swing drop target is multicast
        }